4
1
mirror of https://github.com/DragonOS-Community/DragonOS.git synced 2025-06-20 10:06:31 +00:00

feat(debug)[WIP]: add static-keys support (#1025)

* feat: add static-keys support
This commit is contained in:
linfeng
2024-11-16 16:21:20 +08:00
committed by GitHub
parent e232830c18
commit 750b3b5d91
5 changed files with 36 additions and 1 deletions

@ -0,0 +1,28 @@
#[cfg(feature = "static_keys_test")]
mod tests {
use static_keys::{define_static_key_false, static_branch_unlikely};
define_static_key_false!(MY_STATIC_KEY);
#[inline(always)]
fn foo() {
println!("Entering foo function");
if static_branch_unlikely!(MY_STATIC_KEY) {
println!("A branch");
} else {
println!("B branch");
}
}
pub(super) fn static_keys_test() {
foo();
unsafe {
MY_STATIC_KEY.enable();
}
foo();
}
}
pub fn static_keys_init() {
static_keys::global_init();
#[cfg(feature = "static_keys_test")]
tests::static_keys_test();
}

@ -1,2 +1,3 @@
pub mod jump_label;
pub mod klog;
pub mod kprobe;

@ -57,6 +57,7 @@ fn do_start_kernel() {
unsafe { mm_init() };
// crate::debug::jump_label::static_keys_init();
if scm_reinit().is_ok() {
if let Err(e) = textui_init() {
warn!("Failed to init textui: {:?}", e);
@ -90,6 +91,7 @@ fn do_start_kernel() {
clocksource_boot_finish();
Futex::init();
crate::bpf::init_bpf_system();
crate::debug::jump_label::static_keys_init();
#[cfg(all(target_arch = "x86_64", feature = "kvm"))]
crate::virt::kvm::kvm_init();
}

@ -17,6 +17,7 @@
#![feature(sync_unsafe_cell)]
#![feature(vec_into_raw_parts)]
#![feature(c_variadic)]
#![feature(asm_goto)]
#![cfg_attr(target_os = "none", no_std)]
#![allow(static_mut_refs, non_local_definitions, internal_features)]
// clippy的配置