|
@@ -37,7 +37,7 @@ cdef extern from 'windows.h':
|
|
|
ctypedef void* HANDLE
|
|
|
struct _ACL:
|
|
|
uint16_t AceCount
|
|
|
-
|
|
|
+
|
|
|
cdef enum _SID_NAME_USE:
|
|
|
SidTypeUser,
|
|
|
SidTypeGroup,
|
|
@@ -59,8 +59,19 @@ cdef extern from 'windows.h':
|
|
|
_LARGE_INTEGER StreamSize
|
|
|
wchar_t[296] cStreamName # MAX_PATH + 36
|
|
|
|
|
|
+ struct _LUID:
|
|
|
+ pass
|
|
|
+
|
|
|
+ struct _LUID_AND_ATTRIBUTES:
|
|
|
+ _LUID Luid
|
|
|
+ DWORD Attributes
|
|
|
+
|
|
|
+ struct _TOKEN_PRIVILEGES:
|
|
|
+ DWORD PrivilegeCount
|
|
|
+ _LUID_AND_ATTRIBUTES Privileges[1]
|
|
|
+
|
|
|
HLOCAL LocalFree(HLOCAL)
|
|
|
- DWORD GetLastError();
|
|
|
+ DWORD GetLastError()
|
|
|
void SetLastError(DWORD)
|
|
|
|
|
|
DWORD FormatMessageW(DWORD, void*, DWORD, DWORD, wchar_t*, DWORD, void*)
|
|
@@ -74,6 +85,15 @@ cdef extern from 'windows.h':
|
|
|
BOOL LookupAccountNameW(LPCTSTR, LPCTSTR, PSID, LPDWORD, LPCTSTR, LPDWORD, _SID_NAME_USE*)
|
|
|
BOOL GetSecurityDescriptorDacl(PSID, BOOL*, _ACL**, BOOL*)
|
|
|
|
|
|
+ BOOL OpenProcessToken(HANDLE, DWORD, HANDLE*)
|
|
|
+ BOOL OpenThreadToken(HANDLE, DWORD, BOOL, HANDLE*)
|
|
|
+ BOOL LookupPrivilegeValueW(wchar_t*, wchar_t*, _LUID*)
|
|
|
+ BOOL AdjustTokenPrivileges(HANDLE, BOOL, _TOKEN_PRIVILEGES*, DWORD, _TOKEN_PRIVILEGES*, DWORD*)
|
|
|
+
|
|
|
+ HANDLE GetCurrentThread()
|
|
|
+ HANDLE GetCurrentProcess()
|
|
|
+
|
|
|
+ cdef extern int ERROR_SUCCESS
|
|
|
cdef extern int ERROR_INSUFFICIENT_BUFFER
|
|
|
cdef extern int ERROR_INVALID_SID
|
|
|
cdef extern int ERROR_NONE_MAPPED
|
|
@@ -100,6 +120,11 @@ cdef extern from 'windows.h':
|
|
|
|
|
|
cdef extern int INVALID_HANDLE_VALUE
|
|
|
|
|
|
+ cdef extern DWORD SE_PRIVILEGE_ENABLED
|
|
|
+
|
|
|
+ cdef extern int TOKEN_ADJUST_PRIVILEGES
|
|
|
+ cdef extern int TOKEN_QUERY
|
|
|
+
|
|
|
cdef extern from 'accctrl.h':
|
|
|
ctypedef enum _SE_OBJECT_TYPE:
|
|
|
SE_FILE_OBJECT
|
|
@@ -119,6 +144,11 @@ cdef extern from 'accctrl.h':
|
|
|
cdef extern uint16_t TRUSTEE_IS_NAME
|
|
|
cdef extern uint16_t TRUSTEE_BAD_FORM
|
|
|
|
|
|
+ cdef extern int NO_INHERITANCE
|
|
|
+ cdef extern int INHERIT_NO_PROPAGATE
|
|
|
+ cdef extern int INHERIT_ONLY
|
|
|
+ cdef extern int INHERITED_ACCESS_ENTRY
|
|
|
+
|
|
|
DWORD GetExplicitEntriesFromAclW(_ACL*, uint32_t*, _EXPLICIT_ACCESS_W**)
|
|
|
|
|
|
|
|
@@ -140,6 +170,7 @@ cdef extern from 'Aclapi.h':
|
|
|
DWORD GetNamedSecurityInfoW(LPCTSTR, SE_OBJECT_TYPE, DWORD, PSID*, PSID*, PACL*, PACL*, _ACL**)
|
|
|
DWORD SetNamedSecurityInfoW(LPCTSTR, int, int, PSID, PSID, PACL, PACL)
|
|
|
DWORD SetEntriesInAclW(unsigned int, _EXPLICIT_ACCESS_W*, PACL, _ACL**)
|
|
|
+ DWORD LookupSecurityDescriptorPartsW(_TRUSTEE_W**, _TRUSTEE_W**, uint32_t*, _EXPLICIT_ACCESS_W**, uint32_t*, _EXPLICIT_ACCESS_W**, PSID)
|
|
|
|
|
|
|
|
|
def raise_error(api, path=''):
|
|
@@ -158,6 +189,52 @@ def raise_error(api, path=''):
|
|
|
raise OSError(error, error_string)
|
|
|
|
|
|
|
|
|
+permissions_enabled = False # Have we tried to acquire permissions for SACL
|
|
|
+permissions_granted = False # Did we get them
|
|
|
+
|
|
|
+
|
|
|
+cdef enable_permissions():
|
|
|
+ global permissions_enabled
|
|
|
+ global permissions_granted
|
|
|
+ if permissions_enabled:
|
|
|
+ return
|
|
|
+ permissions_enabled = True
|
|
|
+ cdef HANDLE hToken
|
|
|
+ OpenProcessToken(GetCurrentProcess() , TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)
|
|
|
+
|
|
|
+ cdef _TOKEN_PRIVILEGES tp
|
|
|
+ cdef _LUID luid
|
|
|
+ cdef _TOKEN_PRIVILEGES tpPrevious
|
|
|
+ cdef DWORD cbPrevious=sizeof(_TOKEN_PRIVILEGES)
|
|
|
+
|
|
|
+ cdef wchar_t* privilege = PyUnicode_AsWideCharString("SeSecurityPrivilege", NULL)
|
|
|
+ if not LookupPrivilegeValueW( NULL, privilege, &luid ):
|
|
|
+ permissions_granted = False
|
|
|
+ print("Warning: permissions to read auditing settings (SACL) denied. Try running as admin.")
|
|
|
+ return
|
|
|
+
|
|
|
+ tp.PrivilegeCount = 1
|
|
|
+ tp.Privileges[0].Luid = luid
|
|
|
+ tp.Privileges[0].Attributes = 0
|
|
|
+
|
|
|
+ AdjustTokenPrivileges(hToken, 0, &tp, sizeof(_TOKEN_PRIVILEGES), &tpPrevious, &cbPrevious)
|
|
|
+ if GetLastError() != ERROR_SUCCESS:
|
|
|
+ permissions_granted = False
|
|
|
+ print("Warning: permissions to read auditing settings (SACL) denied. Try running as admin.")
|
|
|
+ return
|
|
|
+
|
|
|
+ tpPrevious.PrivilegeCount = 1
|
|
|
+ tpPrevious.Privileges[0].Luid = luid
|
|
|
+ tpPrevious.Privileges[0].Attributes = tpPrevious.Privileges[0].Attributes | SE_PRIVILEGE_ENABLED
|
|
|
+
|
|
|
+ AdjustTokenPrivileges(hToken, 0, &tpPrevious, cbPrevious, NULL, NULL)
|
|
|
+
|
|
|
+ if GetLastError() != ERROR_SUCCESS:
|
|
|
+ permissions_granted = False
|
|
|
+ print("Warning: permissions to read auditing settings (SACL) denied. Try running as admin.")
|
|
|
+ return
|
|
|
+
|
|
|
+
|
|
|
cdef PSID _get_file_security(filename, int request):
|
|
|
cdef DWORD length = 0
|
|
|
# N.B. This query may fail with ERROR_INVALID_FUNCTION
|
|
@@ -267,93 +344,181 @@ def set_owner(path, owner, sidstring = None):
|
|
|
free(newsid)
|
|
|
|
|
|
|
|
|
-def acl_get(path, item, st, numeric_owner=False):
|
|
|
- cdef int request = DACL_SECURITY_INFORMATION
|
|
|
+def acl_get(path, item, st, numeric_owner=False, depth = 0):
|
|
|
+ if not permissions_enabled:
|
|
|
+ enable_permissions()
|
|
|
+ pyDACL = []
|
|
|
+ pySACL = []
|
|
|
+ if not os.path.samefile(os.path.abspath(path), os.path.abspath(os.path.join(path, ".."))):
|
|
|
+ pyDACL, pySACL = acl_get(os.path.abspath(os.path.join(path, "..")), item, st, numeric_owner, depth + 1)
|
|
|
|
|
|
+ cdef int request = DACL_SECURITY_INFORMATION
|
|
|
+ if permissions_granted:
|
|
|
+ request = DACL_SECURITY_INFORMATION | SACL_SECURITY_INFORMATION
|
|
|
cdef BYTE* SD = _get_file_security(path, request)
|
|
|
if SD == NULL:
|
|
|
return
|
|
|
|
|
|
- cdef BOOL daclFound
|
|
|
- cdef _ACL* DACL
|
|
|
- cdef BOOL DACLDefaulted
|
|
|
- GetSecurityDescriptorDacl(SD, &daclFound, &DACL, &DACLDefaulted)
|
|
|
-
|
|
|
- cdef uint32_t length
|
|
|
- cdef _EXPLICIT_ACCESS_W* ACEs
|
|
|
+ cdef uint32_t dacllength
|
|
|
+ cdef _EXPLICIT_ACCESS_W* DACL
|
|
|
+ cdef uint32_t sacllength
|
|
|
+ cdef _EXPLICIT_ACCESS_W* SACL
|
|
|
|
|
|
- GetExplicitEntriesFromAclW(DACL, &length, &ACEs)
|
|
|
+ # LookupSecurityDescriptorPartsW(&owner, &group, &dacllength, &DACL, &sacllength, &sacl, SD)
|
|
|
+ LookupSecurityDescriptorPartsW(NULL, NULL, &dacllength, &DACL, &sacllength, &SACL, SD)
|
|
|
|
|
|
- pyDACL = []
|
|
|
cdef PSID newsid
|
|
|
cdef uint32_t domainlength
|
|
|
+ cdef uint32_t sidlength
|
|
|
cdef _SID_NAME_USE sid_type
|
|
|
- for i in range(length):
|
|
|
+
|
|
|
+ # DACL
|
|
|
+ for i in range(dacllength):
|
|
|
permissions = None
|
|
|
name = ""
|
|
|
sidstr = ""
|
|
|
- if ACEs[i].Trustee.TrusteeForm == TRUSTEE_IS_SID:
|
|
|
- name, domain, type = _look_up_account_sid(<BYTE*>(ACEs[i].Trustee.ptstrName))
|
|
|
- sidstr = sid2string(<PSID>(ACEs[i].Trustee.ptstrName))
|
|
|
+ if DACL[i].Trustee.TrusteeForm == TRUSTEE_IS_SID:
|
|
|
+ name, domain, type = _look_up_account_sid(<BYTE*>(DACL[i].Trustee.ptstrName))
|
|
|
+ sidstr = sid2string(<PSID>(DACL[i].Trustee.ptstrName))
|
|
|
|
|
|
- elif ACEs[i].Trustee.TrusteeForm == TRUSTEE_IS_NAME:
|
|
|
+ elif DACL[i].Trustee.TrusteeForm == TRUSTEE_IS_NAME:
|
|
|
sid_type = SidTypeInvalid
|
|
|
domainlength = 0
|
|
|
- LookupAccountNameW(NULL, ACEs[i].Trustee.ptstrName, NULL, &(length), NULL, &domainlength, &sid_type)
|
|
|
+ LookupAccountNameW(NULL, DACL[i].Trustee.ptstrName, NULL, &sidlength, NULL, &domainlength, &sid_type)
|
|
|
|
|
|
- newsid = <PSID>malloc((length) * sizeof(BYTE))
|
|
|
+ newsid = <PSID>malloc((sidlength) * sizeof(BYTE))
|
|
|
domainlength = 0
|
|
|
- LookupAccountNameW(NULL, ACEs[i].Trustee.ptstrName, newsid, &length, NULL, &domainlength, &sid_type)
|
|
|
+ LookupAccountNameW(NULL, DACL[i].Trustee.ptstrName, newsid, &sidlength, NULL, &domainlength, &sid_type)
|
|
|
trusteeName, domain, type = _look_up_account_sid(newsid)
|
|
|
|
|
|
name = trusteeName
|
|
|
sidstr = sid2string(newsid)
|
|
|
free(newsid)
|
|
|
|
|
|
- elif ACEs[i].Trustee.TrusteeForm == TRUSTEE_BAD_FORM:
|
|
|
+ elif DACL[i].Trustee.TrusteeForm == TRUSTEE_BAD_FORM:
|
|
|
continue
|
|
|
- permissions = {'user': {'name': name, 'sid': sidstr}, 'permissions': (ACEs[i].grfAccessPermissions, ACEs[i].grfAccessMode, ACEs[i].grfInheritance)}
|
|
|
- pyDACL.append(permissions)
|
|
|
- item['win_dacl'] = json.dumps(pyDACL)
|
|
|
+ if ((depth == 0 and DACL[i].grfInheritance & INHERIT_ONLY != 0)
|
|
|
+ or (DACL[i].grfInheritance & INHERIT_NO_PROPAGATE and depth == 1)
|
|
|
+ or (DACL[i].grfInheritance != NO_INHERITANCE and DACL[i].grfInheritance & INHERIT_NO_PROPAGATE == 0)):
|
|
|
+ permissions = {'user': {'name': name, 'sid': sidstr}, 'permissions': (DACL[i].grfAccessPermissions, DACL[i].grfAccessMode, NO_INHERITANCE)}
|
|
|
+ pyDACL.append(permissions)
|
|
|
+
|
|
|
+ if permissions_granted:
|
|
|
+ for i in range(sacllength):
|
|
|
+ permissions = None
|
|
|
+ name = ""
|
|
|
+ sidstr = ""
|
|
|
+ if DACL[i].Trustee.TrusteeForm == TRUSTEE_IS_SID:
|
|
|
+ name, domain, type = _look_up_account_sid(<BYTE*>(SACL[i].Trustee.ptstrName))
|
|
|
+ sidstr = sid2string(<PSID>(SACL[i].Trustee.ptstrName))
|
|
|
+
|
|
|
+ elif SACL[i].Trustee.TrusteeForm == TRUSTEE_IS_NAME:
|
|
|
+ sid_type = SidTypeInvalid
|
|
|
+ domainlength = 0
|
|
|
+ LookupAccountNameW(NULL, SACL[i].Trustee.ptstrName, NULL, &sidlength, NULL, &domainlength, &sid_type)
|
|
|
+
|
|
|
+ newsid = <PSID>malloc((sidlength) * sizeof(BYTE))
|
|
|
+ domainlength = 0
|
|
|
+ LookupAccountNameW(NULL, SACL[i].Trustee.ptstrName, newsid, &sidlength, NULL, &domainlength, &sid_type)
|
|
|
+ trusteeName, domain, type = _look_up_account_sid(newsid)
|
|
|
+
|
|
|
+ name = trusteeName
|
|
|
+ sidstr = sid2string(newsid)
|
|
|
+ free(newsid)
|
|
|
+ else:
|
|
|
+ continue
|
|
|
+ if ((depth == 0 and SACL[i].grfInheritance & INHERIT_ONLY != 0)
|
|
|
+ or (SACL[i].grfInheritance & INHERIT_NO_PROPAGATE and depth == 1)
|
|
|
+ or (SACL[i].grfInheritance != NO_INHERITANCE and SACL[i].grfInheritance & INHERIT_NO_PROPAGATE == 0)):
|
|
|
+ permissions = {'user': {'name': name, 'sid': sidstr}, 'permissions': (SACL[i].grfAccessPermissions, SACL[i].grfAccessMode, NO_INHERITANCE)}
|
|
|
+ pySACL.append(permissions)
|
|
|
+
|
|
|
+ if depth == 0:
|
|
|
+ item['win_dacl'] = json.dumps(pyDACL)
|
|
|
+ item['win_sacl'] = json.dumps(pySACL)
|
|
|
|
|
|
free(SD)
|
|
|
- LocalFree(<HLOCAL>ACEs)
|
|
|
+ LocalFree(<HLOCAL>DACL)
|
|
|
+ LocalFree(<HLOCAL>SACL)
|
|
|
+ return pyDACL,pySACL
|
|
|
|
|
|
|
|
|
def acl_set(path, item, numeric_owner=False):
|
|
|
- if 'win_dacl' not in item:
|
|
|
- return
|
|
|
+ if not permissions_enabled:
|
|
|
+ enable_permissions()
|
|
|
|
|
|
- pyDACL = json.loads(item.win_dacl)
|
|
|
- cdef _EXPLICIT_ACCESS_W* ACEs = <_EXPLICIT_ACCESS_W*>calloc(sizeof(_EXPLICIT_ACCESS_W), len(pyDACL))
|
|
|
+ cdef _EXPLICIT_ACCESS_W* DACL
|
|
|
cdef wchar_t* temp
|
|
|
cdef PSID newsid
|
|
|
- for i in range(len(pyDACL)):
|
|
|
- if pyDACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
- ACEs[i].Trustee.TrusteeForm = TRUSTEE_IS_SID
|
|
|
- temp = PyUnicode_AsWideCharString(pyDACL[i]['user']['sid'], NULL)
|
|
|
- ConvertStringSidToSidW(temp, &newsid)
|
|
|
- ACEs[i].Trustee.ptstrName = <LPCTSTR>newsid
|
|
|
- PyMem_Free(temp)
|
|
|
- else:
|
|
|
- ACEs[i].Trustee.TrusteeForm = TRUSTEE_IS_NAME
|
|
|
- ACEs[i].Trustee.ptstrName = PyUnicode_AsWideCharString(pyDACL[i]['user']['name'], NULL)
|
|
|
- ACEs[i].grfAccessPermissions = pyDACL[i]['permissions'][0]
|
|
|
- ACEs[i].grfAccessMode = pyDACL[i]['permissions'][1]
|
|
|
- ACEs[i].grfInheritance = pyDACL[i]['permissions'][2]
|
|
|
cdef _ACL* newDACL
|
|
|
- SetEntriesInAclW(len(pyDACL), ACEs, NULL, &newDACL)
|
|
|
- cdef wchar_t* cstrPath = PyUnicode_AsWideCharString(path, NULL)
|
|
|
- SetNamedSecurityInfoW(cstrPath, SE_FILE_OBJECT, DACL_SECURITY_INFORMATION, NULL, NULL, newDACL, NULL)
|
|
|
|
|
|
- for i in range(len(pyDACL)):
|
|
|
- if pyDACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
- LocalFree(<HLOCAL>ACEs[i].Trustee.ptstrName)
|
|
|
- else:
|
|
|
- PyMem_Free(ACEs[i].Trustee.ptstrName)
|
|
|
- free(ACEs)
|
|
|
- PyMem_Free(cstrPath)
|
|
|
- LocalFree(<HLOCAL>newDACL)
|
|
|
+ cdef wchar_t* cstrPath
|
|
|
+
|
|
|
+ if 'win_dacl' in item:
|
|
|
+ pyDACL = json.loads(item.win_dacl)
|
|
|
+ if len(pyDACL) > 0:
|
|
|
+ DACL = <_EXPLICIT_ACCESS_W*>calloc(sizeof(_EXPLICIT_ACCESS_W), len(pyDACL))
|
|
|
+
|
|
|
+ for i in range(len(pyDACL)):
|
|
|
+ if pyDACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
+ DACL[i].Trustee.TrusteeForm = TRUSTEE_IS_SID
|
|
|
+ temp = PyUnicode_AsWideCharString(pyDACL[i]['user']['sid'], NULL)
|
|
|
+ ConvertStringSidToSidW(temp, &newsid)
|
|
|
+ DACL[i].Trustee.ptstrName = <LPCTSTR>newsid
|
|
|
+ PyMem_Free(temp)
|
|
|
+ else:
|
|
|
+ DACL[i].Trustee.TrusteeForm = TRUSTEE_IS_NAME
|
|
|
+ DACL[i].Trustee.ptstrName = PyUnicode_AsWideCharString(pyDACL[i]['user']['name'], NULL)
|
|
|
+ DACL[i].grfAccessPermissions = pyDACL[i]['permissions'][0]
|
|
|
+ DACL[i].grfAccessMode = pyDACL[i]['permissions'][1]
|
|
|
+ DACL[i].grfInheritance = pyDACL[i]['permissions'][2]
|
|
|
+
|
|
|
+ SetEntriesInAclW(len(pyDACL), DACL, NULL, &newDACL)
|
|
|
+ cstrPath = PyUnicode_AsWideCharString(path, NULL)
|
|
|
+ SetNamedSecurityInfoW(cstrPath, SE_FILE_OBJECT, PROTECTED_DACL_SECURITY_INFORMATION, NULL, NULL, newDACL, NULL)
|
|
|
+
|
|
|
+ for i in range(len(pyDACL)):
|
|
|
+ if pyDACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
+ LocalFree(<HLOCAL>DACL[i].Trustee.ptstrName)
|
|
|
+ else:
|
|
|
+ PyMem_Free(DACL[i].Trustee.ptstrName)
|
|
|
+ free(DACL)
|
|
|
+ PyMem_Free(cstrPath)
|
|
|
+ LocalFree(<HLOCAL>newDACL)
|
|
|
+
|
|
|
+ cdef _EXPLICIT_ACCESS_W* SACL
|
|
|
+ cdef _ACL* newSACL
|
|
|
+ if permissions_granted and 'win_sacl' in item:
|
|
|
+ pySACL = json.loads(item.win_sacl)
|
|
|
+ if len(pySACL) > 0:
|
|
|
+ SACL = <_EXPLICIT_ACCESS_W*>calloc(sizeof(_EXPLICIT_ACCESS_W), len(pySACL))
|
|
|
+
|
|
|
+ for i in range(len(pyDACL)):
|
|
|
+ if pySACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
+ SACL[i].Trustee.TrusteeForm = TRUSTEE_IS_SID
|
|
|
+ temp = PyUnicode_AsWideCharString(pySACL[i]['user']['sid'], NULL)
|
|
|
+ ConvertStringSidToSidW(temp, &newsid)
|
|
|
+ SACL[i].Trustee.ptstrName = <LPCTSTR>newsid
|
|
|
+ PyMem_Free(temp)
|
|
|
+ else:
|
|
|
+ SACL[i].Trustee.TrusteeForm = TRUSTEE_IS_NAME
|
|
|
+ SACL[i].Trustee.ptstrName = PyUnicode_AsWideCharString(pySACL[i]['user']['name'], NULL)
|
|
|
+ SACL[i].grfAccessPermissions = pySACL[i]['permissions'][0]
|
|
|
+ SACL[i].grfAccessMode = pySACL[i]['permissions'][1]
|
|
|
+ SACL[i].grfInheritance = pySACL[i]['permissions'][2]
|
|
|
+
|
|
|
+ SetEntriesInAclW(len(pySACL), SACL, NULL, &newSACL)
|
|
|
+ cstrPath = PyUnicode_AsWideCharString(path, NULL)
|
|
|
+ SetNamedSecurityInfoW(cstrPath, SE_FILE_OBJECT, PROTECTED_SACL_SECURITY_INFORMATION, NULL, NULL, newSACL, NULL)
|
|
|
+
|
|
|
+ for i in range(len(pySACL)):
|
|
|
+ if pySACL[i]['user']['name'] == '' or numeric_owner:
|
|
|
+ LocalFree(<HLOCAL>SACL[i].Trustee.ptstrName)
|
|
|
+ else:
|
|
|
+ PyMem_Free(SACL[i].Trustee.ptstrName)
|
|
|
+ free(SACL)
|
|
|
+ PyMem_Free(cstrPath)
|
|
|
+ LocalFree(<HLOCAL>newSACL)
|
|
|
|
|
|
|
|
|
def sync_dir(path):
|