Merge pull request #75 from dmitris/fix-uprobe

Fix uprobe programs and VSCode spacing

{{project-name}}-ebpf/src/main.rs

@@ -2,10 +2,7 @@
 #![no_main]
 {% case program_type -%}
 {%- when "kprobe" %}
-use aya_bpf::{
-    macros::kprobe,
-    programs::ProbeContext,
-};
+use aya_bpf::{macros::kprobe, programs::ProbeContext};
 use aya_log_ebpf::info;
 
 #[kprobe(name = "{{crate_name}}")]
@@ -21,10 +18,7 @@ fn try_{{crate_name}}(ctx: ProbeContext) -> Result<u32, u32> {
     Ok(0)
 }
 {%- when "kretprobe" %}
-use aya_bpf::{
-    macros::kretprobe,
-    programs::ProbeContext,
-};
+use aya_bpf::{macros::kretprobe, programs::ProbeContext};
 use aya_log_ebpf::info;
 
 #[kretprobe(name = "{{crate_name}}")]
@@ -160,11 +154,7 @@ fn try_{{crate_name}}(ctx: SkMsgContext) -> Result<u32, u32> {
     Ok(0)
 }
 {%- when "xdp" %}
-use aya_bpf::{
-    bindings::xdp_action,
-    macros::xdp,
-    programs::XdpContext,
-};
+use aya_bpf::{bindings::xdp_action, macros::xdp, programs::XdpContext};
 use aya_log_ebpf::info;
 
 #[xdp(name = "{{crate_name}}")]
@@ -180,10 +170,7 @@ fn try_{{crate_name}}(ctx: XdpContext) -> Result<u32, u32> {
     Ok(xdp_action::XDP_PASS)
 }
 {%- when "classifier" %}
-use aya_bpf::{
-    macros::classifier,
-    programs::TcContext,
-};
+use aya_bpf::{macros::classifier, programs::TcContext};
 use aya_log_ebpf::info;
 
 #[classifier(name = "{{crate_name}}")]
@@ -304,10 +291,7 @@ fn try_{{crate_name}}(ctx: SysctlContext) -> Result<i32, i32> {
     Ok(0)
 }
 {%- when "cgroup_sockopt" %}
-use aya_bpf::{
-    macros::cgroup_sockopt,
-    programs::SockoptContext,
-};
+use aya_bpf::{macros::cgroup_sockopt, programs::SockoptContext};
 use aya_log_ebpf::info;
 
 #[cgroup_sockopt({{sockopt_target}}, name = "{{crate_name}}")]
@@ -323,10 +307,7 @@ fn try_{{crate_name}}(ctx: SockoptContext) -> Result<i32, i32> {
     Ok(0)
 }
 {%- when "raw_tracepoint" %}
-use aya_bpf::{
-    macros::raw_tracepoint,
-    programs::RawTracePointContext,
-};
+use aya_bpf::{macros::raw_tracepoint, programs::RawTracePointContext};
 use aya_log_ebpf::info;
 
 #[raw_tracepoint(name = "{{crate_name}}")]
@@ -343,10 +324,7 @@ fn try_{{crate_name}}(ctx: RawTracePointContext) -> Result<i32, i32> {
 }
 {%- when "perf_event" %}
 use aya_bpf::{
-    helpers::bpf_get_smp_processor_id,
-    macros::perf_event,
-    programs::PerfEventContext,
-    BpfContext,
+    helpers::bpf_get_smp_processor_id, macros::perf_event, programs::PerfEventContext, BpfContext,
 };
 use aya_log_ebpf::info;
 
@@ -361,8 +339,14 @@ pub fn {{crate_name}}(ctx: PerfEventContext) -> u32 {
 fn try_{{crate_name}}(ctx: PerfEventContext) -> Result<u32, u32> {
     let cpu = unsafe { bpf_get_smp_processor_id() };
     match ctx.pid() {
-        0 => info!(&ctx, "perf_event '{{crate_name}}' triggered on CPU {}, running a kernel task", cpu),
-        pid => info!(&ctx, "perf_event '{{crate_name}}' triggered on CPU {}, running PID {}", cpu, pid),
+        0 => info!(
+            &ctx,
+            "perf_event 'perftest' triggered on CPU {}, running a kernel task", cpu
+        ),
+        pid => info!(
+            &ctx,
+            "perf_event 'perftest' triggered on CPU {}, running PID {}", cpu, pid
+        ),
     }
 
     Ok(0)

{{project-name}}/Cargo.toml

@@ -6,10 +6,10 @@ publish = false
 
 [dependencies]
 aya = { version = ">=0.11", features=["async_tokio"] }
+aya-log = "0.1"
 {% if program_types_with_opts contains program_type -%}
 clap = { version = "4.1", features = ["derive"] }
 {% endif -%}
-aya-log = "0.1"
 {{project-name}}-common = { path = "../{{project-name}}-common", features = ["user"] }
 anyhow = "1.0.68"
 env_logger = "0.10"

{{project-name}}/src/main.rs

@@ -59,7 +59,7 @@ struct Opt {
     cgroup_path: String,
 {% elsif program_type == "uprobe" or program_type == "uretprobe" %}
     #[clap(short, long)]
-    pid: Option<i32>
+    pid: Option<i32>,
 {% endif -%}
 }
 
@@ -105,7 +105,7 @@ async fn main() -> Result<(), anyhow::Error> {
     {%- when "uprobe", "uretprobe" -%}
     let program: &mut UProbe = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;
     program.load()?;
-    program.attach(Some("{{uprobe_fn_name}}"), 0, "{{uprobe_target}}", opt.pid.try_into()?)?;
+    program.attach(Some("{{uprobe_fn_name}}"), 0, "{{uprobe_target}}", opt.pid)?;
     {%- when "sock_ops" -%}
     let program: &mut SockOps = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;
     let cgroup = std::fs::File::open(opt.cgroup_path)?;