Changeset 121 for trunk/Utilities/Miscellaneous/SetFileOwner

Timestamp:

Sep 9, 2012, 8:35:58 AM (11 years ago)

Author:

roman

Message:

 

Location:

trunk/Utilities/Miscellaneous/SetFileOwner

Files:

• . (modified) (1 prop)
• SetFileOwner.cpp (modified) (2 diffs)

trunk/Utilities/Miscellaneous/SetFileOwner

@@ -1 +1 @@
 *.suo
+Debug
+ipch
+*.sdf

@@ -1 +1 @@
 *.suo
+Debug
+ipch
+*.sdf

trunk/Utilities/Miscellaneous/SetFileOwner/SetFileOwner.cpp

@@ -10 +10 @@
 #include <atlfile.h>
 
-VOID PrintOwner(LPCTSTR pszPath)
+VOID EnablePrivilege(LPCTSTR pszPrivilege)
 {
-        PSID pSecurityIdentifer = NULL;
-        ATLENSURE_SUCCEEDED(HRESULT_FROM_WIN32(GetNamedSecurityInfo(const_cast<LPTSTR>(pszPath), SE_FILE_OBJECT, OWNER_SECURITY_INFORMATION, &pSecurityIdentifer, NULL, NULL, NULL, NULL)));
-        CSid SecurityIdentifer((const SID*) pSecurityIdentifer);
-        _tprintf(_T("Owner: %s\\%s\n"), SecurityIdentifer.Domain(), SecurityIdentifer.AccountName());
-}
-
-int _tmain(int argc, _TCHAR* argv[])
-{
-        #pragma region Enable SeRestorePrivilege
         CAccessToken Token;
         if(!Token.GetThreadToken(TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES))
@@ -28 +19 @@
                 ATLENSURE_THROW(Token.GetProcessToken(TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES), AtlHresultFromLastError());
         }
-        ATLENSURE_THROW(Token.EnablePrivilege(SE_RESTORE_NAME), AtlHresultFromLastError());
-        #pragma endregion
+        ATLENSURE_THROW(Token.EnablePrivilege(pszPrivilege), AtlHresultFromLastError());
+}
+VOID PrintOwner(LPCTSTR pszPath)
+{
+        PSID pSecurityIdentifer = NULL;
+        CSecurityDesc SecurityDescriptor;
+        ATLENSURE_SUCCEEDED(HRESULT_FROM_WIN32(GetNamedSecurityInfo(const_cast<LPTSTR>(pszPath), SE_FILE_OBJECT, OWNER_SECURITY_INFORMATION, &pSecurityIdentifer, NULL, NULL, NULL, reinterpret_cast<PSECURITY_DESCRIPTOR*>(&SecurityDescriptor))));
+        CSid SecurityIdentifer((const SID*) pSecurityIdentifer);
+        _tprintf(_T("Owner: %s\\%s\n"), SecurityIdentifer.Domain(), SecurityIdentifer.AccountName());
+}
+VOID SetOwner(LPCTSTR pszPath, const CSid& SecurityIdentifer)
+{
+        ATLENSURE_SUCCEEDED(HRESULT_FROM_WIN32(SetNamedSecurityInfo(const_cast<LPTSTR>(pszPath), SE_FILE_OBJECT, OWNER_SECURITY_INFORMATION, const_cast<SID*>((const SID*) SecurityIdentifer), NULL, NULL, NULL)));
+}
+
+int _tmain(int argc, _TCHAR* argv[])
+{
         static LPCTSTR g_pszPath = _T("D:\\Projects\\Alax.Info\\Repository-Public\\Utilities\\Miscellaneous\\SetFileOwner\\ReadMe.txt");
+        EnablePrivilege(SE_RESTORE_NAME);
         PrintOwner(g_pszPath);
         CSid SecurityIdentifer;
         ATLENSURE_THROW(SecurityIdentifer.LoadAccount(_T("Guest")), AtlHresultFromLastError());
         _tprintf(_T("New Owner: %s\\%s\n"), SecurityIdentifer.Domain(), SecurityIdentifer.AccountName());
-        ATLENSURE_SUCCEEDED(HRESULT_FROM_WIN32(SetNamedSecurityInfo(const_cast<LPTSTR>(g_pszPath), SE_FILE_OBJECT, OWNER_SECURITY_INFORMATION, const_cast<SID*>((const SID*) SecurityIdentifer), NULL, NULL, NULL)));
+        _ATLTRY
+        {
+                SetOwner(g_pszPath, SecurityIdentifer);
+        }
+        _ATLCATCH(Exception)
+        {
+                _tprintf(_T("Exception 0x%08x\n"), (HRESULT) Exception);
+                // NOTE: This might be the worst case scenario: changing ownership is perhaps protected by DACL
+                EnablePrivilege(SE_TAKE_OWNERSHIP_NAME);
+                SetOwner(g_pszPath, SecurityIdentifer);
+        }
         PrintOwner(g_pszPath);
         return 0;

@@ -1 +1 @@
 *.suo
+Debug
+ipch
+*.sdf