general/occlum
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix affinity array size incorrect issue

Browse Source
This commit is contained in:
zongmin.gu 2021-03-01 10:44:05 +08:00 committed by Zongmin.Gu
parent 7deeccb03b
commit 7122529b6b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/libos/src/sched/syscalls.rs
View File

@ -10,7 +10,7 @@ pub fn do_sched_yield() -> Result<isize> {
pub fn do_sched_getaffinity(pid: pid_t, buf_size: size_t, buf_ptr: *mut u8) -> Result<isize> { pub fn do_sched_getaffinity(pid: pid_t, buf_size: size_t, buf_ptr: *mut u8) -> Result<isize> {
// Construct safe Rust types // Construct safe Rust types
let buf_size = { let buf_size = {
if buf_size * 8 < AVAIL_CPUSET.cpu_count() { if buf_size < CpuSet::len() {
return_errno!(EINVAL, "buf size is not big enough"); return_errno!(EINVAL, "buf size is not big enough");
} }
@ -39,7 +39,7 @@ pub fn do_sched_getaffinity(pid: pid_t, buf_size: size_t, buf_ptr: *mut u8) -> R
pub fn do_sched_setaffinity(pid: pid_t, buf_size: size_t, buf_ptr: *const u8) -> Result<isize> { pub fn do_sched_setaffinity(pid: pid_t, buf_size: size_t, buf_ptr: *const u8) -> Result<isize> {
// Convert unsafe C types into safe Rust types // Convert unsafe C types into safe Rust types
let buf_size = { let buf_size = {
if buf_size * 8 < AVAIL_CPUSET.cpu_count() { if buf_size < CpuSet::len() {
return_errno!(EINVAL, "buf size is not big enough"); return_errno!(EINVAL, "buf size is not big enough");
} }
CpuSet::len() CpuSet::len()