mirror of https://github.com/aya-rs/aya
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.7 KiB
Rust
108 lines
2.7 KiB
Rust
//! [](https://aya-rs.dev)
|
|
//!
|
|
//! A library to write eBPF programs.
|
|
//!
|
|
//! Aya-bpf is an eBPF library built with a focus on operability and developer experience.
|
|
//! It is the kernel-space counterpart of [Aya](https://docs.rs/aya)
|
|
#![doc(
|
|
html_logo_url = "https://aya-rs.dev/assets/images/crabby.svg",
|
|
html_favicon_url = "https://aya-rs.dev/assets/images/crabby.svg"
|
|
)]
|
|
#![cfg_attr(
|
|
feature = "const_assert",
|
|
allow(incomplete_features),
|
|
feature(generic_const_exprs)
|
|
)]
|
|
#![cfg_attr(unstable, feature(never_type))]
|
|
#![cfg_attr(target_arch = "bpf", feature(asm_experimental_arch))]
|
|
#![allow(clippy::missing_safety_doc)]
|
|
#![deny(warnings)]
|
|
#![warn(clippy::cast_lossless, clippy::cast_sign_loss)]
|
|
#![no_std]
|
|
|
|
pub use aya_ebpf_bindings::bindings;
|
|
|
|
mod args;
|
|
pub use args::PtRegs;
|
|
pub mod helpers;
|
|
pub mod maps;
|
|
pub mod programs;
|
|
|
|
use core::ffi::c_void;
|
|
|
|
pub use aya_ebpf_cty as cty;
|
|
pub use aya_ebpf_macros as macros;
|
|
use cty::{c_int, c_long};
|
|
use helpers::{bpf_get_current_comm, bpf_get_current_pid_tgid, bpf_get_current_uid_gid};
|
|
|
|
pub const TASK_COMM_LEN: usize = 16;
|
|
|
|
pub trait EbpfContext {
|
|
fn as_ptr(&self) -> *mut c_void;
|
|
|
|
#[inline]
|
|
fn command(&self) -> Result<[u8; TASK_COMM_LEN], c_long> {
|
|
bpf_get_current_comm()
|
|
}
|
|
|
|
fn pid(&self) -> u32 {
|
|
bpf_get_current_pid_tgid() as u32
|
|
}
|
|
|
|
fn tgid(&self) -> u32 {
|
|
(bpf_get_current_pid_tgid() >> 32) as u32
|
|
}
|
|
|
|
fn uid(&self) -> u32 {
|
|
bpf_get_current_uid_gid() as u32
|
|
}
|
|
|
|
fn gid(&self) -> u32 {
|
|
(bpf_get_current_uid_gid() >> 32) as u32
|
|
}
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub unsafe extern "C" fn memset(s: *mut u8, c: c_int, n: usize) {
|
|
#[allow(clippy::cast_sign_loss)]
|
|
let b = c as u8;
|
|
for i in 0..n {
|
|
*s.add(i) = b;
|
|
}
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub unsafe extern "C" fn memcpy(dest: *mut u8, src: *mut u8, n: usize) {
|
|
for i in 0..n {
|
|
*dest.add(i) = *src.add(i);
|
|
}
|
|
}
|
|
|
|
/// Check if a value is within a range, using conditional forms compatible with
|
|
/// the verifier.
|
|
#[inline(always)]
|
|
pub fn check_bounds_signed(value: i64, lower: i64, upper: i64) -> bool {
|
|
#[cfg(target_arch = "bpf")]
|
|
unsafe {
|
|
let mut in_bounds = 0u64;
|
|
core::arch::asm!(
|
|
"if {value} s< {lower} goto +2",
|
|
"if {value} s> {upper} goto +1",
|
|
"{i} = 1",
|
|
i = inout(reg) in_bounds,
|
|
lower = in(reg) lower,
|
|
upper = in(reg) upper,
|
|
value = in(reg) value,
|
|
);
|
|
in_bounds == 1
|
|
}
|
|
// We only need this for doc tests which are compiled for the host target
|
|
#[cfg(not(target_arch = "bpf"))]
|
|
{
|
|
let _ = value;
|
|
let _ = lower;
|
|
let _ = upper;
|
|
unimplemented!()
|
|
}
|
|
}
|