Update stackcollapse-bpftrace to work with book examples

sravotto · Aug 21, 2021 · ecc931c · ecc931c
1 parent a258e78
commit ecc931c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/stackcollapse-bpftrace.pl b/stackcollapse-bpftrace.pl
@@ -51,15 +51,21 @@
 foreach (<>) {
   chomp;
   if (!$in_stack) {
-    if (/^@\[/) {
+    if (/^@\[$/) {
       $in_stack = 1;
+    } elsif (/^@\[,\s(.*)\]: (\d+)/) {
+      print $1 . " $2\n";
     }
   } else {
-    if (m/^\]: (\d+)/) {
-      print join(';', reverse(@stack)) . " $1\n";
+    if (m/^,?\s?(.*)\]: (\d+)/) {
+      if (length $1) {
+        push(@stack, $1);
+      }
+      print join(';', reverse(@stack)) . " $2\n";
       $in_stack = 0;
       @stack = ();
     } else {
+      $_ =~ s/^\s+//;
       push(@stack, $_);
     }
   }