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