checkstack.pl: fix arch autodetection
[oweals/busybox.git] / scripts / checkstack.pl
1 #!/usr/bin/perl
2
3 # Stolen from Linux kernel :)
4
5 #       Check the stack usage of functions
6 #
7 #       Copyright Joern Engel <joern@wh.fh-wedel.de>
8 #       Inspired by Linus Torvalds
9 #       Original idea maybe from Keith Owens
10 #       s390 port and big speedup by Arnd Bergmann <arnd@bergmann-dalldorf.de>
11 #       Mips port by Juan Quintela <quintela@mandrakesoft.com>
12 #       IA64 port via Andreas Dilger
13 #       Arm port by Holger Schurig
14 #       sh64 port by Paul Mundt
15 #       Random bits by Matt Mackall <mpm@selenic.com>
16 #       M68k port by Geert Uytterhoeven and Andreas Schwab
17 #
18 #       Usage:
19 #       objdump -d vmlinux | checkstack.pl [arch]
20 #
21 #       TODO :  Port to all architectures (one regex per arch)
22
23 # check for arch
24 #
25 # $re is used for two matches:
26 # $& (whole re) matches the complete objdump line with the stack growth
27 # $1 (first bracket) matches the size of the stack growth
28 #
29 # use anything else and feel the pain ;)
30 my (@stack, $re, $x, $xs);
31 {
32         my $arch = shift;
33         if ($arch eq "") {
34                 $arch = `uname -m`;
35                 1 while chomp $arch;
36         }
37
38         $x      = "[0-9a-f]";   # hex character
39         $xs     = "[0-9a-f ]";  # hex character or space
40         if ($arch eq 'arm') {
41                 #c0008ffc:      e24dd064        sub     sp, sp, #100    ; 0x64
42                 $re = qr/.*sub.*sp, sp, #(([0-9]{2}|[3-9])[0-9]{2})/o;
43         } elsif ($arch eq 'blackfin') {
44                 #      52:       00 e8 03 00     LINK 0xc;
45                 $re = qr/.*LINK (0x$x{1,5});$/o;
46         } elsif ($arch =~ /^i[3456]86$/) {
47                 #c0105234:       81 ec ac 05 00 00       sub    $0x5ac,%esp
48                 $re = qr/^.*[as][du][db]    \$(0x$x{1,8}),\%esp$/o;
49         } elsif ($arch eq 'x86_64') {
50                 #    2f60:      48 81 ec e8 05 00 00    sub    $0x5e8,%rsp
51                 $re = qr/^.*[as][du][db]    \$(0x$x{1,8}),\%rsp$/o;
52         } elsif ($arch eq 'ia64') {
53                 #e0000000044011fc:       01 0f fc 8c     adds r12=-384,r12
54                 $re = qr/.*adds.*r12=-(([0-9]{2}|[3-9])[0-9]{2}),r12/o;
55         } elsif ($arch eq 'm68k') {
56                 #    2b6c:       4e56 fb70       linkw %fp,#-1168
57                 #  1df770:       defc ffe4       addaw #-28,%sp
58                 $re = qr/.*(?:linkw %fp,|addaw )#-([0-9]{1,4})(?:,%sp)?$/o;
59         } elsif ($arch eq 'mips64') {
60                 #8800402c:       67bdfff0        daddiu  sp,sp,-16
61                 $re = qr/.*daddiu.*sp,sp,-(([0-9]{2}|[3-9])[0-9]{2})/o;
62         } elsif ($arch eq 'mips') {
63                 #88003254:       27bdffe0        addiu   sp,sp,-32
64                 $re = qr/.*addiu.*sp,sp,-(([0-9]{2}|[3-9])[0-9]{2})/o;
65         } elsif ($arch eq 'ppc') {
66                 #c00029f4:       94 21 ff 30     stwu    r1,-208(r1)
67                 $re = qr/.*stwu.*r1,-($x{1,8})\(r1\)/o;
68         } elsif ($arch eq 'ppc64') {
69                 #XXX
70                 $re = qr/.*stdu.*r1,-($x{1,8})\(r1\)/o;
71         } elsif ($arch eq 'powerpc') {
72                 $re = qr/.*st[dw]u.*r1,-($x{1,8})\(r1\)/o;
73         } elsif ($arch =~ /^s390x?$/) {
74                 #   11160:       a7 fb ff 60             aghi   %r15,-160
75                 $re = qr/.*ag?hi.*\%r15,-(([0-9]{2}|[3-9])[0-9]{2})/o;
76         } elsif ($arch =~ /^sh64$/) {
77                 #XXX: we only check for the immediate case presently,
78                 #     though we will want to check for the movi/sub
79                 #     pair for larger users. -- PFM.
80                 #a00048e0:       d4fc40f0        addi.l  r15,-240,r15
81                 $re = qr/.*addi\.l.*r15,-(([0-9]{2}|[3-9])[0-9]{2}),r15/o;
82         } else {
83                 print("wrong or unknown architecture\n");
84                 exit
85         }
86 }
87
88 sub bysize($) {
89         my ($asize, $bsize);
90         ($asize = $a) =~ s/.*:  *(.*)$/$1/;
91         ($bsize = $b) =~ s/.*:  *(.*)$/$1/;
92         $bsize <=> $asize
93 }
94
95 #
96 # main()
97 #
98 my $funcre = qr/^$x* <(.*)>:$/;
99 my $func;
100 my $file, $lastslash;
101
102 while (my $line = <STDIN>) {
103         if ($line =~ m/$funcre/) {
104                 $func = $1;
105         }
106         elsif ($line =~ m/(.*):\s*file format/) {
107                 $file = $1;
108                 $file =~ s/\.ko//;
109                 $lastslash = rindex($file, "/");
110                 if ($lastslash != -1) {
111                         $file = substr($file, $lastslash + 1);
112                 }
113         }
114         elsif ($line =~ m/$re/) {
115                 my $size = $1;
116                 $size = hex($size) if ($size =~ /^0x/);
117
118                 if ($size > 0xf0000000) {
119                         $size = - $size;
120                         $size += 0x80000000;
121                         $size += 0x80000000;
122                 }
123                 next if ($size > 0x10000000);
124
125                 next if $line !~ m/^($xs*)/;
126                 my $addr = $1;
127                 $addr =~ s/ /0/g;
128                 $addr = "0x$addr";
129
130                 # bbox: was: my $intro = "$addr $func [$file]:";
131                 my $intro = "$func [$file]:";
132                 my $padlen = 56 - length($intro);
133                 while ($padlen > 0) {
134                         $intro .= '     ';
135                         $padlen -= 8;
136                 }
137                 next if ($size < 100);
138                 push @stack, "$intro$size\n";
139         }
140 }
141
142 print sort bysize @stack;