Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357/* * COPYRIGHT: See COPYING in the top level directory * PROJECT: ReactOS net command * FILE: base/applications/network/net/main.c * PURPOSE: * * PROGRAMMERS: Magnus Olsen (greatlord@reactos.org) */
#include "net.h"
#define MAX_BUFFER_SIZE 4096
typedef struct _COMMAND{ WCHAR *name; INT (*func)(INT, WCHAR**);// VOID (*help)(INT, WCHAR**);} COMMAND, *PCOMMAND;
COMMAND cmds[] ={ {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}};
HMODULE hModuleNetMsg = NULL;
VOIDPrintPaddedResourceString( UINT uID, INT nPaddedLength){ INT nLength;
nLength = ConResPuts(StdOut, uID); if (nLength < nPaddedLength) PrintPadding(L' ', nPaddedLength - nLength);}
VOIDPrintPadding( WCHAR chr, INT nPaddedLength){ INT i; WCHAR szMsgBuffer[MAX_BUFFER_SIZE];
for (i = 0; i < nPaddedLength; i++) szMsgBuffer[i] = chr; szMsgBuffer[nPaddedLength] = UNICODE_NULL;
ConPuts(StdOut, szMsgBuffer);}
DWORDTranslateAppMessage( DWORD dwMessage){ switch (dwMessage) { case NERR_Success: return 3500; // APPERR_3500 case ERROR_MORE_DATA: return 3513; // APPERR_3513 } return dwMessage;}
VOIDPrintMessageStringV( DWORD dwMessage, ...){ PWSTR pBuffer; va_list args = NULL;
va_start(args, dwMessage);
FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE, hModuleNetMsg, dwMessage, LANG_USER_DEFAULT, (LPWSTR)&pBuffer, 0, &args); va_end(args);
if (pBuffer) { ConPuts(StdOut, pBuffer); LocalFree(pBuffer); pBuffer = NULL; }}
VOIDPrintMessageString( DWORD dwMessage){ PWSTR pBuffer;
FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, hModuleNetMsg, dwMessage, LANG_USER_DEFAULT, (LPWSTR)&pBuffer, 0, NULL); if (pBuffer) { ConPuts(StdOut, pBuffer); LocalFree(pBuffer); pBuffer = NULL; }}
VOIDPrintPaddedMessageString( DWORD dwMessage, INT nPaddedLength){ PWSTR pBuffer; DWORD dwLength;
dwLength = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, hModuleNetMsg, dwMessage, LANG_USER_DEFAULT, (LPWSTR)&pBuffer, 0, NULL); if (pBuffer) { ConPuts(StdOut, pBuffer); LocalFree(pBuffer); pBuffer = NULL; }
if (dwLength < (DWORD)nPaddedLength) PrintPadding(L' ', (DWORD)nPaddedLength - dwLength);}
VOIDPrintErrorMessage( DWORD dwError){ 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; } }}
VOIDPrintNetMessage( DWORD dwMessage){ 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; }}
VOIDReadFromConsole( LPWSTR lpInput, DWORD dwLength, BOOL bEcho){ 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);}
int wmain(int argc, WCHAR **argv){ 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;}
INT unimplemented(INT argc, WCHAR **argv){ ConPuts(StdOut, L"This command is not implemented yet\n"); return 1;}