1 #pragma once 2 3 #include <DragonOS/signal.h> 4 #include <common/wait_queue.h> 5 #include <DragonOS/stdint.h> 6 #include "ptrace.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 /** 73 * @brief 进程控制块 74 * 75 */ 76 struct process_control_block 77 { 78 // 进程的状态 79 volatile uint64_t state; 80 // 进程标志：进程、线程、内核线程 81 uint64_t flags; 82 int32_t preempt_count; // 持有的自旋锁的数量 83 84 uint32_t cpu_id; // 当前进程在哪个CPU核心上运行 85 char name[PCB_NAME_LEN]; 86 87 // 内存空间分布结构体， 记录内存页表和程序段信息 88 struct mm_struct *mm; 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 109 // 进程拥有的文件描述符的指针数组 110 // todo: 改用动态指针数组 111 struct vfs_file_t *fds[PROC_MAX_FD_NUM]; 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 }; 134 135 // 将进程的pcb和内核栈融合到一起,8字节对齐 136 union proc_union { 137 struct process_control_block pcb; 138 ul stack[STACK_SIZE / sizeof(ul)]; 139 } __attribute__((aligned(8))); 140 141 struct tss_struct 142 { 143 unsigned int reserved0; 144 ul rsp0; 145 ul rsp1; 146 ul rsp2; 147 ul reserved1; 148 ul ist1; 149 ul ist2; 150 ul ist3; 151 ul ist4; 152 ul ist5; 153 ul ist6; 154 ul ist7; 155 ul reserved2; 156 unsigned short reserved3; 157 // io位图基地址 158 unsigned short io_map_base_addr; 159 } __attribute__((packed)); // 使用packed表明是紧凑结构，编译器不会对成员变量进行字节对齐。