Lakshminarayana R. Kammath | 27693a4 | 2019-06-24 00:51:47 -0500 | [diff] [blame] | 1 | #include "pldm_base_cmd.hpp" |
| 2 | |
| 3 | #include "pldm_cmd_helper.hpp" |
| 4 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 5 | #include "libpldm/utils.h" |
Lakshminarayana R. Kammath | 27693a4 | 2019-06-24 00:51:47 -0500 | [diff] [blame] | 6 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 7 | namespace pldmtool |
Lakshminarayana R. Kammath | 27693a4 | 2019-06-24 00:51:47 -0500 | [diff] [blame] | 8 | { |
Lakshminarayana R. Kammath | 27693a4 | 2019-06-24 00:51:47 -0500 | [diff] [blame] | 9 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 10 | namespace base |
Lakshminarayana R. Kammath | 27693a4 | 2019-06-24 00:51:47 -0500 | [diff] [blame] | 11 | { |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 12 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 13 | namespace |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 14 | { |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 15 | |
| 16 | using namespace pldmtool::helper; |
| 17 | |
| 18 | std::vector<std::unique_ptr<CommandInterface>> commands; |
| 19 | const std::map<const char*, pldm_supported_types> pldmTypes{ |
| 20 | {"base", PLDM_BASE}, {"platform", PLDM_PLATFORM}, {"bios", PLDM_BIOS}, |
| 21 | {"fru", PLDM_FRU}, {"oem", PLDM_OEM}, |
| 22 | }; |
| 23 | |
| 24 | } // namespace |
| 25 | |
| 26 | class GetPLDMTypes : public CommandInterface |
| 27 | { |
| 28 | public: |
| 29 | ~GetPLDMTypes() = default; |
| 30 | GetPLDMTypes() = delete; |
| 31 | GetPLDMTypes(const GetPLDMTypes&) = delete; |
| 32 | GetPLDMTypes(GetPLDMTypes&&) = default; |
| 33 | GetPLDMTypes& operator=(const GetPLDMTypes&) = delete; |
| 34 | GetPLDMTypes& operator=(GetPLDMTypes&&) = default; |
| 35 | |
| 36 | using CommandInterface::CommandInterface; |
| 37 | |
| 38 | std::pair<int, std::vector<uint8_t>> createRequestMsg() override |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 39 | { |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 40 | std::vector<uint8_t> requestMsg(sizeof(pldm_msg_hdr)); |
| 41 | auto request = reinterpret_cast<pldm_msg*>(requestMsg.data()); |
| 42 | auto rc = encode_get_types_req(PLDM_LOCAL_INSTANCE_ID, request); |
| 43 | return {rc, requestMsg}; |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 44 | } |
| 45 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 46 | void parseResponseMsg(pldm_msg* responsePtr, size_t payloadLength) override |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 47 | { |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 48 | uint8_t cc = 0; |
| 49 | std::vector<bitfield8_t> types(8); |
| 50 | auto rc = decode_get_types_resp(responsePtr, payloadLength, &cc, |
| 51 | types.data()); |
| 52 | if (rc != PLDM_SUCCESS || cc != PLDM_SUCCESS) |
| 53 | { |
| 54 | std::cerr << "Response Message Error: " |
Sampa Misra | aa8ae72 | 2019-12-12 03:20:40 -0600 | [diff] [blame] | 55 | << "rc=" << rc << ",cc=" << (int)cc << "\n"; |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 56 | return; |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 57 | } |
| 58 | |
| 59 | printPldmTypes(types); |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 60 | } |
| 61 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 62 | private: |
| 63 | void printPldmTypes(std::vector<bitfield8_t>& types) |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 64 | { |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 65 | std::cout << "Supported types:"; |
| 66 | for (int i = 0; i < PLDM_MAX_TYPES; i++) |
| 67 | { |
| 68 | bitfield8_t b = types[i / 8]; |
| 69 | if (b.byte & (1 << i % 8)) |
| 70 | { |
| 71 | std::cout << " " << i; |
| 72 | auto it = std::find_if( |
| 73 | pldmTypes.begin(), pldmTypes.end(), |
| 74 | [i](const auto& typePair) { return typePair.second == i; }); |
| 75 | if (it != pldmTypes.end()) |
| 76 | { |
| 77 | |
| 78 | std::cout << "(" << it->first << ")"; |
| 79 | } |
| 80 | } |
| 81 | } |
| 82 | |
| 83 | std::cout << std::endl; |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 84 | } |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 85 | }; |
| 86 | |
| 87 | class GetPLDMVersion : public CommandInterface |
| 88 | { |
| 89 | public: |
| 90 | ~GetPLDMVersion() = default; |
| 91 | GetPLDMVersion() = delete; |
| 92 | GetPLDMVersion(const GetPLDMVersion&) = delete; |
| 93 | GetPLDMVersion(GetPLDMVersion&&) = default; |
| 94 | GetPLDMVersion& operator=(const GetPLDMVersion&) = delete; |
| 95 | GetPLDMVersion& operator=(GetPLDMVersion&&) = default; |
| 96 | |
| 97 | explicit GetPLDMVersion(const char* type, const char* name, CLI::App* app) : |
| 98 | CommandInterface(type, name, app) |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 99 | { |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 100 | app->add_option("-t,--type", pldmType, "pldm supported type") |
| 101 | ->required() |
| 102 | ->transform(CLI::CheckedTransformer(pldmTypes, CLI::ignore_case)); |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 103 | } |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 104 | std::pair<int, std::vector<uint8_t>> createRequestMsg() override |
| 105 | { |
| 106 | std::vector<uint8_t> requestMsg(sizeof(pldm_msg_hdr) + |
| 107 | PLDM_GET_VERSION_REQ_BYTES); |
| 108 | auto request = reinterpret_cast<pldm_msg*>(requestMsg.data()); |
| 109 | |
| 110 | auto rc = encode_get_version_req(PLDM_LOCAL_INSTANCE_ID, 0, |
| 111 | PLDM_GET_FIRSTPART, pldmType, request); |
| 112 | return {rc, requestMsg}; |
| 113 | } |
| 114 | |
| 115 | void parseResponseMsg(pldm_msg* responsePtr, size_t payloadLength) override |
| 116 | { |
| 117 | uint8_t cc = 0, transferFlag = 0; |
| 118 | uint32_t transferHandle = 0; |
| 119 | ver32_t version; |
| 120 | auto rc = |
| 121 | decode_get_version_resp(responsePtr, payloadLength, &cc, |
| 122 | &transferHandle, &transferFlag, &version); |
| 123 | if (rc != PLDM_SUCCESS || cc != PLDM_SUCCESS) |
| 124 | { |
| 125 | std::cerr << "Response Message Error: " |
Sampa Misra | aa8ae72 | 2019-12-12 03:20:40 -0600 | [diff] [blame] | 126 | << "rc=" << rc << ",cc=" << (int)cc << "\n"; |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 127 | return; |
| 128 | } |
| 129 | char buffer[16] = {0}; |
| 130 | ver2str(&version, buffer, sizeof(buffer)); |
| 131 | std::cout << "Type " << pldmType; |
| 132 | auto it = std::find_if( |
| 133 | pldmTypes.begin(), pldmTypes.end(), |
| 134 | [&](const auto& typePair) { return typePair.second == pldmType; }); |
| 135 | |
| 136 | if (it != pldmTypes.end()) |
| 137 | { |
| 138 | std::cout << "(" << it->first << ")"; |
| 139 | } |
| 140 | std::cout << ": " << buffer << std::endl; |
| 141 | } |
| 142 | |
| 143 | private: |
| 144 | pldm_supported_types pldmType; |
| 145 | }; |
| 146 | |
Sridevi Ramesh | add56f0 | 2020-01-20 23:44:22 -0600 | [diff] [blame] | 147 | class GetTID : public CommandInterface |
| 148 | { |
| 149 | public: |
| 150 | ~GetTID() = default; |
| 151 | GetTID() = delete; |
| 152 | GetTID(const GetTID&) = delete; |
| 153 | GetTID(GetTID&&) = default; |
| 154 | GetTID& operator=(const GetTID&) = delete; |
| 155 | GetTID& operator=(GetTID&&) = default; |
| 156 | |
| 157 | using CommandInterface::CommandInterface; |
| 158 | |
| 159 | std::pair<int, std::vector<uint8_t>> createRequestMsg() override |
| 160 | { |
| 161 | std::vector<uint8_t> requestMsg(sizeof(pldm_msg_hdr)); |
| 162 | auto request = reinterpret_cast<pldm_msg*>(requestMsg.data()); |
| 163 | auto rc = encode_get_tid_req(PLDM_LOCAL_INSTANCE_ID, request); |
| 164 | return {rc, requestMsg}; |
| 165 | } |
| 166 | |
| 167 | void parseResponseMsg(pldm_msg* responsePtr, size_t payloadLength) override |
| 168 | { |
| 169 | uint8_t cc = 0; |
| 170 | uint8_t tid = 0; |
| 171 | std::vector<bitfield8_t> types(8); |
| 172 | auto rc = decode_get_tid_resp(responsePtr, payloadLength, &cc, &tid); |
| 173 | if (rc != PLDM_SUCCESS || cc != PLDM_SUCCESS) |
| 174 | { |
| 175 | std::cerr << "Response Message Error: " |
| 176 | << "rc=" << rc << ",cc=" << (int)cc << "\n"; |
| 177 | return; |
| 178 | } |
| 179 | std::cout << "Parsed Response Msg: " << std::endl; |
| 180 | std::cout << "TID : " << static_cast<uint32_t>(tid) << std::endl; |
| 181 | } |
| 182 | }; |
| 183 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 184 | void registerCommand(CLI::App& app) |
| 185 | { |
| 186 | auto base = app.add_subcommand("base", "base type command"); |
| 187 | base->require_subcommand(1); |
Sridevi Ramesh | add56f0 | 2020-01-20 23:44:22 -0600 | [diff] [blame] | 188 | |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 189 | auto getPLDMTypes = |
| 190 | base->add_subcommand("GetPLDMTypes", "get pldm supported types"); |
| 191 | commands.push_back( |
| 192 | std::make_unique<GetPLDMTypes>("base", "GetPLDMTypes", getPLDMTypes)); |
| 193 | |
| 194 | auto getPLDMVersion = |
| 195 | base->add_subcommand("GetPLDMVersion", "get version of a certain type"); |
| 196 | commands.push_back(std::make_unique<GetPLDMVersion>( |
| 197 | "base", "GetPLDMVersion", getPLDMVersion)); |
Sridevi Ramesh | add56f0 | 2020-01-20 23:44:22 -0600 | [diff] [blame] | 198 | |
| 199 | auto getPLDMTID = base->add_subcommand("GetTID", "get Terminus ID (TID)"); |
| 200 | commands.push_back(std::make_unique<GetTID>("base", "GetTID", getPLDMTID)); |
Lakshminarayana R. Kammath | 58229b2 | 2019-08-09 06:30:04 -0500 | [diff] [blame] | 201 | } |
John Wang | 58a0e06 | 2019-11-08 15:38:15 +0800 | [diff] [blame] | 202 | |
| 203 | } // namespace base |
| 204 | } // namespace pldmtool |