如何为运行中的进程按PID设置mempolicy？

确实，set_mempolicy()本身只能修改当前进程或线程的内存策略，没有直接指定PID的参数，但有几种可行的方式给运行中的进程设置NUMA策略：

1. 用numactl的--pid选项直接修改 #

你可能之前没留意到numactl支持--pid参数，专门用来修改已运行进程的内存策略。举几个常用例子：

• 给PID为1234的进程设置全局交错内存策略：

  numactl --interleave=all --pid 1234
  

• 将进程绑定到NUMA节点0和1：

  numactl --membind=0,1 --pid 1234
  

numactl实现这个的核心逻辑是通过ptrace附着到目标进程，注入代码让目标进程自身调用set_mempolicy()或mbind()系统调用——毕竟只有进程自己才能修改自身的内存策略。你可以再查看numactl源码里的attach.c文件，里面专门处理了--pid的附着和系统调用注入流程。

2. 手动基于ptrace编写工具 #

如果需要自定义逻辑，可以自己写工具用ptrace实现：

• 核心步骤：
  • 调用ptrace(PTRACE_ATTACH, pid, NULL, NULL)附着到目标进程（需要root权限）
  • 读取进程当前的寄存器状态，保存现场
  • 修改寄存器，将__NR_set_mempolicy（系统调用号）存入rax寄存器，同时设置好策略类型、节点掩码等参数
  • 触发进程执行系统调用，完成后恢复寄存器状态
  • 调用ptrace(PTRACE_DETACH, pid, NULL, NULL)解除附着

这种方法需要熟悉ptrace的使用和系统调用的参数传递规则，避免破坏目标进程的正常运行。

3. 针对特定内存区域使用mbind #

如果不需要修改进程的全局默认策略，而是要给特定内存区域设置策略，可以让目标进程调用mbind()。同样需要通过ptrace注入调用，或者如果目标进程有内置的IPC机制，也可以触发它主动调用mbind()，但这种场景比较少见。

另外，move_pages()系统调用可以配合内存策略使用，用来将进程已有的内存页面迁移到指定NUMA节点。

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