10 use Data::Dump qw(dump);
15 use CloudStore::Couchbase;
17 my $dir = $ENV{RSYNC_DIR} || '/srv/cloudstore';
18 my $port = $ENV{RSYNC_PORT} || 6501;
21 my $log_fifo = "$dir/var/$port.log";
22 my $pid_file = "$dir/var/$port.pid";
23 my $cfg_file = "$dir/var/$port.conf";
26 $rsync = 'bin/rsync' if -x 'bin/rsync'; # use 3.1dev version!
29 timestamp:%t:timestamp
45 $transfer[2] = "port:$port:int"; # expand $port
47 my @transfer_names = map { ( split(/:/,$_,3) )[0] } @transfer;
48 my $transfer_log = join('|',map { ( split(/:/,$_,3) )[1] } @transfer );
51 print "CREATE TABLE rsync_transfer (\n\t",
52 join(",\n\t", map { my @m = split(/:/,$_,3); "$m[0] $m[2]" } @transfer),
57 my $store = CloudStore::Couchbase->new;
59 mkdir "$dir/var" if ! -e "$dir/var";
61 unlink $log_fifo if -f $log_fifo;
62 mkfifo $log_fifo, 0700 unless -p $log_fifo;
64 my $rsync_config = qq{
72 lock file = $dir/var/$port.lock
74 #syslog facility = local5
77 transfer logging = yes
78 log format = transfer-log:$transfer_log
83 # don't check secrets file permission (uid)
86 #pre-xfer exec = /srv/cloudstore/pre-xfer.sh
87 #post-xfer exec = /srv/cloudstore/post-xfer.sh
91 foreach my $path ( glob "$users/*" ) {
94 $login =~ s{^.+/([^/]+)$}{$1};
96 if ( -d $path && -d "$path/blob" && -f "$path/secrets" ) {
97 my @secrets = map { chomp; $_ } read_file "$path/secrets";
98 my $auth_users = join(', ', map { s/:.+$//; $_ } @secrets );
100 $rsync_config .= <<__RSYNC_MODULE__;
103 path = $dir/users/$login/blob
104 auth users = $auth_users
105 secrets file = $dir/users/$login/secrets
110 print "INFO: added $login = $auth_users\n";
112 my $quota = read_file "$path/quota" if -e "$path/quota";
113 chomp $quota && $quota * 1;
114 $quota ||= 100 * 1024; # Kb - XXX default quota
119 secrets => [ @secrets ],
125 warn "skipped $login: $!";
130 write_file $cfg_file, $rsync_config;
131 warn "created $cfg_file ", -s $cfg_file, " bytes\n";
133 if ( -e $pid_file ) {
134 my $pid = read_file $pid_file;
136 if ( kill 0, $pid ) {
137 warn "found rsync pid $pid\n";
139 while ( -e $pid_file ) {
140 warn "waiting for rsync to die...\n";
144 open(my $fifo, '<', $log_fifo);
145 while ( kill 0, $pid ) {
150 kill 0, $pid && die "can't kill it!";
156 use POSIX ":sys_wait_h";
159 while ((my $waitedpid = waitpid(-1,WNOHANG)) > 0) {
160 warn "reaped $waitedpid" . ($? ? " with exit $?" : '');
162 $SIG{CHLD} = \&REAPER; # loathe SysV
165 $SIG{CHLD} = \&REAPER;
168 if ( ! -e $pid_file ) {
169 my $exec = "$rsync --daemon --config $cfg_file --no-detach --port=$port";
170 warn "START $exec\n";
172 die "could not fork\n" unless defined(my $pid = fork);
174 warn "start server with $exec\n";
175 exec $exec || die $!;
178 warn "wait for pid file";
179 while ( ! -e $pid_file ) {
186 my $gearman = Gearman::Client->new;
187 $gearman->job_servers('127.0.0.1:4730');
191 warn "# reading log output from $log_fifo\n";
192 open(my $fifo, '<', $log_fifo);
193 while( my $line = <$fifo> ) {
194 Module::Refresh->refresh;
195 die "ERROR: $line" if $line =~ /rsync error:/;
199 if ( $line =~ /transfer-log:(.+)/ ) {
201 $transfer =~ s|(\d\d\d\d)/(\d\d)/(\d\d)[-\s](\d\d:\d\d:\d\d)|$1-$2-$3T$4|g;
202 my ( $yyyy,$mm,$dd,undef,$login,undef ) = split( /[\-T\|]/, $transfer, 6 );
203 my $host = $1 if $login =~ s/\+(.+)//;
205 my $path = "users/$login/log";
206 mkdir $path unless -d $path;
207 $path .= "/$yyyy-$mm-$dd";
208 my $new_log = ! -e $path;
209 open( my $log, '>>', $path );
210 print $log join('|',@transfer),"\n" if $new_log; # store header
211 print $log "$transfer\n";
214 my @v = split(/\|/,$transfer,$#transfer + 1);
216 @data{@transfer_names} = @v ; # FIXME validate?
218 print ">>> data ",dump( \%data ) if $ENV{DEBUG};
220 $store->transfer( \%data );
222 $gearman->dispatch_background( 'rsync_transfer' => $json );