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