Skip to content

Commit

Permalink
aya: update VerifierLogLevel to use bitflags
Browse files Browse the repository at this point in the history
  • Loading branch information
conectado committed Sep 1, 2022
1 parent 906c25f commit 7b14319
Showing 1 changed file with 22 additions and 18 deletions.
40 changes: 22 additions & 18 deletions aya/src/bpf.rs
Original file line number Diff line number Diff line change
Expand Up @@ -196,22 +196,26 @@ pub struct BpfLoader<'a> {
verifier_log_level: VerifierLogLevel,
}

/// Used to set the verifier log level in [BpfLoader](BpfLoader::verifier_log_level()).
#[repr(u32)]
#[non_exhaustive]
#[derive(Debug, Clone, Copy)]
pub enum VerifierLogLevel {
/// Disable all logging.
Disable = 0,

/// Default level of logging, shows verifier stats.
Default = 4,

/// Prints verbose logs showing tracing.
Verbose = 1,
bitflags! {
/// Used to set the verifier log level in [BpfLoader](BpfLoader::verifier_log_level()).
pub struct VerifierLogLevel: u32 {
/// Used to disable all logging.
const DISABLE = 0;
/// Logs tracing with details level 1
const LEVEL1 = 1;
/// Log tracing with details level 2
const LEVEL2 = 2 | Self::LEVEL1.bits;
/// Logs eBPF verifier stats
const STATS = 4;
}
}

/// Prints full debug details.
Debug = 7,
impl Default for VerifierLogLevel {
fn default() -> Self {
Self {
bits: Self::LEVEL1.bits | Self::STATS.bits,
}
}
}

impl<'a> BpfLoader<'a> {
Expand All @@ -225,7 +229,7 @@ impl<'a> BpfLoader<'a> {
globals: HashMap::new(),
features,
extensions: HashSet::new(),
verifier_log_level: VerifierLogLevel::Default,
verifier_log_level: VerifierLogLevel::default(),
}
}

Expand Down Expand Up @@ -341,7 +345,7 @@ impl<'a> BpfLoader<'a> {
/// use aya::{BpfLoader, VerifierLogLevel};
///
/// let bpf = BpfLoader::new()
/// .verifier_log_level(VerifierLogLevel::Verbose)
/// .verifier_log_level(VerifierLogLevel::LEVEL2 | VerifierLogLevel::STATS)
/// .load_file("file.o")?;
/// # Ok::<(), aya::BpfError>(())
/// ```
Expand Down Expand Up @@ -382,7 +386,7 @@ impl<'a> BpfLoader<'a> {
/// # Ok::<(), aya::BpfError>(())
/// ```
pub fn load(&mut self, data: &[u8]) -> Result<Bpf, BpfError> {
let verifier_log_level = self.verifier_log_level as u32;
let verifier_log_level = self.verifier_log_level.bits;
let mut obj = Object::parse(data)?;
obj.patch_map_data(self.globals.clone())?;

Expand Down

0 comments on commit 7b14319

Please sign in to comment.