10 use Data::Dump qw(dump);
16 use CloudStore::Store;
18 my $slice = $ARGV[0] || 's1';
20 my ( undef, $dir, $port, undef ) = getgrnam($slice);
22 my $log_fifo = "$dir/$port.log";
23 my $pid_file = "$dir/$port.pid";
24 my $cfg_file = "$dir/$port.conf";
27 $rsync = 'bin/rsync' if -x 'bin/rsync'; # use 3.1dev version!
30 timestamp:%t:timestamp
46 $transfer[2] = "port:$port:int"; # expand $port
48 my @transfer_names = map { ( split(/:/,$_,3) )[0] } @transfer;
49 my $transfer_log = join('|',map { ( split(/:/,$_,3) )[1] } @transfer );
52 print "CREATE TABLE rsync_transfer (\n\t",
53 join(",\n\t", map { my @m = split(/:/,$_,3); "$m[0] $m[2]" } @transfer),
58 my $store = CloudStore::Store->new( $slice );
60 unlink $log_fifo if -f $log_fifo;
61 mkfifo $log_fifo, 0700 unless -p $log_fifo;
63 sub rsync_rebuild_config {
65 my $rsync_config = qq{
73 lock file = $dir/$port.lock
75 #syslog facility = local5
78 transfer logging = yes
79 log format = transfer-log:$transfer_log
84 # don't check secrets file permission (uid)
87 #pre-xfer exec = /srv/cloudstore/rsync-xfer-trigger.pl
88 #post-xfer exec = /srv/cloudstore/rsync-xfer-trigger.pl
92 open(my $p, '<', '/var/lib/extrausers/passwd');
96 my ( $login, undef, $uid, $gid, $email, $path, $shell ) = split(/:/,$_,7);
98 if ( -d $path && -f "$path/.secrets" && ! -e "$path/.meta/secrets" ) {
99 $store->api->mkbasepath( "$path/.meta/secrets" );
100 rename "$path/.secrets", "$path/.meta/secrets";
101 warn "UPGRADE $login rsync secrets location\n";
104 if ( -d $path && -f "$path/.meta/secrets" ) {
105 my @secrets = map { chomp; $_ } read_file "$path/.meta/secrets";
106 my $auth_users = join(', ', map { s/:.+$//; $_ } @secrets );
108 $rsync_config .= <<__RSYNC_MODULE__;
112 auth users = $auth_users
113 secrets file = $path/.meta/secrets
118 # refuse options = c delete
120 incoming chmod = u=rwX,g+rX,o+rX
125 print "INFO: added $login = $auth_users\n";
128 warn "skipped $login: $!";
133 write_file $cfg_file, $rsync_config;
134 warn "created $cfg_file ", -s $cfg_file, " bytes\n";
136 } # sub rsync_rebuild_config
138 rsync_rebuild_config;
140 sub rsync_running_pid {
141 return unless -e $pid_file;
142 my $pid = read_file $pid_file;
147 if ( my $pid = rsync_running_pid ) {
148 if ( kill 0, $pid ) {
149 warn "found rsync pid $pid\n";
150 kill 1, $pid && warn "reload config";
153 while ( -e $pid_file ) {
154 warn "waiting for rsync to die...\n";
158 open(my $fifo, '<', $log_fifo);
159 while ( kill 0, $pid ) {
164 kill 0, $pid && die "can't kill it!";
171 use POSIX ":sys_wait_h";
174 while ((my $waitedpid = waitpid(-1,WNOHANG)) > 0) {
175 warn "reaped $waitedpid" . ($? ? " with exit $?" : '');
177 $SIG{CHLD} = \&REAPER; # loathe SysV
180 $SIG{CHLD} = \&REAPER;
183 if ( ! -e $pid_file ) {
184 my $exec = "$rsync --daemon --config $cfg_file --no-detach --port=$port";
185 warn "START $exec\n";
187 die "could not fork\n" unless defined(my $pid = fork);
189 warn "start server with $exec\n";
190 exec $exec || die $!;
193 warn "wait for pid file";
194 while ( ! -e $pid_file ) {
201 my $gearman = Gearman::Client->new;
202 $gearman->job_servers('127.0.0.1:4730');
206 die "no rsync running" unless kill 0, rsync_running_pid;
207 warn "waiting for log from $log_fifo\n";
208 open(my $fifo, '<', $log_fifo);
209 while( my $line = <$fifo> ) {
210 Module::Refresh->refresh;
214 if ( $line =~ /\[(\d+)\] transfer-log:(.+)/ ) {
217 $transfer =~ s|(\d\d\d\d)/(\d\d)/(\d\d)[-\s](\d\d:\d\d:\d\d)|$1-$2-$3T$4|g;
218 my ( $yyyy,$mm,$dd,undef,$login,undef ) = split( /[\-T\|]/, $transfer, 6 );
219 my $host = $1 if $login =~ s/\+(.+)//;
222 my $path = "users/$login/log";
223 mkdir $path unless -d $path;
224 $path .= "/$yyyy-$mm-$dd";
225 my $new_log = ! -e $path;
226 open( my $log, '>>', $path );
227 print $log join('|',@transfer),"\n" if $new_log; # store header
228 print $log "$transfer\n";
232 my @v = split(/\|/,$transfer,$#transfer + 1);
234 @data{@transfer_names} = @v ; # FIXME validate?
237 # overwrite pid from transfer log with consistant one for start/stop
239 print ">>> data ",dump( \%data ) if $ENV{DEBUG};
241 $store->rsync_transfer( \%data );
243 $gearman->dispatch_background( 'rsync_transfer' => $json );
246 die "no rsync running" unless kill 0, rsync_running_pid;
247 } elsif ( $line =~ m/(unknown module|rebuild|reload|config)/ ) {
248 warn "refresh modules, rebuild config and HUP rsync";
249 Module::Refresh->refresh;
250 rsync_rebuild_config;
251 my $pid = rsync_running_pid;
252 kill 1, $pid && warn "reload config";
254 $store->rsync_log( $line );