funcs.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847
  1. /*
  2. * Copyright (c) Christos Zoulas 2003.
  3. * All Rights Reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions
  7. * are met:
  8. * 1. Redistributions of source code must retain the above copyright
  9. * notice immediately at the beginning of the file, without modification,
  10. * this list of conditions, and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
  19. * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25. * SUCH DAMAGE.
  26. */
  27. #include "file.h"
  28. #ifndef lint
  29. FILE_RCSID("@(#)$File: funcs.c,v 1.122 2021/06/30 10:08:48 christos Exp $")
  30. #endif /* lint */
  31. #include "magic.h"
  32. #include <assert.h>
  33. #include <stdarg.h>
  34. #include <stdlib.h>
  35. #include <string.h>
  36. #include <ctype.h>
  37. #ifdef HAVE_UNISTD_H
  38. #include <unistd.h> /* for pipe2() */
  39. #endif
  40. #if defined(HAVE_WCHAR_H)
  41. #include <wchar.h>
  42. #endif
  43. #if defined(HAVE_WCTYPE_H)
  44. #include <wctype.h>
  45. #endif
  46. #include <limits.h>
  47. #ifndef SIZE_MAX
  48. #define SIZE_MAX ((size_t)~0)
  49. #endif
  50. protected char *
  51. file_copystr(char *buf, size_t blen, size_t width, const char *str)
  52. {
  53. if (++width > blen)
  54. width = blen;
  55. strlcpy(buf, str, width);
  56. return buf;
  57. }
  58. private void
  59. file_clearbuf(struct magic_set *ms)
  60. {
  61. free(ms->o.buf);
  62. ms->o.buf = NULL;
  63. ms->o.blen = 0;
  64. }
  65. private int
  66. file_checkfield(char *msg, size_t mlen, const char *what, const char **pp)
  67. {
  68. const char *p = *pp;
  69. int fw = 0;
  70. while (*p && isdigit((unsigned char)*p))
  71. fw = fw * 10 + (*p++ - '0');
  72. *pp = p;
  73. if (fw < 1024)
  74. return 1;
  75. if (msg)
  76. snprintf(msg, mlen, "field %s too large: %d", what, fw);
  77. return 0;
  78. }
  79. protected int
  80. file_checkfmt(char *msg, size_t mlen, const char *fmt)
  81. {
  82. for (const char *p = fmt; *p; p++) {
  83. if (*p != '%')
  84. continue;
  85. if (*++p == '%')
  86. continue;
  87. // Skip uninteresting.
  88. while (strchr("#0.'+- ", *p) != NULL)
  89. p++;
  90. if (*p == '*') {
  91. if (msg)
  92. snprintf(msg, mlen, "* not allowed in format");
  93. return -1;
  94. }
  95. if (!file_checkfield(msg, mlen, "width", &p))
  96. return -1;
  97. if (*p == '.') {
  98. p++;
  99. if (!file_checkfield(msg, mlen, "precision", &p))
  100. return -1;
  101. }
  102. if (!isalpha((unsigned char)*p)) {
  103. if (msg)
  104. snprintf(msg, mlen, "bad format char: %c", *p);
  105. return -1;
  106. }
  107. }
  108. return 0;
  109. }
  110. /*
  111. * Like printf, only we append to a buffer.
  112. */
  113. protected int
  114. file_vprintf(struct magic_set *ms, const char *fmt, va_list ap)
  115. {
  116. int len;
  117. char *buf, *newstr;
  118. char tbuf[1024];
  119. if (ms->event_flags & EVENT_HAD_ERR)
  120. return 0;
  121. if (file_checkfmt(tbuf, sizeof(tbuf), fmt)) {
  122. file_clearbuf(ms);
  123. file_error(ms, 0, "Bad magic format `%s' (%s)", fmt, tbuf);
  124. return -1;
  125. }
  126. len = vasprintf(&buf, fmt, ap);
  127. if (len < 0 || (size_t)len > 1024 || len + ms->o.blen > 1024 * 1024) {
  128. size_t blen = ms->o.blen;
  129. free(buf);
  130. file_clearbuf(ms);
  131. file_error(ms, 0, "Output buffer space exceeded %d+%zu", len,
  132. blen);
  133. return -1;
  134. }
  135. if (ms->o.buf != NULL) {
  136. len = asprintf(&newstr, "%s%s", ms->o.buf, buf);
  137. free(buf);
  138. if (len < 0)
  139. goto out;
  140. free(ms->o.buf);
  141. buf = newstr;
  142. }
  143. ms->o.buf = buf;
  144. ms->o.blen = len;
  145. return 0;
  146. out:
  147. file_clearbuf(ms);
  148. file_error(ms, errno, "vasprintf failed");
  149. return -1;
  150. }
  151. protected int
  152. file_printf(struct magic_set *ms, const char *fmt, ...)
  153. {
  154. int rv;
  155. va_list ap;
  156. va_start(ap, fmt);
  157. rv = file_vprintf(ms, fmt, ap);
  158. va_end(ap);
  159. return rv;
  160. }
  161. /*
  162. * error - print best error message possible
  163. */
  164. /*VARARGS*/
  165. __attribute__((__format__(__printf__, 3, 0)))
  166. private void
  167. file_error_core(struct magic_set *ms, int error, const char *f, va_list va,
  168. size_t lineno)
  169. {
  170. /* Only the first error is ok */
  171. if (ms->event_flags & EVENT_HAD_ERR)
  172. return;
  173. if (lineno != 0) {
  174. file_clearbuf(ms);
  175. (void)file_printf(ms, "line %" SIZE_T_FORMAT "u:", lineno);
  176. }
  177. if (ms->o.buf && *ms->o.buf)
  178. (void)file_printf(ms, " ");
  179. (void)file_vprintf(ms, f, va);
  180. if (error > 0)
  181. (void)file_printf(ms, " (%s)", strerror(error));
  182. ms->event_flags |= EVENT_HAD_ERR;
  183. ms->error = error;
  184. }
  185. /*VARARGS*/
  186. protected void
  187. file_error(struct magic_set *ms, int error, const char *f, ...)
  188. {
  189. va_list va;
  190. va_start(va, f);
  191. file_error_core(ms, error, f, va, 0);
  192. va_end(va);
  193. }
  194. /*
  195. * Print an error with magic line number.
  196. */
  197. /*VARARGS*/
  198. protected void
  199. file_magerror(struct magic_set *ms, const char *f, ...)
  200. {
  201. va_list va;
  202. va_start(va, f);
  203. file_error_core(ms, 0, f, va, ms->line);
  204. va_end(va);
  205. }
  206. protected void
  207. file_oomem(struct magic_set *ms, size_t len)
  208. {
  209. file_error(ms, errno, "cannot allocate %" SIZE_T_FORMAT "u bytes",
  210. len);
  211. }
  212. protected void
  213. file_badseek(struct magic_set *ms)
  214. {
  215. file_error(ms, errno, "error seeking");
  216. }
  217. protected void
  218. file_badread(struct magic_set *ms)
  219. {
  220. file_error(ms, errno, "error reading");
  221. }
  222. #ifndef COMPILE_ONLY
  223. #define FILE_SEPARATOR "\n- "
  224. protected int
  225. file_separator(struct magic_set *ms)
  226. {
  227. return file_printf(ms, FILE_SEPARATOR);
  228. }
  229. static void
  230. trim_separator(struct magic_set *ms)
  231. {
  232. size_t l;
  233. if (ms->o.buf == NULL)
  234. return;
  235. l = strlen(ms->o.buf);
  236. if (l < sizeof(FILE_SEPARATOR))
  237. return;
  238. l -= sizeof(FILE_SEPARATOR) - 1;
  239. if (strcmp(ms->o.buf + l, FILE_SEPARATOR) != 0)
  240. return;
  241. ms->o.buf[l] = '\0';
  242. }
  243. static int
  244. checkdone(struct magic_set *ms, int *rv)
  245. {
  246. if ((ms->flags & MAGIC_CONTINUE) == 0)
  247. return 1;
  248. if (file_separator(ms) == -1)
  249. *rv = -1;
  250. return 0;
  251. }
  252. protected int
  253. file_default(struct magic_set *ms, size_t nb)
  254. {
  255. if (ms->flags & MAGIC_MIME) {
  256. if ((ms->flags & MAGIC_MIME_TYPE) &&
  257. file_printf(ms, "application/%s",
  258. nb ? "octet-stream" : "x-empty") == -1)
  259. return -1;
  260. return 1;
  261. }
  262. if (ms->flags & MAGIC_APPLE) {
  263. if (file_printf(ms, "UNKNUNKN") == -1)
  264. return -1;
  265. return 1;
  266. }
  267. if (ms->flags & MAGIC_EXTENSION) {
  268. if (file_printf(ms, "???") == -1)
  269. return -1;
  270. return 1;
  271. }
  272. return 0;
  273. }
  274. /*
  275. * The magic detection functions return:
  276. * 1: found
  277. * 0: not found
  278. * -1: error
  279. */
  280. /*ARGSUSED*/
  281. protected int
  282. file_buffer(struct magic_set *ms, int fd, struct stat *st,
  283. const char *inname __attribute__ ((__unused__)),
  284. const void *buf, size_t nb)
  285. {
  286. int m = 0, rv = 0, looks_text = 0;
  287. const char *code = NULL;
  288. const char *code_mime = "binary";
  289. const char *def = "data";
  290. const char *ftype = NULL;
  291. char *rbuf = NULL;
  292. struct buffer b;
  293. buffer_init(&b, fd, st, buf, nb);
  294. ms->mode = b.st.st_mode;
  295. if (nb == 0) {
  296. def = "empty";
  297. goto simple;
  298. } else if (nb == 1) {
  299. def = "very short file (no magic)";
  300. goto simple;
  301. }
  302. if ((ms->flags & MAGIC_NO_CHECK_ENCODING) == 0) {
  303. looks_text = file_encoding(ms, &b, NULL, 0,
  304. &code, &code_mime, &ftype);
  305. }
  306. #ifdef __EMX__
  307. if ((ms->flags & MAGIC_NO_CHECK_APPTYPE) == 0 && inname) {
  308. m = file_os2_apptype(ms, inname, &b);
  309. if ((ms->flags & MAGIC_DEBUG) != 0)
  310. (void)fprintf(stderr, "[try os2_apptype %d]\n", m);
  311. switch (m) {
  312. case -1:
  313. return -1;
  314. case 0:
  315. break;
  316. default:
  317. return 1;
  318. }
  319. }
  320. #endif
  321. #if HAVE_FORK
  322. /* try compression stuff */
  323. if ((ms->flags & MAGIC_NO_CHECK_COMPRESS) == 0) {
  324. m = file_zmagic(ms, &b, inname);
  325. if ((ms->flags & MAGIC_DEBUG) != 0)
  326. (void)fprintf(stderr, "[try zmagic %d]\n", m);
  327. if (m) {
  328. goto done_encoding;
  329. }
  330. }
  331. #endif
  332. /* Check if we have a tar file */
  333. if ((ms->flags & MAGIC_NO_CHECK_TAR) == 0) {
  334. m = file_is_tar(ms, &b);
  335. if ((ms->flags & MAGIC_DEBUG) != 0)
  336. (void)fprintf(stderr, "[try tar %d]\n", m);
  337. if (m) {
  338. if (checkdone(ms, &rv))
  339. goto done;
  340. }
  341. }
  342. /* Check if we have a JSON file */
  343. if ((ms->flags & MAGIC_NO_CHECK_JSON) == 0) {
  344. m = file_is_json(ms, &b);
  345. if ((ms->flags & MAGIC_DEBUG) != 0)
  346. (void)fprintf(stderr, "[try json %d]\n", m);
  347. if (m) {
  348. if (checkdone(ms, &rv))
  349. goto done;
  350. }
  351. }
  352. /* Check if we have a CSV file */
  353. if ((ms->flags & MAGIC_NO_CHECK_CSV) == 0) {
  354. m = file_is_csv(ms, &b, looks_text);
  355. if ((ms->flags & MAGIC_DEBUG) != 0)
  356. (void)fprintf(stderr, "[try csv %d]\n", m);
  357. if (m) {
  358. if (checkdone(ms, &rv))
  359. goto done;
  360. }
  361. }
  362. /* Check if we have a CDF file */
  363. if ((ms->flags & MAGIC_NO_CHECK_CDF) == 0) {
  364. m = file_trycdf(ms, &b);
  365. if ((ms->flags & MAGIC_DEBUG) != 0)
  366. (void)fprintf(stderr, "[try cdf %d]\n", m);
  367. if (m) {
  368. if (checkdone(ms, &rv))
  369. goto done;
  370. }
  371. }
  372. #ifdef BUILTIN_ELF
  373. if ((ms->flags & MAGIC_NO_CHECK_ELF) == 0 && nb > 5 && fd != -1) {
  374. file_pushbuf_t *pb;
  375. /*
  376. * We matched something in the file, so this
  377. * *might* be an ELF file, and the file is at
  378. * least 5 bytes long, so if it's an ELF file
  379. * it has at least one byte past the ELF magic
  380. * number - try extracting information from the
  381. * ELF headers that cannot easily be extracted
  382. * with rules in the magic file. We we don't
  383. * print the information yet.
  384. */
  385. if ((pb = file_push_buffer(ms)) == NULL)
  386. return -1;
  387. rv = file_tryelf(ms, &b);
  388. rbuf = file_pop_buffer(ms, pb);
  389. if (rv == -1) {
  390. free(rbuf);
  391. rbuf = NULL;
  392. }
  393. if ((ms->flags & MAGIC_DEBUG) != 0)
  394. (void)fprintf(stderr, "[try elf %d]\n", m);
  395. }
  396. #endif
  397. /* try soft magic tests */
  398. if ((ms->flags & MAGIC_NO_CHECK_SOFT) == 0) {
  399. m = file_softmagic(ms, &b, NULL, NULL, BINTEST, looks_text);
  400. if ((ms->flags & MAGIC_DEBUG) != 0)
  401. (void)fprintf(stderr, "[try softmagic %d]\n", m);
  402. if (m == 1 && rbuf) {
  403. if (file_printf(ms, "%s", rbuf) == -1)
  404. goto done;
  405. }
  406. if (m) {
  407. if (checkdone(ms, &rv))
  408. goto done;
  409. }
  410. }
  411. /* try text properties */
  412. if ((ms->flags & MAGIC_NO_CHECK_TEXT) == 0) {
  413. m = file_ascmagic(ms, &b, looks_text);
  414. if ((ms->flags & MAGIC_DEBUG) != 0)
  415. (void)fprintf(stderr, "[try ascmagic %d]\n", m);
  416. if (m) {
  417. goto done;
  418. }
  419. }
  420. simple:
  421. /* give up */
  422. if (m == 0) {
  423. m = 1;
  424. rv = file_default(ms, nb);
  425. if (rv == 0)
  426. if (file_printf(ms, "%s", def) == -1)
  427. rv = -1;
  428. }
  429. done:
  430. trim_separator(ms);
  431. if ((ms->flags & MAGIC_MIME_ENCODING) != 0) {
  432. if (ms->flags & MAGIC_MIME_TYPE)
  433. if (file_printf(ms, "; charset=") == -1)
  434. rv = -1;
  435. if (file_printf(ms, "%s", code_mime) == -1)
  436. rv = -1;
  437. }
  438. #if HAVE_FORK
  439. done_encoding:
  440. #endif
  441. free(rbuf);
  442. buffer_fini(&b);
  443. if (rv)
  444. return rv;
  445. return m;
  446. }
  447. #endif
  448. protected int
  449. file_reset(struct magic_set *ms, int checkloaded)
  450. {
  451. if (checkloaded && ms->mlist[0] == NULL) {
  452. file_error(ms, 0, "no magic files loaded");
  453. return -1;
  454. }
  455. file_clearbuf(ms);
  456. if (ms->o.pbuf) {
  457. free(ms->o.pbuf);
  458. ms->o.pbuf = NULL;
  459. }
  460. ms->event_flags &= ~EVENT_HAD_ERR;
  461. ms->error = -1;
  462. return 0;
  463. }
  464. #define OCTALIFY(n, o) \
  465. /*LINTED*/ \
  466. (void)(*(n)++ = '\\', \
  467. *(n)++ = ((CAST(uint32_t, *(o)) >> 6) & 3) + '0', \
  468. *(n)++ = ((CAST(uint32_t, *(o)) >> 3) & 7) + '0', \
  469. *(n)++ = ((CAST(uint32_t, *(o)) >> 0) & 7) + '0', \
  470. (o)++)
  471. protected const char *
  472. file_getbuffer(struct magic_set *ms)
  473. {
  474. char *pbuf, *op, *np;
  475. size_t psize, len;
  476. if (ms->event_flags & EVENT_HAD_ERR)
  477. return NULL;
  478. if (ms->flags & MAGIC_RAW)
  479. return ms->o.buf;
  480. if (ms->o.buf == NULL)
  481. return NULL;
  482. /* * 4 is for octal representation, + 1 is for NUL */
  483. len = strlen(ms->o.buf);
  484. if (len > (SIZE_MAX - 1) / 4) {
  485. file_oomem(ms, len);
  486. return NULL;
  487. }
  488. psize = len * 4 + 1;
  489. if ((pbuf = CAST(char *, realloc(ms->o.pbuf, psize))) == NULL) {
  490. file_oomem(ms, psize);
  491. return NULL;
  492. }
  493. ms->o.pbuf = pbuf;
  494. #if defined(HAVE_WCHAR_H) && defined(HAVE_MBRTOWC) && defined(HAVE_WCWIDTH)
  495. {
  496. mbstate_t state;
  497. wchar_t nextchar;
  498. int mb_conv = 1;
  499. size_t bytesconsumed;
  500. char *eop;
  501. (void)memset(&state, 0, sizeof(mbstate_t));
  502. np = ms->o.pbuf;
  503. op = ms->o.buf;
  504. eop = op + len;
  505. while (op < eop) {
  506. bytesconsumed = mbrtowc(&nextchar, op,
  507. CAST(size_t, eop - op), &state);
  508. if (bytesconsumed == CAST(size_t, -1) ||
  509. bytesconsumed == CAST(size_t, -2)) {
  510. mb_conv = 0;
  511. break;
  512. }
  513. if (iswprint(nextchar)) {
  514. (void)memcpy(np, op, bytesconsumed);
  515. op += bytesconsumed;
  516. np += bytesconsumed;
  517. } else {
  518. while (bytesconsumed-- > 0)
  519. OCTALIFY(np, op);
  520. }
  521. }
  522. *np = '\0';
  523. /* Parsing succeeded as a multi-byte sequence */
  524. if (mb_conv != 0)
  525. return ms->o.pbuf;
  526. }
  527. #endif
  528. for (np = ms->o.pbuf, op = ms->o.buf; *op;) {
  529. if (isprint(CAST(unsigned char, *op))) {
  530. *np++ = *op++;
  531. } else {
  532. OCTALIFY(np, op);
  533. }
  534. }
  535. *np = '\0';
  536. return ms->o.pbuf;
  537. }
  538. protected int
  539. file_check_mem(struct magic_set *ms, unsigned int level)
  540. {
  541. size_t len;
  542. if (level >= ms->c.len) {
  543. len = (ms->c.len = 20 + level) * sizeof(*ms->c.li);
  544. ms->c.li = CAST(struct level_info *, (ms->c.li == NULL) ?
  545. malloc(len) :
  546. realloc(ms->c.li, len));
  547. if (ms->c.li == NULL) {
  548. file_oomem(ms, len);
  549. return -1;
  550. }
  551. }
  552. ms->c.li[level].got_match = 0;
  553. #ifdef ENABLE_CONDITIONALS
  554. ms->c.li[level].last_match = 0;
  555. ms->c.li[level].last_cond = COND_NONE;
  556. #endif /* ENABLE_CONDITIONALS */
  557. return 0;
  558. }
  559. protected size_t
  560. file_printedlen(const struct magic_set *ms)
  561. {
  562. return ms->o.blen;
  563. }
  564. protected int
  565. file_replace(struct magic_set *ms, const char *pat, const char *rep)
  566. {
  567. file_regex_t rx;
  568. int rc, rv = -1;
  569. rc = file_regcomp(&rx, pat, REG_EXTENDED);
  570. if (rc) {
  571. file_regerror(&rx, rc, ms);
  572. } else {
  573. regmatch_t rm;
  574. int nm = 0;
  575. while (file_regexec(&rx, ms->o.buf, 1, &rm, 0) == 0) {
  576. ms->o.buf[rm.rm_so] = '\0';
  577. if (file_printf(ms, "%s%s", rep,
  578. rm.rm_eo != 0 ? ms->o.buf + rm.rm_eo : "") == -1)
  579. goto out;
  580. nm++;
  581. }
  582. rv = nm;
  583. }
  584. out:
  585. file_regfree(&rx);
  586. return rv;
  587. }
  588. protected int
  589. file_regcomp(file_regex_t *rx, const char *pat, int flags)
  590. {
  591. #ifdef USE_C_LOCALE
  592. rx->c_lc_ctype = newlocale(LC_CTYPE_MASK, "C", 0);
  593. assert(rx->c_lc_ctype != NULL);
  594. rx->old_lc_ctype = uselocale(rx->c_lc_ctype);
  595. assert(rx->old_lc_ctype != NULL);
  596. #else
  597. rx->old_lc_ctype = setlocale(LC_CTYPE, NULL);
  598. assert(rx->old_lc_ctype != NULL);
  599. rx->old_lc_ctype = strdup(rx->old_lc_ctype);
  600. assert(rx->old_lc_ctype != NULL);
  601. (void)setlocale(LC_CTYPE, "C");
  602. #endif
  603. rx->pat = pat;
  604. return rx->rc = regcomp(&rx->rx, pat, flags);
  605. }
  606. protected int
  607. file_regexec(file_regex_t *rx, const char *str, size_t nmatch,
  608. regmatch_t* pmatch, int eflags)
  609. {
  610. assert(rx->rc == 0);
  611. /* XXX: force initialization because glibc does not always do this */
  612. if (nmatch != 0)
  613. memset(pmatch, 0, nmatch * sizeof(*pmatch));
  614. return regexec(&rx->rx, str, nmatch, pmatch, eflags);
  615. }
  616. protected void
  617. file_regfree(file_regex_t *rx)
  618. {
  619. if (rx->rc == 0)
  620. regfree(&rx->rx);
  621. #ifdef USE_C_LOCALE
  622. (void)uselocale(rx->old_lc_ctype);
  623. freelocale(rx->c_lc_ctype);
  624. #else
  625. (void)setlocale(LC_CTYPE, rx->old_lc_ctype);
  626. free(rx->old_lc_ctype);
  627. #endif
  628. }
  629. protected void
  630. file_regerror(file_regex_t *rx, int rc, struct magic_set *ms)
  631. {
  632. char errmsg[512];
  633. (void)regerror(rc, &rx->rx, errmsg, sizeof(errmsg));
  634. file_magerror(ms, "regex error %d for `%s', (%s)", rc, rx->pat,
  635. errmsg);
  636. }
  637. protected file_pushbuf_t *
  638. file_push_buffer(struct magic_set *ms)
  639. {
  640. file_pushbuf_t *pb;
  641. if (ms->event_flags & EVENT_HAD_ERR)
  642. return NULL;
  643. if ((pb = (CAST(file_pushbuf_t *, malloc(sizeof(*pb))))) == NULL)
  644. return NULL;
  645. pb->buf = ms->o.buf;
  646. pb->blen = ms->o.blen;
  647. pb->offset = ms->offset;
  648. ms->o.buf = NULL;
  649. ms->o.blen = 0;
  650. ms->offset = 0;
  651. return pb;
  652. }
  653. protected char *
  654. file_pop_buffer(struct magic_set *ms, file_pushbuf_t *pb)
  655. {
  656. char *rbuf;
  657. if (ms->event_flags & EVENT_HAD_ERR) {
  658. free(pb->buf);
  659. free(pb);
  660. return NULL;
  661. }
  662. rbuf = ms->o.buf;
  663. ms->o.buf = pb->buf;
  664. ms->o.blen = pb->blen;
  665. ms->offset = pb->offset;
  666. free(pb);
  667. return rbuf;
  668. }
  669. /*
  670. * convert string to ascii printable format.
  671. */
  672. protected char *
  673. file_printable(struct magic_set *ms, char *buf, size_t bufsiz,
  674. const char *str, size_t slen)
  675. {
  676. char *ptr, *eptr = buf + bufsiz - 1;
  677. const unsigned char *s = RCAST(const unsigned char *, str);
  678. const unsigned char *es = s + slen;
  679. for (ptr = buf; ptr < eptr && s < es && *s; s++) {
  680. if ((ms->flags & MAGIC_RAW) != 0 || isprint(*s)) {
  681. *ptr++ = *s;
  682. continue;
  683. }
  684. if (ptr >= eptr - 3)
  685. break;
  686. *ptr++ = '\\';
  687. *ptr++ = ((CAST(unsigned int, *s) >> 6) & 7) + '0';
  688. *ptr++ = ((CAST(unsigned int, *s) >> 3) & 7) + '0';
  689. *ptr++ = ((CAST(unsigned int, *s) >> 0) & 7) + '0';
  690. }
  691. *ptr = '\0';
  692. return buf;
  693. }
  694. struct guid {
  695. uint32_t data1;
  696. uint16_t data2;
  697. uint16_t data3;
  698. uint8_t data4[8];
  699. };
  700. protected int
  701. file_parse_guid(const char *s, uint64_t *guid)
  702. {
  703. struct guid *g = CAST(struct guid *, CAST(void *, guid));
  704. return sscanf(s,
  705. "%8x-%4hx-%4hx-%2hhx%2hhx-%2hhx%2hhx%2hhx%2hhx%2hhx%2hhx",
  706. &g->data1, &g->data2, &g->data3, &g->data4[0], &g->data4[1],
  707. &g->data4[2], &g->data4[3], &g->data4[4], &g->data4[5],
  708. &g->data4[6], &g->data4[7]) == 11 ? 0 : -1;
  709. }
  710. protected int
  711. file_print_guid(char *str, size_t len, const uint64_t *guid)
  712. {
  713. const struct guid *g = CAST(const struct guid *,
  714. CAST(const void *, guid));
  715. return snprintf(str, len, "%.8X-%.4hX-%.4hX-%.2hhX%.2hhX-"
  716. "%.2hhX%.2hhX%.2hhX%.2hhX%.2hhX%.2hhX",
  717. g->data1, g->data2, g->data3, g->data4[0], g->data4[1],
  718. g->data4[2], g->data4[3], g->data4[4], g->data4[5],
  719. g->data4[6], g->data4[7]);
  720. }
  721. protected int
  722. file_pipe_closexec(int *fds)
  723. {
  724. #ifdef HAVE_PIPE2
  725. return pipe2(fds, O_CLOEXEC);
  726. #else
  727. if (pipe(fds) == -1)
  728. return -1;
  729. (void)fcntl(fds[0], F_SETFD, FD_CLOEXEC);
  730. (void)fcntl(fds[1], F_SETFD, FD_CLOEXEC);
  731. return 0;
  732. #endif
  733. }
  734. protected int
  735. file_clear_closexec(int fd) {
  736. return fcntl(fd, F_SETFD, 0);
  737. }
  738. protected char *
  739. file_strtrim(char *str)
  740. {
  741. char *last;
  742. while (isspace(CAST(unsigned char, *str)))
  743. str++;
  744. last = str;
  745. while (*last)
  746. last++;
  747. --last;
  748. while (isspace(CAST(unsigned char, *last)))
  749. last--;
  750. *++last = '\0';
  751. return str;
  752. }