在Windows平臺(tái)莫矗,如果需要以某個(gè)指定用戶來運(yùn)行程序碱工,此時(shí)可以通過使用 Windows CreateProcessAsUser API來實(shí)現(xiàn)糖荒。但是要實(shí)現(xiàn)這個(gè)功能首先需要有要切換用戶的用戶名和密碼舶沿。
import psutil
import win32process
import win32security
import win32con
import win32api
import win32service
import win32file
import win32profile
import pywintypes
try:
username = 'admin'
password = 'admin'
domain = ''
token = win32security.LogonUser (
username,
domain,
password,
win32con.LOGON32_LOGON_SERVICE,
win32con.LOGON32_PROVIDER_DEFAULT
)
win32security.ImpersonateLoggedOnUser(token)
cmd = '<a command line programe>'
cwd = '<a work dir>'
env = os.environ
priority = win32con.NORMAL_PRIORITY_CLASS
startup = win32process.STARTUPINFO()
(hProcess, hThread, dwProcessId, dwThreadId) = \
win32process.CreateProcessAsUser(token, None, cmd, None, None, True, priority, env, cwd, startup)
# win32process.GetExitCodeProcess(hProcess) == win32con.STILL_ACTIVE
process = psutil.Process(dwProcessId)
return_code = process.wait()
print 'return code: %s' % str(return_code)
except win32security.error as e:
print e
- 要使上面的程序生效,首先需要使用一個(gè)windows service來執(zhí)行這個(gè)程序(目前我是這么測試的蛾坯,有人說也可以在命令行以administrator來運(yùn)行,但是我沒有測試成功)疏遏。關(guān)于Python創(chuàng)建Windows Service的方法脉课,可以參考我前面文章。
- 要切換用戶執(zhí)行腳本财异,需要有被切換用戶的用戶名和密碼倘零。
- 在用 CreateProcessAsUser API執(zhí)行命令后,可以通過 GetExitCodeProcess api 來檢查進(jìn)程狀態(tài)戳寸,如果返回結(jié)果是 win32con.STILL_ACTIVE呈驶,我們可以繼續(xù)等待。
- 顯然疫鹊,通過 GetExitCodeProcess api來檢查進(jìn)程是否退出不是很好用袖瞻,所以這里我使用了psutil庫來等待進(jìn)程退出。
參考