fix BEGIN / END probes

BEGIN/END probes were broken after https://github.com/iovisor/bpftrace/pull/100.

fix BEGIN / END probes
BEGIN/END probes were broken after https://github.com/iovisor/bpftrace/pull/100.
c5bcf861 · Matheus Marchini · 2562ff60 · c5bcf861
Commit c5bcf861 authored Sep 23, 2018 by Matheus Marchini
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;
    }