1 // SPDX-License-Identifier: Apache-2.0
2 // Copyright (C) 2018 IBM Corp.
3 
4 #include "config.h"
5 
6 extern "C" {
7 #include "common.h"
8 #include "flash.h"
9 #include "mboxd.h"
10 }
11 
12 #include "vpnor/test/tmpd.hpp"
13 
14 #include <fcntl.h>
15 #include <stdint.h>
16 #include <sys/ioctl.h>
17 #include <sys/mman.h>
18 #include <sys/syslog.h>
19 #include <unistd.h>
20 
21 #include <cassert>
22 #include <experimental/filesystem>
23 
24 static constexpr auto BLOCK_SIZE = 0x1000;
25 static constexpr auto PART_SIZE = BLOCK_SIZE;
26 static constexpr auto PATCH_SIZE = BLOCK_SIZE / 2;
27 static constexpr auto UPDATE_SIZE = BLOCK_SIZE;
28 
29 const std::string toc[] = {
30     "partition01=TEST1,00001000,00002000,80,ECC,READWRITE",
31 };
32 
33 int main(void)
34 {
35     namespace fs = std::experimental::filesystem;
36     namespace test = openpower::virtual_pnor::test;
37 
38     struct mbox_context _ctx, *ctx = &_ctx;
39     void* map;
40     int rc;
41     int fd;
42 
43     /* Setup */
44     memset(ctx, 0, sizeof(mbox_context));
45 
46     mbox_vlog = &mbox_log_console;
47     verbosity = (verbose)2;
48 
49     test::VpnorRoot root(ctx, toc, BLOCK_SIZE);
50     std::vector<uint8_t> roContent(PART_SIZE, 0xff);
51     root.write("TEST1", roContent.data(), roContent.size());
52     /* flash_write doesn't copy the file for us */
53     std::vector<uint8_t> patchContent(PATCH_SIZE, 0xaa);
54     root.patch("TEST1", patchContent.data(), patchContent.size());
55 
56     init_vpnor_from_paths(ctx);
57 
58     /* Test */
59     std::vector<uint8_t> update(UPDATE_SIZE, 0x55);
60     rc = flash_write(ctx, 0x1000, update.data(), update.size());
61     assert(rc == 0);
62 
63     /* Check that PATCH is modified with the new data */
64     fs::path patch = root.patch() / "TEST1";
65     assert(UPDATE_SIZE == fs::file_size(patch));
66     fd = open(patch.c_str(), O_RDONLY);
67     map = mmap(NULL, UPDATE_SIZE, PROT_READ, MAP_SHARED, fd, 0);
68     assert(map != MAP_FAILED);
69     rc = memcmp(update.data(), map, update.size());
70     assert(rc == 0);
71     munmap(map, update.size());
72     close(fd);
73 
74     destroy_vpnor(ctx);
75     free(ctx->flash_bmap);
76 
77     return rc;
78 }
79