1 /* 2 * Copyright 2019 Google Inc. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #include "buildjson.hpp" 17 18 #include "file_handler.hpp" 19 #include "fs.hpp" 20 #include "prepare_systemd.hpp" 21 #include "update_systemd.hpp" 22 #include "verify_systemd.hpp" 23 24 #include <algorithm> 25 #include <cstdio> 26 #include <exception> 27 #include <fstream> 28 #include <nlohmann/json.hpp> 29 #include <phosphor-logging/log.hpp> 30 #include <sdbusplus/bus.hpp> 31 #include <string> 32 #include <vector> 33 34 namespace ipmi_flash 35 { 36 37 std::vector<HandlerConfig> buildHandlerFromJson(const nlohmann::json& data) 38 { 39 std::vector<HandlerConfig> handlers; 40 41 for (const auto& item : data) 42 { 43 try 44 { 45 HandlerConfig output; 46 47 /* at() throws an exception when the key is not present. */ 48 item.at("blob").get_to(output.blobId); 49 50 /* handler is required. */ 51 const auto& h = item.at("handler"); 52 const std::string handlerType = h.at("type"); 53 if (handlerType == "file") 54 { 55 const auto& path = h.at("path"); 56 output.handler = std::make_unique<FileHandler>(path); 57 } 58 else 59 { 60 throw std::runtime_error("Invalid handler type: " + 61 handlerType); 62 } 63 64 /* actions are required (presently). */ 65 const auto& a = item.at("actions"); 66 std::unique_ptr<ActionPack> pack = std::make_unique<ActionPack>(); 67 68 /* It hasn't been fully determined if any action being optional is 69 * useful, so for now they will be required. 70 * TODO: Evaluate how the behaviors change if some actions are 71 * missing, does the code just assume it was successful? I would 72 * think not. 73 */ 74 const auto& prep = a.at("preparation"); 75 const std::string prepareType = prep.at("type"); 76 if (prepareType == "systemd") 77 { 78 const auto& unit = prep.at("unit"); 79 pack->preparation = SystemdPreparation::CreatePreparation( 80 sdbusplus::bus::new_default(), unit); 81 } 82 else 83 { 84 throw std::runtime_error("Invalid preparation type: " + 85 prepareType); 86 } 87 88 const auto& verify = a.at("verification"); 89 const std::string verifyType = verify.at("type"); 90 if (verifyType == "fileSystemdVerify") 91 { 92 const auto& path = verify.at("path"); 93 const auto& unit = verify.at("unit"); 94 95 /* the mode parameter is optional. */ 96 std::string systemdMode = "replace"; 97 const auto& mode = verify.find("mode"); 98 if (mode != verify.end()) 99 { 100 systemdMode = verify.at("mode").get<std::string>(); 101 } 102 103 pack->verification = SystemdVerification::CreateVerification( 104 sdbusplus::bus::new_default(), path, unit, systemdMode); 105 } 106 else 107 { 108 throw std::runtime_error("Invalid verification type:" + 109 verifyType); 110 } 111 112 const auto& update = a.at("update"); 113 const std::string updateType = update.at("type"); 114 if (updateType == "reboot") 115 { 116 static constexpr auto rebootTarget = "reboot.target"; 117 static constexpr auto rebootMode = "replace-irreversibly"; 118 pack->update = SystemdUpdateMechanism::CreateSystemdUpdate( 119 sdbusplus::bus::new_default(), rebootTarget, rebootMode); 120 } 121 else if (updateType == "systemd") 122 { 123 const auto& unit = update.at("unit"); 124 125 /* the mode parameter is optional. */ 126 std::string systemdMode = "replace"; 127 const auto& mode = update.find("mode"); 128 if (mode != update.end()) 129 { 130 systemdMode = update.at("mode").get<std::string>(); 131 } 132 133 pack->update = SystemdUpdateMechanism::CreateSystemdUpdate( 134 sdbusplus::bus::new_default(), unit, systemdMode); 135 } 136 else 137 { 138 throw std::runtime_error("Invalid update type: " + updateType); 139 } 140 141 output.actions = std::move(pack); 142 handlers.push_back(std::move(output)); 143 } 144 catch (const std::exception& e) 145 { 146 /* TODO: Once phosphor-logging supports unit-test injection, fix 147 * this to log. 148 */ 149 std::fprintf(stderr, 150 "Excepted building HandlerConfig from json: %s\n", 151 e.what()); 152 } 153 } 154 155 return handlers; 156 } 157 158 std::vector<HandlerConfig> BuildHandlerConfigs(const std::string& directory) 159 { 160 using namespace phosphor::logging; 161 162 std::vector<HandlerConfig> output; 163 164 std::vector<std::string> jsonPaths = GetJsonList(directory); 165 166 for (const auto& path : jsonPaths) 167 { 168 std::ifstream jsonFile(path); 169 if (!jsonFile.is_open()) 170 { 171 log<level::ERR>("Unable to open json file", 172 entry("PATH=%s", path.c_str())); 173 continue; 174 } 175 176 auto data = nlohmann::json::parse(jsonFile, nullptr, false); 177 if (data.is_discarded()) 178 { 179 log<level::ERR>("Parsing json failed", 180 entry("PATH=%s", path.c_str())); 181 continue; 182 } 183 184 std::vector<HandlerConfig> configs = buildHandlerFromJson(data); 185 std::move(configs.begin(), configs.end(), std::back_inserter(output)); 186 } 187 188 return output; 189 } 190 191 } // namespace ipmi_flash 192