Linux Ring3 设置当前进程cpu亲和性

最近在研究设置进程cpu亲和性的相关工作,这里一起分享一下，这个主要还是参考博客大佬；

主要还是使用以下几个函数，具体的使用方法，这里就不多加说明，大佬们说的比较清楚，我这里直接分享写的代码；

//进程
sched_setaffinity(0, sizeof(cpu_set_t), &mask);
sched_getaffinity(0, sizeof(cpu_set_t), &get);
//线程
pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask）
pthread_getaffinity_np(pthread_self(), sizeof(get), &get）
sysconf(_SC_NPROCESSORS_CONF);
CPU_ZERO(&mask);
CPU_ISSET(i, &get)
CPU_SET(cpu, &mask);

//set_task_cpu.c
#define _GNU_SOURCE
#include <sched.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>


static int set_task_cpu(int cpu)
{
	cpu_set_t mask;
	int ret = 0;
	CPU_ZERO(&mask);
	CPU_SET(cpu, &mask);
	ret = sched_setaffinity(0, sizeof(cpu_set_t), &mask);
	if(ret < 0)
	{
		perror("sched_setaffinity");
	}

	printf("set pid %d to cpu %d success\n",getpid(),cpu);
	return ret;
}

static int get_task_cpu(void)
{
	unsigned long int cpus = 0;
	int ret = 0;
	int i = 0;
	cpu_set_t get;

	printf("system cpu num is %ld\n",sysconf(_SC_NPROCESSORS_CONF));
	printf("system enable cpu num is %ld\n",sysconf(_SC_NPROCESSORS_ONLN));
    
    cpus = sysconf(_SC_NPROCESSORS_CONF);

	CPU_ZERO(&get);
	ret = sched_getaffinity(0, sizeof(cpu_set_t), &get);

	if(ret < 0)
	{
		perror("sched_getaffinity");
	}

	for(i = 0; i < cpus; i++)
	{
		if(CPU_ISSET(i, &get))
		{
			printf("This porcess %d is runing processor %d\n",getpid(),i);
		}
	}

	return 0;

}

int main(char argc, char **argv)
{
	int cpu = 0;

	printf("Enter into set task cpu\n");
	if(argc<2)
	{
		printf("Please input what you want set task cpu number!\n");
		return ;
	}

	get_task_cpu();
	cpu = atoi(argv[1]);
	set_task_cpu(cpu);	
	get_task_cpu();

	return 0;

}

打印信息：

curtis@curtis-virtual-machine:/mnt/hgfs/share/write_code/set_task_cpu$ ./a.out 0
Enter into set task cpu
system cpu num is 2
system enable cpu num is 2
This porcess 3914 is runing processor 0
This porcess 3914 is runing processor 1
set pid 3914 to cpu 0 success
system cpu num is 2
system enable cpu num is 2
This porcess 3914 is runing processor 0

主要逻辑：清空cpu集合 --> 添加所需cpu集合 --> 设置进程亲和性；
首先获得当前系统所有cpu的核数，我这里是虚拟机分配了两个核，sysconf()函数返回的数据是正确的，然后去判断当前进程可以在哪些cpu上执行，也可以写个驱动查看task_struct -> cpus_allowed的mask,这个位掩码由 n 位组成，与系统中的 n 个逻辑处理器一一对应，Linux缺省状态下mask位全为1，也就是说当前进程可以在cpu0,cpu1上跑，然后调用set_task_set（）函数，设置cpu的亲和性，这里传入的参数是0，再次检查cpu亲和位cpu0，说明设置成功。