在Windows上将^ C发送到Python子进程对象

Question

我有一个测试工具（用Python编写），需要通过发送它来关闭被测程序（用C语言编写）。在Unix上，

proc.send_signal(signal.SIGINT)

工作得很好。在Windows上，会抛出错误（“不支持信号2”或类似的东西）。我正在使用Python 2.7 for Windows，所以我觉得我应该能够做到

proc.send_signal(signal.CTRL_C_EVENT)

但这根本不起作用。我需要做什么？这是创建子进程的代码：

# Windows needs an extra argument passed to subprocess.Popen,
# but the constant isn't defined on Unix.
try: kwargs['creationflags'] = subprocess.CREATE_NEW_PROCESS_GROUP
except AttributeError: pass
proc = subprocess.Popen(argv,
                        stdin=open(os.path.devnull, "r"),
                        stdout=subprocess.PIPE,
                        stderr=subprocess.PIPE,
                        **kwargs)

Answer 1

通过使用包装器（如提供的链接Vinay中所述），可以使用Windows启动命令在新的控制台窗口中启动解决方案。

包装器代码：

#wrapper.py
import subprocess, time, signal, sys, os

def signal_handler(signal, frame):
  time.sleep(1)
  print 'Ctrl+C received in wrapper.py'

signal.signal(signal.SIGINT, signal_handler)
print "wrapper.py started"
subprocess.Popen("python demo.py")
time.sleep(3) #Replace with your IPC code here, which waits on a fire CTRL-C request
os.kill(signal.CTRL_C_EVENT, 0)

捕获CTRL-C的程序代码：

#demo.py

import signal, sys, time

def signal_handler(signal, frame):
  print 'Ctrl+C received in demo.py'
  time.sleep(1)
  sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)
print 'demo.py started'
#signal.pause() # does not work under Windows
while(True):
  time.sleep(1)

启动包装器，例如：

PythonPrompt> import subprocess
PythonPrompt> subprocess.Popen("start python wrapper.py", shell=True)

您需要添加一些IPC代码，允许您控制包装器触发os.kill（signal.CTRL_C_EVENT，0）命令。我在我的应用程序中使用了套接字。

说明：

Preinformation

send_signal(CTRL_C_EVENT)不起作用，因为CTRL_C_EVENT仅适用于os.kill。 [REF1]
os.kill(CTRL_C_EVENT)将信号发送到当前cmd窗口[REF2]中运行的所有进程
Popen(..., creationflags=CREATE_NEW_PROCESS_GROUP)不起作用，因为进程组忽略了CTRL_C_EVENT。 [REF2]这是python文档[REF3]中的一个错误

实施解决方案

让您的程序在Windows shell命令start的不同cmd窗口中运行。
在控制应用程序和应该获得CTRL-C信号的应用程序之间添加一个CTRL-C请求包装器。包装器将在与应该获得CTRL-C信号的应用程序相同的cmd窗口中运行。
包装器将关闭自身以及应该通过向cmd窗口中的所有进程发送CTRL_C_EVENT来获取CTRL-C信号的程序。
控制程序应该能够请求包装器发送CTRL-C信号。这可以通过IPC方式实现，例如插座。

有用的帖子是：

我不得不删除链接前面的http，因为我是新用户，不允许发布两个以上的链接。

更新：基于IPC的CTRL-C Wrapper

在这里，您可以找到一个自写的python模块，提供CTRL-C包装，包括基于套接字的IPC。语法与子进程模块非常相似。

用法：

>>> import winctrlc
>>> p1 = winctrlc.Popen("python demo.py")
>>> p2 = winctrlc.Popen("python demo.py")
>>> p3 = winctrlc.Popen("python demo.py")
>>> p2.send_ctrl_c()
>>> p1.send_ctrl_c()
>>> p3.send_ctrl_c()

码

import socket
import subprocess
import time
import random
import signal, os, sys


class Popen:
  _port = random.randint(10000, 50000)
  _connection = ''

  def _start_ctrl_c_wrapper(self, cmd):
    cmd_str = "start \"\" python winctrlc.py "+"\""+cmd+"\""+" "+str(self._port)
    subprocess.Popen(cmd_str, shell=True)

  def _create_connection(self):
    self._connection = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self._connection.connect(('localhost', self._port))

  def send_ctrl_c(self):
    self._connection.send(Wrapper.TERMINATION_REQ)
    self._connection.close()

  def __init__(self, cmd):
    self._start_ctrl_c_wrapper(cmd)
    self._create_connection()


class Wrapper:
  TERMINATION_REQ = "Terminate with CTRL-C"

  def _create_connection(self, port):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(('localhost', port))
    s.listen(1)
    conn, addr = s.accept()
    return conn

  def _wait_on_ctrl_c_request(self, conn):
    while True:
      data = conn.recv(1024)
      if data == self.TERMINATION_REQ:
        ctrl_c_received = True
        break
      else:
        ctrl_c_received = False
    return ctrl_c_received

  def _cleanup_and_fire_ctrl_c(self, conn):
    conn.close()
    os.kill(signal.CTRL_C_EVENT, 0)

  def _signal_handler(self, signal, frame):
    time.sleep(1)
    sys.exit(0)

  def __init__(self, cmd, port):
    signal.signal(signal.SIGINT, self._signal_handler)
    subprocess.Popen(cmd)
    conn = self._create_connection(port)
    ctrl_c_req_received = self._wait_on_ctrl_c_request(conn)
    if ctrl_c_req_received:
      self._cleanup_and_fire_ctrl_c(conn)
    else:
      sys.exit(0)


if __name__ == "__main__":
  command_string = sys.argv[1]
  port_no = int(sys.argv[2])
  Wrapper(command_string, port_no)

Answer 2

尝试使用GenerateConsoleCtrlEvent调用ctypes函数。在创建新进程组时，进程组ID应与pid相同。所以，像

import ctypes

ctypes.windll.kernel32.GenerateConsoleCtrlEvent(0, proc.pid) # 0 => Ctrl-C

应该管用。

更新：你是对的，我错过了那部分细节。这是a post，它提出了一个可能的解决方案，尽管它有点笨拙。更多细节在this answer。

Answer 3

这是一个完整的示例，不需要在目标脚本中进行任何修改。

这会覆盖sitecustomize模块，因此可能不适合所有情况。但是，在这种情况下，您可以使用site-packages中的* .pth文件在子进程启动时执行代码（请参阅https://nedbatchelder.com/blog/201001/running_code_at_python_startup.html）。

编辑这只适用于Python中子进程的开箱即用。其他进程必须手动调用SetConsoleCtrlHandler(NULL, FALSE)。

卖弄.朋友

import os
import signal
import subprocess
import sys
import time


def main():
    env = os.environ.copy()
    env['PYTHONPATH'] = '%s%s%s' % ('custom-site', os.pathsep,
                                    env.get('PYTHONPATH', ''))
    proc = subprocess.Popen(
        [sys.executable, 'sub.py'],
        env=env,
        creationflags=subprocess.CREATE_NEW_PROCESS_GROUP,
        )
    time.sleep(1)
    proc.send_signal(signal.CTRL_C_EVENT)
    proc.wait()


if __name__ == '__main__':
    main()

定制网站\ sitecustomize.py

import ctypes
import sys
kernel32 = ctypes.WinDLL('kernel32', use_last_error=True)

if not kernel32.SetConsoleCtrlHandler(None, False):
    print('SetConsoleCtrlHandler Error: ', ctypes.get_last_error(),
          file=sys.stderr)

苏北.朋友

import atexit
import time


def cleanup():
    print ('cleanup')

atexit.register(cleanup)


while True:
    time.sleep(1)

Answer 4

我一直在尝试这个但由于某种原因ctrl + break工作，而ctrl + c没有。所以使用os.kill(signal.CTRL_C_EVENT, 0)失败，但做os.kill(signal.CTRL_C_EVENT, 1)工作。我被告知这与创建进程所有者是唯一可以传递ctrl c的人有关吗？那有意义吗？

为了澄清，在命令窗口中手动运行fio时，它似乎正在按预期运行。使用CTRL + BREAK中断而不按预期存储日志，CTRL + C也按预期完成写入文件。问题似乎出现在CTRL_C_EVENT的信号中。

它几乎似乎是Python中的一个错误，但可能是Windows中的一个错误。还有一件事，我有一个cygwin版本运行并在python中发送ctrl + c也有效，但是我们再没有真正在那里运行本机窗口。

例：

import subprocess, time, signal, sys, os
command = '"C:\\Program Files\\fio\\fio.exe" --rw=randrw --bs=1M --numjobs=8 --iodepth=64 --direct=1 ' \
    '--sync=0 --ioengine=windowsaio --name=test --loops=10000 ' \
    '--size=99901800 --rwmixwrite=100 --do_verify=0 --filename=I\\:\\test ' \
    '--thread --output=C:\\output.txt'
def signal_handler(signal, frame):
  time.sleep(1)
  print 'Ctrl+C received in wrapper.py'

signal.signal(signal.SIGINT, signal_handler)
print 'command Starting'
subprocess.Popen(command)
print 'command started'
time.sleep(15) 
print 'Timeout Completed'
os.kill(signal.CTRL_C_EVENT, 0)

在Windows上将^ C发送到Python子进程对象

问题描述投票：22回答：4

4个回答

最新问题

在Windows上将^ C发送到Python子进程对象

问题描述 投票：22回答：4

4个回答

最新问题

问题描述投票：22回答：4