1971462beSGnoCiYeH use alloc::{string::String, sync::Arc, vec::Vec}; 21496ba7bSLoGin 31496ba7bSLoGin use crate::{ 41496ba7bSLoGin arch::{ 51496ba7bSLoGin interrupt::TrapFrame, 61496ba7bSLoGin process::table::{USER_CS, USER_DS}, 71496ba7bSLoGin CurrentIrqArch, 81496ba7bSLoGin }, 91496ba7bSLoGin exception::InterruptArch, 101496ba7bSLoGin mm::ucontext::AddressSpace, 111496ba7bSLoGin process::{ 121496ba7bSLoGin exec::{load_binary_file, ExecParam, ExecParamFlags}, 13971462beSGnoCiYeH ProcessControlBlock, ProcessManager, 141496ba7bSLoGin }, 15971462beSGnoCiYeH syscall::{user_access::UserBufferWriter, Syscall, SystemError}, 161496ba7bSLoGin }; 171496ba7bSLoGin 181496ba7bSLoGin impl Syscall { 191496ba7bSLoGin pub fn do_execve( 201496ba7bSLoGin path: String, 211496ba7bSLoGin argv: Vec<String>, 221496ba7bSLoGin envp: Vec<String>, 231496ba7bSLoGin regs: &mut TrapFrame, 241496ba7bSLoGin ) -> Result<(), SystemError> { 25*bf4a4899SLoGin // 关中断,防止在设置地址空间的时候,发生中断,然后进调度器,出现错误。 26*bf4a4899SLoGin let irq_guard = unsafe { CurrentIrqArch::save_and_disable_irq() }; 27*bf4a4899SLoGin let pcb = ProcessManager::current_pcb(); 28*bf4a4899SLoGin // crate::kdebug!( 29*bf4a4899SLoGin // "pid: {:?} do_execve: path: {:?}, argv: {:?}, envp: {:?}\n", 30*bf4a4899SLoGin // pcb.pid(), 311496ba7bSLoGin // path, 321496ba7bSLoGin // argv, 331496ba7bSLoGin // envp 341496ba7bSLoGin // ); 351496ba7bSLoGin 361496ba7bSLoGin let mut basic_info = pcb.basic_mut(); 371496ba7bSLoGin // 暂存原本的用户地址空间的引用(因为如果在切换页表之前释放了它,可能会造成内存use after free) 381496ba7bSLoGin let old_address_space = basic_info.user_vm(); 391496ba7bSLoGin 401496ba7bSLoGin // 在pcb中原来的用户地址空间 411496ba7bSLoGin unsafe { 421496ba7bSLoGin basic_info.set_user_vm(None); 431496ba7bSLoGin } 441496ba7bSLoGin // 创建新的地址空间并设置为当前地址空间 451496ba7bSLoGin let address_space = AddressSpace::new(true).expect("Failed to create new address space"); 461496ba7bSLoGin unsafe { 471496ba7bSLoGin basic_info.set_user_vm(Some(address_space.clone())); 481496ba7bSLoGin } 491496ba7bSLoGin 501496ba7bSLoGin // to avoid deadlock 511496ba7bSLoGin drop(basic_info); 521496ba7bSLoGin 531496ba7bSLoGin assert!( 541496ba7bSLoGin AddressSpace::is_current(&address_space), 551496ba7bSLoGin "Failed to set address space" 561496ba7bSLoGin ); 571496ba7bSLoGin // kdebug!("Switch to new address space"); 581496ba7bSLoGin 591496ba7bSLoGin // 切换到新的用户地址空间 601496ba7bSLoGin unsafe { address_space.read().user_mapper.utable.make_current() }; 611496ba7bSLoGin 621496ba7bSLoGin drop(old_address_space); 631496ba7bSLoGin drop(irq_guard); 641496ba7bSLoGin // kdebug!("to load binary file"); 651496ba7bSLoGin let mut param = ExecParam::new(path.as_str(), address_space.clone(), ExecParamFlags::EXEC); 661496ba7bSLoGin 671496ba7bSLoGin // 加载可执行文件 681496ba7bSLoGin let load_result = load_binary_file(&mut param) 691496ba7bSLoGin .unwrap_or_else(|e| panic!("Failed to load binary file: {:?}, path: {:?}", e, path)); 701496ba7bSLoGin // kdebug!("load binary file done"); 711496ba7bSLoGin // kdebug!("argv: {:?}, envp: {:?}", argv, envp); 721496ba7bSLoGin param.init_info_mut().args = argv; 731496ba7bSLoGin param.init_info_mut().envs = envp; 741496ba7bSLoGin 751496ba7bSLoGin // 把proc_init_info写到用户栈上 761496ba7bSLoGin 771496ba7bSLoGin let (user_sp, argv_ptr) = unsafe { 781496ba7bSLoGin param 791496ba7bSLoGin .init_info() 801496ba7bSLoGin .push_at( 811496ba7bSLoGin address_space 821496ba7bSLoGin .write() 831496ba7bSLoGin .user_stack_mut() 841496ba7bSLoGin .expect("No user stack found"), 851496ba7bSLoGin ) 861496ba7bSLoGin .expect("Failed to push proc_init_info to user stack") 871496ba7bSLoGin }; 881496ba7bSLoGin 891496ba7bSLoGin // kdebug!("write proc_init_info to user stack done"); 901496ba7bSLoGin 911496ba7bSLoGin // (兼容旧版libc)把argv的指针写到寄存器内 921496ba7bSLoGin // TODO: 改写旧版libc,不再需要这个兼容 931496ba7bSLoGin regs.rdi = param.init_info().args.len() as u64; 941496ba7bSLoGin regs.rsi = argv_ptr.data() as u64; 951496ba7bSLoGin 961496ba7bSLoGin // 设置系统调用返回时的寄存器状态 971496ba7bSLoGin // TODO: 中断管理重构后,这里的寄存器状态设置要删掉!!!改为对trap frame的设置。要增加架构抽象。 981496ba7bSLoGin regs.rsp = user_sp.data() as u64; 991496ba7bSLoGin regs.rbp = user_sp.data() as u64; 1001496ba7bSLoGin regs.rip = load_result.entry_point().data() as u64; 1011496ba7bSLoGin 1021496ba7bSLoGin regs.cs = USER_CS.bits() as u64; 1031496ba7bSLoGin regs.ds = USER_DS.bits() as u64; 1041496ba7bSLoGin regs.ss = USER_DS.bits() as u64; 1051496ba7bSLoGin regs.es = 0; 1061496ba7bSLoGin regs.rflags = 0x200; 1071496ba7bSLoGin regs.rax = 1; 1081496ba7bSLoGin 109971462beSGnoCiYeH drop(param); 110971462beSGnoCiYeH 1111496ba7bSLoGin // kdebug!("regs: {:?}\n", regs); 1121496ba7bSLoGin 113*bf4a4899SLoGin // crate::kdebug!( 1141496ba7bSLoGin // "tmp_rs_execve: done, load_result.entry_point()={:?}", 1151496ba7bSLoGin // load_result.entry_point() 1161496ba7bSLoGin // ); 117876cb89eSGnoCiYeH 1181496ba7bSLoGin return Ok(()); 1191496ba7bSLoGin } 120971462beSGnoCiYeH 121971462beSGnoCiYeH /// ## 用于控制和查询与体系结构相关的进程特定选项 122971462beSGnoCiYeH pub fn arch_prctl(option: usize, arg2: usize) -> Result<usize, SystemError> { 123971462beSGnoCiYeH let pcb = ProcessManager::current_pcb(); 124971462beSGnoCiYeH if let Err(SystemError::EINVAL) = Self::do_arch_prctl_64(&pcb, option, arg2, true) { 125971462beSGnoCiYeH Self::do_arch_prctl_common(option, arg2)?; 1261496ba7bSLoGin } 127971462beSGnoCiYeH Ok(0) 128971462beSGnoCiYeH } 129971462beSGnoCiYeH 130971462beSGnoCiYeH /// ## 64位下控制fs/gs base寄存器的方法 131971462beSGnoCiYeH pub fn do_arch_prctl_64( 132971462beSGnoCiYeH pcb: &Arc<ProcessControlBlock>, 133971462beSGnoCiYeH option: usize, 134971462beSGnoCiYeH arg2: usize, 135971462beSGnoCiYeH from_user: bool, 136971462beSGnoCiYeH ) -> Result<usize, SystemError> { 137971462beSGnoCiYeH let mut arch_info = pcb.arch_info_irqsave(); 138971462beSGnoCiYeH match option { 139971462beSGnoCiYeH ARCH_GET_FS => { 140971462beSGnoCiYeH unsafe { arch_info.save_fsbase() }; 141971462beSGnoCiYeH let mut writer = UserBufferWriter::new( 142971462beSGnoCiYeH arg2 as *mut usize, 143971462beSGnoCiYeH core::mem::size_of::<usize>(), 144971462beSGnoCiYeH from_user, 145971462beSGnoCiYeH )?; 146971462beSGnoCiYeH writer.copy_one_to_user(&arch_info.fsbase, 0)?; 147971462beSGnoCiYeH } 148971462beSGnoCiYeH ARCH_GET_GS => { 149971462beSGnoCiYeH unsafe { arch_info.save_gsbase() }; 150971462beSGnoCiYeH let mut writer = UserBufferWriter::new( 151971462beSGnoCiYeH arg2 as *mut usize, 152971462beSGnoCiYeH core::mem::size_of::<usize>(), 153971462beSGnoCiYeH from_user, 154971462beSGnoCiYeH )?; 155971462beSGnoCiYeH writer.copy_one_to_user(&arch_info.gsbase, 0)?; 156971462beSGnoCiYeH } 157971462beSGnoCiYeH ARCH_SET_FS => { 158971462beSGnoCiYeH arch_info.fsbase = arg2; 159971462beSGnoCiYeH // 如果是当前进程则直接写入寄存器 160971462beSGnoCiYeH if pcb.pid() == ProcessManager::current_pcb().pid() { 161971462beSGnoCiYeH unsafe { arch_info.restore_fsbase() } 162971462beSGnoCiYeH } 163971462beSGnoCiYeH } 164971462beSGnoCiYeH ARCH_SET_GS => { 165971462beSGnoCiYeH arch_info.gsbase = arg2; 166971462beSGnoCiYeH if pcb.pid() == ProcessManager::current_pcb().pid() { 167971462beSGnoCiYeH unsafe { arch_info.restore_gsbase() } 168971462beSGnoCiYeH } 169971462beSGnoCiYeH } 170971462beSGnoCiYeH _ => { 171971462beSGnoCiYeH return Err(SystemError::EINVAL); 172971462beSGnoCiYeH } 173971462beSGnoCiYeH } 174971462beSGnoCiYeH Ok(0) 175971462beSGnoCiYeH } 176971462beSGnoCiYeH 177971462beSGnoCiYeH #[allow(dead_code)] 178971462beSGnoCiYeH pub fn do_arch_prctl_common(_option: usize, _arg2: usize) -> Result<usize, SystemError> { 179971462beSGnoCiYeH todo!("do_arch_prctl_common not unimplemented"); 180971462beSGnoCiYeH } 181971462beSGnoCiYeH } 182971462beSGnoCiYeH 183971462beSGnoCiYeH pub const ARCH_SET_GS: usize = 0x1001; 184971462beSGnoCiYeH pub const ARCH_SET_FS: usize = 0x1002; 185971462beSGnoCiYeH pub const ARCH_GET_FS: usize = 0x1003; 186971462beSGnoCiYeH pub const ARCH_GET_GS: usize = 0x1004; 187