Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-22 08:53:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement system call sched_get/set_affinity

Browse Source
This commit is contained in:
Zhang Junyang
2024-09-23 14:38:17 +08:00
committed by Tate, Hongliang Tian
parent 3468ec213b
commit e319641b4d
8 changed files with 167 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
kernel/src/thread/mod.rs
View File

@ -4,7 +4,10 @@
use core::sync::atomic::Ordering;
use ostd::{cpu::CpuSet, sync::PreemptDisabled, task::Task};
use ostd::{
cpu::{AtomicCpuSet, CpuSet},
task::Task,
};
use self::status::{AtomicThreadStatus, ThreadStatus};
use crate::{
@ -34,8 +37,8 @@ pub struct Thread {
status: AtomicThreadStatus,
/// Thread priority
priority: AtomicPriority,
/// Thread cpu affinity
cpu_affinity: SpinLock<CpuSet>,
/// Thread CPU affinity
cpu_affinity: AtomicCpuSet,
}
impl Thread {
@ -52,7 +55,7 @@ impl Thread {
data: Box::new(data),
status: AtomicThreadStatus::new(status),
priority: AtomicPriority::new(priority),
cpu_affinity: SpinLock::new(cpu_affinity),
cpu_affinity: AtomicCpuSet::new(cpu_affinity),
}
}
@ -67,6 +70,11 @@ impl Thread {
.cloned()
}
/// Returns the task associated with this thread.
pub fn task(&self) -> Arc<Task> {
self.task.upgrade().unwrap()
}
/// Gets the Thread from task's data.
///
/// # Panics
@ -116,14 +124,9 @@ impl Thread {
self.priority.store(new_priority, Ordering::Relaxed)
}
/// Acquires the lock of cpu affinity.
pub fn lock_cpu_affinity(&self) -> SpinLockGuard<CpuSet, PreemptDisabled> {
self.cpu_affinity.lock()
}
/// Updates the cpu affinity with the new value.
pub fn set_cpu_affinity(&self, new_cpu_affinity: CpuSet) {
*self.cpu_affinity.lock() = new_cpu_affinity;
/// Returns the reference to the atomic CPU affinity.
pub fn atomic_cpu_affinity(&self) -> &AtomicCpuSet {
&self.cpu_affinity
}
pub fn yield_now() {