开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
UTILITY.CPP
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：12k
源码类别：

Windows编程

开发平台：

Visual C++

UTILITY.CPP：源码内容
  1. /**************************************************************************
  2.    THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
  3.    ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
  4.    THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
  5.    PARTICULAR PURPOSE.
  7.    Copyright 1997 Microsoft Corporation.  All Rights Reserved.
  8. **************************************************************************/
  10. /**************************************************************************
  12.    File:          Utility.cpp
  13.    
  14.    Description:   
  16. **************************************************************************/
  18. /**************************************************************************
  19.    #include statements
  20. **************************************************************************/
  22. #include "Utility.h"
  23. #include "ShlFldr.h"
  24. #include "resource.h"
  26. /**************************************************************************
  27.    private function prototypes
  28. **************************************************************************/
  30. /**************************************************************************
  31.    global variables
  32. **************************************************************************/
  34. extern HIMAGELIST g_himlLarge;
  35. extern HIMAGELIST g_himlSmall;
  37. #define MAIN_KEY_STRING       (TEXT("Software\RegView"))
  38. #define VALUE_STRING          (TEXT("Display Settings"))
  39. #define INITIAL_COLUMN_SIZE   100
  41. /**************************************************************************
  43.    GetKeyName()
  44.    
  45. **************************************************************************/
  47. DWORD GetKeyName( HKEY hKeyRoot, 
  48.                   LPCTSTR pszSubKey, 
  49.                   DWORD dwIndex, 
  50.                   LPTSTR pszOut, 
  51.                   DWORD dwSize)
  52. {
  53. HKEY     hKey;
  54. LONG     lResult;
  55. FILETIME ft;
  57. if(!pszOut)
  58.    return 0;
  60. if(!pszSubKey)
  61.    pszSubKey = TEXT("");
  63. //open the specified key
  64. lResult = RegOpenKeyEx( hKeyRoot,
  65.                         pszSubKey,
  66.                         0,
  67.                         KEY_ENUMERATE_SUB_KEYS,
  68.                         &hKey);
  70. if(ERROR_SUCCESS != lResult)
  71.    return 0;
  73. //try to get the specified subkey
  74. lResult = RegEnumKeyEx( hKey,
  75.                         dwIndex,
  76.                         pszOut,
  77.                         &dwSize,
  78.                         NULL,
  79.                         NULL,
  80.                         NULL,
  81.                         &ft);
  83. RegCloseKey(hKey);
  85. if(ERROR_SUCCESS == lResult)
  86.    return dwSize;
  88. return 0;
  89. }
  91. /**************************************************************************
  93.    GetValueName()
  94.    
  95. **************************************************************************/
  97. BOOL GetValueName(   HKEY hKeyRoot, 
  98.                      LPCTSTR pszSubKey, 
  99.                      DWORD dwIndex, 
  100.                      LPTSTR pszOut, 
  101.                      DWORD dwSize)
  102. {
  103. HKEY     hKey;
  104. LONG     lResult;
  105. DWORD    dwType;
  107. if(!pszOut)
  108.    return 0;
  110. if(!pszSubKey)
  111.    pszSubKey = TEXT("");
  113. //open the specified key
  114. lResult = RegOpenKeyEx( hKeyRoot,
  115.                         pszSubKey,
  116.                         0,
  117.                         KEY_QUERY_VALUE,
  118.                         &hKey);
  120. if(ERROR_SUCCESS != lResult)
  121.    return 0;
  123. //try to get the specified subkey
  124. lResult = RegEnumValue( hKey,
  125.                         dwIndex,
  126.                         pszOut,
  127.                         &dwSize,
  128.                         NULL,
  129.                         &dwType,
  130.                         NULL,
  131.                         NULL);
  133. RegCloseKey(hKey);
  135. return BOOL(ERROR_SUCCESS == lResult);
  137. if(ERROR_SUCCESS == lResult)
  138.    return dwSize;
  140. return 0;
  141. }
  144. /**************************************************************************
  146.    GetRootKeyText()
  147.    
  148. **************************************************************************/
  150. DWORD GetRootKeyText(HKEY hKeyRoot, LPTSTR lpszOut, DWORD dwOutSize)
  151. {
  152. *lpszOut = 0;
  154. if(hKeyRoot == HKEY_CLASSES_ROOT)
  155.    {
  156.    lstrcpyn(lpszOut, TEXT("HKEY_CLASSES_ROOT"), dwOutSize);
  157.    }
  159. else if(hKeyRoot == HKEY_CURRENT_USER)
  160.    {
  161.    lstrcpyn(lpszOut, TEXT("HKEY_CURRENT_USER"), dwOutSize);
  162.    }
  164. else if(hKeyRoot == HKEY_LOCAL_MACHINE)
  165.    {
  166.    lstrcpyn(lpszOut, TEXT("HKEY_LOCAL_MACHINE"), dwOutSize);
  167.    }
  169. else if(hKeyRoot == HKEY_USERS)
  170.    {
  171.    lstrcpyn(lpszOut, TEXT("HKEY_USERS"), dwOutSize);
  172.    }
  174. else if(hKeyRoot == HKEY_PERFORMANCE_DATA)
  175.    {
  176.    lstrcpyn(lpszOut, TEXT("HKEY_PERFORMANCE_DATA"), dwOutSize);
  177.    }
  179. else if(hKeyRoot == HKEY_CURRENT_CONFIG)
  180.    {
  181.    lstrcpyn(lpszOut, TEXT("HKEY_CURRENT_CONFIG"), dwOutSize);
  182.    }
  184. else if(hKeyRoot == HKEY_DYN_DATA)
  185.    {
  186.    lstrcpyn(lpszOut, TEXT("HKEY_DYN_DATA"), dwOutSize);
  187.    }
  188.    
  189. return lstrlen(lpszOut) + 1;
  190. }
  192. /**************************************************************************
  194.    RootKeyExists()
  195.    
  196. **************************************************************************/
  198. BOOL RootKeyExists(HKEY hKeyRoot)
  199. {
  200. LONG  lResult;
  201. HKEY  hKey;
  203. //open the specified key
  204. lResult = RegOpenKeyEx( hKeyRoot,
  205.                         NULL,
  206.                         0,
  207.                         KEY_ENUMERATE_SUB_KEYS,
  208.                         &hKey);
  210. if(ERROR_SUCCESS != lResult)
  211.    return FALSE;
  213. RegCloseKey(hKey);
  215. return TRUE;
  216. }
  218. /**************************************************************************
  220.    SaveGlobalSettings()
  221.    
  222. **************************************************************************/
  224. BOOL SaveGlobalSettings(void)
  225. {
  226. HKEY  hKey;
  227. LONG  lResult;
  228. DWORD dwDisp;
  230. lResult = RegCreateKeyEx(  HKEY_CURRENT_USER,
  231.                            MAIN_KEY_STRING,
  232.                            0,
  233.                            NULL,
  234.                            REG_OPTION_NON_VOLATILE, 
  235.                            KEY_ALL_ACCESS,
  236.                            NULL, 
  237.                            &hKey,
  238.                            &dwDisp);
  240. if(lResult != ERROR_SUCCESS)
  241.    return FALSE;
  243. //create an array to put our data in
  244. DWORD dwArray[4];
  245. dwArray[0] = g_nColumn1;
  246. dwArray[1] = g_nColumn2;
  247. dwArray[2] = g_bViewKeys;
  248. dwArray[3] = g_bShowIDW;
  250. //save the last printer selected
  251. lResult = RegSetValueEx(   hKey,
  252.                            VALUE_STRING,
  253.                            0,
  254.                            REG_BINARY,
  255.                            (LPBYTE)dwArray,
  256.                            sizeof(dwArray));
  258. RegCloseKey(hKey);
  260. if(lResult != ERROR_SUCCESS)
  261.    return FALSE;
  263. return TRUE;
  264. }
  266. /**************************************************************************
  268.    GetGlobalSettings()
  269.    
  270. **************************************************************************/
  272. BOOL GetGlobalSettings(void)
  273. {
  274. HKEY  hKey;
  275. LONG  lResult;
  277. //set up the default data
  278. g_nColumn1 = INITIAL_COLUMN_SIZE;
  279. g_nColumn2 = INITIAL_COLUMN_SIZE;
  280. g_bViewKeys = TRUE;
  281. g_bShowIDW = FALSE;
  283. lResult = RegOpenKeyEx( HKEY_CURRENT_USER,
  284.                         MAIN_KEY_STRING,
  285.                         0,
  286.                         KEY_ALL_ACCESS,
  287.                         &hKey);
  289. if(lResult != ERROR_SUCCESS)
  290.    return FALSE;
  292. //create an array to put our data in
  293. DWORD dwArray[4];
  294. DWORD dwType;
  295. DWORD dwSize = sizeof(dwArray);
  297. //save the last printer selected
  298. lResult = RegQueryValueEx( hKey,
  299.                            VALUE_STRING,
  300.                            NULL,
  301.                            &dwType,
  302.                            (LPBYTE)dwArray,
  303.                            &dwSize);
  305. RegCloseKey(hKey);
  307. if(lResult != ERROR_SUCCESS)
  308.    return FALSE;
  310. g_nColumn1 = dwArray[0];
  311. g_nColumn2 = dwArray[1];
  312. g_bViewKeys = dwArray[2];
  313. g_bShowIDW = dwArray[3];
  315. return TRUE;
  316. }
  318. /**************************************************************************
  320.    CompareItems()
  321.    
  322. **************************************************************************/
  324. int CALLBACK CompareItems(LPARAM lParam1, LPARAM lParam2, LPARAM lpData)
  325. {
  326. CShellFolder  *pFolder = (CShellFolder*)lpData;
  328. if(!pFolder)
  329.    return 0;
  331. return (int)pFolder->CompareIDs(0, (LPITEMIDLIST)lParam1, (LPITEMIDLIST)lParam2);
  332. }
  334. /**************************************************************************
  336.    CreateImageLists()
  337.    
  338. **************************************************************************/
  340. VOID CreateImageLists(VOID)
  341. {
  342. int   cx;
  343. int   cy;
  345. cx = GetSystemMetrics(SM_CXSMICON);
  346. cy = GetSystemMetrics(SM_CYSMICON);
  348. cx = 16;
  349. cy = 16;
  351. if(g_himlSmall)
  352.    ImageList_Destroy(g_himlSmall);
  354. //set the small image list
  355. g_himlSmall = ImageList_Create(cx, cy, ILC_COLORDDB | ILC_MASK, 4, 0);
  357. if(g_himlSmall)
  358.    {
  359.    HICON hIcon;
  360.    
  361.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_BINARY), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  362.    ImageList_AddIcon(g_himlSmall, hIcon);
  364.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_STRING), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  365.    ImageList_AddIcon(g_himlSmall, hIcon);
  367.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_FOLDER), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  368.    ImageList_AddIcon(g_himlSmall, hIcon);
  370.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_FOLDEROPEN), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  371.    ImageList_AddIcon(g_himlSmall, hIcon);
  372.    }
  374. if(g_himlLarge)
  375.    ImageList_Destroy(g_himlLarge);
  377. cx = GetSystemMetrics(SM_CXICON);
  378. cy = GetSystemMetrics(SM_CYICON);
  380. cx = 32;
  381. cy = 32;
  383. //set the large image list
  384. g_himlLarge = ImageList_Create(cx, cy, ILC_COLORDDB | ILC_MASK, 4, 0);
  386. if(g_himlSmall)
  387.    {
  388.    HICON hIcon;
  389.    
  390.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_BINARY), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  391.    ImageList_AddIcon(g_himlLarge, hIcon);
  393.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_STRING), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  394.    ImageList_AddIcon(g_himlLarge, hIcon);
  396.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_FOLDER), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  397.    ImageList_AddIcon(g_himlLarge, hIcon);
  399.    hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_FOLDEROPEN), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
  400.    ImageList_AddIcon(g_himlLarge, hIcon);
  401.    }
  403. }
  405. /**************************************************************************
  407.    DestroyImageLists()
  408.    
  409. **************************************************************************/
  411. VOID DestroyImageLists(VOID)
  412. {
  413. if(g_himlSmall)
  414.    ImageList_Destroy(g_himlSmall);
  416. if(g_himlLarge)
  417.    ImageList_Destroy(g_himlLarge);
  418. }
  420. /**************************************************************************
  422.    WideCharToLocal()
  423.    
  424. **************************************************************************/
  426. int WideCharToLocal(LPTSTR pLocal, LPWSTR pWide, DWORD dwChars)
  427. {
  428. *pLocal = 0;
  430. #ifdef UNICODE
  431. lstrcpyn(pLocal, pWide, dwChars);
  432. #else
  433. WideCharToMultiByte( CP_ACP, 
  434.                      0, 
  435.                      pWide, 
  436.                      -1, 
  437.                      pLocal, 
  438. dwChars, 
  439.                      NULL, 
  440.                      NULL);
  441. #endif
  443. return lstrlen(pLocal);
  444. }
  446. /**************************************************************************
  448.    LocalToWideChar()
  449.    
  450. **************************************************************************/
  452. int LocalToWideChar(LPWSTR pWide, LPTSTR pLocal, DWORD dwChars)
  453. {
  454. *pWide = 0;
  456. #ifdef UNICODE
  457. lstrcpyn(pWide, pLocal, dwChars);
  458. #else
  459. MultiByteToWideChar( CP_ACP, 
  460.                      0, 
  461.                      pLocal, 
  462.                      -1, 
  463.                      pWide, 
  464.                      dwChars); 
  465. #endif
  467. return lstrlenW(pWide);
  468. }