【转载】python 并发subprocess.Popen的坑

转自:http://www.west999.com/cms/wiki/code/2018-07-20/41564.html

表现

一个父进程里多个线程并发地调用 subprocess.Popen 来创建子进程的时候, 会有几率出现 Popen 长时间不返回的情况.

这个问题是由于fd被多个子进程同时继承导致的.

重现问题的代码

下面这个小程序启动2个线程, 每个线程各自(通过 subprocess.Popen )启动一个子进程, 一个子进程执行 echo 1 后就直接返回; 另一个子进程启动后, sleep 0.03 秒后返回.

程序里统计了2个调用 Popen 花的时间, 运行后可以发现, echo的进程有时启动很快(小于预期的0.01秒, 仅仅是启动, 不包括执行时间), 有时会很慢(超过0.03秒), 刚好和另一个sleep的进程执行时间吻合. 调大sleep子进程的时间可以看到echo也会同样有几率返回慢.

# > cat slow.py

import threading

import subprocess

import time

def _open(cmd, expect):

    t0 = time.time()

    proc = subprocess.Popen(

            cmd,

            shell=True,

            # # without this line, some Popen does not return at once as expected

            # close_fds=True,

            stderr=subprocess.PIPE,

            stdout=subprocess.PIPE)

    spent = time.time() - t0

    if spent > expect:

        print cmd + ' spent: ' + str(spent)

    proc.wait()

for ii in range(100):

    ths = [

            threading.Thread(target=_open, args=('echo 1', 0.01)),

            threading.Thread(target=_open, args=('sleep 0.03', 0.05)),

    ]

    for th in ths:

        th.start()

    for th in ths:

        th.join()

# > python2 slow.py

# echo 1 spent: 0.0381829738617

# echo 1 spent: 0.041118144989

# echo 1 spent: 0.0417079925537

# echo 1 spent: 0.0421600341797

# echo 1 spent: 0.039479970932

# ...

运行环境:

# uname -a

Linux ** 3.10.0-327.el7.x86_64 #1 SMP Thu Nov 19 22:10:57 UTC 2015 x86_64 x86_64 x86_64 GNU/Linux

# python2 --version

Python 2.7.5

# cat /etc/*-release

CentOS Linux release 7.2.1511 (Core)

解决方法

Popen 时加上 close_fds=True , 保证fd不会被多个子进程继承.

proc = subprocess.Popen(

        cmd,

        close_fds=True, # here

        ...

        )

原因

直接原因是因为有并发时, Popen中创建的pipe没有被关闭, 导致父进程认为子进程还没启动成功而一直阻塞.

这里的 Popen 的过程, 包括:

父进程创建通信的管道(调用 os.pipe() )

fork子进程

父进程通过pipe阻塞读取子进程的启动后的错误消息, 确认失败;

或读取到EOF(pipe在子进程exec时被关闭), 确认成功.

Popen 调用的最核心的代码是 subprocess.py 中的 _execute_child , 问题的原因可以从下面这段简化版的代码中看到:

def _execute_child(self, args, executable, preexec_fn, close_fds,

                  cwd, env, universal_newlines,

                  startupinfo, creationflags, shell, to_close,

                  p2cread, p2cwrite,

                  c2pread, c2pwrite,

                  errread, errwrite):

    # 1) 创建pipe用于父子进程通信...

    errpipe_read, errpipe_write = self.pipe_cloexec()

    try:

        try:

            # 2) 3)

            self.pid = os.fork()

            if self.pid == 0:

                # 子进程流程

                try:

                    # 5) 子进程关闭读的pipe: 不需要接受父进程的消息

                    os.close(errpipe_read)

                    # 如果需要, 子进程关闭所有打开的文件描述符.

                    # 只留下用于告之父进程错误的pipe

                    if close_fds:

                        self._close_fds(but=errpipe_write)

                    # 7) 子进程加载程序

                    os.execvp(executable, args)

                except:

                    exc_type, exc_value, tb = sys.exc_info()

                    exc_lines = traceback.format_exception(exc_type,

                                                          exc_value,

                                                          tb)

                    exc_value.child_traceback = ''.join(exc_lines)

                    # 通过pipe通知父进程错误信息

                    os.write(errpipe_write, pickle.dumps(exc_value))

                # 子进程如果出错, 直接退出. 退出会关闭所有pipe

                os._exit(255)

            # 以下是父进程的流程

        finally:

            # 4) 父进程不需要通知子进程错误消息, 直接关闭pipe的写入端.

            os.close(errpipe_write)

        # 6) 父进程阻塞的读取子进程发来的错误消息.

        data = _eintr_retry_call(os.read, errpipe_read, 1048576)

    finally:

        # 无论成功与否, 父进程不再需要从子进程读取任何消息了, 关闭pipe的读取端.

        os.close(errpipe_read)

    if data != "":

        # 父进程的错误处理...

    # 8) 9)

我们把上面的代码的执行过程整理成时间表, 如下:

步骤时间动作echo线程sleep线程父进程打开的fdecho子进程的fdsleep子进程的fd

10.00创建通信的pipe4,5=os.pipe()6,7=os.pipe()4,5,6,7--

20.00echo线程forkfork-4,5,6,74,5,6,7-

30.00sleep线程fork-fork4,5,6,74,5,6,74,5,6,7

40.00父进程关闭写pipeclose(5)close(7)4,-,6,-4,5,6,74,5,6,7

50.00子进程关闭读pipe--4,-,6,--,5,6,74,5,-,7

60.00父进程阻塞读read(4)…read(6)…4,-,6,--,5,6,74,5,-,7

70.00子进程exec, 关闭自己的的pipe-fd--4,-,6,--,-,6,74,5,-,-

80.01echo子进程结束, sleep线程返回-read(6):done4,-,6,--,-,-,-4,5,-,-

90.03sleep子进程结束, echo线程返回read(4):done-4,-,6,- -,-,-,-

Popen 调用时创建1对pipe和子进程通信, Popen 返回时, 子进程就已经创建成功, pipe也在子进程exec时关闭了.

但如果同时有2个 Popen 在调用, 父进程中会同时出现2对或几对pipe.

这几对pipe会在fork时被继承到子进程. 子进程在进行exec之前(创建pipe时), 已经将pipe的fd设置FD_CLOEXEC: 执行exec时自动关闭fd. 但并没有对其他fd进行这个设置 .

因此, 如果并发地调用 Popen , 1个子进程会在fork时带着为别的子进程准备的pipe fd, 并且不会关闭它们(因为子进程只知道自己的pipe fd, 没有设置 close_fds 时, 它不会鲁莽地关闭其他fd)! 这样1个pipe fd本应在父子进程这2个进程之间共享, 却意外地会在3个或更多的进程中处于打开状态.

而父进程中的Popen在阻塞的read pipe, 自己的子进程exec后自动关闭了这个pipe fd, 进而让父进程结束read, 但还有另外一个进程在打开着这个pipe fd(步骤7)(我们的例子中是sleep子进程继承了echo 子进程的pipe fd并且没有关闭), 父进程中的read不会检查到fd关闭, 一直保持阻塞读的状态.

直到所有继承了这个pipe fd的进程都退出了, 父进程的read才能结束(sleep子进程退出时, 自动关闭了所有fd, 包括为echo子进程准备的pipe fd, 到此时父进程read才能收到1个EOF并退出read的阻塞).


来自:http://drmingdrmer.github.io/tech/programming/2017/11/20/python-concurrent-popen.html

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,245评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,749评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,960评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,575评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,668评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,670评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,664评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,422评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,864评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,178评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,340评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,015评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,646评论 3 323
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,265评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,494评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,261评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,206评论 2 352

推荐阅读更多精彩内容