git.sur5r.net Git - bacula/bacula/blob - bacula/src/lib/util.c

   1 /*
   2  *   util.c  miscellaneous utility subroutines for Bacula
   3  *
   4  *    Kern Sibbald, MM
   5  *
   6  *   Version $Id$
   7  */
   8
   9 /*
  10    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
  11
  12    This program is free software; you can redistribute it and/or
  13    modify it under the terms of the GNU General Public License as
  14    published by the Free Software Foundation; either version 2 of
  15    the License, or (at your option) any later version.
  16
  17    This program is distributed in the hope that it will be useful,
  18    but WITHOUT ANY WARRANTY; without even the implied warranty of
  19    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  20    General Public License for more details.
  21
  22    You should have received a copy of the GNU General Public
  23    License along with this program; if not, write to the Free
  24    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  25    MA 02111-1307, USA.
  26
  27  */
  28
  29 #include "bacula.h"
  30 #include "jcr.h"
  31 #include "findlib/find.h"
  32
  33 /*
  34  * Various Bacula Utility subroutines
  35  *
  36  */
  37
  38 /* Return true of buffer has all zero bytes */
  39 int is_buf_zero(char *buf, int len)
  40 {
  41    uint64_t *ip = (uint64_t *)buf;
  42    char *p;
  43    int i, len64, done, rem;
  44
  45    /* Optimize by checking uint64_t for zero */
  46    len64 = len >> sizeof(uint64_t);
  47    for (i=0; i < len64; i++) {
  48       if (ip[i] != 0) {
  49          return 0;
  50       }
  51    }
  52    done = len64 << sizeof(uint64_t);  /* bytes already checked */
  53    p = buf + done;
  54    rem = len - done;
  55    for (i = 0; i < rem; i++) {
  56       if (p[i] != 0) {
  57          return 0;
  58       }
  59    }
  60    return 1;
  61 }
  62
  63
  64 /* Convert a string in place to lower case */
  65 void lcase(char *str)
  66 {
  67    while (*str) {
  68       if (B_ISUPPER(*str))
  69          *str = tolower((int)(*str));
  70        str++;
  71    }
  72 }
  73
  74 /* Convert spaces to non-space character.
  75  * This makes scanf of fields containing spaces easier.
  76  */
  77 void
  78 bash_spaces(char *str)
  79 {
  80    while (*str) {
  81       if (*str == ' ')
  82          *str = 0x1;
  83       str++;
  84    }
  85 }
  86
  87 /* Convert non-space characters (0x1) back into spaces */
  88 void
  89 unbash_spaces(char *str)
  90 {
  91    while (*str) {
  92      if (*str == 0x1)
  93         *str = ' ';
  94      str++;
  95    }
  96 }
  97
  98 /* Strip any trailing junk from the command */
  99 void strip_trailing_junk(char *cmd)
 100 {
 101    char *p;
 102    p = cmd + strlen(cmd) - 1;
 103
 104    /* strip trailing junk from command */
 105    while ((p >= cmd) && (*p == '\n' || *p == '\r' || *p == ' '))
 106       *p-- = 0;
 107 }
 108
 109 /* Strip any trailing slashes from a directory path */
 110 void strip_trailing_slashes(char *dir)
 111 {
 112    char *p;
 113    p = dir + strlen(dir) - 1;
 114
 115    /* strip trailing slashes */
 116    while ((p >= dir) && (*p == '/'))
 117       *p-- = 0;
 118 }
 119
 120 /*
 121  * Skip spaces
 122  *  Returns: 0 on failure (EOF)
 123  *           1 on success
 124  *           new address in passed parameter
 125  */
 126 int skip_spaces(char **msg)
 127 {
 128    char *p = *msg;
 129    if (!p) {
 130       return 0;
 131    }
 132    while (*p && *p == ' ') {
 133       p++;
 134    }
 135    *msg = p;
 136    return *p ? 1 : 0;
 137 }
 138
 139 /*
 140  * Skip nonspaces
 141  *  Returns: 0 on failure (EOF)
 142  *           1 on success
 143  *           new address in passed parameter
 144  */
 145 int skip_nonspaces(char **msg)
 146 {
 147    char *p = *msg;
 148
 149    if (!p) {
 150       return 0;
 151    }
 152    while (*p && *p != ' ') {
 153       p++;
 154    }
 155    *msg = p;
 156    return *p ? 1 : 0;
 157 }
 158
 159 /* folded search for string - case insensitive */
 160 int
 161 fstrsch(char *a, char *b)   /* folded case search */
 162 {
 163    register char *s1,*s2;
 164    register char c1, c2;
 165
 166    s1=a;
 167    s2=b;
 168    while (*s1) {                      /* do it the fast way */
 169       if ((*s1++ | 0x20) != (*s2++ | 0x20))
 170          return 0;                    /* failed */
 171    }
 172    while (*a) {                       /* do it over the correct slow way */
 173       if (B_ISUPPER(c1 = *a)) {
 174          c1 = tolower((int)c1);
 175       }
 176       if (B_ISUPPER(c2 = *b)) {
 177          c2 = tolower((int)c2);
 178       }
 179       if (c1 != c2) {
 180          return 0;
 181       }
 182       a++;
 183       b++;
 184    }
 185    return 1;
 186 }
 187
 188
 189 char *encode_time(time_t time, char *buf)
 190 {
 191    struct tm tm;
 192    int n = 0;
 193
 194    if (localtime_r(&time, &tm)) {
 195       n = sprintf(buf, "%04d-%02d-%02d %02d:%02d:%02d",
 196                    tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
 197                    tm.tm_hour, tm.tm_min, tm.tm_sec);
 198    }
 199    return buf+n;
 200 }
 201
 202 /*
 203  * Concatenate a string (str) onto a pool memory buffer pm
 204  */
 205 void pm_strcat(POOLMEM **pm, char *str)
 206 {
 207    int pmlen = strlen(*pm);
 208    int len = strlen(str) + 1;
 209
 210    *pm = check_pool_memory_size(*pm, pmlen + len);
 211    memcpy(*pm+pmlen, str, len);
 212 }
 213
 214
 215 /*
 216  * Copy a string (str) into a pool memory buffer pm
 217  */
 218 void pm_strcpy(POOLMEM **pm, char *str)
 219 {
 220    int len = strlen(str) + 1;
 221
 222    *pm = check_pool_memory_size(*pm, len);
 223    memcpy(*pm, str, len);
 224 }
 225
 226
 227 /*
 228  * Convert a JobStatus code into a human readable form
 229  */
 230 void jobstatus_to_ascii(int JobStatus, char *msg, int maxlen)
 231 {
 232    char *termstat, jstat[2];
 233
 234    switch (JobStatus) {
 235       case JS_Terminated:
 236          termstat = _("OK");
 237          break;
 238      case JS_FatalError:
 239      case JS_ErrorTerminated:
 240          termstat = _("Error");
 241          break;
 242      case JS_Error:
 243          termstat = _("Non-fatal error");
 244          break;
 245      case JS_Canceled:
 246          termstat = _("Canceled");
 247          break;
 248      case JS_Differences:
 249          termstat = _("Verify differences");
 250          break;
 251      default:
 252          jstat[0] = last_job.JobStatus;
 253          jstat[1] = 0;
 254          termstat = jstat;
 255          break;
 256    }
 257    bstrncpy(msg, termstat, maxlen);
 258 }
 259
 260 /*
 261  * Convert Job Termination Status into a string
 262  */
 263 char *job_status_to_str(int stat)
 264 {
 265    char *str;
 266
 267    switch (stat) {
 268    case JS_Terminated:
 269       str = _("OK");
 270       break;
 271    case JS_ErrorTerminated:
 272    case JS_Error:
 273       str = _("Error");
 274       break;
 275    case JS_FatalError:
 276       str = _("Fatal Error");
 277       break;
 278    case JS_Canceled:
 279       str = _("Canceled");
 280       break;
 281    case JS_Differences:
 282       str = _("Differences");
 283       break;
 284    default:
 285       str = _("Unknown term code");
 286       break;
 287    }
 288    return str;
 289 }
 290
 291
 292 /*
 293  * Convert Job Type into a string
 294  */
 295 char *job_type_to_str(int type)
 296 {
 297    char *str;
 298
 299    switch (type) {
 300    case JT_BACKUP:
 301       str = _("Backup");
 302       break;
 303    case JT_VERIFY:
 304       str = _("Verify");
 305       break;
 306    case JT_RESTORE:
 307       str = _("Restore");
 308       break;
 309    case JT_ADMIN:
 310       str = _("Admin");
 311       break;
 312    default:
 313       str = _("Unknown Type");
 314       break;
 315    }
 316    return str;
 317 }
 318
 319 /*
 320  * Convert Job Level into a string
 321  */
 322 char *job_level_to_str(int level)
 323 {
 324    char *str;
 325
 326    switch (level) {
 327    case L_BASE:
 328       str = _("Base");
 329    case L_FULL:
 330       str = _("Full");
 331       break;
 332    case L_INCREMENTAL:
 333       str = _("Incremental");
 334       break;
 335    case L_DIFFERENTIAL:
 336       str = _("Differential");
 337       break;
 338    case L_SINCE:
 339       str = _("Since");
 340       break;
 341    case L_VERIFY_CATALOG:
 342       str = _("Verify Catalog");
 343       break;
 344    case L_VERIFY_INIT:
 345       str = _("Verify Init Catalog");
 346       break;
 347    case L_VERIFY_VOLUME_TO_CATALOG:
 348       str = _("Verify Volume to Catalog");
 349       break;
 350    case L_VERIFY_DATA:
 351       str = _("Verify Data");
 352       break;
 353    default:
 354       str = _("Unknown Job Level");
 355       break;
 356    }
 357    return str;
 358 }
 359
 360
 361 /***********************************************************************
 362  * Encode the mode bits into a 10 character string like LS does
 363  ***********************************************************************/
 364
 365 char *encode_mode(mode_t mode, char *buf)
 366 {
 367   char *cp = buf;
 368
 369   *cp++ = S_ISDIR(mode) ? 'd' : S_ISBLK(mode)  ? 'b' : S_ISCHR(mode)  ? 'c' :
 370           S_ISLNK(mode) ? 'l' : S_ISFIFO(mode) ? 'f' : S_ISSOCK(mode) ? 's' : '-';
 371   *cp++ = mode & S_IRUSR ? 'r' : '-';
 372   *cp++ = mode & S_IWUSR ? 'w' : '-';
 373   *cp++ = (mode & S_ISUID
 374                ? (mode & S_IXUSR ? 's' : 'S')
 375                : (mode & S_IXUSR ? 'x' : '-'));
 376   *cp++ = mode & S_IRGRP ? 'r' : '-';
 377   *cp++ = mode & S_IWGRP ? 'w' : '-';
 378   *cp++ = (mode & S_ISGID
 379                ? (mode & S_IXGRP ? 's' : 'S')
 380                : (mode & S_IXGRP ? 'x' : '-'));
 381   *cp++ = mode & S_IROTH ? 'r' : '-';
 382   *cp++ = mode & S_IWOTH ? 'w' : '-';
 383   *cp++ = (mode & S_ISVTX
 384                ? (mode & S_IXOTH ? 't' : 'T')
 385                : (mode & S_IXOTH ? 'x' : '-'));
 386   *cp = '\0';
 387   return cp;
 388 }
 389
 390
 391 int do_shell_expansion(char *name, int name_len)
 392 {
 393 /*  ****FIXME***** this should work for Win32 too */
 394 #define UNIX
 395 #ifdef UNIX
 396 #ifndef PATH_MAX
 397 #define PATH_MAX 512
 398 #endif
 399
 400    int pid, wpid, stat;
 401    int waitstatus;
 402    char *shellcmd;
 403    int i;
 404    char echout[PATH_MAX + 256];
 405    int pfd[2];
 406    static char meta[] = "~\\$[]*?`'<>\"";
 407    int found = FALSE;
 408    int len;
 409
 410    /* Check if any meta characters are present */
 411    len = strlen(meta);
 412    for (i = 0; i < len; i++) {
 413       if (strchr(name, meta[i])) {
 414          found = TRUE;
 415          break;
 416       }
 417    }
 418    stat = 0;
 419    if (found) {
 420 #ifdef nt
 421        /* If the filename appears to be a DOS filename,
 422           convert all backward slashes \ to Unix path
 423           separators / and insert a \ infront of spaces. */
 424        len = strlen(name);
 425        if (len >= 3 && name[1] == ':' && name[2] == '\\') {
 426           for (i=2; i<len; i++)
 427              if (name[i] == '\\')
 428                 name[i] = '/';
 429        }
 430 #else
 431        /* Pass string off to the shell for interpretation */
 432        if (pipe(pfd) == -1)
 433           return 0;
 434        switch(pid = fork()) {
 435        case -1:
 436           break;
 437
 438        case 0:                            /* child */
 439           /* look for shell */
 440           if ((shellcmd = getenv("SHELL")) == NULL) {
 441              shellcmd = "/bin/sh";
 442           }
 443           close(1); dup(pfd[1]);          /* attach pipes to stdin and stdout */
 444           close(2); dup(pfd[1]);
 445           for (i = 3; i < 32; i++)        /* close everything else */
 446              close(i);
 447           strcpy(echout, "echo ");        /* form echo command */
 448           bstrncat(echout, name, sizeof(echout));
 449           execl(shellcmd, shellcmd, "-c", echout, NULL); /* give to shell */
 450           exit(127);                      /* shouldn't get here */
 451
 452        default:                           /* parent */
 453           /* read output from child */
 454           echout[0] = 0;
 455           i = read(pfd[0], echout, sizeof echout);
 456           if (i > 0) {
 457              echout[--i] = 0;                /* set end of string */
 458              /* look for first line. */
 459              while (--i >= 0) {
 460                 if (echout[i] == '\n') {
 461                    echout[i] = 0;            /* keep only first one */
 462                 }
 463              }
 464           }
 465           /* wait for child to exit */
 466           while ((wpid = wait(&waitstatus)) != pid && wpid != -1)
 467              { ; }
 468           strip_trailing_junk(echout);
 469           if (strlen(echout) > 0) {
 470              bstrncpy(name, echout, name_len);
 471           }
 472           stat = 1;
 473           break;
 474        }
 475        close(pfd[0]);                     /* close pipe */
 476        close(pfd[1]);
 477 #endif /* nt */
 478    }
 479    return stat;
 480
 481 #endif /* UNIX */
 482
 483 #if  MSC | MSDOS | __WATCOMC__
 484
 485    char prefix[100], *env, *getenv();
 486
 487    /* Home directory reference? */
 488    if (*name == '~' && (env=getenv("HOME"))) {
 489       strcpy(prefix, env);            /* copy HOME directory name */
 490       name++;                         /* skip over ~ in name */
 491       strcat(prefix, name);
 492       name--;                         /* get back to beginning */
 493       strcpy(name, prefix);           /* move back into name */
 494    }
 495    return 1;
 496 #endif
 497
 498 }
 499
 500
 501 /*  MAKESESSIONKEY  --  Generate session key with optional start
 502                         key.  If mode is TRUE, the key will be
 503                         translated to a string, otherwise it is
 504                         returned as 16 binary bytes.
 505
 506     from SpeakFreely by John Walker */
 507
 508 void makeSessionKey(char *key, char *seed, int mode)
 509 {
 510      int j, k;
 511      struct MD5Context md5c;
 512      unsigned char md5key[16], md5key1[16];
 513      char s[1024];
 514
 515      s[0] = 0;
 516      if (seed != NULL) {
 517         strcat(s, seed);
 518      }
 519
 520      /* The following creates a seed for the session key generator
 521         based on a collection of volatile and environment-specific
 522         information unlikely to be vulnerable (as a whole) to an
 523         exhaustive search attack.  If one of these items isn't
 524         available on your machine, replace it with something
 525         equivalent or, if you like, just delete it. */
 526
 527      sprintf(s + strlen(s), "%lu", (unsigned long) getpid());
 528      sprintf(s + strlen(s), "%lu", (unsigned long) getppid());
 529      getcwd(s + strlen(s), 256);
 530      sprintf(s + strlen(s), "%lu", (unsigned long) clock());
 531      sprintf(s + strlen(s), "%lu", (unsigned long) time(NULL));
 532 #ifdef Solaris
 533      sysinfo(SI_HW_SERIAL,s + strlen(s), 12);
 534 #endif
 535 #ifdef HAVE_GETHOSTID
 536      sprintf(s + strlen(s), "%lu", (unsigned long) gethostid());
 537 #endif
 538 #ifdef HAVE_GETDOMAINNAME
 539      getdomainname(s + strlen(s), 256);
 540 #endif
 541      gethostname(s + strlen(s), 256);
 542      sprintf(s + strlen(s), "%u", (unsigned)getuid());
 543      sprintf(s + strlen(s), "%u", (unsigned)getgid());
 544      MD5Init(&md5c);
 545      MD5Update(&md5c, (unsigned char *)s, strlen(s));
 546      MD5Final(md5key, &md5c);
 547      sprintf(s + strlen(s), "%lu", (unsigned long) ((time(NULL) + 65121) ^ 0x375F));
 548      MD5Init(&md5c);
 549      MD5Update(&md5c, (unsigned char *)s, strlen(s));
 550      MD5Final(md5key1, &md5c);
 551 #define nextrand    (md5key[j] ^ md5key1[j])
 552      if (mode) {
 553         for (j = k = 0; j < 16; j++) {
 554             unsigned char rb = nextrand;
 555
 556 #define Rad16(x) ((x) + 'A')
 557             key[k++] = Rad16((rb >> 4) & 0xF);
 558             key[k++] = Rad16(rb & 0xF);
 559 #undef Rad16
 560             if (j & 1) {
 561                  key[k++] = '-';
 562             }
 563         }
 564         key[--k] = 0;
 565      } else {
 566         for (j = 0; j < 16; j++) {
 567             key[j] = nextrand;
 568         }
 569      }
 570 }
 571 #undef nextrand
 572
 573
 574
 575 /*
 576  * Edit job codes into main command line
 577  *  %% = %
 578  *  %j = Job name
 579  *  %t = Job type (Backup, ...)
 580  *  %e = Job Exit code
 581  *  %i = JobId
 582  *  %l = job level
 583  *  %c = Client's name
 584  *  %r = Recipients
 585  *  %d = Director's name
 586  *
 587  *  omsg = edited output message
 588  *  imsg = input string containing edit codes (%x)
 589  *  to = recepients list
 590  *
 591  */
 592 POOLMEM *edit_job_codes(void *mjcr, char *omsg, char *imsg, char *to)
 593 {
 594    char *p, *str;
 595    char add[20];
 596    JCR *jcr = (JCR *)mjcr;
 597
 598    *omsg = 0;
 599    Dmsg1(200, "edit_job_codes: %s\n", imsg);
 600    for (p=imsg; *p; p++) {
 601       if (*p == '%') {
 602          switch (*++p) {
 603          case '%':
 604             str = "%";
 605             break;
 606          case 'c':
 607             str = jcr->client_name;
 608             if (!str) {
 609                str = "";
 610             }
 611             break;
 612          case 'd':
 613             str = my_name;            /* Director's name */
 614             break;
 615          case 'e':
 616             str = job_status_to_str(jcr->JobStatus);
 617             break;
 618          case 'i':
 619             sprintf(add, "%d", jcr->JobId);
 620             str = add;
 621             break;
 622          case 'j':                    /* Job name */
 623             str = jcr->Job;
 624             break;
 625          case 'l':
 626             str = job_level_to_str(jcr->JobLevel);
 627             break;
 628          case 'r':
 629             str = to;
 630             break;
 631          case 't':
 632             str = job_type_to_str(jcr->JobType);
 633             break;
 634          default:
 635             add[0] = '%';
 636             add[1] = *p;
 637             add[2] = 0;
 638             str = add;
 639             break;
 640          }
 641       } else {
 642          add[0] = *p;
 643          add[1] = 0;
 644          str = add;
 645       }
 646       Dmsg1(1200, "add_str %s\n", str);
 647       pm_strcat(&omsg, str);
 648       Dmsg1(1200, "omsg=%s\n", omsg);
 649    }
 650    return omsg;
 651 }
 652
 653 /*
 654  * Return next argument from command line.  Note, this
 655  * routine is destructive.
 656  */
 657 char *next_arg(char **s)
 658 {
 659    char *p, *q, *n;
 660    int in_quote = 0;
 661
 662    /* skip past spaces to next arg */
 663    for (p=*s; *p && *p == ' '; ) {
 664       p++;
 665    }
 666    Dmsg1(400, "Next arg=%s\n", p);
 667    for (n = q = p; *p ; ) {
 668       if (*p == '\\') {
 669          p++;
 670          if (*p) {
 671             *q++ = *p++;
 672          } else {
 673             *q++ = *p;
 674          }
 675          continue;
 676       }
 677       if (*p == '"') {                  /* start or end of quote */
 678          if (in_quote) {
 679             p++;                        /* skip quote */
 680             in_quote = 0;
 681             continue;
 682          }
 683          in_quote = 1;
 684          p++;
 685          continue;
 686       }
 687       if (!in_quote && *p == ' ') {     /* end of field */
 688          p++;
 689          break;
 690       }
 691       *q++ = *p++;
 692    }
 693    *q = 0;
 694    *s = p;
 695    Dmsg2(400, "End arg=%s next=%s\n", n, p);
 696    return n;
 697 }
 698
 699 /*
 700  * This routine parses the input command line.
 701  * It makes a copy in args, then builds an
 702  *  argc, argv like list where
 703  *
 704  *  argc = count of arguments
 705  *  argk[i] = argument keyword (part preceding =)
 706  *  argv[i] = argument value (part after =)
 707  *
 708  *  example:  arg1 arg2=abc arg3=
 709  *
 710  *  argc = c
 711  *  argk[0] = arg1
 712  *  argv[0] = NULL
 713  *  argk[1] = arg2
 714  *  argv[1] = abc
 715  *  argk[2] = arg3
 716  *  argv[2] =
 717  */
 718
 719 void parse_command_args(POOLMEM *cmd, POOLMEM *args, int *argc,
 720                         char **argk, char **argv)
 721 {
 722    char *p, *q, *n;
 723    int len;
 724
 725    len = strlen(cmd) + 1;
 726    args = check_pool_memory_size(args, len);
 727    bstrncpy(args, cmd, len);
 728    strip_trailing_junk(args);
 729    *argc = 0;
 730    p = args;
 731    /* Pick up all arguments */
 732    while (*argc < MAX_CMD_ARGS) {
 733       n = next_arg(&p);
 734       if (*n) {
 735          argk[*argc] = n;
 736          argv[(*argc)++] = NULL;
 737       } else {
 738          break;
 739       }
 740    }
 741    /* Separate keyword and value */
 742    for (int i=0; i < *argc; i++) {
 743       p = strchr(argk[i], '=');
 744       if (p) {
 745          *p++ = 0;                    /* terminate keyword and point to value */
 746          /* Unquote quoted values */
 747          if (*p == '"') {
 748             for (n = q = ++p; *p && *p != '"'; ) {
 749                if (*p == '\\') {
 750                   p++;
 751                }
 752                *q++ = *p++;
 753             }
 754             *q = 0;                   /* terminate string */
 755             p = n;                    /* point to string */
 756          }
 757          if (strlen(p) > MAX_NAME_LENGTH-1) {
 758             p[MAX_NAME_LENGTH-1] = 0; /* truncate to max len */
 759          }
 760       }
 761       argv[i] = p;                    /* save ptr to value or NULL */
 762    }
 763 #ifdef xxxx
 764    for (i=0; i<argc; i++) {
 765       Dmsg3(000, "Arg %d: kw=%s val=%s\n", i, argk[i], argv[i]?argv[i]:"NULL");
 766    }
 767 #endif
 768 }
 769
 770 void set_working_directory(char *wd)
 771 {
 772    struct stat stat_buf;
 773
 774    if (wd == NULL) {
 775       Emsg0(M_ERROR_TERM, 0, _("Working directory not defined. Cannot continue.\n"));
 776    }
 777    if (stat(wd, &stat_buf) != 0) {
 778       Emsg1(M_ERROR_TERM, 0, _("Working Directory: \"%s\" not found. Cannot continue.\n"),
 779          wd);
 780    }
 781    if (!S_ISDIR(stat_buf.st_mode)) {
 782       Emsg1(M_ERROR_TERM, 0, _("Working Directory: \"%s\" is not a directory. Cannot continue.\n"),
 783          wd);
 784    }
 785    working_directory = wd;            /* set global */
 786 }