系统相关
首页 > 系统相关> > 将子进程作为不同用户从长时间运行的Python进程运行

将子进程作为不同用户从长时间运行的Python进程运行

作者:互联网

我有一个长期运行的,守护进程的Python进程,它使用子进程在发生某些事件时生成新的子进程.长时间运行的进程由具有超级用户权限的用户启动.我需要它生成的子进程作为不同的用户(例如,“nobody”)运行,同时保留父进程的超级用户权限.

我正在使用

su -m nobody -c <program to execute as a child>

但这似乎是重量级的,并不会非常干净地死去.

有没有办法以编程方式完成此操作而不是使用su?我正在查看os.set * uid方法,但Python std lib中的doc在该领域非常稀少.

解决方法:

既然你提到了守护进程,我可以断定你是在类Unix操作系统上运行的.这很重要,因为如何做到这一点取决于操作系统的种类.这个答案仅适用于Unix,包括Linux和Mac OS X.

>定义一个函数,用于设置正在运行的进程的gid和uid.
>将此函数作为preexec_fn参数传递给subprocess.Popen

subprocess.Popen将使用fork / exec模型来使用preexec_fn.这相当于按顺序调用os.fork(),preexec_fn()(在子进程中)和os.exec()(在子进程中).由于os.setuid,os.setgid和preexec_fn都只在Unix上受支持,因此该解决方案无法移植到其他类型的操作系统.

以下代码是一个脚本(Python 2.4),演示了如何执行此操作:

import os
import pwd
import subprocess
import sys


def main(my_args=None):
    if my_args is None: my_args = sys.argv[1:]
    user_name, cwd = my_args[:2]
    args = my_args[2:]
    pw_record = pwd.getpwnam(user_name)
    user_name      = pw_record.pw_name
    user_home_dir  = pw_record.pw_dir
    user_uid       = pw_record.pw_uid
    user_gid       = pw_record.pw_gid
    env = os.environ.copy()
    env[ 'HOME'     ]  = user_home_dir
    env[ 'LOGNAME'  ]  = user_name
    env[ 'PWD'      ]  = cwd
    env[ 'USER'     ]  = user_name
    report_ids('starting ' + str(args))
    process = subprocess.Popen(
        args, preexec_fn=demote(user_uid, user_gid), cwd=cwd, env=env
    )
    result = process.wait()
    report_ids('finished ' + str(args))
    print 'result', result


def demote(user_uid, user_gid):
    def result():
        report_ids('starting demotion')
        os.setgid(user_gid)
        os.setuid(user_uid)
        report_ids('finished demotion')
    return result


def report_ids(msg):
    print 'uid, gid = %d, %d; %s' % (os.getuid(), os.getgid(), msg)


if __name__ == '__main__':
    main()

您可以像这样调用此脚本:

从根开始……

(hale)/tmp/demo$sudo bash --norc
(root)/tmp/demo$ls -l
total 8
drwxr-xr-x  2 hale  wheel    68 May 17 16:26 inner
-rw-r--r--  1 hale  staff  1836 May 17 15:25 test-child.py

在子进程中成为非root用户…

(root)/tmp/demo$python test-child.py hale inner /bin/bash --norc
uid, gid = 0, 0; starting ['/bin/bash', '--norc']
uid, gid = 0, 0; starting demotion
uid, gid = 501, 20; finished demotion
(hale)/tmp/demo/inner$pwd
/tmp/demo/inner
(hale)/tmp/demo/inner$whoami
hale

当子进程退出时,我们回到父进程的根…

(hale)/tmp/demo/inner$exit
exit
uid, gid = 0, 0; finished ['/bin/bash', '--norc']
result 0
(root)/tmp/demo$pwd
/tmp/demo
(root)/tmp/demo$whoami
root

请注意,让父进程等待子进程退出仅用于演示目的.我这样做是为了让父母和孩子共享一个终端.一个守护进程没有终端,很少等待子进程退出.

标签:setuid,python,fork,subprocess
来源: https://codeday.me/bug/20190916/1808557.html