mirror of
https://github.com/reactos/reactos.git
synced 2026-06-08 08:52:59 +08:00
no message
svn path=/trunk/; revision=1500
This commit is contained in:
Nedko Arnaudov
@@ -1,4 +1,4 @@
|
||||
/* $Id: RegistryExplorer.cpp,v 1.2 2000/10/24 20:17:41 narnaoud Exp $
|
||||
/* $Id: RegistryExplorer.cpp,v 1.3 2001/01/10 01:25:29 narnaoud Exp $
|
||||
*
|
||||
* regexpl - Console Registry Explorer
|
||||
*
|
||||
@@ -40,7 +40,7 @@
|
||||
#include "ShellCommandOwner.h"
|
||||
#include "ShellCommandDACL.h"
|
||||
#include "ShellCommandSACL.h"
|
||||
#include "ShellCommandDOKA.h"
|
||||
//#include "ShellCommandDOKA.h"
|
||||
#include "ShellCommandConnect.h"
|
||||
#include "ShellCommandNewKey.h"
|
||||
#include "ShellCommandDeleteKey.h"
|
||||
@@ -49,310 +49,9 @@
|
||||
|
||||
TCHAR pchCurrentKey[PROMPT_BUFFER_SIZE];
|
||||
|
||||
CRegistryTree Tree(PROMPT_BUFFER_SIZE+1);
|
||||
CRegistryTree Tree;
|
||||
CConsole Console;
|
||||
|
||||
const TCHAR * ppchRootKeys[7] =
|
||||
{
|
||||
_T("HKEY_CLASSES_ROOT"),
|
||||
_T("HKEY_CURRENT_USER"),
|
||||
_T("HKEY_LOCAL_MACHINE"),
|
||||
_T("HKEY_USERS"),
|
||||
_T("HKEY_PERFORMANCE_DATA"),
|
||||
_T("HKEY_CURRENT_CONFIG"),
|
||||
_T("HKEY_DYN_DATA") //win9x only?
|
||||
};
|
||||
|
||||
BOOL g_blnCompletionCycle = TRUE;
|
||||
|
||||
const TCHAR * CompletionCallback(unsigned __int64 & rnIndex, const BOOL *pblnForward, const TCHAR *pchContext, const TCHAR *pchBegin)
|
||||
{
|
||||
#define COMPLETION_BUFFER_SIZE 4096
|
||||
static TCHAR pchBuffer[COMPLETION_BUFFER_SIZE];
|
||||
CRegistryKey *pKey = NULL;
|
||||
CRegistryTree *pTree = NULL;
|
||||
unsigned __int64 nTotalKeys = 0;
|
||||
unsigned __int64 nTotalValues = 0;
|
||||
unsigned __int64 nTotalItems = 0;
|
||||
class CompletionMatch
|
||||
{
|
||||
public:
|
||||
CompletionMatch(const TCHAR *pchText, CompletionMatch **pNext)
|
||||
{
|
||||
BOOL b = _tcschr(pchText,_T(' ')) != NULL;
|
||||
size_t s = _tcslen(pchText);
|
||||
m_pchText = new TCHAR [s+(b?3:1)];
|
||||
if (b)
|
||||
{
|
||||
m_pchText[0] = _T('\"');
|
||||
_tcscpy(m_pchText+1,pchText);
|
||||
m_pchText[s+1] = _T('\"');
|
||||
m_pchText[s+2] = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
_tcscpy(m_pchText,pchText);
|
||||
}
|
||||
if (m_pchText)
|
||||
{
|
||||
m_pNext = *pNext;
|
||||
*pNext = this;
|
||||
}
|
||||
}
|
||||
~CompletionMatch()
|
||||
{
|
||||
if (m_pchText)
|
||||
delete m_pchText;
|
||||
if (m_pNext)
|
||||
delete m_pNext;
|
||||
}
|
||||
const TCHAR *GetText(unsigned __int64 dwReverseIndex)
|
||||
{
|
||||
if (dwReverseIndex)
|
||||
{
|
||||
if (m_pNext)
|
||||
return m_pNext->GetText(dwReverseIndex-1);
|
||||
else
|
||||
return NULL;
|
||||
}
|
||||
return m_pchText;
|
||||
}
|
||||
private:
|
||||
TCHAR *m_pchText;
|
||||
CompletionMatch *m_pNext;
|
||||
};
|
||||
CompletionMatch *pRootMatch = NULL;
|
||||
|
||||
BOOL blnCompletionOnKeys = TRUE;
|
||||
BOOL blnCompletionOnValues = TRUE;
|
||||
|
||||
while(*pchContext && _istspace(*pchContext))
|
||||
{
|
||||
pchContext++;
|
||||
}
|
||||
|
||||
/* if ((_tcsnicmp(pchContext,DIR_CMD,DIR_CMD_LENGTH) == 0)||
|
||||
(_tcsnicmp(pchContext,CD_CMD,CD_CMD_LENGTH) == 0)||
|
||||
(_tcsnicmp(pchContext,OWNER_CMD,OWNER_CMD_LENGTH) == 0)||
|
||||
(_tcsnicmp(pchContext,DACL_CMD,DACL_CMD_LENGTH) == 0)||
|
||||
(_tcsnicmp(pchContext,SACL_CMD,SACL_CMD_LENGTH) == 0))
|
||||
{
|
||||
blnCompletionOnValues = FALSE;
|
||||
}*/
|
||||
// else if (_tcsnicmp(pchContext,VALUE_CMD,VALUE_CMD_LENGTH) == 0)
|
||||
// {
|
||||
// blnCompletionOnKeys = FALSE;
|
||||
// }
|
||||
|
||||
const TCHAR *pchKey = _tcsrchr(pchBegin,_T('\\'));
|
||||
DWORD nBufferOffset = 0;
|
||||
if (pchKey)
|
||||
{
|
||||
nBufferOffset = pchKey-pchBegin+1;
|
||||
if (nBufferOffset >= COMPLETION_BUFFER_SIZE-1)
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("internal error");
|
||||
}
|
||||
_tcsncpy(pchBuffer,pchBegin,nBufferOffset);
|
||||
pchBuffer[nBufferOffset] = 0;
|
||||
pchBegin = pchKey+1;
|
||||
pTree = new CRegistryTree(Tree);
|
||||
if ((_tcscmp(pTree->GetCurrentPath(),Tree.GetCurrentPath()) != 0)||(!pTree->ChangeCurrentKey(pchBuffer)))
|
||||
{
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return NULL;
|
||||
}
|
||||
else
|
||||
{
|
||||
pKey = pTree->GetCurrentKey();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
pKey = Tree.GetCurrentKey();
|
||||
}
|
||||
|
||||
if (!pKey)
|
||||
{
|
||||
for(unsigned int i = 0 ; i < sizeof(ppchRootKeys)/sizeof(TCHAR *) ; i++)
|
||||
{
|
||||
nTotalKeys++;
|
||||
nTotalItems++;
|
||||
CompletionMatch *p = new CompletionMatch(ppchRootKeys[i],&pRootMatch);
|
||||
if (!p || !p->GetText(0))
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("Out of memory");
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
CompletionMatch *p;
|
||||
DWORD dwError;
|
||||
if (blnCompletionOnKeys)
|
||||
{
|
||||
/* if (_tcslen(pchBegin) == 0)
|
||||
{
|
||||
p = new CompletionMatch(_T(".."),&pRootMatch);
|
||||
if (!p || !p->GetText(0))
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("Out of memory");
|
||||
}
|
||||
nTotalKeys++;
|
||||
nTotalItems++;
|
||||
}
|
||||
*/
|
||||
pKey->InitSubKeyEnumeration();
|
||||
TCHAR *pchSubKeyName;
|
||||
while ((pchSubKeyName = pKey->GetSubKeyName(dwError)) != NULL)
|
||||
{
|
||||
if (_tcsnicmp(pchSubKeyName,pchBegin,_tcslen(pchBegin)) == 0)
|
||||
{
|
||||
nTotalKeys++;
|
||||
nTotalItems++;
|
||||
p = new CompletionMatch(pchSubKeyName,&pRootMatch);
|
||||
if (!p || !p->GetText(0))
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("Out of memory");
|
||||
}
|
||||
}
|
||||
}
|
||||
if ((dwError != ERROR_SUCCESS)&&(dwError != ERROR_NO_MORE_ITEMS))
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("error");
|
||||
}
|
||||
}
|
||||
|
||||
if (blnCompletionOnValues)
|
||||
{
|
||||
pKey->InitValueEnumeration();
|
||||
TCHAR *pchValueName;
|
||||
DWORD dwValueNameLength, dwMaxValueNameLength;
|
||||
dwError = pKey->GetMaxValueNameLength(dwMaxValueNameLength);
|
||||
if (dwError != ERROR_SUCCESS)
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("error");
|
||||
}
|
||||
|
||||
dwMaxValueNameLength++;
|
||||
pchValueName = new TCHAR [dwMaxValueNameLength];
|
||||
if (!pchValueName)
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("Out of memory");
|
||||
}
|
||||
for(;;)
|
||||
{
|
||||
dwValueNameLength = dwMaxValueNameLength;
|
||||
//dwValueSize = dwMaxValueSize;
|
||||
dwError = pKey->GetNextValue(pchValueName,dwValueNameLength,NULL,NULL,NULL);
|
||||
if (dwError == ERROR_NO_MORE_ITEMS) break;
|
||||
if (dwError != ERROR_SUCCESS)
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("error");
|
||||
}
|
||||
|
||||
if (((dwValueNameLength == 0) && (_tcslen(pchBegin) == 0))||
|
||||
(_tcsnicmp(pchValueName,pchBegin,_tcslen(pchBegin)) == 0))
|
||||
{
|
||||
nTotalValues++;
|
||||
nTotalItems++;
|
||||
p = new CompletionMatch((dwValueNameLength == 0)?_T("(Default)"):pchValueName,&pRootMatch);
|
||||
if (!p || !p->GetText(0))
|
||||
{
|
||||
if (pRootMatch) delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("Out of memory");
|
||||
}
|
||||
}
|
||||
} // for
|
||||
delete [] pchValueName;
|
||||
}
|
||||
}
|
||||
if (rnIndex >= nTotalItems)
|
||||
rnIndex = nTotalItems-1;
|
||||
if (pblnForward)
|
||||
{
|
||||
if (*pblnForward)
|
||||
{
|
||||
rnIndex++;
|
||||
if (rnIndex >= nTotalItems)
|
||||
{
|
||||
if (g_blnCompletionCycle)
|
||||
rnIndex = 0;
|
||||
else
|
||||
rnIndex--;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (rnIndex)
|
||||
rnIndex--;
|
||||
else if (g_blnCompletionCycle)
|
||||
rnIndex = nTotalItems-1;
|
||||
}
|
||||
}
|
||||
|
||||
const TCHAR *pchName;
|
||||
if (nTotalItems == 0)
|
||||
{
|
||||
if (pRootMatch)
|
||||
delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return NULL;
|
||||
}
|
||||
ASSERT(rnIndex < nTotalItems);
|
||||
pchName = pRootMatch->GetText(nTotalItems-rnIndex-1);
|
||||
if (pchName == NULL)
|
||||
{
|
||||
if (pRootMatch)
|
||||
delete pRootMatch;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
return _T("internal error");
|
||||
}
|
||||
DWORD dwBufferFull = _tcslen(pchName);
|
||||
if (dwBufferFull >= COMPLETION_BUFFER_SIZE-nBufferOffset)
|
||||
{
|
||||
dwBufferFull = COMPLETION_BUFFER_SIZE-1-nBufferOffset;
|
||||
}
|
||||
_tcsncpy(pchBuffer+nBufferOffset,pchName,dwBufferFull);
|
||||
if ((dwBufferFull < COMPLETION_BUFFER_SIZE-1)&&(rnIndex < nTotalKeys))
|
||||
pchBuffer[nBufferOffset+dwBufferFull++] = _T('\\');
|
||||
pchBuffer[nBufferOffset+dwBufferFull] = 0;
|
||||
if (pTree)
|
||||
delete pTree;
|
||||
if (pRootMatch)
|
||||
delete pRootMatch;
|
||||
return pchBuffer;
|
||||
}
|
||||
|
||||
BOOL blnCommandExecutionInProgress = FALSE;
|
||||
|
||||
BOOL WINAPI HandlerRoutine(DWORD dwCtrlType)
|
||||
@@ -413,8 +112,8 @@ int main ()
|
||||
CShellCommandSACL SACLCommand(Tree);
|
||||
CommandsList.AddCommand(&SACLCommand);
|
||||
|
||||
CShellCommandDOKA DOKACommand(Tree);
|
||||
CommandsList.AddCommand(&DOKACommand);
|
||||
//CShellCommandDOKA DOKACommand(Tree);
|
||||
//CommandsList.AddCommand(&DOKACommand);
|
||||
|
||||
CShellCommandConnect ConnectCommand(Tree);
|
||||
CommandsList.AddCommand(&ConnectCommand);
|
||||
@@ -463,14 +162,15 @@ int main ()
|
||||
//(_L(__TIMESTAMP__))
|
||||
)) goto Abort;
|
||||
|
||||
Tree.SetDesiredOpenKeyAccess(KEY_READ);
|
||||
//Tree.SetDesiredOpenKeyAccess(KEY_READ);
|
||||
|
||||
const TCHAR *pszCurrentPath;
|
||||
GetCommand:
|
||||
// prompt
|
||||
// TODO: make prompt user-customizable
|
||||
Console.EnableWrite();
|
||||
_tcscpy(pchCurrentKey,Tree.GetCurrentPath());
|
||||
Console.Write(pchCurrentKey);
|
||||
pszCurrentPath = Tree.GetCurrentPath();
|
||||
Console.Write(pszCurrentPath?pszCurrentPath:_T("NULL (Internal Error)"));
|
||||
Console.Write(_T("\n# "));
|
||||
Console.FlushInputBuffer();
|
||||
|
||||
@@ -478,7 +178,8 @@ GetCommand:
|
||||
|
||||
// Set command line color
|
||||
// Console.SetTextAttribute(/*FOREGROUND_BLUE|*/FOREGROUND_GREEN|FOREGROUND_RED|FOREGROUND_INTENSITY);
|
||||
if (!Console.ReadLine()) goto Abort;
|
||||
if (!Console.ReadLine())
|
||||
goto Abort;
|
||||
|
||||
// Set normal color
|
||||
// Console.SetTextAttribute(FOREGROUND_BLUE|FOREGROUND_GREEN|FOREGROUND_RED|FOREGROUND_INTENSITY);
|
||||
@@ -486,7 +187,6 @@ GetCommand:
|
||||
Console.BeginScrollingOperation();
|
||||
blnCommandExecutionInProgress = TRUE;
|
||||
|
||||
|
||||
// Parse command line (1st step - convert to multi sz)
|
||||
Parser.SetArgumentList(pchCommand);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user