Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

156 lines
4.0 KiB

#include "globals.h"
kNT5NetWalk::kNT5NetWalk(kLogFile *Proc, HWND hIn)
{
LogProc=Proc;
hMainWnd=hIn;
}
BOOL kNT5NetWalk::Begin()
{
DWORD dwRet=0;
dwCurrentKey=0;
dwLevel2Key=0;
strcpy(szRootKeyString, "SYSTEM\\CurrentControlSet\\Control\\Network");
dwRet=RegOpenKeyEx(
HKEY_LOCAL_MACHINE,
szRootKeyString,
0,
KEY_READ,
&hkeyRoot);
if (dwRet==ERROR_SUCCESS)
{
//MessageBox(GetFocus(), "Returning REG_SUCCESS", "SP", MB_OK);
return REG_SUCCESS;
}
else
{
char szMessage[1024];
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, 0, dwRet,
NULL, szMessage, 1024, 0);
printf("**********\r\n");
printf("%s[%d] Error: %s", __FILE__, __LINE__, szMessage);
printf("**********\r\n");
//MessageBox(GetFocus(), "Returning REG_FAIL", "SP", MB_OK);
return REG_FAILURE;
}
//else if (dwPlatform==PLATFORM_NT)
//MessageBox(GetFocus(), "Returning REG_FAIL", "SP", MB_OK);
return REG_FAILURE;
}
BOOL kNT5NetWalk::Walk()
{
DWORD dwIndex = 0;
PTCHAR szName = NULL;
DWORD dwSizeName = MAX_PATH * 4;
LogProc->LogString(",#NT5_Net_Components,,\r\n");
szName = (PTCHAR)malloc(dwSizeName);
if(!szName) {
return FALSE;
}
while (ERROR_SUCCESS == RegEnumKeyEx(hkeyRoot, dwIndex, szName, &dwSizeName, NULL, NULL, NULL, NULL))
{
TCHAR szFull[MAX_PATH * 4];
wsprintf(szFull, "SYSTEM\\CurrentControlSet\\Control\\Network\\%s", szName);
GetKeyValues(szFull);
SearchSubKeys(szFull);
dwSizeName = MAX_PATH * 4;
dwIndex++;
}
return TRUE;
}
BOOL kNT5NetWalk::SearchSubKeys(PTCHAR szName)
{
HKEY hKeyTemp;
DWORD dwRet = 0;
PTCHAR szName2 = NULL;
DWORD dwIndex = 0;
DWORD dwSizeName = MAX_PATH * 4;
TCHAR szFull[MAX_PATH * 4];
if(!szName)
return FALSE;
szName2 = (PTCHAR)malloc(MAX_PATH * 4);
if(!szName2)
return FALSE;
if(ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, szName, 0, KEY_READ, &hKeyTemp)) {
free(szName2);
return FALSE;
}
while (ERROR_SUCCESS == RegEnumKeyEx(hKeyTemp, dwIndex, szName2, &dwSizeName, NULL, NULL, NULL, NULL)) {
wsprintf(szFull, "%s\\%s", szName, szName2);
GetKeyValues(szFull);
SearchSubKeys(szFull);
dwSizeName = MAX_PATH * 4;
dwIndex++;
}
free(szName2);
return TRUE;
}
BOOL kNT5NetWalk::GetKeyValues(char *szName)
{
HKEY hkeyUninstallKey;
char szFullKey[1024];
PUCHAR szProductName=(PUCHAR)malloc(1024);
DWORD dwProductSize=1024;
DWORD dwRet=0;
DWORD dwType=REG_SZ;
strcpy(szFullKey, szRootKeyString);
strcat(szFullKey, "\\");
strcat(szFullKey, szName);
dwRet=RegOpenKeyEx(HKEY_LOCAL_MACHINE, szName, 0,
KEY_READ, &hkeyUninstallKey);
if (dwRet==ERROR_SUCCESS)
{
dwRet=RegQueryValueEx(hkeyUninstallKey, "ComponentId", NULL, &dwType,
szProductName, &dwProductSize);
if (dwRet==ERROR_SUCCESS)
{
printf("Product = %s\r\n", szProductName);
LogProc->StripCommas((char*)szProductName);
LogProc->LogString(",%s,\r\n", szProductName);
free(szProductName);
RegCloseKey(hkeyUninstallKey);
return REG_SUCCESS;
}
/*
else
{
printf("Product = %s\r\n", szName);
LogProc->StripCommas(szName);
LogProc->LogString(",%s,\r\n", szName);
free(szProductName);
RegCloseKey(hkeyUninstallKey);
return REG_SUCCESS;
//Check for other ways to get product name
}
*/
}
else
{
char szMessage[1024];
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, 0, dwRet,
NULL, szMessage, 1024, 0);
printf("**********\r\n");
printf("%s[%d] Error: %s",__FILE__, __LINE__, szMessage);
printf("**********\r\n");
free(szProductName);
RegCloseKey(hkeyUninstallKey);
return REG_FAILURE;
}
free(szProductName);
RegCloseKey(hkeyUninstallKey);
return REG_FAILURE;
}