Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283/* * PROJECT: ReactOS NetSh * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later) * PURPOSE: Network Shell alias management functions * COPYRIGHT: Copyright 2025 Eric Kohl <eric.kohl@reactos.org> */
/* INCLUDES *******************************************************************/
#include "precomp.h"
#define NDEBUG#include <debug.h>
/* GLOBALS ********************************************************************/
typedef struct _ALIAS_ENTRY{ struct _ALIAS_ENTRY *pPrev; struct _ALIAS_ENTRY *pNext;
PWSTR pszAliasName; PWSTR pszAlias;} ALIAS_ENTRY, *PALIAS_ENTRY;
PALIAS_ENTRY AliasListHead = NULL;PALIAS_ENTRY AliasListTail = NULL;
/* FUNCTIONS ******************************************************************/
staticVOIDShowAliases(VOID){ PALIAS_ENTRY pAliasEntry = NULL;
DPRINT1("ShowAliases()\n");
pAliasEntry = AliasListHead; while (pAliasEntry) { ConPrintf(StdOut, L"%s : %s\n", pAliasEntry->pszAliasName, pAliasEntry->pszAlias); pAliasEntry = pAliasEntry->pNext; }}
staticVOIDShowAlias( PWSTR pszAliasName){ PALIAS_ENTRY pAliasEntry = NULL;
DPRINT1("ShowAlias(%S)\n", pszAliasName);
pAliasEntry = AliasListHead; while (pAliasEntry) { if (wcscmp(pAliasEntry->pszAliasName, pszAliasName) == 0) { ConPrintf(StdOut, L"%s\n", pAliasEntry->pszAlias); return; }
pAliasEntry = pAliasEntry->pNext; }
ConResPrintf(StdOut, IDS_ALIAS_NOT_FOUND, pszAliasName);}
staticPALIAS_ENTRYGetAliasEntry( PWSTR pszAliasName){ PALIAS_ENTRY pAliasEntry = NULL;
pAliasEntry = AliasListHead; while (pAliasEntry) { if (wcscmp(pAliasEntry->pszAliasName, pszAliasName) == 0) return pAliasEntry;
pAliasEntry = pAliasEntry->pNext; }
return NULL;}
VOIDInitAliases(VOID){ AliasListHead = NULL; AliasListTail = NULL;}
VOIDDestroyAliases(VOID){ PALIAS_ENTRY pAliasEntry;
while (AliasListHead != NULL) { pAliasEntry = AliasListHead; AliasListHead = AliasListHead->pNext;
HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName); HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias); HeapFree(GetProcessHeap(), 0, pAliasEntry); }
AliasListTail = NULL;}
DWORDWINAPIAliasCommand( LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone){ PALIAS_ENTRY pAliasEntry = NULL; PWSTR pszAlias;
DPRINT("AliasCommand(dwCurrentIndex %lu dwArgCount %lu)\n", dwCurrentIndex, dwArgCount);
/* Show aliases */ if (dwArgCount - dwCurrentIndex == 0) { ShowAliases(); return ERROR_SUCCESS; }
if (dwArgCount - dwCurrentIndex == 1) { ShowAlias(argv[dwCurrentIndex]); return ERROR_SUCCESS; }
/* TODO: Check builtin commands */
pAliasEntry = GetAliasEntry(argv[dwCurrentIndex]); if (pAliasEntry) { pszAlias = MergeStrings(&argv[dwCurrentIndex + 1], dwArgCount - dwCurrentIndex - 1); DPRINT("Alias: %S\n", pszAlias); if (pszAlias == NULL) return ERROR_NOT_ENOUGH_MEMORY;
if (pAliasEntry->pszAlias) HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias);
pAliasEntry->pszAlias = pszAlias; } else { pAliasEntry = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ALIAS_ENTRY)); if (pAliasEntry == NULL) return ERROR_NOT_ENOUGH_MEMORY;
pAliasEntry->pszAliasName = HeapAlloc(GetProcessHeap(), 0, (wcslen(argv[dwCurrentIndex]) + 1) * sizeof(WCHAR)); if (pAliasEntry->pszAliasName == NULL) { HeapFree(GetProcessHeap(), 0, pAliasEntry); return ERROR_NOT_ENOUGH_MEMORY; }
wcscpy(pAliasEntry->pszAliasName, argv[dwCurrentIndex]); DPRINT("AliasName: %S\n", pAliasEntry->pszAliasName);
pAliasEntry->pszAlias = MergeStrings(&argv[dwCurrentIndex + 1], dwArgCount - dwCurrentIndex - 1); DPRINT("Alias: %S\n", pAliasEntry->pszAlias); if (pAliasEntry->pszAlias == NULL) { HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName); HeapFree(GetProcessHeap(), 0, pAliasEntry); return ERROR_NOT_ENOUGH_MEMORY; }
if (AliasListHead == NULL) { AliasListHead = pAliasEntry; AliasListTail = pAliasEntry; } else { pAliasEntry->pPrev = AliasListTail; AliasListTail->pNext = pAliasEntry; AliasListTail = pAliasEntry; } }
DPRINT("Done\n");
return ERROR_OKAY;}
DWORDWINAPIShowAliasCommand( LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone){ DPRINT("ShowAliasCommand()\n");
ShowAliases();
return ERROR_SUCCESS;}
DWORDWINAPIUnaliasCommand( LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone){ PALIAS_ENTRY pAliasEntry = NULL;
DPRINT("UnaliasCommand()\n");
if (dwArgCount - dwCurrentIndex != 1) return ERROR_INVALID_SYNTAX;
DPRINT("Alias %S\n", argv[dwCurrentIndex]);
pAliasEntry = AliasListHead; while (pAliasEntry) { if (wcscmp(pAliasEntry->pszAliasName, argv[dwCurrentIndex]) == 0) { DPRINT("Alias found %S\n", argv[dwCurrentIndex]);
if (pAliasEntry->pNext != NULL) pAliasEntry->pNext->pPrev = pAliasEntry->pPrev; else AliasListTail = pAliasEntry->pPrev;
if (pAliasEntry->pPrev != NULL) pAliasEntry->pPrev->pNext = pAliasEntry->pNext; else AliasListHead = pAliasEntry->pNext;
HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName); HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias); HeapFree(GetProcessHeap(), 0, pAliasEntry);
return ERROR_SUCCESS; }
pAliasEntry = pAliasEntry->pNext; }
ConResPrintf(StdOut, IDS_ALIAS_NOT_FOUND, argv[dwCurrentIndex]);
return ERROR_SUCCESS;}