Emit vma callbacks for itrace.

* tapsets.cxx (itrace_derived_probe_group::emit_module_decls): Emit vma
  callbacks.
  (itrace_derived_probe_group::emit_module_init): Activate vma callbacks.

1 files changed, 25 insertions, 0 deletions

diff --git a/tapsets.cxx b/tapsets.cxx
index a668f016..8118c838 100644
--- a/tapsets.cxx
+++ b/tapsets.cxx
@@ -6453,6 +6453,23 @@ itrace_derived_probe_group::emit_module_decls (systemtap_session& s)
   s.op->newline(-1) << "return rc;";
   s.op->newline(-1) << "}";
 
+  // Emit vma callbacks.
+  s.op->newline() << "static struct stap_task_finder_target stap_itrace_vmcbs[] = {";
+  s.op->indent(1);
+  if (! probes_by_path.empty())
+    {
+      for (p_b_path_iterator it = probes_by_path.begin();
+           it != probes_by_path.end(); it++)
+        emit_vma_callback_probe_decl (s, it->first, (int64_t)0);
+    }
+  if (! probes_by_pid.empty())
+    {
+      for (p_b_pid_iterator it = probes_by_pid.begin();
+           it != probes_by_pid.end(); it++)
+        emit_vma_callback_probe_decl (s, "", it->first);
+    }
+  s.op->newline(-1) << "};";
+
   s.op->newline() << "static struct stap_itrace_probe stap_itrace_probes[] = {";
   s.op->indent(1);
 
@@ -6494,6 +6511,14 @@ itrace_derived_probe_group::emit_module_init (systemtap_session& s)
     return;
 
   s.op->newline();
+  s.op->newline() << "/* ---- itrace vma callbacks ---- */";
+  s.op->newline() << "for (i=0; i<ARRAY_SIZE(stap_itrace_vmcbs); i++) {";
+  s.op->indent(1);
+  s.op->newline() << "struct stap_task_finder_target *r = &stap_itrace_vmcbs[i];";
+  s.op->newline() << "rc = stap_register_task_finder_target(r);";
+  s.op->newline(-1) << "}";
+
+  s.op->newline();
   s.op->newline() << "/* ---- itrace probes ---- */";
 
   s.op->newline() << "for (i=0; i<" << num_probes << "; i++) {";