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
runTest(unsigned char * b1,unsigned char * b2)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);
ATF_TC_HEAD(memcpy_basic,tc)79 ATF_TC_HEAD(memcpy_basic, tc)
80 {
81           atf_tc_set_md_var(tc, "descr", "Test memcpy results");
82 }
83 
ATF_TC_BODY(memcpy_basic,tc)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);
ATF_TC_HEAD(memccpy_simple,tc)104 ATF_TC_HEAD(memccpy_simple, tc)
105 {
106         atf_tc_set_md_var(tc, "descr", "Test memccpy(3) results");
107 }
108 
ATF_TC_BODY(memccpy_simple,tc)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);
ATF_TC_HEAD(memcpy_return,tc)130 ATF_TC_HEAD(memcpy_return, tc)
131 {
132           atf_tc_set_md_var(tc, "descr", "Test memcpy(3) return value");
133 }
134 
ATF_TC_BODY(memcpy_return,tc)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 
ATF_TP_ADD_TCS(tp)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