oleview: Added TreeView entries read from registry.

This commit is contained in:
Piotr Caban 2006-06-14 20:09:12 +02:00 committed by Alexandre Julliard
parent 3348ecdddc
commit 6c7a65c1d3
1 changed files with 336 additions and 0 deletions

View File

@ -22,6 +22,14 @@
TREE tree;
static const WCHAR wszCLSID[] = { 'C','L','S','I','D','\\','\0' };
static const WCHAR wszInProcServer32[] =
{ 'I','n','P','r','o','c','S','e','r','v','e','r','3','2','\0' };
static const WCHAR wszOle32dll[] = { 'o','l','e','3','2','.','d','l','l','\0' };
static const WCHAR wszOleAut32dll[] =
{ 'o','l','e','a','u','t','3','2','.','d','l','l','\0' };
static const WCHAR wszImplementedCategories[] =
{ 'I','m','p','l','e','m','e','n','t','e','d',' ',
'C','a','t','e','g','o','r','i','e','s','\0' };
static const WCHAR wszAppID[] = { 'A','p','p','I','D','\\','\0' };
static const WCHAR wszTypeLib[] = { 'T','y','p','e','L','i','b','\\','\0' };
static const WCHAR wszInterface[] = { 'I','n','t','e','r','f','a','c','e','\\','\0' };
@ -42,6 +50,322 @@ LPARAM CreateITEM_INFO(INT flag, const WCHAR *info, const WCHAR *clsid)
return (LPARAM)reg;
}
BOOL CreateRegPath(HTREEITEM item, WCHAR *buffer, int bufSize)
{
TVITEM tvi;
int bufLen;
BOOL ret;
memset(buffer, 0, sizeof(WCHAR[bufSize]));
memset(&tvi, 0, sizeof(TVITEM));
tvi.hItem = item;
SendMessage(globals.hTree, TVM_GETITEM, 0, (LPARAM)&tvi);
ret = (tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGPATH);
while(TRUE)
{
SendMessage(globals.hTree, TVM_GETITEM, 0, (LPARAM)&tvi);
if(tvi.lParam && (((ITEM_INFO *)tvi.lParam)->cFlag & (REGPATH|REGTOP)))
{
bufLen = strlenW(((ITEM_INFO *)tvi.lParam)->info);
memmove(&buffer[bufLen], buffer, sizeof(WCHAR[bufSize-bufLen]));
memcpy(buffer, ((ITEM_INFO *)tvi.lParam)->info, sizeof(WCHAR[bufLen]));
}
if(tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGTOP) break;
if(!tvi.lParam) return FALSE;
tvi.hItem = TreeView_GetParent(globals.hTree, tvi.hItem);
}
return ret;
}
void AddCOMandAll(void)
{
TVINSERTSTRUCT tvis;
TVITEM tvi;
HTREEITEM curSearch;
HKEY hKey, hCurKey, hInfo;
WCHAR valName[MAX_LOAD_STRING];
WCHAR buffer[MAX_LOAD_STRING];
WCHAR wszComp[MAX_LOAD_STRING];
LONG lenBuffer;
int i=-1;
memset(&tvi, 0, sizeof(TVITEM));
tvis.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_CHILDREN;
tvis.item.cchTextMax = MAX_LOAD_STRING;
tvis.item.cChildren = 1;
tvis.hInsertAfter = (HTREEITEM)TVI_FIRST;
if(RegOpenKey(HKEY_CLASSES_ROOT, wszCLSID, &hKey) != ERROR_SUCCESS) return;
while(TRUE)
{
i++;
if(RegEnumKey(hKey, i, valName, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
tvis.hParent = tree.hAO;
if(RegOpenKey(hCurKey, wszInProcServer32, &hInfo) == ERROR_SUCCESS)
{
if(RegQueryValue(hInfo, NULL, buffer, &lenBuffer) == ERROR_SUCCESS
&& *buffer)
if(!memcmp(buffer, wszOle32dll, sizeof(WCHAR[9]))
||!memcmp(buffer, wszOleAut32dll, sizeof(WCHAR[12])))
tvis.hParent = tree.hCLO;
RegCloseKey(hInfo);
}
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
if(RegQueryValue(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
tvis.item.pszText = buffer;
else tvis.item.pszText = valName;
tvis.item.lParam = CreateITEM_INFO(REGPATH|SHOWALL, valName, valName);
if(tvis.hParent) SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
if(RegOpenKey(hCurKey, wszImplementedCategories, &hInfo) == ERROR_SUCCESS)
{
if(RegEnumKey(hInfo, 0, wszComp, -1) != ERROR_SUCCESS) break;
RegCloseKey(hInfo);
if(tree.hGBCC) curSearch = TreeView_GetChild(globals.hTree, tree.hGBCC);
else curSearch = TreeView_GetChild(globals.hTree, TVI_ROOT);
while(curSearch)
{
tvi.hItem = curSearch;
SendMessage(globals.hTree, TVM_GETITEM, 0, (LPARAM)&tvi);
if(tvi.lParam && !strcmpW(((ITEM_INFO *)tvi.lParam)->info, wszComp))
{
tvis.hParent = curSearch;
memmove(&valName[6], valName, sizeof(WCHAR[MAX_LOAD_STRING-6]));
memmove(valName, wszCLSID, sizeof(WCHAR[6]));
tvis.item.lParam = CreateITEM_INFO(REGTOP|REGPATH|SHOWALL,
valName, &valName[6]);
SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
break;
}
curSearch = TreeView_GetNextSibling(globals.hTree, curSearch);
}
}
RegCloseKey(hCurKey);
}
RegCloseKey(hKey);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hCLO);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hAO);
}
void AddApplicationID(void)
{
TVINSERTSTRUCT tvis;
HKEY hKey, hCurKey;
WCHAR valName[MAX_LOAD_STRING];
WCHAR buffer[MAX_LOAD_STRING];
LONG lenBuffer;
int i=-1;
tvis.item.mask = TVIF_TEXT|TVIF_PARAM;
tvis.item.cchTextMax = MAX_LOAD_STRING;
tvis.hInsertAfter = (HTREEITEM)TVI_FIRST;
tvis.hParent = tree.hAID;
if(RegOpenKey(HKEY_CLASSES_ROOT, wszAppID, &hKey) != ERROR_SUCCESS) return;
while(TRUE)
{
i++;
if(RegEnumKey(hKey, i, valName, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
if(RegQueryValue(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
tvis.item.pszText = buffer;
else tvis.item.pszText = valName;
RegCloseKey(hCurKey);
tvis.item.lParam = CreateITEM_INFO(REGPATH, valName, valName);
SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
}
RegCloseKey(hKey);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hAID);
}
void AddTypeLib(void)
{
TVINSERTSTRUCT tvis;
HKEY hKey, hCurKey, hInfoKey;
WCHAR valName[MAX_LOAD_STRING];
WCHAR valParent[MAX_LOAD_STRING];
WCHAR buffer[MAX_LOAD_STRING];
WCHAR wszVer[MAX_LOAD_STRING];
const WCHAR wszFormat[] = { ' ','(','%','s',' ','%','s',')','\0' };
const WCHAR wszFormat2[] = { '%','s','\\','%','s','\0' };
LONG lenBuffer;
int i=-1, j;
tvis.item.mask = TVIF_TEXT|TVIF_PARAM;
tvis.item.cchTextMax = MAX_LOAD_STRING;
tvis.hInsertAfter = (HTREEITEM)TVI_FIRST;
tvis.hParent = tree.hTL;
if(RegOpenKey(HKEY_CLASSES_ROOT, wszTypeLib, &hKey) != ERROR_SUCCESS) return;
while(TRUE)
{
i++;
if(RegEnumKey(hKey, i, valParent, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hKey, valParent, &hCurKey) != ERROR_SUCCESS) continue;
j = -1;
while(TRUE)
{
j++;
if(RegEnumKey(hCurKey, j, valName, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hCurKey, valName, &hInfoKey) != ERROR_SUCCESS) continue;
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
if(RegQueryValue(hInfoKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS
&& *buffer)
{
LoadString(globals.hMainInst, IDS_TL_VER, wszVer,
sizeof(WCHAR[MAX_LOAD_STRING]));
wsprintfW(&buffer[strlenW(buffer)], wszFormat, wszVer, valName);
tvis.item.pszText = buffer;
}
else tvis.item.pszText = valName;
RegCloseKey(hInfoKey);
wsprintfW(wszVer, wszFormat2, valParent, valName);
tvis.item.lParam = CreateITEM_INFO(REGPATH, wszVer, valParent);
SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
}
RegCloseKey(hCurKey);
}
RegCloseKey(hKey);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hTL);
}
void AddInterfaces(void)
{
TVINSERTSTRUCT tvis;
HKEY hKey, hCurKey;
WCHAR valName[MAX_LOAD_STRING];
WCHAR buffer[MAX_LOAD_STRING];
LONG lenBuffer;
int i=-1;
tvis.item.mask = TVIF_TEXT|TVIF_PARAM;
tvis.item.cchTextMax = MAX_LOAD_STRING;
tvis.hInsertAfter = (HTREEITEM)TVI_FIRST;
tvis.hParent = tree.hI;
if(RegOpenKey(HKEY_CLASSES_ROOT, wszInterface, &hKey) != ERROR_SUCCESS) return;
while(TRUE)
{
i++;
if(RegEnumKey(hKey, i, valName, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
if(RegQueryValue(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
tvis.item.pszText = buffer;
else tvis.item.pszText = valName;
RegCloseKey(hCurKey);
tvis.item.lParam = CreateITEM_INFO(REGPATH|INTERFACE, valName, valName);
SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
}
RegCloseKey(hKey);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hI);
}
void AddComponentCategories(void)
{
TVINSERTSTRUCT tvis;
HKEY hKey, hCurKey;
WCHAR valName[MAX_LOAD_STRING];
WCHAR buffer[MAX_LOAD_STRING];
LONG lenBuffer;
DWORD lenBufferHlp;
int i=-1;
tvis.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_CHILDREN;
tvis.item.cchTextMax = MAX_LOAD_STRING;
tvis.hInsertAfter = (HTREEITEM)TVI_FIRST;
if(tree.hGBCC) tvis.hParent = tree.hGBCC;
else tvis.hParent = TVI_ROOT;
tvis.item.cChildren = 1;
if(RegOpenKey(HKEY_CLASSES_ROOT, wszComponentCategories, &hKey) != ERROR_SUCCESS)
return;
while(TRUE)
{
i++;
if(RegEnumKey(hKey, i, valName, -1) != ERROR_SUCCESS) break;
if(RegOpenKey(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
lenBufferHlp = sizeof(WCHAR[MAX_LOAD_STRING]);
if(RegQueryValue(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
tvis.item.pszText = buffer;
else if(RegEnumValue(hCurKey, 0, NULL, NULL, NULL, NULL,
(LPBYTE)buffer, &lenBufferHlp) == ERROR_SUCCESS && *buffer)
tvis.item.pszText = buffer;
else continue;
RegCloseKey(hCurKey);
tvis.item.lParam = CreateITEM_INFO(REGTOP, valName, valName);
SendMessage(globals.hTree, TVM_INSERTITEM, 0, (LPARAM)&tvis);
}
RegCloseKey(hKey);
SendMessage(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hGBCC);
}
void AddBaseEntries(void)
{
TVINSERTSTRUCT tvis;
@ -134,6 +458,18 @@ void EmptyTree(void)
void AddTreeEx(void)
{
AddBaseEntries();
AddComponentCategories();
AddCOMandAll();
AddApplicationID();
AddTypeLib();
AddInterfaces();
}
void AddTree(void)
{
memset(&tree, 0, sizeof(TREE));
AddComponentCategories();
AddCOMandAll();
}
LRESULT CALLBACK TreeProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)