SYS-OSS/FRET-LibAFL
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dragonflybsd build fix for core affinity. (#753)

Browse Source 
supporting most of linux sched api here.
This commit is contained in:
David CARLIER 2022-08-30 02:37:17 +01:00 committed by GitHub
parent ebae4d3ce8
commit ebfe414a27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libafl/src/bolts/core_affinity.rs
View File

@ -193,24 +193,29 @@ pub fn parse_core_bind_arg(args: &str) -> Result<Vec<usize>, Error> {
// Linux Section // Linux Section
#[cfg(any(target_os = "android", target_os = "linux"))] #[cfg(any(target_os = "android", target_os = "linux", target_os = "dragonfly"))]
#[inline] #[inline]
fn get_core_ids_helper() -> Result<Vec<CoreId>, Error> { fn get_core_ids_helper() -> Result<Vec<CoreId>, Error> {
linux::get_core_ids() linux::get_core_ids()
} }
#[cfg(any(target_os = "android", target_os = "linux"))] #[cfg(any(target_os = "android", target_os = "linux", target_os = "dragonfly"))]
#[inline] #[inline]
fn set_for_current_helper(core_id: CoreId) -> Result<(), Error> { fn set_for_current_helper(core_id: CoreId) -> Result<(), Error> {
linux::set_for_current(core_id) linux::set_for_current(core_id)
} }
#[cfg(any(target_os = "android", target_os = "linux"))] #[cfg(any(target_os = "android", target_os = "linux", target_os = "dragonfly"))]
mod linux { mod linux {
use alloc::{string::ToString, vec::Vec}; use alloc::{string::ToString, vec::Vec};
use std::mem; use std::mem;
#[cfg(target_os = "dragonfly")]
use libc::{cpu_set_t, sched_getaffinity, sched_setaffinity, CPU_ISSET, CPU_SET};
#[cfg(not(target_os = "dragonfly"))]
use libc::{cpu_set_t, sched_getaffinity, sched_setaffinity, CPU_ISSET, CPU_SET, CPU_SETSIZE}; use libc::{cpu_set_t, sched_getaffinity, sched_setaffinity, CPU_ISSET, CPU_SET, CPU_SETSIZE};
#[cfg(target_os = "dragonfly")]
const CPU_SETSIZE: libc::c_int = 256;
use super::CoreId; use super::CoreId;
use crate::Error; use crate::Error;