base/applications/network/net/main.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS net command
   4  * FILE:            base/applications/network/net/main.c
   5  * PURPOSE:
   6  *
   7  * PROGRAMMERS:     Magnus Olsen (greatlord@reactos.org)
   8  */
   9
  10 #include "net.h"
  11
  12 #define MAX_BUFFER_SIZE 4096
  13
  14 typedef struct _COMMAND
  15 {
  16     WCHAR *name;
  17     INT (*func)(INT, WCHAR**);
  18 //    VOID (*help)(INT, WCHAR**);
  19 } COMMAND, *PCOMMAND;
  20
  21 COMMAND cmds[] =
  22 {
  23     {L"accounts",   cmdAccounts},
  24     {L"computer",   cmdComputer},
  25     {L"config",     cmdConfig},
  26     {L"continue",   cmdContinue},
  27     {L"file",       unimplemented},
  28     {L"group",      cmdGroup},
  29     {L"help",       cmdHelp},
  30     {L"helpmsg",    cmdHelpMsg},
  31     {L"localgroup", cmdLocalGroup},
  32     {L"name",       unimplemented},
  33     {L"pause",      cmdPause},
  34     {L"print",      unimplemented},
  35     {L"send",       unimplemented},
  36     {L"session",    unimplemented},
  37     {L"share",      unimplemented},
  38     {L"start",      cmdStart},
  39     {L"statistics", cmdStatistics},
  40     {L"stop",       cmdStop},
  41     {L"syntax",     cmdSyntax},
  42     {L"time",       unimplemented},
  43     {L"use",        cmdUse},
  44     {L"user",       cmdUser},
  45     {L"view",       unimplemented},
  46     {NULL,          NULL}
  47 };
  48
  49 HMODULE hModuleNetMsg = NULL;
  50
  51
  52 VOID
  53 PrintPaddedResourceString(
  54     UINT uID,
  55     INT nPaddedLength)
  56 {
  57     INT nLength;
  58
  59     nLength = ConResPuts(StdOut, uID);
  60     if (nLength < nPaddedLength)
  61         PrintPadding(L' ', nPaddedLength - nLength);
  62 }
  63
  64
  65 VOID
  66 PrintPadding(
  67     WCHAR chr,
  68     INT nPaddedLength)
  69 {
  70     INT i;
  71     WCHAR szMsgBuffer[MAX_BUFFER_SIZE];
  72
  73     for (i = 0; i < nPaddedLength; i++)
  74          szMsgBuffer[i] = chr;
  75     szMsgBuffer[nPaddedLength] = UNICODE_NULL;
  76
  77     ConPuts(StdOut, szMsgBuffer);
  78 }
  79
  80
  81 VOID
  82 PrintMessageStringV(
  83     DWORD dwMessage,
  84     ...)
  85 {
  86     PWSTR pBuffer;
  87     va_list args = NULL;
  88
  89     va_start(args, dwMessage);
  90
  91     FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE,
  92                    hModuleNetMsg,
  93                    dwMessage,
  94                    LANG_USER_DEFAULT,
  95                    (LPWSTR)&pBuffer,
  96                    0,
  97                    &args);
  98     va_end(args);
  99
 100     if (pBuffer)
 101     {
 102         ConPuts(StdOut, pBuffer);
 103         LocalFree(pBuffer);
 104         pBuffer = NULL;
 105     }
 106 }
 107
 108 VOID
 109 PrintMessageString(
 110     DWORD dwMessage)
 111 {
 112     PWSTR pBuffer;
 113
 114     FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
 115                    FORMAT_MESSAGE_IGNORE_INSERTS,
 116                    hModuleNetMsg,
 117                    dwMessage,
 118                    LANG_USER_DEFAULT,
 119                    (LPWSTR)&pBuffer,
 120                    0,
 121                    NULL);
 122     if (pBuffer)
 123     {
 124         ConPuts(StdOut, pBuffer);
 125         LocalFree(pBuffer);
 126         pBuffer = NULL;
 127     }
 128 }
 129
 130
 131 VOID
 132 PrintPaddedMessageString(
 133     DWORD dwMessage,
 134     INT nPaddedLength)
 135 {
 136     PWSTR pBuffer;
 137     DWORD dwLength;
 138
 139     dwLength = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
 140                               FORMAT_MESSAGE_IGNORE_INSERTS,
 141                               hModuleNetMsg,
 142                               dwMessage,
 143                               LANG_USER_DEFAULT,
 144                               (LPWSTR)&pBuffer,
 145                               0,
 146                               NULL);
 147     if (pBuffer)
 148     {
 149         ConPuts(StdOut, pBuffer);
 150         LocalFree(pBuffer);
 151         pBuffer = NULL;
 152     }
 153
 154     if (dwLength < (DWORD)nPaddedLength)
 155         PrintPadding(L' ', (DWORD)nPaddedLength - dwLength);
 156 }
 157
 158
 159 VOID
 160 PrintErrorMessage(
 161     DWORD dwError)
 162 {
 163     WCHAR szErrorBuffer[16];
 164     PWSTR pBuffer;
 165     PWSTR pErrorInserts[2] = {NULL, NULL};
 166
 167     if (dwError >= MIN_LANMAN_MESSAGE_ID && dwError <= MAX_LANMAN_MESSAGE_ID)
 168     {
 169         FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
 170                        FORMAT_MESSAGE_IGNORE_INSERTS,
 171                        hModuleNetMsg,
 172                        dwError,
 173                        LANG_USER_DEFAULT,
 174                        (LPWSTR)&pBuffer,
 175                        0,
 176                        NULL);
 177         if (pBuffer)
 178         {
 179             ConPrintf(StdErr, L"%s\n", pBuffer);
 180             LocalFree(pBuffer);
 181             pBuffer = NULL;
 182         }
 183     }
 184     else
 185     {
 186         FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
 187                        FORMAT_MESSAGE_IGNORE_INSERTS,
 188                        NULL,
 189                        dwError,
 190                        LANG_USER_DEFAULT,
 191                        (LPWSTR)&pBuffer,
 192                        0,
 193                        NULL);
 194         if (pBuffer)
 195         {
 196             ConPrintf(StdErr, L"%s\n", pBuffer);
 197             LocalFree(pBuffer);
 198             pBuffer = NULL;
 199         }
 200     }
 201
 202     if (dwError != ERROR_SUCCESS)
 203     {
 204         /* Format insert for the 3514 message */
 205         swprintf(szErrorBuffer, L"%lu", dwError);
 206         pErrorInserts[0] = szErrorBuffer;
 207
 208         /* Format and print the 3514 message */
 209         FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE |
 210                        FORMAT_MESSAGE_ARGUMENT_ARRAY,
 211                        hModuleNetMsg,
 212                        3514,
 213                        LANG_USER_DEFAULT,
 214                        (LPWSTR)&pBuffer,
 215                        0,
 216                        (va_list *)pErrorInserts);
 217         if (pBuffer)
 218         {
 219             ConPrintf(StdErr, L"%s\n", pBuffer);
 220             LocalFree(pBuffer);
 221             pBuffer = NULL;
 222         }
 223     }
 224 }
 225
 226
 227 VOID
 228 PrintNetMessage(
 229     DWORD dwMessage)
 230 {
 231     PWSTR pBuffer;
 232
 233     FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
 234                    FORMAT_MESSAGE_FROM_HMODULE |
 235                    FORMAT_MESSAGE_IGNORE_INSERTS,
 236                    GetModuleHandleW(NULL),
 237                    dwMessage,
 238                    LANG_USER_DEFAULT,
 239                    (LPWSTR)&pBuffer,
 240                    0,
 241                    NULL);
 242     if (pBuffer)
 243     {
 244         ConPuts(StdOut, pBuffer);
 245         ConPuts(StdOut, L"\n");
 246         LocalFree(pBuffer);
 247         pBuffer = NULL;
 248     }
 249 }
 250
 251
 252 VOID
 253 ReadFromConsole(
 254     LPWSTR lpInput,
 255     DWORD dwLength,
 256     BOOL bEcho)
 257 {
 258     DWORD dwOldMode;
 259     DWORD dwRead = 0;
 260     HANDLE hFile;
 261     LPWSTR p;
 262     PCHAR pBuf;
 263
 264     pBuf = HeapAlloc(GetProcessHeap(), 0, dwLength - 1);
 265     ZeroMemory(lpInput, dwLength * sizeof(WCHAR));
 266     hFile = GetStdHandle(STD_INPUT_HANDLE);
 267     GetConsoleMode(hFile, &dwOldMode);
 268
 269     SetConsoleMode(hFile, ENABLE_LINE_INPUT | (bEcho ? ENABLE_ECHO_INPUT : 0));
 270
 271     ReadFile(hFile, (PVOID)pBuf, dwLength - 1, &dwRead, NULL);
 272
 273     MultiByteToWideChar(CP_OEMCP, 0, pBuf, dwRead, lpInput, dwLength - 1);
 274     HeapFree(GetProcessHeap(), 0, pBuf);
 275
 276     for (p = lpInput; *p; p++)
 277     {
 278         if (*p == L'\x0d')
 279         {
 280             *p = L'\0';
 281             break;
 282         }
 283     }
 284
 285     SetConsoleMode(hFile, dwOldMode);
 286 }
 287
 288
 289 int wmain(int argc, WCHAR **argv)
 290 {
 291     WCHAR szDllBuffer[MAX_PATH];
 292     PCOMMAND cmdptr;
 293     int nResult = 0;
 294     BOOL bRun = FALSE;
 295
 296     /* Initialize the Console Standard Streams */
 297     ConInitStdStreams();
 298
 299     /* Load netmsg.dll */
 300     GetSystemDirectoryW(szDllBuffer, ARRAYSIZE(szDllBuffer));
 301     wcscat(szDllBuffer, L"\\netmsg.dll");
 302
 303     hModuleNetMsg = LoadLibrary(szDllBuffer);
 304     if (hModuleNetMsg == NULL)
 305     {
 306         ConPrintf(StdErr, L"Failed to load netmsg.dll\n");
 307         return 1;
 308     }
 309
 310     if (argc < 2)
 311     {
 312         nResult = 1;
 313         goto done;
 314     }
 315
 316     /* Scan the command table */
 317     for (cmdptr = cmds; cmdptr->name; cmdptr++)
 318     {
 319         if (_wcsicmp(argv[1], cmdptr->name) == 0)
 320         {
 321             nResult = cmdptr->func(argc, argv);
 322             bRun = TRUE;
 323             break;
 324         }
 325     }
 326
 327 done:
 328     if (bRun == FALSE)
 329     {
 330         ConResPuts(StdOut, IDS_GENERIC_SYNTAX);
 331         PrintNetMessage(MSG_NET_SYNTAX);
 332     }
 333
 334     if (hModuleNetMsg != NULL)
 335         FreeLibrary(hModuleNetMsg);
 336
 337     return nResult;
 338 }
 339
 340 INT unimplemented(INT argc, WCHAR **argv)
 341 {
 342     ConPuts(StdOut, L"This command is not implemented yet\n");
 343     return 1;
 344 }