Windows2000 服务器端应用程序开发设计指南－系统登录

RegScan.cpp 

/******************************************************************** 

模组：RegScan.cpp 

通告：Copyright (c)2000 Jeffrey Richter 

********************************************************************/ 

#include "..\CmnHdr.h "      //请参阅附录A 

#include  

#include  

#include "Resource.h" 

#define UILAYOUT_IMPL 

#include "..\ClassLib \UILayout.h "    //请参阅附录B 

#define PRINTBUF_IMPL 

#include "..\ClassLib \PrintBuf.h "    //请参阅附录B 

#define AUTOBUF_IMPL 

#include "..\ClassLib \AutoBuf.h "    //请参阅附录B 

#define REGWALK_IMPL 

#include "RegWalk.h" 

/////////////////////////////////////////////////////////////////////////////// 

class CRegScan :private CRegWalk { 

public: 

 CRegScan():m_pb(256 *1024){} 

 BOOL Go(PCTSTR pszMachine,HKEY hkeyRoot,PCTSTR pszSubkey, 

  PCTSTR pszString,BOOL fSearchKeyNames,BOOL fSearchValueNames, 

  BOOL fSearchValueData,BOOL fCaseSensitive); 

 PCTSTR Result(){return(m_pb);} 

 void ForceSearchStop(){m_fStopSearch =TRUE;} 

 BOOL WasSearchStopped(){return(m_fStopSearch);} 

private: 

 PCTSTR m_pszString;   //要搜寻的字串 

 BOOL m_fSearchKeyNames;  //要搜寻机码名称？ 

 BOOL m_fSearchValueNames;  //要搜寻登录值名称？ 

 BOOL m_fSearchValueData;  //要搜寻登录值之字串资料？ 

 BOOL m_fShownThisSubkey;  //任何符合现行子机码的？ 

 BOOL m_fStopSearch;   //提早停止搜寻？ 

 CPrintBuf m_pb;//Growable results buffer 

 typedef PTSTR (WINAPI*PFNSTRCMP)(PCTSTR pszFirst,PCTSTR pszSearch); 

 PFNSTRCMP m_pfnStrCmp;//String comparison function 

protected: 

 REGWALKSTATUS onSubkey(PCTSTR pszSubkey,int nDepth, 

  BOOL fRecurseRequested); 

 REGWALKSTATUS onValue(HKEY hkey,PCTSTR pszValue,int nDepth); 

 void ProcessUI(); 

}; 

/////////////////////////////////////////////////////////////////////////////// 

BOOL CRegScan::Go(PCTSTR pszMachine,HKEY hkeyRoot,PCTSTR pszSubkey, 

 PCTSTR pszString,BOOL fSearchKeyNames,BOOL fSearchValueNames, 

 BOOL fSearchValueData,BOOL fCaseSensitive){ 

 m_pszString    =pszString; 

 m_fSearchKeyNames =fSearchKeyNames; 

 m_fSearchValueNames =fSearchValueNames; 

 m_fSearchValueData =fSearchValueData; 

 m_pfnStrCmp   =fCaseSensitive ?StrStr :StrStrI; 

 m_fShownThisSubkey =FALSE; 

 m_fStopSearch   =FALSE; 

 m_pb.Clear(); 

 BOOL fOk =TRUE; 

 if (!m_fSearchKeyNames &&!m_fSearchValueNames &&!m_fSearchValueData){ 

  chMB("You must at least select one field to search."); 

 }else fOk =CRegWalk::Go(pszMachine,hkeyRoot,pszSubkey,TRUE); 

 return(fOk); 

} 

/////////////////////////////////////////////////////////////////////////////// 

void CRegScan::ProcessUI(){ 

 MSG msg; 

 while (PeekMessage(&msg,0,0,0,PM_REMOVE)){ 

  //这里是讯息的使用者介面，处理它们 

  if (!IsDialogMessage(GetActiveWindow(),&msg)){ 

   TranslateMessage(&msg); 

   DispatchMessage(&msg); 

  } 

 } 

} 

/////////////////////////////////////////////////////////////////////////////// 

CRegWalk::REGWALKSTATUS CRegScan::onSubkey(PCTSTR pszSubkey,int nDepth, 

 BOOL fRecurseRequested){ 

 REGWALKSTATUS rws =RWS_FULLSTOP; 

 if (fRecurseRequested ||(nDepth ==0))rws =RWS_RECURSE; 

 //不使用完整路径即取得子机码的名称 

 PCTSTR pszSubkeyName =PathFindFileName(pszSubkey); 

 if (m_fSearchKeyNames) 

  m_fShownThisSubkey =(m_pfnStrCmp(pszSubkeyName,m_pszString)!=NULL); 

 else m_fShownThisSubkey =FALSE; 

 if (m_fShownThisSubkey){ 

  m_pb.Print(TEXT("%s \r \n "),pszSubkey); 

 } 

 ProcessUI(); 

 return(WasSearchStopped()?RWS_FULLSTOP :rws); 

} 

/////////////////////////////////////////////////////////////////////////////// 

CRegWalk::REGWALKSTATUS CRegScan::onValue( 

 HKEY hkey,PCTSTR pszValue,int nDepth){ 

 if (m_fSearchValueNames &&(m_pfnStrCmp(pszValue,m_pszString)!=NULL)){ 

  if (!m_fShownThisSubkey){ 

   m_pb.Print(TEXT("%s \r \n "),m_szSubkeyPath); 

   m_fShownThisSubkey =TRUE; 

  } 

  m_pb.Print(TEXT("\t%s \r \n "),pszValue); 

 } 

 if (m_fSearchValueData){ 

  //检查登录值的资料 

  DWORD dwType; 

  RegQueryValueEx(hkey,pszValue,NULL,&dwType,NULL,NULL); 

  if ((dwType ==REG_EXPAND_SZ)||(dwType ==REG_SZ)){ 

   CAutoBufszData; 

   //给一个大于0的缓冲区以便RegQueryValueEx返回 

   //ERROR_MORE_DATA代替ERROR_SUCCESS 

   szData =1; 

   while (RegQueryValueEx(hkey,pszValue,NULL,NULL,szData,szData) 

    ==ERROR_MORE_DATA); 

   //szData为NULL表示这里没有登录值资料 

   if (((PCTSTR)szData !=NULL)&& 

    (m_pfnStrCmp(szData,m_pszString)!=NULL)){ 

    if (!m_fShownThisSubkey){ 

     m_pb.Print(TEXT("%s \r \n "),m_szSubkeyPath); 

     m_fShownThisSubkey =TRUE; 

    } 

    m_pb.Print(TEXT("\t%s (%s)\r \n "), 

     ((pszValue [0 ] ===0)?TEXT("(default)"):pszValue), 

     (PCTSTR)szData); 

   } 

  } 

 } 

 ProcessUI(); 

 return(WasSearchStopped()?RWS_FULLSTOP :RWS_CONTINUE); 

} 

/////////////////////////////////////////////////////////////////////////////// 

CUILayout g_UILayout;  //当对话方块大小改变时，负责控制 

/////////////////////////////////////////////////////////////////////////////// 

BOOL Dlg_OnInitDialog(HWND hwnd,HWND hwndFocus,LPARAM lParam){ 

 chSETDLGICONS(hwnd,IDI_REGSCAN); 

 HWND hwndRootKey =GetDlgItem(hwnd,IDC_ROOTKEY); 

 int n =0; 

 n =ComboBox_AddString(hwndRootKey,TEXT("HKEY_LOCAL_MACHINE ")); 

 ComboBox_SetItemData(hwndRootKey,n,HKEY_LOCAL_MACHINE); 

 ComboBox_SetCurSel(hwndRootKey,n); //HKLM为预设值 

 n =ComboBox_AddString(hwndRootKey,TEXT("HKEY_CURRENT_CONFIG ")); 

 ComboBox_SetItemData(hwndRootKey,n,HKEY_CURRENT_CONFIG); 

 n =ComboBox_AddString(hwndRootKey,TEXT("HKEY_CLASSES_ROOT ")); 

 ComboBox_SetItemData(hwndRootKey,n,HKEY_CLASSES_ROOT); 

 n =ComboBox_AddString(hwndRootKey,TEXT("HKEY_USERS ")); 

 ComboBox_SetItemData(hwndRootKey,n,HKEY_USERS); 

 n =ComboBox_AddString(hwndRootKey,TEXT("HKEY_CURRENT_USER ")); 

 ComboBox_SetItemData(hwndRootKey,n,HKEY_CURRENT_USER); 

 //重新建立控制的大小 

 g_UILayout.Initialize(hwnd); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPLEFT,CUILayout::AP_TOPRIGHT, 

  IDC_MACHINE,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPLEFT,CUILayout::AP_TOPRIGHT, 

  IDC_ROOTKEY,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPLEFT,CUILayout::AP_TOPRIGHT, 

  IDC_SUBKEY,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPLEFT,CUILayout::AP_TOPRIGHT, 

  IDC_STRING,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_SEARCHKEYNAMES,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_SEARCHVALUENAMES,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_SEARCHVALUEDATA,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_CASESENSITIVE,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_SEARCHSTART,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPRIGHT,CUILayout::AP_TOPRIGHT, 

  IDC_SEARCHSTOP,FALSE); 

 g_UILayout.AnchorControl(CUILayout::AP_TOPLEFT,CUILayout::AP_BOTTOMRIGHT, 

  IDC_SEARCHRESULTS,FALSE); 

 CheckDlgButton(hwnd,IDC_SEARCHKEYNAMES,TRUE); 

 CheckDlgButton(hwnd,IDC_SEARCHVALUENAMES,TRUE); 

 CheckDlgButton(hwnd,IDC_SEARCHVALUEDATA,TRUE); 

 return(TRUE); 

} 

/////////////////////////////////////////////////////////////////////////////// 

void EnableControls(HWND hwnd,BOOL fEnable){ 

 EnableWindow(GetDlgItem(hwnd,IDC_MACHINE),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_ROOTKEY),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_SUBKEY),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_STRING),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_SEARCHKEYNAMES),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_SEARCHVALUENAMES),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_SEARCHVALUEDATA),fEnable); 

 EnableWindow(GetDlgItem(hwnd,IDC_CASESENSITIVE),fEnable); 

 ShowWindow(GetDlgItem(hwnd,IDC_SEARCHSTART),fEnable ?SW_SHOW :SW_HIDE); 

 ShowWindow(GetDlgItem(hwnd,IDC_SEARCHSTOP),fEnable ?SW_HIDE :SW_SHOW); 

} 

/////////////////////////////////////////////////////////////////////////////// 

void Dlg_OnCommand(HWND hwnd,int id,HWND hwndCtl,UINT codeNotify){ 

 static CRegScan x; 

 switch (id){ 

 case IDCANCEL: 

  EndDialog(hwnd,id); 

  break; 

 case IDC_SEARCHSTOP: 

  x.ForceSearchStop(); 

  break; 

 case IDC_SEARCHSTART: 

  SetDlgItemText(hwnd,IDC_SEARCHRESULTS,TEXT("Scanning Registry...")); 

  EnableControls(hwnd,FALSE); 

  TCHAR szString [1000 ]; 

  GetDlgItemText(hwnd,IDC_STRING,szString,chDIMOF(szString)); 

  TCHAR szMachine [100 ],szSubkey [1000 ]; 

  GetDlgItemText(hwnd,IDC_MACHINE,szMachine,chDIMOF(szMachine)); 

  GetDlgItemText(hwnd,IDC_SUBKEY,szSubkey,chDIMOF(szSubkey)); 

  HWND hwndRootKey =GetDlgItem(hwnd,IDC_ROOTKEY); 

  int nIndex =ComboBox_GetCurSel(hwndRootKey); 

  HKEY hkeyRoot =(HKEY)ComboBox_GetItemData(hwndRootKey,nIndex); 

  if (!x.Go( 

   (szMachine [0 ] ==0)?NULL :szMachine,hkeyRoot,szSubkey, 

   szString, 

   IsDlgButtonChecked(hwnd,IDC_SEARCHKEYNAMES), 

   IsDlgButtonChecked(hwnd,IDC_SEARCHVALUENAMES), 

   IsDlgButtonChecked(hwnd,IDC_SEARCHVALUEDATA), 

   IsDlgButtonChecked(hwnd,IDC_CASESENSITIVE))){ 

   chMB("Couldn ’t access the registry "); 

  } 

  SetDlgItemText(hwnd,IDC_SEARCHRESULTS, 

   x.WasSearchStopped()?TEXT("Scan Canceled "): 

    ((x.Result()[0 ] ===0)?TEXT("No entries found "):x.Result())); 

  EnableControls(hwnd,TRUE); 

  break; 

 } 

} 

/////////////////////////////////////////////////////////////////////////////// 

void Dlg_OnSize(HWND hwnd,UINT state,int cx,int cy){ 

 //改变子控制的位置 

 g_UILayout.AdjustControls(cx,cy); 

} 

/////////////////////////////////////////////////////////////////////////////// 

void Dlg_OnGetMinMaxInfo(HWND hwnd,PMINMAXINFO pMinMaxInfo){ 

 //回传对话方块的最小尺寸 

 g_UILayout.HandleMinMax(pMinMaxInfo); 

} 

/////////////////////////////////////////////////////////////////////////////// 

INT_PTR WINAPI Dlg_Proc(HWND hwnd,UINT uMsg,WPARAM wParam,LPARAM lParam){ 

 switch (uMsg){ 

 chHANDLE_DLGMSG(hwnd,WM_INITDIALOG,Dlg_OnInitDialog); 

 chHANDLE_DLGMSG(hwnd,WM_COMMAND,Dlg_OnCommand); 

 chHANDLE_DLGMSG(hwnd,WM_SIZE,Dlg_OnSize); 

 chHANDLE_DLGMSG(hwnd,WM_GETMINMAXINFO,Dlg_OnGetMinMaxInfo); 

 } 

 return(FALSE); 

} 

/////////////////////////////////////////////////////////////////////////////// 

int WINAPI _tWinMain(HINSTANCE hinstExe,HINSTANCE,PTSTR pszCmdLine,int){ 

 DialogBox(hinstExe,MAKEINTRESOURCE(IDD_REGSCAN),NULL,Dlg_Proc); 

 return(0); 

} 

////////////////////////////////End of File //////////////////////////////////

RegWalk.h 

/******************************************************************** 

模组：RegWalk.h 

通告：Copyright (c)2000 Jeffrey Richter 

********************************************************************/ 

#pragma once  //每一次编译单元时即包含此标头档 

/////////////////////////////////////////////////////////////////////////////// 

#include  

/////////////////////////////////////////////////////////////////////////////// 

class CRegWalk { 

public: 

 CRegWalk(){} 

 virtual ~CRegWalk(){} 

 BOOL Go(PCTSTR pszMachine,HKEY hkeyRoot,PCTSTR pszSubkey,BOOL fRecurse); 

 enum REGWALKSTATUS {RWS_FULLSTOP,RWS_CONTINUE,RWS_RECURSE }; 

protected: 

 virtual REGWALKSTATUS onSubkey(PCTSTR pszSubkey,int nDepth, 

  BOOL fRecurseRequested); 

 virtual REGWALKSTATUS onValue(HKEY hkey,PCTSTR pszValue,int nDepth); 

protected: 

 HKEY m_hkeyRootMachine;  //机器上的根机码 

 BOOL m_fRecurse;    //递回至子机码中？ 

 int m_nDepth;      //递回深度 

 TCHAR m_szSubkeyPath [MAX_PATH ]; //子机码路径 

private: 

 REGWALKSTATUS RegWalkRecurse(); 

 REGWALKSTATUS EnumValuesInSubkey(); 

}; 

/////////////////////////////////////////////////////////////////////////////// 

#ifdef REGWALK_IMPL 

/////////////////////////////////////////////////////////////////////////////// 

#pragma comment(lib,"shlwapi ") 

/////////////////////////////////////////////////////////////////////////////// 

CRegWalk::REGWALKSTATUS CRegWalk::onSubkey(PCTSTR pszSubkey,int nDepth, 

 BOOL fRecurseRequested){ 

 return(fRecurseRequested ?RWS_RECURSE :RWS_CONTINUE); 

} 

CRegWalk::REGWALKSTATUS CRegWalk::onValue(HKEY hkey,PCTSTR pszValue, 

 int nDepth){ 

 return(RWS_CONTINUE); 

} 

/////////////////////////////////////////////////////////////////////////////// 

CRegWalk::REGWALKSTATUS CRegWalk::EnumValuesInSubkey(){ 

 HKEY hkey =NULL; 

 REGWALKSTATUS rws =RWS_CONTINUE; 

 if (ERROR_SUCCESS ==RegOpenKeyEx(m_hkeyRootMachine,m_szSubkeyPath,0, 

  KEY_QUERY_VALUE,&hkey)){ 

  for (int nIndex =0;rws !=RWS_FULLSTOP;nIndex++){ 

   TCHAR szValueName [256 ]; //没有超过255个字元的登录值名称 

   DWORD cbValueName =chDIMOF(szValueName); 

   if (ERROR_SUCCESS !=RegEnumValue(hkey,nIndex, 

    szValueName,&cbValueName,NULL,NULL,NULL,NULL)) 

    break; 

   rws =onValue(hkey,, szValueName,m_nDepth); 

  } 

  chVERIFY(RegCloseKey(hkey)==ERROR_SUCCESS); 

 } 

 return(rws); 

} 

/////////////////////////////////////////////////////////////////////////////// 

CRegWalk::REGWALKSTATUS CRegWalk::RegWalkRecurse(){ 

 //回报此机码 

 REGWALKSTATUS rws =onSubkey(m_szSubkeyPath,++m_nDepth,m_fRecurse); 

 //列举在子机码中的登录值？ 

 if (rws ==RWS_RECURSE)rws =EnumValuesInSubkey(); 

 //继续列举子机码？ 

 if (rws !=RWS_FULLSTOP){ 

  HKEY hkey =NULL; 

  if (ERROR_SUCCESS == RegOpenKeyEx(m_hkeyRootMachine,m_szSubkeyPath,0, 

  KEY_ENUMERATE_SUB_KEYS,&hkey)){ 

  for (int nIndex =0;rws !=RWS_FULLSTOP;nIndex++){ 

   TCHAR szSubkeyName [256 ]; //没有超过255个字元的子机码名称 

   DWORD cbSubkeyName =chDIMOF(szSubkeyName); 

   if (ERROR_SUCCESS !=RegEnumKeyEx(hkey,nIndex, 

    szSubkeyName,&cbSubkeyName,NULL,NULL,NULL,NULL)) 

    break; 

    //新增子机码至路径中 

    if (m_szSubkeyPath [0] !!=0)StrCat(m_szSubkeyPath,TEXT("\\")); 

StrCat(m_szSubkeyPath,szSubkeyName); 

rws =RegWalkRecurse(); 

    //从路径中截断最后的子机码 

    PTSTR p =StrRChr(m_szSubkeyPath,NULL,TEXT(’\\’)); 

    if (p !=NULL)*p =0; 

    else m_szSubkeyPath [0 ] =0; 

   } 

   chVERIFY(RegCloseKey(hkey)==ERROR_SUCCESS); 

  } 

 } 

 m_nDepth--; 

 return(rws); 

} 

/////////////////////////////////////////////////////////////////////////////// 

BOOL CRegWalk::Go(PCTSTR pszMachine,HKEY hkeyRoot,PCTSTR pszSubkey, 

 BOOL fRecurse){ 

 //nDepth指示了从上至下共有多少阶级 

 m_nDepth =-1; 

 m_fRecurse =fRecurse; 

 m_hkeyRootMachine =NULL; 

 REGWALKSTATUS rws =RWS_FULLSTOP; 

 __try { 

  if (ERROR_SUCCESS != 

   RegConnectRegistry(pszMachine,hkeyRoot,&m_hkeyRootMachine)) 

   __leave; 

  lstrcpy(m_szSubkeyPath,pszSubkey); 

  //呼叫递回函数开始沿着子机码走 

  rws =RegWalkRecurse(); 

 } 

 __finally { 

  if (m_hkeyRootMachine !=NULL) 

   RegCloseKey(m_hkeyRootMachine); 

 } 

 return(rws !=RWS_FULLSTOP); 

} 

/////////////////////////////////////////////////////////////////////////////// 

#endif //REGWALK_IMPL 

////////////////////////////////End of File //////////////////////////////////