GPU虚拟化实现（七）

章节回顾

在上一章，分析了项目的主要代码模块功能：共享内存和初始化、GPU 内存管理、GPU 利用率管理以及锁机制，在这一章将继续分析其他的代码模块功能。

进程管理

void init_proc_slot_withlock() {
   
    int32_t current_pid = getpid(); // 获取当前进程的 PID
    lock_shrreg(); // 加锁
    shared_region_t* region = region_info.shared_region; // 获取共享内存区域 shared_region 的指针
    // 检查共享内存中的进程计数 proc_num 是否达到最大限制
    if (region->proc_num >= SHARED_REGION_MAX_PROCESS_NUM) {
   
        exit_withlock(-1); // 退出程序
    }

	// 调用 signal() 函数为 SIGUSR1 和 SIGUSR2 信号注册处理函数
    signal(SIGUSR2,sig_swap_stub);
    signal(SIGUSR1,sig_restore_stub);
    // If, by any means a pid of itself is found in region->proces, then it is probably caused by crashloop
    // we need to reset it.
    int i,found=0;
    // 遍历共享内存中的进程槽位数组 region->procs，检查是否存在当前进程的 PID（current_pid）
    for (i=0; i<region->proc_num; i++) {
   
	    // 如果找到，将该槽位的状态 status 设置为 1（表示活跃），并用 memset() 清空该槽位的两个数组
        if (region->procs[i].pid == current_pid) {
   
            region->procs[i].status = 1;
            memset(region->procs[i].used,0,sizeof(device_memory_t)*CUDA_DEVICE_MAX_COUNT);
            memset(region->procs[i].device_util,0,sizeof(device_util_t)*CUDA_DEVICE_MAX_COUNT);
            found = 1;
            break;
        }
    }
    // 如果未找到自己的 PID（found == 0），在当前进程计数 proc_num 对应的槽位上分配一个新槽位
    if (!found) {
   
        region->procs[region->proc_num].pid = current_pid;
        region->procs[region->proc_num].status = 1;
        memset(region->procs[region->proc_num].used,0,sizeof(device_memory_t)*CUDA_DEVICE_MAX_COUNT);
        memset(region->procs[region->proc_num].device_util,0,sizeof(device_util_t)*CUDA_DEVICE_MAX_COUNT);
        region->proc_num++;
    }

    clear_proc_slot_nolock(current_pid, 1);
    unlock_shrreg();
}

• 为当前进程在共享内存区域中分配或重用一个进程槽位，初始化该槽位的状态和数据，并设置信号处理函数以支持进程间通信或恢复机制

int rm_quitted_process(){
   
	// 使用 popen() 调用系统命令 ps ax，并以只读模式（"r"）打开一个文件流 wstream，用于读取命令输出。ps ax 会列出系统中所有运行进程的信息
    FILE *wstream;
    wstream=popen("ps ax","r");
    char tmp[256];
    char *atmp;
    int pidmap[SHARED_REGION_MAX_PROCESS_NUM];
    memset(pidmap,0,sizeof(int)*SHARED_REGION_MAX_PROCESS_NUM);
    ensure_initialized();

    int32_t pid;
    int i = 0,cnt=0,ret=0;
    LOG_INFO("rm_quitted_process");
    lock_shrreg();
    if (wstream!=NULL){
   
        while (fgets(tmp,256,wstream)) {
   
            atmp = strtok(tmp," ");
            pid = atoi(atmp);
            if (pid!=0)
                for (i=0;i<region_info.shared_region->proc_num;i++)
                    if (region_info.shared_region->procs[i].pid==pid){
   
                        pidmap[i]=1;
                    }
        }
        // 遍历共享内存中的进程列表 procs，检查 pidmap[i] 的值
        for (i=0;i<region_info