contrib/file/apprentice.c

   1 /*
   2  * Copyright (c) Ian F. Darwin 1986-1995.
   3  * Software written by Ian F. Darwin and others;
   4  * maintained 1995-present by Christos Zoulas and others.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice immediately at the beginning of the file, without modification,
  11  *    this list of conditions, and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
  20  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28 /*
  29  * apprentice - make one pass through /etc/magic, learning its secrets.
  30  */
  31
  32 #include "file.h"
  33
  34 #ifndef lint
  35 FILE_RCSID("@(#)$File: apprentice.c,v 1.173 2011/12/08 12:38:24 rrt Exp $")
  36 #endif  /* lint */
  37
  38 #include "magic.h"
  39 #include <stdlib.h>
  40 #ifdef HAVE_UNISTD_H
  41 #include <unistd.h>
  42 #endif
  43 #include <string.h>
  44 #include <assert.h>
  45 #include <ctype.h>
  46 #include <fcntl.h>
  47 #ifdef QUICK
  48 #include <sys/mman.h>
  49 #endif
  50 #include <dirent.h>
  51
  52 #define EATAB {while (isascii((unsigned char) *l) && \
  53                       isspace((unsigned char) *l))  ++l;}
  54 #define LOWCASE(l) (isupper((unsigned char) (l)) ? \
  55                         tolower((unsigned char) (l)) : (l))
  56 /*
  57  * Work around a bug in headers on Digital Unix.
  58  * At least confirmed for: OSF1 V4.0 878
  59  */
  60 #if defined(__osf__) && defined(__DECC)
  61 #ifdef MAP_FAILED
  62 #undef MAP_FAILED
  63 #endif
  64 #endif
  65
  66 #ifndef MAP_FAILED
  67 #define MAP_FAILED (void *) -1
  68 #endif
  69
  70 #ifndef MAP_FILE
  71 #define MAP_FILE 0
  72 #endif
  73
  74 struct magic_entry {
  75         struct magic *mp;
  76         uint32_t cont_count;
  77         uint32_t max_count;
  78 };
  79
  80 int file_formats[FILE_NAMES_SIZE];
  81 const size_t file_nformats = FILE_NAMES_SIZE;
  82 const char *file_names[FILE_NAMES_SIZE];
  83 const size_t file_nnames = FILE_NAMES_SIZE;
  84
  85 private int getvalue(struct magic_set *ms, struct magic *, const char **, int);
  86 private int hextoint(int);
  87 private const char *getstr(struct magic_set *, struct magic *, const char *,
  88     int);
  89 private int parse(struct magic_set *, struct magic_entry **, uint32_t *,
  90     const char *, size_t, int);
  91 private void eatsize(const char **);
  92 private int apprentice_1(struct magic_set *, const char *, int, struct mlist *);
  93 private size_t apprentice_magic_strength(const struct magic *);
  94 private int apprentice_sort(const void *, const void *);
  95 #ifndef COMPILE_ONLY
  96 private void apprentice_list(struct mlist *, int );
  97 #endif
  98 private int apprentice_load(struct magic_set *, struct magic **, uint32_t *,
  99     const char *, int);
 100 #ifndef COMPILE_ONLY
 101 private void byteswap(struct magic *, uint32_t);
 102 private void bs1(struct magic *);
 103 private uint16_t swap2(uint16_t);
 104 private uint32_t swap4(uint32_t);
 105 private uint64_t swap8(uint64_t);
 106 #endif
 107 private char *mkdbname(struct magic_set *, const char *, int);
 108 #ifndef COMPILE_ONLY
 109 private int apprentice_map(struct magic_set *, struct magic **, uint32_t *,
 110     const char *);
 111 #endif
 112 private int apprentice_compile(struct magic_set *, struct magic **, uint32_t *,
 113     const char *);
 114 private int check_format_type(const char *, int);
 115 private int check_format(struct magic_set *, struct magic *);
 116 private int get_op(char);
 117 private int parse_mime(struct magic_set *, struct magic_entry *, const char *);
 118 private int parse_strength(struct magic_set *, struct magic_entry *, const char *);
 119 private int parse_apple(struct magic_set *, struct magic_entry *, const char *);
 120
 121
 122 private size_t maxmagic = 0;
 123 private size_t magicsize = sizeof(struct magic);
 124
 125 private const char usg_hdr[] = "cont\toffset\ttype\topcode\tmask\tvalue\tdesc";
 126
 127 private struct {
 128         const char *name;
 129         size_t len;
 130         int (*fun)(struct magic_set *, struct magic_entry *, const char *);
 131 } bang[] = {
 132 #define DECLARE_FIELD(name) { # name, sizeof(# name) - 1, parse_ ## name }
 133         DECLARE_FIELD(mime),
 134         DECLARE_FIELD(apple),
 135         DECLARE_FIELD(strength),
 136 #undef  DECLARE_FIELD
 137         { NULL, 0, NULL }
 138 };
 139
 140 #ifdef COMPILE_ONLY
 141
 142 int main(int, char *[]);
 143
 144 int
 145 main(int argc, char *argv[])
 146 {
 147         int ret;
 148         struct magic_set *ms;
 149         char *progname;
 150
 151         if ((progname = strrchr(argv[0], '/')) != NULL)
 152                 progname++;
 153         else
 154                 progname = argv[0];
 155
 156         if (argc != 2) {
 157                 (void)fprintf(stderr, "Usage: %s file\n", progname);
 158                 return 1;
 159         }
 160
 161         if ((ms = magic_open(MAGIC_CHECK)) == NULL) {
 162                 (void)fprintf(stderr, "%s: %s\n", progname, strerror(errno));
 163                 return 1;
 164         }
 165         ret = magic_compile(ms, argv[1]) == -1 ? 1 : 0;
 166         if (ret == 1)
 167                 (void)fprintf(stderr, "%s: %s\n", progname, magic_error(ms));
 168         magic_close(ms);
 169         return ret;
 170 }
 171 #endif /* COMPILE_ONLY */
 172
 173 static const struct type_tbl_s {
 174         const char name[16];
 175         const size_t len;
 176         const int type;
 177         const int format;
 178 } type_tbl[] = {
 179 # define XX(s)          s, (sizeof(s) - 1)
 180 # define XX_NULL        "", 0
 181         { XX("byte"),           FILE_BYTE,              FILE_FMT_NUM },
 182         { XX("short"),          FILE_SHORT,             FILE_FMT_NUM },
 183         { XX("default"),        FILE_DEFAULT,           FILE_FMT_STR },
 184         { XX("long"),           FILE_LONG,              FILE_FMT_NUM },
 185         { XX("string"),         FILE_STRING,            FILE_FMT_STR },
 186         { XX("date"),           FILE_DATE,              FILE_FMT_STR },
 187         { XX("beshort"),        FILE_BESHORT,           FILE_FMT_NUM },
 188         { XX("belong"),         FILE_BELONG,            FILE_FMT_NUM },
 189         { XX("bedate"),         FILE_BEDATE,            FILE_FMT_STR },
 190         { XX("leshort"),        FILE_LESHORT,           FILE_FMT_NUM },
 191         { XX("lelong"),         FILE_LELONG,            FILE_FMT_NUM },
 192         { XX("ledate"),         FILE_LEDATE,            FILE_FMT_STR },
 193         { XX("pstring"),        FILE_PSTRING,           FILE_FMT_STR },
 194         { XX("ldate"),          FILE_LDATE,             FILE_FMT_STR },
 195         { XX("beldate"),        FILE_BELDATE,           FILE_FMT_STR },
 196         { XX("leldate"),        FILE_LELDATE,           FILE_FMT_STR },
 197         { XX("regex"),          FILE_REGEX,             FILE_FMT_STR },
 198         { XX("bestring16"),     FILE_BESTRING16,        FILE_FMT_STR },
 199         { XX("lestring16"),     FILE_LESTRING16,        FILE_FMT_STR },
 200         { XX("search"),         FILE_SEARCH,            FILE_FMT_STR },
 201         { XX("medate"),         FILE_MEDATE,            FILE_FMT_STR },
 202         { XX("meldate"),        FILE_MELDATE,           FILE_FMT_STR },
 203         { XX("melong"),         FILE_MELONG,            FILE_FMT_NUM },
 204         { XX("quad"),           FILE_QUAD,              FILE_FMT_QUAD },
 205         { XX("lequad"),         FILE_LEQUAD,            FILE_FMT_QUAD },
 206         { XX("bequad"),         FILE_BEQUAD,            FILE_FMT_QUAD },
 207         { XX("qdate"),          FILE_QDATE,             FILE_FMT_STR },
 208         { XX("leqdate"),        FILE_LEQDATE,           FILE_FMT_STR },
 209         { XX("beqdate"),        FILE_BEQDATE,           FILE_FMT_STR },
 210         { XX("qldate"),         FILE_QLDATE,            FILE_FMT_STR },
 211         { XX("leqldate"),       FILE_LEQLDATE,          FILE_FMT_STR },
 212         { XX("beqldate"),       FILE_BEQLDATE,          FILE_FMT_STR },
 213         { XX("float"),          FILE_FLOAT,             FILE_FMT_FLOAT },
 214         { XX("befloat"),        FILE_BEFLOAT,           FILE_FMT_FLOAT },
 215         { XX("lefloat"),        FILE_LEFLOAT,           FILE_FMT_FLOAT },
 216         { XX("double"),         FILE_DOUBLE,            FILE_FMT_DOUBLE },
 217         { XX("bedouble"),       FILE_BEDOUBLE,          FILE_FMT_DOUBLE },
 218         { XX("ledouble"),       FILE_LEDOUBLE,          FILE_FMT_DOUBLE },
 219         { XX("leid3"),          FILE_LEID3,             FILE_FMT_NUM },
 220         { XX("beid3"),          FILE_BEID3,             FILE_FMT_NUM },
 221         { XX("indirect"),       FILE_INDIRECT,          FILE_FMT_NONE },
 222         { XX_NULL,              FILE_INVALID,           FILE_FMT_NONE },
 223 # undef XX
 224 # undef XX_NULL
 225 };
 226
 227 private int
 228 get_type(const char *l, const char **t)
 229 {
 230         const struct type_tbl_s *p;
 231
 232         for (p = type_tbl; p->len; p++) {
 233                 if (strncmp(l, p->name, p->len) == 0) {
 234                         if (t)
 235                                 *t = l + p->len;
 236                         break;
 237                 }
 238         }
 239         return p->type;
 240 }
 241
 242 private void
 243 init_file_tables(void)
 244 {
 245         static int done = 0;
 246         const struct type_tbl_s *p;
 247
 248         if (done)
 249                 return;
 250         done++;
 251
 252         for (p = type_tbl; p->len; p++) {
 253                 assert(p->type < FILE_NAMES_SIZE);
 254                 file_names[p->type] = p->name;
 255                 file_formats[p->type] = p->format;
 256         }
 257 }
 258
 259 /*
 260  * Handle one file or directory.
 261  */
 262 private int
 263 apprentice_1(struct magic_set *ms, const char *fn, int action,
 264     struct mlist *mlist)
 265 {
 266         struct magic *magic = NULL;
 267         uint32_t nmagic = 0;
 268 #ifndef COMPILE_ONLY
 269         struct mlist *ml;
 270 #endif
 271         int rv = -1;
 272 #ifndef COMPILE_ONLY
 273         int mapped;
 274 #endif
 275
 276         if (magicsize != FILE_MAGICSIZE) {
 277                 file_error(ms, 0, "magic element size %lu != %lu",
 278                     (unsigned long)sizeof(*magic),
 279                     (unsigned long)FILE_MAGICSIZE);
 280                 return -1;
 281         }
 282
 283         if (action == FILE_COMPILE) {
 284                 rv = apprentice_load(ms, &magic, &nmagic, fn, action);
 285                 if (rv != 0)
 286                         return -1;
 287                 rv = apprentice_compile(ms, &magic, &nmagic, fn);
 288                 free(magic);
 289                 return rv;
 290         }
 291
 292 #ifndef COMPILE_ONLY
 293         if ((rv = apprentice_map(ms, &magic, &nmagic, fn)) == -1) {
 294                 if (ms->flags & MAGIC_CHECK)
 295                         file_magwarn(ms, "using regular magic file `%s'", fn);
 296                 rv = apprentice_load(ms, &magic, &nmagic, fn, action);
 297                 if (rv != 0)
 298                         return -1;
 299         }
 300
 301         mapped = rv;
 302
 303         if (magic == NULL) {
 304                 file_delmagic(magic, mapped, nmagic);
 305                 return -1;
 306         }
 307
 308         if ((ml = CAST(struct mlist *, malloc(sizeof(*ml)))) == NULL) {
 309                 file_delmagic(magic, mapped, nmagic);
 310                 file_oomem(ms, sizeof(*ml));
 311                 return -1;
 312         }
 313
 314         ml->magic = magic;
 315         ml->nmagic = nmagic;
 316         ml->mapped = mapped;
 317
 318         mlist->prev->next = ml;
 319         ml->prev = mlist->prev;
 320         ml->next = mlist;
 321         mlist->prev = ml;
 322
 323         if (action == FILE_LIST) {
 324                 printf("Binary patterns:\n");
 325                 apprentice_list(mlist, BINTEST);
 326                 printf("Text patterns:\n");
 327                 apprentice_list(mlist, TEXTTEST);
 328         }
 329 #endif /* COMPILE_ONLY */
 330         return 0;
 331 }
 332
 333 protected void
 334 file_delmagic(struct magic *p, int type, size_t entries)
 335 {
 336         if (p == NULL)
 337                 return;
 338         switch (type) {
 339         case 2:
 340 #ifdef QUICK
 341                 p--;
 342                 (void)munmap((void *)p, sizeof(*p) * (entries + 1));
 343                 break;
 344 #else
 345                 (void)&entries;
 346                 abort();
 347                 /*NOTREACHED*/
 348 #endif
 349         case 1:
 350                 p--;
 351                 /*FALLTHROUGH*/
 352         case 0:
 353                 free(p);
 354                 break;
 355         default:
 356                 abort();
 357         }
 358 }
 359
 360 /* const char *fn: list of magic files and directories */
 361 protected struct mlist *
 362 file_apprentice(struct magic_set *ms, const char *fn, int action)
 363 {
 364         char *p, *mfn;
 365         int file_err, errs = -1;
 366         struct mlist *mlist;
 367
 368         if ((fn = magic_getpath(fn, action)) == NULL)
 369                 return NULL;
 370
 371         init_file_tables();
 372
 373         if ((mfn = strdup(fn)) == NULL) {
 374                 file_oomem(ms, strlen(fn));
 375                 return NULL;
 376         }
 377         fn = mfn;
 378
 379         if ((mlist = CAST(struct mlist *, malloc(sizeof(*mlist)))) == NULL) {
 380                 free(mfn);
 381                 file_oomem(ms, sizeof(*mlist));
 382                 return NULL;
 383         }
 384         mlist->next = mlist->prev = mlist;
 385
 386         while (fn) {
 387                 p = strchr(fn, PATHSEP);
 388                 if (p)
 389                         *p++ = '\0';
 390                 if (*fn == '\0')
 391                         break;
 392                 file_err = apprentice_1(ms, fn, action, mlist);
 393                 errs = MAX(errs, file_err);
 394                 fn = p;
 395         }
 396         if (errs == -1) {
 397                 free(mfn);
 398                 free(mlist);
 399                 mlist = NULL;
 400                 file_error(ms, 0, "could not find any magic files!");
 401                 return NULL;
 402         }
 403         free(mfn);
 404         return mlist;
 405 }
 406
 407 /*
 408  * Get weight of this magic entry, for sorting purposes.
 409  */
 410 private size_t
 411 apprentice_magic_strength(const struct magic *m)
 412 {
 413 #define MULT 10
 414         size_t val = 2 * MULT;  /* baseline strength */
 415
 416         switch (m->type) {
 417         case FILE_DEFAULT:      /* make sure this sorts last */
 418                 if (m->factor_op != FILE_FACTOR_OP_NONE)
 419                         abort();
 420                 return 0;
 421
 422         case FILE_BYTE:
 423                 val += 1 * MULT;
 424                 break;
 425
 426         case FILE_SHORT:
 427         case FILE_LESHORT:
 428         case FILE_BESHORT:
 429                 val += 2 * MULT;
 430                 break;
 431
 432         case FILE_LONG:
 433         case FILE_LELONG:
 434         case FILE_BELONG:
 435         case FILE_MELONG:
 436                 val += 4 * MULT;
 437                 break;
 438
 439         case FILE_PSTRING:
 440         case FILE_STRING:
 441                 val += m->vallen * MULT;
 442                 break;
 443
 444         case FILE_BESTRING16:
 445         case FILE_LESTRING16:
 446                 val += m->vallen * MULT / 2;
 447                 break;
 448
 449         case FILE_SEARCH:
 450         case FILE_REGEX:
 451                 val += m->vallen * MAX(MULT / m->vallen, 1);
 452                 break;
 453
 454         case FILE_DATE:
 455         case FILE_LEDATE:
 456         case FILE_BEDATE:
 457         case FILE_MEDATE:
 458         case FILE_LDATE:
 459         case FILE_LELDATE:
 460         case FILE_BELDATE:
 461         case FILE_MELDATE:
 462         case FILE_FLOAT:
 463         case FILE_BEFLOAT:
 464         case FILE_LEFLOAT:
 465                 val += 4 * MULT;
 466                 break;
 467
 468         case FILE_QUAD:
 469         case FILE_BEQUAD:
 470         case FILE_LEQUAD:
 471         case FILE_QDATE:
 472         case FILE_LEQDATE:
 473         case FILE_BEQDATE:
 474         case FILE_QLDATE:
 475         case FILE_LEQLDATE:
 476         case FILE_BEQLDATE:
 477         case FILE_DOUBLE:
 478         case FILE_BEDOUBLE:
 479         case FILE_LEDOUBLE:
 480                 val += 8 * MULT;
 481                 break;
 482
 483         default:
 484                 val = 0;
 485                 (void)fprintf(stderr, "Bad type %d\n", m->type);
 486                 abort();
 487         }
 488
 489         switch (m->reln) {
 490         case 'x':       /* matches anything penalize */
 491         case '!':       /* matches almost anything penalize */
 492                 val = 0;
 493                 break;
 494
 495         case '=':       /* Exact match, prefer */
 496                 val += MULT;
 497                 break;
 498
 499         case '>':
 500         case '<':       /* comparison match reduce strength */
 501                 val -= 2 * MULT;
 502                 break;
 503
 504         case '^':
 505         case '&':       /* masking bits, we could count them too */
 506                 val -= MULT;
 507                 break;
 508
 509         default:
 510                 (void)fprintf(stderr, "Bad relation %c\n", m->reln);
 511                 abort();
 512         }
 513
 514         if (val == 0)   /* ensure we only return 0 for FILE_DEFAULT */
 515                 val = 1;
 516
 517         switch (m->factor_op) {
 518         case FILE_FACTOR_OP_NONE:
 519                 break;
 520         case FILE_FACTOR_OP_PLUS:
 521                 val += m->factor;
 522                 break;
 523         case FILE_FACTOR_OP_MINUS:
 524                 val -= m->factor;
 525                 break;
 526         case FILE_FACTOR_OP_TIMES:
 527                 val *= m->factor;
 528                 break;
 529         case FILE_FACTOR_OP_DIV:
 530                 val /= m->factor;
 531                 break;
 532         default:
 533                 abort();
 534         }
 535
 536         /*
 537          * Magic entries with no description get a bonus because they depend
 538          * on subsequent magic entries to print something.
 539          */
 540         if (m->desc[0] == '\0')
 541                 val++;
 542         return val;
 543 }
 544
 545 /*
 546  * Sort callback for sorting entries by "strength" (basically length)
 547  */
 548 private int
 549 apprentice_sort(const void *a, const void *b)
 550 {
 551         const struct magic_entry *ma = CAST(const struct magic_entry *, a);
 552         const struct magic_entry *mb = CAST(const struct magic_entry *, b);
 553         size_t sa = apprentice_magic_strength(ma->mp);
 554         size_t sb = apprentice_magic_strength(mb->mp);
 555         if (sa == sb)
 556                 return 0;
 557         else if (sa > sb)
 558                 return -1;
 559         else
 560                 return 1;
 561 }
 562
 563 /*
 564  * Shows sorted patterns list in the order which is used for the matching
 565  */
 566 #ifndef COMPILE_ONLY
 567 private void
 568 apprentice_list(struct mlist *mlist, int mode)
 569 {
 570         uint32_t magindex = 0;
 571         struct mlist *ml;
 572         for (ml = mlist->next; ml != mlist; ml = ml->next) {
 573                 for (magindex = 0; magindex < ml->nmagic; magindex++) {
 574                         struct magic *m = &ml->magic[magindex];
 575                         if ((m->flag & mode) != mode) {
 576                                 /* Skip sub-tests */
 577                                 while (magindex + 1 < ml->nmagic &&
 578                                        ml->magic[magindex + 1].cont_level != 0)
 579                                         ++magindex;
 580                                 continue; /* Skip to next top-level test*/
 581                         }
 582
 583                         /*
 584                          * Try to iterate over the tree until we find item with
 585                          * description/mimetype.
 586                          */
 587                         while (magindex + 1 < ml->nmagic &&
 588                                ml->magic[magindex + 1].cont_level != 0 &&
 589                                *ml->magic[magindex].desc == '\0' &&
 590                                *ml->magic[magindex].mimetype == '\0')
 591                                 magindex++;
 592
 593                         printf("Strength = %3" SIZE_T_FORMAT "u : %s [%s]\n",
 594                             apprentice_magic_strength(m),
 595                             ml->magic[magindex].desc,
 596                             ml->magic[magindex].mimetype);
 597                 }
 598         }
 599 }
 600 #endif /* COMPILE_ONLY */
 601
 602 private void
 603 set_test_type(struct magic *mstart, struct magic *m)
 604 {
 605         switch (m->type) {
 606         case FILE_BYTE:
 607         case FILE_SHORT:
 608         case FILE_LONG:
 609         case FILE_DATE:
 610         case FILE_BESHORT:
 611         case FILE_BELONG:
 612         case FILE_BEDATE:
 613         case FILE_LESHORT:
 614         case FILE_LELONG:
 615         case FILE_LEDATE:
 616         case FILE_LDATE:
 617         case FILE_BELDATE:
 618         case FILE_LELDATE:
 619         case FILE_MEDATE:
 620         case FILE_MELDATE:
 621         case FILE_MELONG:
 622         case FILE_QUAD:
 623         case FILE_LEQUAD:
 624         case FILE_BEQUAD:
 625         case FILE_QDATE:
 626         case FILE_LEQDATE:
 627         case FILE_BEQDATE:
 628         case FILE_QLDATE:
 629         case FILE_LEQLDATE:
 630         case FILE_BEQLDATE:
 631         case FILE_FLOAT:
 632         case FILE_BEFLOAT:
 633         case FILE_LEFLOAT:
 634         case FILE_DOUBLE:
 635         case FILE_BEDOUBLE:
 636         case FILE_LEDOUBLE:
 637                 mstart->flag |= BINTEST;
 638                 break;
 639         case FILE_STRING:
 640         case FILE_PSTRING:
 641         case FILE_BESTRING16:
 642         case FILE_LESTRING16:
 643                 /* Allow text overrides */
 644                 if (mstart->str_flags & STRING_TEXTTEST)
 645                         mstart->flag |= TEXTTEST;
 646                 else
 647                         mstart->flag |= BINTEST;
 648                 break;
 649         case FILE_REGEX:
 650         case FILE_SEARCH:
 651 #ifndef COMPILE_ONLY
 652                 /* Check for override */
 653                 if (mstart->str_flags & STRING_BINTEST)
 654                         mstart->flag |= BINTEST;
 655                 if (mstart->str_flags & STRING_TEXTTEST)
 656                         mstart->flag |= TEXTTEST;
 657
 658                 if (mstart->flag & (TEXTTEST|BINTEST))
 659                         break;
 660
 661                 /* binary test if pattern is not text */
 662                 if (file_looks_utf8(m->value.us, (size_t)m->vallen, NULL,
 663                     NULL) <= 0)
 664                         mstart->flag |= BINTEST;
 665                 else
 666                         mstart->flag |= TEXTTEST;
 667 #endif
 668                 break;
 669         case FILE_DEFAULT:
 670                 /* can't deduce anything; we shouldn't see this at the
 671                    top level anyway */
 672                 break;
 673         case FILE_INVALID:
 674         default:
 675                 /* invalid search type, but no need to complain here */
 676                 break;
 677         }
 678 }
 679
 680 /*
 681  * Load and parse one file.
 682  */
 683 private void
 684 load_1(struct magic_set *ms, int action, const char *fn, int *errs,
 685    struct magic_entry **marray, uint32_t *marraycount)
 686 {
 687         size_t lineno = 0, llen = 0;
 688         char *line = NULL;
 689         ssize_t len;
 690
 691         FILE *f = fopen(ms->file = fn, "r");
 692         if (f == NULL) {
 693                 if (errno != ENOENT)
 694                         file_error(ms, errno, "cannot read magic file `%s'",
 695                                    fn);
 696                 (*errs)++;
 697                 return;
 698         }
 699
 700         /* read and parse this file */
 701         for (ms->line = 1; (len = getline(&line, &llen, f)) != -1;
 702             ms->line++) {
 703                 if (len == 0) /* null line, garbage, etc */
 704                         continue;
 705                 if (line[len - 1] == '\n') {
 706                         lineno++;
 707                         line[len - 1] = '\0'; /* delete newline */
 708                 }
 709                 switch (line[0]) {
 710                 case '\0':      /* empty, do not parse */
 711                 case '#':       /* comment, do not parse */
 712                         continue;
 713                 case '!':
 714                         if (line[1] == ':') {
 715                                 size_t i;
 716
 717                                 for (i = 0; bang[i].name != NULL; i++) {
 718                                         if ((size_t)(len - 2) > bang[i].len &&
 719                                             memcmp(bang[i].name, line + 2,
 720                                             bang[i].len) == 0)
 721                                                 break;
 722                                 }
 723                                 if (bang[i].name == NULL) {
 724                                         file_error(ms, 0,
 725                                             "Unknown !: entry `%s'", line);
 726                                         (*errs)++;
 727                                         continue;
 728                                 }
 729                                 if (*marraycount == 0) {
 730                                         file_error(ms, 0,
 731                                             "No current entry for :!%s type",
 732                                                 bang[i].name);
 733                                         (*errs)++;
 734                                         continue;
 735                                 }
 736                                 if ((*bang[i].fun)(ms,
 737                                     &(*marray)[*marraycount - 1],
 738                                     line + bang[i].len + 2) != 0) {
 739                                         (*errs)++;
 740                                         continue;
 741                                 }
 742                                 continue;
 743                         }
 744                         /*FALLTHROUGH*/
 745                 default:
 746                         if (parse(ms, marray, marraycount, line, lineno,
 747                             action) != 0)
 748                                 (*errs)++;
 749                         break;
 750                 }
 751         }
 752         free(line);
 753         (void)fclose(f);
 754 }
 755
 756 /*
 757  * parse a file or directory of files
 758  * const char *fn: name of magic file or directory
 759  */
 760 private int
 761 cmpstrp(const void *p1, const void *p2)
 762 {
 763         return strcmp(*(char *const *)p1, *(char *const *)p2);
 764 }
 765
 766 private int
 767 apprentice_load(struct magic_set *ms, struct magic **magicp, uint32_t *nmagicp,
 768     const char *fn, int action)
 769 {
 770         int errs = 0;
 771         struct magic_entry *marray;
 772         uint32_t marraycount, i, mentrycount = 0, starttest;
 773         size_t slen, files = 0, maxfiles = 0;
 774         char **filearr = NULL, *mfn;
 775         struct stat st;
 776         DIR *dir;
 777         struct dirent *d;
 778
 779         ms->flags |= MAGIC_CHECK;       /* Enable checks for parsed files */
 780
 781         maxmagic = MAXMAGIS;
 782         if ((marray = CAST(struct magic_entry *, calloc(maxmagic,
 783             sizeof(*marray)))) == NULL) {
 784                 file_oomem(ms, maxmagic * sizeof(*marray));
 785                 return -1;
 786         }
 787         marraycount = 0;
 788
 789         /* print silly verbose header for USG compat. */
 790         if (action == FILE_CHECK)
 791                 (void)fprintf(stderr, "%s\n", usg_hdr);
 792
 793         /* load directory or file */
 794         if (stat(fn, &st) == 0 && S_ISDIR(st.st_mode)) {
 795                 dir = opendir(fn);
 796                 if (!dir) {
 797                         errs++;
 798                         goto out;
 799                 }
 800                 while ((d = readdir(dir)) != NULL) {
 801                         if (asprintf(&mfn, "%s/%s", fn, d->d_name) < 0) {
 802                                 file_oomem(ms,
 803                                     strlen(fn) + strlen(d->d_name) + 2);
 804                                 errs++;
 805                                 closedir(dir);
 806                                 goto out;
 807                         }
 808                         if (stat(mfn, &st) == -1 || !S_ISREG(st.st_mode)) {
 809                                 free(mfn);
 810                                 continue;
 811                         }
 812                         if (files >= maxfiles) {
 813                                 size_t mlen;
 814                                 maxfiles = (maxfiles + 1) * 2;
 815                                 mlen = maxfiles * sizeof(*filearr);
 816                                 if ((filearr = CAST(char **,
 817                                     realloc(filearr, mlen))) == NULL) {
 818                                         file_oomem(ms, mlen);
 819                                         free(mfn);
 820                                         closedir(dir);
 821                                         errs++;
 822                                         goto out;
 823                                 }
 824                         }
 825                         filearr[files++] = mfn;
 826                 }
 827                 closedir(dir);
 828                 qsort(filearr, files, sizeof(*filearr), cmpstrp);
 829                 for (i = 0; i < files; i++) {
 830                         load_1(ms, action, filearr[i], &errs, &marray,
 831                             &marraycount);
 832                         free(filearr[i]);
 833                 }
 834                 free(filearr);
 835         } else
 836                 load_1(ms, action, fn, &errs, &marray, &marraycount);
 837         if (errs)
 838                 goto out;
 839
 840         /* Set types of tests */
 841         for (i = 0; i < marraycount; ) {
 842                 if (marray[i].mp->cont_level != 0) {
 843                         i++;
 844                         continue;
 845                 }
 846
 847                 starttest = i;
 848                 do {
 849                         static const char text[] = "text";
 850                         static const char binary[] = "binary";
 851                         static const size_t len = sizeof(text);
 852                         set_test_type(marray[starttest].mp, marray[i].mp);
 853                         if ((ms->flags & MAGIC_DEBUG) == 0)
 854                                 continue;
 855                         (void)fprintf(stderr, "%s%s%s: %s\n",
 856                             marray[i].mp->mimetype,
 857                             marray[i].mp->mimetype[0] == '\0' ? "" : "; ",
 858                             marray[i].mp->desc[0] ? marray[i].mp->desc :
 859                             "(no description)",
 860                             marray[i].mp->flag & BINTEST ? binary : text);
 861                         if (marray[i].mp->flag & BINTEST) {
 862                                 char *p = strstr(marray[i].mp->desc, text);
 863                                 if (p && (p == marray[i].mp->desc ||
 864                                     isspace((unsigned char)p[-1])) &&
 865                                     (p + len - marray[i].mp->desc ==
 866                                     MAXstring || (p[len] == '\0' ||
 867                                     isspace((unsigned char)p[len]))))
 868                                         (void)fprintf(stderr, "*** Possible "
 869                                             "binary test for text type\n");
 870                         }
 871                 } while (++i < marraycount && marray[i].mp->cont_level != 0);
 872         }
 873
 874         qsort(marray, marraycount, sizeof(*marray), apprentice_sort);
 875
 876         /*
 877          * Make sure that any level 0 "default" line is last (if one exists).
 878          */
 879         for (i = 0; i < marraycount; i++) {
 880                 if (marray[i].mp->cont_level == 0 &&
 881                     marray[i].mp->type == FILE_DEFAULT) {
 882                         while (++i < marraycount)
 883                                 if (marray[i].mp->cont_level == 0)
 884                                         break;
 885                         if (i != marraycount) {
 886                                 /* XXX - Ugh! */
 887                                 ms->line = marray[i].mp->lineno;
 888                                 file_magwarn(ms,
 889                                     "level 0 \"default\" did not sort last");
 890                         }
 891                         break;
 892                 }
 893         }
 894
 895         for (i = 0; i < marraycount; i++)
 896                 mentrycount += marray[i].cont_count;
 897
 898         slen = sizeof(**magicp) * mentrycount;
 899         if ((*magicp = CAST(struct magic *, malloc(slen))) == NULL) {
 900                 file_oomem(ms, slen);
 901                 errs++;
 902                 goto out;
 903         }
 904
 905         mentrycount = 0;
 906         for (i = 0; i < marraycount; i++) {
 907                 (void)memcpy(*magicp + mentrycount, marray[i].mp,
 908                     marray[i].cont_count * sizeof(**magicp));
 909                 mentrycount += marray[i].cont_count;
 910         }
 911 out:
 912         for (i = 0; i < marraycount; i++)
 913                 free(marray[i].mp);
 914         free(marray);
 915         if (errs) {
 916                 *magicp = NULL;
 917                 *nmagicp = 0;
 918                 return errs;
 919         } else {
 920                 *nmagicp = mentrycount;
 921                 return 0;
 922         }
 923
 924 }
 925
 926 /*
 927  * extend the sign bit if the comparison is to be signed
 928  */
 929 protected uint64_t
 930 file_signextend(struct magic_set *ms, struct magic *m, uint64_t v)
 931 {
 932         if (!(m->flag & UNSIGNED)) {
 933                 switch(m->type) {
 934                 /*
 935                  * Do not remove the casts below.  They are
 936                  * vital.  When later compared with the data,
 937                  * the sign extension must have happened.
 938                  */
 939                 case FILE_BYTE:
 940                         v = (char) v;
 941                         break;
 942                 case FILE_SHORT:
 943                 case FILE_BESHORT:
 944                 case FILE_LESHORT:
 945                         v = (short) v;
 946                         break;
 947                 case FILE_DATE:
 948                 case FILE_BEDATE:
 949                 case FILE_LEDATE:
 950                 case FILE_MEDATE:
 951                 case FILE_LDATE:
 952                 case FILE_BELDATE:
 953                 case FILE_LELDATE:
 954                 case FILE_MELDATE:
 955                 case FILE_LONG:
 956                 case FILE_BELONG:
 957                 case FILE_LELONG:
 958                 case FILE_MELONG:
 959                 case FILE_FLOAT:
 960                 case FILE_BEFLOAT:
 961                 case FILE_LEFLOAT:
 962                         v = (int32_t) v;
 963                         break;
 964                 case FILE_QUAD:
 965                 case FILE_BEQUAD:
 966                 case FILE_LEQUAD:
 967                 case FILE_QDATE:
 968                 case FILE_QLDATE:
 969                 case FILE_BEQDATE:
 970                 case FILE_BEQLDATE:
 971                 case FILE_LEQDATE:
 972                 case FILE_LEQLDATE:
 973                 case FILE_DOUBLE:
 974                 case FILE_BEDOUBLE:
 975                 case FILE_LEDOUBLE:
 976                         v = (int64_t) v;
 977                         break;
 978                 case FILE_STRING:
 979                 case FILE_PSTRING:
 980                 case FILE_BESTRING16:
 981                 case FILE_LESTRING16:
 982                 case FILE_REGEX:
 983                 case FILE_SEARCH:
 984                 case FILE_DEFAULT:
 985                 case FILE_INDIRECT:
 986                         break;
 987                 default:
 988                         if (ms->flags & MAGIC_CHECK)
 989                             file_magwarn(ms, "cannot happen: m->type=%d\n",
 990                                     m->type);
 991                         return ~0U;
 992                 }
 993         }
 994         return v;
 995 }
 996
 997 private int
 998 string_modifier_check(struct magic_set *ms, struct magic *m)
 999 {
1000         if ((ms->flags & MAGIC_CHECK) == 0)
1001                 return 0;
1002
1003         if (m->type != FILE_PSTRING && (m->str_flags & PSTRING_LEN) != 0) {
1004                 file_magwarn(ms,
1005                     "'/BHhLl' modifiers are only allowed for pascal strings\n");
1006                 return -1;
1007         }
1008         switch (m->type) {
1009         case FILE_BESTRING16:
1010         case FILE_LESTRING16:
1011                 if (m->str_flags != 0) {
1012                         file_magwarn(ms,
1013                             "no modifiers allowed for 16-bit strings\n");
1014                         return -1;
1015                 }
1016                 break;
1017         case FILE_STRING:
1018         case FILE_PSTRING:
1019                 if ((m->str_flags & REGEX_OFFSET_START) != 0) {
1020                         file_magwarn(ms,
1021                             "'/%c' only allowed on regex and search\n",
1022                             CHAR_REGEX_OFFSET_START);
1023                         return -1;
1024                 }
1025                 break;
1026         case FILE_SEARCH:
1027                 if (m->str_range == 0) {
1028                         file_magwarn(ms,
1029                             "missing range; defaulting to %d\n",
1030                             STRING_DEFAULT_RANGE);
1031                         m->str_range = STRING_DEFAULT_RANGE;
1032                         return -1;
1033                 }
1034                 break;
1035         case FILE_REGEX:
1036                 if ((m->str_flags & STRING_COMPACT_WHITESPACE) != 0) {
1037                         file_magwarn(ms, "'/%c' not allowed on regex\n",
1038                             CHAR_COMPACT_WHITESPACE);
1039                         return -1;
1040                 }
1041                 if ((m->str_flags & STRING_COMPACT_OPTIONAL_WHITESPACE) != 0) {
1042                         file_magwarn(ms, "'/%c' not allowed on regex\n",
1043                             CHAR_COMPACT_OPTIONAL_WHITESPACE);
1044                         return -1;
1045                 }
1046                 break;
1047         default:
1048                 file_magwarn(ms, "coding error: m->type=%d\n",
1049                     m->type);
1050                 return -1;
1051         }
1052         return 0;
1053 }
1054
1055 private int
1056 get_op(char c)
1057 {
1058         switch (c) {
1059         case '&':
1060                 return FILE_OPAND;
1061         case '|':
1062                 return FILE_OPOR;
1063         case '^':
1064                 return FILE_OPXOR;
1065         case '+':
1066                 return FILE_OPADD;
1067         case '-':
1068                 return FILE_OPMINUS;
1069         case '*':
1070                 return FILE_OPMULTIPLY;
1071         case '/':
1072                 return FILE_OPDIVIDE;
1073         case '%':
1074                 return FILE_OPMODULO;
1075         default:
1076                 return -1;
1077         }
1078 }
1079
1080 #ifdef ENABLE_CONDITIONALS
1081 private int
1082 get_cond(const char *l, const char **t)
1083 {
1084         static const struct cond_tbl_s {
1085                 char name[8];
1086                 size_t len;
1087                 int cond;
1088         } cond_tbl[] = {
1089                 { "if",         2,      COND_IF },
1090                 { "elif",       4,      COND_ELIF },
1091                 { "else",       4,      COND_ELSE },
1092                 { "",           0,      COND_NONE },
1093         };
1094         const struct cond_tbl_s *p;
1095
1096         for (p = cond_tbl; p->len; p++) {
1097                 if (strncmp(l, p->name, p->len) == 0 &&
1098                     isspace((unsigned char)l[p->len])) {
1099                         if (t)
1100                                 *t = l + p->len;
1101                         break;
1102                 }
1103         }
1104         return p->cond;
1105 }
1106
1107 private int
1108 check_cond(struct magic_set *ms, int cond, uint32_t cont_level)
1109 {
1110         int last_cond;
1111         last_cond = ms->c.li[cont_level].last_cond;
1112
1113         switch (cond) {
1114         case COND_IF:
1115                 if (last_cond != COND_NONE && last_cond != COND_ELIF) {
1116                         if (ms->flags & MAGIC_CHECK)
1117                                 file_magwarn(ms, "syntax error: `if'");
1118                         return -1;
1119                 }
1120                 last_cond = COND_IF;
1121                 break;
1122
1123         case COND_ELIF:
1124                 if (last_cond != COND_IF && last_cond != COND_ELIF) {
1125                         if (ms->flags & MAGIC_CHECK)
1126                                 file_magwarn(ms, "syntax error: `elif'");
1127                         return -1;
1128                 }
1129                 last_cond = COND_ELIF;
1130                 break;
1131
1132         case COND_ELSE:
1133                 if (last_cond != COND_IF && last_cond != COND_ELIF) {
1134                         if (ms->flags & MAGIC_CHECK)
1135                                 file_magwarn(ms, "syntax error: `else'");
1136                         return -1;
1137                 }
1138                 last_cond = COND_NONE;
1139                 break;
1140
1141         case COND_NONE:
1142                 last_cond = COND_NONE;
1143                 break;
1144         }
1145
1146         ms->c.li[cont_level].last_cond = last_cond;
1147         return 0;
1148 }
1149 #endif /* ENABLE_CONDITIONALS */
1150
1151 /*
1152  * parse one line from magic file, put into magic[index++] if valid
1153  */
1154 private int
1155 parse(struct magic_set *ms, struct magic_entry **mentryp, uint32_t *nmentryp,
1156     const char *line, size_t lineno, int action)
1157 {
1158 #ifdef ENABLE_CONDITIONALS
1159         static uint32_t last_cont_level = 0;
1160 #endif
1161         size_t i;
1162         struct magic_entry *me;
1163         struct magic *m;
1164         const char *l = line;
1165         char *t;
1166         int op;
1167         uint32_t cont_level;
1168
1169         cont_level = 0;
1170
1171         while (*l == '>') {
1172                 ++l;            /* step over */
1173                 cont_level++;
1174         }
1175 #ifdef ENABLE_CONDITIONALS
1176         if (cont_level == 0 || cont_level > last_cont_level)
1177                 if (file_check_mem(ms, cont_level) == -1)
1178                         return -1;
1179         last_cont_level = cont_level;
1180 #endif
1181
1182 #define ALLOC_CHUNK     (size_t)10
1183 #define ALLOC_INCR      (size_t)200
1184
1185         if (cont_level != 0) {
1186                 if (*nmentryp == 0) {
1187                         file_error(ms, 0, "No current entry for continuation");
1188                         return -1;
1189                 }
1190                 me = &(*mentryp)[*nmentryp - 1];
1191                 if (me->cont_count == me->max_count) {
1192                         struct magic *nm;
1193                         size_t cnt = me->max_count + ALLOC_CHUNK;
1194                         if ((nm = CAST(struct magic *, realloc(me->mp,
1195                             sizeof(*nm) * cnt))) == NULL) {
1196                                 file_oomem(ms, sizeof(*nm) * cnt);
1197                                 return -1;
1198                         }
1199                         me->mp = m = nm;
1200                         me->max_count = CAST(uint32_t, cnt);
1201                 }
1202                 m = &me->mp[me->cont_count++];
1203                 (void)memset(m, 0, sizeof(*m));
1204                 m->cont_level = cont_level;
1205         } else {
1206                 if (*nmentryp == maxmagic) {
1207                         struct magic_entry *mp;
1208
1209                         maxmagic += ALLOC_INCR;
1210                         if ((mp = CAST(struct magic_entry *,
1211                             realloc(*mentryp, sizeof(*mp) * maxmagic))) ==
1212                             NULL) {
1213                                 file_oomem(ms, sizeof(*mp) * maxmagic);
1214                                 return -1;
1215                         }
1216                         (void)memset(&mp[*nmentryp], 0, sizeof(*mp) *
1217                             ALLOC_INCR);
1218                         *mentryp = mp;
1219                 }
1220                 me = &(*mentryp)[*nmentryp];
1221                 if (me->mp == NULL) {
1222                         size_t len = sizeof(*m) * ALLOC_CHUNK;
1223                         if ((m = CAST(struct magic *, malloc(len))) == NULL) {
1224                                 file_oomem(ms, len);
1225                                 return -1;
1226                         }
1227                         me->mp = m;
1228                         me->max_count = ALLOC_CHUNK;
1229                 } else
1230                         m = me->mp;
1231                 (void)memset(m, 0, sizeof(*m));
1232                 m->factor_op = FILE_FACTOR_OP_NONE;
1233                 m->cont_level = 0;
1234                 me->cont_count = 1;
1235         }
1236         m->lineno = CAST(uint32_t, lineno);
1237
1238         if (*l == '&') {  /* m->cont_level == 0 checked below. */
1239                 ++l;            /* step over */
1240                 m->flag |= OFFADD;
1241         }
1242         if (*l == '(') {
1243                 ++l;            /* step over */
1244                 m->flag |= INDIR;
1245                 if (m->flag & OFFADD)
1246                         m->flag = (m->flag & ~OFFADD) | INDIROFFADD;
1247
1248                 if (*l == '&') {  /* m->cont_level == 0 checked below */
1249                         ++l;            /* step over */
1250                         m->flag |= OFFADD;
1251                 }
1252         }
1253         /* Indirect offsets are not valid at level 0. */
1254         if (m->cont_level == 0 && (m->flag & (OFFADD | INDIROFFADD)))
1255                 if (ms->flags & MAGIC_CHECK)
1256                         file_magwarn(ms, "relative offset at level 0");
1257
1258         /* get offset, then skip over it */
1259         m->offset = (uint32_t)strtoul(l, &t, 0);
1260         if (l == t)
1261                 if (ms->flags & MAGIC_CHECK)
1262                         file_magwarn(ms, "offset `%s' invalid", l);
1263         l = t;
1264
1265         if (m->flag & INDIR) {
1266                 m->in_type = FILE_LONG;
1267                 m->in_offset = 0;
1268                 /*
1269                  * read [.lbs][+-]nnnnn)
1270                  */
1271                 if (*l == '.') {
1272                         l++;
1273                         switch (*l) {
1274                         case 'l':
1275                                 m->in_type = FILE_LELONG;
1276                                 break;
1277                         case 'L':
1278                                 m->in_type = FILE_BELONG;
1279                                 break;
1280                         case 'm':
1281                                 m->in_type = FILE_MELONG;
1282                                 break;
1283                         case 'h':
1284                         case 's':
1285                                 m->in_type = FILE_LESHORT;
1286                                 break;
1287                         case 'H':
1288                         case 'S':
1289                                 m->in_type = FILE_BESHORT;
1290                                 break;
1291                         case 'c':
1292                         case 'b':
1293                         case 'C':
1294                         case 'B':
1295                                 m->in_type = FILE_BYTE;
1296                                 break;
1297                         case 'e':
1298                         case 'f':
1299                         case 'g':
1300                                 m->in_type = FILE_LEDOUBLE;
1301                                 break;
1302                         case 'E':
1303                         case 'F':
1304                         case 'G':
1305                                 m->in_type = FILE_BEDOUBLE;
1306                                 break;
1307                         case 'i':
1308                                 m->in_type = FILE_LEID3;
1309                                 break;
1310                         case 'I':
1311                                 m->in_type = FILE_BEID3;
1312                                 break;
1313                         default:
1314                                 if (ms->flags & MAGIC_CHECK)
1315                                         file_magwarn(ms,
1316                                             "indirect offset type `%c' invalid",
1317                                             *l);
1318                                 break;
1319                         }
1320                         l++;
1321                 }
1322
1323                 m->in_op = 0;
1324                 if (*l == '~') {
1325                         m->in_op |= FILE_OPINVERSE;
1326                         l++;
1327                 }
1328                 if ((op = get_op(*l)) != -1) {
1329                         m->in_op |= op;
1330                         l++;
1331                 }
1332                 if (*l == '(') {
1333                         m->in_op |= FILE_OPINDIRECT;
1334                         l++;
1335                 }
1336                 if (isdigit((unsigned char)*l) || *l == '-') {
1337                         m->in_offset = (int32_t)strtol(l, &t, 0);
1338                         if (l == t)
1339                                 if (ms->flags & MAGIC_CHECK)
1340                                         file_magwarn(ms,
1341                                             "in_offset `%s' invalid", l);
1342                         l = t;
1343                 }
1344                 if (*l++ != ')' ||
1345                     ((m->in_op & FILE_OPINDIRECT) && *l++ != ')'))
1346                         if (ms->flags & MAGIC_CHECK)
1347                                 file_magwarn(ms,
1348                                     "missing ')' in indirect offset");
1349         }
1350         EATAB;
1351
1352 #ifdef ENABLE_CONDITIONALS
1353         m->cond = get_cond(l, &l);
1354         if (check_cond(ms, m->cond, cont_level) == -1)
1355                 return -1;
1356
1357         EATAB;
1358 #endif
1359
1360         if (*l == 'u') {
1361                 ++l;
1362                 m->flag |= UNSIGNED;
1363         }
1364
1365         m->type = get_type(l, &l);
1366         if (m->type == FILE_INVALID) {
1367                 if (ms->flags & MAGIC_CHECK)
1368                         file_magwarn(ms, "type `%s' invalid", l);
1369                 return -1;
1370         }
1371
1372         /* New-style anding: "0 byte&0x80 =0x80 dynamically linked" */
1373         /* New and improved: ~ & | ^ + - * / % -- exciting, isn't it? */
1374
1375         m->mask_op = 0;
1376         if (*l == '~') {
1377                 if (!IS_STRING(m->type))
1378                         m->mask_op |= FILE_OPINVERSE;
1379                 else if (ms->flags & MAGIC_CHECK)
1380                         file_magwarn(ms, "'~' invalid for string types");
1381                 ++l;
1382         }
1383         m->str_range = 0;
1384         m->str_flags = m->type == FILE_PSTRING ? PSTRING_1_LE : 0;
1385         if ((op = get_op(*l)) != -1) {
1386                 if (!IS_STRING(m->type)) {
1387                         uint64_t val;
1388                         ++l;
1389                         m->mask_op |= op;
1390                         val = (uint64_t)strtoull(l, &t, 0);
1391                         l = t;
1392                         m->num_mask = file_signextend(ms, m, val);
1393                         eatsize(&l);
1394                 }
1395                 else if (op == FILE_OPDIVIDE) {
1396                         int have_range = 0;
1397                         while (!isspace((unsigned char)*++l)) {
1398                                 switch (*l) {
1399                                 case '0':  case '1':  case '2':
1400                                 case '3':  case '4':  case '5':
1401                                 case '6':  case '7':  case '8':
1402                                 case '9':
1403                                         if (have_range &&
1404                                             (ms->flags & MAGIC_CHECK))
1405                                                 file_magwarn(ms,
1406                                                     "multiple ranges");
1407                                         have_range = 1;
1408                                         m->str_range = CAST(uint32_t,
1409                                             strtoul(l, &t, 0));
1410                                         if (m->str_range == 0)
1411                                                 file_magwarn(ms,
1412                                                     "zero range");
1413                                         l = t - 1;
1414                                         break;
1415                                 case CHAR_COMPACT_WHITESPACE:
1416                                         m->str_flags |=
1417                                             STRING_COMPACT_WHITESPACE;
1418                                         break;
1419                                 case CHAR_COMPACT_OPTIONAL_WHITESPACE:
1420                                         m->str_flags |=
1421                                             STRING_COMPACT_OPTIONAL_WHITESPACE;
1422                                         break;
1423                                 case CHAR_IGNORE_LOWERCASE:
1424                                         m->str_flags |= STRING_IGNORE_LOWERCASE;
1425                                         break;
1426                                 case CHAR_IGNORE_UPPERCASE:
1427                                         m->str_flags |= STRING_IGNORE_UPPERCASE;
1428                                         break;
1429                                 case CHAR_REGEX_OFFSET_START:
1430                                         m->str_flags |= REGEX_OFFSET_START;
1431                                         break;
1432                                 case CHAR_BINTEST:
1433                                         m->str_flags |= STRING_BINTEST;
1434                                         break;
1435                                 case CHAR_TEXTTEST:
1436                                         m->str_flags |= STRING_TEXTTEST;
1437                                         break;
1438                                 case CHAR_PSTRING_1_LE:
1439                                         if (m->type != FILE_PSTRING)
1440                                                 goto bad;
1441                                         m->str_flags = (m->str_flags & ~PSTRING_LEN) | PSTRING_1_LE;
1442                                         break;
1443                                 case CHAR_PSTRING_2_BE:
1444                                         if (m->type != FILE_PSTRING)
1445                                                 goto bad;
1446                                         m->str_flags = (m->str_flags & ~PSTRING_LEN) | PSTRING_2_BE;
1447                                         break;
1448                                 case CHAR_PSTRING_2_LE:
1449                                         if (m->type != FILE_PSTRING)
1450                                                 goto bad;
1451                                         m->str_flags = (m->str_flags & ~PSTRING_LEN) | PSTRING_2_LE;
1452                                         break;
1453                                 case CHAR_PSTRING_4_BE:
1454                                         if (m->type != FILE_PSTRING)
1455                                                 goto bad;
1456                                         m->str_flags = (m->str_flags & ~PSTRING_LEN) | PSTRING_4_BE;
1457                                         break;
1458                                 case CHAR_PSTRING_4_LE:
1459                                         if (m->type != FILE_PSTRING)
1460                                                 goto bad;
1461                                         m->str_flags = (m->str_flags & ~PSTRING_LEN) | PSTRING_4_LE;
1462                                         break;
1463                                 case CHAR_PSTRING_LENGTH_INCLUDES_ITSELF:
1464                                         if (m->type != FILE_PSTRING)
1465                                                 goto bad;
1466                                         m->str_flags |= PSTRING_LENGTH_INCLUDES_ITSELF;
1467                                         break;
1468                                 default:
1469                                 bad:
1470                                         if (ms->flags & MAGIC_CHECK)
1471                                                 file_magwarn(ms,
1472                                                     "string extension `%c' "
1473                                                     "invalid", *l);
1474                                         return -1;
1475                                 }
1476                                 /* allow multiple '/' for readability */
1477                                 if (l[1] == '/' &&
1478                                     !isspace((unsigned char)l[2]))
1479                                         l++;
1480                         }
1481                         if (string_modifier_check(ms, m) == -1)
1482                                 return -1;
1483                 }
1484                 else {
1485                         if (ms->flags & MAGIC_CHECK)
1486                                 file_magwarn(ms, "invalid string op: %c", *t);
1487                         return -1;
1488                 }
1489         }
1490         /*
1491          * We used to set mask to all 1's here, instead let's just not do
1492          * anything if mask = 0 (unless you have a better idea)
1493          */
1494         EATAB;
1495
1496         switch (*l) {
1497         case '>':
1498         case '<':
1499                 m->reln = *l;
1500                 ++l;
1501                 if (*l == '=') {
1502                         if (ms->flags & MAGIC_CHECK) {
1503                                 file_magwarn(ms, "%c= not supported",
1504                                     m->reln);
1505                                 return -1;
1506                         }
1507                    ++l;
1508                 }
1509                 break;
1510         /* Old-style anding: "0 byte &0x80 dynamically linked" */
1511         case '&':
1512         case '^':
1513         case '=':
1514                 m->reln = *l;
1515                 ++l;
1516                 if (*l == '=') {
1517                    /* HP compat: ignore &= etc. */
1518                    ++l;
1519                 }
1520                 break;
1521         case '!':
1522                 m->reln = *l;
1523                 ++l;
1524                 break;
1525         default:
1526                 m->reln = '=';  /* the default relation */
1527                 if (*l == 'x' && ((isascii((unsigned char)l[1]) &&
1528                     isspace((unsigned char)l[1])) || !l[1])) {
1529                         m->reln = *l;
1530                         ++l;
1531                 }
1532                 break;
1533         }
1534         /*
1535          * Grab the value part, except for an 'x' reln.
1536          */
1537         if (m->reln != 'x' && getvalue(ms, m, &l, action))
1538                 return -1;
1539
1540         /*
1541          * TODO finish this macro and start using it!
1542          * #define offsetcheck {if (offset > HOWMANY-1)
1543          *      magwarn("offset too big"); }
1544          */
1545
1546         /*
1547          * Now get last part - the description
1548          */
1549         EATAB;
1550         if (l[0] == '\b') {
1551                 ++l;
1552                 m->flag |= NOSPACE;
1553         } else if ((l[0] == '\\') && (l[1] == 'b')) {
1554                 ++l;
1555                 ++l;
1556                 m->flag |= NOSPACE;
1557         }
1558         for (i = 0; (m->desc[i++] = *l++) != '\0' && i < sizeof(m->desc); )
1559                 continue;
1560         if (i == sizeof(m->desc)) {
1561                 m->desc[sizeof(m->desc) - 1] = '\0';
1562                 if (ms->flags & MAGIC_CHECK)
1563                         file_magwarn(ms, "description `%s' truncated", m->desc);
1564         }
1565
1566         /*
1567          * We only do this check while compiling, or if any of the magic
1568          * files were not compiled.
1569          */
1570         if (ms->flags & MAGIC_CHECK) {
1571                 if (check_format(ms, m) == -1)
1572                         return -1;
1573         }
1574 #ifndef COMPILE_ONLY
1575         if (action == FILE_CHECK) {
1576                 file_mdump(m);
1577         }
1578 #endif
1579         m->mimetype[0] = '\0';          /* initialise MIME type to none */
1580         if (m->cont_level == 0)
1581                 ++(*nmentryp);          /* make room for next */
1582         return 0;
1583 }
1584
1585 /*
1586  * parse a STRENGTH annotation line from magic file, put into magic[index - 1]
1587  * if valid
1588  */
1589 private int
1590 parse_strength(struct magic_set *ms, struct magic_entry *me, const char *line)
1591 {
1592         const char *l = line;
1593         char *el;
1594         unsigned long factor;
1595         struct magic *m = &me->mp[0];
1596
1597         if (m->factor_op != FILE_FACTOR_OP_NONE) {
1598                 file_magwarn(ms,
1599                     "Current entry already has a strength type: %c %d",
1600                     m->factor_op, m->factor);
1601                 return -1;
1602         }
1603         EATAB;
1604         switch (*l) {
1605         case FILE_FACTOR_OP_NONE:
1606         case FILE_FACTOR_OP_PLUS:
1607         case FILE_FACTOR_OP_MINUS:
1608         case FILE_FACTOR_OP_TIMES:
1609         case FILE_FACTOR_OP_DIV:
1610                 m->factor_op = *l++;
1611                 break;
1612         default:
1613                 file_magwarn(ms, "Unknown factor op `%c'", *l);
1614                 return -1;
1615         }
1616         EATAB;
1617         factor = strtoul(l, &el, 0);
1618         if (factor > 255) {
1619                 file_magwarn(ms, "Too large factor `%lu'", factor);
1620                 goto out;
1621         }
1622         if (*el && !isspace((unsigned char)*el)) {
1623                 file_magwarn(ms, "Bad factor `%s'", l);
1624                 goto out;
1625         }
1626         m->factor = (uint8_t)factor;
1627         if (m->factor == 0 && m->factor_op == FILE_FACTOR_OP_DIV) {
1628                 file_magwarn(ms, "Cannot have factor op `%c' and factor %u",
1629                     m->factor_op, m->factor);
1630                 goto out;
1631         }
1632         return 0;
1633 out:
1634         m->factor_op = FILE_FACTOR_OP_NONE;
1635         m->factor = 0;
1636         return -1;
1637 }
1638
1639 /*
1640  * Parse an Apple CREATOR/TYPE annotation from magic file and put it into
1641  * magic[index - 1]
1642  */
1643 private int
1644 parse_apple(struct magic_set *ms, struct magic_entry *me, const char *line)
1645 {
1646         size_t i;
1647         const char *l = line;
1648         struct magic *m = &me->mp[me->cont_count == 0 ? 0 : me->cont_count - 1];
1649
1650         if (m->apple[0] != '\0') {
1651                 file_magwarn(ms, "Current entry already has a APPLE type "
1652                     "`%.8s', new type `%s'", m->mimetype, l);
1653                 return -1;
1654         }
1655
1656         EATAB;
1657         for (i = 0; *l && ((isascii((unsigned char)*l) &&
1658             isalnum((unsigned char)*l)) || strchr("-+/.", *l)) &&
1659             i < sizeof(m->apple); m->apple[i++] = *l++)
1660                 continue;
1661         if (i == sizeof(m->apple) && *l) {
1662                 /* We don't need to NUL terminate here, printing handles it */
1663                 if (ms->flags & MAGIC_CHECK)
1664                         file_magwarn(ms, "APPLE type `%s' truncated %"
1665                             SIZE_T_FORMAT "u", line, i);
1666         }
1667
1668         if (i > 0)
1669                 return 0;
1670         else
1671                 return -1;
1672 }
1673
1674 /*
1675  * parse a MIME annotation line from magic file, put into magic[index - 1]
1676  * if valid
1677  */
1678 private int
1679 parse_mime(struct magic_set *ms, struct magic_entry *me, const char *line)
1680 {
1681         size_t i;
1682         const char *l = line;
1683         struct magic *m = &me->mp[me->cont_count == 0 ? 0 : me->cont_count - 1];
1684
1685         if (m->mimetype[0] != '\0') {
1686                 file_magwarn(ms, "Current entry already has a MIME type `%s',"
1687                     " new type `%s'", m->mimetype, l);
1688                 return -1;
1689         }
1690
1691         EATAB;
1692         for (i = 0; *l && ((isascii((unsigned char)*l) &&
1693             isalnum((unsigned char)*l)) || strchr("-+/.", *l)) &&
1694             i < sizeof(m->mimetype); m->mimetype[i++] = *l++)
1695                 continue;
1696         if (i == sizeof(m->mimetype)) {
1697                 m->mimetype[sizeof(m->mimetype) - 1] = '\0';
1698                 if (ms->flags & MAGIC_CHECK)
1699                         file_magwarn(ms, "MIME type `%s' truncated %"
1700                             SIZE_T_FORMAT "u", m->mimetype, i);
1701         } else
1702                 m->mimetype[i] = '\0';
1703
1704         if (i > 0)
1705                 return 0;
1706         else
1707                 return -1;
1708 }
1709
1710 private int
1711 check_format_type(const char *ptr, int type)
1712 {
1713         int quad = 0;
1714         if (*ptr == '\0') {
1715                 /* Missing format string; bad */
1716                 return -1;
1717         }
1718
1719         switch (type) {
1720         case FILE_FMT_QUAD:
1721                 quad = 1;
1722                 /*FALLTHROUGH*/
1723         case FILE_FMT_NUM:
1724                 if (*ptr == '-')
1725                         ptr++;
1726                 if (*ptr == '.')
1727                         ptr++;
1728                 while (isdigit((unsigned char)*ptr)) ptr++;
1729                 if (*ptr == '.')
1730                         ptr++;
1731                 while (isdigit((unsigned char)*ptr)) ptr++;
1732                 if (quad) {
1733                         if (*ptr++ != 'l')
1734                                 return -1;
1735                         if (*ptr++ != 'l')
1736                                 return -1;
1737                 }
1738
1739                 switch (*ptr++) {
1740                 case 'l':
1741                         switch (*ptr++) {
1742                         case 'i':
1743                         case 'd':
1744                         case 'u':
1745                         case 'x':
1746                         case 'X':
1747                                 return 0;
1748                         default:
1749                                 return -1;
1750                         }
1751
1752                 case 'h':
1753                         switch (*ptr++) {
1754                         case 'h':
1755                                 switch (*ptr++) {
1756                                 case 'i':
1757                                 case 'd':
1758                                 case 'u':
1759                                 case 'x':
1760                                 case 'X':
1761                                         return 0;
1762                                 default:
1763                                         return -1;
1764                                 }
1765                         case 'd':
1766                                 return 0;
1767                         default:
1768                                 return -1;
1769                         }
1770
1771                 case 'i':
1772                 case 'c':
1773                 case 'd':
1774                 case 'u':
1775                 case 'x':
1776                 case 'X':
1777                         return 0;
1778
1779                 default:
1780                         return -1;
1781                 }
1782
1783         case FILE_FMT_FLOAT:
1784         case FILE_FMT_DOUBLE:
1785                 if (*ptr == '-')
1786                         ptr++;
1787                 if (*ptr == '.')
1788                         ptr++;
1789                 while (isdigit((unsigned char)*ptr)) ptr++;
1790                 if (*ptr == '.')
1791                         ptr++;
1792                 while (isdigit((unsigned char)*ptr)) ptr++;
1793
1794                 switch (*ptr++) {
1795                 case 'e':
1796                 case 'E':
1797                 case 'f':
1798                 case 'F':
1799                 case 'g':
1800                 case 'G':
1801                         return 0;
1802
1803                 default:
1804                         return -1;
1805                 }
1806
1807
1808         case FILE_FMT_STR:
1809                 if (*ptr == '-')
1810                         ptr++;
1811                 while (isdigit((unsigned char )*ptr))
1812                         ptr++;
1813                 if (*ptr == '.') {
1814                         ptr++;
1815                         while (isdigit((unsigned char )*ptr))
1816                                 ptr++;
1817                 }
1818
1819                 switch (*ptr++) {
1820                 case 's':
1821                         return 0;
1822                 default:
1823                         return -1;
1824                 }
1825
1826         default:
1827                 /* internal error */
1828                 abort();
1829         }
1830         /*NOTREACHED*/
1831         return -1;
1832 }
1833
1834 /*
1835  * Check that the optional printf format in description matches
1836  * the type of the magic.
1837  */
1838 private int
1839 check_format(struct magic_set *ms, struct magic *m)
1840 {
1841         char *ptr;
1842
1843         for (ptr = m->desc; *ptr; ptr++)
1844                 if (*ptr == '%')
1845                         break;
1846         if (*ptr == '\0') {
1847                 /* No format string; ok */
1848                 return 1;
1849         }
1850
1851         assert(file_nformats == file_nnames);
1852
1853         if (m->type >= file_nformats) {
1854                 file_magwarn(ms, "Internal error inconsistency between "
1855                     "m->type and format strings");
1856                 return -1;
1857         }
1858         if (file_formats[m->type] == FILE_FMT_NONE) {
1859                 file_magwarn(ms, "No format string for `%s' with description "
1860                     "`%s'", m->desc, file_names[m->type]);
1861                 return -1;
1862         }
1863
1864         ptr++;
1865         if (check_format_type(ptr, file_formats[m->type]) == -1) {
1866                 /*
1867                  * TODO: this error message is unhelpful if the format
1868                  * string is not one character long
1869                  */
1870                 file_magwarn(ms, "Printf format `%c' is not valid for type "
1871                     "`%s' in description `%s'", *ptr ? *ptr : '?',
1872                     file_names[m->type], m->desc);
1873                 return -1;
1874         }
1875
1876         for (; *ptr; ptr++) {
1877                 if (*ptr == '%') {
1878                         file_magwarn(ms,
1879                             "Too many format strings (should have at most one) "
1880                             "for `%s' with description `%s'",
1881                             file_names[m->type], m->desc);
1882                         return -1;
1883                 }
1884         }
1885         return 0;
1886 }
1887
1888 /*
1889  * Read a numeric value from a pointer, into the value union of a magic
1890  * pointer, according to the magic type.  Update the string pointer to point
1891  * just after the number read.  Return 0 for success, non-zero for failure.
1892  */
1893 private int
1894 getvalue(struct magic_set *ms, struct magic *m, const char **p, int action)
1895 {
1896         switch (m->type) {
1897         case FILE_BESTRING16:
1898         case FILE_LESTRING16:
1899         case FILE_STRING:
1900         case FILE_PSTRING:
1901         case FILE_REGEX:
1902         case FILE_SEARCH:
1903                 *p = getstr(ms, m, *p, action == FILE_COMPILE);
1904                 if (*p == NULL) {
1905                         if (ms->flags & MAGIC_CHECK)
1906                                 file_magwarn(ms, "cannot get string from `%s'",
1907                                     m->value.s);
1908                         return -1;
1909                 }
1910                 return 0;
1911         case FILE_FLOAT:
1912         case FILE_BEFLOAT:
1913         case FILE_LEFLOAT:
1914                 if (m->reln != 'x') {
1915                         char *ep;
1916 #ifdef HAVE_STRTOF
1917                         m->value.f = strtof(*p, &ep);
1918 #else
1919                         m->value.f = (float)strtod(*p, &ep);
1920 #endif
1921                         *p = ep;
1922                 }
1923                 return 0;
1924         case FILE_DOUBLE:
1925         case FILE_BEDOUBLE:
1926         case FILE_LEDOUBLE:
1927                 if (m->reln != 'x') {
1928                         char *ep;
1929                         m->value.d = strtod(*p, &ep);
1930                         *p = ep;
1931                 }
1932                 return 0;
1933         default:
1934                 if (m->reln != 'x') {
1935                         char *ep;
1936                         m->value.q = file_signextend(ms, m,
1937                             (uint64_t)strtoull(*p, &ep, 0));
1938                         *p = ep;
1939                         eatsize(p);
1940                 }
1941                 return 0;
1942         }
1943 }
1944
1945 /*
1946  * Convert a string containing C character escapes.  Stop at an unescaped
1947  * space or tab.
1948  * Copy the converted version to "m->value.s", and the length in m->vallen.
1949  * Return updated scan pointer as function result. Warn if set.
1950  */
1951 private const char *
1952 getstr(struct magic_set *ms, struct magic *m, const char *s, int warn)
1953 {
1954         const char *origs = s;
1955         char    *p = m->value.s;
1956         size_t  plen = sizeof(m->value.s);
1957         char    *origp = p;
1958         char    *pmax = p + plen - 1;
1959         int     c;
1960         int     val;
1961
1962         while ((c = *s++) != '\0') {
1963                 if (isspace((unsigned char) c))
1964                         break;
1965                 if (p >= pmax) {
1966                         file_error(ms, 0, "string too long: `%s'", origs);
1967                         return NULL;
1968                 }
1969                 if (c == '\\') {
1970                         switch(c = *s++) {
1971
1972                         case '\0':
1973                                 if (warn)
1974                                         file_magwarn(ms, "incomplete escape");
1975                                 goto out;
1976
1977                         case '\t':
1978                                 if (warn) {
1979                                         file_magwarn(ms,
1980                                             "escaped tab found, use \\t instead");
1981                                         warn = 0;       /* already did */
1982                                 }
1983                                 /*FALLTHROUGH*/
1984                         default:
1985                                 if (warn) {
1986                                         if (isprint((unsigned char)c)) {
1987                                                 /* Allow escaping of
1988                                                  * ``relations'' */
1989                                                 if (strchr("<>&^=!", c) == NULL
1990                                                     && (m->type != FILE_REGEX ||
1991                                                     strchr("[]().*?^$|{}", c)
1992                                                     == NULL)) {
1993                                                         file_magwarn(ms, "no "
1994                                                             "need to escape "
1995                                                             "`%c'", c);
1996                                                 }
1997                                         } else {
1998                                                 file_magwarn(ms,
1999                                                     "unknown escape sequence: "
2000                                                     "\\%03o", c);
2001                                         }
2002                                 }
2003                                 /*FALLTHROUGH*/
2004                         /* space, perhaps force people to use \040? */
2005                         case ' ':
2006 #if 0
2007                         /*
2008                          * Other things people escape, but shouldn't need to,
2009                          * so we disallow them
2010                          */
2011                         case '\'':
2012                         case '"':
2013                         case '?':
2014 #endif
2015                         /* Relations */
2016                         case '>':
2017                         case '<':
2018                         case '&':
2019                         case '^':
2020                         case '=':
2021                         case '!':
2022                         /* and baskslash itself */
2023                         case '\\':
2024                                 *p++ = (char) c;
2025                                 break;
2026
2027                         case 'a':
2028                                 *p++ = '\a';
2029                                 break;
2030
2031                         case 'b':
2032                                 *p++ = '\b';
2033                                 break;
2034
2035                         case 'f':
2036                                 *p++ = '\f';
2037                                 break;
2038
2039                         case 'n':
2040                                 *p++ = '\n';
2041                                 break;
2042
2043                         case 'r':
2044                                 *p++ = '\r';
2045                                 break;
2046
2047                         case 't':
2048                                 *p++ = '\t';
2049                                 break;
2050
2051                         case 'v':
2052                                 *p++ = '\v';
2053                                 break;
2054
2055                         /* \ and up to 3 octal digits */
2056                         case '0':
2057                         case '1':
2058                         case '2':
2059                         case '3':
2060                         case '4':
2061                         case '5':
2062                         case '6':
2063                         case '7':
2064                                 val = c - '0';
2065                                 c = *s++;  /* try for 2 */
2066                                 if (c >= '0' && c <= '7') {
2067                                         val = (val << 3) | (c - '0');
2068                                         c = *s++;  /* try for 3 */
2069                                         if (c >= '0' && c <= '7')
2070                                                 val = (val << 3) | (c-'0');
2071                                         else
2072                                                 --s;
2073                                 }
2074                                 else
2075                                         --s;
2076                                 *p++ = (char)val;
2077                                 break;
2078
2079                         /* \x and up to 2 hex digits */
2080                         case 'x':
2081                                 val = 'x';      /* Default if no digits */
2082                                 c = hextoint(*s++);     /* Get next char */
2083                                 if (c >= 0) {
2084                                         val = c;
2085                                         c = hextoint(*s++);
2086                                         if (c >= 0)
2087                                                 val = (val << 4) + c;
2088                                         else
2089                                                 --s;
2090                                 } else
2091                                         --s;
2092                                 *p++ = (char)val;
2093                                 break;
2094                         }
2095                 } else
2096                         *p++ = (char)c;
2097         }
2098 out:
2099         *p = '\0';
2100         m->vallen = CAST(unsigned char, (p - origp));
2101         if (m->type == FILE_PSTRING)
2102                 m->vallen += (unsigned char)file_pstring_length_size(m);
2103         return s;
2104 }
2105
2106
2107 /* Single hex char to int; -1 if not a hex char. */
2108 private int
2109 hextoint(int c)
2110 {
2111         if (!isascii((unsigned char) c))
2112                 return -1;
2113         if (isdigit((unsigned char) c))
2114                 return c - '0';
2115         if ((c >= 'a') && (c <= 'f'))
2116                 return c + 10 - 'a';
2117         if (( c>= 'A') && (c <= 'F'))
2118                 return c + 10 - 'A';
2119         return -1;
2120 }
2121
2122
2123 /*
2124  * Print a string containing C character escapes.
2125  */
2126 protected void
2127 file_showstr(FILE *fp, const char *s, size_t len)
2128 {
2129         char    c;
2130
2131         for (;;) {
2132                 if (len == ~0U) {
2133                         c = *s++;
2134                         if (c == '\0')
2135                                 break;
2136                 }
2137                 else  {
2138                         if (len-- == 0)
2139                                 break;
2140                         c = *s++;
2141                 }
2142                 if (c >= 040 && c <= 0176)      /* TODO isprint && !iscntrl */
2143                         (void) fputc(c, fp);
2144                 else {
2145                         (void) fputc('\\', fp);
2146                         switch (c) {
2147                         case '\a':
2148                                 (void) fputc('a', fp);
2149                                 break;
2150
2151                         case '\b':
2152                                 (void) fputc('b', fp);
2153                                 break;
2154
2155                         case '\f':
2156                                 (void) fputc('f', fp);
2157                                 break;
2158
2159                         case '\n':
2160                                 (void) fputc('n', fp);
2161                                 break;
2162
2163                         case '\r':
2164                                 (void) fputc('r', fp);
2165                                 break;
2166
2167                         case '\t':
2168                                 (void) fputc('t', fp);
2169                                 break;
2170
2171                         case '\v':
2172                                 (void) fputc('v', fp);
2173                                 break;
2174
2175                         default:
2176                                 (void) fprintf(fp, "%.3o", c & 0377);
2177                                 break;
2178                         }
2179                 }
2180         }
2181 }
2182
2183 /*
2184  * eatsize(): Eat the size spec from a number [eg. 10UL]
2185  */
2186 private void
2187 eatsize(const char **p)
2188 {
2189         const char *l = *p;
2190
2191         if (LOWCASE(*l) == 'u')
2192                 l++;
2193
2194         switch (LOWCASE(*l)) {
2195         case 'l':    /* long */
2196         case 's':    /* short */
2197         case 'h':    /* short */
2198         case 'b':    /* char/byte */
2199         case 'c':    /* char/byte */
2200                 l++;
2201                 /*FALLTHROUGH*/
2202         default:
2203                 break;
2204         }
2205
2206         *p = l;
2207 }
2208
2209 #ifndef COMPILE_ONLY
2210 /*
2211  * handle a compiled file.
2212  */
2213 private int
2214 apprentice_map(struct magic_set *ms, struct magic **magicp, uint32_t *nmagicp,
2215     const char *fn)
2216 {
2217         int fd;
2218         struct stat st;
2219         uint32_t *ptr;
2220         uint32_t version;
2221         int needsbyteswap;
2222         char *dbname = NULL;
2223         void *mm = NULL;
2224
2225         dbname = mkdbname(ms, fn, 0);
2226         if (dbname == NULL)
2227                 goto error2;
2228
2229         if ((fd = open(dbname, O_RDONLY|O_BINARY)) == -1)
2230                 goto error2;
2231
2232         if (fstat(fd, &st) == -1) {
2233                 file_error(ms, errno, "cannot stat `%s'", dbname);
2234                 goto error1;
2235         }
2236         if (st.st_size < 8) {
2237                 file_error(ms, 0, "file `%s' is too small", dbname);
2238                 goto error1;
2239         }
2240
2241 #ifdef QUICK
2242         if ((mm = mmap(0, (size_t)st.st_size, PROT_READ|PROT_WRITE,
2243             MAP_PRIVATE|MAP_FILE, fd, (off_t)0)) == MAP_FAILED) {
2244                 file_error(ms, errno, "cannot map `%s'", dbname);
2245                 goto error1;
2246         }
2247 #define RET     2
2248 #else
2249         if ((mm = CAST(void *, malloc((size_t)st.st_size))) == NULL) {
2250                 file_oomem(ms, (size_t)st.st_size);
2251                 goto error1;
2252         }
2253         if (read(fd, mm, (size_t)st.st_size) != (ssize_t)st.st_size) {
2254                 file_badread(ms);
2255                 goto error1;
2256         }
2257 #define RET     1
2258 #endif
2259         *magicp = CAST(struct magic *, mm);
2260         (void)close(fd);
2261         fd = -1;
2262         ptr = (uint32_t *)(void *)*magicp;
2263         if (*ptr != MAGICNO) {
2264                 if (swap4(*ptr) != MAGICNO) {
2265                         file_error(ms, 0, "bad magic in `%s'", dbname);
2266                         goto error1;
2267                 }
2268                 needsbyteswap = 1;
2269         } else
2270                 needsbyteswap = 0;
2271         if (needsbyteswap)
2272                 version = swap4(ptr[1]);
2273         else
2274                 version = ptr[1];
2275         if (version != VERSIONNO) {
2276                 file_error(ms, 0, "File %s supports only version %d magic "
2277                     "files. `%s' is version %d", VERSION,
2278                     VERSIONNO, dbname, version);
2279                 goto error1;
2280         }
2281         *nmagicp = (uint32_t)(st.st_size / sizeof(struct magic));
2282         if (*nmagicp > 0)
2283                 (*nmagicp)--;
2284         (*magicp)++;
2285         if (needsbyteswap)
2286                 byteswap(*magicp, *nmagicp);
2287         free(dbname);
2288         return RET;
2289
2290 error1:
2291         if (fd != -1)
2292                 (void)close(fd);
2293         if (mm) {
2294 #ifdef QUICK
2295                 (void)munmap((void *)mm, (size_t)st.st_size);
2296 #else
2297                 free(mm);
2298 #endif
2299         } else {
2300                 *magicp = NULL;
2301                 *nmagicp = 0;
2302         }
2303 error2:
2304         free(dbname);
2305         return -1;
2306 }
2307 #endif /* COMPILE_ONLY */
2308
2309 private const uint32_t ar[] = {
2310     MAGICNO, VERSIONNO
2311 };
2312 /*
2313  * handle an mmaped file.
2314  */
2315 private int
2316 apprentice_compile(struct magic_set *ms, struct magic **magicp,
2317     uint32_t *nmagicp, const char *fn)
2318 {
2319         int fd = -1;
2320         char *dbname;
2321         int rv = -1;
2322
2323         dbname = mkdbname(ms, fn, 1);
2324
2325         if (dbname == NULL)
2326                 goto out;
2327
2328         if ((fd = open(dbname, O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644)) == -1) {
2329                 file_error(ms, errno, "cannot open `%s'", dbname);
2330                 goto out;
2331         }
2332
2333         if (write(fd, ar, sizeof(ar)) != (ssize_t)sizeof(ar)) {
2334                 file_error(ms, errno, "error writing `%s'", dbname);
2335                 goto out;
2336         }
2337
2338         if (lseek(fd, (off_t)sizeof(struct magic), SEEK_SET)
2339             != sizeof(struct magic)) {
2340                 file_error(ms, errno, "error seeking `%s'", dbname);
2341                 goto out;
2342         }
2343
2344         if (write(fd, *magicp, (sizeof(struct magic) * *nmagicp))
2345             != (ssize_t)(sizeof(struct magic) * *nmagicp)) {
2346                 file_error(ms, errno, "error writing `%s'", dbname);
2347                 goto out;
2348         }
2349
2350         if (fd != -1)
2351                 (void)close(fd);
2352         rv = 0;
2353 out:
2354         free(dbname);
2355         return rv;
2356 }
2357
2358 private const char ext[] = ".mgc";
2359 /*
2360  * make a dbname
2361  */
2362 private char *
2363 mkdbname(struct magic_set *ms, const char *fn, int strip)
2364 {
2365         const char *p, *q;
2366         char *buf;
2367
2368         if (strip) {
2369                 if ((p = strrchr(fn, '/')) != NULL)
2370                         fn = ++p;
2371         }
2372
2373         for (q = fn; *q; q++)
2374                 continue;
2375         /* Look for .mgc */
2376         for (p = ext + sizeof(ext) - 1; p >= ext && q >= fn; p--, q--)
2377                 if (*p != *q)
2378                         break;
2379
2380         /* Did not find .mgc, restore q */
2381         if (p >= ext)
2382                 while (*q)
2383                         q++;
2384
2385         q++;
2386         /* Compatibility with old code that looked in .mime */
2387         if (ms->flags & MAGIC_MIME) {
2388                 asprintf(&buf, "%.*s.mime%s", (int)(q - fn), fn, ext);
2389                 if (access(buf, R_OK) != -1) {
2390                         ms->flags &= MAGIC_MIME_TYPE;
2391                         return buf;
2392                 }
2393                 free(buf);
2394         }
2395         asprintf(&buf, "%.*s%s", (int)(q - fn), fn, ext);
2396
2397         /* Compatibility with old code that looked in .mime */
2398         if (strstr(p, ".mime") != NULL)
2399                 ms->flags &= MAGIC_MIME_TYPE;
2400         return buf;
2401 }
2402
2403 #ifndef COMPILE_ONLY
2404 /*
2405  * Byteswap an mmap'ed file if needed
2406  */
2407 private void
2408 byteswap(struct magic *magic, uint32_t nmagic)
2409 {
2410         uint32_t i;
2411         for (i = 0; i < nmagic; i++)
2412                 bs1(&magic[i]);
2413 }
2414
2415 /*
2416  * swap a short
2417  */
2418 private uint16_t
2419 swap2(uint16_t sv)
2420 {
2421         uint16_t rv;
2422         uint8_t *s = (uint8_t *)(void *)&sv;
2423         uint8_t *d = (uint8_t *)(void *)&rv;
2424         d[0] = s[1];
2425         d[1] = s[0];
2426         return rv;
2427 }
2428
2429 /*
2430  * swap an int
2431  */
2432 private uint32_t
2433 swap4(uint32_t sv)
2434 {
2435         uint32_t rv;
2436         uint8_t *s = (uint8_t *)(void *)&sv;
2437         uint8_t *d = (uint8_t *)(void *)&rv;
2438         d[0] = s[3];
2439         d[1] = s[2];
2440         d[2] = s[1];
2441         d[3] = s[0];
2442         return rv;
2443 }
2444
2445 /*
2446  * swap a quad
2447  */
2448 private uint64_t
2449 swap8(uint64_t sv)
2450 {
2451         uint64_t rv;
2452         uint8_t *s = (uint8_t *)(void *)&sv;
2453         uint8_t *d = (uint8_t *)(void *)&rv;
2454 #if 0
2455         d[0] = s[3];
2456         d[1] = s[2];
2457         d[2] = s[1];
2458         d[3] = s[0];
2459         d[4] = s[7];
2460         d[5] = s[6];
2461         d[6] = s[5];
2462         d[7] = s[4];
2463 #else
2464         d[0] = s[7];
2465         d[1] = s[6];
2466         d[2] = s[5];
2467         d[3] = s[4];
2468         d[4] = s[3];
2469         d[5] = s[2];
2470         d[6] = s[1];
2471         d[7] = s[0];
2472 #endif
2473         return rv;
2474 }
2475
2476 /*
2477  * byteswap a single magic entry
2478  */
2479 private void
2480 bs1(struct magic *m)
2481 {
2482         m->cont_level = swap2(m->cont_level);
2483         m->offset = swap4((uint32_t)m->offset);
2484         m->in_offset = swap4((uint32_t)m->in_offset);
2485         m->lineno = swap4((uint32_t)m->lineno);
2486         if (IS_STRING(m->type)) {
2487                 m->str_range = swap4(m->str_range);
2488                 m->str_flags = swap4(m->str_flags);
2489         }
2490         else {
2491                 m->value.q = swap8(m->value.q);
2492                 m->num_mask = swap8(m->num_mask);
2493         }
2494 }
2495 #endif /* COMPILE_ONLY */
2496
2497 protected size_t
2498 file_pstring_length_size(const struct magic *m)
2499 {
2500         switch (m->str_flags & PSTRING_LEN) {
2501         case PSTRING_1_LE:
2502                 return 1;
2503         case PSTRING_2_LE:
2504         case PSTRING_2_BE:
2505                 return 2;
2506         case PSTRING_4_LE:
2507         case PSTRING_4_BE:
2508                 return 4;
2509         default:
2510                 abort();        /* Impossible */
2511                 return 1;
2512         }
2513 }
2514 protected size_t
2515 file_pstring_get_length(const struct magic *m, const char *s)
2516 {
2517         size_t len = 0;
2518
2519         switch (m->str_flags & PSTRING_LEN) {
2520         case PSTRING_1_LE:
2521                 len = *s;
2522                 break;
2523         case PSTRING_2_LE:
2524                 len = (s[1] << 8) | s[0];
2525                 break;
2526         case PSTRING_2_BE:
2527                 len = (s[0] << 8) | s[1];
2528                 break;
2529         case PSTRING_4_LE:
2530                 len = (s[3] << 24) | (s[2] << 16) | (s[1] << 8) | s[0];
2531                 break;
2532         case PSTRING_4_BE:
2533                 len = (s[0] << 24) | (s[1] << 16) | (s[2] << 8) | s[3];
2534                 break;
2535         default:
2536                 abort();        /* Impossible */
2537         }
2538
2539         if (m->str_flags & PSTRING_LENGTH_INCLUDES_ITSELF)
2540                 len -= file_pstring_length_size(m);
2541
2542         return len;
2543 }