prdhcp.c32: Print DHCP packets
[profile/ivi/syslinux.git] / mbr / checksize.pl
index 4feb20a..4b42327 100755 (executable)
@@ -1,6 +1,7 @@
 ## -----------------------------------------------------------------------
 ##
-##   Copyright 2007 H. Peter Anvin - All Rights Reserved
+##   Copyright 2007-2009 H. Peter Anvin - All Rights Reserved
+##   Copyright 2009 Intel Corporation; author: H. Peter Anvin
 ##
 ##   This program is free software; you can redistribute it and/or modify
 ##   it under the terms of the GNU General Public License as published by
 ##
 ## checksize.pl
 ##
-## Check the size of a binary file
+## Check the size of a binary file and pad it with zeroes to that size
 ##
 
-($file, $maxsize) = @ARGV;
+use bytes;
 
-@st = stat($file);
+($file, $maxsize, $padsize) = @ARGV;
+
+if (!defined($maxsize)) {
+    # Defaults based on the filename
+    if ($file =~ /^mbr[^0-9a-z]/) {
+       $maxsize = $padsize = 440;
+    } elsif ($file =~ /^gptmbr[^0-9a-z]/) {
+       $maxsize = $padsize = 440;
+    } elsif ($file =~ /^isohdp[fp]x[^0-9a-z]/) {
+       $maxsize = $padsize = 432;
+    } elsif ($file =~ /^altmbr[^0-9a-z]/) {
+       $maxsize = $padsize = 439;
+    } else {
+       die "$0: no default size for filename: $file\n";
+    }
+}
+
+$padsize = $maxsize unless(defined($padsize));
+
+open(FILE, '+<', $file) or die;
+@st = stat(FILE);
 if (!defined($size = $st[7])) {
     die "$0: $file: $!\n";
 }
 if ($size > $maxsize) {
     print STDERR "$file: too big ($size > $maxsize)\n";
     exit 1;
-} else {
-    exit 0;
+} elsif ($size < $padsize) {
+    seek(FILE, $size, 0);
+    print FILE "\0" x ($padsize-$size);
 }
+
+exit 0;