2014-11-15 12:13:40 +01:00
|
|
|
# Helper function
|
|
|
|
# Get the MD5 sum of the file open at the filehandle
|
|
|
|
sub md5fh
|
|
|
|
{
|
|
|
|
my $fh_orig = shift;
|
|
|
|
open(my $fh, "<&", $fh_orig); # Duplicate the file handle so the seek
|
|
|
|
seek($fh, 0, 0); # does not affect the caller
|
|
|
|
my $md5 = Digest::MD5->new->addfile($fh)->hexdigest;
|
|
|
|
close($fh);
|
|
|
|
return $md5;
|
|
|
|
}
|
|
|
|
|
2014-11-17 00:31:33 +01:00
|
|
|
# Get the file size from stat() (by file handle or name)
|
|
|
|
sub statSize
|
|
|
|
{
|
|
|
|
my $f = shift;
|
|
|
|
my @s = stat($f) or die("stat on '$f' failed");
|
|
|
|
return $s[7];
|
|
|
|
}
|
|
|
|
|
|
|
|
# Get the file size by read()ing the whole file
|
|
|
|
sub readSize
|
|
|
|
{
|
|
|
|
my $fh = shift;
|
|
|
|
seek($fh, 0, 0);
|
2014-11-14 23:01:41 +01:00
|
|
|
my $block = 4*1024;
|
|
|
|
my $s;
|
|
|
|
my $data;
|
|
|
|
my $sum = read($fh, $data, $block);
|
|
|
|
while ( $s = read($fh, $data, $block) )
|
|
|
|
{
|
|
|
|
$sum += $s;
|
|
|
|
}
|
|
|
|
$data = "";
|
|
|
|
return $sum;
|
2014-11-17 00:31:33 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
# Verify that the size of the file passed by filehandle matches the target size s0
|
|
|
|
# Checks both stat() and read()
|
|
|
|
sub sizeVerify
|
|
|
|
{
|
|
|
|
my $ok = 1;
|
|
|
|
my $fh = shift;
|
|
|
|
my $s0 = shift;
|
|
|
|
$ss = statSize($fh);
|
|
|
|
if ($s0 != $ss) {
|
|
|
|
$ok = 0;
|
|
|
|
print("# stat size $ss, expected $s0\n");
|
|
|
|
}
|
|
|
|
$sr = readSize($fh);
|
|
|
|
if ($s0 != $sr) {
|
|
|
|
$ok = 0;
|
|
|
|
print("# read size $sr, expected $s0\n");
|
|
|
|
}
|
|
|
|
return $ok;
|
|
|
|
}
|
|
|
|
|
|
|
|
# Wait for a file to appear
|
|
|
|
use Time::HiRes qw(usleep);
|
|
|
|
sub waitForFile
|
|
|
|
{
|
|
|
|
my $file = shift;
|
|
|
|
my $timeout;
|
|
|
|
$timeout = shift or $timeout = 5;
|
|
|
|
for(my $i = $timeout*10; $i > 0; $i--)
|
|
|
|
{
|
|
|
|
-f $file and return 1;
|
|
|
|
usleep(100000); # 0.1 seconds
|
|
|
|
}
|
|
|
|
print "# timeout waiting for '$file' to appear\n";
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-11-15 12:13:40 +01:00
|
|
|
# As this file will be require()'d, it needs to return true
|
|
|
|
return 1;
|