Pārlūkot izejas kodu

[util] Rewrite catrom.pl to use Option::ROM library

Signed-off-by: Michael Brown <mcb30@ipxe.org>
tags/v1.20.1
Michael Brown 12 gadus atpakaļ
vecāks
revīzija
12be8bc544
1 mainītis faili ar 15 papildinājumiem un 34 dzēšanām
  1. 15
    34
      src/util/catrom.pl

+ 15
- 34
src/util/catrom.pl Parādīt failu

@@ -3,46 +3,27 @@
3 3
 use warnings;
4 4
 use strict;
5 5
 
6
-use bytes;
6
+use FindBin;
7
+use lib "$FindBin::Bin";
8
+use Option::ROM qw ( :all );
7 9
 
8
-use constant MAX_ROM_LEN => 1024*1024;
9
-use constant PCI_OFF => 0x18;
10
-use constant INDICATOR_OFF => 0x15;
11
-
12
-my $total_len = 0;
13 10
 my @romfiles = @ARGV
14 11
     or die "Usage: $0 rom-file-1 rom-file-2 ... > multi-rom-file\n";
15 12
 
16 13
 while ( my $romfile = shift @romfiles ) {
17
-  my $last = @romfiles ? 0 : 1;
18
-
19
-  open ROM, "<$romfile" or die "Could not open $romfile: $!\n";
20
-  my $len = read ( ROM, my $romdata, MAX_ROM_LEN )
21
-      or die "Could not read $romfile: $!\n";
22
-  close ROM;
23
-
24
-  die "$romfile is not a ROM file\n"
25
-      unless substr ( $romdata, 0, 2 ) eq "\x55\xAA";
26 14
 
27
-  ( my $checklen ) = unpack ( 'C', substr ( $romdata, 2, 1 ) );
28
-  $checklen *= 512;
29
-  die "$romfile has incorrect length field $checklen (should be $len)\n"
30
-      unless $len == $checklen;
15
+  # Read ROM file
16
+  my $rom = new Option::ROM;
17
+  $rom->load ( $romfile );
31 18
 
32
-  ( my $pci ) = unpack ( 'v', substr ( $romdata, PCI_OFF, 2 ) );
33
-  die "Invalid PCI offset field in $romfile\n"
34
-      if $pci >= $len;
35
-  die "No PCIR signature in $romfile\n"
36
-      unless substr ( $romdata, $pci, 4 ) eq "PCIR";
37
-  
38
-  ( my $indicator ) =
39
-      unpack ( 'C', substr ( $romdata, $pci + INDICATOR_OFF, 1 ) );
40
-  my $msg = sprintf ( "$romfile: indicator was %02x, ", $indicator );
41
-  $indicator &= ! ( 1 << 7 );
42
-  $indicator |= ( $last << 7 );
43
-  $msg .= sprintf ( "now %02x\n", $indicator );
44
-  substr ( $romdata, $pci + INDICATOR_OFF, 1 ) = pack ( 'C', $indicator );
45
-  warn $msg;
19
+  # Tag final image as non-final in all except the final ROM
20
+  if ( @romfiles ) {
21
+    my $image = $rom;
22
+    $image = $image->next_image() while $image->next_image();
23
+    $image->pci_header->{last_image} &= ~PCI_LAST_IMAGE;
24
+    $image->fix_checksum();
25
+  }
46 26
 
47
-  print $romdata;
27
+  # Write ROM file to STDOUT
28
+  $rom->save ( "-" );
48 29
 }

Notiek ielāde…
Atcelt
Saglabāt