1 /**
2 * Copyright © 2019 IBM Corporation
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 "model.hpp"
17 #include "updater.hpp"
18 #include "utility.hpp"
19 #include "validator.hpp"
20 #include "version.hpp"
21
22 #include <CLI/CLI.hpp>
23 #include <phosphor-logging/lg2.hpp>
24 #include <sdbusplus/bus.hpp>
25
26 #include <cassert>
27 #include <filesystem>
28
29 using namespace phosphor::logging;
30
main(int argc,char ** argv)31 int main(int argc, char** argv)
32 {
33 std::string psuPathVersion, psuPathModel;
34 std::vector<std::string> versions;
35 bool validateUpdate = false;
36 bool rawOutput = false;
37 std::vector<std::string> updateArguments;
38
39 CLI::App app{"PSU utils app for OpenBMC"};
40 auto action = app.add_option_group("Action");
41 action->add_option("-g,--get-version", psuPathVersion,
42 "Get PSU version from inventory path");
43 action->add_option("-m,--get-model", psuPathModel,
44 "Get PSU model from inventory path");
45 action->add_option("-c,--compare", versions,
46 "Compare and get the latest version");
47 auto updateOpt =
48 action
49 ->add_option("-u,--update", updateArguments,
50 "Update PSU firmware, expecting two arguments: "
51 "<PSU inventory path> <image-dir>")
52 ->expected(2)
53 ->type_name("PSU_PATH IMAGE_DIR");
54 app.add_flag(
55 "--validate", validateUpdate,
56 "Validate number of present PSU vs number of required PSUs and all PSUs have same model before updating firmware")
57 ->needs(updateOpt);
58 action->require_option(1); // Only one option is supported
59 app.add_flag("--raw", rawOutput, "Output raw text without linefeed");
60 CLI11_PARSE(app, argc, argv);
61
62 std::string ret;
63
64 auto bus = sdbusplus::bus::new_default();
65 if (!psuPathVersion.empty())
66 {
67 ret = version::getVersion(bus, psuPathVersion);
68 }
69 if (!psuPathModel.empty())
70 {
71 ret = model::getModel(bus, psuPathModel);
72 }
73 if (!versions.empty())
74 {
75 ret = version::getLatest(versions);
76 }
77 if (!updateArguments.empty())
78 {
79 assert(updateArguments.size() == 2);
80 bool updateStatus = false;
81 if (validateUpdate)
82 {
83 updateStatus = updater::validateAndUpdate(bus, updateArguments[0],
84 updateArguments[1]);
85 }
86 else
87 {
88 updateStatus =
89 updater::update(bus, updateArguments[0], updateArguments[1]);
90 }
91 if (updateStatus)
92 {
93 ret = "Update successful";
94 lg2::info("Successful update to PSU: {PSU}", "PSU",
95 updateArguments[0]);
96 }
97 else
98 {
99 lg2::error("Failed to update PSU: {PSU}", "PSU",
100 updateArguments[0]);
101 }
102 }
103
104 printf("%s", ret.c_str());
105 if (!rawOutput)
106 {
107 printf("\n");
108 }
109 return ret.empty() ? 1 : 0;
110 }
111