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 "mslocusr.h"
  2. #include "msluglob.h"
  3. #include <md5.h>
  5. HRESULT VerifySupervisorPassword(LPCSTR pszPassword)
  6. {
  7. #ifdef MSLOCUSR_USE_SUPERVISOR_PASSWORD
  8. if (!::fSupervisorKeyInit) {
  9. HKEY hkeyRating;
  10. LONG err;
  12. err = RegOpenKey(HKEY_LOCAL_MACHINE, ::szRATINGS, &hkeyRating);
  13. if (hkeyRating !=  NULL) {
  14. DWORD cbData = sizeof(::abSupervisorKey);
  15. DWORD dwType;
  16. err = ::RegQueryValueEx(hkeyRating, ::szRatingsSupervisorKeyName, NULL,
  17. &dwType, (LPBYTE)::abSupervisorKey, &cbData);
  18.             if (err != ERROR_SUCCESS)
  19.      err = ::RegQueryValueEx(hkeyRating, ::szUsersSupervisorKeyName, NULL,
  20.      &dwType, (LPBYTE)::abSupervisorKey, &cbData);
  21. ::RegCloseKey(hkeyRating);
  22. if (err == ERROR_SUCCESS) {
  23. if (dwType != REG_BINARY || cbData != sizeof(::abSupervisorKey)) {
  24. return E_UNEXPECTED;
  25. }
  26. ::fSupervisorKeyInit = TRUE;
  27. }
  28. }
  29. else
  30. err = ERROR_FILE_NOT_FOUND;
  32. if (err != ERROR_SUCCESS) {
  33. return HRESULT_FROM_WIN32(err);
  34. }
  35. }
  37. if (pszPassword == NULL)
  38. return ResultFromScode(S_FALSE);
  40. MD5_CTX ctx;
  42. MD5Init(&ctx);
  43. MD5Update(&ctx, (const BYTE *)pszPassword, ::strlenf(pszPassword)+1);
  44. MD5Final(&ctx);
  46. return ResultFromScode(::memcmpf(::abSupervisorKey, ctx.digest, sizeof(::abSupervisorKey)) ? S_FALSE : S_OK);
  48. #else
  50.     return S_OK;        /* everybody's a supervisor */
  52. #endif
  53. }
  56. HRESULT ChangeSupervisorPassword(LPCSTR pszOldPassword, LPCSTR pszNewPassword)
  57. {
  58. #ifdef MSLOCUSR_USE_SUPERVISOR_PASSWORD
  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. HKEY hkeyRating;
  77. LONG err = RegOpenKey(HKEY_LOCAL_MACHINE, ::szRATINGS, &hkeyRating);
  78. if (err == ERROR_SUCCESS) {
  79.         char abTemp[sizeof(::abSupervisorKey)];
  80.         LPCSTR pszValueToSet;
  81.         DWORD dwType;
  82.         DWORD cbData = sizeof(abTemp);
  83.         if (::RegQueryValueEx(hkeyRating, ::szRatingsSupervisorKeyName, NULL,
  84.   &dwType, (LPBYTE)abTemp, &cbData) == ERROR_SUCCESS)
  85.             pszValueToSet = ::szRatingsSupervisorKeyName;
  86.         else
  87.             pszValueToSet = ::szUsersSupervisorKeyName;
  89. ::RegSetValueEx(hkeyRating, pszValueToSet, NULL,
  90. REG_BINARY, (const BYTE *)::abSupervisorKey, sizeof(::abSupervisorKey));
  91. ::RegCloseKey(hkeyRating);
  92. }
  93. #endif
  95. return NOERROR;
  96. }