Windows平台使用Python切换用户运行程序

kongxx

发布于 2018-05-14 11:47:22

1.8K0

发布于 2018-05-14 11:47:22

文章被收录于专栏：有困难要上，没有困难创造困难也要上！有困难要上，没有困难创造困难也要上！

在Windows平台，如果需要以某个指定用户来运行程序，此时可以通过使用 Windows CreateProcessAsUser API来实现。但是要实现这个功能首先需要有要切换用户的用户名和密码。

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

要使上面的程序生效，首先需要使用一个windows service来执行这个程序（目前我是这么测试的，有人说也可以在命令行以administrator来运行，但是我没有测试成功）。关于Python创建Windows Service的方法，可以参考我前面文章。
要切换用户执行脚本，需要有被切换用户的用户名和密码。
在用 CreateProcessAsUser API执行命令后，可以通过 GetExitCodeProcess api 来检查进程状态，如果返回结果是 win32con.STILL_ACTIVE，我们可以继续等待。
显然，通过 GetExitCodeProcess api来检查进程是否退出不是很好用，所以这里我使用了psutil库来等待进程退出。