X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/0b76cd63ee0eacb95285dfb5d2cac2992e1794ef..08f15335b5b469b29cbdacd0fc5ba48d76acf280:/log.c diff --git a/log.c b/log.c index 20dc702a..788225f5 100644 --- a/log.c +++ b/log.c @@ -23,50 +23,142 @@ */ #include "rsync.h" -/* this is the rsync debugging function. Call it with FINFO or FERROR */ -void rprintf(int fd, const char *format, ...) +static FILE *logfile; +static int log_error_fd = -1; + +static void logit(int priority, char *buf) { - va_list ap; - char buf[1024]; - int len; - FILE *f=NULL; - extern int am_daemon; - - if (am_daemon) { - static FILE *logf; - if (!logf) logf = fopen(RSYNCD_LOG, "a"); - f = logf; - if (!f) return; + if (logfile) { + fprintf(logfile,"%s [%d] %s", + timestring(time(NULL)), (int)getpid(), buf); + fflush(logfile); + } else { + syslog(priority, "%s", buf); } +} - va_start(ap, format); +void log_open(void) +{ + static int initialised; + int options = LOG_PID; + time_t t; + char *logf; -#if HAVE_VSNPRINTF - len = vsnprintf(buf, sizeof(buf)-1, format, ap); + if (initialised) return; + initialised = 1; + + /* this looks pointless, but it is needed in order for the + C library on some systems to fetch the timezone info + before the chroot */ + t = time(NULL); + localtime(&t); + + /* optionally use a log file instead of syslog */ + logf = lp_log_file(); + if (logf && *logf) { + extern int orig_umask; + int old_umask = umask(022 | orig_umask); + logfile = fopen(logf, "a"); + umask(old_umask); + return; + } + +#ifdef LOG_NDELAY + options |= LOG_NDELAY; +#endif + +#ifdef LOG_DAEMON + openlog("rsyncd", options, lp_syslog_facility()); #else - len = vsprintf(buf, format, ap); + openlog("rsyncd", options); #endif - va_end(ap); - if (len < 0) exit_cleanup(1); +#ifndef LOG_NDELAY + logit(LOG_INFO,"rsyncd started\n"); +#endif +} - if (!am_daemon) { - if (fd == FERROR) { - f = stderr; - } - - if (fd == FINFO) { - extern int am_server; - if (am_server) - f = stderr; - else - f = stdout; - } +/* setup the error file descriptor - used when we are a server + that is receiving files */ +void set_error_fd(int fd) +{ + log_error_fd = fd; +} + +/* this is the underlying (unformatted) rsync debugging function. Call + it with FINFO, FERROR or FLOG */ +void rwrite(int fd, char *buf, int len) +{ + FILE *f=NULL; + extern int am_daemon; + extern int quiet; + /* recursion can happen with certain fatal conditions */ + + if (quiet != 0 && fd == FINFO) return; + + if (len < 0) exit_cleanup(RERR_MESSAGEIO); + + buf[len] = 0; + + if (fd == FLOG) { + if (am_daemon) logit(LOG_INFO, buf); + return; } - if (!f) exit_cleanup(1); + if (am_daemon) { + static int depth; + int priority = LOG_INFO; + if (fd == FERROR) priority = LOG_WARNING; + + if (depth) return; + + depth++; + + log_open(); + + if (!io_error_write(log_error_fd, buf, strlen(buf)) && + !io_multiplex_write(fd, buf, strlen(buf))) { + logit(priority, buf); + } + + depth--; + return; + } + + if (fd == FERROR) { + f = stderr; + } + + if (fd == FINFO) { + extern int am_server; + if (am_server) + f = stderr; + else + f = stdout; + } + + if (!f) exit_cleanup(RERR_MESSAGEIO); - if (fwrite(buf, len, 1, f) != 1) exit_cleanup(1); + if (fwrite(buf, len, 1, f) != 1) exit_cleanup(RERR_MESSAGEIO); + + if (buf[len-1] == '\r' || buf[len-1] == '\n') fflush(f); +} + + +/* this is the rsync debugging function. Call it with FINFO, FERROR or FLOG */ + void rprintf(int fd, const char *format, ...) +{ + va_list ap; + char buf[1024]; + int len; + + va_start(ap, format); + len = vslprintf(buf, sizeof(buf), format, ap); + va_end(ap); + + if (len > sizeof(buf)-1) exit_cleanup(RERR_MESSAGEIO); + + rwrite(fd, buf, len); } void rflush(int fd) @@ -78,6 +170,10 @@ void rflush(int fd) return; } + if (fd == FLOG) { + return; + } + if (fd == FERROR) { f = stderr; } @@ -90,7 +186,161 @@ void rflush(int fd) f = stdout; } - if (!f) exit_cleanup(1); + if (!f) exit_cleanup(RERR_MESSAGEIO); fflush(f); } + + +/* a generic logging routine for send/recv, with parameter + substitiution */ +static void log_formatted(int fd, + char *format, char *op, struct file_struct *file, + struct stats *initial_stats) +{ + extern int module_id; + extern char *auth_user; + char buf[1024]; + char buf2[1024]; + char *p, *s, *n; + int l; + extern struct stats stats; + extern int am_sender; + extern int am_daemon; + int64 b; + + strlcpy(buf, format, sizeof(buf)); + + for (s=&buf[0]; + s && (p=strchr(s,'%')); ) { + n = NULL; + s = p + 1; + + switch (p[1]) { + case 'h': if (am_daemon) n = client_name(0); break; + case 'a': if (am_daemon) n = client_addr(0); break; + case 'l': + slprintf(buf2,sizeof(buf2),"%.0f", + (double)file->length); + n = buf2; + break; + case 'p': + slprintf(buf2,sizeof(buf2),"%d", + (int)getpid()); + n = buf2; + break; + case 'o': n = op; break; + case 'f': + slprintf(buf2, sizeof(buf2), "%s/%s", + file->basedir?file->basedir:"", + f_name(file)); + clean_fname(buf2); + n = buf2; + if (*n == '/') n++; + break; + case 'm': n = lp_name(module_id); break; + case 't': n = timestring(time(NULL)); break; + case 'P': n = lp_path(module_id); break; + case 'u': n = auth_user; break; + case 'b': + if (am_sender) { + b = stats.total_written - + initial_stats->total_written; + } else { + b = stats.total_read - + initial_stats->total_read; + } + slprintf(buf2,sizeof(buf2),"%.0f", (double)b); + n = buf2; + break; + case 'c': + if (!am_sender) { + b = stats.total_written - + initial_stats->total_written; + } else { + b = stats.total_read - + initial_stats->total_read; + } + slprintf(buf2,sizeof(buf2),"%.0f", (double)b); + n = buf2; + break; + } + + if (!n) continue; + + l = strlen(n); + + if ((l-1) + ((int)(s - &buf[0])) > sizeof(buf)) { + rprintf(FERROR,"buffer overflow expanding %%%c - exiting\n", + p[0]); + exit_cleanup(RERR_MESSAGEIO); + } + + if (l != 2) { + memmove(s+(l-1), s+1, strlen(s+1)+1); + } + memcpy(p, n, l); + + s = p+l; + } + + rprintf(fd,"%s\n", buf); +} + +/* log the outgoing transfer of a file */ +void log_send(struct file_struct *file, struct stats *initial_stats) +{ + extern int module_id; + extern int am_server; + extern char *log_format; + + if (lp_transfer_logging(module_id)) { + log_formatted(FLOG, lp_log_format(module_id), "send", file, initial_stats); + } else if (log_format && !am_server) { + log_formatted(FINFO, log_format, "send", file, initial_stats); + } +} + +/* log the incoming transfer of a file */ +void log_recv(struct file_struct *file, struct stats *initial_stats) +{ + extern int module_id; + extern int am_server; + extern char *log_format; + + if (lp_transfer_logging(module_id)) { + log_formatted(FLOG, lp_log_format(module_id), "recv", file, initial_stats); + } else if (log_format && !am_server) { + log_formatted(FINFO, log_format, "recv", file, initial_stats); + } +} + +/* called when the transfer is interrupted for some reason */ +void log_exit(int code, const char *file, int line) +{ + if (code == 0) { + extern struct stats stats; + rprintf(FLOG,"wrote %.0f bytes read %.0f bytes total size %.0f\n", + (double)stats.total_written, + (double)stats.total_read, + (double)stats.total_size); + } else { + rprintf(FLOG,"transfer interrupted (code %d) at %s(%d)\n", + code, file, line); + } +} + +/* log the incoming transfer of a file for interactive use, this + will be called at the end where the client was run + + it i called when a file starts to be transferred +*/ +void log_transfer(struct file_struct *file, const char *fname) +{ + extern int verbose; + + if (!verbose) return; + + rprintf(FINFO,"%s\n", fname); +} +