Python学习笔记15：标准库之获取进程信息（os包）

Python的os包中有查询和修改进程信息的函数。

进程信息

os包中相关函数如下：

uname() 返回操作系统相关信息，类似于Linux上的uname命令。

umask() 设置该进程创建文件时的权限mask，类似于Linux上的umask命令。

get*() 查询 (*由以下代替)

uid, euid, resuid, gid, egid, resgid ：权限相关，其中resuid主要用来返回saved UID。

pid, pgid, ppid, sid ：进程相关。

put*() 设置 (*由以下代替)

euid, egid： 用于更改euid，egid。

uid, gid ： 改变进程的uid, gid。只有super user才有权改变进程uid和gid (意味着要以$sudo python的方式运行Python)。

pgid, sid ： 改变进程所在的进程组(process group)和会话(session)。

getenviron()：获得进程的环境变量

setenviron()：更改进程的环境变量

例1，进程的real UID和real GID

import os
print(os.getuid())
print(os.getgid())<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"> </span>

将上面的程序保存为py_id.py文件，分别用$python py_id.py和$sudo python py_id.py看一下运行结果

saved UID和saved GID

当我们写一个Python脚本后，我们实际运行的是python这个解释器，而不是Python脚本文件。

对比C，C语言直接运行由C语言编译成的执行文件。我们必须更改python解释器本身的权限来运用saved UID机制，然而这么做又是异常危险的。

比如说，我们的python执行文件为/usr/bin/python (你可以通过$which python获知)

先看一下python解释器的权限：

$ls -l /usr/bin/python

输出：

-rwxr-xr-x root root

我们修改权限以设置set UID和set GID位

$sudo chmod 6755 /usr/bin/python

/usr/bin/python的权限成为:

-rwsr-sr-x root root

随后，我们运行文件下面test.py文件，这个文件可以是由普通用户所有:

import os
print(os.getresuid())

输出：

(1000, 0, 0)

上面分别是UID，EUID，saved UID。

我们只用执行一个由普通用户拥有的python脚本，就可以得到super user的权限！

所以，这样做是极度危险的，我们相当于交出了系统的保护系统。

想像一下Python强大的功能，别人现在可以用这些强大的功能作为攻击你的武器了！使用下面命令来恢复到从前:

$sudo chmod 0755 /usr/bin/python