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