forked from rurban/safeclib
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtest_memrchr_s.c
146 lines (114 loc) · 3.14 KB
/
test_memrchr_s.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
/*------------------------------------------------------------------
* test_memrchr_s
* File 'memrchr_s.c'
* Lines executed:100.00% of 17
*
*------------------------------------------------------------------
*/
#include "test_private.h"
#include "safe_mem_lib.h"
#include <string.h>
#ifdef HAVE_MEMRCHR
#ifndef _ISOC11_SOURCE
extern void *memrchr(const void *, int, size_t);
#endif
#endif
#define LEN (128)
static uint8_t mem[LEN];
int test_memrchr_s(void);
int test_memrchr_s(void) {
errno_t rc;
int ch;
void *sub;
#ifdef HAVE_MEMRCHR
void *std_sub;
#endif
size_t len;
int errs = 0;
/*--------------------------------------------------*/
ch = 0;
#ifndef HAVE_CT_BOS_OVR
EXPECT_BOS("empty dest")
rc = memrchr_s(NULL, LEN, ch, &sub);
ERR(ESNULLP);
SUBNULL();
EXPECT_BOS("empty result")
rc = memrchr_s(mem, LEN, ch, NULL);
ERR(ESNULLP);
EXPECT_BOS("empty dest or dmax")
rc = memrchr_s(mem, 0, ch, &sub);
ERR(ESZEROL)
SUBNULL();
EXPECT_BOS("dest overflow")
rc = memrchr_s(mem, RSIZE_MAX_MEM + 1, ch, &sub);
ERR(ESLEMAX)
SUBNULL();
#ifdef HAVE___BUILTIN_OBJECT_SIZE
EXPECT_BOS("dest overflow")
rc = memrchr_s(mem, LEN + 1, ch, &sub);
ERR(EOVERFLOW)
SUBNULL();
#endif
EXPECT_BOS("ch overflow >255")
rc = memrchr_s(mem, LEN, 256, &sub);
ERR(ESLEMAX)
SUBNULL();
#endif
/*--------------------------------------------------*/
mem[LEN - 1] = '\0';
rc = memrchr_s(mem, LEN, 0, &sub);
ERR(EOK);
PTREQ(sub, &mem[LEN - 1]);
/*--------------------------------------------------*/
len = 20;
/* 012345678901234567890 */
memcpy(mem, "keep it all together", len);
rc = memrchr_s(mem, len + 1, 0, &sub);
ERR(EOK);
PTREQ(sub, &mem[len]);
/*--------------------------------------------------*/
/* at beginning */
rc = memrchr_s(mem, len, 'k', &sub);
ERR(EOK)
PTREQ(sub, &mem[0]);
/* in the middle - left */
rc = memrchr_s(mem, 5, 'e', &sub);
ERR(EOK)
PTREQ(sub, &mem[2]);
/* in the middle - right */
rc = memrchr_s(mem, len, 'o', &sub);
ERR(EOK)
PTREQ(sub, &mem[13]);
/*--------------------------------------------------*/
rc = memrchr_s(mem, len, 'h', &sub);
ERR(EOK)
PTREQ(sub, &mem[17]);
rc = memrchr_s(mem, len, 'r', &sub);
ERR(EOK)
PTREQ(sub, &mem[19]);
/*--------------------------------------------------*/
rc = memrchr_s(mem, 3, 'i', &sub);
ERR(ESNOTFND)
SUBNULL();
rc = memrchr_s(mem, len, 'i', &sub);
ERR(EOK)
/*--------------------------------------------------*/
rc = memrchr_s(mem, len, 1, &sub);
ERR(ESNOTFND);
SUBNULL();
rc = memrchr_s(mem, len, 'e', &sub);
ERR(EOK)
PTREQ(sub, &mem[18]);
#ifdef HAVE_MEMRCHR
/* compare to legacy */
std_sub = (void *)memrchr(mem, 'e', len);
PTREQ(sub, std_sub);
#endif
/*--------------------------------------------------*/
return (errs);
}
#ifndef __KERNEL__
/* simple hack to get this to work for both userspace and Linux kernel,
until a better solution can be created. */
int main(void) { return (test_memrchr_s()); }
#endif