1 /* $NetBSD: test.c,v 1.1.1.2 2014/06/13 01:48:26 christos Exp $ */ 2 /* 3 * Copyright (c) Christos Zoulas 2003. 4 * All Rights Reserved. 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 #include <stdio.h> 30 #include <stdlib.h> 31 #include <string.h> 32 #include "magic.h" 33 34 static void * 35 xrealloc(void *p, size_t n) 36 { 37 p = realloc(p, n); 38 if (p == NULL) { 39 (void)fprintf(stderr, "ERROR slurping file: out of memory\n"); 40 exit(10); 41 } 42 return p; 43 } 44 45 static char * 46 slurp(FILE *fp, size_t *final_len) 47 { 48 size_t len = 256; 49 int c; 50 char *l = (char *)xrealloc(NULL, len), *s = l; 51 52 for (c = getc(fp); c != EOF; c = getc(fp)) { 53 if (s == l + len) { 54 l = (char *)xrealloc(l, len * 2); 55 len *= 2; 56 } 57 *s++ = c; 58 } 59 if (s == l + len) 60 l = (char *)xrealloc(l, len + 1); 61 *s++ = '\0'; 62 63 *final_len = s - l; 64 l = (char *)xrealloc(l, s - l); 65 return l; 66 } 67 68 int 69 main(int argc, char **argv) 70 { 71 struct magic_set *ms; 72 const char *result; 73 char *desired; 74 size_t desired_len; 75 int i; 76 FILE *fp; 77 78 ms = magic_open(MAGIC_NONE); 79 if (ms == NULL) { 80 (void)fprintf(stderr, "ERROR opening MAGIC_NONE: out of memory\n"); 81 return 10; 82 } 83 if (magic_load(ms, NULL) == -1) { 84 (void)fprintf(stderr, "ERROR loading with NULL file: %s\n", magic_error(ms)); 85 return 11; 86 } 87 88 if (argc > 1) { 89 if (argc != 3) { 90 (void)fprintf(stderr, "Usage: test TEST-FILE RESULT\n"); 91 } else { 92 if ((result = magic_file(ms, argv[1])) == NULL) { 93 (void)fprintf(stderr, "ERROR loading file %s: %s\n", argv[1], magic_error(ms)); 94 return 12; 95 } else { 96 fp = fopen(argv[2], "r"); 97 if (fp == NULL) { 98 (void)fprintf(stderr, "ERROR opening `%s': ", argv[2]); 99 perror(NULL); 100 return 13; 101 } 102 desired = slurp(fp, &desired_len); 103 fclose(fp); 104 (void)printf("%s: %s\n", argv[1], result); 105 if (strcmp(result, desired) != 0) { 106 (void)fprintf(stderr, "Error: result was\n%s\nexpected:\n%s\n", result, desired); 107 return 1; 108 } 109 } 110 } 111 } 112 113 magic_close(ms); 114 return 0; 115 } 116