cirros-testvm/src-cirros/buildroot-2015.05/package/rpi-firmware/mkknlimg

   1 #!/usr/bin/env perl
   2 #
   3 # Originaly from: https://github.com/raspberrypi/tools/blob/master/mkimage/mkknlimg
   4 # Original cset : 92a2df13b887f4799554bf92c4fc78d53668c868
   5
   6 use strict;
   7 use integer;
   8
   9 my $trailer_magic = 'RPTL';
  10
  11 my $tmpfile1 = "/tmp/mkknlimg_$$.1";
  12 my $tmpfile2 = "/tmp/mkknlimg_$$.2";
  13
  14 my $dtok = 0;
  15
  16 while ($ARGV[0] =~ /^-/)
  17 {
  18     my $arg = shift(@ARGV);
  19     if ($arg eq '--dtok')
  20     {
  21         $dtok = 1;
  22     }
  23     else
  24     {
  25         print ("* Unknown option '$arg'\n");
  26         usage();
  27     }
  28 }
  29
  30 usage() if (@ARGV != 2);
  31
  32 my $kernel_file = $ARGV[0];
  33 my $out_file = $ARGV[1];
  34
  35 my @wanted_config_lines =
  36 (
  37         'CONFIG_BCM2708_DT'
  38 );
  39
  40 my @wanted_strings =
  41 (
  42         'bcm2708_fb',
  43         'brcm,bcm2708-pinctrl',
  44         'brcm,bcm2835-gpio',
  45         'of_find_property'
  46 );
  47
  48 my $res = try_extract($kernel_file, $tmpfile1);
  49
  50 $res = try_decompress('\037\213\010', 'xy',    'gunzip', 0,
  51                       $kernel_file, $tmpfile1, $tmpfile2) if (!$res);
  52 $res = try_decompress('\3757zXZ\000', 'abcde', 'unxz --single-stream', -1,
  53                       $kernel_file, $tmpfile1, $tmpfile2) if (!$res);
  54 $res = try_decompress('BZh',          'xy',    'bunzip2', 0,
  55                       $kernel_file, $tmpfile1, $tmpfile2) if (!$res);
  56 $res = try_decompress('\135\0\0\0',   'xxx',   'unlzma', 0,
  57                       $kernel_file, $tmpfile1, $tmpfile2) if (!$res);
  58 $res = try_decompress('\211\114\132', 'xy',    'lzop -d', 0,
  59                       $kernel_file, $tmpfile1, $tmpfile2) if (!$res);
  60
  61 my $append_trailer;
  62 my $trailer;
  63
  64 if ($res)
  65 {
  66     print("Version: $res->{''}\n");
  67
  68     $append_trailer = $dtok;
  69     if (!$dtok)
  70     {
  71         if (config_bool($res, 'bcm2708_fb'))
  72         {
  73             $dtok ||= config_bool($res, 'CONFIG_BCM2708_DT');
  74             $dtok ||= config_bool($res, 'brcm,bcm2708-pinctrl');
  75             $dtok ||= config_bool($res, 'brcm,bcm2835-gpio');
  76             $append_trailer = 1;
  77         }
  78         else
  79         {
  80             print ("* This doesn't look like a Raspberry Pi kernel. In pass-through mode.\n");
  81         }
  82     }
  83 }
  84 else
  85 {
  86     print ("* Is this a valid kernel? In pass-through mode.\n");
  87 }
  88
  89 if ($append_trailer)
  90 {
  91     printf("DT: %s\n", $dtok ? "y" : "n");
  92
  93     my @atoms;
  94
  95     push @atoms, [ $trailer_magic, pack('V', 0) ];
  96     push @atoms, [ 'KVer', $res->{''} ];
  97     push @atoms, [ 'DTOK', pack('V', $dtok) ];
  98
  99     $trailer = pack_trailer(\@atoms);
 100     $atoms[0]->[1] = pack('V', length($trailer));
 101
 102     $trailer = pack_trailer(\@atoms);
 103 }
 104
 105 die "* Failed to open '$kernel_file'\n" if (!open(my $ifh, '<', $kernel_file));
 106 die "* Failed to create '$out_file'\n" if (!open(my $ofh, '>', $out_file));
 107
 108 my $copybuf;
 109 my $total_len = 0;
 110 while (1)
 111 {
 112         my $bytes = sysread($ifh, $copybuf, 64*1024);
 113         last if (!$bytes);
 114         syswrite($ofh, $copybuf, $bytes);
 115         $total_len += $bytes;
 116 }
 117
 118 if ($trailer)
 119 {
 120     # Pad to word-alignment
 121     syswrite($ofh, "\x000\x000\x000", (-$total_len & 0x3));
 122     syswrite($ofh, $trailer);
 123 }
 124
 125 close($ifh);
 126 close($ofh);
 127
 128 exit($trailer ? 0 : 1);
 129
 130 END {
 131         unlink($tmpfile1) if ($tmpfile1);
 132         unlink($tmpfile2) if ($tmpfile2);
 133 }
 134
 135
 136 sub usage
 137 {
 138         print ("Usage: mkknlimg [--dtok] <vmlinux|zImage|bzImage> <outfile>\n");
 139         exit(1);
 140 }
 141
 142 sub try_extract
 143 {
 144         my ($knl, $tmp) = @_;
 145
 146         my $ver = `strings "$knl" | grep -a -E "^Linux version [1-9]"`;
 147
 148         return undef if (!$ver);
 149
 150         chomp($ver);
 151
 152         my $res = { ''=>$ver };
 153         my $string_pattern = '^('.join('|', @wanted_strings).')$';
 154
 155         my @matches = `strings \"$knl\" | grep -E \"$string_pattern\"`;
 156         foreach my $match (@matches)
 157         {
 158             chomp($match);
 159             $res->{$match} = 1;
 160         }
 161
 162         my $config_pattern = '^('.join('|', @wanted_config_lines).')=(.*)$';
 163         my $cf1 = 'IKCFG_ST\037\213\010';
 164         my $cf2 = '0123456789';
 165
 166         my $pos = `tr "$cf1\n$cf2" "\n$cf2=" < "$knl" | grep -abo "^$cf2"`;
 167         if ($pos)
 168         {
 169                 $pos =~ s/:.*[\r\n]*$//s;
 170                 $pos += 8;
 171                 my $err = (system("tail -c+$pos \"$knl\" | zcat > $tmp 2> /dev/null") >> 8);
 172                 if (($err == 0) || ($err == 2))
 173                 {
 174                         if (open(my $fh, '<', $tmp))
 175                         {
 176                                 while (my $line = <$fh>)
 177                                 {
 178                                         chomp($line);
 179                                         $res->{$1} = $2 if ($line =~ /$config_pattern/);
 180                                 }
 181
 182                                 close($fh);
 183                         }
 184                 }
 185         }
 186
 187         return $res;
 188 }
 189
 190
 191 sub try_decompress
 192 {
 193         my ($magic, $subst, $zcat, $idx, $knl, $tmp1, $tmp2) = @_;
 194
 195         my $pos = `tr "$magic\n$subst" "\n$subst=" < "$knl" | grep -abo "^$subst"`;
 196         if ($pos)
 197         {
 198                 chomp($pos);
 199                 $pos = (split(/[\r\n]+/, $pos))[$idx];
 200                 $pos =~ s/:.*[\r\n]*$//s;
 201                 my $cmd = "tail -c+$pos \"$knl\" | $zcat > $tmp2 2> /dev/null";
 202                 my $err = (system($cmd) >> 8);
 203                 return undef if (($err != 0) && ($err != 2));
 204
 205                 return try_extract($tmp2, $tmp1);
 206         }
 207
 208         return undef;
 209 }
 210
 211 sub pack_trailer
 212 {
 213         my ($atoms) = @_;
 214         my $trailer = pack('VV', 0, 0);
 215         for (my $i = $#$atoms; $i>=0; $i--)
 216         {
 217                 my $atom = $atoms->[$i];
 218                 $trailer .= pack('a*x!4Va4', $atom->[1], length($atom->[1]), $atom->[0]);
 219         }
 220         return $trailer;
 221 }
 222
 223 sub config_bool
 224 {
 225         my ($configs, $wanted) = @_;
 226         return (($configs->{$wanted} eq 'y') || ($configs->{$wanted} eq '1'));
 227 }