xref: /openbmc/hiomapd/vpnor/test/write_patch.cpp (revision 261f61a1)
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 <sys/ioctl.h>
16 #include <sys/mman.h>
17 #include <sys/syslog.h>
18 #include <unistd.h>
19 
20 #include <cassert>
21 #include <experimental/filesystem>
22 
23 static constexpr auto BLOCK_SIZE = 0x1000;
24 static constexpr auto DATA_SIZE = 8;
25 
26 const uint8_t data[DATA_SIZE] = {0xa0, 0xa1, 0xa2, 0xa3,
27                                  0xa4, 0xa5, 0xa6, 0xa7};
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     char src[DATA_SIZE]{0};
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     root.write("TEST1", data, sizeof(data));
52     /* flash_write doesn't copy the file for us */
53     assert(fs::copy_file(root.ro() / "TEST1", root.rw() / "TEST1"));
54     fs::path patch = root.patch() / "TEST1";
55     assert(fs::copy_file(root.ro() / "TEST1", patch));
56 
57     init_vpnor_from_paths(ctx);
58 
59     /* Test */
60     memset(src, 0x33, sizeof(src));
61     rc = flash_write(ctx, 0x1000, src, sizeof(src));
62     assert(rc == 0);
63 
64     /* Check that RW file is unmodified after the patch write */
65     fd = open((root.rw() / "TEST1").c_str(), O_RDONLY);
66     map = mmap(NULL, sizeof(src), PROT_READ, MAP_SHARED, fd, 0);
67     assert(map != MAP_FAILED);
68     rc = memcmp(data, map, sizeof(src));
69     assert(rc == 0);
70     munmap(map, sizeof(src));
71     close(fd);
72 
73     /* Check that PATCH is modified with the new data */
74     fd = open(patch.c_str(), O_RDONLY);
75     map = mmap(NULL, sizeof(src), PROT_READ, MAP_SHARED, fd, 0);
76     assert(map != MAP_FAILED);
77     rc = memcmp(src, map, sizeof(src));
78     assert(rc == 0);
79     munmap(map, sizeof(src));
80     close(fd);
81 
82     destroy_vpnor(ctx);
83     free(ctx->flash_bmap);
84 
85     return rc;
86 }
87