main.c File Reference

#include "net.h"

Include dependency graph for main.c:

Go to the source code of this file.

Classes
struct	_COMMAND

Macros
#define	MAX_BUFFER_SIZE   4096

Typedefs
typedef struct _COMMAND	COMMAND
typedef struct _COMMAND *	PCOMMAND

Functions
VOID	PrintPaddedResourceString (UINT uID, DWORD nPaddedLength)
VOID	PrintPadding (WCHAR chr, DWORD nPaddedLength)
DWORD	TranslateAppMessage (DWORD dwMessage)
VOID	PrintMessageStringV (DWORD dwMessage,...)
VOID	PrintMessageString (DWORD dwMessage)
VOID	PrintPaddedMessageString (DWORD dwMessage, DWORD nPaddedLength)
VOID	PrintErrorMessage (DWORD dwError)
VOID	PrintNetMessage (DWORD dwMessage)
VOID	ReadFromConsole (LPWSTR lpInput, DWORD dwLength, BOOL bEcho)
int	wmain (int argc, WCHAR **argv)
INT	unimplemented (INT argc, WCHAR **argv)

Variables
COMMAND	cmds []
HMODULE	hModuleNetMsg = NULL

Macro Definition Documentation

MAX_BUFFER_SIZE

#define MAX_BUFFER_SIZE   4096

Definition at line 12 of file main.c.

Typedef Documentation

COMMAND

typedef struct _COMMAND COMMAND

PCOMMAND

typedef struct _COMMAND * PCOMMAND

Function Documentation

PrintErrorMessage()

VOID PrintErrorMessage

(

DWORD

dwError

)

Definition at line 140 of file main.c.

{
    WCHAR szErrorBuffer[16];
    PWSTR pBuffer;
    PWSTR pErrorInserts[2] = {NULL, NULL};
 
    if (dwError >= MIN_LANMAN_MESSAGE_ID && dwError <= MAX_LANMAN_MESSAGE_ID)
    {
        FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
                       FORMAT_MESSAGE_IGNORE_INSERTS,
                       hModuleNetMsg,
                       dwError,
                       LANG_USER_DEFAULT,
                       (LPWSTR)&pBuffer,
                       0,
                       NULL);
        if (pBuffer)
        {
            ConPrintf(StdErr, L"%s\n", pBuffer);
            LocalFree(pBuffer);
            pBuffer = NULL;
        }
    }
    else
    {
        FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
                       FORMAT_MESSAGE_IGNORE_INSERTS,
                       NULL,
                       dwError,
                       LANG_USER_DEFAULT,
                       (LPWSTR)&pBuffer,
                       0,
                       NULL);
        if (pBuffer)
        {
            ConPrintf(StdErr, L"%s\n", pBuffer);
            LocalFree(pBuffer);
            pBuffer = NULL;
        }
    }
 
    if (dwError != ERROR_SUCCESS)
    {
        /* Format insert for the 3514 message */
        swprintf(szErrorBuffer, L"%lu", dwError);
        pErrorInserts[0] = szErrorBuffer;
 
        /* Format and print the 3514 message */
        FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
                       FORMAT_MESSAGE_ARGUMENT_ARRAY,
                       hModuleNetMsg,
                       3514,
                       LANG_USER_DEFAULT,
                       (LPWSTR)&pBuffer,
                       0,
                       (va_list *)pErrorInserts);
        if (pBuffer)
        {
            ConPrintf(StdErr, L"%s\n", pBuffer);
            LocalFree(pBuffer);
            pBuffer = NULL;
        }
    }
}

PrintMessageString()

VOID PrintMessageString

(

DWORD

dwMessage

)

Definition at line 110 of file main.c.

{
    ConMsgPuts(StdOut,
               FORMAT_MESSAGE_FROM_HMODULE,
               hModuleNetMsg,
               dwMessage,
               LANG_USER_DEFAULT);
}

Referenced by cmdAccounts(), cmdComputer(), cmdConfig(), cmdContinue(), cmdGroup(), cmdHelp(), cmdHelpMsg(), cmdLocalGroup(), cmdPause(), cmdSession(), cmdShare(), cmdStart(), cmdStatistics(), cmdStop(), cmdSyntax(), cmdUser(), DisplayGroup(), DisplayLocalGroup(), DisplayServerConfig(), DisplayServerStatistics(), DisplayShare(), DisplayUser(), DisplayWorkstationStatistics(), EnumerateShares(), EnumSessions(), PrintLogonHours(), ReadPassword(), and wmain().

PrintMessageStringV()

VOID PrintMessageStringV	(	DWORD	dwMessage,
			...
	)

Definition at line 93 of file main.c.

{
    va_list args;
 
    va_start(args, dwMessage);
    ConMsgPrintfV(StdOut,
                  FORMAT_MESSAGE_FROM_HMODULE,
                  hModuleNetMsg,
                  dwMessage,
                  LANG_USER_DEFAULT,
                  &args);
    va_end(args);
}

Referenced by cmdAccounts(), cmdConfig(), cmdUse(), cmdUser(), DisplayServerStatistics(), DisplayWorkstationStatistics(), EnumerateGroups(), EnumerateLocalGroups(), EnumerateUsers(), and PrintError().

PrintNetMessage()

VOID PrintNetMessage

(

DWORD

dwMessage

)

Definition at line 208 of file main.c.

{
    PWSTR pBuffer;
 
    FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
                   FORMAT_MESSAGE_FROM_HMODULE |
                   FORMAT_MESSAGE_IGNORE_INSERTS,
                   GetModuleHandleW(NULL),
                   dwMessage,
                   LANG_USER_DEFAULT,
                   (LPWSTR)&pBuffer,
                   0,
                   NULL);
    if (pBuffer)
    {
        ConPuts(StdOut, pBuffer);
        ConPuts(StdOut, L"\n");
        LocalFree(pBuffer);
        pBuffer = NULL;
    }
}

Referenced by cmdAccounts(), cmdComputer(), cmdConfig(), cmdContinue(), cmdGroup(), cmdHelp(), cmdHelpMsg(), cmdLocalGroup(), cmdPause(), cmdSession(), cmdShare(), cmdStart(), cmdStatistics(), cmdStop(), cmdSyntax(), cmdUser(), and wmain().

PrintPaddedMessageString()

VOID PrintPaddedMessageString	(	DWORD	dwMessage,
		DWORD	nPaddedLength
	)

Definition at line 122 of file main.c.

{
    DWORD dwLength;
 
    dwLength = (DWORD)ConMsgPuts(StdOut,
                                 FORMAT_MESSAGE_FROM_HMODULE,
                                 hModuleNetMsg,
                                 dwMessage,
                                 LANG_USER_DEFAULT);
 
    if (dwLength < nPaddedLength)
        PrintPadding(L' ', nPaddedLength - dwLength);
}

Referenced by cmdAccounts(), DisplayGroup(), DisplayLocalGroup(), DisplayServerConfig(), DisplayServerStatistics(), DisplayShare(), DisplayUser(), DisplayWorkstationConfig(), and DisplayWorkstationStatistics().

PrintPaddedResourceString()

VOID PrintPaddedResourceString	(	UINT	uID,
		DWORD	nPaddedLength
	)

Definition at line 50 of file main.c.

{
    DWORD dwLength;
 
    dwLength = (DWORD)ConResPuts(StdOut, uID);
    if (dwLength < nPaddedLength)
        PrintPadding(L' ', nPaddedLength - dwLength);
}

PrintPadding()

VOID PrintPadding	(	WCHAR	chr,
		DWORD	nPaddedLength
	)

Definition at line 63 of file main.c.

{
    DWORD i;
    WCHAR szMsgBuffer[MAX_BUFFER_SIZE];
 
    for (i = 0; i < nPaddedLength; i++)
         szMsgBuffer[i] = chr;
    szMsgBuffer[nPaddedLength] = UNICODE_NULL;
 
    ConPuts(StdOut, szMsgBuffer);
}

Referenced by DisplayGroup(), DisplayLocalGroup(), DisplayUser(), EnumerateGroups(), EnumerateLocalGroups(), EnumerateShares(), EnumerateUsers(), EnumSessions(), PrintLogonHours(), PrintPaddedMessageString(), and PrintPaddedResourceString().

ReadFromConsole()

VOID ReadFromConsole	(	LPWSTR	lpInput,
		DWORD	dwLength,
		BOOL	bEcho
	)

Definition at line 233 of file main.c.

{
    DWORD dwOldMode;
    DWORD dwRead = 0;
    HANDLE hFile;
    LPWSTR p;
    PCHAR pBuf;
 
    pBuf = HeapAlloc(GetProcessHeap(), 0, dwLength - 1);
    ZeroMemory(lpInput, dwLength * sizeof(WCHAR));
    hFile = GetStdHandle(STD_INPUT_HANDLE);
    GetConsoleMode(hFile, &dwOldMode);
 
    SetConsoleMode(hFile, ENABLE_LINE_INPUT | (bEcho ? ENABLE_ECHO_INPUT : 0));
 
    ReadFile(hFile, (PVOID)pBuf, dwLength - 1, &dwRead, NULL);
 
    MultiByteToWideChar(CP_OEMCP, 0, pBuf, dwRead, lpInput, dwLength - 1);
    HeapFree(GetProcessHeap(), 0, pBuf);
 
    for (p = lpInput; *p; p++)
    {
        if (*p == L'\x0d')
        {
            *p = L'\0';
            break;
        }
    }
 
    SetConsoleMode(hFile, dwOldMode);
}

Referenced by ReadPassword().

TranslateAppMessage()

DWORD TranslateAppMessage

(

DWORD

dwMessage

)

Definition at line 79 of file main.c.

{
    switch (dwMessage)
    {
        case NERR_Success:
            return 3500; // APPERR_3500
        case ERROR_MORE_DATA:
            return 3513; // APPERR_3513
    }
    return dwMessage;
}

Referenced by cmdUser().

unimplemented()

INT unimplemented	(	INT	argc,
		WCHAR **	argv
	)

Definition at line 321 of file main.c.

{
    ConPuts(StdOut, L"This command is not implemented yet\n");
    return 1;
}

wmain()

int wmain	(	int	argc,
		WCHAR **	argv
	)

Definition at line 269 of file main.c.

{
    WCHAR szDllBuffer[MAX_PATH];
    PCOMMAND cmdptr;
    int nResult = 0;
    BOOL bRun = FALSE;
 
    /* Initialize the Console Standard Streams */
    ConInitStdStreams();
 
    /* Load netmsg.dll */
    GetSystemDirectoryW(szDllBuffer, ARRAYSIZE(szDllBuffer));
    wcscat(szDllBuffer, L"\\netmsg.dll");
 
    hModuleNetMsg = LoadLibrary(szDllBuffer);
    if (hModuleNetMsg == NULL)
    {
        ConPrintf(StdErr, L"Failed to load netmsg.dll\n");
        return 1;
    }
 
    if (argc < 2)
    {
        nResult = 1;
        goto done;
    }
 
    /* Scan the command table */
    for (cmdptr = cmds; cmdptr->name; cmdptr++)
    {
        if (_wcsicmp(argv[1], cmdptr->name) == 0)
        {
            nResult = cmdptr->func(argc, argv);
            bRun = TRUE;
            break;
        }
    }
 
done:
    if (bRun == FALSE)
    {
        PrintMessageString(4381);
        ConPuts(StdOut, L"\n");
        PrintNetMessage(MSG_NET_SYNTAX);
    }
 
    if (hModuleNetMsg != NULL)
        FreeLibrary(hModuleNetMsg);
 
    return nResult;
}

Variable Documentation

cmds

COMMAND cmds[]

Initial value:

=
{
    {L"accounts",   cmdAccounts},
    {L"computer",   cmdComputer},
    {L"config",     cmdConfig},
    {L"continue",   cmdContinue},
    {L"file",       unimplemented},
    {L"group",      cmdGroup},
    {L"help",       cmdHelp},
    {L"helpmsg",    cmdHelpMsg},
    {L"localgroup", cmdLocalGroup},
    {L"pause",      cmdPause},
    {L"session",    cmdSession},
    {L"share",      cmdShare},
    {L"start",      cmdStart},
    {L"statistics", cmdStatistics},
    {L"stop",       cmdStop},
    {L"syntax",     cmdSyntax},
    {L"time",       unimplemented},
    {L"use",        cmdUse},
    {L"user",       cmdUser},
    {L"view",       unimplemented},
    {NULL,          NULL}
}

Definition at line 21 of file main.c.

Referenced by DoCommand(), handle_arrow_key(), handle_edit_event(), help_main(), HelpCommand(), HelpCommandList(), TEST_IsQSForward(), TEST_MayQSForwardMockup(), test_OleCommandTarget(), and wmain().

hModuleNetMsg

HMODULE hModuleNetMsg = NULL

Definition at line 46 of file main.c.

Referenced by cmdHelpMsg(), GetCountryFromCountryCode(), ParseLogonHours(), PrintErrorMessage(), PrintMessageString(), PrintMessageStringV(), PrintPaddedMessageString(), and wmain().