Python如何实现进程迁移？能否向其他进程内存空间注入内容？

Python对进程操作能力的说明 #

Python本身作为高层语言，没有直接封装内核级的进程操作API，但可以通过调用系统原生接口、第三方扩展库实现你提到的两类操作，具体方案如下：

1. 跨进程内存注入 #

• Windows平台：可以使用pymem库完成，核心操作包括打开目标进程、获取进程句柄、调用VirtualAllocEx申请远程内存、WriteProcessMemory写入注入内容、CreateRemoteThread启动执行逻辑，简单的示例代码结构如下：

import pymem

# 打开目标进程，假设目标进程名是notepad.exe
pm = pymem.Pymem("notepad.exe")
# 申请远程内存空间
remote_addr = pm.allocate(1024)
# 向远程内存写入数据
pm.write_bytes(remote_addr, b"your inject content", 1024)

• Linux平台：可以通过ptrace系统调用实现，第三方库python-ptrace封装了相关接口，核心流程是attach到目标进程、读取/修改寄存器、写入内存数据，完成注入后detach即可。

注意：跨进程注入操作需要当前进程拥有管理员/root权限，部分系统的安全策略会拦截此类操作，仅可用于合法的调试、二次开发场景，禁止用于未授权的恶意操作。

2. 进程迁移 #

如果是指将运行中的进程从一个主机迁移到另一个主机，Python没有原生支持，可行的方案是配合系统级的进程 checkpoint 工具实现：

• 可以调用Linux下的criu（Checkpoint/Restore In Userspace）工具完成进程的快照、迁移、恢复，Python侧只需要用subprocess模块封装对应的命令行调用即可，示例结构如下：

import subprocess

# 对指定PID的进程做快照
subprocess.run(["criu", "dump", "-D", "/tmp/checkpoint_dir", "-t", str(pid)], check=True)
# 快照文件迁移到目标主机后执行恢复
subprocess.run(["criu", "restore", "-D", "/tmp/checkpoint_dir"], check=True)

如果是指同一主机内的进程CPU核心绑定类的迁移，可以用psutil库的cpu_affinity接口直接设置，代码示例：

import psutil

p = psutil.Process(pid=1234)
# 将进程绑定到0、1号CPU核心
p.cpu_affinity([0, 1])

参考说明 #

所有提到的第三方库的使用说明都可以在PyPI的对应包介绍页面查询。

内容的提问来源于stack exchange，提问作者John