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