1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# -*- coding: utf-8 -*-
# !/usr/bin/python
import os
import ctypes
import sys
from foreign.client_handling.lazagne.config.winstructure import get_os_version
from foreign.client_handling.lazagne.config.constant import constant
def get_user_list_on_filesystem(impersonated_user=[]):
"""
Get user list to retrieve their passwords
"""
# Check users existing on the system (get only directories)
user_path = u'{drive}:\\Users'.format(drive=constant.drive)
if float(get_os_version()) < 6:
user_path = u'{drive}:\\Documents and Settings'.format(drive=constant.drive)
all_users = []
if os.path.exists(user_path):
all_users = [filename for filename in os.listdir(user_path) if os.path.isdir(os.path.join(user_path, filename))]
# Remove default users
for user in ['All Users', 'Default User', 'Default', 'Public', 'desktop.ini']:
if user in all_users:
all_users.remove(user)
# Removing user that have already been impersonated
for imper_user in impersonated_user:
if imper_user in all_users:
all_users.remove(imper_user)
return all_users
def set_env_variables(user, to_impersonate=False):
# Restore template path
template_path = {
'APPDATA': u'{drive}:\\Users\\{user}\\AppData\\Roaming\\',
'USERPROFILE': u'{drive}:\\Users\\{user}\\',
'HOMEDRIVE': u'{drive}:',
'HOMEPATH': u'{drive}:\\Users\\{user}',
'ALLUSERSPROFILE': u'{drive}:\\ProgramData',
'COMPOSER_HOME': u'{drive}:\\Users\\{user}\\AppData\\Roaming\\Composer\\',
'LOCALAPPDATA': u'{drive}:\\Users\\{user}\\AppData\\Local',
}
constant.profile = template_path
if not to_impersonate:
# Get value from environment variables
for env in constant.profile:
if os.environ.get(env):
try:
constant.profile[env] = os.environ.get(env).decode(sys.getfilesystemencoding())
except Exception:
constant.profile[env] = os.environ.get(env)
# Replace "drive" and "user" with the correct values
for env in constant.profile:
constant.profile[env] = constant.profile[env].format(drive=constant.drive, user=user)
def get_username_winapi():
GetUserNameW = ctypes.windll.advapi32.GetUserNameW
GetUserNameW.argtypes = [ctypes.c_wchar_p, ctypes.POINTER(ctypes.c_uint)]
GetUserNameW.restype = ctypes.c_uint
_buffer = ctypes.create_unicode_buffer(1)
size = ctypes.c_uint(len(_buffer))
while not GetUserNameW(_buffer, ctypes.byref(size)):
# WinError.h
# define ERROR_INSUFFICIENT_BUFFER 122L // dderror
if ctypes.GetLastError() == 122:
_buffer = ctypes.create_unicode_buffer(len(_buffer)*2)
size.value = len(_buffer)
else:
return os.getenv('username') # Unusual error
return _buffer.value
|
