Merge pull request #112 from iovisor/fix-begin-end

fix BEGIN / END probes

Merge pull request #112 from iovisor/fix-begin-end
fix BEGIN / END probes
56eb2e63 · Brendan Gregg · GitHub · 0b72a90d · c5bcf861 · 56eb2e63
Commit 56eb2e63 authored Sep 23, 2018 by Brendan Gregg Committed by GitHub Sep 23, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/bpftrace.cpp src/bpftrace.cpp +4 -0

No files found.
--- a/src/bpftrace.cpp
+++ b/src/bpftrace.cpp
@@ -33,6 +33,8 @@ int BPFtrace::add_probe(ast::Probe &p)
      probe.orig_name = p.name();
      probe.name = p.name();
      probe.loc = 0;
+      probe.index = attach_point->index(probe.name) > 0 ?
+          attach_point->index(probe.name) : p.index();
      special_probes_.push_back(probe);
      continue;
    }
@@ -45,6 +47,8 @@ int BPFtrace::add_probe(ast::Probe &p)
      probe.orig_name = p.name();
      probe.name = p.name();
      probe.loc = 0;
+      probe.index = attach_point->index(probe.name) > 0 ?
+          attach_point->index(probe.name) : p.index();
      special_probes_.push_back(probe);
      continue;
    }