## -----------------------------------------------------------------------
##
-## 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;