【调试工具】taskset 设置处理器的亲和度

一、功能介绍

taskset命令用于设置进程（或 线程）的处理器亲和性（Processor Affinity），可以将进程（或 线程）绑定到特定的一个 或 多个CPU上去执行，而不允许将进程（或 线程）调度到其他的CPU上。

---

二、用户进程使用

1.编写测试程序

#include <stdio.h>
#include <stdlib.h>

int main(int argc, char **argv)
{
    char *env;

    env = getenv("PATH");
    
    printf("PATH: %s\n", env);
    
    while(1);
    
   return 0;
}

2.查看进程

输出结构处理器亲和性掩码是f，表示进程（或线程）可以在Host上让任何一个CPU运行。查看进程（或 线程）允许允许CPU范围使用-c参数。

cpu核心数为4，分别为cpu0、cpu1、cpu2、cpu3

3.查看当前在运行在哪个cpu上

pidstat -p 2778

可以看到当前是在cpu0上运行。

4.绑定到其它单个cpu

taskset -p 01 2778   // 绑定到 cpu 1

taskset -p 02 2778  // 绑定到 cpu2

5.绑定CPU亲和性指定一组范围

taskset -pc 1,2,3,4 pid  // -c指定cpu 1 2 3 4
taskset -p 0x7 pid    // 0x7 – 0111, 指定cpu 1 2 3

---

三、内核线程使用

内核中是调用 set_cpus_allowed_ptr 函数去实现的，示例如下：

#include<linux/kernel.h>
#include<linux/module.h>
#include<linux/kthread.h>
 
struct task_struct *task0;
static spinlock_t spinlock;
int val;
 
int task(void *arg)
{
    printk(KERN_INFO "%s:%d\n",__func__,__LINE__);
    /* To generate panic uncomment following */
    /* panic("softlockup: hung tasks"); */
 
    while(!kthread_should_stop()) {
        printk(KERN_INFO "%s:%d\n",__func__,__LINE__);
        spin_lock(&spinlock);
        /* busy loop in critical section */
        while(1) {
            printk(KERN_INFO "%s:%d\n",__func__,__LINE__);
        }
 
        spin_unlock(&spinlock);
    }
 
    return val;
}
 
static int softlockup_init(void)
{
    printk(KERN_INFO "%s:%d\n",__func__,__LINE__);
 
    val = 1;
    spin_lock_init(&spinlock);
    task0 = kthread_run(&task,(void *)val,"softlockup_thread");
    set_cpus_allowed_ptr(task0, cpumask_of(0));  //调用该接口绑定
 
    return 0;
}
 
static void softlockup_exit(void)
{
    printk(KERN_INFO "%s:%d\n",__func__,__LINE__);
    kthread_stop(task0);
}
 
module_init(softlockup_init);
module_exit(softlockup_exit);

四、总结

在项目开发过程中，某些cpu的使用率很高，而有些使用率较低，我们可以尝试使用设置cpu的亲和度来调整不同cpu上的task