blob: fa8c92c36d6f07a92e843663035fcfbf73be5a6a [file] [log] [blame]
Andrew Jefferyf37ae022017-04-12 14:21:37 +09301/*
2 * MBox Daemon Test File
3 *
4 * Copyright 2017 IBM
5 *
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 *
18 */
19
20#include <assert.h>
21#include <stdarg.h>
22#include <stdlib.h>
23#include <stdio.h>
24#include <sys/ioctl.h>
25#include <unistd.h>
26
27#include <linux/types.h>
28
29#include "mbox.h"
30#include "mboxd_flash.h"
31
32#include "test/tmpf.h"
33
34static struct tmpf mtd;
35
36void cleanup_mtd(void)
37{
38 tmpf_destroy(&mtd);
39}
40
41char *get_dev_mtd(void)
42{
43 int rc;
44
Andrew Jefferyc3144042018-02-26 13:24:52 +103045 rc = tmpf_init(&mtd, "flash-store.XXXXXX");
Andrew Jefferyf37ae022017-04-12 14:21:37 +093046 if (rc < 0)
47 return NULL;
48
49 return strdup(mtd.path);
50}
51
52struct erase_info_user *recorded;
53int n_ioctls;
54
55#define MEM_SIZE 3
56#define ERASE_SIZE 1
57
58int ioctl(int fd, unsigned long request, ...)
59{
60 va_list ap;
61 struct erase_info_user *provided, *alloced;
62
63 if (!(request == MEMERASE || request == MEMGETINFO)) {
64 printf("Uh-oh: ioctl() called with request 0x%08lx\n", request);
65 return -1;
66 }
67
68 switch (request) {
69 case MEMGETINFO:
70 {
71 struct mtd_info_user *info;
72
73 va_start(ap, request);
74 info = va_arg(ap, struct mtd_info_user *);
75 info->size = MEM_SIZE;
76 info->erasesize = ERASE_SIZE;
77 va_end(ap);
78 break;
79 }
80 case MEMERASE:
81 va_start(ap, request);
82 provided = va_arg(ap, struct erase_info_user *);
83
84 n_ioctls++;
85
86 alloced = realloc(recorded, n_ioctls * sizeof(*recorded));
87 if (!alloced)
88 return -1;
89 recorded = alloced;
90
91 memcpy(&recorded[n_ioctls - 1], provided, sizeof(*provided));
92
93 va_end(ap);
94 break;
95 default:
96 break;
97 }
98
99 return 0;
100}
101
102void dump_ioctls(void)
103{
104 int i;
105
106 printf("n_ioctls: %d\n", n_ioctls);
107
108 for (i = 0; i < n_ioctls; i++)
109 printf("%d: start: %d, length %d\n",
110 i, recorded[i].start, recorded[i].length);
111}
112
113int main(void)
114{
115 struct mbox_context context;
116 char data[MEM_SIZE];
117 int rc;
118
119 rc = atexit(cleanup_mtd);
120 if (rc)
121 return rc;
122
Suraj Jitindar Singh6f3197d2017-05-03 16:57:25 +1000123 mbox_vlog = &mbox_log_console;
124
Andrew Jefferyf37ae022017-04-12 14:21:37 +0930125 n_ioctls = 0;
126 recorded = NULL;
127
128 init_flash_dev(&context);
129
130 /* Erase from an unknown state */
131 rc = erase_flash(&context, 0, sizeof(data));
132
133 assert(rc == 0);
134 assert(n_ioctls == 1);
135 assert(recorded[0].start == 0);
136 assert(recorded[0].length == sizeof(data));
137
138 free(recorded);
139 recorded = NULL;
140 n_ioctls = 0;
141
142 /* Erase an erased flash */
143 rc = erase_flash(&context, 0, sizeof(data));
144
145 assert(rc == 0);
146 assert(n_ioctls == 0);
147
148 memset(data, 0xaa, sizeof(data));
149
150 /* Erase written flash */
151 rc = write_flash(&context, 0, data, sizeof(data));
152 assert(rc == 0);
153 rc = erase_flash(&context, 0, sizeof(data));
154
155 assert(rc == 0);
156 assert(n_ioctls == 1);
157 assert(recorded[0].start == 0);
158 assert(recorded[0].length == sizeof(data));
159
160 free(recorded);
161 recorded = NULL;
162 n_ioctls = 0;
163
164 /* Erase the start of flash */
165 rc = write_flash(&context, 0, data, sizeof(data) - 1);
166 assert(rc == 0);
167 rc = erase_flash(&context, 0, sizeof(data));
168
169 assert(rc == 0);
170 assert(n_ioctls == 1);
171 assert(recorded[0].start == 0);
172 assert(recorded[0].length == sizeof(data) - 1);
173
174 free(recorded);
175 recorded = NULL;
176 n_ioctls = 0;
177
178 /* Erase the end of flash */
179 rc = write_flash(&context, 1, data, sizeof(data) - 1);
180 assert(rc == 0);
181 rc = erase_flash(&context, 0, sizeof(data));
182
183 assert(rc == 0);
184 assert(n_ioctls == 1);
185 assert(recorded[0].start == 1);
186 assert(recorded[0].length == sizeof(data) - 1);
187
188 free(recorded);
189 recorded = NULL;
190 n_ioctls = 0;
191
192 /* Erase each end of flash */
193 rc = write_flash(&context, 0, data, 1);
194 rc = write_flash(&context, 2, data, 1);
195 assert(rc == 0);
196 rc = erase_flash(&context, 0, sizeof(data));
197
198 assert(rc == 0);
199 assert(n_ioctls == 2);
200 assert(recorded[0].start == 0);
201 assert(recorded[0].length == 1);
202 assert(recorded[1].start == 2);
203 assert(recorded[1].length == 1);
204
205 free(recorded);
206 recorded = NULL;
207 n_ioctls = 0;
208
209 /* Erase the middle of flash */
210 rc = write_flash(&context, 1, data, 1);
211 assert(rc == 0);
212 rc = erase_flash(&context, 0, sizeof(data));
213
214 assert(rc == 0);
215 assert(n_ioctls == 1);
216 assert(recorded[0].start == 1);
217 assert(recorded[0].length == 1);
218
219 free(recorded);
220 recorded = NULL;
221 n_ioctls = 0;
222
223 free_flash_dev(&context);
224
225 return rc;
226}