Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 1 | /* SPDX-License-Identifier: Apache-2.0 */ |
| 2 | /* Copyright (C) 2018 IBM Corp. */ |
| 3 | |
| 4 | #include <assert.h> |
| 5 | #include <string.h> |
| 6 | #include <vector> |
| 7 | #include <fstream> |
| 8 | #include <experimental/filesystem> |
| 9 | |
| 10 | #include "config.h" |
| 11 | #include "pnor_partition_table.hpp" |
| 12 | |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 13 | namespace openpower |
| 14 | { |
| 15 | namespace virtual_pnor |
| 16 | { |
| 17 | namespace test |
| 18 | { |
| 19 | |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 20 | namespace fs = std::experimental::filesystem; |
| 21 | |
| 22 | class VpnorRoot |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 23 | { |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 24 | public: |
| 25 | template <std::size_t N> |
| 26 | VpnorRoot(const std::string (&toc)[N], size_t blockSize) |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 27 | { |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 28 | char tmplt[] = "/tmp/vpnor_root.XXXXXX"; |
| 29 | char* tmpdir = mkdtemp(tmplt); |
| 30 | root = fs::path{tmpdir}; |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 31 | |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 32 | for (const auto& attr : attributes) |
| 33 | { |
| 34 | fs::create_directory(root / attr); |
| 35 | } |
| 36 | |
| 37 | fs::path tocFilePath = root / "ro" / PARTITION_TOC_FILE; |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 38 | |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 39 | for (const std::string& line : toc) |
| 40 | { |
| 41 | pnor_partition part; |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 42 | |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 43 | openpower::virtual_pnor::parseTocLine(line, blockSize, part); |
| 44 | |
| 45 | /* Populate the partition in the tree */ |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 46 | std::vector<char> zeroed(part.data.actual, 0); |
| 47 | fs::path partitionFilePath = root / "ro" / part.data.name; |
| 48 | std::ofstream(partitionFilePath) |
| 49 | .write(zeroed.data(), zeroed.size()); |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 50 | |
| 51 | /* Update the ToC if the partition file was created */ |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 52 | std::ofstream(tocFilePath, std::ofstream::app) << line << "\n"; |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 53 | } |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 54 | } |
| 55 | |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 56 | VpnorRoot(const VpnorRoot&) = delete; |
| 57 | VpnorRoot& operator=(const VpnorRoot&) = delete; |
| 58 | VpnorRoot(VpnorRoot&&) = delete; |
| 59 | VpnorRoot& operator=(VpnorRoot&&) = delete; |
| 60 | |
| 61 | ~VpnorRoot() |
| 62 | { |
| 63 | fs::remove_all(root); |
| 64 | } |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 65 | fs::path ro() |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 66 | { |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 67 | return fs::path{root} / "ro"; |
| 68 | } |
| 69 | fs::path rw() |
| 70 | { |
| 71 | return fs::path{root} / "rw"; |
| 72 | } |
| 73 | fs::path prsv() |
| 74 | { |
| 75 | return fs::path{root} / "prsv"; |
| 76 | } |
| 77 | fs::path patch() |
| 78 | { |
| 79 | return fs::path{root} / "patch"; |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 80 | } |
| 81 | size_t write(const std::string& name, const void* data, size_t len); |
| 82 | |
| 83 | private: |
| 84 | fs::path root; |
Andrew Jeffery | 097495c | 2018-02-22 11:36:13 +1030 | [diff] [blame] | 85 | const std::string attributes[4] = {"ro", "rw", "prsv", "patch"}; |
Andrew Jeffery | 812923d | 2018-02-22 11:59:52 +1030 | [diff] [blame] | 86 | }; |
Andrew Jeffery | d083efd | 2018-02-21 21:19:00 +1030 | [diff] [blame] | 87 | |
| 88 | } // test |
| 89 | } // virtual_pnor |
| 90 | } // openpower |