/* $Id$
*
* 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 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 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;
}