apprentice.c 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852
  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. #include "file.h"
  32. #include "magic.h"
  33. #include <stdlib.h>
  34. #ifdef HAVE_UNISTD_H
  35. #include <unistd.h>
  36. #endif
  37. #include <string.h>
  38. #include <assert.h>
  39. #include <ctype.h>
  40. #include <fcntl.h>
  41. #include <sys/stat.h>
  42. #include <sys/param.h>
  43. #ifdef QUICK
  44. #include <sys/mman.h>
  45. #endif
  46. #ifndef lint
  47. FILE_RCSID("@(#)$File: apprentice.c,v 1.104 2007/01/19 19:54:39 christos Exp $")
  48. #endif /* lint */
  49. #define EATAB {while (isascii((unsigned char) *l) && \
  50. isspace((unsigned char) *l)) ++l;}
  51. #define LOWCASE(l) (isupper((unsigned char) (l)) ? \
  52. tolower((unsigned char) (l)) : (l))
  53. /*
  54. * Work around a bug in headers on Digital Unix.
  55. * At least confirmed for: OSF1 V4.0 878
  56. */
  57. #if defined(__osf__) && defined(__DECC)
  58. #ifdef MAP_FAILED
  59. #undef MAP_FAILED
  60. #endif
  61. #endif
  62. #ifndef MAP_FAILED
  63. #define MAP_FAILED (void *) -1
  64. #endif
  65. #ifndef MAP_FILE
  66. #define MAP_FILE 0
  67. #endif
  68. #ifndef MAXPATHLEN
  69. #define MAXPATHLEN 1024
  70. #endif
  71. struct magic_entry {
  72. struct magic *mp;
  73. uint32_t cont_count;
  74. uint32_t max_count;
  75. };
  76. int file_formats[FILE_NAMES_SIZE];
  77. const size_t file_nformats = FILE_NAMES_SIZE;
  78. const char *file_names[FILE_NAMES_SIZE];
  79. const size_t file_nnames = FILE_NAMES_SIZE;
  80. private int getvalue(struct magic_set *ms, struct magic *, const char **, int);
  81. private int hextoint(int);
  82. private const char *getstr(struct magic_set *, const char *, char *, int,
  83. int *, int);
  84. private int parse(struct magic_set *, struct magic_entry **, uint32_t *,
  85. const char *, size_t, int);
  86. private void eatsize(const char **);
  87. private int apprentice_1(struct magic_set *, const char *, int, struct mlist *);
  88. private size_t apprentice_magic_strength(const struct magic *);
  89. private int apprentice_sort(const void *, const void *);
  90. private int apprentice_file(struct magic_set *, struct magic **, uint32_t *,
  91. const char *, int);
  92. private void byteswap(struct magic *, uint32_t);
  93. private void bs1(struct magic *);
  94. private uint16_t swap2(uint16_t);
  95. private uint32_t swap4(uint32_t);
  96. private uint64_t swap8(uint64_t);
  97. private char *mkdbname(const char *, char *, size_t, int);
  98. private int apprentice_map(struct magic_set *, struct magic **, uint32_t *,
  99. const char *);
  100. private int apprentice_compile(struct magic_set *, struct magic **, uint32_t *,
  101. const char *);
  102. private int check_format_type(const char *, int);
  103. private int check_format(struct magic_set *, struct magic *);
  104. private size_t maxmagic = 0;
  105. private size_t magicsize = sizeof(struct magic);
  106. #ifdef COMPILE_ONLY
  107. int main(int, char *[]);
  108. int
  109. main(int argc, char *argv[])
  110. {
  111. int ret;
  112. struct magic_set *ms;
  113. char *progname;
  114. if ((progname = strrchr(argv[0], '/')) != NULL)
  115. progname++;
  116. else
  117. progname = argv[0];
  118. if (argc != 2) {
  119. (void)fprintf(stderr, "Usage: %s file\n", progname);
  120. return 1;
  121. }
  122. if ((ms = magic_open(MAGIC_CHECK)) == NULL) {
  123. (void)fprintf(stderr, "%s: %s\n", progname, strerror(errno));
  124. return 1;
  125. }
  126. ret = magic_compile(ms, argv[1]) == -1 ? 1 : 0;
  127. if (ret == 1)
  128. (void)fprintf(stderr, "%s: %s\n", progname, magic_error(ms));
  129. magic_close(ms);
  130. return ret;
  131. }
  132. #endif /* COMPILE_ONLY */
  133. static const struct type_tbl_s {
  134. const char *name;
  135. const size_t len;
  136. const int type;
  137. const int format;
  138. } type_tbl[] = {
  139. # define XX(s) s, (sizeof(s) - 1)
  140. # define XX_NULL NULL, 0
  141. { XX("byte"), FILE_BYTE, FILE_FMT_NUM },
  142. { XX("short"), FILE_SHORT, FILE_FMT_NUM },
  143. { XX("default"), FILE_DEFAULT, FILE_FMT_STR },
  144. { XX("long"), FILE_LONG, FILE_FMT_NUM },
  145. { XX("string"), FILE_STRING, FILE_FMT_STR },
  146. { XX("date"), FILE_DATE, FILE_FMT_STR },
  147. { XX("beshort"), FILE_BESHORT, FILE_FMT_NUM },
  148. { XX("belong"), FILE_BELONG, FILE_FMT_NUM },
  149. { XX("bedate"), FILE_BEDATE, FILE_FMT_STR },
  150. { XX("leshort"), FILE_LESHORT, FILE_FMT_NUM },
  151. { XX("lelong"), FILE_LELONG, FILE_FMT_NUM },
  152. { XX("ledate"), FILE_LEDATE, FILE_FMT_STR },
  153. { XX("pstring"), FILE_PSTRING, FILE_FMT_STR },
  154. { XX("ldate"), FILE_LDATE, FILE_FMT_STR },
  155. { XX("beldate"), FILE_BELDATE, FILE_FMT_STR },
  156. { XX("leldate"), FILE_LELDATE, FILE_FMT_STR },
  157. { XX("regex"), FILE_REGEX, FILE_FMT_STR },
  158. { XX("bestring16"), FILE_BESTRING16, FILE_FMT_STR },
  159. { XX("lestring16"), FILE_LESTRING16, FILE_FMT_STR },
  160. { XX("search"), FILE_SEARCH, FILE_FMT_STR },
  161. { XX("medate"), FILE_MEDATE, FILE_FMT_STR },
  162. { XX("meldate"), FILE_MELDATE, FILE_FMT_STR },
  163. { XX("melong"), FILE_MELONG, FILE_FMT_NUM },
  164. { XX("quad"), FILE_QUAD, FILE_FMT_QUAD },
  165. { XX("lequad"), FILE_LEQUAD, FILE_FMT_QUAD },
  166. { XX("bequad"), FILE_BEQUAD, FILE_FMT_QUAD },
  167. { XX("qdate"), FILE_QDATE, FILE_FMT_STR },
  168. { XX("leqdate"), FILE_LEQDATE, FILE_FMT_STR },
  169. { XX("beqdate"), FILE_BEQDATE, FILE_FMT_STR },
  170. { XX("qldate"), FILE_QLDATE, FILE_FMT_STR },
  171. { XX("leqldate"), FILE_LEQLDATE, FILE_FMT_STR },
  172. { XX("beqldate"), FILE_BEQLDATE, FILE_FMT_STR },
  173. { XX_NULL, FILE_INVALID, FILE_FMT_NONE },
  174. # undef XX
  175. # undef XX_NULL
  176. };
  177. private int
  178. get_type(const char *l, const char **t)
  179. {
  180. const struct type_tbl_s *p;
  181. for (p = type_tbl; p->name; p++) {
  182. if (strncmp(l, p->name, p->len) == 0) {
  183. if (t)
  184. *t = l + p->len;
  185. break;
  186. }
  187. }
  188. return p->type;
  189. }
  190. private void
  191. init_file_tables(void)
  192. {
  193. static int done = 0;
  194. const struct type_tbl_s *p;
  195. if (done)
  196. return;
  197. done++;
  198. for (p = type_tbl; p->name; p++) {
  199. assert(p->type < FILE_NAMES_SIZE);
  200. file_names[p->type] = p->name;
  201. file_formats[p->type] = p->format;
  202. }
  203. }
  204. /*
  205. * Handle one file.
  206. */
  207. private int
  208. apprentice_1(struct magic_set *ms, const char *fn, int action,
  209. struct mlist *mlist)
  210. {
  211. struct magic *magic = NULL;
  212. uint32_t nmagic = 0;
  213. struct mlist *ml;
  214. int rv = -1;
  215. int mapped;
  216. if (magicsize != FILE_MAGICSIZE) {
  217. file_error(ms, 0, "magic element size %lu != %lu",
  218. (unsigned long)sizeof(*magic),
  219. (unsigned long)FILE_MAGICSIZE);
  220. return -1;
  221. }
  222. if (action == FILE_COMPILE) {
  223. rv = apprentice_file(ms, &magic, &nmagic, fn, action);
  224. if (rv != 0)
  225. return -1;
  226. rv = apprentice_compile(ms, &magic, &nmagic, fn);
  227. free(magic);
  228. return rv;
  229. }
  230. #ifndef COMPILE_ONLY
  231. if ((rv = apprentice_map(ms, &magic, &nmagic, fn)) == -1) {
  232. if (ms->flags & MAGIC_CHECK)
  233. file_magwarn(ms, "using regular magic file `%s'", fn);
  234. rv = apprentice_file(ms, &magic, &nmagic, fn, action);
  235. if (rv != 0)
  236. return -1;
  237. mapped = 0;
  238. }
  239. mapped = rv;
  240. if (magic == NULL || nmagic == 0) {
  241. file_delmagic(magic, mapped, nmagic);
  242. return -1;
  243. }
  244. if ((ml = malloc(sizeof(*ml))) == NULL) {
  245. file_delmagic(magic, mapped, nmagic);
  246. file_oomem(ms, sizeof(*ml));
  247. return -1;
  248. }
  249. ml->magic = magic;
  250. ml->nmagic = nmagic;
  251. ml->mapped = mapped;
  252. mlist->prev->next = ml;
  253. ml->prev = mlist->prev;
  254. ml->next = mlist;
  255. mlist->prev = ml;
  256. return 0;
  257. #endif /* COMPILE_ONLY */
  258. }
  259. protected void
  260. file_delmagic(struct magic *p, int type, size_t entries)
  261. {
  262. if (p == NULL)
  263. return;
  264. switch (type) {
  265. case 2:
  266. p--;
  267. (void)munmap((void *)p, sizeof(*p) * (entries + 1));
  268. break;
  269. case 1:
  270. p--;
  271. /*FALLTHROUGH*/
  272. case 0:
  273. free(p);
  274. break;
  275. default:
  276. abort();
  277. }
  278. }
  279. /* const char *fn: list of magic files */
  280. protected struct mlist *
  281. file_apprentice(struct magic_set *ms, const char *fn, int action)
  282. {
  283. char *p, *mfn, *afn = NULL;
  284. int file_err, errs = -1;
  285. struct mlist *mlist;
  286. static const char mime[] = ".mime";
  287. init_file_tables();
  288. if (fn == NULL)
  289. fn = getenv("MAGIC");
  290. if (fn == NULL)
  291. fn = MAGIC;
  292. if ((fn = mfn = strdup(fn)) == NULL) {
  293. file_oomem(ms, strlen(fn));
  294. return NULL;
  295. }
  296. if ((mlist = malloc(sizeof(*mlist))) == NULL) {
  297. free(mfn);
  298. file_oomem(ms, sizeof(*mlist));
  299. return NULL;
  300. }
  301. mlist->next = mlist->prev = mlist;
  302. while (fn) {
  303. p = strchr(fn, PATHSEP);
  304. if (p)
  305. *p++ = '\0';
  306. if (*fn == '\0')
  307. break;
  308. if (ms->flags & MAGIC_MIME) {
  309. size_t len = strlen(fn) + sizeof(mime);
  310. if ((afn = malloc(len)) == NULL) {
  311. free(mfn);
  312. free(mlist);
  313. file_oomem(ms, len);
  314. return NULL;
  315. }
  316. (void)strcpy(afn, fn);
  317. (void)strcat(afn, mime);
  318. fn = afn;
  319. }
  320. file_err = apprentice_1(ms, fn, action, mlist);
  321. if (file_err > errs)
  322. errs = file_err;
  323. if (afn) {
  324. free(afn);
  325. afn = NULL;
  326. }
  327. fn = p;
  328. }
  329. if (errs == -1) {
  330. free(mfn);
  331. free(mlist);
  332. mlist = NULL;
  333. file_error(ms, 0, "could not find any magic files!");
  334. return NULL;
  335. }
  336. free(mfn);
  337. return mlist;
  338. }
  339. /*
  340. * Get weight of this magic entry, for sorting purposes.
  341. */
  342. private size_t
  343. apprentice_magic_strength(const struct magic *m)
  344. {
  345. #define MULT 10
  346. size_t val = 2 * MULT; /* baseline strength */
  347. switch (m->type) {
  348. case FILE_DEFAULT: /* make sure this sorts last */
  349. return 0;
  350. case FILE_BYTE:
  351. val += 1 * MULT;
  352. break;
  353. case FILE_SHORT:
  354. case FILE_LESHORT:
  355. case FILE_BESHORT:
  356. val += 2 * MULT;
  357. break;
  358. case FILE_LONG:
  359. case FILE_LELONG:
  360. case FILE_BELONG:
  361. case FILE_MELONG:
  362. val += 4 * MULT;
  363. break;
  364. case FILE_PSTRING:
  365. case FILE_STRING:
  366. val += m->vallen * MULT;
  367. break;
  368. case FILE_BESTRING16:
  369. case FILE_LESTRING16:
  370. val += m->vallen * MULT / 2;
  371. break;
  372. case FILE_SEARCH:
  373. case FILE_REGEX:
  374. val += m->vallen;
  375. break;
  376. case FILE_DATE:
  377. case FILE_LEDATE:
  378. case FILE_BEDATE:
  379. case FILE_MEDATE:
  380. case FILE_LDATE:
  381. case FILE_LELDATE:
  382. case FILE_BELDATE:
  383. case FILE_MELDATE:
  384. val += 4 * MULT;
  385. break;
  386. case FILE_QUAD:
  387. case FILE_BEQUAD:
  388. case FILE_LEQUAD:
  389. case FILE_QDATE:
  390. case FILE_LEQDATE:
  391. case FILE_BEQDATE:
  392. case FILE_QLDATE:
  393. case FILE_LEQLDATE:
  394. case FILE_BEQLDATE:
  395. val += 8 * MULT;
  396. break;
  397. default:
  398. val = 0;
  399. (void)fprintf(stderr, "Bad type %d\n", m->type);
  400. abort();
  401. }
  402. switch (m->reln) {
  403. case 'x': /* matches anything penalize */
  404. val = 0;
  405. break;
  406. case '!':
  407. case '=': /* Exact match, prefer */
  408. val += MULT;
  409. break;
  410. case '>':
  411. case '<': /* comparison match reduce strength */
  412. val -= 2 * MULT;
  413. break;
  414. case '^':
  415. case '&': /* masking bits, we could count them too */
  416. val -= MULT;
  417. break;
  418. default:
  419. (void)fprintf(stderr, "Bad relation %c\n", m->reln);
  420. abort();
  421. }
  422. if (val == 0) /* ensure we only return 0 for FILE_DEFAULT */
  423. val = 1;
  424. return val;
  425. }
  426. /*
  427. * Sort callback for sorting entries by "strength" (basically length)
  428. */
  429. private int
  430. apprentice_sort(const void *a, const void *b)
  431. {
  432. const struct magic_entry *ma = a;
  433. const struct magic_entry *mb = b;
  434. size_t sa = apprentice_magic_strength(ma->mp);
  435. size_t sb = apprentice_magic_strength(mb->mp);
  436. if (sa == sb)
  437. return 0;
  438. else if (sa > sb)
  439. return -1;
  440. else
  441. return 1;
  442. }
  443. /*
  444. * parse from a file
  445. * const char *fn: name of magic file
  446. */
  447. private int
  448. apprentice_file(struct magic_set *ms, struct magic **magicp, uint32_t *nmagicp,
  449. const char *fn, int action)
  450. {
  451. private const char hdr[] =
  452. "cont\toffset\ttype\topcode\tmask\tvalue\tdesc";
  453. FILE *f;
  454. char line[BUFSIZ+1];
  455. int errs = 0;
  456. struct magic_entry *marray;
  457. uint32_t marraycount, i, mentrycount = 0;
  458. size_t lineno = 0;
  459. ms->flags |= MAGIC_CHECK; /* Enable checks for parsed files */
  460. f = fopen(ms->file = fn, "r");
  461. if (f == NULL) {
  462. if (errno != ENOENT)
  463. file_error(ms, errno, "cannot read magic file `%s'",
  464. fn);
  465. return -1;
  466. }
  467. maxmagic = MAXMAGIS;
  468. if ((marray = calloc(maxmagic, sizeof(*marray))) == NULL) {
  469. (void)fclose(f);
  470. file_oomem(ms, maxmagic * sizeof(*marray));
  471. return -1;
  472. }
  473. marraycount = 0;
  474. /* print silly verbose header for USG compat. */
  475. if (action == FILE_CHECK)
  476. (void)fprintf(stderr, "%s\n", hdr);
  477. /* read and parse this file */
  478. for (ms->line = 1; fgets(line, BUFSIZ, f) != NULL; ms->line++) {
  479. size_t len;
  480. len = strlen(line);
  481. if (len == 0) /* null line, garbage, etc */
  482. continue;
  483. if (line[len - 1] == '\n') {
  484. lineno++;
  485. line[len - 1] = '\0'; /* delete newline */
  486. }
  487. if (line[0] == '\0') /* empty, do not parse */
  488. continue;
  489. if (line[0] == '#') /* comment, do not parse */
  490. continue;
  491. if (parse(ms, &marray, &marraycount, line, lineno, action) != 0)
  492. errs++;
  493. }
  494. (void)fclose(f);
  495. if (errs)
  496. goto out;
  497. #ifndef NOORDER
  498. qsort(marray, marraycount, sizeof(*marray), apprentice_sort);
  499. /*
  500. * Make sure that any level 0 "default" line is last (if one exists).
  501. */
  502. for (i = 0; i < marraycount; i++) {
  503. if (marray[i].mp->cont_level == 0 &&
  504. marray[i].mp->type == FILE_DEFAULT) {
  505. while (++i < marraycount)
  506. if (marray[i].mp->cont_level == 0)
  507. break;
  508. if (i != marraycount) {
  509. ms->line = marray[i].mp->lineno; /* XXX - Ugh! */
  510. file_magwarn(ms,
  511. "level 0 \"default\" did not sort last");
  512. }
  513. break;
  514. }
  515. }
  516. #endif
  517. for (i = 0; i < marraycount; i++)
  518. mentrycount += marray[i].cont_count;
  519. if ((*magicp = malloc(sizeof(**magicp) * mentrycount)) == NULL) {
  520. file_oomem(ms, sizeof(**magicp) * mentrycount);
  521. errs++;
  522. goto out;
  523. }
  524. mentrycount = 0;
  525. for (i = 0; i < marraycount; i++) {
  526. (void)memcpy(*magicp + mentrycount, marray[i].mp,
  527. marray[i].cont_count * sizeof(**magicp));
  528. mentrycount += marray[i].cont_count;
  529. }
  530. out:
  531. for (i = 0; i < marraycount; i++)
  532. free(marray[i].mp);
  533. free(marray);
  534. if (errs) {
  535. *magicp = NULL;
  536. *nmagicp = 0;
  537. return errs;
  538. } else {
  539. *nmagicp = mentrycount;
  540. return 0;
  541. }
  542. }
  543. /*
  544. * extend the sign bit if the comparison is to be signed
  545. */
  546. protected uint64_t
  547. file_signextend(struct magic_set *ms, struct magic *m, uint64_t v)
  548. {
  549. if (!(m->flag & UNSIGNED)) {
  550. switch(m->type) {
  551. /*
  552. * Do not remove the casts below. They are
  553. * vital. When later compared with the data,
  554. * the sign extension must have happened.
  555. */
  556. case FILE_BYTE:
  557. v = (char) v;
  558. break;
  559. case FILE_SHORT:
  560. case FILE_BESHORT:
  561. case FILE_LESHORT:
  562. v = (short) v;
  563. break;
  564. case FILE_DATE:
  565. case FILE_BEDATE:
  566. case FILE_LEDATE:
  567. case FILE_MEDATE:
  568. case FILE_LDATE:
  569. case FILE_BELDATE:
  570. case FILE_LELDATE:
  571. case FILE_MELDATE:
  572. case FILE_LONG:
  573. case FILE_BELONG:
  574. case FILE_LELONG:
  575. case FILE_MELONG:
  576. v = (int32_t) v;
  577. break;
  578. case FILE_QUAD:
  579. case FILE_BEQUAD:
  580. case FILE_LEQUAD:
  581. case FILE_QDATE:
  582. case FILE_QLDATE:
  583. case FILE_BEQDATE:
  584. case FILE_BEQLDATE:
  585. case FILE_LEQDATE:
  586. case FILE_LEQLDATE:
  587. v = (int64_t) v;
  588. break;
  589. case FILE_STRING:
  590. case FILE_PSTRING:
  591. case FILE_BESTRING16:
  592. case FILE_LESTRING16:
  593. case FILE_REGEX:
  594. case FILE_SEARCH:
  595. case FILE_DEFAULT:
  596. break;
  597. default:
  598. if (ms->flags & MAGIC_CHECK)
  599. file_magwarn(ms, "cannot happen: m->type=%d\n",
  600. m->type);
  601. return ~0U;
  602. }
  603. }
  604. return v;
  605. }
  606. private int
  607. string_modifier_check(struct magic_set *ms, struct magic const *m)
  608. {
  609. if ((ms->flags & MAGIC_CHECK) == 0)
  610. return 0;
  611. switch (m->type) {
  612. case FILE_BESTRING16:
  613. case FILE_LESTRING16:
  614. if (m->str_flags != 0) {
  615. file_magwarn(ms, "no modifiers allowed for 16-bit strings\n");
  616. return -1;
  617. }
  618. break;
  619. case FILE_STRING:
  620. case FILE_PSTRING:
  621. if ((m->str_flags & REGEX_OFFSET_START) != 0) {
  622. file_magwarn(ms, "'/%c' only allowed on regex and search\n",
  623. CHAR_REGEX_OFFSET_START);
  624. return -1;
  625. }
  626. break;
  627. case FILE_SEARCH:
  628. break;
  629. case FILE_REGEX:
  630. if ((m->str_flags & STRING_COMPACT_BLANK) != 0) {
  631. file_magwarn(ms, "'/%c' not allowed on regex\n",
  632. CHAR_COMPACT_BLANK);
  633. return -1;
  634. }
  635. if ((m->str_flags & STRING_COMPACT_OPTIONAL_BLANK) != 0) {
  636. file_magwarn(ms, "'/%c' not allowed on regex\n",
  637. CHAR_COMPACT_OPTIONAL_BLANK);
  638. return -1;
  639. }
  640. break;
  641. default:
  642. file_magwarn(ms, "coding error: m->type=%d\n",
  643. m->type);
  644. return -1;
  645. }
  646. return 0;
  647. }
  648. private int
  649. get_op(char c)
  650. {
  651. switch (c) {
  652. case '&':
  653. return FILE_OPAND;
  654. case '|':
  655. return FILE_OPOR;
  656. case '^':
  657. return FILE_OPXOR;
  658. case '+':
  659. return FILE_OPADD;
  660. case '-':
  661. return FILE_OPMINUS;
  662. case '*':
  663. return FILE_OPMULTIPLY;
  664. case '/':
  665. return FILE_OPDIVIDE;
  666. case '%':
  667. return FILE_OPMODULO;
  668. default:
  669. return -1;
  670. }
  671. }
  672. #ifdef ENABLE_CONDITIONALS
  673. private int
  674. get_cond(const char *l, const char **t)
  675. {
  676. static struct cond_tbl_s {
  677. const char *name;
  678. const size_t len;
  679. const int cond;
  680. } cond_tbl[] = {
  681. { "if", 2, COND_IF },
  682. { "elif", 4, COND_ELIF },
  683. { "else", 4, COND_ELSE },
  684. { NULL, 0, COND_NONE },
  685. };
  686. struct cond_tbl_s *p;
  687. for (p = cond_tbl; p->name; p++) {
  688. if (strncmp(l, p->name, p->len) == 0 &&
  689. isspace((unsigned char)l[p->len])) {
  690. if (t)
  691. *t = l + p->len;
  692. break;
  693. }
  694. }
  695. return p->cond;
  696. }
  697. private int
  698. check_cond(struct magic_set *ms, int cond, uint32_t cont_level)
  699. {
  700. int last_cond;
  701. last_cond = ms->c.li[cont_level].last_cond;
  702. switch (cond) {
  703. case COND_IF:
  704. if (last_cond != COND_NONE && last_cond != COND_ELIF) {
  705. if (ms->flags & MAGIC_CHECK)
  706. file_magwarn(ms, "syntax error: `if'");
  707. return -1;
  708. }
  709. last_cond = COND_IF;
  710. break;
  711. case COND_ELIF:
  712. if (last_cond != COND_IF && last_cond != COND_ELIF) {
  713. if (ms->flags & MAGIC_CHECK)
  714. file_magwarn(ms, "syntax error: `elif'");
  715. return -1;
  716. }
  717. last_cond = COND_ELIF;
  718. break;
  719. case COND_ELSE:
  720. if (last_cond != COND_IF && last_cond != COND_ELIF) {
  721. if (ms->flags & MAGIC_CHECK)
  722. file_magwarn(ms, "syntax error: `else'");
  723. return -1;
  724. }
  725. last_cond = COND_NONE;
  726. break;
  727. case COND_NONE:
  728. last_cond = COND_NONE;
  729. break;
  730. }
  731. ms->c.li[cont_level].last_cond = last_cond;
  732. return 0;
  733. }
  734. #endif /* ENABLE_CONDITIONALS */
  735. /*
  736. * parse one line from magic file, put into magic[index++] if valid
  737. */
  738. private int
  739. parse(struct magic_set *ms, struct magic_entry **mentryp, uint32_t *nmentryp,
  740. const char *line, size_t lineno, int action)
  741. {
  742. #ifdef ENABLE_CONDITIONALS
  743. static uint32_t last_cont_level = 0;
  744. #endif
  745. size_t i;
  746. struct magic_entry *me;
  747. struct magic *m;
  748. const char *l = line;
  749. char *t;
  750. int op;
  751. uint32_t cont_level;
  752. cont_level = 0;
  753. while (*l == '>') {
  754. ++l; /* step over */
  755. cont_level++;
  756. }
  757. #ifdef ENABLE_CONDITIONALS
  758. if (cont_level == 0 || cont_level > last_cont_level)
  759. if (file_check_mem(ms, cont_level) == -1)
  760. return -1;
  761. last_cont_level = cont_level;
  762. #endif
  763. #define ALLOC_CHUNK (size_t)10
  764. #define ALLOC_INCR (size_t)200
  765. if (cont_level != 0) {
  766. if (*nmentryp == 0) {
  767. file_error(ms, 0, "No current entry for continuation");
  768. return -1;
  769. }
  770. me = &(*mentryp)[*nmentryp - 1];
  771. if (me->cont_count == me->max_count) {
  772. struct magic *nm;
  773. size_t cnt = me->max_count + ALLOC_CHUNK;
  774. if ((nm = realloc(me->mp, sizeof(*nm) * cnt)) == NULL) {
  775. file_oomem(ms, sizeof(*nm) * cnt);
  776. return -1;
  777. }
  778. me->mp = m = nm;
  779. me->max_count = cnt;
  780. }
  781. m = &me->mp[me->cont_count++];
  782. (void)memset(m, 0, sizeof(*m));
  783. m->cont_level = cont_level;
  784. } else {
  785. if (*nmentryp == maxmagic) {
  786. struct magic_entry *mp;
  787. maxmagic += ALLOC_INCR;
  788. if ((mp = realloc(*mentryp, sizeof(*mp) * maxmagic)) ==
  789. NULL) {
  790. file_oomem(ms, sizeof(*mp) * maxmagic);
  791. return -1;
  792. }
  793. (void)memset(&mp[*nmentryp], 0, sizeof(*mp) *
  794. ALLOC_INCR);
  795. *mentryp = mp;
  796. }
  797. me = &(*mentryp)[*nmentryp];
  798. if (me->mp == NULL) {
  799. if ((m = malloc(sizeof(*m) * ALLOC_CHUNK)) == NULL) {
  800. file_oomem(ms, sizeof(*m) * ALLOC_CHUNK);
  801. return -1;
  802. }
  803. me->mp = m;
  804. me->max_count = ALLOC_CHUNK;
  805. } else
  806. m = me->mp;
  807. (void)memset(m, 0, sizeof(*m));
  808. m->cont_level = 0;
  809. me->cont_count = 1;
  810. }
  811. m->lineno = lineno;
  812. if (*l == '&') { /* m->cont_level == 0 checked below. */
  813. ++l; /* step over */
  814. m->flag |= OFFADD;
  815. }
  816. if (*l == '(') {
  817. ++l; /* step over */
  818. m->flag |= INDIR;
  819. if (m->flag & OFFADD)
  820. m->flag = (m->flag & ~OFFADD) | INDIROFFADD;
  821. if (*l == '&') { /* m->cont_level == 0 checked below */
  822. ++l; /* step over */
  823. m->flag |= OFFADD;
  824. }
  825. }
  826. /* Indirect offsets are not valid at level 0. */
  827. if (m->cont_level == 0 && (m->flag & (OFFADD | INDIROFFADD)))
  828. if (ms->flags & MAGIC_CHECK)
  829. file_magwarn(ms, "relative offset at level 0");
  830. /* get offset, then skip over it */
  831. m->offset = (uint32_t)strtoul(l, &t, 0);
  832. if (l == t)
  833. if (ms->flags & MAGIC_CHECK)
  834. file_magwarn(ms, "offset `%s' invalid", l);
  835. l = t;
  836. if (m->flag & INDIR) {
  837. m->in_type = FILE_LONG;
  838. m->in_offset = 0;
  839. /*
  840. * read [.lbs][+-]nnnnn)
  841. */
  842. if (*l == '.') {
  843. l++;
  844. switch (*l) {
  845. case 'l':
  846. m->in_type = FILE_LELONG;
  847. break;
  848. case 'L':
  849. m->in_type = FILE_BELONG;
  850. break;
  851. case 'm':
  852. m->in_type = FILE_MELONG;
  853. break;
  854. case 'h':
  855. case 's':
  856. m->in_type = FILE_LESHORT;
  857. break;
  858. case 'H':
  859. case 'S':
  860. m->in_type = FILE_BESHORT;
  861. break;
  862. case 'c':
  863. case 'b':
  864. case 'C':
  865. case 'B':
  866. m->in_type = FILE_BYTE;
  867. break;
  868. default:
  869. if (ms->flags & MAGIC_CHECK)
  870. file_magwarn(ms,
  871. "indirect offset type `%c' invalid",
  872. *l);
  873. break;
  874. }
  875. l++;
  876. }
  877. m->in_op = 0;
  878. if (*l == '~') {
  879. m->in_op |= FILE_OPINVERSE;
  880. l++;
  881. }
  882. if ((op = get_op(*l)) != -1) {
  883. m->in_op |= op;
  884. l++;
  885. }
  886. if (*l == '(') {
  887. m->in_op |= FILE_OPINDIRECT;
  888. l++;
  889. }
  890. if (isdigit((unsigned char)*l) || *l == '-') {
  891. m->in_offset = (int32_t)strtol(l, &t, 0);
  892. if (l == t)
  893. if (ms->flags & MAGIC_CHECK)
  894. file_magwarn(ms,
  895. "in_offset `%s' invalid", l);
  896. l = t;
  897. }
  898. if (*l++ != ')' ||
  899. ((m->in_op & FILE_OPINDIRECT) && *l++ != ')'))
  900. if (ms->flags & MAGIC_CHECK)
  901. file_magwarn(ms,
  902. "missing ')' in indirect offset");
  903. }
  904. EATAB;
  905. #ifdef ENABLE_CONDITIONALS
  906. m->cond = get_cond(l, &l);
  907. if (check_cond(ms, m->cond, cont_level) == -1)
  908. return -1;
  909. EATAB;
  910. #endif
  911. if (*l == 'u') {
  912. ++l;
  913. m->flag |= UNSIGNED;
  914. }
  915. m->type = get_type(l, &l);
  916. if (m->type == FILE_INVALID) {
  917. if (ms->flags & MAGIC_CHECK)
  918. file_magwarn(ms, "type `%s' invalid", l);
  919. return -1;
  920. }
  921. /* New-style anding: "0 byte&0x80 =0x80 dynamically linked" */
  922. /* New and improved: ~ & | ^ + - * / % -- exciting, isn't it? */
  923. m->mask_op = 0;
  924. if (*l == '~') {
  925. if (!IS_STRING(m->type))
  926. m->mask_op |= FILE_OPINVERSE;
  927. else if (ms->flags & MAGIC_CHECK)
  928. file_magwarn(ms, "'~' invalid for string types");
  929. ++l;
  930. }
  931. m->str_count = 0;
  932. m->str_flags = 0;
  933. m->num_mask = 0;
  934. if ((op = get_op(*l)) != -1) {
  935. if (!IS_STRING(m->type)) {
  936. uint64_t val;
  937. ++l;
  938. m->mask_op |= op;
  939. val = (uint64_t)strtoull(l, &t, 0);
  940. l = t;
  941. m->num_mask = file_signextend(ms, m, val);
  942. eatsize(&l);
  943. }
  944. else if (op == FILE_OPDIVIDE) {
  945. int have_count = 0;
  946. while (!isspace((unsigned char)*++l)) {
  947. switch (*l) {
  948. /* for portability avoid "case '0' ... '9':" */
  949. case '0': case '1': case '2':
  950. case '3': case '4': case '5':
  951. case '6': case '7': case '8':
  952. case '9': {
  953. if (have_count && ms->flags & MAGIC_CHECK)
  954. file_magwarn(ms,
  955. "multiple counts");
  956. have_count = 1;
  957. m->str_count = strtoul(l, &t, 0);
  958. l = t - 1;
  959. break;
  960. }
  961. case CHAR_COMPACT_BLANK:
  962. m->str_flags |= STRING_COMPACT_BLANK;
  963. break;
  964. case CHAR_COMPACT_OPTIONAL_BLANK:
  965. m->str_flags |=
  966. STRING_COMPACT_OPTIONAL_BLANK;
  967. break;
  968. case CHAR_IGNORE_LOWERCASE:
  969. m->str_flags |= STRING_IGNORE_LOWERCASE;
  970. break;
  971. case CHAR_IGNORE_UPPERCASE:
  972. m->str_flags |= STRING_IGNORE_UPPERCASE;
  973. break;
  974. case CHAR_REGEX_OFFSET_START:
  975. m->str_flags |= REGEX_OFFSET_START;
  976. break;
  977. default:
  978. if (ms->flags & MAGIC_CHECK)
  979. file_magwarn(ms,
  980. "string extension `%c' invalid",
  981. *l);
  982. return -1;
  983. }
  984. /* allow multiple '/' for readability */
  985. if (l[1] == '/' && !isspace((unsigned char)l[2]))
  986. l++;
  987. }
  988. if (string_modifier_check(ms, m) == -1)
  989. return -1;
  990. }
  991. else {
  992. if (ms->flags & MAGIC_CHECK)
  993. file_magwarn(ms, "invalid string op: %c", *t);
  994. return -1;
  995. }
  996. }
  997. /*
  998. * We used to set mask to all 1's here, instead let's just not do
  999. * anything if mask = 0 (unless you have a better idea)
  1000. */
  1001. EATAB;
  1002. switch (*l) {
  1003. case '>':
  1004. case '<':
  1005. /* Old-style anding: "0 byte &0x80 dynamically linked" */
  1006. case '&':
  1007. case '^':
  1008. case '=':
  1009. m->reln = *l;
  1010. ++l;
  1011. if (*l == '=') {
  1012. /* HP compat: ignore &= etc. */
  1013. ++l;
  1014. }
  1015. break;
  1016. case '!':
  1017. m->reln = *l;
  1018. ++l;
  1019. break;
  1020. default:
  1021. m->reln = '='; /* the default relation */
  1022. if (*l == 'x' && ((isascii((unsigned char)l[1]) &&
  1023. isspace((unsigned char)l[1])) || !l[1])) {
  1024. m->reln = *l;
  1025. ++l;
  1026. }
  1027. break;
  1028. }
  1029. /*
  1030. * Grab the value part, except for an 'x' reln.
  1031. */
  1032. if (m->reln != 'x' && getvalue(ms, m, &l, action))
  1033. return -1;
  1034. /*
  1035. * TODO finish this macro and start using it!
  1036. * #define offsetcheck {if (offset > HOWMANY-1)
  1037. * magwarn("offset too big"); }
  1038. */
  1039. /*
  1040. * Now get last part - the description
  1041. */
  1042. EATAB;
  1043. if (l[0] == '\b') {
  1044. ++l;
  1045. m->nospflag = 1;
  1046. } else if ((l[0] == '\\') && (l[1] == 'b')) {
  1047. ++l;
  1048. ++l;
  1049. m->nospflag = 1;
  1050. } else
  1051. m->nospflag = 0;
  1052. for (i = 0; (m->desc[i++] = *l++) != '\0' && i < sizeof(m->desc); )
  1053. continue;
  1054. if (i == sizeof(m->desc)) {
  1055. m->desc[sizeof(m->desc) - 1] = '\0';
  1056. if (ms->flags & MAGIC_CHECK)
  1057. file_magwarn(ms, "description `%s' truncated", m->desc);
  1058. }
  1059. /*
  1060. * We only do this check while compiling, or if any of the magic
  1061. * files were not compiled.
  1062. */
  1063. if (ms->flags & MAGIC_CHECK) {
  1064. if (check_format(ms, m) == -1)
  1065. return -1;
  1066. }
  1067. #ifndef COMPILE_ONLY
  1068. if (action == FILE_CHECK) {
  1069. file_mdump(m);
  1070. }
  1071. #endif
  1072. if (m->cont_level == 0)
  1073. ++(*nmentryp); /* make room for next */
  1074. return 0;
  1075. }
  1076. private int
  1077. check_format_type(const char *ptr, int type)
  1078. {
  1079. int quad = 0;
  1080. if (*ptr == '\0') {
  1081. /* Missing format string; bad */
  1082. return -1;
  1083. }
  1084. switch (type) {
  1085. case FILE_FMT_QUAD:
  1086. quad = 1;
  1087. /*FALLTHROUGH*/
  1088. case FILE_FMT_NUM:
  1089. if (*ptr == '-')
  1090. ptr++;
  1091. if (*ptr == '.')
  1092. ptr++;
  1093. while (isdigit((unsigned char)*ptr)) ptr++;
  1094. if (*ptr == '.')
  1095. ptr++;
  1096. while (isdigit((unsigned char)*ptr)) ptr++;
  1097. if (quad) {
  1098. if (*ptr++ != 'l')
  1099. return -1;
  1100. if (*ptr++ != 'l')
  1101. return -1;
  1102. }
  1103. switch (*ptr++) {
  1104. case 'l':
  1105. switch (*ptr++) {
  1106. case 'i':
  1107. case 'd':
  1108. case 'u':
  1109. case 'x':
  1110. case 'X':
  1111. return 0;
  1112. default:
  1113. return -1;
  1114. }
  1115. case 'h':
  1116. switch (*ptr++) {
  1117. case 'h':
  1118. switch (*ptr++) {
  1119. case 'i':
  1120. case 'd':
  1121. case 'u':
  1122. case 'x':
  1123. case 'X':
  1124. return 0;
  1125. default:
  1126. return -1;
  1127. }
  1128. case 'd':
  1129. return 0;
  1130. default:
  1131. return -1;
  1132. }
  1133. case 'i':
  1134. case 'c':
  1135. case 'd':
  1136. case 'u':
  1137. case 'x':
  1138. case 'X':
  1139. return 0;
  1140. default:
  1141. return -1;
  1142. }
  1143. case FILE_FMT_STR:
  1144. if (*ptr == '-')
  1145. ptr++;
  1146. while (isdigit((unsigned char )*ptr))
  1147. ptr++;
  1148. if (*ptr == '.') {
  1149. ptr++;
  1150. while (isdigit((unsigned char )*ptr))
  1151. ptr++;
  1152. }
  1153. switch (*ptr++) {
  1154. case 's':
  1155. return 0;
  1156. default:
  1157. return -1;
  1158. }
  1159. default:
  1160. /* internal error */
  1161. abort();
  1162. }
  1163. /*NOTREACHED*/
  1164. return -1;
  1165. }
  1166. /*
  1167. * Check that the optional printf format in description matches
  1168. * the type of the magic.
  1169. */
  1170. private int
  1171. check_format(struct magic_set *ms, struct magic *m)
  1172. {
  1173. char *ptr;
  1174. for (ptr = m->desc; *ptr; ptr++)
  1175. if (*ptr == '%')
  1176. break;
  1177. if (*ptr == '\0') {
  1178. /* No format string; ok */
  1179. return 1;
  1180. }
  1181. assert(file_nformats == file_nnames);
  1182. if (m->type >= file_nformats) {
  1183. file_error(ms, 0, "Internal error inconsistency between "
  1184. "m->type and format strings");
  1185. return -1;
  1186. }
  1187. if (file_formats[m->type] == FILE_FMT_NONE) {
  1188. file_error(ms, 0, "No format string for `%s' with description "
  1189. "`%s'", m->desc, file_names[m->type]);
  1190. return -1;
  1191. }
  1192. ptr++;
  1193. if (check_format_type(ptr, file_formats[m->type]) == -1) {
  1194. /*
  1195. * TODO: this error message is unhelpful if the format
  1196. * string is not one character long
  1197. */
  1198. file_error(ms, 0, "Printf format `%c' is not valid for type "
  1199. " `%s' in description `%s'", *ptr,
  1200. file_names[m->type], m->desc);
  1201. return -1;
  1202. }
  1203. for (; *ptr; ptr++) {
  1204. if (*ptr == '%') {
  1205. file_error(ms, 0,
  1206. "Too many format strings (should have at most one) "
  1207. "for `%s' with description `%s'",
  1208. file_names[m->type], m->desc);
  1209. return -1;
  1210. }
  1211. }
  1212. return 0;
  1213. }
  1214. /*
  1215. * Read a numeric value from a pointer, into the value union of a magic
  1216. * pointer, according to the magic type. Update the string pointer to point
  1217. * just after the number read. Return 0 for success, non-zero for failure.
  1218. */
  1219. private int
  1220. getvalue(struct magic_set *ms, struct magic *m, const char **p, int action)
  1221. {
  1222. int slen;
  1223. switch (m->type) {
  1224. case FILE_BESTRING16:
  1225. case FILE_LESTRING16:
  1226. case FILE_STRING:
  1227. case FILE_PSTRING:
  1228. case FILE_REGEX:
  1229. case FILE_SEARCH:
  1230. *p = getstr(ms, *p, m->value.s, sizeof(m->value.s), &slen, action);
  1231. if (*p == NULL) {
  1232. if (ms->flags & MAGIC_CHECK)
  1233. file_magwarn(ms, "cannot get string from `%s'",
  1234. m->value.s);
  1235. return -1;
  1236. }
  1237. m->vallen = slen;
  1238. return 0;
  1239. default:
  1240. if (m->reln != 'x') {
  1241. char *ep;
  1242. m->value.q = file_signextend(ms, m,
  1243. (uint64_t)strtoull(*p, &ep, 0));
  1244. *p = ep;
  1245. eatsize(p);
  1246. }
  1247. return 0;
  1248. }
  1249. }
  1250. /*
  1251. * Convert a string containing C character escapes. Stop at an unescaped
  1252. * space or tab.
  1253. * Copy the converted version to "p", returning its length in *slen.
  1254. * Return updated scan pointer as function result.
  1255. */
  1256. private const char *
  1257. getstr(struct magic_set *ms, const char *s, char *p, int plen, int *slen, int action)
  1258. {
  1259. const char *origs = s;
  1260. char *origp = p;
  1261. char *pmax = p + plen - 1;
  1262. int c;
  1263. int val;
  1264. while ((c = *s++) != '\0') {
  1265. if (isspace((unsigned char) c))
  1266. break;
  1267. if (p >= pmax) {
  1268. file_error(ms, 0, "string too long: `%s'", origs);
  1269. return NULL;
  1270. }
  1271. if (c == '\\') {
  1272. switch(c = *s++) {
  1273. case '\0':
  1274. if (action == FILE_COMPILE)
  1275. file_magwarn(ms, "incomplete escape");
  1276. goto out;
  1277. case '\t':
  1278. if (action == FILE_COMPILE) {
  1279. file_magwarn(ms,
  1280. "escaped tab found, use \\t instead");
  1281. action++;
  1282. }
  1283. /*FALLTHROUGH*/
  1284. default:
  1285. if (action == FILE_COMPILE) {
  1286. if (isprint((unsigned char)c))
  1287. file_magwarn(ms,
  1288. "no need to escape `%c'", c);
  1289. else
  1290. file_magwarn(ms,
  1291. "unknown escape sequence: \\%03o", c);
  1292. }
  1293. /*FALLTHROUGH*/
  1294. /* space, perhaps force people to use \040? */
  1295. case ' ':
  1296. #if 0
  1297. /*
  1298. * Other things people escape, but shouldn't need to,
  1299. * so we disallow them
  1300. */
  1301. case '\'':
  1302. case '"':
  1303. case '?':
  1304. #endif
  1305. /* Relations */
  1306. case '>':
  1307. case '<':
  1308. case '&':
  1309. case '^':
  1310. case '=':
  1311. case '!':
  1312. /* and baskslash itself */
  1313. case '\\':
  1314. *p++ = (char) c;
  1315. break;
  1316. case 'a':
  1317. *p++ = '\a';
  1318. break;
  1319. case 'b':
  1320. *p++ = '\b';
  1321. break;
  1322. case 'f':
  1323. *p++ = '\f';
  1324. break;
  1325. case 'n':
  1326. *p++ = '\n';
  1327. break;
  1328. case 'r':
  1329. *p++ = '\r';
  1330. break;
  1331. case 't':
  1332. *p++ = '\t';
  1333. break;
  1334. case 'v':
  1335. *p++ = '\v';
  1336. break;
  1337. /* \ and up to 3 octal digits */
  1338. case '0':
  1339. case '1':
  1340. case '2':
  1341. case '3':
  1342. case '4':
  1343. case '5':
  1344. case '6':
  1345. case '7':
  1346. val = c - '0';
  1347. c = *s++; /* try for 2 */
  1348. if (c >= '0' && c <= '7') {
  1349. val = (val << 3) | (c - '0');
  1350. c = *s++; /* try for 3 */
  1351. if (c >= '0' && c <= '7')
  1352. val = (val << 3) | (c-'0');
  1353. else
  1354. --s;
  1355. }
  1356. else
  1357. --s;
  1358. *p++ = (char)val;
  1359. break;
  1360. /* \x and up to 2 hex digits */
  1361. case 'x':
  1362. val = 'x'; /* Default if no digits */
  1363. c = hextoint(*s++); /* Get next char */
  1364. if (c >= 0) {
  1365. val = c;
  1366. c = hextoint(*s++);
  1367. if (c >= 0)
  1368. val = (val << 4) + c;
  1369. else
  1370. --s;
  1371. } else
  1372. --s;
  1373. *p++ = (char)val;
  1374. break;
  1375. }
  1376. } else
  1377. *p++ = (char)c;
  1378. }
  1379. out:
  1380. *p = '\0';
  1381. *slen = p - origp;
  1382. return s;
  1383. }
  1384. /* Single hex char to int; -1 if not a hex char. */
  1385. private int
  1386. hextoint(int c)
  1387. {
  1388. if (!isascii((unsigned char) c))
  1389. return -1;
  1390. if (isdigit((unsigned char) c))
  1391. return c - '0';
  1392. if ((c >= 'a') && (c <= 'f'))
  1393. return c + 10 - 'a';
  1394. if (( c>= 'A') && (c <= 'F'))
  1395. return c + 10 - 'A';
  1396. return -1;
  1397. }
  1398. /*
  1399. * Print a string containing C character escapes.
  1400. */
  1401. protected void
  1402. file_showstr(FILE *fp, const char *s, size_t len)
  1403. {
  1404. char c;
  1405. for (;;) {
  1406. c = *s++;
  1407. if (len == ~0U) {
  1408. if (c == '\0')
  1409. break;
  1410. }
  1411. else {
  1412. if (len-- == 0)
  1413. break;
  1414. }
  1415. if (c >= 040 && c <= 0176) /* TODO isprint && !iscntrl */
  1416. (void) fputc(c, fp);
  1417. else {
  1418. (void) fputc('\\', fp);
  1419. switch (c) {
  1420. case '\a':
  1421. (void) fputc('a', fp);
  1422. break;
  1423. case '\b':
  1424. (void) fputc('b', fp);
  1425. break;
  1426. case '\f':
  1427. (void) fputc('f', fp);
  1428. break;
  1429. case '\n':
  1430. (void) fputc('n', fp);
  1431. break;
  1432. case '\r':
  1433. (void) fputc('r', fp);
  1434. break;
  1435. case '\t':
  1436. (void) fputc('t', fp);
  1437. break;
  1438. case '\v':
  1439. (void) fputc('v', fp);
  1440. break;
  1441. default:
  1442. (void) fprintf(fp, "%.3o", c & 0377);
  1443. break;
  1444. }
  1445. }
  1446. }
  1447. }
  1448. /*
  1449. * eatsize(): Eat the size spec from a number [eg. 10UL]
  1450. */
  1451. private void
  1452. eatsize(const char **p)
  1453. {
  1454. const char *l = *p;
  1455. if (LOWCASE(*l) == 'u')
  1456. l++;
  1457. switch (LOWCASE(*l)) {
  1458. case 'l': /* long */
  1459. case 's': /* short */
  1460. case 'h': /* short */
  1461. case 'b': /* char/byte */
  1462. case 'c': /* char/byte */
  1463. l++;
  1464. /*FALLTHROUGH*/
  1465. default:
  1466. break;
  1467. }
  1468. *p = l;
  1469. }
  1470. /*
  1471. * handle a compiled file.
  1472. */
  1473. private int
  1474. apprentice_map(struct magic_set *ms, struct magic **magicp, uint32_t *nmagicp,
  1475. const char *fn)
  1476. {
  1477. int fd;
  1478. struct stat st;
  1479. uint32_t *ptr;
  1480. uint32_t version;
  1481. int needsbyteswap;
  1482. char buf[MAXPATHLEN];
  1483. char *dbname = mkdbname(fn, buf, sizeof(buf), 0);
  1484. void *mm = NULL;
  1485. if (dbname == NULL)
  1486. return -1;
  1487. if ((fd = open(dbname, O_RDONLY|O_BINARY)) == -1)
  1488. return -1;
  1489. if (fstat(fd, &st) == -1) {
  1490. file_error(ms, errno, "cannot stat `%s'", dbname);
  1491. goto error;
  1492. }
  1493. if (st.st_size < 16) {
  1494. file_error(ms, 0, "file `%s' is too small", dbname);
  1495. goto error;
  1496. }
  1497. #ifdef QUICK
  1498. if ((mm = mmap(0, (size_t)st.st_size, PROT_READ|PROT_WRITE,
  1499. MAP_PRIVATE|MAP_FILE, fd, (off_t)0)) == MAP_FAILED) {
  1500. file_error(ms, errno, "cannot map `%s'", dbname);
  1501. goto error;
  1502. }
  1503. #define RET 2
  1504. #else
  1505. if ((mm = malloc((size_t)st.st_size)) == NULL) {
  1506. file_oomem(ms, (size_t)st.st_size);
  1507. goto error;
  1508. }
  1509. if (read(fd, mm, (size_t)st.st_size) != (size_t)st.st_size) {
  1510. file_badread(ms);
  1511. goto error;
  1512. }
  1513. #define RET 1
  1514. #endif
  1515. *magicp = mm;
  1516. (void)close(fd);
  1517. fd = -1;
  1518. ptr = (uint32_t *)(void *)*magicp;
  1519. if (*ptr != MAGICNO) {
  1520. if (swap4(*ptr) != MAGICNO) {
  1521. file_error(ms, 0, "bad magic in `%s'");
  1522. goto error;
  1523. }
  1524. needsbyteswap = 1;
  1525. } else
  1526. needsbyteswap = 0;
  1527. if (needsbyteswap)
  1528. version = swap4(ptr[1]);
  1529. else
  1530. version = ptr[1];
  1531. if (version != VERSIONNO) {
  1532. file_error(ms, 0, "version mismatch (%d != %d) in `%s'",
  1533. version, VERSIONNO, dbname);
  1534. goto error;
  1535. }
  1536. *nmagicp = (uint32_t)(st.st_size / sizeof(struct magic)) - 1;
  1537. (*magicp)++;
  1538. if (needsbyteswap)
  1539. byteswap(*magicp, *nmagicp);
  1540. return RET;
  1541. error:
  1542. if (fd != -1)
  1543. (void)close(fd);
  1544. if (mm) {
  1545. #ifdef QUICK
  1546. (void)munmap((void *)mm, (size_t)st.st_size);
  1547. #else
  1548. free(mm);
  1549. #endif
  1550. } else {
  1551. *magicp = NULL;
  1552. *nmagicp = 0;
  1553. }
  1554. return -1;
  1555. }
  1556. private const uint32_t ar[] = {
  1557. MAGICNO, VERSIONNO
  1558. };
  1559. /*
  1560. * handle an mmaped file.
  1561. */
  1562. private int
  1563. apprentice_compile(struct magic_set *ms, struct magic **magicp,
  1564. uint32_t *nmagicp, const char *fn)
  1565. {
  1566. int fd;
  1567. char buf[MAXPATHLEN];
  1568. char *dbname = mkdbname(fn, buf, sizeof(buf), 1);
  1569. if (dbname == NULL)
  1570. return -1;
  1571. if ((fd = open(dbname, O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644)) == -1) {
  1572. file_error(ms, errno, "cannot open `%s'", dbname);
  1573. return -1;
  1574. }
  1575. if (write(fd, ar, sizeof(ar)) != (ssize_t)sizeof(ar)) {
  1576. file_error(ms, errno, "error writing `%s'", dbname);
  1577. return -1;
  1578. }
  1579. if (lseek(fd, (off_t)sizeof(struct magic), SEEK_SET)
  1580. != sizeof(struct magic)) {
  1581. file_error(ms, errno, "error seeking `%s'", dbname);
  1582. return -1;
  1583. }
  1584. if (write(fd, *magicp, (sizeof(struct magic) * *nmagicp))
  1585. != (ssize_t)(sizeof(struct magic) * *nmagicp)) {
  1586. file_error(ms, errno, "error writing `%s'", dbname);
  1587. return -1;
  1588. }
  1589. (void)close(fd);
  1590. return 0;
  1591. }
  1592. private const char ext[] = ".mgc";
  1593. /*
  1594. * make a dbname
  1595. */
  1596. private char *
  1597. mkdbname(const char *fn, char *buf, size_t bufsiz, int strip)
  1598. {
  1599. if (strip) {
  1600. const char *p;
  1601. if ((p = strrchr(fn, '/')) != NULL)
  1602. fn = ++p;
  1603. }
  1604. (void)snprintf(buf, bufsiz, "%s%s", fn, ext);
  1605. return buf;
  1606. }
  1607. /*
  1608. * Byteswap an mmap'ed file if needed
  1609. */
  1610. private void
  1611. byteswap(struct magic *magic, uint32_t nmagic)
  1612. {
  1613. uint32_t i;
  1614. for (i = 0; i < nmagic; i++)
  1615. bs1(&magic[i]);
  1616. }
  1617. /*
  1618. * swap a short
  1619. */
  1620. private uint16_t
  1621. swap2(uint16_t sv)
  1622. {
  1623. uint16_t rv;
  1624. uint8_t *s = (uint8_t *)(void *)&sv;
  1625. uint8_t *d = (uint8_t *)(void *)&rv;
  1626. d[0] = s[1];
  1627. d[1] = s[0];
  1628. return rv;
  1629. }
  1630. /*
  1631. * swap an int
  1632. */
  1633. private uint32_t
  1634. swap4(uint32_t sv)
  1635. {
  1636. uint32_t rv;
  1637. uint8_t *s = (uint8_t *)(void *)&sv;
  1638. uint8_t *d = (uint8_t *)(void *)&rv;
  1639. d[0] = s[3];
  1640. d[1] = s[2];
  1641. d[2] = s[1];
  1642. d[3] = s[0];
  1643. return rv;
  1644. }
  1645. /*
  1646. * swap a quad
  1647. */
  1648. private uint64_t
  1649. swap8(uint64_t sv)
  1650. {
  1651. uint32_t rv;
  1652. uint8_t *s = (uint8_t *)(void *)&sv;
  1653. uint8_t *d = (uint8_t *)(void *)&rv;
  1654. d[0] = s[3];
  1655. d[1] = s[2];
  1656. d[2] = s[1];
  1657. d[3] = s[0];
  1658. d[4] = s[7];
  1659. d[5] = s[6];
  1660. d[6] = s[5];
  1661. d[7] = s[4];
  1662. return rv;
  1663. }
  1664. /*
  1665. * byteswap a single magic entry
  1666. */
  1667. private void
  1668. bs1(struct magic *m)
  1669. {
  1670. m->cont_level = swap2(m->cont_level);
  1671. m->offset = swap4((uint32_t)m->offset);
  1672. m->in_offset = swap4((uint32_t)m->in_offset);
  1673. m->lineno = swap4((uint32_t)m->lineno);
  1674. if (IS_STRING(m->type)) {
  1675. m->str_count = swap4(m->str_count);
  1676. m->str_flags = swap4(m->str_flags);
  1677. }
  1678. else {
  1679. m->value.q = swap8(m->value.q);
  1680. m->num_mask = swap8(m->num_mask);
  1681. }
  1682. }