2

我有这个以root身份运行的简单python脚本，但将以其他用户身份执行子进程：

#!/usr/bin/env python2
import subprocess
import os

def demote(user_uid):
   def result():
      os.setuid(user_uid)
   return result

cmd = "echo $USER"
proc = subprocess.Popen(cmd, preexec_fn=demote(1000), stdout=subprocess.PIPE, shell=True)
output = proc.communicate()[0]

print output

如果 cmd 为“sleep 60”，则脚本以用户身份生成 sleep：

# ps -ef | grep sleep
dave   17812 17811  0 17:05 pts/5    00:00:00 /usr/bin/sleep 60

但是，如果 cmd 是“echo $USER”，则输出为“root”。如果我将子进程作为另一个用户生成，我需要做些什么来获取用户的环境吗？

4

3 回答 3

2

这里的 subprocess 并没有真正的问题。子进程正在从您当前环境中传递环境，因此当您“回显 $USER”时，它会报告当前环境中的用户。

看看当您将命令更改为whoami实际检查您的 UID 时会发生什么：

我的 python shell 以 root 身份运行：

>>> print os.getuid()
0

但是更改正在运行的命令会输出正确的新用户：

>>> cmd = "whoami"
>>> proc = subprocess.Popen(cmd, preexec_fn=demote(1000), stdout=subprocess.PIPE, shell=True)
>>> output = proc.communicate()[0]
>>> print output
voodoonofx

如果您真的想将环境修改为该用户，则可以将新字典传递给 Popen 调用。查看通过 help(subprocess.Popen) 传递的 env 关键字：

  __init__(self, args, bufsize=0, executable=None, stdin=None, stdout=None, stderr=None, preexec_fn=None, close_fds=False, shell=False, cwd=None, env=None, universal_newlines=False, startupinfo=None, creationflags=0)
      Create new Popen instance.
于 2013-10-25T22:42:04.480 回答
0

在运行任何命令之前更改用户，这可能会有所帮助：

su otheruser # this will prompt you to enter the 'otheruser' password.
于 2013-10-25T21:13:00.710 回答
0

一种选择是运行一个完整的登录外壳......例如

cmd = "/bin/bash -l -c 'echo $USER'"
于 2013-10-25T21:23:08.350 回答