1 /* $NetBSD: memtest.c,v 1.1.1.1 2022/09/24 20:07:55 christos Exp $ */ 2 3 /* 4 * Copyright (c) Christos Zoulas 2021. 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 #include "file.h" 30 31 #ifndef lint 32 #if 0 33 FILE_RCSID("@(#)$File: memtest.c,v 1.4 2022/09/16 14:12:14 christos Exp $") 34 #else 35 __RCSID("$NetBSD: memtest.c,v 1.1.1.1 2022/09/24 20:07:55 christos Exp $"); 36 #endif 37 #endif 38 39 #include <sys/types.h> 40 #include <sys/stat.h> 41 #include <sys/mman.h> 42 #include <stdio.h> 43 #include <stdbool.h> 44 #include <string.h> 45 #include <stdlib.h> 46 #include <err.h> 47 #include <fcntl.h> 48 #include <unistd.h> 49 #include <dlfcn.h> 50 #include <magic.h> 51 52 void * 53 malloc(size_t len) 54 { 55 char buf[512]; 56 void *(*orig)(size_t) = dlsym(RTLD_NEXT, "malloc"); 57 void *p = (*orig)(len); 58 int l = snprintf(buf, sizeof(buf), "malloc %zu %p\n", len, p); 59 write(2, buf, l); 60 return p; 61 } 62 63 void 64 free(void *p) 65 { 66 char buf[512]; 67 void (*orig)(void *) = dlsym(RTLD_NEXT, "free"); 68 (*orig)(p); 69 int l = snprintf(buf, sizeof(buf), "free %p\n", p); 70 write(2, buf, l); 71 } 72 73 void * 74 calloc(size_t len, size_t nitems) 75 { 76 char buf[512]; 77 void *(*orig)(size_t, size_t) = dlsym(RTLD_NEXT, "calloc"); 78 void *p = (*orig)(len, nitems); 79 size_t tot = len * nitems; 80 int l = snprintf(buf, sizeof(buf), "calloc %zu %p\n", tot, p); 81 write(2, buf, l); 82 return p; 83 } 84 void * 85 realloc(void *q, size_t len) 86 { 87 char buf[512]; 88 void *(*orig)(void *, size_t) = dlsym(RTLD_NEXT, "realloc"); 89 void *p = (*orig)(q, len); 90 int l = snprintf(buf, sizeof(buf), "realloc %zu %p\n", len, p); 91 write(2, buf, l); 92 return p; 93 } 94 95 static void 96 usage(void) 97 { 98 fprintf(stderr, "Usage: test [-b] <filename>\n"); 99 exit(EXIT_FAILURE); 100 } 101 102 int 103 main(int argc, char *argv[]) 104 { 105 bool buf = false; 106 int c; 107 108 while ((c = getopt(argc, argv, "b")) != -1) 109 switch (c) { 110 case 'b': 111 buf = true; 112 break; 113 default: 114 usage(); 115 } 116 117 argc -= optind; 118 argv += optind; 119 120 if (argc == 0) 121 usage(); 122 123 magic_t m = magic_open(0); 124 if (m == NULL) 125 err(EXIT_FAILURE, "magic_open"); 126 127 magic_load(m, NULL); 128 129 const char *r; 130 if (buf) { 131 int fd = open(argv[0], O_RDONLY); 132 if (fd == -1) 133 err(EXIT_FAILURE, "Cannot open `%s'", argv[0]); 134 135 struct stat st; 136 if (fstat(fd, &st) == -1) 137 err(EXIT_FAILURE, "Cannot stat `%s'", argv[0]); 138 size_t l = (size_t)st.st_size; 139 void *p = mmap(NULL, l, PROT_READ, MAP_FILE | MAP_PRIVATE, fd, 140 (off_t)0); 141 if (p == MAP_FAILED) 142 err(EXIT_FAILURE, "Cannot map `%s'", argv[0]); 143 close(fd); 144 r = magic_buffer(m, p, l); 145 munmap(p, l); 146 } else { 147 r = magic_file(m, argv[0]); 148 } 149 magic_close(m); 150 151 printf("%s\n", r ? r : "(null)"); 152 153 return 0; 154 } 155