1 #pragma once 2 3 #include "ptrace.h" 4 #include <DragonOS/signal.h> 5 #include <DragonOS/stdint.h> 6 #include <common/wait_queue.h> 7 8 // 进程最大可拥有的文件描述符数量 9 #define PROC_MAX_FD_NUM 16 10 11 // 进程的内核栈大小 32K 12 #define STACK_SIZE 32768 13 14 // 进程的运行状态 15 // 正在运行 16 #define PROC_RUNNING (1 << 0) 17 // 可被信号打断 18 #define PROC_INTERRUPTIBLE (1 << 1) 19 // 不可被信号打断 20 #define PROC_UNINTERRUPTIBLE (1 << 2) 21 // 挂起 22 #define PROC_ZOMBIE (1 << 3) 23 // 已停止 24 #define PROC_STOPPED (1 << 4) 25 26 // 内核代码段基地址 27 #define KERNEL_CS (0x08) 28 // 内核数据段基地址 29 #define KERNEL_DS (0x10) 30 // 用户代码段基地址 31 #define USER_CS (0x28) 32 // 用户数据段基地址 33 #define USER_DS (0x30) 34 35 // 进程初始化时的数据拷贝标志位 36 #define CLONE_FS (1UL << 0) // 在进程间共享打开的文件 37 #define CLONE_SIGNAL (1UL << 1) // 克隆时，与父进程共享信号结构体 38 #define CLONE_VM (1UL << 2) // 在进程间共享虚拟内存空间 39 #define CLONE_SIGHAND (1UL << 3) // 克隆时，与父进程共享信号处理结构体 40 #define CLONE_CLEAR_SIGHAND (1UL << 4) // 克隆时，将原本被设置为SIG_IGNORE的信号，设置回SIG_DEFAULT 41 #define CLONE_THREAD (1UL << 5) // 拷贝线程 42 #define PCB_NAME_LEN 16 43 44 struct thread_struct 45 { 46 // 内核层栈基指针 47 ul rbp; // in tss rsp0 48 // 内核层代码指针 49 ul rip; 50 // 内核层栈指针 51 ul rsp; 52 53 ul fs, gs; 54 55 ul cr2; 56 // 异常号 57 ul trap_num; 58 // 错误码 59 ul err_code; 60 }; 61 62 // ========= pcb->flags ========= 63 // 进程标志位 64 #define PF_KTHREAD (1UL << 0) // 内核线程 65 #define PF_NEED_SCHED (1UL << 1) // 进程需要被调度 66 #define PF_VFORK (1UL << 2) // 标志进程是否由于vfork而存在资源共享 67 #define PF_KFORK (1UL << 3) // 标志在内核态下调用fork（临时标记，do_fork()结束后会将其复位） 68 #define PF_NOFREEZE (1UL << 4) // 当前进程不能被冻结 69 #define PF_EXITING (1UL << 5) // 进程正在退出 70 #define PF_WAKEKILL (1UL << 6) // 进程由于接收到终止信号唤醒 71 #define PF_SIGNALED (1UL << 7) // 进程由于接收到信号而退出 72 #define PF_NEED_MIGRATE (1UL << 8) // 进程需要迁移到其他的核心 73 74 /** 75 * @brief 进程控制块 76 * 77 */ 78 struct process_control_block 79 { 80 // 进程的状态 81 volatile uint64_t state; 82 // 进程标志：进程、线程、内核线程 83 uint64_t flags; 84 int32_t preempt_count; // 持有的自旋锁的数量 85 86 uint32_t cpu_id; // 当前进程在哪个CPU核心上运行 87 // pcb的名字 88 char name[PCB_NAME_LEN]; 89 90 // 进程切换时保存的状态信息 91 struct thread_struct *thread; 92 93 // pcb加入调度队列时，所使用的链表节点 94 struct List list; 95 96 // todo:给pcb中加一个spinlock_t成员 97 // 进程自旋锁 98 // spinlock_t alloc_lock; 99 100 // 地址空间范围 101 // 用户空间： 0x0000 0000 0000 0000 ~ 0x0000 7fff ffff ffff 102 // 内核空间： 0xffff 8000 0000 0000 ~ 0xffff ffff ffff ffff 103 uint64_t addr_limit; 104 105 long pid; 106 long priority; // 优先级 107 int64_t virtual_runtime; // 虚拟运行时间 108 int64_t rt_time_slice; // 由实时调度器管理的时间片 109 110 // 进程拥有的文件描述符的指针数组(由Rust进行管理) 111 void *fds; 112 113 // 链表中的下一个pcb 114 struct process_control_block *prev_pcb, *next_pcb; 115 // 父进程的pcb 116 struct process_control_block *parent_pcb; 117 118 int32_t exit_code; // 进程退出时的返回码 119 uint32_t policy; // 进程调度策略标志位 120 wait_queue_node_t wait_child_proc_exit; // 子进程退出等待队列 121 122 /* PF_kTHREAD | PF_IO_WORKER 的进程，worker_private不为NULL*/ 123 void *worker_private; 124 125 // ==== 信号处理相关 ===== 126 struct signal_struct *signal; 127 struct sighand_struct *sighand; 128 // 一个bitmap，表示当前进程被禁用的信号 129 sigset_t sig_blocked; 130 // 正在等待的信号的标志位，表示某个信号正在等待处理 131 struct sigpending sig_pending; 132 133 // 如果当前进程等待被迁移到另一个cpu核心上（也就是flags中的PF_NEED_MIGRATE被置位）， 134 // 该字段存储要被迁移到的目标处理器核心号 135 uint32_t migrate_to; 136 // Fpstate 用于用户态切换到内核态时保存浮点寄存器里面的值 137 void *fp_state; 138 // 指向进程的地址空间的arc指针. 139 void *address_space; 140 }; 141 142 // 将进程的pcb和内核栈融合到一起,8字节对齐 143 union proc_union 144 { 145 struct process_control_block pcb; 146 ul stack[STACK_SIZE / sizeof(ul)]; 147 } __attribute__((aligned(8))); 148 149 struct tss_struct 150 { 151 unsigned int reserved0; 152 ul rsp0; 153 ul rsp1; 154 ul rsp2; 155 ul reserved1; 156 ul ist1; 157 ul ist2; 158 ul ist3; 159 ul ist4; 160 ul ist5; 161 ul ist6; 162 ul ist7; 163 ul reserved2; 164 unsigned short reserved3; 165 // io位图基地址 166 unsigned short io_map_base_addr; 167 } __attribute__((packed)); // 使用packed表明是紧凑结构，编译器不会对成员变量进行字节对齐。