CPU亲和性设置视频解析，代码示例 sched_setaffinity sched_getaffinity， CPU_ZERO、CPU_SET、CPU_IS

#include <sched.h>  

int sched_getaffinity(pid_t pid, size_t cpusetsize, cpu_set_t *mask);
int sched_setaffinity(pid_t pid, size_t cpusetsize, const cpu_set_t *mask);

cpu_set_t cpuset;  
CPU_ZERO(&cpuset); // 初始化cpuset，所有位都设为0

 CPU_SET(1, &cpuset); // 将cpuset中CPU 1对应的位设置为1

 CPU_CLR(1, &cpuset); // 将cpuset中CPU 1对应的位清除

if (CPU_ISSET(1, &cpuset)) {  
    // 如果cpuset中CPU 1对应的位被设置，则执行此代码块  
}

#include <stdio.h>  // 引入标准输入输出头文件  
#include <stdlib.h> // 引入标准库头文件，用于exit函数  
#include <sched.h>  // 引入调度头文件，提供CPU亲和性相关的函数和宏  
#include <string.h> // 虽然在这个示例中未直接使用，但通常用于字符串操作  
  
int main() {    
    cpu_set_t cpuset;  // 定义一个cpu_set_t类型的变量cpuset，用于表示CPU集合  
    size_t cpusetsize = sizeof(cpu_set_t);  // 获取cpu_set_t类型的大小，用作sched_setaffinity和sched_getaffinity的参数  

    // 验证并打印当前进程的CPU亲和性  
    cpu_set_t get_cpuset;  // 定义一个变量get_cpuset用于获取当前进程的CPU亲和性  
    CPU_ZERO(&get_cpuset);  // 初始化get_cpuset  
  
    // 尝试获取当前进程的CPU亲和性  
    if (sched_getaffinity(0, cpusetsize, &get_cpuset) == -1) {    
        perror("sched_getaffinity failed");  // 如果获取失败，则打印错误信息  
        exit(EXIT_FAILURE);  // 并以失败状态退出程序  
    }    
  
    // 打印当前进程的CPU亲和性  
    printf("Current CPU affinity: ");    
    for (size_t i = 0; i < CPU_SETSIZE; i++) {  // 遍历CPU_SETSIZE范围内的所有CPU编号  
        if (CPU_ISSET(i, &get_cpuset)) {  // 检查get_cpuset中对应CPU的位是否被设置  
            printf("CPU %zu ", i);  // 如果被设置，则打印该CPU编号  
        }    
    }    
    printf("\n");  // 换行  

  
    // 初始化CPU集合，将所有位清零  
    CPU_ZERO(&cpuset);    
  
    // 设置CPU亲和性，将CPU 0和CPU 1的位设置为1，表示希望将进程绑定到这两个CPU上  
    CPU_SET(0, &cpuset);    
    CPU_SET(1, &cpuset);    
  
    // 尝试设置当前进程的CPU亲和性  
    if (sched_setaffinity(0, cpusetsize, &cpuset) == -1) {    
        perror("sched_setaffinity failed");  // 如果设置失败，则打印错误信息  
        exit(EXIT_FAILURE);  // 并以失败状态退出程序  
    }    
  
    // 验证并打印当前进程的CPU亲和性  
    CPU_ZERO(&get_cpuset);  // 初始化get_cpuset  
  
    // 尝试获取当前进程的CPU亲和性  
    if (sched_getaffinity(0, cpusetsize, &get_cpuset) == -1) {    
        perror("sched_getaffinity failed");  // 如果获取失败，则打印错误信息  
        exit(EXIT_FAILURE);  // 并以失败状态退出程序  
    }    
  
    // 打印当前进程的CPU亲和性  
    printf("Current CPU affinity: ");    
    for (size_t i = 0; i < CPU_SETSIZE; i++) {  // 遍历CPU_SETSIZE范围内的所有CPU编号  
        if (CPU_ISSET(i, &get_cpuset)) {  // 检查get_cpuset中对应CPU的位是否被设置  
            printf("CPU %zu ", i);  // 如果被设置，则打印该CPU编号  
        }    
    }    
    printf("\n");  // 换行  
  
    // 假设我们想要清除CPU 1的亲和性设置（仅作为演示，不会实际影响进程）  
    CPU_CLR(1, &cpuset);  // 修改cpuset变量，清除CPU 1的位  
  
    // 注意：这里的CPU_CLR只是修改了cpuset变量，并不会影响已经设置的进程亲和性  
    // 如果要重新设置进程的CPU亲和性，需要再次调用sched_setaffinity  
  
    // （可选）重新打印修改后的cpuset，但请注意它不会反映进程的当前亲和性  
    printf("Modified cpuset (not applied to process): ");    
    for (size_t i = 0; i < CPU_SETSIZE; i++) {  // 再次遍历CPU_SETSIZE范围内的所有CPU编号  
        if (CPU_ISSET(i, &cpuset)) {  // 检查cpuset中对应CPU的位是否被设置  
            printf("CPU %zu ", i);  // 如果被设置，则打印该CPU编号  
        }    
    }    
    printf("\n");  // 换行  
  
    return 0;  // 程序正常结束  
}