test: Add erase_flash
Change-Id: Idbe0e8a9d7c62eeb5ac90fce055fc87de80ab1fa
Signed-off-by: Andrew Jeffery <andrew@aj.id.au>
diff --git a/Makefile.am b/Makefile.am
index f1491d5..7212558 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -19,7 +19,10 @@
test_copy_flash_SOURCES = test/copy_flash.c mboxd_flash.c common.c mtd.c test/tmpf.c
+test_erase_flash_SOURCES = test/erase_flash.c mboxd_flash.c common.c test/tmpf.c
+
check_PROGRAMS = test/sanity \
- test/copy_flash
+ test/copy_flash \
+ test/erase_flash
TESTS = $(check_PROGRAMS)
diff --git a/test/erase_flash.c b/test/erase_flash.c
new file mode 100644
index 0000000..23a536f
--- /dev/null
+++ b/test/erase_flash.c
@@ -0,0 +1,224 @@
+/*
+ * MBox Daemon Test File
+ *
+ * Copyright 2017 IBM
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+#include <assert.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+#include <linux/types.h>
+
+#include "mbox.h"
+#include "mboxd_flash.h"
+
+#include "test/tmpf.h"
+
+static struct tmpf mtd;
+
+void cleanup_mtd(void)
+{
+ tmpf_destroy(&mtd);
+}
+
+char *get_dev_mtd(void)
+{
+ int rc;
+
+ rc = tmpf_init(&mtd, "flashXXXXXX");
+ if (rc < 0)
+ return NULL;
+
+ return strdup(mtd.path);
+}
+
+struct erase_info_user *recorded;
+int n_ioctls;
+
+#define MEM_SIZE 3
+#define ERASE_SIZE 1
+
+int ioctl(int fd, unsigned long request, ...)
+{
+ va_list ap;
+ struct erase_info_user *provided, *alloced;
+
+ if (!(request == MEMERASE || request == MEMGETINFO)) {
+ printf("Uh-oh: ioctl() called with request 0x%08lx\n", request);
+ return -1;
+ }
+
+ switch (request) {
+ case MEMGETINFO:
+ {
+ struct mtd_info_user *info;
+
+ va_start(ap, request);
+ info = va_arg(ap, struct mtd_info_user *);
+ info->size = MEM_SIZE;
+ info->erasesize = ERASE_SIZE;
+ va_end(ap);
+ break;
+ }
+ case MEMERASE:
+ va_start(ap, request);
+ provided = va_arg(ap, struct erase_info_user *);
+
+ n_ioctls++;
+
+ alloced = realloc(recorded, n_ioctls * sizeof(*recorded));
+ if (!alloced)
+ return -1;
+ recorded = alloced;
+
+ memcpy(&recorded[n_ioctls - 1], provided, sizeof(*provided));
+
+ va_end(ap);
+ break;
+ default:
+ break;
+ }
+
+ return 0;
+}
+
+void dump_ioctls(void)
+{
+ int i;
+
+ printf("n_ioctls: %d\n", n_ioctls);
+
+ for (i = 0; i < n_ioctls; i++)
+ printf("%d: start: %d, length %d\n",
+ i, recorded[i].start, recorded[i].length);
+}
+
+int main(void)
+{
+ struct mbox_context context;
+ char data[MEM_SIZE];
+ int rc;
+
+ rc = atexit(cleanup_mtd);
+ if (rc)
+ return rc;
+
+ n_ioctls = 0;
+ recorded = NULL;
+
+ init_flash_dev(&context);
+
+ /* Erase from an unknown state */
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 1);
+ assert(recorded[0].start == 0);
+ assert(recorded[0].length == sizeof(data));
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ /* Erase an erased flash */
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 0);
+
+ memset(data, 0xaa, sizeof(data));
+
+ /* Erase written flash */
+ rc = write_flash(&context, 0, data, sizeof(data));
+ assert(rc == 0);
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 1);
+ assert(recorded[0].start == 0);
+ assert(recorded[0].length == sizeof(data));
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ /* Erase the start of flash */
+ rc = write_flash(&context, 0, data, sizeof(data) - 1);
+ assert(rc == 0);
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 1);
+ assert(recorded[0].start == 0);
+ assert(recorded[0].length == sizeof(data) - 1);
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ /* Erase the end of flash */
+ rc = write_flash(&context, 1, data, sizeof(data) - 1);
+ assert(rc == 0);
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 1);
+ assert(recorded[0].start == 1);
+ assert(recorded[0].length == sizeof(data) - 1);
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ /* Erase each end of flash */
+ rc = write_flash(&context, 0, data, 1);
+ rc = write_flash(&context, 2, data, 1);
+ assert(rc == 0);
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 2);
+ assert(recorded[0].start == 0);
+ assert(recorded[0].length == 1);
+ assert(recorded[1].start == 2);
+ assert(recorded[1].length == 1);
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ /* Erase the middle of flash */
+ rc = write_flash(&context, 1, data, 1);
+ assert(rc == 0);
+ rc = erase_flash(&context, 0, sizeof(data));
+
+ assert(rc == 0);
+ assert(n_ioctls == 1);
+ assert(recorded[0].start == 1);
+ assert(recorded[0].length == 1);
+
+ free(recorded);
+ recorded = NULL;
+ n_ioctls = 0;
+
+ free_flash_dev(&context);
+
+ return rc;
+}