mirror of
https://github.com/reactos/reactos.git
synced 2024-08-31 07:38:48 +00:00
297 lines
7.7 KiB
C++
297 lines
7.7 KiB
C++
/*
|
|
* regexpl - Console Registry Explorer
|
|
*
|
|
* Copyright (C) 2000-2005 Nedko Arnaudov <nedko@users.sourceforge.net>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
// ShellCommandDir.cpp: implementation of the CShellCommandDir class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#include "ph.h"
|
|
#include "RegistryExplorer.h"
|
|
#include "ShellCommandDir.h"
|
|
#include "RegistryTree.h"
|
|
#include "RegistryKey.h"
|
|
#include "Pattern.h"
|
|
|
|
// *** THIS SHOULD GO IN A MINGW/ROS HEADER (tchar.h ???) - Begin
|
|
#if 1 // #ifndef _ui64tot ???
|
|
#ifdef _UNICODE
|
|
#define _ui64tot _ui64tow
|
|
#else
|
|
#define _ui64tot _ui64toa
|
|
#endif
|
|
#endif
|
|
// *** THIS SHOULD GO IN A MINGW/ROS HEADER - End
|
|
|
|
#define DIR_CMD _T("DIR")
|
|
#define DIR_CMD_LENGTH COMMAND_LENGTH(DIR_CMD)
|
|
#define DIR_CMD_SHORT_DESC DIR_CMD _T(" command lists keys and values of any key.\n")
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Construction/Destruction
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
CShellCommandDir::CShellCommandDir(CRegistryTree& rTree):m_rTree(rTree)
|
|
{
|
|
}
|
|
|
|
CShellCommandDir::~CShellCommandDir()
|
|
{
|
|
}
|
|
|
|
BOOL CShellCommandDir::Match(const TCHAR *pchCommand)
|
|
{
|
|
if (_tcsicmp(pchCommand,DIR_CMD) == 0)
|
|
return TRUE;
|
|
if (_tcsnicmp(pchCommand,DIR_CMD _T(".."),DIR_CMD_LENGTH+2*sizeof(TCHAR)) == 0)
|
|
return TRUE;
|
|
if (_tcsnicmp(pchCommand,DIR_CMD _T("/"),DIR_CMD_LENGTH+1*sizeof(TCHAR)) == 0)
|
|
return TRUE;
|
|
if (_tcsnicmp(pchCommand,DIR_CMD _T("\\"),DIR_CMD_LENGTH+1*sizeof(TCHAR)) == 0)
|
|
return TRUE;
|
|
return FALSE;
|
|
}
|
|
|
|
int CShellCommandDir::Execute(CConsole &rConsole, CArgumentParser& rArguments)
|
|
{
|
|
rArguments.ResetArgumentIteration();
|
|
|
|
BOOL blnDo = TRUE,blnBadParameter, blnHelp = FALSE;
|
|
TCHAR *pszParameter;
|
|
TCHAR *pszCommandItself = rArguments.GetNextArgument();
|
|
TCHAR *pszKey = NULL;
|
|
|
|
if ((_tcsnicmp(pszCommandItself,DIR_CMD _T(".."),DIR_CMD_LENGTH+2*sizeof(TCHAR)) == 0)||
|
|
(_tcsnicmp(pszCommandItself,DIR_CMD _T("\\"),DIR_CMD_LENGTH+1*sizeof(TCHAR)) == 0))
|
|
{
|
|
pszKey = pszCommandItself + DIR_CMD_LENGTH;
|
|
}
|
|
else if (_tcsnicmp(pszCommandItself,DIR_CMD _T("/"),DIR_CMD_LENGTH+1*sizeof(TCHAR)) == 0)
|
|
{
|
|
pszParameter = pszCommandItself + DIR_CMD_LENGTH;
|
|
goto CheckDirArgument;
|
|
}
|
|
|
|
while((pszParameter = rArguments.GetNextArgument()) != NULL)
|
|
{
|
|
CheckDirArgument:
|
|
blnBadParameter = FALSE;
|
|
if ((_tcsicmp(pszParameter,_T("/?")) == 0)
|
|
||(_tcsicmp(pszParameter,_T("-?")) == 0))
|
|
{
|
|
blnHelp = TRUE;
|
|
blnDo = pszKey != NULL;
|
|
}
|
|
else if (!pszKey)
|
|
{
|
|
pszKey = pszParameter;
|
|
blnDo = TRUE;
|
|
}
|
|
else
|
|
{
|
|
blnBadParameter = TRUE;
|
|
}
|
|
if (blnBadParameter)
|
|
{
|
|
rConsole.Write(_T("Bad parameter: "));
|
|
rConsole.Write(pszParameter);
|
|
rConsole.Write(_T("\n"));
|
|
}
|
|
}
|
|
|
|
const TCHAR *pszPattern = PATTERN_MATCH_ALL;
|
|
const TCHAR *pszPath = _T(".");
|
|
|
|
if (pszKey)
|
|
{
|
|
pszPath = pszKey;
|
|
|
|
TCHAR *pch = pszKey;
|
|
while(*pch) // search end of string
|
|
pch++;
|
|
|
|
if (pch > pszKey) // last non-null char
|
|
pch--;
|
|
|
|
if (*pch != _T('\\'))
|
|
{
|
|
while ((pch > pszKey) && (*pch != _T('\\')))
|
|
pch--;
|
|
|
|
if (*pch == _T('\\'))
|
|
{
|
|
pszPattern = pch+1;
|
|
|
|
if (pch > pszKey)
|
|
{
|
|
ASSERT(*pch == _T('\\'));
|
|
*pch = 0;
|
|
}
|
|
else if (*pch == _T('\\'))
|
|
{
|
|
pszPath = _T("\\");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pszPattern = pszKey;
|
|
pszPath = _T(".");
|
|
}
|
|
}
|
|
}
|
|
|
|
CRegistryKey Key;
|
|
|
|
if (!m_rTree.GetKey(pszPath,KEY_ENUMERATE_SUB_KEYS|KEY_QUERY_VALUE,Key))
|
|
{
|
|
const TCHAR *pszErrorMsg = m_rTree.GetLastErrorDescription();
|
|
rConsole.Write(pszErrorMsg);
|
|
blnDo = FALSE;
|
|
}
|
|
|
|
if (blnHelp)
|
|
{
|
|
rConsole.Write(GetHelpString());
|
|
}
|
|
|
|
LONG nError;
|
|
|
|
if (!blnDo)
|
|
return 0;
|
|
|
|
rConsole.Write(_T("\n Key is "));
|
|
rConsole.Write(Key.GetKeyName());
|
|
|
|
if (!Key.IsRoot())
|
|
{
|
|
rConsole.Write(_T("\n Last modify time is "));
|
|
rConsole.Write(Key.GetLastWriteTime());
|
|
}
|
|
|
|
rConsole.Write(_T("\n\n"));
|
|
unsigned __int64 nTotalItems = 0;
|
|
|
|
try
|
|
{
|
|
ASSERT(nTotalItems == 0);
|
|
rConsole.Write(_T("\t(KEY)\t\t\t\t..\\\n")); // parent key abstraction
|
|
nTotalItems = 1;
|
|
|
|
DWORD dwMaxSubkeyNameLength;
|
|
nError = Key.GetSubkeyNameMaxLength(dwMaxSubkeyNameLength);
|
|
if (nError != ERROR_SUCCESS)
|
|
throw nError;
|
|
|
|
TCHAR *pszSubkeyNameBuffer = new (std::nothrow) TCHAR[dwMaxSubkeyNameLength];
|
|
if (!pszSubkeyNameBuffer)
|
|
throw ERROR_OUTOFMEMORY;
|
|
|
|
Key.InitSubkeyEnumeration(pszSubkeyNameBuffer,dwMaxSubkeyNameLength);
|
|
while ((nError = Key.GetNextSubkeyName()) == ERROR_SUCCESS)
|
|
{
|
|
if (PatternMatch(pszPattern,pszSubkeyNameBuffer))
|
|
{
|
|
rConsole.Write(_T("\t(KEY)\t\t\t\t"));
|
|
rConsole.Write(pszSubkeyNameBuffer);
|
|
rConsole.Write(_T("\\\n"));
|
|
nTotalItems++;
|
|
}
|
|
}
|
|
|
|
delete[] pszSubkeyNameBuffer;
|
|
|
|
if (nError != ERROR_NO_MORE_ITEMS)
|
|
throw nError;
|
|
|
|
DWORD dwMaxValueNameBufferSize;
|
|
nError = Key.GetMaxValueNameLength(dwMaxValueNameBufferSize);
|
|
if (nError != ERROR_SUCCESS)
|
|
throw nError;
|
|
|
|
TCHAR *pchValueNameBuffer = new (std::nothrow) TCHAR[dwMaxValueNameBufferSize];
|
|
if (!pchValueNameBuffer)
|
|
throw ERROR_OUTOFMEMORY;
|
|
|
|
|
|
DWORD Type;
|
|
Key.InitValueEnumeration(pchValueNameBuffer,
|
|
dwMaxValueNameBufferSize,
|
|
NULL,
|
|
0,
|
|
&Type);
|
|
|
|
DWORD dwValueNameActualLength;
|
|
const TCHAR *pszValueTypeName;
|
|
unsigned int nTabSize = rConsole.GetTabWidth();
|
|
unsigned int nTabs;
|
|
while((nError = Key.GetNextValue(&dwValueNameActualLength)) == ERROR_SUCCESS)
|
|
{
|
|
if (PatternMatch(pszPattern,pchValueNameBuffer))
|
|
{
|
|
rConsole.Write(_T("\t"));
|
|
pszValueTypeName = CRegistryKey::GetValueTypeName(Type);
|
|
nTabs = _tcslen(pszValueTypeName)/nTabSize;
|
|
nTabs = (nTabs < 4)?(4-nTabs):1;
|
|
rConsole.Write(pszValueTypeName);
|
|
while(nTabs--)
|
|
rConsole.Write(_T("\t"));
|
|
rConsole.Write((dwValueNameActualLength == 0)?_T("(Default)"):pchValueNameBuffer);
|
|
rConsole.Write(_T("\n"));
|
|
nTotalItems++;
|
|
}
|
|
}
|
|
|
|
delete[] pchValueNameBuffer;
|
|
|
|
if (nError != ERROR_NO_MORE_ITEMS)
|
|
throw nError;
|
|
|
|
} // try
|
|
catch (LONG nError)
|
|
{
|
|
rConsole.Write(_T("Error "));
|
|
TCHAR Buffer[256];
|
|
rConsole.Write(_itoa(nError,Buffer,10));
|
|
rConsole.Write(_T("\n"));
|
|
}
|
|
|
|
rConsole.Write(_T("\n Total: "));
|
|
TCHAR Buffer[256];
|
|
rConsole.Write(_ui64tot(nTotalItems,Buffer,10));
|
|
rConsole.Write(_T(" item(s) listed.\n"));
|
|
|
|
return 0;
|
|
}
|
|
|
|
const TCHAR * CShellCommandDir::GetHelpString()
|
|
{
|
|
return DIR_CMD_SHORT_DESC
|
|
_T("Syntax: ") DIR_CMD _T(" [<PATH>\\][<PATTERN>] [/?]\n\n")
|
|
_T(" <PATH> - Optional relative path to the key on which command will be executed\n")
|
|
_T(" <PATTERN> - Optional pattern. Default is the all matching pattern.")
|
|
_T(" /? - This help.\n\n")
|
|
_T("Without parameters, command lists keys and values of current key.\n");
|
|
}
|
|
|
|
const TCHAR * CShellCommandDir::GetHelpShortDescriptionString()
|
|
{
|
|
return DIR_CMD_SHORT_DESC;
|
|
}
|