Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
superpw.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 3k
Category:

Windows Kernel

Development Platform:

Visual C++

superpw.cpp：Code Content
  1. #include "msrating.h"
  2. #include "msluglob.h"
  3. #include <md5.h>
  5. extern HKEY CreateRegKeyNT(LPCSTR pszKey);
  6. extern BOOL RunningOnNT(void);
  8. HRESULT VerifySupervisorPassword(LPCSTR pszPassword)
  9. {
  10. if (!::fSupervisorKeyInit) {
  11. HKEY hkeyRating;
  12. LONG err;
  14. hkeyRating = CreateRegKeyNT(::szRATINGS);
  15. if (hkeyRating !=  NULL) {
  16. DWORD cbData = sizeof(::abSupervisorKey);
  17. DWORD dwType;
  18. err = ::RegQueryValueEx(hkeyRating, ::szRatingsSupervisorKeyName, NULL,
  19. &dwType, (LPBYTE)::abSupervisorKey, &cbData);
  20.             if (err != ERROR_SUCCESS) {
  21.                 if (pszPassword == NULL) {
  22.                     ::RegCloseKey(hkeyRating);
  23.                     return E_FAIL;
  24.                 }                    
  25.      err = ::RegQueryValueEx(hkeyRating, ::szUsersSupervisorKeyName, NULL,
  26.      &dwType, (LPBYTE)::abSupervisorKey, &cbData);
  27.             }
  28. ::RegCloseKey(hkeyRating);
  29. if (err == ERROR_SUCCESS) {
  30. if (dwType != REG_BINARY || cbData != sizeof(::abSupervisorKey)) {
  31. return E_UNEXPECTED;
  32. }
  33. ::fSupervisorKeyInit = TRUE;
  34. }
  35. }
  36. else
  37. err = ERROR_FILE_NOT_FOUND;
  39. if (err != ERROR_SUCCESS) {
  40. return HRESULT_FROM_WIN32(err);
  41. }
  42. }
  44. if (pszPassword == NULL)
  45. return ResultFromScode(S_FALSE);
  47. MD5_CTX ctx;
  49. MD5Init(&ctx);
  50. MD5Update(&ctx, (const BYTE *)pszPassword, ::strlenf(pszPassword)+1);
  51. MD5Final(&ctx);
  53. return ResultFromScode(::memcmpf(::abSupervisorKey, ctx.digest, sizeof(::abSupervisorKey)) ? S_FALSE : S_OK);
  54. }
  57. HRESULT ChangeSupervisorPassword(LPCSTR pszOldPassword, LPCSTR pszNewPassword)
  58. {
  59. HRESULT hres;
  61. hres = ::VerifySupervisorPassword(pszOldPassword);
  62. if (hres == S_FALSE) {
  63. return E_ACCESSDENIED;
  64. }
  66. MD5_CTX ctx;
  68. MD5Init(&ctx);
  69. MD5Update(&ctx, (const BYTE *)pszNewPassword, ::strlenf(pszNewPassword)+1);
  70. MD5Final(&ctx);
  72. ::memcpyf(::abSupervisorKey, ctx.digest, sizeof(::abSupervisorKey));
  73. ::fSupervisorKeyInit = TRUE;
  75.     hres = NOERROR;
  77. HKEY hkeyRating;
  79. hkeyRating = CreateRegKeyNT(::szRATINGS);
  80. if (hkeyRating != NULL) {
  81.         BYTE abTemp[sizeof(::abSupervisorKey)];
  82. DWORD cbData = sizeof(::abSupervisorKey);
  83. DWORD dwType;
  84. if (::RegQueryValueEx(hkeyRating, ::szRatingsSupervisorKeyName, NULL,
  85.   &dwType, abTemp, &cbData) != ERROR_SUCCESS)
  86.             hres = S_FALSE; /* tell caller we're creating the new key */
  88. ::RegSetValueEx(hkeyRating, ::szRatingsSupervisorKeyName, NULL,
  89. REG_BINARY, (const BYTE *)::abSupervisorKey, sizeof(::abSupervisorKey));
  90. ::RegCloseKey(hkeyRating);
  91. }
  93. return hres;
  94. }
  97. HRESULT RemoveSupervisorPassword(void)
  98. {
  99. HKEY hkeyRating;
  100. LONG err;
  102. hkeyRating = CreateRegKeyNT(::szRATINGS);
  103. if (hkeyRating !=  NULL) {
  104. err = ::RegDeleteValue(hkeyRating, ::szRatingsSupervisorKeyName);
  105. ::RegCloseKey(hkeyRating);
  106. }
  108. ::fSupervisorKeyInit = FALSE;
  109. return HRESULT_FROM_WIN32(err);
  110. }