python安全管理子进程-subprocess - Python - language - ITeye论坛

python安全管理子进程-subprocess - Python - language - ITeye论坛

    锁定老帖子 主题：python安全管理子进程-subprocess
    精华帖 (0) :: 良好帖 (0) :: 新手帖 (0) :: 隐藏帖 (0)
    作者 正文

        pypy
        等级: 一星会员
        pypy的博客
        性别: [男]
        文章: 20
        积分: 100
        来自: 上海

       发表时间：2009-06-11   最后修改：2010-04-03
    < > 猎头职位: 北京: ITeye网站诚聘网站编辑
    相关文章: 

        一个多进程python代码产生了僵尸进程，晕。
        Python 信号(linux) 源代码放附件中
        python语音控制电脑_uusee全屏


    推荐群组: Python
    更多相关推荐
    Python
    经常会用到python去调用外部 工具或者命令去干活

    有的时候子进程并不按预期退出

    比如，子进程由于某种原因挂在那里，

    这时候也许，我们有这样一种需求：需要父进程对子进程有监控动作，即，超过一定的时间，就不再等待子进程自己退出，而是去kill子进程，回收资源


    以下会列出几张实现方法

    1.os.system

    http://docs.python.org/library/os.html

    Python代码  收藏代码

        Python 2.5.2 (r252:60911, Jan  4 2009, 17:40:26) 
        [GCC 4.3.2] on linux2 
        Type "help", "copyright", "credits" or "license" for more information. 
        >>> import os 
        >>> os.system("date") 
        Wed Jun 10 19:34:23 CST 2009 
        0 
        >>> 



    其实是执行 linux shell 命令

    Shell代码  收藏代码

        $Snbsp;date 
        Wed Jun 10 19:36:02 CST 2009 



    缺点：
    A. os.system() 是新起一个shell去干活的，对系统的开销比较大
    B. 获得输出等信息比较麻烦,不能与外部命令或工具交互
    C. 无法控制，（如果调用的外部命令，挂死或者执行时间很长），主进程无法控制os.system(), 因为调用os.system(cmd) 调用进程会block， until os.system() 自己退出

    2.commands
    [url]
    http://docs.python.org/library/commands.html[/url]

    Python代码  收藏代码

        tommy@lab3:~$Snbsp;python 
        Python 2.5.2 (r252:60911, Jan  4 2009, 17:40:26) 
        [GCC 4.3.2] on linux2 
        Type "help", "copyright", "credits" or "license" for more information. 
        >>> import commands 
        >>> dir(commands) 
        ['__all__', '__builtins__', '__doc__', '__file__', '__name__', 'getoutput', 'getstatus', 'getstatusoutput', 'mk2arg', 'mkarg'] 
        >>> commands.getoutput("date") 
        'Wed Jun 10 19:39:57 CST 2009' 
        >>> 
        >>> commands.getstatusoutput("date") 
        (0, 'Wed Jun 10 19:40:41 CST 2009') 



    优点：
    A. 容易获得外部命令的输出，已经退出状态
    缺点：
    同os.system()中的B,C

    3.subprocess

    http://docs.python.org/library/subprocess.html

    Python代码  收藏代码

        tommy@lab3:~$Snbsp;python 
        Python 2.5.2 (r252:60911, Jan  4 2009, 17:40:26) 
        [GCC 4.3.2] on linux2 
        Type "help", "copyright", "credits" or "license" for more information. 
        >>> import subprocess 
        >>> dir(subprocess) 
        ['CalledProcessError', 'MAXFD', 'PIPE', 'Popen', 'STDOUT', '__all__', '__builtins__', '__doc__', '__file__', '__name__', '_active', '_cleanup', '_demo_posix', '_demo_windows', 'call', 'check_call', 'errno', 'fcntl', 'gc', 'list2cmdline', 'mswindows', 'os', 'pickle', 'select', 'signal', 'sys', 'traceback', 'types'] 
        >>> Popen = subprocess.Popen(["date"]) 
        Wed Jun 10 19:48:41 CST 2009 
        >>> Popen.pid 
        24723 
        >>> 



    优点：
    看文档吧，可以支持和子进程交互等等

    虽然 python2.6中的subprocess模块增加了
    kill()
    terminate()
    来控制子进程退出

    但是在实际的使用过程中会发现

    如果子进程并不是自己退出，而是调用 kill()/terminate() 给子进程发信退出

    通过 top 或者 ps -A 看到，子进程的确是释放资源了，但是却变成了 zombie（僵尸进程）

    于是分析 subprocess.py模块
    Python代码  收藏代码

        1201 
        1202         def send_signal(self, sig): 
        1203             """Send a signal to the process
        1204             """ 
        1205             os.kill(self.pid, sig) 
        1206 
        1207         def terminate(self): 
        1208             """Terminate the process with SIGTERM
        1209             """ 
        1210             self.send_signal(signal.SIGTERM) 
        1211 
        1212         def kill(self): 
        1213             """Kill the process with SIGKILL
        1214             """ 
        1215             self.send_signal(signal.SIGKILL) 



    程序仅仅是 调用 os.kill(self.pid, sig) 向子进程发送了一个信号后，标准subprocess.py库 父进程并没有显示区 wait() 子进程，导致了 zombie（僵尸进程） 的生成

    所以问题找到，

    修改subprocess.py模块，显然不妥，
    那就封装一下(继承subprocess)，

    我是用这个subprocess去调用mencoder 做批量转码，所以为子进程超时，要有很好控制，

    具体实现见附件

    显示的封装成两个函数

    1.
    shell_2_tty(_cmd=cmds, _cwd=None, _timeout=10*60)
    # _cmd 是要执行的外面命令行，要是一个 list， 如果是str，shell=True，会启动一个新的shell去干活的，这样，不利于进程的控制
    # _cwd 是执行这个命令行前，cd到这个路径下面，这个，对我的用应很重要，如果不需要可以用默认值
    # _timeout 这个是主角，设置超时时间（秒单位），从真重执行命令行开始计时，墙上时间超过 _timeout后，父进程会kill掉子进程，回收资源，并避免产生 zombie（僵尸进程）
    # 并将调用的命令行输出，直接输出到stdout,即是屏幕的终端上，
    （如果对输出比较讨厌，可以将 stdout = open("/dev/null", "w"), stderr=open("/dev/null"),等等）

    2.
    shell_2_tempfile(_cmd=cmds, _cwd=None, _timeout=10)
    类同1，主要是增加，对命令行的输出，捕获，并返回给父进程，留作分析