int CShellCommandChangeKey::Execute(CConsole &rConsole, CArgumentParser& rArguments) { BOOL blnHelp = FALSE; rArguments.ResetArgumentIteration(); const TCHAR *pchCommandItself = rArguments.GetNextArgument(); const TCHAR *pchPath = rArguments.GetNextArgument(); if ((_tcsnicmp(pchCommandItself,CD_CMD _T(".."),CD_CMD_LENGTH+2*sizeof(TCHAR)) == 0)|| (_tcsnicmp(pchCommandItself,CD_CMD _T("\\"),CD_CMD_LENGTH+1*sizeof(TCHAR)) == 0)) { if (!pchPath) pchPath = pchCommandItself + CD_CMD_LENGTH; else blnHelp = TRUE; } if ((!blnHelp)&&(pchPath != NULL)&&(!rArguments.GetNextArgument())) { size_t size = _tcslen(pchPath); ASSERT(size <= PROMPT_BUFFER_SIZE); if (!m_rTree.ChangeCurrentKey(pchPath)) { rConsole.Write(m_rTree.GetLastErrorDescription()); rConsole.Write(_T("\n")); } } else { rConsole.Write(GetHelpString()); } return 0; }
int CShellCommandConnect::Execute(CConsole &rConsole, CArgumentParser& rArguments) { const TCHAR *pchArg; const TCHAR *pchMachine = NULL; BOOL blnHelp = FALSE; VERIFY(m_rTree.ChangeCurrentKey(_T("\\"))); while ((pchArg = rArguments.GetNextArgument()) != NULL) { if ((_tcsicmp(pchArg,_T("/?")) == 0) ||(_tcsicmp(pchArg,_T("-?")) == 0)) { blnHelp = TRUE; } // else if ((_tcsicmp(pchArg,_T("-a")) == 0)|| // (_tcsicmp(pchArg,_T("/a")) == 0)) // { // } else { pchMachine = pchArg; } } if (blnHelp) rConsole.Write(GetHelpString()); if (!m_rTree.SetMachineName(pchMachine)) { rConsole.Write(m_rTree.GetLastErrorDescription()); rConsole.Write(_T("\n")); } return 0; }
int CShellCommandDeleteValue::Execute(CConsole &rConsole, CArgumentParser& rArguments) { rArguments.ResetArgumentIteration(); TCHAR *pszCommandItself = rArguments.GetNextArgument(); TCHAR *pszParameter; TCHAR *pszValueFull = NULL; BOOL blnHelp = FALSE; if ((_tcsnicmp(pszCommandItself,DV_CMD _T(".."),DV_CMD_LENGTH+2*sizeof(TCHAR)) == 0)|| (_tcsnicmp(pszCommandItself,DV_CMD _T("\\"),DV_CMD_LENGTH+1*sizeof(TCHAR)) == 0)) { pszValueFull = pszCommandItself + DV_CMD_LENGTH; } else if (_tcsnicmp(pszCommandItself,DV_CMD _T("/"),DV_CMD_LENGTH+1*sizeof(TCHAR)) == 0) { pszParameter = pszCommandItself + DV_CMD_LENGTH; goto CheckValueArgument; } while((pszParameter = rArguments.GetNextArgument()) != NULL) { CheckValueArgument: if ((_tcsicmp(pszParameter,_T("/?")) == 0) ||(_tcsicmp(pszParameter,_T("-?")) == 0)) { blnHelp = TRUE; break; } else if (!pszValueFull) { pszValueFull = pszParameter; } else { rConsole.Write(_T("Bad parameter: ")); rConsole.Write(pszParameter); rConsole.Write(_T("\n")); } } CRegistryKey Key; TCHAR *pszValueNamePattern; const TCHAR *pszEmpty = _T(""); const TCHAR *pszPath; if (blnHelp) { rConsole.Write(GetHelpString()); return 0; } if (pszValueFull) { if (_tcscmp(pszValueFull,_T("\\")) == 0) goto CommandNAonRoot; TCHAR *pchSep = _tcsrchr(pszValueFull,_T('\\')); pszValueNamePattern = pchSep?(pchSep+1):(pszValueFull); pszPath = pchSep?pszValueFull:_T("."); if (pchSep) *pchSep = 0; } else { pszValueNamePattern = (TCHAR*)pszEmpty; pszPath = _T("."); } { size_t s = _tcslen(pszValueNamePattern); if (s && (pszValueNamePattern[0] == _T('\"'))&&(pszValueNamePattern[s-1] == _T('\"'))) { pszValueNamePattern[s-1] = 0; pszValueNamePattern++; } } if (!m_rTree.GetKey(pszPath,KEY_QUERY_VALUE|KEY_SET_VALUE,Key)) { rConsole.Write(m_rTree.GetLastErrorDescription()); return 0; } if (!Key.IsRoot()) { // not root key ??? TCHAR Buffer[254]; DWORD dwMaxValueNameLength; LONG nError = Key.GetMaxValueNameLength(dwMaxValueNameLength); if (nError != ERROR_SUCCESS) { _stprintf(Buffer,_T("Cannot query info about %s key. Error is %u\n"),Key.GetKeyName(),(unsigned int)nError); rConsole.Write(Buffer); return 0; } TCHAR *pszValueName = new (std::nothrow) TCHAR[dwMaxValueNameLength]; if (!pszValueName) { rConsole.Write("Out of memory."); return 0; } Key.InitValueEnumeration(pszValueName,dwMaxValueNameLength,NULL,0,NULL); while ((nError = Key.GetNextValue()) == ERROR_SUCCESS) { if (PatternMatch(pszValueNamePattern,pszValueName)) { nError = Key.DeleteValue(pszValueName); if (nError != ERROR_SUCCESS) { _stprintf(Buffer,_T("Cannot delete value. Error is %u\n"),(unsigned int)nError); rConsole.Write(Buffer); } else { InvalidateCompletion(); } Key.InitValueEnumeration(pszValueName,dwMaxValueNameLength,NULL,0,NULL); // reset iteration } } } // if (pKey) else { CommandNAonRoot: rConsole.Write(DV_CMD COMMAND_NA_ON_ROOT); } return 0; }
int CShellCommandOwner::Execute(CConsole &rConsole, CArgumentParser& rArguments) { const TCHAR *pchKey = NULL; BOOL blnDo = TRUE; BOOL blnBadParameter = FALSE; BOOL blnHelp = FALSE; const TCHAR *pchParameter; DWORD dwError; rArguments.ResetArgumentIteration(); const TCHAR *pchCommandItself = rArguments.GetNextArgument(); if ((_tcsnicmp(pchCommandItself,OWNER_CMD _T(".."),OWNER_CMD_LENGTH+2*sizeof(TCHAR)) == 0)|| (_tcsnicmp(pchCommandItself,OWNER_CMD _T("\\"),OWNER_CMD_LENGTH+1*sizeof(TCHAR)) == 0)) { pchKey = pchCommandItself + OWNER_CMD_LENGTH; } else if (_tcsnicmp(pchCommandItself,OWNER_CMD _T("/"),OWNER_CMD_LENGTH+1*sizeof(TCHAR)) == 0) { pchParameter = pchCommandItself + OWNER_CMD_LENGTH; goto CheckOwnerArgument; } while((pchParameter = rArguments.GetNextArgument()) != NULL) { CheckOwnerArgument: blnBadParameter = FALSE; if ((_tcsicmp(pchParameter,_T("/?")) == 0) ||(_tcsicmp(pchParameter,_T("-?")) == 0)) { blnHelp = TRUE; blnDo = pchKey != NULL; } else if (!pchKey) { pchKey = pchParameter; blnDo = TRUE; } else { blnBadParameter = TRUE; } if (blnBadParameter) { rConsole.Write(_T("Bad parameter: ")); rConsole.Write(pchParameter); rConsole.Write(_T("\n")); } } CRegistryKey Key; if (!m_rTree.GetKey(pchKey?pchKey:_T("."),KEY_QUERY_VALUE|READ_CONTROL,Key)) { rConsole.Write(m_rTree.GetLastErrorDescription()); blnDo = FALSE; } if (blnHelp) { rConsole.Write(GetHelpString()); } if (blnDo&&blnHelp) rConsole.Write(_T("\n")); if (!blnDo) return 0; if (Key.IsRoot()) { // root key rConsole.Write(OWNER_CMD COMMAND_NA_ON_ROOT); return 0; } PISECURITY_DESCRIPTOR pSecurityDescriptor = NULL; TCHAR *pchName = NULL, *pchDomainName = NULL; try { DWORD dwSecurityDescriptorLength; rConsole.Write(_T("Key : ")); rConsole.Write(_T("\\")); rConsole.Write(Key.GetKeyName()); rConsole.Write(_T("\n")); dwError = Key.GetSecurityDescriptorLength(&dwSecurityDescriptorLength); if (dwError != ERROR_SUCCESS) throw dwError; pSecurityDescriptor = (PISECURITY_DESCRIPTOR) new unsigned char [dwSecurityDescriptorLength]; DWORD dwSecurityDescriptorLength1 = dwSecurityDescriptorLength; dwError = Key.GetSecurityDescriptor((SECURITY_INFORMATION)OWNER_SECURITY_INFORMATION,pSecurityDescriptor,&dwSecurityDescriptorLength1); if (dwError != ERROR_SUCCESS) throw dwError; PSID psidOwner; BOOL blnOwnerDefaulted; if (!GetSecurityDescriptorOwner(pSecurityDescriptor,&psidOwner,&blnOwnerDefaulted)) throw GetLastError(); if (psidOwner == NULL) { rConsole.Write(_T("Key has no owner.")); } else { if (!IsValidSid(psidOwner)) { rConsole.Write(_T("Key has invalid owner SID.")); } else { rConsole.Write(_T("Key Owner: \n")); DWORD dwSIDStringSize = 0; BOOL blnRet = GetTextualSid(psidOwner,NULL,&dwSIDStringSize); ASSERT(!blnRet); ASSERT(GetLastError() == ERROR_INSUFFICIENT_BUFFER); TCHAR *pchSID = new TCHAR[dwSIDStringSize]; if(!GetTextualSid(psidOwner,pchSID,&dwSIDStringSize)) { dwError = GetLastError(); ASSERT(dwError != ERROR_INSUFFICIENT_BUFFER); rConsole.Write(_T("Error ")); TCHAR Buffer[256]; rConsole.Write(_itoa(dwError,Buffer,10)); rConsole.Write(_T("\nGetting string representation of SID\n")); } else { rConsole.Write(_T("\tSID: ")); rConsole.Write(pchSID); rConsole.Write(_T("\n")); } delete [] pchSID; DWORD dwNameBufferLength, dwDomainNameBufferLength; dwNameBufferLength = 1024; dwDomainNameBufferLength = 1024; pchName = new TCHAR [dwNameBufferLength]; pchDomainName = new TCHAR [dwDomainNameBufferLength]; DWORD dwNameLength = dwNameBufferLength, dwDomainNameLength = dwDomainNameBufferLength; SID_NAME_USE Use; if (!LookupAccountSid(NULL,psidOwner,pchName,&dwNameLength,pchDomainName,&dwDomainNameLength,&Use)) throw GetLastError(); else { rConsole.Write(_T("\tOwner Domain: ")); rConsole.Write(pchDomainName); rConsole.Write(_T("\n")); rConsole.Write(_T("\tOwner Name: ")); rConsole.Write(pchName); rConsole.Write(_T("\n\tSID type: ")); rConsole.Write(GetSidTypeName(Use)); rConsole.Write(_T("\n")); rConsole.Write(_T("\tOwner defaulted: ")); rConsole.Write(blnOwnerDefaulted?_T("Yes"):_T("No")); rConsole.Write(_T("\n")); } delete [] pchName; pchName = NULL; delete [] pchDomainName; pchDomainName = NULL; } } delete [] pSecurityDescriptor; } catch (DWORD dwError) { rConsole.Write(_T("Error ")); TCHAR Buffer[256]; rConsole.Write(_itoa(dwError,Buffer,10)); rConsole.Write(_T("\n")); if (pchName) delete [] pchName; if (pchDomainName) delete [] pchDomainName; if (pSecurityDescriptor) delete [] pSecurityDescriptor; } return 0; }
int CShellCommandDOKA::Execute(CConsole &rConsole, CArgumentParser& rArguments) { REGSAM Access = m_rTree.GetDesiredOpenKeyAccess(); const TCHAR *pchParameter; BOOL blnBadParameter = FALSE; BOOL blnHelp = FALSE; while((pchParameter = rArguments.GetNextArgument()) != NULL) { blnBadParameter = FALSE; // Console.Write(_T("Processing parameter: \")"); // Console.Write(pchParameter); // Console.Write(_T("\")\n"); if ((_tcsicmp(pchParameter,_T("/?")) == 0) ||(_tcsicmp(pchParameter,_T("-?")) == 0)) { blnHelp = TRUE; } else if (*pchParameter == _T('-')) { TCHAR a = *(pchParameter+1); if (a == 0) { Access = 0; } else { if (*(pchParameter+2) != 0) { blnBadParameter = TRUE; } else { switch(a) { case _T('l'): // KEY_CREATE_LINK case _T('L'): Access &= ~KEY_CREATE_LINK; break; case _T('c'): // KEY_CREATE_SUB_KEY case _T('C'): Access &= ~KEY_CREATE_SUB_KEY; break; case _T('e'): // KEY_ENUMERATE_SUB_KEYS case _T('E'): Access &= ~KEY_ENUMERATE_SUB_KEYS; break; case _T('n'): // KEY_NOTIFY case _T('N'): Access &= ~KEY_NOTIFY; break; case _T('q'): // KEY_QUERY_VALUE case _T('Q'): Access &= ~KEY_QUERY_VALUE; break; case _T('s'): // KEY_SET_VALUE case _T('S'): Access &= ~KEY_SET_VALUE; break; default: blnBadParameter = TRUE; } // switch } // else (*(pchParameter+2) != 0) } // else (a == 0) } // if (*pchParameter == _T('-')) else if (*pchParameter == _T('+')) { TCHAR a = *(pchParameter+1); if (a == 0) { blnBadParameter = TRUE; } else { if (*(pchParameter+2) != 0) { blnBadParameter = TRUE; } else { switch(a) { case _T('a'): // KEY_ALL_ACCESS case _T('A'): Access |= KEY_ALL_ACCESS; break; case _T('l'): // KEY_CREATE_LINK case _T('L'): Access |= KEY_CREATE_LINK; break; case _T('c'): // KEY_CREATE_SUB_KEY case _T('C'): Access |= KEY_CREATE_SUB_KEY; break; case _T('e'): // KEY_ENUMERATE_SUB_KEYS case _T('E'): Access |= KEY_ENUMERATE_SUB_KEYS; break; case _T('n'): // KEY_NOTIFY case _T('N'): Access |= KEY_NOTIFY; break; case _T('q'): // KEY_QUERY_VALUE case _T('Q'): Access |= KEY_QUERY_VALUE; break; case _T('s'): // KEY_SET_VALUE case _T('S'): Access |= KEY_SET_VALUE; break; // case _T('X'): // KEY_EXECUTE // case _T('x'): // Access |= KEY_EXECUTE; // break; case _T('R'): // KEY_READ case _T('r'): Access |= KEY_READ; break; default: blnBadParameter = TRUE; } // switch } // else (*(pchParameter+2) != 0) } // else (a == 0) } // if (*pchParameter == _T('-')) else { blnBadParameter = TRUE; } if (blnBadParameter) { rConsole.Write(_T("Bad parameter: ")); rConsole.Write(pchParameter); rConsole.Write(_T("\n")); blnHelp = TRUE; } } // while((pchParameter = Parser.GetNextArgument()) != NULL) if (blnHelp) { rConsole.Write(GetHelpString()); } else { m_rTree.SetDesiredOpenKeyAccess(Access); rConsole.Write(_T("Desired open key access:\n")); REGSAM Access = m_rTree.GetDesiredOpenKeyAccess(); if (Access & KEY_CREATE_LINK) { rConsole.Write(_T("\tKEY_CREATE_LINK - Permission to create a symbolic link.\n")); } if (Access & KEY_CREATE_SUB_KEY) { rConsole.Write(_T("\tKEY_CREATE_SUB_KEY - Permission to create subkeys.\n")); } if (Access & KEY_ENUMERATE_SUB_KEYS) { rConsole.Write(_T("\tKEY_ENUMERATE_SUB_KEYS - Permission to enumerate subkeys.\n")); } if (Access & KEY_NOTIFY) { rConsole.Write(_T("\tKEY_NOTIFY - Permission for change notification.\n")); } if (Access & KEY_QUERY_VALUE) { rConsole.Write(_T("\tKEY_QUERY_VALUE - Permission to query subkey data.\n")); } if (Access & KEY_SET_VALUE) { rConsole.Write(_T("\tKEY_SET_VALUE - Permission to set subkey data.\n")); } } return 0; }
int CShellCommandValue::Execute(CConsole &rConsole, CArgumentParser& rArguments) { rArguments.ResetArgumentIteration(); TCHAR *pchCommandItself = rArguments.GetNextArgument(); TCHAR *pchParameter; TCHAR *pchValueFull = NULL; BOOL blnUnicodeDump = FALSE; BOOL blnBadParameter = FALSE; BOOL blnHelp = FALSE; LONG nError; DWORD dwValueSize; DWORD dwType = REG_NONE; BYTE *pDataBuffer = NULL; TCHAR *pchFilename = NULL; if ((_tcsnicmp(pchCommandItself,VALUE_CMD _T(".."),VALUE_CMD_LENGTH+2*sizeof(TCHAR)) == 0)|| (_tcsnicmp(pchCommandItself,VALUE_CMD _T("\\"),VALUE_CMD_LENGTH+1*sizeof(TCHAR)) == 0)) { pchValueFull = pchCommandItself + VALUE_CMD_LENGTH; } else if (_tcsnicmp(pchCommandItself,VALUE_CMD _T("/"),VALUE_CMD_LENGTH+1*sizeof(TCHAR)) == 0) { pchParameter = pchCommandItself + VALUE_CMD_LENGTH; goto CheckValueArgument; } while((pchParameter = rArguments.GetNextArgument()) != NULL) { CheckValueArgument: blnBadParameter = FALSE; if ((_tcsicmp(pchParameter,_T("/?")) == 0) ||(_tcsicmp(pchParameter,_T("-?")) == 0)) { blnHelp = TRUE; break; } else if (_tcsicmp(pchParameter,_T("/u")) == 0) { blnUnicodeDump = TRUE; } else if ((*pchParameter == _T('/'))&&(*(pchParameter+1) == _T('f'))) { pchFilename = pchParameter+2; } else if (!pchValueFull) { pchValueFull = pchParameter; } else { blnBadParameter = TRUE; } if (blnBadParameter) { rConsole.Write(_T("Bad parameter: ")); rConsole.Write(pchParameter); rConsole.Write(_T("\n")); } } CRegistryKey Key; TCHAR *pchValueName; const TCHAR *pszEmpty = _T(""); const TCHAR *pszPath; if (blnHelp) { rConsole.Write(GetHelpString()); if (pDataBuffer) delete pDataBuffer; return 0; } if (pchValueFull) { if (_tcscmp(pchValueFull,_T("\\")) == 0) goto ValueCommandNAonRoot; TCHAR *pchSep = _tcsrchr(pchValueFull,_T('\\')); pchValueName = pchSep?(pchSep+1):(pchValueFull); pszPath = pchSep?pchValueFull:_T("."); //if (_tcsrchr(pchValueName,_T('.'))) //{ // pchValueName = _T(""); // pchPath = pchValueFull; //} //else if (pchSep) *pchSep = 0; } else { pchValueName = (TCHAR*)pszEmpty; pszPath = _T("."); } if (!m_rTree.GetKey(pszPath,KEY_READ,Key)) { rConsole.Write(m_rTree.GetLastErrorDescription()); goto SkipValueCommand; } if (Key.IsRoot()) goto ValueCommandNAonRoot; { rConsole.Write(_T("Value name : \"")); rConsole.Write(_T("\\")); rConsole.Write(Key.GetKeyName()); size_t l = _tcslen(pchValueName); if (l&& (*pchValueName == _T('\"'))&& (pchValueName[l-1] == _T('\"'))) { pchValueName[l-1] = 0; pchValueName++; } rConsole.Write(pchValueName); rConsole.Write(_T("\"\n")); nError = Key.GetValue(pchValueName,NULL,NULL,&dwValueSize); if (nError == ERROR_SUCCESS) { pDataBuffer = new BYTE [dwValueSize]; Key.GetValue(pchValueName,&dwType,pDataBuffer,&dwValueSize); rConsole.Write(_T("Value type : ")); rConsole.Write(CRegistryKey::GetValueTypeName(dwType)); rConsole.Write(_T("\nValue data : ")); switch(dwType) { case REG_DWORD_LITTLE_ENDIAN: { TCHAR Buffer[3]; rConsole.Write(_T("0x")); for (unsigned int i = 0 ; i < dwValueSize ; i++) { _stprintf(Buffer,_T("%02X"),*(pDataBuffer+((dwValueSize-1)-i))); rConsole.Write(Buffer); } } rConsole.Write(_T("\n")); break; case REG_DWORD_BIG_ENDIAN: { TCHAR Buffer[3]; rConsole.Write(_T("0x")); for (unsigned int i = 0 ; i < dwValueSize ; i++) { _stprintf(Buffer,_T("%02X"),*(pDataBuffer+i)); rConsole.Write(Buffer); } } rConsole.Write(_T("\n")); break; case REG_LINK: break; case REG_MULTI_SZ: { TCHAR *pchCurrentString = (TCHAR *)pDataBuffer; rConsole.Write(_T("\n")); while(*pchCurrentString) { rConsole.Write(_T("\"")); rConsole.Write(pchCurrentString); rConsole.Write(_T("\"\n")); pchCurrentString += _tcslen(pchCurrentString)+1; } } break; case REG_RESOURCE_LIST: break; case REG_SZ: case REG_EXPAND_SZ: rConsole.Write(_T("\"")); rConsole.Write((TCHAR *)pDataBuffer); rConsole.Write(_T("\"\n")); break; case REG_BINARY: default: { TCHAR Buffer[256]; DWORD i, j; for (i = 0 ; i < dwValueSize ; i++) { if (i%16 == 0) { // ok this is begining of line rConsole.Write(_T("\n")); // print offset _stprintf(Buffer,_T("0x%08X "),(unsigned int)i); rConsole.Write(Buffer); } else if (i%8 == 0) { // this is the additional space between 7th and 8th byte in current line rConsole.Write(_T(" ")); } // print current byte unsigned int n = *(pDataBuffer+i); _stprintf(Buffer,_T("%02X "),n); rConsole.Write(Buffer); if (i && (i%16 == 15)) { // if this is the last byte in line // Dump text representation for (j = i-15; j <= i; j += blnUnicodeDump?2:1)\ { if ((j%8 == 0)&&(j%16 != 0)) { // this is the additional space between 7th and 8th byte in current line rConsole.Write(_T(" ")); } ASSERT(i-j < 16); // write current char representation if (blnUnicodeDump) { ASSERT(j%2 == 0); wchar_t ch = *(TCHAR *)(pDataBuffer+j); _stprintf(Buffer, #ifdef _UNICODE _T("%c"), #else // g++ may print warnings here (warning: __wchar_t format, different type arg (arg 3)) // %C in format string is a Microsoft extension. _T("%C"), #endif iswprint(ch)?ch:L'.'); } else { unsigned char ch = *(pDataBuffer+j); _stprintf(Buffer, #ifdef _UNICODE // g++ may print warnings here (warning: __wchar_t format, different type arg (arg 3)) // %C in format string is a Microsoft extension. _T("%C"), #else _T("%c"), #endif isprint(ch)?ch:'.'); } rConsole.Write(Buffer); } // for } // if } // for // print text representation of last line if it is not full (it have less than 16 bytes) // k is pseudo offset for (DWORD k = i; k%16 != 0; k++) { if (k%8 == 0) { // this is the additional space between 7th and 8th byte in current line rConsole.Write(_T(" ")); } _tcscpy(Buffer,_T(" ")); // the replacement of two digit of current byte + spacing rConsole.Write(Buffer); if (k && (k%16 == 15)) { // if this is the last byte in line ASSERT((k-15)%16 == 0); // k-15 must point at begin of last line for (j = k-15; j < i; j += blnUnicodeDump?2:1) { if (blnUnicodeDump&&(j+1 >= i)) { // ok, buffer size is odd number, so we don't display last byte. ASSERT(j+1 == i); break; } if ((j%8 == 0)&&(j%16 != 0)) { // this is the additional space between 7th and 8th byte in current line rConsole.Write(_T(" ")); } // write current char representation if (blnUnicodeDump) { ASSERT(j%2 == 0); wchar_t ch = *(TCHAR *)(pDataBuffer+j); _stprintf(Buffer, #ifdef _UNICODE _T("%c"), #else // g++ may print warnings here (warning: __wchar_t format, different type arg (arg 3)) // %C in format string is a Microsoft extension. _T("%C"), #endif iswprint(ch)?ch:L'.'); } else { unsigned char ch = *(pDataBuffer+j); _stprintf(Buffer, #ifdef _UNICODE // g++ may print warnings here (warning: __wchar_t format, different type arg (arg 3)) // %C in format string is a Microsoft extension. _T("%C"), #else _T("%c"), #endif isprint(ch)?ch:'.'); } rConsole.Write(Buffer); } // for } // if } // for } // default: rConsole.Write(_T("\n")); } // switch rConsole.Write(_T("\n")); if (pchFilename) { rConsole.Write(_T("Exporting value data to ")); rConsole.Write(pchFilename); rConsole.Write(_T(" ...\n")); HANDLE hFile = CreateFile(pchFilename,GENERIC_WRITE,0,NULL,CREATE_NEW,FILE_ATTRIBUTE_NORMAL,NULL); if (hFile == INVALID_HANDLE_VALUE) { rConsole.Write(_T("Cannot create new file ")); rConsole.Write(pchFilename); rConsole.Write(_T("\n")); goto SkipValueCommand; } DWORD dwBytesWritten; if (!WriteFile(hFile,pDataBuffer,dwValueSize,&dwBytesWritten,NULL)) { rConsole.Write(_T("Error writting file.\n")); VERIFY(CloseHandle(hFile)); goto SkipValueCommand; } ASSERT(dwBytesWritten == dwValueSize); VERIFY(CloseHandle(hFile)); } } else { rConsole.Write(_T("Error ")); TCHAR Buffer[256]; rConsole.Write(_itoa(nError,Buffer,10)); rConsole.Write(_T("\n")); if (nError == ERROR_FILE_NOT_FOUND) { rConsole.Write(_T("(System cannot find the value specified)\n")); } } } SkipValueCommand: if (pDataBuffer) delete[] pDataBuffer; return 0; ValueCommandNAonRoot: rConsole.Write(VALUE_CMD COMMAND_NA_ON_ROOT); return 0; }
int CShellCommandSetValue::Execute(CConsole &rConsole, CArgumentParser& rArguments) { LONG nError; rArguments.ResetArgumentIteration(); TCHAR *pszCommandItself = rArguments.GetNextArgument(); TCHAR *pszParameter; TCHAR *pszValueFull = NULL; TCHAR *pszValueData = NULL; BOOL blnBadParameter = FALSE; BOOL blnHelp = FALSE; DWORD dwValueSize = 0; DWORD dwType = REG_NONE; BYTE *pDataBuffer = NULL; if ((_tcsnicmp(pszCommandItself,SET_VALUE_CMD _T(".."),SET_VALUE_CMD_LENGTH+2*sizeof(TCHAR)) == 0)|| (_tcsnicmp(pszCommandItself,SET_VALUE_CMD _T("\\"),SET_VALUE_CMD_LENGTH+1*sizeof(TCHAR)) == 0)) { pszValueFull = pszCommandItself + SET_VALUE_CMD_LENGTH; } else if (_tcsnicmp(pszCommandItself,SET_VALUE_CMD _T("/"),SET_VALUE_CMD_LENGTH+1*sizeof(TCHAR)) == 0) { pszParameter = pszCommandItself + SET_VALUE_CMD_LENGTH; goto CheckValueArgument; } while((pszParameter = rArguments.GetNextArgument()) != NULL) { CheckValueArgument: blnBadParameter = FALSE; if (((*pszParameter == _T('/'))||(*pszParameter == _T('-'))) &&(*(pszParameter+1) == _T('?'))) { blnHelp = TRUE; } else if (dwType == REG_NONE) { if (_tcsicmp(pszParameter,_T("b")) == 0) { dwType = REG_BINARY; } else if (_tcsicmp(pszParameter,_T("dw")) == 0) { dwType = REG_DWORD; } else if (_tcsicmp(pszParameter,_T("dwle")) == 0) { dwType = REG_DWORD_LITTLE_ENDIAN; } else if (_tcsicmp(pszParameter,_T("dwbe")) == 0) { dwType = REG_DWORD_BIG_ENDIAN; } else if (_tcsicmp(pszParameter,_T("sz")) == 0) { dwType = REG_SZ; } else if (_tcsicmp(pszParameter,_T("esz")) == 0) { dwType = REG_EXPAND_SZ; } else { blnBadParameter = TRUE; } } else if (pszValueData == NULL) { pszValueData = pszParameter; } else if (!pszValueFull) { pszValueFull = pszParameter; } else { blnBadParameter = TRUE; } if (blnBadParameter) { rConsole.Write(_T("Bad parameter: ")); rConsole.Write(pszParameter); rConsole.Write(_T("\n")); } } if (!pszValueData) blnHelp = TRUE; CRegistryKey Key; TCHAR *pszValueName; const TCHAR *pszEmpty = _T(""); const TCHAR *pszPath; if (blnHelp) { rConsole.Write(GetHelpString()); if (pDataBuffer) delete pDataBuffer; return 0; } if (pszValueFull) { if (_tcscmp(pszValueFull,_T("\\")) == 0) goto CommandNAonRoot; TCHAR *pchSep = _tcsrchr(pszValueFull,_T('\\')); pszValueName = pchSep?(pchSep+1):(pszValueFull); pszPath = pchSep?pszValueFull:_T("."); //if (_tcsrchr(pszValueName,_T('.'))) //{ // pszValueName = _T(""); // pszPath = pszValueFull; //} //else if (pchSep) *pchSep = 0; } else { pszValueName = (TCHAR*)pszEmpty; pszPath = _T("."); } if (!m_rTree.GetKey(pszPath,KEY_SET_VALUE,Key)) { rConsole.Write(m_rTree.GetLastErrorDescription()); goto SkipCommand; } if (Key.IsRoot()) goto CommandNAonRoot; switch (dwType) { case REG_BINARY: { HANDLE hFile; DWORD dwBytesReaded; hFile = CreateFile(pszValueData,GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,0,NULL); if (hFile == INVALID_HANDLE_VALUE) { rConsole.Write(_T("Cannot open file ")); rConsole.Write(pszValueData); rConsole.Write(_T("\n")); goto SkipCommand; } dwValueSize = GetFileSize(hFile,NULL); if (dwValueSize == (DWORD)-1) // ok, that's right, we compare signed with unsigned here. // GetFileSize is documented and declared to return DWORD. // Error is indicated by checking if return is -1. Design->documentation bug ??? { rConsole.Write(_T("Cannot get size of file ")); rConsole.Write(pszValueData); rConsole.Write(_T("\n")); VERIFY(CloseHandle(hFile)); goto SkipCommand; } pDataBuffer = new BYTE [dwValueSize]; if (!pDataBuffer) { rConsole.Write(_T("Cannot load file into memory. Out of memory.\n")); VERIFY(CloseHandle(hFile)); goto SkipCommand; } if (!ReadFile(hFile,pDataBuffer,dwValueSize,&dwBytesReaded,NULL)) { rConsole.Write(_T("Cannot load file into memory. Error reading file.\n")); VERIFY(CloseHandle(hFile)); goto SkipCommand; } VERIFY(CloseHandle(hFile)); ASSERT(dwBytesReaded == dwValueSize); } break; case REG_DWORD_LITTLE_ENDIAN: case REG_DWORD_BIG_ENDIAN: dwValueSize = 4; pDataBuffer = (BYTE *) new BYTE [dwValueSize]; if (!StringToDWORD(*(DWORD *)pDataBuffer,pszValueData)) { rConsole.Write(_T("Cannot convert ")); rConsole.Write(pszValueData); rConsole.Write(_T(" to DWORD \n")); goto SkipCommand; } if (dwType == REG_DWORD_BIG_ENDIAN) { unsigned char nByte; nByte = *pDataBuffer; *pDataBuffer = *(pDataBuffer+3); *(pDataBuffer+3) = nByte; nByte = *(pDataBuffer+1); *(pDataBuffer+1) = *(pDataBuffer+2); *(pDataBuffer+2) = nByte; } break; case REG_SZ: case REG_EXPAND_SZ: dwValueSize = _tcslen(pszValueData)+1; if (*pszValueData == _T('\"')) { dwValueSize -= 2; *(pszValueData+dwValueSize) = 0; pszValueData++; } dwValueSize *= sizeof(TCHAR); pDataBuffer = (BYTE *) new BYTE [dwValueSize]; { const TCHAR *pchSrc = pszValueData; TCHAR *pchDest = (TCHAR *)pDataBuffer; while(*pchSrc) { if (pchSrc[0] == _T('^')) { if (pchSrc[1] == _T('a')) *pchDest = _T('\a'); else if (pchSrc[1] == _T('b')) *pchDest = _T('\b'); else if (pchSrc[1] == _T('f')) *pchDest = _T('\f'); else if (pchSrc[1] == _T('n')) *pchDest = _T('\n'); else if (pchSrc[1] == _T('r')) *pchDest = _T('\r'); else if (pchSrc[1] == _T('t')) *pchDest = _T('\t'); else *pchDest = pchSrc[1]; pchSrc +=2; pchDest++; dwValueSize--; } else { *pchDest = *pchSrc; pchSrc++; pchDest++; } } *pchDest = _T('\0'); } break; default: ASSERT(FALSE); } { size_t s = _tcslen(pszValueName); if (s && (pszValueName[0] == _T('\"'))&&(pszValueName[s-1] == _T('\"'))) { pszValueName[s-1] = 0; pszValueName++; } } nError = Key.SetValue(pszValueName,dwType,pDataBuffer,dwValueSize); if (nError != ERROR_SUCCESS) { char Buffer[254]; _stprintf(Buffer,_T("Cannot set value. Error is %u\n"),(unsigned int)nError); rConsole.Write(Buffer); } else { InvalidateCompletion(); } SkipCommand: if (pDataBuffer) delete[] pDataBuffer; return 0; CommandNAonRoot: rConsole.Write(SET_VALUE_CMD COMMAND_NA_ON_ROOT); return 0; }
//int _tmain(/*int argc, TCHAR* argv[], TCHAR* envp[]*/) int main () { int nRetCode = 0; HRESULT hr; CSettings *pSettings = NULL; CPrompt *pPrompt = NULL; CShellCommandsLinkedList CommandsList(Console); CShellCommandExit ExitCommand; CommandsList.AddCommand(&ExitCommand); CShellCommandVersion VersionCommand; CommandsList.AddCommand(&VersionCommand); CShellCommandHelp HelpCommand(CommandsList); CommandsList.AddCommand(&HelpCommand); CShellCommandDir DirCommand(Tree); CommandsList.AddCommand(&DirCommand); CShellCommandChangeKey ChangeKeyCommand(Tree); CommandsList.AddCommand(&ChangeKeyCommand); CShellCommandValue ValueCommand(Tree); CommandsList.AddCommand(&ValueCommand); CShellCommandOwner OwnerCommand(Tree); CommandsList.AddCommand(&OwnerCommand); CShellCommandDACL DACLCommand(Tree); CommandsList.AddCommand(&DACLCommand); CShellCommandSACL SACLCommand(Tree); CommandsList.AddCommand(&SACLCommand); CShellCommandConnect ConnectCommand(Tree); CommandsList.AddCommand(&ConnectCommand); CShellCommandNewKey NewKeyCommand(Tree); CommandsList.AddCommand(&NewKeyCommand); CShellCommandDeleteKey DeleteKeyCommand(Tree); CommandsList.AddCommand(&DeleteKeyCommand); CShellCommandSetValue SetValueCommand(Tree); CommandsList.AddCommand(&SetValueCommand); CShellCommandDeleteValue DeleteValueCommand(Tree); CommandsList.AddCommand(&DeleteValueCommand); CArgumentParser Parser; pSettings = new (std::nothrow) CSettings(); if (!pSettings) { _ftprintf(stderr,_T("Cannot initialize settings. Out of memory.\n")); goto Abort; } hr = pSettings->Load(SETTINGS_REGISTRY_KEY); if (FAILED(hr)) { _ftprintf(stderr,_T("Cannot load settings. Error is 0x%X.\n"),(unsigned int)hr); goto Abort; } pPrompt = new (std::nothrow) CPrompt(Tree,hr); if (!pPrompt) { _ftprintf(stderr,_T("Cannot initialize prompt. Out of memory.\n")); goto Abort; } if (FAILED(hr)) { _ftprintf(stderr,_T("Cannot initialize prompt. Error is 0x%X.\n"),(unsigned int)hr); goto Abort; } // input buffer size in chars #define INPUT_BUFFER_SIZE 1024 //#define INPUT_BUFFER_SIZE 128 //#define INPUT_BUFFER_SIZE 10 TCHAR *pchCommand; pchCommand = Console.Init(INPUT_BUFFER_SIZE,10); if (pchCommand == NULL) { _ftprintf(stderr,_T("Cannot initialize console.\n")); nRetCode = 1; goto Exit; } Console.SetReplaceCompletionCallback(CompletionCallback); WORD wOldConsoleAttribute; if (!Console.GetTextAttribute(wOldConsoleAttribute)) goto Abort; Console.SetTitle(_T("Registry Explorer")); Console.SetTextAttribute(pSettings->GetNormalTextAttributes()); VERIFY(SetConsoleCtrlHandler((PHANDLER_ROUTINE)HandlerRoutine,TRUE)); if (!Console.Write(HELLO_MSG //(_L(__TIMESTAMP__)) )) goto Abort; //Tree.SetDesiredOpenKeyAccess(KEY_READ); hr = pPrompt->SetPrompt(pSettings->GetPrompt()); if (FAILED(hr)) { _ftprintf(stderr,_T("Cannot initialize prompt. Error is 0x%X.\n"),(unsigned int)hr); goto Abort; } GetCommand: // prompt // TODO: make prompt user-customizable Console.EnableWrite(); pPrompt->ShowPrompt(Console); Console.FlushInputBuffer(); blnCommandExecutionInProgress = FALSE; // Set command line color Console.SetTextAttribute(pSettings->GetCommandTextAttributes()); if (!Console.ReadLine()) goto Abort; // Set normal color Console.SetTextAttribute(pSettings->GetNormalTextAttributes()); Console.BeginScrollingOperation(); blnCommandExecutionInProgress = TRUE; // Parse command line (1st step - convert to multi sz) Parser.SetArgumentList(pchCommand); int nCommandReturnValue; switch(CommandsList.Execute(Parser,nCommandReturnValue)) { case -1: // not recognized command { Parser.ResetArgumentIteration(); TCHAR *pchCommandItself = Parser.GetNextArgument(); size_t cmdlen = _tcslen(pchCommandItself); if ((!cmdlen)|| (pchCommandItself[cmdlen-1] != _T('\\'))|| (Parser.GetNextArgument())|| (!Tree.ChangeCurrentKey(pchCommandItself))) { Console.Write(_T("Unknown command \"")); Console.Write(pchCommandItself); Console.Write(_T("\"\n")); } } case -2: // empty line goto GetCommand; case 0: // exit command nRetCode = 0; Console.SetTextAttribute(wOldConsoleAttribute); goto Exit; default: Console.Write(_T("\n")); goto GetCommand; } Abort: _ftprintf(stderr,_T("Abnormal program termination.\nPlease report bugs to ") EMAIL _T("\n")); nRetCode = 1; Exit: if (pSettings) delete pSettings; if (pPrompt) delete pPrompt; return nRetCode; }
int CShellCommandHelp::Execute(CConsole &rConsole, CArgumentParser& rArguments) { const TCHAR *pchArg = rArguments.GetNextArgument(); CShellCommand *pCommand; if (pchArg == NULL) { POSITION pos = m_rCommandsLinkedList.GetFirstCommandPosition(); while(pos) { pCommand = m_rCommandsLinkedList.GetNextCommand(pos); rConsole.Write(pCommand->GetHelpShortDescriptionString()); } return 0; } if (_tcsicmp(pchArg,_T("GPL")) == 0) { HRSRC hrcGPL; HGLOBAL hGPL; char *pchGPL; DWORD dwSize; if ((hrcGPL = FindResource(NULL, _T("GPL"), _T("LICENSE")))&& (hGPL = LoadResource(NULL,hrcGPL))&& (pchGPL = (char *)LockResource(hGPL))&& (dwSize = SizeofResource(NULL,hrcGPL))) { // save last char char pchSaved[2]; pchSaved[0] = pchGPL[dwSize-1]; pchSaved[1] = 0; // make null-terminated string pchGPL[dwSize-1] = 0; // replace all non-printable chars except CR, LF and HTAB with spaces for (DWORD i = 0; i < dwSize ; i++) if ((!isprint(pchGPL[i]))&&(pchGPL[i] != '\r')&&(pchGPL[i] != '\n')&&(pchGPL[i] != '\t')) pchGPL[i] = ' '; rConsole.Write(pchGPL); rConsole.Write(pchSaved); } else { rConsole.Write(_T("Internal error cannot load GPL.\n")); } return 0; } while (pchArg) { pCommand = m_rCommandsLinkedList.Match(pchArg); if ((!pCommand)&&((_tcsicmp(pchArg,_T("/?")) == 0)||(_tcsicmp(pchArg,_T("-?")) == 0))) pCommand = this; if (pCommand) { rConsole.Write(pCommand->GetHelpString()); } else { rConsole.Write(_T("HELP: Unknown command \"")); rConsole.Write(pchArg); rConsole.Write(_T("\"\n")); } pchArg = rArguments.GetNextArgument(); } return 0; }
int CShellCommandDeleteKey::Execute(CConsole &rConsole, CArgumentParser& rArguments) { TCHAR *pchKey = NULL, *pchArg; BOOL blnHelp = FALSE; BOOL blnExitAfterHelp = FALSE; BOOL blnRecursive = FALSE; while((pchArg = rArguments.GetNextArgument()) != NULL) { if ((_tcsicmp(pchArg,_T("/?")) == 0) ||(_tcsicmp(pchArg,_T("-?")) == 0)) { blnHelp = TRUE; } else if ((_tcsicmp(pchArg,_T("/s")) == 0) ||(_tcsicmp(pchArg,_T("-s")) == 0)) { blnRecursive = TRUE; } else { if (pchKey) { rConsole.Write(_T("Wrong parameter : \"")); rConsole.Write(pchArg); rConsole.Write(_T("\"\n\n")); blnHelp = TRUE; } else { pchKey = pchArg; } } } if ((!blnHelp) && (!pchKey)) { rConsole.Write(_T("Key name not specified !\n\n")); blnExitAfterHelp = TRUE; } if (blnHelp) { rConsole.Write(GetHelpString()); if (blnExitAfterHelp) return 0; else rConsole.Write(_T("\n")); } // search for last key name token TCHAR *pch = pchKey; while(*pch) pch++; if (pch > pchKey) pch--; while(*pch == _T('\\')) *pch = 0; while((pch > pchKey)&&(*pch != _T('\\'))) pch--; ASSERT(pch >= pchKey); const TCHAR *pszPath; TCHAR *pszPattern = pch; if (*pch == _T('\\')) pszPattern++; if (pch == pchKey) { pszPath = _T("."); } else { if (pch-1 == pchKey) { rConsole.Write(DK_CMD COMMAND_NA_ON_ROOT); return 0; } else { *pch = 0; pszPath = pchKey; } } { size_t s = _tcslen(pszPattern); if (s && (pszPattern[0] == _T('\"'))&&(pszPattern[s-1] == _T('\"'))) { pszPattern[s-1] = 0; pszPattern++; } } if (!m_rTree.DeleteSubkeys(pszPattern,pszPath,blnRecursive)) { rConsole.Write(_T("Cannot delete key(s).\n")); rConsole.Write(m_rTree.GetLastErrorDescription()); } else { InvalidateCompletion(); } return 0; }