log.c

   1 /* -*- c-file-style: "linux"; -*-
   2
   3    Copyright (C) 1998-2001 by Andrew Tridgell <tridge@samba.org>
   4    Copyright (C) 2000-2001 by Martin Pool <mbp@samba.org>
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19 */
  20
  21 /*
  22   Logging and utility functions.
  23   tridge, May 1998
  24
  25   Mapping to human-readable messages added by Martin Pool
  26   <mbp@samba.org>, Oct 2000.
  27   */
  28 #include "rsync.h"
  29
  30 static char *logfname;
  31 static FILE *logfile;
  32 static int log_error_fd = -1;
  33
  34 int log_got_error=0;
  35
  36 struct {
  37         int code;
  38         char const *name;
  39 } const rerr_names[] = {
  40         { RERR_SYNTAX     , "syntax or usage error" },
  41         { RERR_PROTOCOL   , "protocol incompatibility" },
  42         { RERR_FILESELECT , "errors selecting input/output files, dirs" },
  43         { RERR_UNSUPPORTED, "requested action not supported" },
  44         { RERR_SOCKETIO   , "error in socket IO" },
  45         { RERR_FILEIO     , "error in file IO" },
  46         { RERR_STREAMIO   , "error in rsync protocol data stream" },
  47         { RERR_MESSAGEIO  , "errors with program diagnostics" },
  48         { RERR_IPC        , "error in IPC code" },
  49         { RERR_SIGNAL     , "received SIGUSR1 or SIGINT" },
  50         { RERR_WAITCHILD  , "some error returned by waitpid()" },
  51         { RERR_MALLOC     , "error allocating core memory buffers" },
  52         { RERR_PARTIAL    , "partial transfer" },
  53         { RERR_TIMEOUT    , "timeout in data send/receive" },
  54         { RERR_CMD_FAILED , "remote shell failed" },
  55         { RERR_CMD_KILLED , "remote shell killed" },
  56         { RERR_CMD_RUN,     "remote command could not be run" },
  57         { RERR_CMD_NOTFOUND, "remote command not found" },
  58         { 0, NULL }
  59 };
  60
  61
  62
  63 /*
  64  * Map from rsync error code to name, or return NULL.
  65  */
  66 static char const *rerr_name(int code)
  67 {
  68         int i;
  69         for (i = 0; rerr_names[i].name; i++) {
  70                 if (rerr_names[i].code == code)
  71                         return rerr_names[i].name;
  72         }
  73         return NULL;
  74 }
  75
  76 struct err_list {
  77         struct err_list *next;
  78         char *buf;
  79         int len;
  80         int written; /* how many bytes we have written so far */
  81 };
  82
  83 static struct err_list *err_list_head;
  84 static struct err_list *err_list_tail;
  85
  86 /* add an error message to the pending error list */
  87 static void err_list_add(int code, char *buf, int len)
  88 {
  89         struct err_list *el;
  90         el = (struct err_list *)malloc(sizeof(*el));
  91         if (!el) exit_cleanup(RERR_MALLOC);
  92         el->next = NULL;
  93         el->buf = malloc(len+4);
  94         if (!el->buf) exit_cleanup(RERR_MALLOC);
  95         memcpy(el->buf+4, buf, len);
  96         SIVAL(el->buf, 0, ((code+MPLEX_BASE)<<24) | len);
  97         el->len = len+4;
  98         el->written = 0;
  99         if (err_list_tail) {
 100                 err_list_tail->next = el;
 101         } else {
 102                 err_list_head = el;
 103         }
 104         err_list_tail = el;
 105 }
 106
 107
 108 /* try to push errors off the error list onto the wire */
 109 void err_list_push(void)
 110 {
 111         if (log_error_fd == -1) return;
 112
 113         while (err_list_head) {
 114                 struct err_list *el = err_list_head;
 115                 int n = write(log_error_fd, el->buf+el->written, el->len - el->written);
 116                 /* don't check for an error if the best way of handling the error is
 117                    to ignore it */
 118                 if (n == -1) break;
 119                 if (n > 0) {
 120                         el->written += n;
 121                 }
 122                 if (el->written == el->len) {
 123                         free(el->buf);
 124                         err_list_head = el->next;
 125                         if (!err_list_head) err_list_tail = NULL;
 126                         free(el);
 127                 }
 128         }
 129 }
 130
 131
 132 static void logit(int priority, char *buf)
 133 {
 134         if (logfname) {
 135                 if (!logfile)
 136                         log_open();
 137                 fprintf(logfile,"%s [%d] %s",
 138                         timestring(time(NULL)), (int)getpid(), buf);
 139                 fflush(logfile);
 140         } else {
 141                 syslog(priority, "%s", buf);
 142         }
 143 }
 144
 145 void log_init(void)
 146 {
 147         static int initialised;
 148         int options = LOG_PID;
 149         time_t t;
 150
 151         if (initialised) return;
 152         initialised = 1;
 153
 154         /* this looks pointless, but it is needed in order for the
 155            C library on some systems to fetch the timezone info
 156            before the chroot */
 157         t = time(NULL);
 158         localtime(&t);
 159
 160         /* optionally use a log file instead of syslog */
 161         logfname = lp_log_file();
 162         if (logfname) {
 163                 if (*logfname) {
 164                         log_open();
 165                         return;
 166                 }
 167                 logfname = NULL;
 168         }
 169
 170 #ifdef LOG_NDELAY
 171         options |= LOG_NDELAY;
 172 #endif
 173
 174 #ifdef LOG_DAEMON
 175         openlog("rsyncd", options, lp_syslog_facility());
 176 #else
 177         openlog("rsyncd", options);
 178 #endif
 179
 180 #ifndef LOG_NDELAY
 181         logit(LOG_INFO,"rsyncd started\n");
 182 #endif
 183 }
 184
 185 void log_open()
 186 {
 187         if (logfname && !logfile) {
 188                 extern int orig_umask;
 189                 int old_umask = umask(022 | orig_umask);
 190                 logfile = fopen(logfname, "a");
 191                 umask(old_umask);
 192         }
 193 }
 194
 195 void log_close()
 196 {
 197         if (logfile) {
 198                 fclose(logfile);
 199                 logfile = NULL;
 200         }
 201 }
 202
 203 /* setup the error file descriptor - used when we are a server
 204    that is receiving files */
 205 void set_error_fd(int fd)
 206 {
 207         log_error_fd = fd;
 208         set_nonblocking(log_error_fd);
 209 }
 210
 211 /* this is the underlying (unformatted) rsync debugging function. Call
 212    it with FINFO, FERROR or FLOG */
 213 void rwrite(enum logcode code, char *buf, int len)
 214 {
 215         FILE *f=NULL;
 216         extern int am_daemon;
 217         extern int am_server;
 218         extern int quiet;
 219         /* recursion can happen with certain fatal conditions */
 220
 221         if (quiet && code == FINFO) return;
 222
 223         if (len < 0) exit_cleanup(RERR_MESSAGEIO);
 224
 225         buf[len] = 0;
 226
 227         if (code == FLOG) {
 228                 if (am_daemon) logit(LOG_INFO, buf);
 229                 return;
 230         }
 231
 232         /* first try to pass it off to our sibling */
 233         if (am_server && log_error_fd != -1) {
 234                 err_list_add(code, buf, len);
 235                 err_list_push();
 236                 return;
 237         }
 238
 239         /* if that fails, try to pass it to the other end */
 240         if (am_server && io_multiplex_write(code, buf, len)) {
 241                 return;
 242         }
 243
 244         if (am_daemon) {
 245                 static int depth;
 246                 int priority = LOG_INFO;
 247                 if (code == FERROR) priority = LOG_WARNING;
 248
 249                 if (depth) return;
 250
 251                 depth++;
 252
 253                 log_init();
 254                 logit(priority, buf);
 255
 256                 depth--;
 257                 return;
 258         }
 259
 260         if (code == FERROR) {
 261                 log_got_error = 1;
 262                 f = stderr;
 263         }
 264
 265         if (code == FINFO) {
 266                 if (am_server)
 267                         f = stderr;
 268                 else
 269                         f = stdout;
 270         }
 271
 272         if (!f) exit_cleanup(RERR_MESSAGEIO);
 273
 274         if (fwrite(buf, len, 1, f) != 1) exit_cleanup(RERR_MESSAGEIO);
 275
 276         if (buf[len-1] == '\r' || buf[len-1] == '\n') fflush(f);
 277 }
 278
 279
 280 /* This is the rsync debugging function. Call it with FINFO, FERROR or
 281  * FLOG. */
 282 void rprintf(enum logcode code, const char *format, ...)
 283 {
 284         va_list ap;
 285         char buf[1024];
 286         int len;
 287
 288         va_start(ap, format);
 289         len = vslprintf(buf, sizeof(buf), format, ap);
 290         va_end(ap);
 291
 292         if (len > sizeof(buf)-1) exit_cleanup(RERR_MESSAGEIO);
 293
 294         rwrite(code, buf, len);
 295 }
 296
 297
 298 /* This is like rprintf, but it also tries to print some
 299  * representation of the error code.  Normally errcode = errno.
 300  *
 301  * Unlike rprintf, this always adds a newline and there should not be
 302  * one in the format string.
 303  *
 304  * Note that since strerror might involve dynamically loading a
 305  * message catalog we need to call it once before chroot-ing. */
 306 void rsyserr(enum logcode code, int errcode, const char *format, ...)
 307 {
 308         va_list ap;
 309         char buf[1024];
 310         int len, sys_len;
 311         char *sysmsg;
 312
 313         va_start(ap, format);
 314         len = vslprintf(buf, sizeof(buf), format, ap);
 315         va_end(ap);
 316
 317         if (len > sizeof(buf)-1) exit_cleanup(RERR_MESSAGEIO);
 318
 319         sysmsg = strerror(errcode);
 320         sys_len = strlen(sysmsg);
 321         if (len + 3 + sys_len > sizeof(buf) - 1)
 322                 exit_cleanup(RERR_MESSAGEIO);
 323
 324         strcpy(buf + len, ": ");
 325         len += 2;
 326         strcpy(buf + len, sysmsg);
 327         len += sys_len;
 328         strcpy(buf + len, "\n");
 329         len++;
 330
 331         rwrite(code, buf, len);
 332 }
 333
 334
 335
 336 void rflush(enum logcode code)
 337 {
 338         FILE *f = NULL;
 339         extern int am_daemon;
 340
 341         if (am_daemon) {
 342                 return;
 343         }
 344
 345         if (code == FLOG) {
 346                 return;
 347         }
 348
 349         if (code == FERROR) {
 350                 f = stderr;
 351         }
 352
 353         if (code == FINFO) {
 354                 extern int am_server;
 355                 if (am_server)
 356                         f = stderr;
 357                 else
 358                         f = stdout;
 359         }
 360
 361         if (!f) exit_cleanup(RERR_MESSAGEIO);
 362         fflush(f);
 363 }
 364
 365
 366
 367 /* a generic logging routine for send/recv, with parameter
 368    substitiution */
 369 static void log_formatted(enum logcode code,
 370                           char *format, char *op, struct file_struct *file,
 371                           struct stats *initial_stats)
 372 {
 373         extern int module_id;
 374         extern char *auth_user;
 375         char buf[1024];
 376         char buf2[1024];
 377         char *p, *s, *n;
 378         int l;
 379         extern struct stats stats;
 380         extern int am_sender;
 381         extern int am_daemon;
 382         int64 b;
 383
 384         strlcpy(buf, format, sizeof(buf));
 385
 386         for (s=&buf[0];
 387              s && (p=strchr(s,'%')); ) {
 388                 n = NULL;
 389                 s = p + 1;
 390
 391                 switch (p[1]) {
 392                 case 'h': if (am_daemon) n = client_name(0); break;
 393                 case 'a': if (am_daemon) n = client_addr(0); break;
 394                 case 'l':
 395                         slprintf(buf2,sizeof(buf2),"%.0f",
 396                                  (double)file->length);
 397                         n = buf2;
 398                         break;
 399                 case 'p':
 400                         slprintf(buf2,sizeof(buf2),"%d",
 401                                  (int)getpid());
 402                         n = buf2;
 403                         break;
 404                 case 'o': n = op; break;
 405                 case 'f':
 406                         slprintf(buf2, sizeof(buf2), "%s/%s",
 407                                  file->basedir?file->basedir:"",
 408                                  f_name(file));
 409                         clean_fname(buf2);
 410                         n = buf2;
 411                         if (*n == '/') n++;
 412                         break;
 413                 case 'm': n = lp_name(module_id); break;
 414                 case 't': n = timestring(time(NULL)); break;
 415                 case 'P': n = lp_path(module_id); break;
 416                 case 'u': n = auth_user; break;
 417                 case 'b':
 418                         if (am_sender) {
 419                                 b = stats.total_written -
 420                                         initial_stats->total_written;
 421                         } else {
 422                                 b = stats.total_read -
 423                                         initial_stats->total_read;
 424                         }
 425                         slprintf(buf2,sizeof(buf2),"%.0f", (double)b);
 426                         n = buf2;
 427                         break;
 428                 case 'c':
 429                         if (!am_sender) {
 430                                 b = stats.total_written -
 431                                         initial_stats->total_written;
 432                         } else {
 433                                 b = stats.total_read -
 434                                         initial_stats->total_read;
 435                         }
 436                         slprintf(buf2,sizeof(buf2),"%.0f", (double)b);
 437                         n = buf2;
 438                         break;
 439                 }
 440
 441                 if (!n) continue;
 442
 443                 l = strlen(n);
 444
 445                 if ((l-1) + ((int)(s - &buf[0])) > sizeof(buf)) {
 446                         rprintf(FERROR,"buffer overflow expanding %%%c - exiting\n",
 447                                 p[0]);
 448                         exit_cleanup(RERR_MESSAGEIO);
 449                 }
 450
 451                 if (l != 2) {
 452                         memmove(s+(l-1), s+1, strlen(s+1)+1);
 453                 }
 454                 memcpy(p, n, l);
 455
 456                 s = p+l;
 457         }
 458
 459         rprintf(code,"%s\n", buf);
 460 }
 461
 462 /* log the outgoing transfer of a file */
 463 void log_send(struct file_struct *file, struct stats *initial_stats)
 464 {
 465         extern int module_id;
 466         extern int am_server;
 467         extern char *log_format;
 468
 469         if (lp_transfer_logging(module_id)) {
 470                 log_formatted(FLOG, lp_log_format(module_id), "send", file, initial_stats);
 471         } else if (log_format && !am_server) {
 472                 log_formatted(FINFO, log_format, "send", file, initial_stats);
 473         }
 474 }
 475
 476 /* log the incoming transfer of a file */
 477 void log_recv(struct file_struct *file, struct stats *initial_stats)
 478 {
 479         extern int module_id;
 480         extern int am_server;
 481         extern char *log_format;
 482
 483         if (lp_transfer_logging(module_id)) {
 484                 log_formatted(FLOG, lp_log_format(module_id), "recv", file, initial_stats);
 485         } else if (log_format && !am_server) {
 486                 log_formatted(FINFO, log_format, "recv", file, initial_stats);
 487         }
 488 }
 489
 490
 491
 492
 493 /*
 494  * Called when the transfer is interrupted for some reason.
 495  *
 496  * Code is one of the RERR_* codes from errcode.h, or terminating
 497  * successfully.
 498  */
 499 void log_exit(int code, const char *file, int line)
 500 {
 501         if (code == 0) {
 502                 extern struct stats stats;
 503                 rprintf(FLOG,"wrote %.0f bytes  read %.0f bytes  total size %.0f\n",
 504                         (double)stats.total_written,
 505                         (double)stats.total_read,
 506                         (double)stats.total_size);
 507         } else {
 508                 const char *name;
 509
 510                 name = rerr_name(code);
 511                 if (!name)
 512                         name = "unexplained error";
 513
 514                 rprintf(FERROR,"rsync error: %s (code %d) at %s(%d)\n",
 515                         name, code, file, line);
 516         }
 517 }
 518
 519
 520
 521
 522 /* log the incoming transfer of a file for interactive use, this
 523    will be called at the end where the client was run
 524
 525    it i called when a file starts to be transferred
 526 */
 527 void log_transfer(struct file_struct *file, const char *fname)
 528 {
 529         extern int verbose;
 530
 531         if (!verbose) return;
 532
 533         rprintf(FINFO,"%s\n", fname);
 534 }
 535