Skip to content

Commit e08935d

Browse files
committed
Simplify logic
1 parent af56d94 commit e08935d

1 file changed

Lines changed: 8 additions & 14 deletions

File tree

nxc/modules/presence.py

Lines changed: 8 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -84,24 +84,18 @@ def on_admin_login(self, context, connection):
8484
user_handle = samr.hSamrOpenUser(dce, domain_handle, samr.MAXIMUM_ALLOWED, rid)["UserHandle"]
8585
username = samr.hSamrQueryInformationUser2(dce, user_handle, samr.USER_INFORMATION_CLASS.UserAllInformation)["Buffer"]["All"]["UserName"]
8686

87-
full_username = f"{domain}\\{username}"
88-
admin_users.add(f"{full_username} (Member of {group_name})")
87+
admin_users.add(f"{domain}\\{username} (Member of {group_name})")
8988

9089
# map sid string of user to username
9190
user_sid = f"{domain_sid}-{rid}"
92-
self.sid_to_user[user_sid] = full_username
93-
94-
samr.hSamrCloseHandle(dce, user_handle)
95-
except Exception as name_e:
96-
try:
97-
sid_str = domain_sid
98-
full_sid = f"{sid_str}-{rid}"
99-
except Exception:
100-
full_sid = "[unrepresentable SID]"
101-
context.log.debug(f"Failed to get user info for RID {rid}: {name_e!s}")
102-
admin_users.add(f"{domain}\\{full_sid} (Member of {group_name})")
91+
self.sid_to_user[user_sid] = f"{domain}\\{username}"
92+
except Exception as e:
93+
context.log.debug(f"Failed to get user info for RID {rid}: {e!s}")
94+
admin_users.add(f"{domain}\\{domain_sid}-{rid} (Member of {group_name})")
95+
finally:
96+
with suppress(Exception):
97+
samr.hSamrCloseHandle(dce, user_handle)
10398
except Exception as e:
104-
context.log.exception(e)
10599
context.log.debug(f"Failed to get members of group {group_name}: {e!s}")
106100
finally:
107101
with suppress(Exception):

0 commit comments

Comments
 (0)