1 /* 2 * Copyright (c) 2002 Kungliga Tekniska H�gskolan 3 * (Royal Institute of Technology, Stockholm, Sweden). 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 * 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 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 * 3. Neither the name of the Institute nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifdef HAVE_CONFIG_H 35 #include <config.h> 36 RCSID("$Id: parse_reply-test.c 15287 2005-05-29 21:21:12Z lha $"); 37 #endif 38 39 #include <sys/types.h> 40 #ifdef HAVE_SYS_MMAN_H 41 #include <sys/mman.h> 42 #endif 43 #include <fcntl.h> 44 45 #include "roken.h" 46 #include "resolve.h" 47 48 struct dns_reply* 49 parse_reply(const unsigned char *, size_t); 50 51 enum { MAX_BUF = 36}; 52 53 static struct testcase { 54 unsigned char buf[MAX_BUF]; 55 size_t buf_len; 56 } tests[] = { 57 {{0x12, 0x67, 0x84, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 58 0x03, 'f', 'o', 'o', 0x00, 59 0x00, 0x10, 0x00, 0x01, 60 0x03, 'f', 'o', 'o', 0x00, 61 0x00, 0x10, 0x00, 0x01, 62 0x00, 0x00, 0x12, 0x67, 0xff, 0xff}, 36} 63 }; 64 65 #ifndef MAP_FAILED 66 #define MAP_FAILED (-1) 67 #endif 68 69 static sig_atomic_t val = 0; 70 71 static RETSIGTYPE 72 segv_handler(int sig) 73 { 74 val = 1; 75 } 76 77 int 78 main(int argc, char **argv) 79 { 80 #ifndef HAVE_MMAP 81 return 77; /* signal to automake that this test 82 cannot be run */ 83 #else /* HAVE_MMAP */ 84 int ret; 85 int i; 86 struct sigaction sa; 87 88 sigemptyset (&sa.sa_mask); 89 sa.sa_flags = 0; 90 sa.sa_handler = segv_handler; 91 sigaction (SIGSEGV, &sa, NULL); 92 93 for (i = 0; val == 0 && i < sizeof(tests)/sizeof(tests[0]); ++i) { 94 const struct testcase *t = &tests[i]; 95 unsigned char *p1, *p2; 96 int flags; 97 int fd; 98 size_t pagesize = getpagesize(); 99 unsigned char *buf; 100 101 #ifdef MAP_ANON 102 flags = MAP_ANON; 103 fd = -1; 104 #else 105 flags = 0; 106 fd = open ("/dev/zero", O_RDONLY); 107 if(fd < 0) 108 err (1, "open /dev/zero"); 109 #endif 110 flags |= MAP_PRIVATE; 111 112 p1 = (unsigned char *)mmap(0, 2 * pagesize, PROT_READ | PROT_WRITE, 113 flags, fd, 0); 114 if (p1 == (unsigned char *)MAP_FAILED) 115 err (1, "mmap"); 116 p2 = p1 + pagesize; 117 ret = mprotect ((void *)p2, pagesize, 0); 118 if (ret < 0) 119 err (1, "mprotect"); 120 buf = p2 - t->buf_len; 121 memcpy (buf, t->buf, t->buf_len); 122 parse_reply (buf, t->buf_len); 123 ret = munmap ((void *)p1, 2 * pagesize); 124 if (ret < 0) 125 err (1, "munmap"); 126 } 127 return val; 128 #endif /* HAVE_MMAP */ 129 } 130