X-Git-Url: http://git.rot13.org/?p=BackupPC.git;a=blobdiff_plain;f=lib%2FBackupPC%2FLib.pm;h=ecc455dbcfe873df57d39ad9fa6c1a2bb5d19f14;hp=02177a0a8a29c407c51e91b53f8baa5dafb6a7cb;hb=f6257f558390295c581f4e5af8f084341db05d34;hpb=1282f5f90cf092cb406ee186e6bf328875e4c306 diff --git a/lib/BackupPC/Lib.pm b/lib/BackupPC/Lib.pm index 02177a0..ecc455d 100644 --- a/lib/BackupPC/Lib.pm +++ b/lib/BackupPC/Lib.pm @@ -29,7 +29,7 @@ # #======================================================================== # -# Version 1.5.0, released 2 Aug 2002. +# Version 2.0.0beta2, released 13 Apr 2003. # # See http://backuppc.sourceforge.net. # @@ -52,65 +52,78 @@ use Digest::MD5; sub new { my $class = shift; - my($topDir) = @_; - my $self = bless { + my($topDir, $installDir, $noUserCheck) = @_; + + my $bpc = bless { TopDir => $topDir || '/data/BackupPC', - BinDir => '/usr/local/BackupPC/bin', - LibDir => '/usr/local/BackupPC/lib', - Version => '1.5.0', + BinDir => $installDir || '/usr/local/BackupPC', + LibDir => $installDir || '/usr/local/BackupPC', + Version => '2.0.0beta2', BackupFields => [qw( num type startTime endTime nFiles size nFilesExist sizeExist nFilesNew sizeNew xferErrs xferBadFile xferBadShare tarErrs compress sizeExistComp sizeNewComp - noFill fillFromNum mangle + noFill fillFromNum mangle xferMethod level )], RestoreFields => [qw( num startTime endTime result errorMsg nFiles size tarCreateErrs xferErrs )], }, $class; + $bpc->{BinDir} .= "/bin"; + $bpc->{LibDir} .= "/lib"; # # Clean up %ENV and setup other variables. # delete @ENV{qw(IFS CDPATH ENV BASH_ENV)}; - $self->{PoolDir} = "$self->{TopDir}/pool"; - $self->{CPoolDir} = "$self->{TopDir}/cpool"; - if ( defined(my $error = $self->ConfigRead()) ) { + $bpc->{PoolDir} = "$bpc->{TopDir}/pool"; + $bpc->{CPoolDir} = "$bpc->{TopDir}/cpool"; + if ( defined(my $error = $bpc->ConfigRead()) ) { print(STDERR $error, "\n"); return; } - return $self; + # + # Verify we are running as the correct user + # + if ( !$noUserCheck + && $bpc->{Conf}{BackupPCUserVerify} + && $> != (my $uid = (getpwnam($bpc->{Conf}{BackupPCUser}))[2]) ) { + print("Wrong user: my userid is $>, instead of $uid" + . " ($bpc->{Conf}{BackupPCUser})\n"); + return; + } + return $bpc; } sub TopDir { - my($self) = @_; - return $self->{TopDir}; + my($bpc) = @_; + return $bpc->{TopDir}; } sub BinDir { - my($self) = @_; - return $self->{BinDir}; + my($bpc) = @_; + return $bpc->{BinDir}; } sub Version { - my($self) = @_; - return $self->{Version}; + my($bpc) = @_; + return $bpc->{Version}; } sub Conf { - my($self) = @_; - return %{$self->{Conf}}; + my($bpc) = @_; + return %{$bpc->{Conf}}; } sub Lang { - my($self) = @_; - return $self->{Lang}; + my($bpc) = @_; + return $bpc->{Lang}; } sub adminJob @@ -123,9 +136,24 @@ sub trashJob return " trashClean "; } +sub ConfValue +{ + my($bpc, $param) = @_; + + return $bpc->{Conf}{$param}; +} + +sub verbose +{ + my($bpc, $param) = @_; + + $bpc->{verbose} = $param if ( defined($param) ); + return $bpc->{verbose}; +} + sub timeStamp { - my($self, $t, $noPad) = @_; + my($bpc, $t, $noPad) = @_; my($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime($t || time); $year += 1900; @@ -141,7 +169,7 @@ sub timeStamp # sub timeStampISO { - my($self, $t, $noPad) = @_; + my($bpc, $t, $noPad) = @_; my($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime($t || time); $year += 1900; @@ -153,17 +181,17 @@ sub timeStampISO sub BackupInfoRead { - my($self, $host) = @_; + my($bpc, $host) = @_; local(*BK_INFO, *LOCK); my(@Backups); - flock(LOCK, LOCK_EX) if open(LOCK, "$self->{TopDir}/pc/$host/LOCK"); - if ( open(BK_INFO, "$self->{TopDir}/pc/$host/backups") ) { + flock(LOCK, LOCK_EX) if open(LOCK, "$bpc->{TopDir}/pc/$host/LOCK"); + if ( open(BK_INFO, "$bpc->{TopDir}/pc/$host/backups") ) { while ( ) { s/[\n\r]+//; next if ( !/^(\d+\t(incr|full)[\d\t]*$)/ ); $_ = $1; - @{$Backups[@Backups]}{@{$self->{BackupFields}}} = split(/\t/); + @{$Backups[@Backups]}{@{$bpc->{BackupFields}}} = split(/\t/); } close(BK_INFO); } @@ -173,20 +201,20 @@ sub BackupInfoRead sub BackupInfoWrite { - my($self, $host, @Backups) = @_; + my($bpc, $host, @Backups) = @_; local(*BK_INFO, *LOCK); my($i); - flock(LOCK, LOCK_EX) if open(LOCK, "$self->{TopDir}/pc/$host/LOCK"); - unlink("$self->{TopDir}/pc/$host/backups.old") - if ( -f "$self->{TopDir}/pc/$host/backups.old" ); - rename("$self->{TopDir}/pc/$host/backups", - "$self->{TopDir}/pc/$host/backups.old") - if ( -f "$self->{TopDir}/pc/$host/backups" ); - if ( open(BK_INFO, ">$self->{TopDir}/pc/$host/backups") ) { + flock(LOCK, LOCK_EX) if open(LOCK, "$bpc->{TopDir}/pc/$host/LOCK"); + unlink("$bpc->{TopDir}/pc/$host/backups.old") + if ( -f "$bpc->{TopDir}/pc/$host/backups.old" ); + rename("$bpc->{TopDir}/pc/$host/backups", + "$bpc->{TopDir}/pc/$host/backups.old") + if ( -f "$bpc->{TopDir}/pc/$host/backups" ); + if ( open(BK_INFO, ">$bpc->{TopDir}/pc/$host/backups") ) { for ( $i = 0 ; $i < @Backups ; $i++ ) { my %b = %{$Backups[$i]}; - printf(BK_INFO "%s\n", join("\t", @b{@{$self->{BackupFields}}})); + printf(BK_INFO "%s\n", join("\t", @b{@{$bpc->{BackupFields}}})); } close(BK_INFO); } @@ -195,17 +223,17 @@ sub BackupInfoWrite sub RestoreInfoRead { - my($self, $host) = @_; + my($bpc, $host) = @_; local(*RESTORE_INFO, *LOCK); my(@Restores); - flock(LOCK, LOCK_EX) if open(LOCK, "$self->{TopDir}/pc/$host/LOCK"); - if ( open(RESTORE_INFO, "$self->{TopDir}/pc/$host/restores") ) { + flock(LOCK, LOCK_EX) if open(LOCK, "$bpc->{TopDir}/pc/$host/LOCK"); + if ( open(RESTORE_INFO, "$bpc->{TopDir}/pc/$host/restores") ) { while ( ) { s/[\n\r]+//; next if ( !/^(\d+.*)/ ); $_ = $1; - @{$Restores[@Restores]}{@{$self->{RestoreFields}}} = split(/\t/); + @{$Restores[@Restores]}{@{$bpc->{RestoreFields}}} = split(/\t/); } close(RESTORE_INFO); } @@ -215,21 +243,21 @@ sub RestoreInfoRead sub RestoreInfoWrite { - my($self, $host, @Restores) = @_; + my($bpc, $host, @Restores) = @_; local(*RESTORE_INFO, *LOCK); my($i); - flock(LOCK, LOCK_EX) if open(LOCK, "$self->{TopDir}/pc/$host/LOCK"); - unlink("$self->{TopDir}/pc/$host/restores.old") - if ( -f "$self->{TopDir}/pc/$host/restores.old" ); - rename("$self->{TopDir}/pc/$host/restores", - "$self->{TopDir}/pc/$host/restores.old") - if ( -f "$self->{TopDir}/pc/$host/restores" ); - if ( open(RESTORE_INFO, ">$self->{TopDir}/pc/$host/restores") ) { + flock(LOCK, LOCK_EX) if open(LOCK, "$bpc->{TopDir}/pc/$host/LOCK"); + unlink("$bpc->{TopDir}/pc/$host/restores.old") + if ( -f "$bpc->{TopDir}/pc/$host/restores.old" ); + rename("$bpc->{TopDir}/pc/$host/restores", + "$bpc->{TopDir}/pc/$host/restores.old") + if ( -f "$bpc->{TopDir}/pc/$host/restores" ); + if ( open(RESTORE_INFO, ">$bpc->{TopDir}/pc/$host/restores") ) { for ( $i = 0 ; $i < @Restores ; $i++ ) { my %b = %{$Restores[$i]}; printf(RESTORE_INFO "%s\n", - join("\t", @b{@{$self->{RestoreFields}}})); + join("\t", @b{@{$bpc->{RestoreFields}}})); } close(RESTORE_INFO); } @@ -238,13 +266,15 @@ sub RestoreInfoWrite sub ConfigRead { - my($self, $host) = @_; + my($bpc, $host) = @_; my($ret, $mesg, $config, @configs); - $self->{Conf} = (); - push(@configs, "$self->{TopDir}/conf/config.pl"); - push(@configs, "$self->{TopDir}/pc/$host/config.pl") - if ( defined($host) && -f "$self->{TopDir}/pc/$host/config.pl" ); + $bpc->{Conf} = (); + push(@configs, "$bpc->{TopDir}/conf/config.pl"); + push(@configs, "$bpc->{TopDir}/conf/$host.pl") + if ( $host ne "config" && -f "$bpc->{TopDir}/conf/$host.pl" ); + push(@configs, "$bpc->{TopDir}/pc/$host/config.pl") + if ( defined($host) && -f "$bpc->{TopDir}/pc/$host/config.pl" ); foreach $config ( @configs ) { %Conf = (); if ( !defined($ret = do $config) && ($! || $@) ) { @@ -253,16 +283,28 @@ sub ConfigRead $mesg =~ s/[\n\r]+//; return $mesg; } - %{$self->{Conf}} = ( %{$self->{Conf} || {}}, %Conf ); + %{$bpc->{Conf}} = ( %{$bpc->{Conf} || {}}, %Conf ); + } + return if ( !defined($bpc->{Conf}{Language}) ); + if ( defined($bpc->{Conf}{PerlModuleLoad}) ) { + # + # Load any user-specified perl modules. This is for + # optional user-defined extensions. + # + $bpc->{Conf}{PerlModuleLoad} = [$bpc->{Conf}{PerlModuleLoad}] + if ( ref($bpc->{Conf}{PerlModuleLoad}) ne "ARRAY" ); + foreach my $module ( @{$bpc->{Conf}{PerlModuleLoad}} ) { + eval("use $module;"); + } } - my $langFile = "$self->{LibDir}/BackupPC/Lang/$self->{Conf}{Language}.pm"; + my $langFile = "$bpc->{LibDir}/BackupPC/Lang/$bpc->{Conf}{Language}.pm"; if ( !defined($ret = do $langFile) && ($! || $@) ) { $mesg = "Couldn't open language file $langFile: $!" if ( $! ); $mesg = "Couldn't execute language file $langFile: $@" if ( $@ ); $mesg =~ s/[\n\r]+//; return $mesg; } - $self->{Lang} = \%Lang; + $bpc->{Lang} = \%Lang; return; } @@ -271,12 +313,12 @@ sub ConfigRead # sub ConfigMTime { - my($self) = @_; - return (stat("$self->{TopDir}/conf/config.pl"))[9]; + my($bpc) = @_; + return (stat("$bpc->{TopDir}/conf/config.pl"))[9]; } # -# Returns information from the host file in $self->{TopDir}/conf/hosts. +# Returns information from the host file in $bpc->{TopDir}/conf/hosts. # With no argument a ref to a hash of hosts is returned. Each # hash contains fields as specified in the hosts file. With an # argument a ref to a single hash is returned with information @@ -284,21 +326,32 @@ sub ConfigMTime # sub HostInfoRead { - my($self, $host) = @_; + my($bpc, $host) = @_; my(%hosts, @hdr, @fld); local(*HOST_INFO); - if ( !open(HOST_INFO, "$self->{TopDir}/conf/hosts") ) { - print(STDERR $self->timeStamp, - "Can't open $self->{TopDir}/conf/hosts\n"); + if ( !open(HOST_INFO, "$bpc->{TopDir}/conf/hosts") ) { + print(STDERR $bpc->timeStamp, + "Can't open $bpc->{TopDir}/conf/hosts\n"); return {}; } while ( ) { s/[\n\r]+//; s/#.*//; s/\s+$//; - next if ( /^\s*$/ || !/^([\w\.-]+\s+.*)/ ); - @fld = split(/\s+/, $1); + next if ( /^\s*$/ || !/^([\w\.\\-]+\s+.*)/ ); + # + # Split on white space, except if preceded by \ + # using zero-width negative look-behind assertion + # (always wanted to use one of those). + # + @fld = split(/(?{TopDir}/conf/hosts"))[9]; + my($bpc) = @_; + return (stat("$bpc->{TopDir}/conf/hosts"))[9]; } # @@ -335,7 +388,7 @@ sub HostsMTime # sub RmTreeQuiet { - my($self, $pwd, $roots) = @_; + my($bpc, $pwd, $roots) = @_; my(@files, $root); if ( defined($roots) && length($roots) ) { @@ -359,7 +412,7 @@ sub RmTreeQuiet @files = $d->read; $d->close; @files = grep $_!~/^\.{1,2}$/, @files; - $self->RmTreeQuiet("$pwd/$root", \@files); + $bpc->RmTreeQuiet("$pwd/$root", \@files); chdir($pwd); rmdir($root) || rmdir($root); } else { @@ -374,7 +427,7 @@ sub RmTreeQuiet # sub RmTreeDefer { - my($self, $trashDir, $file) = @_; + my($bpc, $trashDir, $file) = @_; my($i, $f); return if ( !-e $file ); @@ -390,28 +443,33 @@ sub RmTreeDefer my($d) = $1; my($f) = $2; my($cwd) = Cwd::fastcwd(); - $self->RmTreeQuiet($d, $f); + $cwd = $1 if ( $cwd =~ /(.*)/ ); + $bpc->RmTreeQuiet($d, $f); chdir($cwd) if ( $cwd ); } } # -# Empty the trash directory. Returns 0 if it did nothing. +# Empty the trash directory. Returns 0 if it did nothing, 1 if it +# did something, -1 if it failed to remove all the files. # sub RmTreeTrashEmpty { - my($self, $trashDir) = @_; + my($bpc, $trashDir) = @_; my(@files); my($cwd) = Cwd::fastcwd(); + $cwd = $1 if ( $cwd =~ /(.*)/ ); return if ( !-d $trashDir ); - my $d = DirHandle->new($trashDir) - or carp "Can't read $trashDir: $!"; + my $d = DirHandle->new($trashDir) or carp "Can't read $trashDir: $!"; @files = $d->read; $d->close; @files = grep $_!~/^\.{1,2}$/, @files; return 0 if ( !@files ); - $self->RmTreeQuiet($trashDir, \@files); + $bpc->RmTreeQuiet($trashDir, \@files); + foreach my $f ( @files ) { + return -1 if ( -e $f ); + } chdir($cwd) if ( $cwd ); return 1; } @@ -422,14 +480,14 @@ sub RmTreeTrashEmpty # sub ServerConnect { - my($self, $host, $port, $justConnect) = @_; + my($bpc, $host, $port, $justConnect) = @_; local(*FH); - return if ( defined($self->{ServerFD}) ); + return if ( defined($bpc->{ServerFD}) ); # # First try the unix-domain socket # - my $sockFile = "$self->{TopDir}/log/BackupPC.sock"; + my $sockFile = "$bpc->{TopDir}/log/BackupPC.sock"; socket(*FH, PF_UNIX, SOCK_STREAM, 0) || return "unix socket: $!"; if ( !connect(*FH, sockaddr_un($sockFile)) ) { my $err = "unix connect: $!"; @@ -447,14 +505,14 @@ sub ServerConnect } } my($oldFH) = select(*FH); $| = 1; select($oldFH); - $self->{ServerFD} = *FH; + $bpc->{ServerFD} = *FH; return if ( $justConnect ); # # Read the seed that we need for our MD5 message digest. See # ServerMesg below. # - sysread($self->{ServerFD}, $self->{ServerSeed}, 1024); - $self->{ServerMesgCnt} = 0; + sysread($bpc->{ServerFD}, $bpc->{ServerSeed}, 1024); + $bpc->{ServerMesgCnt} = 0; return; } @@ -463,13 +521,13 @@ sub ServerConnect # sub ServerOK { - my($self) = @_; + my($bpc) = @_; - return 0 if ( !defined($self->{ServerFD}) ); - vec(my $FDread, fileno($self->{ServerFD}), 1) = 1; + return 0 if ( !defined($bpc->{ServerFD}) ); + vec(my $FDread, fileno($bpc->{ServerFD}), 1) = 1; my $ein = $FDread; return 0 if ( select(my $rout = $FDread, undef, $ein, 0.0) < 0 ); - return 1 if ( !vec($rout, fileno($self->{ServerFD}), 1) ); + return 1 if ( !vec($rout, fileno($bpc->{ServerFD}), 1) ); } # @@ -477,10 +535,10 @@ sub ServerOK # sub ServerDisconnect { - my($self) = @_; - return if ( !defined($self->{ServerFD}) ); - close($self->{ServerFD}); - delete($self->{ServerFD}); + my($bpc) = @_; + return if ( !defined($bpc->{ServerFD}) ); + close($bpc->{ServerFD}); + delete($bpc->{ServerFD}); } # @@ -501,13 +559,13 @@ sub ServerDisconnect # sub ServerMesg { - my($self, $mesg) = @_; - return if ( !defined(my $fh = $self->{ServerFD}) ); + my($bpc, $mesg) = @_; + return if ( !defined(my $fh = $bpc->{ServerFD}) ); my $md5 = Digest::MD5->new; - $md5->add($self->{ServerSeed} . $self->{ServerMesgCnt} - . $self->{Conf}{ServerMesgSecret} . $mesg); + $md5->add($bpc->{ServerSeed} . $bpc->{ServerMesgCnt} + . $bpc->{Conf}{ServerMesgSecret} . $mesg); print($fh $md5->b64digest . " $mesg\n"); - $self->{ServerMesgCnt}++; + $bpc->{ServerMesgCnt}++; return <$fh>; } @@ -516,12 +574,12 @@ sub ServerMesg # sub ChildInit { - my($self) = @_; + my($bpc) = @_; close(STDERR); open(STDERR, ">&STDOUT"); select(STDERR); $| = 1; select(STDOUT); $| = 1; - $ENV{PATH} = $self->{Conf}{MyPath}; + $ENV{PATH} = $bpc->{Conf}{MyPath}; } # @@ -539,7 +597,7 @@ sub ChildInit # sub File2MD5 { - my($self, $md5, $name) = @_; + my($bpc, $md5, $name) = @_; my($data, $fileSize); local(*N); @@ -584,7 +642,7 @@ sub File2MD5 # sub Buffer2MD5 { - my($self, $md5, $fileSize, $dataRef) = @_; + my($bpc, $md5, $fileSize, $dataRef) = @_; $md5->reset(); $md5->add($fileSize); @@ -611,16 +669,16 @@ sub Buffer2MD5 # sub MD52Path { - my($self, $d, $compress, $poolDir) = @_; + my($bpc, $d, $compress, $poolDir) = @_; return if ( $d !~ m{(.)(.)(.)(.*)} ); - $poolDir = ($compress ? $self->{CPoolDir} : $self->{PoolDir}) + $poolDir = ($compress ? $bpc->{CPoolDir} : $bpc->{PoolDir}) if ( !defined($poolDir) ); return "$poolDir/$1/$2/$3/$1$2$3$4"; } # -# For each file, check if the file exists in $self->{TopDir}/pool. +# For each file, check if the file exists in $bpc->{TopDir}/pool. # If so, remove the file and make a hardlink to the file in # the pool. Otherwise, if the newFile flag is set, make a # hardlink in the pool to the new file. @@ -633,12 +691,12 @@ sub MD52Path # sub MakeFileLink { - my($self, $name, $d, $newFile, $compress) = @_; + my($bpc, $name, $d, $newFile, $compress) = @_; my($i, $rawFile); return -1 if ( !-f $name ); for ( $i = -1 ; ; $i++ ) { - return -2 if ( !defined($rawFile = $self->MD52Path($d, $compress)) ); + return -2 if ( !defined($rawFile = $bpc->MD52Path($d, $compress)) ); $rawFile .= "_$i" if ( $i >= 0 ); if ( -f $rawFile ) { if ( !compare($name, $rawFile) ) { @@ -660,75 +718,172 @@ sub MakeFileLink sub CheckHostAlive { - my($self, $host) = @_; - my($s, $pingArgs); + my($bpc, $host) = @_; + my($s, $pingCmd, $ret); - $pingArgs = $self->{Conf}{PingArgs}; # - # Merge variables into $pingArgs + # Return success if the ping cmd is undefined or empty. # - my $vars = { - host => $host, + if ( $bpc->{Conf}{PingCmd} eq "" ) { + print("CheckHostAlive: return ok because \$Conf{PingCmd} is empty\n") + if ( $bpc->{verbose} ); + return 0; + } + + my $args = { + pingPath => $bpc->{Conf}{PingPath}, + host => $host, }; - $pingArgs =~ s/\$(\w+)/defined($vars->{$1}) - ? $self->shellEscape($vars->{$1}) - : \$$1/eg; + $pingCmd = $bpc->cmdVarSubstitute($bpc->{Conf}{PingCmd}, $args); + # # Do a first ping in case the PC needs to wakeup # - $s = `$self->{Conf}{PingPath} $pingArgs 2>&1`; - return -1 if ( $? ); + $s = $bpc->cmdSystemOrEval($pingCmd, undef, $args); + if ( $? ) { + print("CheckHostAlive: first ping failed ($?, $!)\n") + if ( $bpc->{verbose} ); + return -1; + } + # # Do a second ping and get the round-trip time in msec # - $s = `$self->{Conf}{PingPath} $pingArgs 2>&1`; - return -1 if ( $? ); - return $1 if ( $s !~ /time=([\d\.]+)\s*ms/ ); - return $1/1000 if ( $s !~ /time=([\d\.]+)\s*usec/ ); - return 0; + $s = $bpc->cmdSystemOrEval($pingCmd, undef, $args); + if ( $? ) { + print("CheckHostAlive: second ping failed ($?, $!)\n") + if ( $bpc->{verbose} ); + return -1; + } + if ( $s =~ /time=([\d\.]+)\s*ms/i ) { + $ret = $1; + } elsif ( $s =~ /time=([\d\.]+)\s*usec/i ) { + $ret = $1/1000; + } else { + print("CheckHostAlive: can't extract round-trip time (not fatal)\n") + if ( $bpc->{verbose} ); + $ret = 0; + } + print("CheckHostAlive: returning $ret\n") if ( $bpc->{verbose} ); + return $ret; } sub CheckFileSystemUsage { - my($self) = @_; - my($topDir) = $self->{TopDir}; - my($s); - - if ( $^O eq "solaris" ) { - $s = `$self->{Conf}{DfPath} -k $topDir 2>&1`; - return 0 if ( $? || $s !~ /(\d+)%/s ); - return $1; - } elsif ( $^O eq "sunos" ) { - $s = `$self->{Conf}{DfPath} $topDir 2>&1`; - return 0 if ( $? || $s !~ /(\d+)%/s ); - return $1; - } elsif ( $^O eq "linux" ) { - $s = `$self->{Conf}{DfPath} $topDir 2>&1`; - return 0 if ( $? || $s !~ /(\d+)%/s ); - return $1; - } else { - return 0; - } + my($bpc) = @_; + my($topDir) = $bpc->{TopDir}; + my($s, $dfCmd); + + return 0 if ( $bpc->{Conf}{DfCmd} eq "" ); + my $args = { + dfPath => $bpc->{Conf}{DfPath}, + topDir => $bpc->{TopDir}, + }; + $dfCmd = $bpc->cmdVarSubstitute($bpc->{Conf}{DfCmd}, $args); + $s = $bpc->cmdSystemOrEval($dfCmd, undef, $args); + return 0 if ( $? || $s !~ /(\d+)%/s ); + return $1; } +# +# Given an IP address, return the host name and user name via +# NetBios. +# sub NetBiosInfoGet { - my($self, $host) = @_; + my($bpc, $host) = @_; my($netBiosHostName, $netBiosUserName); + my($s, $nmbCmd); - foreach ( split(/[\n\r]+/, `$self->{Conf}{NmbLookupPath} -A $host 2>&1`) ) { - next if ( !/([\w-]+)\s*<(\w{2})\> - .*/i ); + # + # Skip NetBios check if NmbLookupCmd is emtpy + # + if ( $bpc->{Conf}{NmbLookupCmd} eq "" ) { + print("NetBiosInfoGet: return $host because \$Conf{NmbLookupCmd}" + . " is empty\n") + if ( $bpc->{verbose} ); + return ($host, undef); + } + + my $args = { + nmbLookupPath => $bpc->{Conf}{NmbLookupPath}, + host => $host, + }; + $nmbCmd = $bpc->cmdVarSubstitute($bpc->{Conf}{NmbLookupCmd}, $args); + foreach ( split(/[\n\r]+/, $bpc->cmdSystemOrEval($nmbCmd, undef, $args)) ) { + next if ( !/^\s*([\w\s-]+?)\s*<(\w{2})\> - .*/i ); $netBiosHostName ||= $1 if ( $2 eq "00" ); # host is first 00 $netBiosUserName = $1 if ( $2 eq "03" ); # user is last 03 } - return if ( !defined($netBiosHostName) ); - return (lc($netBiosHostName), lc($netBiosUserName)); + if ( !defined($netBiosHostName) ) { + print("NetBiosInfoGet: failed: can't parse return string\n") + if ( $bpc->{verbose} ); + return; + } + $netBiosHostName = lc($netBiosHostName); + $netBiosUserName = lc($netBiosUserName); + print("NetBiosInfoGet: success, returning host $netBiosHostName," + . " user $netBiosUserName\n") + if ( $bpc->{verbose} ); + return ($netBiosHostName, $netBiosUserName); +} + +# +# Given a NetBios name lookup the IP address via NetBios. +# In the case of a host returning multiple interfaces we +# return the first IP address that matches the subnet mask. +# If none match the subnet mask (or nmblookup doesn't print +# the subnet mask) then just the first IP address is returned. +# +sub NetBiosHostIPFind +{ + my($bpc, $host) = @_; + my($netBiosHostName, $netBiosUserName); + my($s, $nmbCmd, $subnet, $ipAddr, $firstIpAddr); + + # + # Skip NetBios lookup if NmbLookupFindHostCmd is emtpy + # + if ( $bpc->{Conf}{NmbLookupFindHostCmd} eq "" ) { + print("NetBiosHostIPFind: return $host because" + . " \$Conf{NmbLookupFindHostCmd} is empty\n") + if ( $bpc->{verbose} ); + return $host; + } + + my $args = { + nmbLookupPath => $bpc->{Conf}{NmbLookupPath}, + host => $host, + }; + $nmbCmd = $bpc->cmdVarSubstitute($bpc->{Conf}{NmbLookupFindHostCmd}, $args); + foreach my $resp ( split(/[\n\r]+/, $bpc->cmdSystemOrEval($nmbCmd, undef, + $args) ) ) { + if ( $resp =~ /querying\s+\Q$host\E\s+on\s+(\d+\.\d+\.\d+\.\d+)/i ) { + $subnet = $1; + $subnet = $1 if ( $subnet =~ /^(.*?)(\.255)+$/ ); + } elsif ( $resp =~ /^\s*(\d+\.\d+\.\d+\.\d+)\s+\Q$host/ ) { + my $ip = $1; + $firstIpAddr = $ip if ( !defined($firstIpAddr) ); + $ipAddr = $ip if ( !defined($ipAddr) && $ip =~ /^\Q$subnet/ ); + } + } + $ipAddr = $firstIpAddr if ( !defined($ipAddr) ); + if ( defined($ipAddr) ) { + print("NetBiosHostIPFind: found IP address $ipAddr for host $host\n") + if ( $bpc->{verbose} ); + return $ipAddr; + } else { + print("NetBiosHostIPFind: couldn't find IP address for host $host\n") + if ( $bpc->{verbose} ); + return; + } } sub fileNameEltMangle { - my($self, $name) = @_; + my($bpc, $name) = @_; + return "" if ( $name eq "" ); $name =~ s{([%/\n\r])}{sprintf("%%%02x", ord($1))}eg; return "f$name"; } @@ -742,10 +897,10 @@ sub fileNameEltMangle # sub fileNameMangle { - my($self, $name) = @_; + my($bpc, $name) = @_; - $name =~ s{/([^/]+)}{"/" . $self->fileNameEltMangle($1)}eg; - $name =~ s{^([^/]+)}{$self->fileNameEltMangle($1)}eg; + $name =~ s{/([^/]+)}{"/" . $bpc->fileNameEltMangle($1)}eg; + $name =~ s{^([^/]+)}{$bpc->fileNameEltMangle($1)}eg; return $name; } @@ -754,7 +909,7 @@ sub fileNameMangle # sub fileNameUnmangle { - my($self, $name) = @_; + my($bpc, $name) = @_; $name =~ s{/f}{/}g; $name =~ s{^f}{}; @@ -769,10 +924,203 @@ sub fileNameUnmangle # sub shellEscape { - my($self, $cmd) = @_; + my($bpc, $cmd) = @_; $cmd =~ s/([][;&()<>{}|^\n\r\t *\$\\'"`?])/\\$1/g; return $cmd; } +# +# For printing exec commands (which don't use a shell) so they look like +# a valid shell command this function should be called with the exec +# args. The shell command string is returned. +# +sub execCmd2ShellCmd +{ + my($bpc, @args) = @_; + my $str; + + foreach my $a ( @args ) { + $str .= " " if ( $str ne "" ); + $str .= $bpc->shellEscape($a); + } + return $str; +} + +# +# Do a URI-style escape to protect/encode special characters +# +sub uriEsc +{ + my($bpc, $s) = @_; + $s =~ s{([^\w.\/-])}{sprintf("%%%02X", ord($1));}eg; + return $s; +} + +# +# Do a URI-style unescape to restore special characters +# +sub uriUnesc +{ + my($bpc, $s) = @_; + $s =~ s{%(..)}{chr(hex($1))}eg; + return $s; +} + +# +# Do variable substitution prior to execution of a command. +# +sub cmdVarSubstitute +{ + my($bpc, $template, $vars) = @_; + my(@cmd); + + # + # Return without any substitution if the first entry starts with "&", + # indicating this is perl code. + # + if ( (ref($template) eq "ARRAY" ? $template->[0] : $template) =~ /^\&/ ) { + return $template; + } + if ( ref($template) ne "ARRAY" ) { + # + # Split at white space, except if escaped by \ + # + $template = [split(/(?{$1}) && ref($vars->{$1}) ne "ARRAY" + ? ($2 eq "+" ? $bpc->shellEscape($vars->{$1}) : $vars->{$1}) + : "\$$1" + }eg; + # + # Now replicate any array arguments; this just works for just one + # array var in each argument. + # + if ( $arg =~ m{(.*)\$(\w+)(\+?)(.*)} && ref($vars->{$2}) eq "ARRAY" ) { + my $pre = $1; + my $var = $2; + my $esc = $3; + my $post = $4; + foreach my $v ( @{$vars->{$var}} ) { + $v = $bpc->shellEscape($v) if ( $esc eq "+" ); + push(@cmd, "$pre$v$post"); + } + } else { + push(@cmd, $arg); + } + } + return \@cmd; +} + +# +# Exec or eval a command. $cmd is either a string on an array ref. +# +# @args are optional arguments for the eval() case; they are not used +# for exec(). +# +sub cmdExecOrEval +{ + my($bpc, $cmd, @args) = @_; + + if ( (ref($cmd) eq "ARRAY" ? $cmd->[0] : $cmd) =~ /^\&/ ) { + $cmd = join(" ", $cmd) if ( ref($cmd) eq "ARRAY" ); + print("cmdExecOrEval: about to eval perl code $cmd\n") + if ( $bpc->{verbose} ); + eval($cmd); + print(STDERR "Perl code fragment for exec shouldn't return!!\n"); + exit(1); + } else { + $cmd = [split(/\s+/, $cmd)] if ( ref($cmd) ne "ARRAY" ); + print("cmdExecOrEval: about to exec ", + $bpc->execCmd2ShellCmd(@$cmd), "\n") + if ( $bpc->{verbose} ); + exec(@$cmd); + print(STDERR "Exec failed for @$cmd\n"); + exit(1); + } +} + +# +# System or eval a command. $cmd is either a string on an array ref. +# $stdoutCB is a callback for output generated by the command. If it +# is undef then output is returned. If it is a code ref then the function +# is called with each piece of output as an argument. If it is a scalar +# ref the output is appended to this variable. +# +# @args are optional arguments for the eval() case; they are not used +# for system(). +# +# Also, $? should be set when the CHILD pipe is closed. +# +sub cmdSystemOrEval +{ + my($bpc, $cmd, $stdoutCB, @args) = @_; + my($pid, $out, $allOut); + local(*CHILD); + + if ( (ref($cmd) eq "ARRAY" ? $cmd->[0] : $cmd) =~ /^\&/ ) { + $cmd = join(" ", $cmd) if ( ref($cmd) eq "ARRAY" ); + print("cmdSystemOrEval: about to eval perl code $cmd\n") + if ( $bpc->{verbose} ); + $out = eval($cmd); + $$stdoutCB .= $out if ( ref($stdoutCB) eq 'SCALAR' ); + &$stdoutCB($out) if ( ref($stdoutCB) eq 'CODE' ); + print("cmdSystemOrEval: finished: got output $out\n") + if ( $bpc->{verbose} ); + return $out if ( !defined($stdoutCB) ); + return; + } else { + $cmd = [split(/\s+/, $cmd)] if ( ref($cmd) ne "ARRAY" ); + print("cmdSystemOrEval: about to system ", + $bpc->execCmd2ShellCmd(@$cmd), "\n") + if ( $bpc->{verbose} ); + if ( !defined($pid = open(CHILD, "-|")) ) { + my $err = "Can't fork to run @$cmd\n"; + $? = 1; + $$stdoutCB .= $err if ( ref($stdoutCB) eq 'SCALAR' ); + &$stdoutCB($err) if ( ref($stdoutCB) eq 'CODE' ); + return $err if ( !defined($stdoutCB) ); + return; + } + if ( !$pid ) { + # + # This is the child + # + close(STDERR); + open(STDERR, ">&STDOUT"); + exec(@$cmd); + print("Exec of @$cmd failed\n"); + exit(1); + } + # + # The parent gathers the output from the child + # + while ( ) { + $$stdoutCB .= $_ if ( ref($stdoutCB) eq 'SCALAR' ); + &$stdoutCB($_) if ( ref($stdoutCB) eq 'CODE' ); + $out .= $_ if ( !defined($stdoutCB) ); + $allOut .= $_ if ( $bpc->{verbose} ); + } + $? = 0; + close(CHILD); + } + print("cmdSystemOrEval: finished: got output $allOut\n") + if ( $bpc->{verbose} ); + return $out; +} + 1;