1 /* $NetBSD: t_memmem.c,v 1.4 2018/10/15 17:55:28 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2005 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Perry E. Metzger of Metzger, Dowdeswell & Co. LLC. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 #include <atf-c.h> 33 34 #include <stdio.h> 35 #include <stdlib.h> 36 #include <string.h> 37 #include <sys/mman.h> 38 39 char p0[] = ""; 40 int lp0 = 0; 41 char p1[] = "0123"; 42 int lp1 = 4; 43 char p2[] = "456"; 44 int lp2 = 3; 45 char p3[] = "789"; 46 int lp3 = 3; 47 char p4[] = "abc"; 48 int lp4 = 3; 49 char p5[] = "0"; 50 int lp5 = 1; 51 char p6[] = "9"; 52 int lp6 = 1; 53 char p7[] = "654"; 54 int lp7 = 3; 55 char p8[] = "89abc"; 56 int lp8 = 5; 57 58 char b0[] = ""; 59 int lb0 = 0; 60 char b1[] = "0"; 61 int lb1 = 1; 62 char b2[] = "0123456789"; 63 int lb2 = 10; 64 65 #define expect(b) \ 66 if (!(b)) { \ 67 fprintf(stderr, "failed on line %d\n", __LINE__); \ 68 atf_tc_fail("Check stderr for test id/line"); \ 69 } 70 71 ATF_TC(memmem_basic); 72 ATF_TC_HEAD(memmem_basic, tc) 73 { 74 75 atf_tc_set_md_var(tc, "descr", "Test memmem results"); 76 } 77 78 ATF_TC_BODY(memmem_basic, tc) 79 { 80 81 expect(memmem(b2, lb2, p0, lp0) == b2); 82 expect(memmem(b0, lb0, p0, lp0) == b0); 83 expect(memmem(b0, lb0, p1, lp1) == NULL); 84 expect(memmem(b1, lb1, p1, lp1) == NULL); 85 86 expect(memmem(b2, lb2, p1, lp1) == b2); 87 expect(memmem(b2, lb2, p2, lp2) == (b2 + 4)); 88 expect(memmem(b2, lb2, p3, lp3) == (b2 + 7)); 89 90 expect(memmem(b2, lb2, p5, lp5) == b2); 91 expect(memmem(b2, lb2, p6, lp6) == (b2 + 9)); 92 93 expect(memmem(b2, lb2, p4, lp4) == NULL); 94 expect(memmem(b2, lb2, p7, lp7) == NULL); 95 expect(memmem(b2, lb2, p8, lp8) == NULL); 96 } 97 98 ATF_TC(memmem_oob); 99 ATF_TC_HEAD(memmem_oob, tc) 100 { 101 atf_tc_set_md_var(tc, "descr", "Test memmem out of bounds read"); 102 } 103 104 ATF_TC_BODY(memmem_oob, tc) 105 { 106 static const char str[] = "abcde"; 107 size_t pg = getpagesize(); 108 char *src = mmap(NULL, pg, PROT_READ|PROT_WRITE, 109 MAP_ANON|MAP_PRIVATE, -1, (off_t)0); 110 ATF_CHECK(src != MAP_FAILED); 111 char *guard = mmap(src + pg, pg, 112 PROT_NONE, MAP_ANON|MAP_PRIVATE|MAP_FIXED, -1, (off_t)0); 113 for (size_t i = 2; i < 5; i++) { 114 char *search = src + pg - i; 115 char match[sizeof(str)]; 116 search[-1] = str[0]; 117 search[0] = str[0]; 118 search[1] = str[0]; 119 memcpy(match, str, i); 120 ATF_CHECK(memmem(search, i, match, i) != search); 121 } 122 munmap(guard, pg); 123 munmap(src, pg); 124 } 125 126 127 ATF_TP_ADD_TCS(tp) 128 { 129 130 ATF_TP_ADD_TC(tp, memmem_basic); 131 ATF_TP_ADD_TC(tp, memmem_oob); 132 133 return atf_no_error(); 134 } 135