Output probe calling sequence comment

* dtrace: Output probe calling sequence comment

1 files changed, 3 insertions, 13 deletions

diff --git a/dtrace b/dtrace
index 114e70a6..a2b495b2 100755
--- a/dtrace
+++ b/dtrace
@@ -17,12 +17,11 @@ from subprocess import call
 from tempfile import mkstemp
 
 class provider:
-    arglist = dict()
     def open(self, provider, header):
         have_provider = False
         self.f = open(provider)
         self.h = open(header,mode='w')
-        self.h.write("// Generated by the Systemtap dtrace wrapper\n")
+        self.h.write("/* Generated by the Systemtap dtrace wrapper */\n")
         self.h.write("\n#include <sys/sdt.h>\n\n")
         in_comment = False
         while (True):
@@ -49,7 +48,6 @@ class provider:
                 new_args = ""
                 i = 0
                 c = 0
-                self.arglist[this_probe] = ""
                 while (i < len(args)):
                     if (args[i:i+1] == ","):
                         new_args = ('%s%s' % (new_args, args[i]))
@@ -57,8 +55,6 @@ class provider:
                     else:
                         new_args = new_args + args[i]
                     i += 1
-                if (len(new_args) > 0):
-                    self.arglist[this_probe] = ('%s arg%d' % (new_args, c))
                 if (len(new_args) == 0):
                     c = 0
                     stap_str = "STAP_PROBE(provider,%s" % (this_probe)
@@ -68,22 +64,16 @@ class provider:
                 define_str = "#define %s(" % (this_probe_canon)
                 i = 1
                 while (i <= c):
-                    if (i != 0):
+                    if (i != 1):
                         define_str += ","
                     define_str = define_str + "arg%s" % (i);
                     stap_str = stap_str + ",arg%s" % (i);
                     i += 1
+                self.h.write ('/* %s (%s) */\n' % (this_probe_canon,new_args))
                 self.h.write ('#define %s_ENABLED() 1\n' % this_probe_canon)
                 self.h.write (define_str + ") \\\n")
                 self.h.write (stap_str + ")\n\n")
 
-    def get(self, arg):
-        print arg
-        if (arg in self.arglist):
-            return self.arglist[arg]
-        else:
-            return ""
-
 def usage ():
     print "Usage " + sys.argv[0] + " [-h | -G] -s File.d -o File {Files}"
     sys.exit(1)