flash: Rename write_flash to flash_write
Change-Id: I4b41aac597299c93369d8998b03a7d3e2a128742
Signed-off-by: Andrew Jeffery <andrew@aj.id.au>
diff --git a/test/flash_write.c b/test/flash_write.c
new file mode 100644
index 0000000..cec6c92
--- /dev/null
+++ b/test/flash_write.c
@@ -0,0 +1,105 @@
+// SPDX-License-Identifier: Apache-2.0
+// Copyright (C) 2018 IBM Corp.
+
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/ioctl.h>
+#include <sys/mman.h>
+#include <unistd.h>
+
+#include "mbox.h"
+#include "flash.h"
+
+#include "test/tmpf.h"
+
+struct tmpf _tmp, *tmp = &_tmp;
+
+void cleanup(void)
+{
+ tmpf_destroy(tmp);
+}
+
+char *get_dev_mtd(void)
+{
+ int rc;
+
+ rc = tmpf_init(tmp, "flash-store.XXXXXX");
+ if (rc < 0)
+ return NULL;
+
+ return strdup(tmp->path);
+}
+
+#define MEM_SIZE 3
+#define ERASE_SIZE 1
+
+int ioctl(int fd, unsigned long request, ...)
+{
+ va_list ap;
+
+ if (request != MEMGETINFO)
+ return -1;
+
+ 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);
+
+ return 0;
+}
+
+int main(void)
+{
+ struct mbox_context _context, *context = &_context;
+ char src[MEM_SIZE];
+ uint8_t *map;
+ int rc;
+
+ atexit(cleanup);
+
+ mbox_vlog = &mbox_log_console;
+
+ rc = flash_dev_init(context);
+ assert(rc == 0);
+
+ map = mmap(NULL, MEM_SIZE, PROT_READ, MAP_PRIVATE, tmp->fd, 0);
+ assert(map != MAP_FAILED);
+
+ memset(src, 0xaa, sizeof(src));
+ rc = flash_write(context, 0, src, sizeof(src));
+ assert(rc == 0);
+ rc = memcmp(src, map, sizeof(src));
+ assert(rc == 0);
+
+ memset(src, 0x55, sizeof(src));
+ rc = flash_write(context, 0, src, sizeof(src));
+ assert(rc == 0);
+ rc = memcmp(src, map, sizeof(src));
+ assert(rc == 0);
+
+ src[0] = 0xff;
+ rc = flash_write(context, 0, src, 1);
+ assert(rc == 0);
+ rc = memcmp(src, map, sizeof(src));
+ assert(rc == 0);
+
+ src[1] = 0xff;
+ rc = flash_write(context, 1, &src[1], 1);
+ assert(rc == 0);
+ rc = memcmp(src, map, sizeof(src));
+ assert(rc == 0);
+
+ src[2] = 0xff;
+ rc = flash_write(context, 2, &src[2], 1);
+ assert(rc == 0);
+ rc = memcmp(src, map, sizeof(src));
+ assert(rc == 0);
+
+ flash_dev_free(context);
+
+ return rc;
+}