test.c revision 1.1.1.5
1/* $NetBSD: test.c,v 1.1.1.5 2019/12/17 02:23:53 christos Exp $ */ 2 3/* 4 * Copyright (c) Christos Zoulas 2003. 5 * All Rights Reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice immediately at the beginning of the file, without modification, 12 * this list of conditions, and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR 21 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 */ 29 30#include <stdio.h> 31#include <stdlib.h> 32#include <string.h> 33#include "magic.h" 34 35static void * 36xrealloc(void *p, size_t n) 37{ 38 p = realloc(p, n); 39 if (p == NULL) { 40 (void)fprintf(stderr, "ERROR slurping file: out of memory\n"); 41 exit(10); 42 } 43 return p; 44} 45 46static char * 47slurp(FILE *fp, size_t *final_len) 48{ 49 size_t len = 256; 50 int c; 51 char *l = (char *)xrealloc(NULL, len), *s = l; 52 53 for (c = getc(fp); c != EOF; c = getc(fp)) { 54 if (s == l + len) { 55 l = (char *)xrealloc(l, len * 2); 56 len *= 2; 57 } 58 *s++ = c; 59 } 60 if (s == l + len) 61 l = (char *)xrealloc(l, len + 1); 62 *s++ = '\0'; 63 64 *final_len = s - l; 65 l = (char *)xrealloc(l, s - l); 66 return l; 67} 68 69int 70main(int argc, char **argv) 71{ 72 struct magic_set *ms; 73 const char *result; 74 char *desired; 75 size_t desired_len; 76 int i; 77 FILE *fp; 78 79 ms = magic_open(MAGIC_NONE); 80 if (ms == NULL) { 81 (void)fprintf(stderr, "ERROR opening MAGIC_NONE: out of memory\n"); 82 return 10; 83 } 84 if (magic_load(ms, NULL) == -1) { 85 (void)fprintf(stderr, "ERROR loading with NULL file: %s\n", 86 magic_error(ms)); 87 return 11; 88 } 89 90 if (argc > 1) { 91 if (argc != 3) { 92 (void)fprintf(stderr, "Usage: test TEST-FILE RESULT\n"); 93 } else { 94 if ((result = magic_file(ms, argv[1])) == NULL) { 95 (void)fprintf(stderr, "ERROR loading file %s: %s\n", argv[1], magic_error(ms)); 96 return 12; 97 } else { 98 fp = fopen(argv[2], "r"); 99 if (fp == NULL) { 100 (void)fprintf(stderr, "ERROR opening `%s': ", argv[2]); 101 perror(NULL); 102 return 13; 103 } 104 desired = slurp(fp, &desired_len); 105 fclose(fp); 106 (void)printf("%s: %s\n", argv[1], result); 107 if (strcmp(result, desired) != 0) { 108 (void)fprintf(stderr, "Error: result was\n%s\nexpected:\n%s\n", result, desired); 109 return 1; 110 } 111 } 112 } 113 } 114 115 magic_close(ms); 116 return 0; 117} 118