/* $NetBSD: test.c,v 1.1.1.8 2023/08/18 18:36:51 christos Exp $ */ /* * Copyright (c) Christos Zoulas 2003. * All Rights Reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice immediately at the beginning of the file, without modification, * this list of conditions, and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #include #include #include #include #include #include #include "magic.h" static const char *prog; static void * xrealloc(void *p, size_t n) { p = realloc(p, n); if (p == NULL) { (void)fprintf(stderr, "%s ERROR slurping file: %s\n", prog, strerror(errno)); exit(10); } return p; } static char * slurp(FILE *fp, size_t *final_len) { size_t len = 256; int c; char *l = xrealloc(NULL, len), *s = l; for (c = getc(fp); c != EOF; c = getc(fp)) { if (s == l + len) { s = l + len; len *= 2; l = xrealloc(l, len); } *s++ = c; } if (s != l && s[-1] == '\n') s--; if (s == l + len) { l = xrealloc(l, len + 1); s = l + len; } *s++ = '\0'; *final_len = s - l; return xrealloc(l, s - l); } int main(int argc, char **argv) { struct magic_set *ms = NULL; const char *result; size_t result_len, desired_len; char *desired = NULL; int e = EXIT_FAILURE, flags, c; FILE *fp; setenv("TZ", "UTC", 1); tzset(); prog = strrchr(argv[0], '/'); if (prog) prog++; else prog = argv[0]; if (argc == 1) return 0; flags = 0; while ((c = getopt(argc, argv, "ek")) != -1) switch (c) { case 'e': flags |= MAGIC_ERROR; break; case 'k': flags |= MAGIC_CONTINUE; break; default: goto usage; } argc -= optind; argv += optind; if (argc != 2) { usage: (void)fprintf(stderr, "Usage: %s [-ek] TEST-FILE RESULT\n", prog); goto bad; } ms = magic_open(flags); if (ms == NULL) { (void)fprintf(stderr, "%s: ERROR opening MAGIC_NONE: %s\n", prog, strerror(errno)); return e; } if (magic_load(ms, NULL) == -1) { (void)fprintf(stderr, "%s: ERROR loading with NULL file: %s\n", prog, magic_error(ms)); goto bad; } if ((result = magic_file(ms, argv[0])) == NULL) { (void)fprintf(stderr, "%s: ERROR loading file %s: %s\n", prog, argv[1], magic_error(ms)); goto bad; } fp = fopen(argv[1], "r"); if (fp == NULL) { (void)fprintf(stderr, "%s: ERROR opening `%s': %s", prog, argv[1], strerror(errno)); goto bad; } desired = slurp(fp, &desired_len); fclose(fp); (void)printf("%s: %s\n", argv[0], result); if (strcmp(result, desired) != 0) { result_len = strlen(result); (void)fprintf(stderr, "%s: ERROR: result was (len %zu)\n%s\n" "expected (len %zu)\n%s\n", prog, result_len, result, desired_len, desired); goto bad; } e = 0; bad: free(desired); if (ms) magic_close(ms); return e; }