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 int log_initialised;
  31 static char *logfname;
  32 static FILE *logfile;
  33 static int log_error_fd = -1;
  34 struct stats stats;
  35
  36 int log_got_error=0;
  37
  38 struct {
  39         int code;
  40         char const *name;
  41 } const rerr_names[] = {
  42         { RERR_SYNTAX     , "syntax or usage error" },
  43         { RERR_PROTOCOL   , "protocol incompatibility" },
  44         { RERR_FILESELECT , "errors selecting input/output files, dirs" },
  45         { RERR_UNSUPPORTED, "requested action not supported" },
  46         { RERR_STARTCLIENT, "error starting client-server protocol" },
  47         { RERR_SOCKETIO   , "error in socket IO" },
  48         { RERR_FILEIO     , "error in file IO" },
  49         { RERR_STREAMIO   , "error in rsync protocol data stream" },
  50         { RERR_MESSAGEIO  , "errors with program diagnostics" },
  51         { RERR_IPC        , "error in IPC code" },
  52         { RERR_SIGNAL     , "received SIGUSR1 or SIGINT" },
  53         { RERR_WAITCHILD  , "some error returned by waitpid()" },
  54         { RERR_MALLOC     , "error allocating core memory buffers" },
  55         { RERR_PARTIAL    , "some files could not be transferred" },
  56         { RERR_TIMEOUT    , "timeout in data send/receive" },
  57         { RERR_CMD_FAILED , "remote shell failed" },
  58         { RERR_CMD_KILLED , "remote shell killed" },
  59         { RERR_CMD_RUN,     "remote command could not be run" },
  60         { RERR_CMD_NOTFOUND, "remote command not found" },
  61         { 0, NULL }
  62 };
  63
  64
  65
  66 /*
  67  * Map from rsync error code to name, or return NULL.
  68  */
  69 static char const *rerr_name(int code)
  70 {
  71         int i;
  72         for (i = 0; rerr_names[i].name; i++) {
  73                 if (rerr_names[i].code == code)
  74                         return rerr_names[i].name;
  75         }
  76         return NULL;
  77 }
  78
  79 struct err_list {
  80         struct err_list *next;
  81         char *buf;
  82         int len;
  83         int written; /* how many bytes we have written so far */
  84 };
  85
  86 static struct err_list *err_list_head;
  87 static struct err_list *err_list_tail;
  88
  89 /* add an error message to the pending error list */
  90 static void err_list_add(int code, char *buf, int len)
  91 {
  92         struct err_list *el;
  93         el = (struct err_list *)malloc(sizeof(*el));
  94         if (!el) exit_cleanup(RERR_MALLOC);
  95         el->next = NULL;
  96         el->buf = malloc(len+4);
  97         if (!el->buf) exit_cleanup(RERR_MALLOC);
  98         memcpy(el->buf+4, buf, len);
  99         SIVAL(el->buf, 0, ((code+MPLEX_BASE)<<24) | len);
 100         el->len = len+4;
 101         el->written = 0;
 102         if (err_list_tail) {
 103                 err_list_tail->next = el;
 104         } else {
 105                 err_list_head = el;
 106         }
 107         err_list_tail = el;
 108 }
 109
 110
 111 /* try to push errors off the error list onto the wire */
 112 void err_list_push(void)
 113 {
 114         if (log_error_fd == -1) return;
 115
 116         while (err_list_head) {
 117                 struct err_list *el = err_list_head;
 118                 int n = write(log_error_fd, el->buf+el->written, el->len - el->written);
 119                 /* don't check for an error if the best way of handling the error is
 120                    to ignore it */
 121                 if (n == -1) break;
 122                 if (n > 0) {
 123                         el->written += n;
 124                 }
 125                 if (el->written == el->len) {
 126                         free(el->buf);
 127                         err_list_head = el->next;
 128                         if (!err_list_head) err_list_tail = NULL;
 129                         free(el);
 130                 }
 131         }
 132 }
 133
 134
 135 static void logit(int priority, char *buf)
 136 {
 137         if (logfname) {
 138                 if (!logfile)
 139                         log_open();
 140                 fprintf(logfile,"%s [%d] %s",
 141                         timestring(time(NULL)), (int)getpid(), buf);
 142                 fflush(logfile);
 143         } else {
 144                 syslog(priority, "%s", buf);
 145         }
 146 }
 147
 148 void log_init(void)
 149 {
 150         int options = LOG_PID;
 151         time_t t;
 152
 153         if (log_initialised) return;
 154         log_initialised = 1;
 155
 156         /* this looks pointless, but it is needed in order for the
 157            C library on some systems to fetch the timezone info
 158            before the chroot */
 159         t = time(NULL);
 160         localtime(&t);
 161
 162         /* optionally use a log file instead of syslog */
 163         logfname = lp_log_file();
 164         if (logfname) {
 165                 if (*logfname) {
 166                         log_open();
 167                         return;
 168                 }
 169                 logfname = NULL;
 170         }
 171
 172 #ifdef LOG_NDELAY
 173         options |= LOG_NDELAY;
 174 #endif
 175
 176 #ifdef LOG_DAEMON
 177         openlog("rsyncd", options, lp_syslog_facility());
 178 #else
 179         openlog("rsyncd", options);
 180 #endif
 181
 182 #ifndef LOG_NDELAY
 183         logit(LOG_INFO,"rsyncd started\n");
 184 #endif
 185 }
 186
 187 void log_open(void)
 188 {
 189         if (logfname && !logfile) {
 190                 extern int orig_umask;
 191                 int old_umask = umask(022 | orig_umask);
 192                 logfile = fopen(logfname, "a");
 193                 umask(old_umask);
 194         }
 195 }
 196
 197 void log_close(void)
 198 {
 199         if (logfile) {
 200                 fclose(logfile);
 201                 logfile = NULL;
 202         }
 203 }
 204
 205 /* setup the error file descriptor - used when we are a server
 206    that is receiving files */
 207 void set_error_fd(int fd)
 208 {
 209         log_error_fd = fd;
 210         set_nonblocking(log_error_fd);
 211 }
 212
 213 /* this is the underlying (unformatted) rsync debugging function. Call
 214    it with FINFO, FERROR or FLOG */
 215 void rwrite(enum logcode code, char *buf, int len)
 216 {
 217         FILE *f=NULL;
 218         extern int am_daemon;
 219         extern int am_server;
 220         extern int quiet;
 221         /* recursion can happen with certain fatal conditions */
 222
 223         if (quiet && code == FINFO) return;
 224
 225         if (len < 0) exit_cleanup(RERR_MESSAGEIO);
 226
 227         buf[len] = 0;
 228
 229         if (code == FLOG) {
 230                 if (am_daemon) logit(LOG_INFO, buf);
 231                 return;
 232         }
 233
 234         /* first try to pass it off to our sibling */
 235         if (am_server && log_error_fd != -1) {
 236                 err_list_add(code, buf, len);
 237                 err_list_push();
 238                 return;
 239         }
 240
 241         /* next, if we are a server and multiplexing is enabled,
 242          * pass it to the other side.  */
 243         if (am_server && io_multiplex_write(code, buf, len)) {
 244                 return;
 245         }
 246
 247         /* otherwise, if in daemon mode and either we are not a server
 248          *  (that is, we are not running --daemon over a remote shell) or
 249          *  the log has already been initialised, log the message on this
 250          *  side because we don't want the client to see most errors for
 251          *  security reasons.  We do want early messages when running daemon
 252          *  mode over a remote shell to go to the remote side; those will
 253          *  fall through to the next case.
 254          * Note that this is only for the time before multiplexing is enabled.
 255          */
 256         if (am_daemon && (!am_server || log_initialised)) {
 257                 static int depth;
 258                 int priority = LOG_INFO;
 259                 if (code == FERROR) priority = LOG_WARNING;
 260
 261                 if (depth) return;
 262
 263                 depth++;
 264
 265                 log_init();
 266                 logit(priority, buf);
 267
 268                 depth--;
 269                 return;
 270         }
 271
 272         if (code == FERROR) {
 273                 log_got_error = 1;
 274                 f = stderr;
 275         }
 276
 277         if (code == FINFO) {
 278                 if (am_server)
 279                         f = stderr;
 280                 else
 281                         f = stdout;
 282         }
 283
 284         if (!f) exit_cleanup(RERR_MESSAGEIO);
 285
 286         if (fwrite(buf, len, 1, f) != 1) exit_cleanup(RERR_MESSAGEIO);
 287
 288         if (buf[len-1] == '\r' || buf[len-1] == '\n') fflush(f);
 289 }
 290
 291
 292 /* This is the rsync debugging function. Call it with FINFO, FERROR or
 293  * FLOG. */
 294 void rprintf(enum logcode code, const char *format, ...)
 295 {
 296         va_list ap;
 297         char buf[1024];
 298         int len;
 299
 300         va_start(ap, format);
 301         /* Note: might return -1 */
 302         len = vsnprintf(buf, sizeof(buf), format, ap);
 303         va_end(ap);
 304
 305         /* Deal with buffer overruns.  Instead of panicking, just
 306          * truncate the resulting string.  Note that some vsnprintf()s
 307          * return -1 on truncation, e.g., glibc 2.0.6 and earlier. */
 308         if ((size_t) len > sizeof(buf)-1  ||  len < 0) {
 309                 const char ellipsis[] = "[...]";
 310
 311                 /* Reset length, and zero-terminate the end of our buffer */
 312                 len = sizeof(buf)-1;
 313                 buf[len] = '\0';
 314
 315                 /* Copy the ellipsis to the end of the string, but give
 316                  * us one extra character:
 317                  *
 318                  *                  v--- null byte at buf[sizeof(buf)-1]
 319                  *        abcdefghij0
 320                  *     -> abcd[...]00  <-- now two null bytes at end
 321                  *
 322                  * If the input format string has a trailing newline,
 323                  * we copy it into that extra null; if it doesn't, well,
 324                  * all we lose is one byte.  */
 325                 strncpy(buf+len-sizeof(ellipsis), ellipsis, sizeof(ellipsis));
 326                 if (format[strlen(format)-1] == '\n') {
 327                         buf[len-1] = '\n';
 328                 }
 329         }
 330
 331         rwrite(code, buf, len);
 332 }
 333
 334
 335 /* This is like rprintf, but it also tries to print some
 336  * representation of the error code.  Normally errcode = errno.
 337  *
 338  * Unlike rprintf, this always adds a newline and there should not be
 339  * one in the format string.
 340  *
 341  * Note that since strerror might involve dynamically loading a
 342  * message catalog we need to call it once before chroot-ing. */
 343 void rsyserr(enum logcode code, int errcode, const char *format, ...)
 344 {
 345         va_list ap;
 346         char buf[1024];
 347         int len;
 348         size_t sys_len;
 349         char *sysmsg;
 350
 351         va_start(ap, format);
 352         /* Note: might return <0 */
 353         len = vsnprintf(buf, sizeof(buf), format, ap);
 354         va_end(ap);
 355
 356         /* TODO: Put in RSYNC_NAME at the start. */
 357
 358         if ((size_t) len > sizeof(buf)-1)
 359                 exit_cleanup(RERR_MESSAGEIO);
 360
 361         sysmsg = strerror(errcode);
 362         sys_len = strlen(sysmsg);
 363         if ((size_t) len + 3 + sys_len > sizeof(buf) - 1)
 364                 exit_cleanup(RERR_MESSAGEIO);
 365
 366         strcpy(buf + len, ": ");
 367         len += 2;
 368         strcpy(buf + len, sysmsg);
 369         len += sys_len;
 370         strcpy(buf + len, "\n");
 371         len++;
 372
 373         rwrite(code, buf, len);
 374 }
 375
 376
 377
 378 void rflush(enum logcode code)
 379 {
 380         FILE *f = NULL;
 381         extern int am_daemon;
 382
 383         if (am_daemon) {
 384                 return;
 385         }
 386
 387         if (code == FLOG) {
 388                 return;
 389         }
 390
 391         if (code == FERROR) {
 392                 f = stderr;
 393         }
 394
 395         if (code == FINFO) {
 396                 extern int am_server;
 397                 if (am_server)
 398                         f = stderr;
 399                 else
 400                         f = stdout;
 401         }
 402
 403         if (!f) exit_cleanup(RERR_MESSAGEIO);
 404         fflush(f);
 405 }
 406
 407
 408
 409 /* a generic logging routine for send/recv, with parameter
 410    substitiution */
 411 static void log_formatted(enum logcode code,
 412                           char *format, char *op, struct file_struct *file,
 413                           struct stats *initial_stats)
 414 {
 415         extern int module_id;
 416         extern char *auth_user;
 417         char buf[1024];
 418         char buf2[1024];
 419         char *p, *s, *n;
 420         size_t l;
 421         extern struct stats stats;
 422         extern int am_sender;
 423         extern int am_daemon;
 424         int64 b;
 425
 426         /* We expand % codes one by one in place in buf.  We don't
 427          * copy in the terminating nul of the inserted strings, but
 428          * rather keep going until we reach the nul of the format.
 429          * Just to make sure we don't clobber that nul and therefore
 430          * accidentally keep going, we zero the buffer now. */
 431         memset(buf, 0, sizeof buf);
 432         strlcpy(buf, format, sizeof(buf));
 433
 434         for (s=&buf[0];
 435              s && (p=strchr(s,'%')); ) {
 436                 n = NULL;
 437                 s = p + 1;
 438
 439                 switch (p[1]) {
 440                 case 'h': if (am_daemon) n = client_name(0); break;
 441                 case 'a': if (am_daemon) n = client_addr(0); break;
 442                 case 'l':
 443                         snprintf(buf2,sizeof(buf2),"%.0f",
 444                                  (double)file->length);
 445                         n = buf2;
 446                         break;
 447                 case 'p':
 448                         snprintf(buf2,sizeof(buf2),"%d",
 449                                  (int)getpid());
 450                         n = buf2;
 451                         break;
 452                 case 'o': n = op; break;
 453                 case 'f':
 454                         snprintf(buf2, sizeof(buf2), "%s/%s",
 455                                  file->basedir?file->basedir:"",
 456                                  f_name(file));
 457                         clean_fname(buf2);
 458                         n = buf2;
 459                         if (*n == '/') n++;
 460                         break;
 461                 case 'm': n = lp_name(module_id); break;
 462                 case 't': n = timestring(time(NULL)); break;
 463                 case 'P': n = lp_path(module_id); break;
 464                 case 'u': n = auth_user; break;
 465                 case 'b':
 466                         if (am_sender) {
 467                                 b = stats.total_written -
 468                                         initial_stats->total_written;
 469                         } else {
 470                                 b = stats.total_read -
 471                                         initial_stats->total_read;
 472                         }
 473                         snprintf(buf2,sizeof(buf2),"%.0f", (double)b);
 474                         n = buf2;
 475                         break;
 476                 case 'c':
 477                         if (!am_sender) {
 478                                 b = stats.total_written -
 479                                         initial_stats->total_written;
 480                         } else {
 481                                 b = stats.total_read -
 482                                         initial_stats->total_read;
 483                         }
 484                         snprintf(buf2,sizeof(buf2),"%.0f", (double)b);
 485                         n = buf2;
 486                         break;
 487                 }
 488
 489                 /* n is the string to be inserted in place of this %
 490                  * code; l is its length not including the trailing
 491                  * NUL */
 492                 if (!n)
 493                         continue;
 494
 495                 l = strlen(n);
 496
 497                 if (l + ((int)(s - &buf[0])) >= sizeof(buf)) {
 498                         rprintf(FERROR,"buffer overflow expanding %%%c - exiting\n",
 499                                 p[0]);
 500                         exit_cleanup(RERR_MESSAGEIO);
 501                 }
 502
 503                 /* Shuffle the rest of the string along to make space for n */
 504                 if (l != 2) {
 505                         memmove(s+(l-1), s+1, strlen(s+1)+1);
 506                 }
 507
 508                 /* Copy in n but NOT its nul, because the format sting
 509                  * probably continues after this. */
 510                 memcpy(p, n, l);
 511
 512                 /* Skip over inserted string; continue looking */
 513                 s = p+l;
 514         }
 515
 516         rprintf(code,"%s\n", buf);
 517 }
 518
 519 /* log the outgoing transfer of a file */
 520 void log_send(struct file_struct *file, struct stats *initial_stats)
 521 {
 522         extern int module_id;
 523         extern int am_server;
 524         extern char *log_format;
 525
 526         if (lp_transfer_logging(module_id)) {
 527                 log_formatted(FLOG, lp_log_format(module_id), "send", file, initial_stats);
 528         } else if (log_format && !am_server) {
 529                 log_formatted(FINFO, log_format, "send", file, initial_stats);
 530         }
 531 }
 532
 533 /* log the incoming transfer of a file */
 534 void log_recv(struct file_struct *file, struct stats *initial_stats)
 535 {
 536         extern int module_id;
 537         extern int am_server;
 538         extern char *log_format;
 539
 540         if (lp_transfer_logging(module_id)) {
 541                 log_formatted(FLOG, lp_log_format(module_id), "recv", file, initial_stats);
 542         } else if (log_format && !am_server) {
 543                 log_formatted(FINFO, log_format, "recv", file, initial_stats);
 544         }
 545 }
 546
 547
 548
 549
 550 /*
 551  * Called when the transfer is interrupted for some reason.
 552  *
 553  * Code is one of the RERR_* codes from errcode.h, or terminating
 554  * successfully.
 555  */
 556 void log_exit(int code, const char *file, int line)
 557 {
 558         if (code == 0) {
 559                 extern struct stats stats;
 560                 rprintf(FLOG,"wrote %.0f bytes  read %.0f bytes  total size %.0f\n",
 561                         (double)stats.total_written,
 562                         (double)stats.total_read,
 563                         (double)stats.total_size);
 564         } else {
 565                 const char *name;
 566
 567                 name = rerr_name(code);
 568                 if (!name)
 569                         name = "unexplained error";
 570
 571                 rprintf(FERROR,"rsync error: %s (code %d) at %s(%d)\n",
 572                         name, code, file, line);
 573         }
 574 }
 575
 576 /*
 577  * Log the incoming transfer of a file for interactive use,
 578  * this will be called at the end where the client was run.
 579  * Called when a file starts to be transferred.
 580  */
 581 void log_transfer(struct file_struct *file, const char *fname)
 582 {
 583         extern int verbose;
 584
 585         if (!verbose) return;
 586
 587         rprintf(FINFO, "%s\n", fname);
 588 }