f9372fd4894e34db7d79eeb1df117788a4b5f668
[BackupPC.git] / bin / BackupPC_ASA_ArchiveStart
1 #!/usr/bin/perl
2 #============================================================= -*-perl-*-
3 #
4 # BackupPC_archiveStart: start an archive request from the
5 # command line.
6 #
7 # DESCRIPTION
8 #  
9 #   Usage: BackupPC_archiveStart archiveHost userName hosts...
10 #
11 #   Initiates an archive request on archive host archiveHost
12 #   for the listed hosts.  The latest backup for each host is
13 #   archived.  The userName is name of the requesting user,
14 #   which appears in the log files.
15 #
16 # AUTHOR
17 #   Craig Barratt  <cbarratt@users.sourceforge.net>
18 #   Dobrica Pavlinusic <dpavlin@rot13.org>
19 #
20 # COPYRIGHT
21 #   Copyright (C) 2007-2009  Craig Barratt
22 #
23 #   This program is free software; you can redistribute it and/or modify
24 #   it under the terms of the GNU General Public License as published by
25 #   the Free Software Foundation; either version 2 of the License, or
26 #   (at your option) any later version.
27 #
28 #   This program is distributed in the hope that it will be useful,
29 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
30 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
31 #   GNU General Public License for more details.
32 #
33 #   You should have received a copy of the GNU General Public License
34 #   along with this program; if not, write to the Free Software
35 #   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
36 #
37 #========================================================================
38 #
39 # Version 3.2.0, released 31 Jul 2010.
40 #
41 # See http://backuppc.sourceforge.net.
42 #
43 #========================================================================
44
45 use strict;
46 no  utf8;
47 use lib "/usr/local/BackupPC/lib";
48 use Getopt::Std;
49 use BackupPC::Lib;
50
51 use DBI;
52 use Data::Dump qw(dump);
53
54 my $debug = $ENV{DEBUG} || 0;
55
56 die("BackupPC::Lib->new failed\n") if ( !(my $bpc = BackupPC::Lib->new) );
57
58 my %opts;
59
60 # no options currently
61 if ( !getopts("", \%opts) || @ARGV < 2 ) {
62     print STDERR <<EOF;
63 usage: $0 archiveHost userName [hosts[:num]...]
64 EOF
65     exit(1);
66 }
67
68 my %Conf = $bpc->Conf();
69
70 my $dbh = DBI->connect($Conf{SearchDSN}, $Conf{SearchUser}, "", { RaiseError => 1, AutoCommit => 0 });
71
72 my $sth = $dbh->prepare(qq{
73         select
74                 backups.id as backup_id,
75                 hosts.name as host,
76                 backups.num as num,
77                 inc_size,
78                 size,
79                 inc_deleted
80         from backups
81         join hosts on hosts.id = hostid
82         where hosts.name = ?
83 });
84
85 my $Hosts       = $bpc->HostInfoRead();
86 my $ArchiveHost = shift @ARGV;
87 my $UserName    = shift @ARGV;
88 my @HostFilter  = @ARGV;
89 my $TopDir      = $bpc->{Conf}{TopDir};
90
91 if ( !defined($Hosts->{$ArchiveHost}) ) {
92     print(STDERR "$0: archive host $ArchiveHost doesn't exist... quitting\n");
93     exit(1);
94 }
95 $bpc->ConfigRead($ArchiveHost);
96
97 if ( ! @HostFilter ) {
98         @HostFilter = keys %$Hosts;
99 }
100
101 warn "archiving hosts ",dump(@HostFilter);
102
103 my(@HostList, @BackupList);
104 foreach ( @HostFilter ) {
105     my ($host,$num) = split(/:/,$_,2);
106     if ( !defined($Hosts->{$host}) ) {
107         print(STDERR "$0: host $host doesn't exist... quitting\n");
108         exit(1);
109     }
110     my @backups = $bpc->BackupInfoRead($host);
111     if ( !@backups ) {
112         warn "$0: host $host doesn't have any backups... skipping\n";
113         next;
114     }
115
116         my $all_backup_numbers;
117         $all_backup_numbers->{ $_->{num} }++ foreach @backups;
118
119         $sth->execute( $host );
120         while ( my $row = $sth->fetchrow_hashref ) {
121                 warn "# row ",dump($row) if $debug;
122                 $all_backup_numbers->{ $row->{num} } =
123                 $row->{inc_deleted}  ? 0 :
124                 $row->{size}    == 0 ? 0 :
125                 $row->{inc_size} > 0 ? 0 :
126                 $row->{size}     > 0 ? 1 :
127                 0;
128         }
129
130         warn "# $host all_backup_numbers = ",dump($all_backup_numbers) if $debug;
131
132         my @backup_nums = 
133                 sort
134                 grep { $all_backup_numbers->{$_} }
135                 keys %$all_backup_numbers;
136
137         foreach my $num ( @backup_nums ) {
138                 warn "+ $host $num\n";
139                 push(@HostList, $host);
140                 push(@BackupList, $num);
141         }
142 }
143
144 $dbh->disconnect;
145
146 my $ReqFileName;
147 for ( my $i = 0 ; ; $i++ ) {
148     $ReqFileName="archiveReq.$$.$i";
149     last if ( !-f "$TopDir/pc/$ArchiveHost/$ReqFileName" );
150 }
151 my %ArchiveReq = (
152     archiveloc  => $bpc->{Conf}{ArchiveDest},
153     archtype    => 0,
154     compression => $bpc->{Conf}{ArchiveComp} eq 'none' ? $bpc->{Conf}{CatPath}
155                     : ($bpc->{Conf}{ArchiveComp} eq 'gzip'
156                       ? $bpc->{Conf}{GzipPath} : $bpc->{Conf}{Bzip2Path}),
157     compext     => $bpc->{Conf}{ArchiveComp} eq 'none' ? ''
158                     : ($bpc->{Conf}{ArchiveComp} eq 'gzip' ? '.gz' : '.bz2'),
159     parfile     => $bpc->{Conf}{ArchivePar},
160     splitsize   => $bpc->{Conf}{ArchiveSplit} . '000000', # mb -> bytes
161     host        => $ArchiveHost,
162     HostList    => \@HostList,
163     BackupList  => \@BackupList,
164     user        => $UserName,
165     reqTime     => time,
166 );
167 my $archive = Data::Dumper->new([\%ArchiveReq], [qw(*ArchiveReq)]);
168 $archive->Indent(1);
169 if ( !open(REQ, ">", "$TopDir/pc/$ArchiveHost/$ReqFileName") ) {
170     print(STDERR "$0: can't open/write request file $TopDir/pc/$ArchiveHost/$ReqFileName... quitting\n");
171     exit(1);
172 }
173 binmode(REQ);
174 print REQ $archive->Dump;
175 close(REQ);
176 $bpc->ServerConnect($bpc->{Conf}{ServerHost}, $bpc->{Conf}{ServerPort});
177 my $reply = $bpc->ServerMesg("archive $UserName $ArchiveHost $ReqFileName");
178 $bpc->ServerDisconnect();
179 print("Sent archive request, reply: $reply\n");
180 exit(0);