diff --git a/{{project-name}}/Cargo.toml b/{{project-name}}/Cargo.toml index 56e094d..a869e6e 100644 --- a/{{project-name}}/Cargo.toml +++ b/{{project-name}}/Cargo.toml @@ -13,9 +13,9 @@ clap = { version = "4.1", features = ["derive"] } {{project-name}}-common = { path = "../{{project-name}}-common", features = ["user"] } anyhow = "1" env_logger = "0.10" +libc = "0.2" log = "0.4" tokio = { version = "1.25", features = ["macros", "rt", "rt-multi-thread", "net", "signal"] } -libc = "0.2" [[bin]] name = "{{project-name}}" diff --git a/{{project-name}}/src/main.rs b/{{project-name}}/src/main.rs index 181927b..103b6eb 100644 --- a/{{project-name}}/src/main.rs +++ b/{{project-name}}/src/main.rs @@ -71,13 +71,14 @@ async fn main() -> Result<(), anyhow::Error> { {% endif %} env_logger::init(); - let rlim = libc::rlimit{ + // Allow current process to lock memory for eBPF resources. + let rlim = libc::rlimit { rlim_cur: libc::RLIM_INFINITY, rlim_max: libc::RLIM_INFINITY, }; let ret = unsafe { libc::setrlimit(libc::RLIMIT_MEMLOCK, &rlim) }; - if !ret == 0{ - warn!("cannot remove mem lock"); + if ret != 0 { + warn!("remove limit on locked memory failed, ret is: {}", ret); } // This will include your eBPF object file as raw bytes at compile-time and load it at