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