contrib/file/apprentice.c

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