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