mirror of
https://github.com/DragonOS-Community/DragonOS.git
synced 2025-06-19 17:26:31 +00:00
signal的发送(暂时父子进程之间共享信号及相应的结构体) (#89)
* 解决由于spinlock.h中包含preempt_enable()带来的循环include问题 * new: 初步实现signal的数据结构 * new:signal相关数据结构 * fix: 解决bindings.rs报一堆警告的问题 * new: rust下的kdebug kinfo kwarn kBUG kerror宏 * 移动asm.h和cmpxchg.h * new: signal的发送(暂时只支持父子进程共享信号及处理函数)
This commit is contained in:
103
kernel/src/process/process.rs
Normal file
103
kernel/src/process/process.rs
Normal file
@ -0,0 +1,103 @@
|
||||
use core::ptr::{read_volatile, write_volatile};
|
||||
|
||||
use crate::{
|
||||
arch::x86_64::asm::current::current_pcb,
|
||||
include::bindings::bindings::{
|
||||
process_control_block, sched_enqueue, PROC_RUNNING, PROC_STOPPED,
|
||||
},
|
||||
sched::core::cpu_executing,
|
||||
smp::core::{smp_get_processor_id, smp_send_reschedule},
|
||||
};
|
||||
|
||||
use super::preempt::{preempt_disable, preempt_enable};
|
||||
|
||||
/// 判断进程是否已经停止
|
||||
#[no_mangle]
|
||||
pub extern "C" fn process_is_stopped(pcb: *const process_control_block) -> bool {
|
||||
let state: u64 = unsafe { read_volatile(&(*pcb).state) } as u64;
|
||||
if (state & (PROC_STOPPED as u64)) != 0 {
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/// @brief 尝试唤醒指定的进程。
|
||||
/// 本函数的行为:If (@_state & @pcb->state) @pcb->state = TASK_RUNNING.
|
||||
///
|
||||
/// @param _pcb 要被唤醒的进程的pcb
|
||||
/// @param _state 如果pcb的state与_state匹配,则唤醒这个进程
|
||||
/// @param _wake_flags 保留,暂未使用,请置为0
|
||||
/// @return true: 成功唤醒
|
||||
/// false: 不符合唤醒条件,无法唤醒
|
||||
#[no_mangle]
|
||||
pub extern "C" fn process_try_to_wake_up(
|
||||
_pcb: *mut process_control_block,
|
||||
_state: u64,
|
||||
_wake_flags: i32,
|
||||
) -> bool {
|
||||
preempt_disable();
|
||||
|
||||
let mut retval = false;
|
||||
// 获取对pcb的可变引用
|
||||
let pcb = unsafe { _pcb.as_mut() }.unwrap();
|
||||
|
||||
// 如果要唤醒的就是当前的进程
|
||||
if current_pcb() as *mut process_control_block as usize == _pcb as usize {
|
||||
unsafe {
|
||||
write_volatile(&mut pcb.state, PROC_RUNNING as u64);
|
||||
}
|
||||
preempt_enable();
|
||||
retval = true;
|
||||
return retval;
|
||||
}
|
||||
// todo: 将来调度器引入ttwu队列之后,需要修改这里的判断条件
|
||||
|
||||
// todo: 为pcb引入pi_lock,然后在这里加锁
|
||||
if unsafe { read_volatile(&pcb.state) } & _state != 0 {
|
||||
// 可以wakeup
|
||||
unsafe {
|
||||
write_volatile(&mut pcb.state, PROC_RUNNING as u64);
|
||||
sched_enqueue(pcb);
|
||||
}
|
||||
retval = true;
|
||||
}
|
||||
// todo: 对pcb的pi_lock放锁
|
||||
preempt_enable();
|
||||
return retval;
|
||||
}
|
||||
|
||||
/// @brief 当进程,满足 (@state & @pcb->state)时,唤醒进程,并设置: @pcb->state = TASK_RUNNING.
|
||||
///
|
||||
/// @return true 唤醒成功
|
||||
/// @return false 唤醒失败
|
||||
#[no_mangle]
|
||||
pub extern "C" fn process_wake_up_state(pcb: *mut process_control_block, state: u64) -> bool {
|
||||
return process_try_to_wake_up(pcb, state, 0);
|
||||
}
|
||||
|
||||
|
||||
/// @brief 让一个正在cpu上运行的进程陷入内核
|
||||
pub fn process_kick(pcb: *mut process_control_block) {
|
||||
preempt_disable();
|
||||
let cpu = process_cpu(pcb);
|
||||
// 如果给定的进程正在别的核心上执行,则立即发送请求,让它陷入内核态,以及时响应信号。
|
||||
if cpu != smp_get_processor_id() && process_is_executing(pcb) {
|
||||
smp_send_reschedule(cpu);
|
||||
}
|
||||
preempt_enable();
|
||||
}
|
||||
|
||||
/// @brief 获取给定的进程在哪个cpu核心上运行(使用volatile避免编译器优化)
|
||||
#[inline]
|
||||
pub fn process_cpu(pcb: *const process_control_block) -> u32 {
|
||||
unsafe { read_volatile(&(*pcb).cpu_id) }
|
||||
}
|
||||
|
||||
/// @brief 判断给定的进程是否正在处理器上执行
|
||||
///
|
||||
/// @param pcb 进程的pcb
|
||||
#[inline]
|
||||
pub fn process_is_executing(pcb: *const process_control_block) -> bool {
|
||||
return cpu_executing(process_cpu(pcb)) == pcb;
|
||||
}
|
Reference in New Issue
Block a user