2 #============================================================= -*-perl-*-
4 # BackupPC_zipCreate: create a zip archive of an existing dump
5 # for restore on a client.
9 # Usage: BackupPC_zipCreate [-t] [-h host] [-n dumpNum] [-s shareName]
10 # [-r pathRemove] [-p pathAdd] [-c compressionLevel]
11 # files/directories...
16 # -h host host from which the zip archive is created
17 # -n dumpNum dump number from which the zip archive is created
18 # -s shareName share name from which the zip archive is created
21 # -t print summary totals
22 # -r pathRemove path prefix that will be replaced with pathAdd
23 # -p pathAdd new path prefix
24 # -c level compression level (default is 0, no compression)
26 # The -h, -n and -s options specify which dump is used to generate
27 # the zip archive. The -r and -p options can be used to relocate
28 # the paths in the zip archive so extracted files can be placed
29 # in a location different from their original location.
32 # Guillaume Filion <gfk@users.sourceforge.net>
33 # Based on Backup_tarCreate by Craig Barratt <cbarratt@users.sourceforge.net>
36 # Copyright (C) 2002 Craig Barratt and Guillaume Filion
38 # This program is free software; you can redistribute it and/or modify
39 # it under the terms of the GNU General Public License as published by
40 # the Free Software Foundation; either version 2 of the License, or
41 # (at your option) any later version.
43 # This program is distributed in the hope that it will be useful,
44 # but WITHOUT ANY WARRANTY; without even the implied warranty of
45 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
46 # GNU General Public License for more details.
48 # You should have received a copy of the GNU General Public License
49 # along with this program; if not, write to the Free Software
50 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
52 #========================================================================
54 # Version 2.0.0beta2, released 13 Apr 2003.
56 # See http://backuppc.sourceforge.net.
58 #========================================================================
61 use lib "/usr/local/BackupPC/lib";
62 use Archive::Zip qw(:ERROR_CODES);
67 use BackupPC::Attrib qw(:all);
68 use BackupPC::FileZIO;
69 use BackupPC::Zip::FileMember;
72 die("BackupPC::Lib->new failed\n") if ( !(my $bpc = BackupPC::Lib->new) );
73 my $TopDir = $bpc->TopDir();
74 my $BinDir = $bpc->BinDir();
75 my %Conf = $bpc->Conf();
79 if ( !getopts("th:n:p:r:s:c:", \%opts) || @ARGV < 1 ) {
80 print(STDERR "usage: $0 [-t] [-h host] [-n dumpNum] [-s shareName]"
81 . " [-r pathRemove] [-p pathAdd] [-c compressionLevel]"
82 . " files/directories...\n");
86 if ( $opts{h} !~ /^([\w\.\s-]+)$/ ) {
87 print(STDERR "$0: bad host name '$opts{h}'\n");
92 if ( $opts{n} !~ /^(\d+)$/ ) {
93 print(STDERR "$0: bad dump number '$opts{n}'\n");
97 $opts{c} = 0 if ( $opts{c} eq "" );
98 if ( $opts{c} !~ /^(\d+)$/ ) {
99 print(STDERR "$0: invalid compression level '$opts{c}'. 0=none, 9=max\n");
102 my $compLevel = $opts{c};
104 my @Backups = $bpc->BackupInfoRead($Host);
112 for ( $i = 0 ; $i < @Backups ; $i++ ) {
113 last if ( $Backups[$i]{num} == $Num );
115 if ( $i >= @Backups ) {
116 print(STDERR "$0: bad backup number $Num for host $Host\n");
120 my $PathRemove = $1 if ( $opts{r} =~ /(.+)/ );
121 my $PathAdd = $1 if ( $opts{p} =~ /(.+)/ );
122 if ( $opts{s} !~ /^([\w\s\.\/\$-]+)$/ ) {
123 print(STDERR "$0: bad share name '$opts{s}'\n");
126 my $ShareName = $opts{s};
128 my $BufSize = 1048576; # 1MB or 2^20
129 my(%UidCache, %GidCache);
131 my $fh = new IO::Handle;
132 $fh->fdopen(fileno(STDOUT),"w");
133 my $zipfh = Archive::Zip->new();
135 foreach my $dir ( @ARGV ) {
136 archiveWrite($zipfh, $dir);
141 my($zipfh, $dir, $zipPathOverride) = @_;
143 my $view = BackupPC::View->new($bpc, $Host, \@Backups);
145 if ( $dir =~ m{(^|/)\.\.(/|$)} || $dir !~ /^(.*)$/ ) {
146 print(STDERR "$0: bad directory '$dir'\n");
150 $view->find($Num, $ShareName, $dir, 0, \&ZipWriteFile,
151 $zipfh, $zipPathOverride);
155 print STDERR "Can't write Zip file\n"
156 unless $zipfh->writeToFileHandle($fh, 0) == Archive::Zip::AZ_OK;
159 # print out totals if requested
162 print STDERR "Done: $FileCnt files, $ByteCnt bytes, $DirCnt dirs,",
163 " $SpecialCnt specials ignored, $ErrorCnt errors\n";
167 ###########################################################################
169 ###########################################################################
175 $UidCache{$uid} = (getpwuid($uid))[0] if ( !exists($UidCache{$uid}) );
176 return $UidCache{$uid};
183 $GidCache{$gid} = (getgrgid($gid))[0] if ( !exists($GidCache{$gid}) );
184 return $GidCache{$gid};
192 my($hdr, $zipfh, $zipPathOverride) = @_;
194 my $tarPath = $hdr->{relPath};
195 $tarPath = $zipPathOverride if ( defined($zipPathOverride) );
197 if ( defined($PathRemove)
198 && substr($tarPath, 0, length($PathRemove)) eq $PathRemove ) {
199 substr($tarPath, 0, length($PathRemove)) = $PathAdd;
201 $tarPath = "./" . $tarPath if ( $tarPath !~ /^\.\// );
202 $tarPath =~ s{//+}{/}g;
203 $hdr->{name} = $tarPath;
205 my $zipmember; # Container to hold the file/directory to zip.
207 if ( $hdr->{type} == BPC_FTYPE_DIR ) {
209 # Directory: just write the header
211 $hdr->{name} .= "/" if ( $hdr->{name} !~ m{/$} );
212 $zipmember = Archive::Zip::Member->newDirectoryNamed($hdr->{name});
214 } elsif ( $hdr->{type} == BPC_FTYPE_FILE ) {
216 # Regular file: write the header and file
218 $zipmember = BackupPC::Zip::FileMember->newFromFileNamed(
225 $ByteCnt += $hdr->{size};
226 } elsif ( $hdr->{type} == BPC_FTYPE_HARDLINK ) {
228 # Hardlink file: not supported by Zip, so just make a copy
229 # of the pointed-to file.
231 # Start by reading the contents of the link.
233 my $f = BackupPC::FileZIO->open($hdr->{fullPath}, 0, $hdr->{compress});
234 if ( !defined($f) ) {
235 print(STDERR "Unable to open file $hdr->{fullPath}\n");
240 while ( $f->read(\$data, $BufSize) > 0 ) {
241 $hdr->{linkname} .= $data;
245 # Dump the original file. Just call the top-level
246 # routine, so that we save the hassle of dealing with
247 # mangling, merging and attributes.
249 archiveWrite($zipfh, $hdr->{linkname}, $hdr->{name});
250 } elsif ( $hdr->{type} == BPC_FTYPE_SYMLINK ) {
252 # Symlinks can't be Zipped. 8(
253 # We could zip the pointed-to dir/file (just like hardlink), but we
254 # have to avoid the infinite-loop case of a symlink pointed to a
255 # directory above us. Ignore for now. Could be a comand-line
259 } elsif ( $hdr->{type} == BPC_FTYPE_CHARDEV
260 || $hdr->{type} == BPC_FTYPE_BLOCKDEV
261 || $hdr->{type} == BPC_FTYPE_FIFO ) {
263 # Special files can't be Zipped. 8(
267 print(STDERR "Got unknown type $hdr->{type} for $hdr->{name}\n");
270 return if ( !$zipmember );
272 # Set the attributes and permissions
273 $zipmember->setLastModFileDateTimeFromUnix($hdr->{mtime});
274 $zipmember->unixFileAttributes($hdr->{mode});
275 # Zip files don't accept uid and gid, so we put them in the comment field.
276 $zipmember->fileComment("uid=".$hdr->{uid}." gid=".$hdr->{gid})
277 if ( $hdr->{uid} || $hdr->{gid} );
279 # Specify the compression level for this member
280 $zipmember->desiredCompressionLevel($compLevel) if ($compLevel =~ /[0-9]/);
282 # Finally Zip the member
283 $zipfh->addMember($zipmember);