1 /* $NetBSD: t_memcpy.c,v 1.6 2017/01/11 18:05:54 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2010 The NetBSD Foundation, Inc. 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, 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 NETBSD FOUNDATION, INC. AND CONTRIBUTORS 17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26 * POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #include <atf-c.h> 30 31 #include <stdio.h> 32 #include <stdlib.h> 33 #include <string.h> 34 #include <assert.h> 35 #include <md5.h> 36 37 #include <sys/types.h> 38 39 #define ALIGNMENTS 16 40 #define LENGTHS 4 41 #define BLOCKTYPES 4 42 43 MD5_CTX mc[1]; 44 45 typedef unsigned char testBlock_t[ALIGNMENTS * LENGTHS]; 46 47 testBlock_t bss1, bss2; 48 49 unsigned char *start[BLOCKTYPES] = { 50 bss1, bss2 51 }; 52 53 char result[100]; 54 const char goodResult[] = "7b405d24bc03195474c70ddae9e1f8fb"; 55 56 static void 57 runTest(unsigned char *b1, unsigned char *b2) 58 { 59 int i, j, k, m; 60 size_t n; 61 62 for (i = 0; i < ALIGNMENTS; ++i) { 63 for (j = 0; j < ALIGNMENTS; ++j) { 64 k = sizeof(testBlock_t) - (i > j ? i : j); 65 for (m = 0; m < k; ++m) { 66 for (n = 0; n < sizeof(testBlock_t); ++n) { 67 b1[n] = (unsigned char)random(); 68 b2[n] = (unsigned char)random(); 69 } 70 memcpy(b1 + i, b2 + j, m); 71 MD5Update(mc, b1, sizeof(testBlock_t)); 72 MD5Update(mc, b2, sizeof(testBlock_t)); 73 } 74 } 75 } 76 } 77 78 ATF_TC(memcpy_basic); 79 ATF_TC_HEAD(memcpy_basic, tc) 80 { 81 atf_tc_set_md_var(tc, "descr", "Test memcpy results"); 82 } 83 84 ATF_TC_BODY(memcpy_basic, tc) 85 { 86 int i, j; 87 testBlock_t auto1, auto2; 88 89 start[2] = auto1; 90 start[3] = auto2; 91 92 srandom(0L); 93 MD5Init(mc); 94 for (i = 0; i < BLOCKTYPES; ++i) 95 for (j = 0; j < BLOCKTYPES; ++j) 96 if (i != j) 97 runTest(start[i], start[j]); 98 MD5End(mc, result); 99 ATF_REQUIRE_EQ_MSG(strcmp(result, goodResult), 0, "%s != %s", 100 result, goodResult); 101 } 102 103 ATF_TC(memccpy_simple); 104 ATF_TC_HEAD(memccpy_simple, tc) 105 { 106 atf_tc_set_md_var(tc, "descr", "Test memccpy(3) results"); 107 } 108 109 ATF_TC_BODY(memccpy_simple, tc) 110 { 111 char buf[100]; 112 char c = ' '; 113 114 (void)memset(buf, c, sizeof(buf)); 115 116 ATF_CHECK(memccpy(buf, "foo bar", c, sizeof(buf)) != NULL); 117 ATF_CHECK(buf[4] == c); 118 119 ATF_CHECK(memccpy(buf, "foo bar", '\0', sizeof(buf) - 1) != NULL); 120 ATF_CHECK(buf[8] == c); 121 122 ATF_CHECK(memccpy(buf, "foo bar", 'x', 7) == NULL); 123 ATF_CHECK(strncmp(buf, "foo bar", 7) == 0); 124 125 ATF_CHECK(memccpy(buf, "xxxxxxx", 'r', 7) == NULL); 126 ATF_CHECK(strncmp(buf, "xxxxxxx", 7) == 0); 127 } 128 129 ATF_TC(memcpy_return); 130 ATF_TC_HEAD(memcpy_return, tc) 131 { 132 atf_tc_set_md_var(tc, "descr", "Test memcpy(3) return value"); 133 } 134 135 ATF_TC_BODY(memcpy_return, tc) 136 { 137 char *b = (char *)0x1; 138 char c[2]; 139 ATF_REQUIRE_EQ(memcpy(b, b, 0), b); 140 ATF_REQUIRE_EQ(memcpy(c, "ab", sizeof(c)), c); 141 } 142 143 ATF_TP_ADD_TCS(tp) 144 { 145 146 ATF_TP_ADD_TC(tp, memcpy_basic); 147 ATF_TP_ADD_TC(tp, memcpy_return); 148 ATF_TP_ADD_TC(tp, memccpy_simple); 149 150 return atf_no_error(); 151 } 152