WebCore:
[WebKit-https.git] / WebKitTools / Scripts / parse-malloc-history
1 #!/usr/bin/perl
2
3 # Parses the callstacks in a file with malloc_history formatted content, sorting
4 # based on total number of bytes allocated, and filtering based on command-line
5 # parameters.
6
7 use Getopt::Long;
8 use File::Basename;
9
10 use strict;
11 use warnings;
12
13 sub commify($);
14
15 sub main()
16 {
17     my $usage =
18         "Usage: " . basename($0) . " [options] malloc_history.txt\n" .
19         "  --callstack-regexp   Include only call stacks that match this regular expression.\n" .
20         "  --byte-minimum       Include only call stacks with allocation sizes >= this value.\n" .
21         "  --merge-depth        Merge all call stacks that match at this stack depth and above.\n";
22
23     my $mergeDepth = -1;
24     my $callstackRegexp = "";
25     my $byteMinimum = 0;
26     my $getOptionsResult = GetOptions(
27         "callstack-regexp:s" => \$callstackRegexp,
28         "byte-minimum:i" => \$byteMinimum,
29         "merge-depth:i" => \$mergeDepth
30     );
31     my $fileName = $ARGV[0];
32     die $usage if (!$getOptionsResult || !$fileName);
33
34     open FILE, "<$fileName" or die "bad file: $fileName";
35     my @file = <FILE>;
36     close FILE;
37
38     my %callstacks = ();
39     my $byteCountTotal = 0;
40
41     for my $line (@file) {
42         my ($callCount, $byteCount) = ($line =~ /(\d+) calls for (\d+) bytes/);
43         if ($callCount && $byteCount) {
44             $byteCountTotal += $byteCount;
45
46             next if !($line =~ $callstackRegexp);
47
48             my $callstackBegin;
49             if ($mergeDepth == -1) { # start at beginning of callstack
50                 $callstackBegin = index($line, "|");
51             } else { # count stack frames backwards from end of callstack
52                 $callstackBegin = length($line);
53                 for (my $pipeCount = 0; $pipeCount < $mergeDepth; $pipeCount++) {
54                     my $rindexResult = rindex($line, "|", $callstackBegin - 1);
55                     last if $rindexResult == -1;
56                     $callstackBegin = $rindexResult;
57                 }
58             }
59
60             my $callstack = substr($line, $callstackBegin + 2); # + 2 skips "| "
61             if (!$callstacks{$callstack}) {
62                 $callstacks{$callstack} = {"callCount" => 0, "byteCount" => 0};
63             }
64
65             $callstacks{$callstack}{"callCount"} += $callCount;
66             $callstacks{$callstack}{"byteCount"} += $byteCount;
67         }
68     }
69
70     my $byteCountTotalReported = 0;
71     for my $callstack (sort { $callstacks{$b}{"byteCount"} <=> $callstacks{$a}{"byteCount"} } keys %callstacks) {
72         my $callCount = $callstacks{$callstack}{"callCount"};
73         my $byteCount = $callstacks{$callstack}{"byteCount"};
74         last if $byteCount < $byteMinimum;
75
76         $byteCountTotalReported += $byteCount;
77         print commify($callCount) . " calls for " . commify($byteCount) . " bytes: $callstack\n";
78     }
79
80     print "total: " . commify($byteCountTotalReported) . " bytes (" . commify($byteCountTotal - $byteCountTotalReported) . " bytes excluded).\n";
81 }
82
83 exit(main());
84
85 # Copied from perldoc -- please excuse the style
86 sub commify($)
87 {
88     local $_  = shift;
89     1 while s/^([-+]?\d+)(\d{3})/$1,$2/;
90     return $_;
91 }