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