xref: /DragonOS/kernel/src/process/syscall.rs (revision edaf015400f83967c2fc940f07be0dbb5792246f) 
1 use core::ffi::c_void;
2 
3 use alloc::{
4     string::{String, ToString},
5     sync::Arc,
6     vec::Vec,
7 };
8 
9 use super::{
10     abi::WaitOption,
11     exit::kernel_wait4,
12     fork::{CloneFlags, KernelCloneArgs},
13     resource::{RLimit64, RLimitID, RUsage, RUsageWho},
14     KernelStack, Pid, ProcessManager,
15 };
16 use crate::{
17     arch::{interrupt::TrapFrame, MMArch},
18     filesystem::{
19         procfs::procfs_register_pid,
20         vfs::{file::FileDescriptorVec, MAX_PATHLEN},
21     },
22     include::bindings::bindings::verify_area,
23     mm::{ucontext::UserStack, MemoryManagementArch, VirtAddr},
24     process::ProcessControlBlock,
25     sched::completion::Completion,
26     syscall::{
27         user_access::{check_and_clone_cstr, check_and_clone_cstr_array, UserBufferWriter},
28         Syscall, SystemError,
29     },
30 };
31 
32 impl Syscall {
33     pub fn fork(frame: &mut TrapFrame) -> Result<usize, SystemError> {
34         let r = ProcessManager::fork(frame, CloneFlags::empty()).map(|pid| pid.into());
35         return r;
36     }
37 
38     pub fn vfork(frame: &mut TrapFrame) -> Result<usize, SystemError> {
39         // 由于Linux vfork需要保证子进程先运行（除非子进程调用execve或者exit），
40         // 而我们目前没有实现这个特性，所以暂时使用fork代替vfork（linux文档表示这样也是也可以的）
41         Self::fork(frame)
42 
43         // 下面是以前的实现，除非我们实现了子进程先运行的特性，否则不要使用，不然会导致父进程数据损坏
44         // ProcessManager::fork(
45         //     frame,
46         //     CloneFlags::CLONE_VM | CloneFlags::CLONE_FS | CloneFlags::CLONE_SIGNAL,
47         // )
48         // .map(|pid| pid.into())
49     }
50 
51     pub fn execve(
52         path: *const u8,
53         argv: *const *const u8,
54         envp: *const *const u8,
55         frame: &mut TrapFrame,
56     ) -> Result<(), SystemError> {
57         // kdebug!(
58         //     "execve path: {:?}, argv: {:?}, envp: {:?}\n",
59         //     path,
60         //     argv,
61         //     envp
62         // );
63         // kdebug!(
64         //     "before execve: strong count: {}",
65         //     Arc::strong_count(&ProcessManager::current_pcb())
66         // );
67 
68         if path.is_null() {
69             return Err(SystemError::EINVAL);
70         }
71 
72         let x = || {
73             let path: String = check_and_clone_cstr(path, Some(MAX_PATHLEN))?;
74             let argv: Vec<String> = check_and_clone_cstr_array(argv)?;
75             let envp: Vec<String> = check_and_clone_cstr_array(envp)?;
76             Ok((path, argv, envp))
77         };
78         let r: Result<(String, Vec<String>, Vec<String>), SystemError> = x();
79         if let Err(e) = r {
80             panic!("Failed to execve: {:?}", e);
81         }
82         let (path, argv, envp) = r.unwrap();
83         ProcessManager::current_pcb()
84             .basic_mut()
85             .set_name(ProcessControlBlock::generate_name(&path, &argv));
86 
87         Self::do_execve(path, argv, envp, frame)?;
88 
89         // 关闭设置了O_CLOEXEC的文件描述符
90         let fd_table = ProcessManager::current_pcb().fd_table();
91         fd_table.write().close_on_exec();
92         // kdebug!(
93         //     "after execve: strong count: {}",
94         //     Arc::strong_count(&ProcessManager::current_pcb())
95         // );
96 
97         return Ok(());
98     }
99 
100     pub fn wait4(
101         pid: i64,
102         wstatus: *mut i32,
103         options: i32,
104         rusage: *mut c_void,
105     ) -> Result<usize, SystemError> {
106         let options = WaitOption::from_bits(options as u32).ok_or(SystemError::EINVAL)?;
107 
108         let wstatus_buf = if wstatus.is_null() {
109             None
110         } else {
111             Some(UserBufferWriter::new(
112                 wstatus,
113                 core::mem::size_of::<i32>(),
114                 true,
115             )?)
116         };
117 
118         let mut tmp_rusage = if rusage.is_null() {
119             None
120         } else {
121             Some(RUsage::default())
122         };
123 
124         let r = kernel_wait4(pid, wstatus_buf, options, tmp_rusage.as_mut())?;
125 
126         if !rusage.is_null() {
127             let mut rusage_buf = UserBufferWriter::new::<RUsage>(
128                 rusage as *mut RUsage,
129                 core::mem::size_of::<RUsage>(),
130                 true,
131             )?;
132             rusage_buf.copy_one_to_user(&tmp_rusage.unwrap(), 0)?;
133         }
134         return Ok(r);
135     }
136 
137     /// # 退出进程
138     ///
139     /// ## 参数
140     ///
141     /// - status: 退出状态
142     pub fn exit(status: usize) -> ! {
143         ProcessManager::exit(status);
144     }
145 
146     /// @brief 获取当前进程的pid
147     pub fn getpid() -> Result<Pid, SystemError> {
148         let current_pcb = ProcessManager::current_pcb();
149         return Ok(current_pcb.tgid());
150     }
151 
152     /// @brief 获取指定进程的pgid
153     ///
154     /// @param pid 指定一个进程号
155     ///
156     /// @return 成功，指定进程的进程组id
157     /// @return 错误，不存在该进程
158     pub fn getpgid(mut pid: Pid) -> Result<Pid, SystemError> {
159         if pid == Pid(0) {
160             let current_pcb = ProcessManager::current_pcb();
161             pid = current_pcb.pid();
162         }
163         let target_proc = ProcessManager::find(pid).ok_or(SystemError::ESRCH)?;
164         return Ok(target_proc.basic().pgid());
165     }
166     /// @brief 获取当前进程的父进程id
167 
168     /// 若为initproc则ppid设置为0
169     pub fn getppid() -> Result<Pid, SystemError> {
170         let current_pcb = ProcessManager::current_pcb();
171         return Ok(current_pcb.basic().ppid());
172     }
173 
174     pub fn clone(
175         current_trapframe: &mut TrapFrame,
176         clone_args: KernelCloneArgs,
177     ) -> Result<usize, SystemError> {
178         let flags = clone_args.flags;
179 
180         let vfork = Arc::new(Completion::new());
181 
182         if flags.contains(CloneFlags::CLONE_PIDFD)
183             && flags.contains(CloneFlags::CLONE_PARENT_SETTID)
184         {
185             return Err(SystemError::EINVAL);
186         }
187 
188         let current_pcb = ProcessManager::current_pcb();
189         let new_kstack = KernelStack::new()?;
190         let name = current_pcb.basic().name().to_string();
191         let pcb = ProcessControlBlock::new(name, new_kstack);
192         // 克隆pcb
193         ProcessManager::copy_process(&current_pcb, &pcb, clone_args, current_trapframe)?;
194         ProcessManager::add_pcb(pcb.clone());
195 
196         // 向procfs注册进程
197         procfs_register_pid(pcb.pid()).unwrap_or_else(|e| {
198             panic!(
199                 "fork: Failed to register pid to procfs, pid: [{:?}]. Error: {:?}",
200                 pcb.pid(),
201                 e
202             )
203         });
204 
205         if flags.contains(CloneFlags::CLONE_VFORK) {
206             pcb.thread.write().vfork_done = Some(vfork.clone());
207         }
208 
209         if pcb.thread.read().set_child_tid.is_some() {
210             let addr = pcb.thread.read().set_child_tid.unwrap();
211             let mut writer =
212                 UserBufferWriter::new(addr.as_ptr::<i32>(), core::mem::size_of::<i32>(), true)?;
213             writer.copy_one_to_user(&(pcb.pid().data() as i32), 0)?;
214         }
215 
216         ProcessManager::wakeup(&pcb).unwrap_or_else(|e| {
217             panic!(
218                 "fork: Failed to wakeup new process, pid: [{:?}]. Error: {:?}",
219                 pcb.pid(),
220                 e
221             )
222         });
223 
224         if flags.contains(CloneFlags::CLONE_VFORK) {
225             // 等待子进程结束或者exec;
226             vfork.wait_for_completion_interruptible()?;
227         }
228 
229         return Ok(pcb.pid().0);
230     }
231 
232     /// 设置线程地址
233     pub fn set_tid_address(ptr: usize) -> Result<usize, SystemError> {
234         if !unsafe { verify_area(ptr as u64, core::mem::size_of::<i32>() as u64) } {
235             return Err(SystemError::EFAULT);
236         }
237 
238         let pcb = ProcessManager::current_pcb();
239         pcb.thread.write().clear_child_tid = Some(VirtAddr::new(ptr));
240         Ok(pcb.pid.0)
241     }
242 
243     pub fn gettid() -> Result<Pid, SystemError> {
244         let pcb = ProcessManager::current_pcb();
245         Ok(pcb.pid)
246     }
247 
248     pub fn getuid() -> Result<usize, SystemError> {
249         // todo: 增加credit功能之后，需要修改
250         return Ok(0);
251     }
252 
253     pub fn getgid() -> Result<usize, SystemError> {
254         // todo: 增加credit功能之后，需要修改
255         return Ok(0);
256     }
257 
258     pub fn geteuid() -> Result<usize, SystemError> {
259         // todo: 增加credit功能之后，需要修改
260         return Ok(0);
261     }
262 
263     pub fn getegid() -> Result<usize, SystemError> {
264         // todo: 增加credit功能之后，需要修改
265         return Ok(0);
266     }
267 
268     pub fn get_rusage(who: i32, rusage: *mut RUsage) -> Result<usize, SystemError> {
269         let who = RUsageWho::try_from(who)?;
270         let mut writer = UserBufferWriter::new(rusage, core::mem::size_of::<RUsage>(), true)?;
271         let pcb = ProcessManager::current_pcb();
272         let rusage = pcb.get_rusage(who).ok_or(SystemError::EINVAL)?;
273 
274         let ubuf = writer.buffer::<RUsage>(0).unwrap();
275         ubuf.copy_from_slice(&[rusage]);
276 
277         return Ok(0);
278     }
279 
280     /// # 设置资源限制
281     ///
282     /// TODO: 目前暂时不支持设置资源限制，只提供读取默认值的功能
283     ///
284     /// ## 参数
285     ///
286     /// - pid: 进程号
287     /// - resource: 资源类型
288     /// - new_limit: 新的资源限制
289     /// - old_limit: 旧的资源限制
290     ///
291     /// ## 返回值
292     ///
293     /// - 成功，0
294     /// - 如果old_limit不为NULL，则返回旧的资源限制到old_limit
295     ///
296     pub fn prlimit64(
297         _pid: Pid,
298         resource: usize,
299         _new_limit: *const RLimit64,
300         old_limit: *mut RLimit64,
301     ) -> Result<usize, SystemError> {
302         let resource = RLimitID::try_from(resource)?;
303         let mut writer = None;
304 
305         if !old_limit.is_null() {
306             writer = Some(UserBufferWriter::new(
307                 old_limit,
308                 core::mem::size_of::<RLimit64>(),
309                 true,
310             )?);
311         }
312 
313         match resource {
314             RLimitID::Stack => {
315                 if let Some(mut writer) = writer {
316                     let mut rlimit = writer.buffer::<RLimit64>(0).unwrap()[0];
317                     rlimit.rlim_cur = UserStack::DEFAULT_USER_STACK_SIZE as u64;
318                     rlimit.rlim_max = UserStack::DEFAULT_USER_STACK_SIZE as u64;
319                 }
320                 return Ok(0);
321             }
322 
323             RLimitID::Nofile => {
324                 if let Some(mut writer) = writer {
325                     let mut rlimit = writer.buffer::<RLimit64>(0).unwrap()[0];
326                     rlimit.rlim_cur = FileDescriptorVec::PROCESS_MAX_FD as u64;
327                     rlimit.rlim_max = FileDescriptorVec::PROCESS_MAX_FD as u64;
328                 }
329                 return Ok(0);
330             }
331 
332             RLimitID::As | RLimitID::Rss => {
333                 if let Some(mut writer) = writer {
334                     let mut rlimit = writer.buffer::<RLimit64>(0).unwrap()[0];
335                     rlimit.rlim_cur = MMArch::USER_END_VADDR.data() as u64;
336                     rlimit.rlim_max = MMArch::USER_END_VADDR.data() as u64;
337                 }
338                 return Ok(0);
339             }
340 
341             _ => {
342                 return Err(SystemError::ENOSYS);
343             }
344         }
345     }
346 }
347