/[BackupPC]/upstream/2.1.0/lib/BackupPC/Xfer/Smb.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Annotation of /upstream/2.1.0/lib/BackupPC/Xfer/Smb.pm

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1 - (hide annotations)
Wed Jun 22 19:12:04 2005 UTC (18 years, 11 months ago) by dpavlin
File size: 12583 byte(s)
import of version 2.1.0

1 dpavlin 1 #============================================================= -*-perl-*-
2     #
3     # BackupPC::Xfer::Smb package
4     #
5     # DESCRIPTION
6     #
7     # This library defines a BackupPC::Xfer::Smb class for managing
8     # the SMB (smbclient) transport of backup data from the client.
9     #
10     # AUTHOR
11     # Craig Barratt <cbarratt@users.sourceforge.net>
12     #
13     # COPYRIGHT
14     # Copyright (C) 2001-2003 Craig Barratt
15     #
16     # This program is free software; you can redistribute it and/or modify
17     # it under the terms of the GNU General Public License as published by
18     # the Free Software Foundation; either version 2 of the License, or
19     # (at your option) any later version.
20     #
21     # This program is distributed in the hope that it will be useful,
22     # but WITHOUT ANY WARRANTY; without even the implied warranty of
23     # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24     # GNU General Public License for more details.
25     #
26     # You should have received a copy of the GNU General Public License
27     # along with this program; if not, write to the Free Software
28     # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
29     #
30     #========================================================================
31     #
32     # Version 2.1.0, released 20 Jun 2004.
33     #
34     # See http://backuppc.sourceforge.net.
35     #
36     #========================================================================
37    
38     package BackupPC::Xfer::Smb;
39    
40     use strict;
41    
42     sub new
43     {
44     my($class, $bpc, $args) = @_;
45    
46     $args ||= {};
47     my $t = bless {
48     bpc => $bpc,
49     conf => { $bpc->Conf },
50     host => "",
51     hostIP => "",
52     shareName => "",
53     pipeRH => undef,
54     pipeWH => undef,
55     badFiles => [],
56     %$args,
57     }, $class;
58    
59     return $t;
60     }
61    
62     sub args
63     {
64     my($t, $args) = @_;
65    
66     foreach my $arg ( keys(%$args) ) {
67     $t->{$arg} = $args->{$arg};
68     }
69     }
70    
71     sub useTar
72     {
73     return 1;
74     }
75    
76     sub start
77     {
78     my($t) = @_;
79     my $bpc = $t->{bpc};
80     my $conf = $t->{conf};
81     my $I_option = $t->{hostIP} eq $t->{host} ? [] : ['-I', $t->{hostIP}];
82     my(@fileList, $X_option, $smbClientCmd, $logMsg);
83     my($timeStampFile);
84     local(*SMB);
85    
86     #
87     # First propagate the PASSWD setting
88     #
89     $ENV{PASSWD} = $ENV{BPC_SMB_PASSWD} if ( defined($ENV{BPC_SMB_PASSWD}) );
90     $ENV{PASSWD} = $conf->{SmbSharePasswd}
91     if ( defined($conf->{SmbSharePasswd}) );
92     if ( !defined($ENV{PASSWD}) ) {
93     $t->{_errStr} = "passwd not set for smbclient";
94     return;
95     }
96     if ( !defined($conf->{SmbClientPath}) || !-x $conf->{SmbClientPath} ) {
97     $t->{_errStr} = '$Conf{SmbClientPath} is not a valid executable';
98     return;
99     }
100     if ( $t->{type} eq "restore" ) {
101     $smbClientCmd = $conf->{SmbClientRestoreCmd};
102     $logMsg = "restore started for share $t->{shareName}";
103     } else {
104     #
105     # Turn $conf->{BackupFilesOnly} and $conf->{BackupFilesExclude}
106     # into a hash of arrays of files, and $conf->{SmbShareName}
107     # to an array
108     #
109     $bpc->backupFileConfFix($conf, "SmbShareName");
110    
111     $t->{fileIncludeHash} = {};
112     if ( defined($conf->{BackupFilesOnly}{$t->{shareName}}) ) {
113     foreach my $file ( @{$conf->{BackupFilesOnly}{$t->{shareName}}} ) {
114     push(@fileList, $file);
115     $t->{fileIncludeHash}{$file} = 1;
116     }
117     } elsif ( defined($conf->{BackupFilesExclude}{$t->{shareName}}) ) {
118     foreach my $file ( @{$conf->{BackupFilesExclude}{$t->{shareName}}} )
119     {
120     push(@fileList, $file);
121     }
122     #
123     # Allow simple wildcards in exclude list by specifying "r" option.
124     #
125     $X_option = "rX";
126     }
127     if ( $t->{type} eq "full" ) {
128     $smbClientCmd = $conf->{SmbClientFullCmd};
129     $logMsg = "full backup started for share $t->{shareName}";
130     } else {
131     $timeStampFile = "$t->{outDir}/timeStamp.level0";
132     open(LEV0, ">", $timeStampFile) && close(LEV0);
133     utime($t->{lastFull} - 3600, $t->{lastFull} - 3600, $timeStampFile);
134     $smbClientCmd = $conf->{SmbClientIncrCmd};
135     $logMsg = "incr backup started back to "
136     . $bpc->timeStamp($t->{lastFull} - 3600, 0)
137     . "for share $t->{shareName}";
138     }
139     }
140     my $args = {
141     smbClientPath => $conf->{SmbClientPath},
142     host => $t->{host},
143     hostIP => $t->{hostIP},
144     client => $t->{client},
145     shareName => $t->{shareName},
146     userName => $conf->{SmbShareUserName},
147     fileList => \@fileList,
148     I_option => $I_option,
149     X_option => $X_option,
150     timeStampFile => $timeStampFile,
151     };
152     $smbClientCmd = $bpc->cmdVarSubstitute($smbClientCmd, $args);
153    
154     if ( !defined($t->{xferPid} = open(SMB, "-|")) ) {
155     $t->{_errStr} = "Can't fork to run smbclient";
156     return;
157     }
158     $t->{pipeSMB} = *SMB;
159     if ( !$t->{xferPid} ) {
160     #
161     # This is the smbclient child.
162     #
163     setpgrp 0,0;
164     if ( $t->{type} eq "restore" ) {
165     #
166     # For restores close the write end of the pipe,
167     # clone STDIN from RH, and STDERR to STDOUT
168     #
169     close($t->{pipeWH});
170     close(STDERR);
171     open(STDERR, ">&STDOUT");
172     close(STDIN);
173     open(STDIN, "<&$t->{pipeRH}");
174     } else {
175     #
176     # For backups close the read end of the pipe,
177     # clone STDOUT to WH, STDERR to STDOUT
178     #
179     close($t->{pipeRH});
180     close(STDERR);
181     open(STDERR, ">&STDOUT");
182     open(STDOUT, ">&$t->{pipeWH}");
183     }
184     #
185     # Run smbclient.
186     #
187     alarm(0);
188     $bpc->cmdExecOrEval($smbClientCmd, $args);
189     # should not be reached, but just in case...
190     $t->{_errStr} = "Can't exec $conf->{SmbClientPath}";
191     return;
192     }
193     my $str = "Running: " . $bpc->execCmd2ShellCmd(@$smbClientCmd) . "\n";
194     $t->{XferLOG}->write(\$str);
195     alarm($conf->{ClientTimeout});
196     $t->{_errStr} = undef;
197     return $logMsg;
198     }
199    
200     sub readOutput
201     {
202     my($t, $FDreadRef, $rout) = @_;
203     my $conf = $t->{conf};
204    
205     if ( vec($rout, fileno($t->{pipeSMB}), 1) ) {
206     my $mesg;
207     if ( sysread($t->{pipeSMB}, $mesg, 8192) <= 0 ) {
208     vec($$FDreadRef, fileno($t->{pipeSMB}), 1) = 0;
209     close($t->{pipeSMB});
210     } else {
211     $t->{smbOut} .= $mesg;
212     }
213     }
214     while ( $t->{smbOut} =~ /(.*?)[\n\r]+(.*)/s ) {
215     $_ = $1;
216     $t->{smbOut} = $2;
217     #
218     # ignore the log file time stamps from smbclient introduced
219     # in version 3.0.0 - don't even write them to the log file.
220     #
221     if ( m{^\[\d+/\d+/\d+ +\d+:\d+:\d+.*\] +(client/cli|lib/util_unistr).*\(\d+\)} ) {
222     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 5 );
223     next;
224     }
225     #
226     # refresh our inactivity alarm
227     #
228     alarm($conf->{ClientTimeout}) if ( !$t->{abort} );
229     $t->{lastOutputLine} = $_ if ( !/^$/ );
230     #
231     # This section is highly dependent on the version of smbclient.
232     # If you upgrade Samba, make sure that these regexp are still valid.
233     #
234     if ( /^\s*(-?\d+) \(\s*\d+[.,]\d kb\/s\) (.*)$/ ) {
235     my $sambaFileSize = $1;
236     my $pcFileName = $2;
237     (my $fileName = $pcFileName) =~ s/\\/\//g;
238     $sambaFileSize += 1024 * 1024 * 4096 if ( $sambaFileSize < 0 );
239     $fileName =~ s/^\/*//;
240     $t->{byteCnt} += $sambaFileSize;
241     $t->{fileCnt}++;
242     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 2 );
243     } elsif ( /restore tar file (.*) of size (\d+) bytes/ ) {
244     $t->{byteCnt} += $2;
245     $t->{fileCnt}++;
246     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 1 );
247     } elsif ( /^\s*tar: dumped \d+ files/ ) {
248     $t->{xferOK} = 1;
249     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 0 );
250     } elsif ( /^\s*tar: restored \d+ files/ ) {
251     $t->{xferOK} = 1;
252     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 0 );
253     } elsif ( /^\s*read_socket_with_timeout: timeout read. /i ) {
254     $t->{hostAbort} = 1;
255     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 0 );
256     } elsif ( /^code 0 listing /
257     || /^\s*code 0 opening /
258     || /^\s*abandoning restore/i
259     || /^\s*Error: Looping in FIND_NEXT/i
260     || /^\s*SUCCESS - 0/i
261     || /^\s*Call timed out: server did not respond/i
262     || /^\s*tree connect failed: ERRDOS - ERRnoaccess \(Access denied\.\)/
263     || /^\s*tree connect failed: NT_STATUS_BAD_NETWORK_NAME/
264     ) {
265     if ( $t->{hostError} eq "" ) {
266     $t->{XferLOG}->write(\"This backup will fail because: $_\n");
267     $t->{hostError} = $_;
268     }
269     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 0 );
270     } elsif ( /^\s*NT_STATUS_ACCESS_DENIED listing (.*)/
271     || /^\s*ERRDOS - ERRnoaccess \(Access denied\.\) listing (.*)/ ) {
272     my $badDir = $1;
273     $badDir =~ s{\\}{/}g;
274     $badDir =~ s{/+}{/}g;
275     $badDir =~ s{/\*$}{};
276     if ( $t->{hostError} eq ""
277     && ($badDir eq "" || $t->{fileIncludeHash}{$badDir}) ) {
278     $t->{XferLOG}->write(\"This backup will fail because: $_\n");
279     $t->{hostError} ||= $_;
280     }
281     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 0 );
282     } elsif ( /^\s*directory \\/i ) {
283     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 2 );
284     } elsif ( /smb: \\>/
285     || /^\s*added interface/i
286     || /^\s*tarmode is now/i
287     || /^\s*Total bytes written/i
288     || /^\s*Domain=/i
289     || /^\([\d\.]* kb\/s\) \(average [\d\.]* kb\/s\)$/i
290     || /^\s*Getting files newer than/i
291     || /^\s*restore directory \\/i
292     || /^\s*Output is \/dev\/null/i
293     || /^\s*Timezone is/i
294     || /^\s*tar_re_search set/i
295     || /^\s*creating lame (up|low)case table/i
296     ) {
297     # ignore these messages
298     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 1 );
299     } else {
300     $t->{xferErrCnt}++;
301     $t->{xferBadShareCnt}++ if ( /^ERRDOS - ERRbadshare/ );
302     $t->{xferBadFileCnt}++ if ( /^ERRDOS - ERRbadfile/ );
303     if ( $t->{xferErrCnt} > 50000 ) {
304     $t->logMsg(
305     "Too many smbtar errors ($t->{xferErrCnt})... giving up");
306     $t->{hostError} = "Too many smbtar errors ($t->{xferErrCnt})";
307     return;
308     }
309     if ( /^Error reading file (.*)\. Got 0 bytes/ ) {
310     #
311     # This happens when a Windoze application has
312     # locked the file. This is a particular problem
313     # with MS-Outlook. smbclient has already written
314     # the tar header to stdout, so all it can do is to
315     # write a dummy file with the correct size, but all
316     # zeros. BackupPC_tarExtract stores these
317     # zero-content files efficiently as a sparse file,
318     # or if compression is on the file will be small
319     # anyhow. After the dump is done we simply delete
320     # the file (it is no use) and try to link it to same
321     # file in any recent backup.
322     #
323     my $badFile = $1;
324     $badFile =~ s{\\}{/}g;
325     $badFile =~ s{^/}{};
326     push(@{$t->{badFiles}}, {
327     share => $t->{shareName},
328     file => $badFile
329     });
330     }
331     $t->{XferLOG}->write(\"$_\n") if ( $t->{logLevel} >= 1 );
332     }
333     }
334     return 1;
335     }
336    
337     sub abort
338     {
339     my($t, $reason) = @_;
340    
341     $t->{abort} = 1;
342     $t->{abortReason} = $reason;
343     }
344    
345     sub setSelectMask
346     {
347     my($t, $FDreadRef) = @_;
348    
349     vec($$FDreadRef, fileno($t->{pipeSMB}), 1) = 1;
350     }
351    
352     sub errStr
353     {
354     my($t) = @_;
355    
356     return $t->{_errStr};
357     }
358    
359     sub xferPid
360     {
361     my($t) = @_;
362    
363     return ($t->{xferPid});
364     }
365    
366     sub logMsg
367     {
368     my($t, $msg) = @_;
369    
370     push(@{$t->{_logMsg}}, $msg);
371     }
372    
373     sub logMsgGet
374     {
375     my($t) = @_;
376    
377     return shift(@{$t->{_logMsg}});
378     }
379    
380     #
381     # Returns a hash ref giving various status information about
382     # the transfer.
383     #
384     sub getStats
385     {
386     my($t) = @_;
387    
388     return { map { $_ => $t->{$_} }
389     qw(byteCnt fileCnt xferErrCnt xferBadShareCnt xferBadFileCnt
390     xferOK hostAbort hostError lastOutputLine)
391     };
392     }
393    
394     sub getBadFiles
395     {
396     my($t) = @_;
397    
398     return @{$t->{badFiles}};
399     }
400    
401     1;

  ViewVC Help
Powered by ViewVC 1.1.26