setuserobjectsecurity.py 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import win32security,win32api,win32con, win32process
  2. new_privs = ((win32security.LookupPrivilegeValue('',win32security.SE_SECURITY_NAME),win32con.SE_PRIVILEGE_ENABLED),
  3. (win32security.LookupPrivilegeValue('',win32security.SE_TCB_NAME),win32con.SE_PRIVILEGE_ENABLED),
  4. (win32security.LookupPrivilegeValue('',win32security.SE_SHUTDOWN_NAME),win32con.SE_PRIVILEGE_ENABLED),
  5. (win32security.LookupPrivilegeValue('',win32security.SE_RESTORE_NAME),win32con.SE_PRIVILEGE_ENABLED),
  6. (win32security.LookupPrivilegeValue('',win32security.SE_TAKE_OWNERSHIP_NAME),win32con.SE_PRIVILEGE_ENABLED),
  7. (win32security.LookupPrivilegeValue('',win32security.SE_CREATE_PERMANENT_NAME),win32con.SE_PRIVILEGE_ENABLED),
  8. (win32security.LookupPrivilegeValue('',win32security.SE_ENABLE_DELEGATION_NAME),win32con.SE_PRIVILEGE_ENABLED),
  9. (win32security.LookupPrivilegeValue('',win32security.SE_CHANGE_NOTIFY_NAME),win32con.SE_PRIVILEGE_ENABLED),
  10. (win32security.LookupPrivilegeValue('',win32security.SE_DEBUG_NAME),win32con.SE_PRIVILEGE_ENABLED),
  11. (win32security.LookupPrivilegeValue('',win32security.SE_PROF_SINGLE_PROCESS_NAME),win32con.SE_PRIVILEGE_ENABLED),
  12. (win32security.LookupPrivilegeValue('',win32security.SE_SYSTEM_PROFILE_NAME),win32con.SE_PRIVILEGE_ENABLED),
  13. (win32security.LookupPrivilegeValue('',win32security.SE_LOCK_MEMORY_NAME),win32con.SE_PRIVILEGE_ENABLED)
  14. )
  15. all_info=win32security.OWNER_SECURITY_INFORMATION|win32security.GROUP_SECURITY_INFORMATION| \
  16. win32security.DACL_SECURITY_INFORMATION|win32security.SACL_SECURITY_INFORMATION
  17. info=win32security.OWNER_SECURITY_INFORMATION|win32security.GROUP_SECURITY_INFORMATION|win32security.DACL_SECURITY_INFORMATION
  18. ph=win32process.GetCurrentProcess()
  19. th = win32security.OpenProcessToken(ph,win32security.TOKEN_ALL_ACCESS) ##win32con.TOKEN_ADJUST_PRIVILEGES)
  20. win32security.AdjustTokenPrivileges(th,0,new_privs)
  21. my_sid = win32security.GetTokenInformation(th,win32security.TokenUser)[0]
  22. pwr_sid=win32security.LookupAccountName('','Power Users')[0]
  23. h=win32process.GetProcessWindowStation()
  24. sd=win32security.GetUserObjectSecurity(h,info)
  25. dacl=sd.GetSecurityDescriptorDacl()
  26. ace_cnt=dacl.GetAceCount()
  27. dacl.AddAccessAllowedAce(dacl.GetAclRevision(),win32con.ACCESS_SYSTEM_SECURITY|win32con.WRITE_DAC,my_sid)
  28. sd.SetSecurityDescriptorDacl(1,dacl,0)
  29. sd.SetSecurityDescriptorGroup(pwr_sid,0)
  30. sd.SetSecurityDescriptorOwner(pwr_sid,0)
  31. win32security.SetUserObjectSecurity(h,info,sd)
  32. new_sd=win32security.GetUserObjectSecurity(h,info)
  33. assert new_sd.GetSecurityDescriptorDacl().GetAceCount()==ace_cnt+1,'Did not add an ace to the Dacl !!!!!!'
  34. assert win32security.LookupAccountSid('',new_sd.GetSecurityDescriptorOwner())[0]=='Power Users','Owner not successfully set to Power Users !!!!!'
  35. assert win32security.LookupAccountSid('',new_sd.GetSecurityDescriptorGroup())[0]=='Power Users','Group not successfully set to Power Users !!!!!'