Fill in phosphor-unit-failure-monitor main()
Obtain the command line arguments, construct the
monitor class, and have it analyze the failure.
Change-Id: I27015b7d485995ac7605f07622112f4b8d0aa621
Signed-off-by: Matt Spinler <spinler@us.ibm.com>
diff --git a/fail-monitor/Makefile.am b/fail-monitor/Makefile.am
index 6c5c5b2..61fde80 100644
--- a/fail-monitor/Makefile.am
+++ b/fail-monitor/Makefile.am
@@ -5,6 +5,7 @@
phosphor-unit-failure-monitor
phosphor_unit_failure_monitor_SOURCES = \
+ argument.cpp \
main.cpp \
monitor.cpp
diff --git a/fail-monitor/argument.cpp b/fail-monitor/argument.cpp
new file mode 100644
index 0000000..4f9003c
--- /dev/null
+++ b/fail-monitor/argument.cpp
@@ -0,0 +1,92 @@
+/**
+ * Copyright © 2017 IBM Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#include <iostream>
+#include <iterator>
+#include <algorithm>
+#include "argument.hpp"
+
+namespace phosphor
+{
+namespace unit
+{
+namespace failure
+{
+
+ArgumentParser::ArgumentParser(int argc, char** argv)
+{
+ int option = 0;
+ while (-1 != (option = getopt_long(argc, argv, optionStr, options, NULL)))
+ {
+ if ((option == '?') || (option == 'h'))
+ {
+ usage(argv);
+ exit(-1);
+ }
+
+ auto i = &options[0];
+ while ((i->val != option) && (i->val != 0))
+ {
+ ++i;
+ }
+
+ if (i->val)
+ {
+ arguments[i->name] = (i->has_arg ? optarg : trueString);
+ }
+ }
+}
+
+const std::string& ArgumentParser::operator[](const std::string& opt)
+{
+ auto i = arguments.find(opt);
+ if (i == arguments.end())
+ {
+ return emptyString;
+ }
+ else
+ {
+ return i->second;
+ }
+}
+
+void ArgumentParser::usage(char** argv)
+{
+ std::cerr << "Usage: " << argv[0] << " [options]\n";
+ std::cerr << "Options:\n";
+ std::cerr << " --help Print this menu\n";
+ std::cerr << " --source=<source> The source unit to monitor\n";
+ std::cerr << " --target=<target> The target unit to start or stop\n";
+ std::cerr << " --action=<action> Target unit action - start or stop\n";
+ std::cerr << std::flush;
+}
+
+const option ArgumentParser::options[] =
+{
+ { "source", required_argument, NULL, 's' },
+ { "action", required_argument, NULL, 'a' },
+ { "target", required_argument, NULL, 't' },
+ { "help", no_argument, NULL, 'h' },
+ { 0, 0, 0, 0},
+};
+
+const char* ArgumentParser::optionStr = "s:a:t:h?";
+
+const std::string ArgumentParser::trueString = "true";
+const std::string ArgumentParser::emptyString = "";
+
+}
+}
+}
diff --git a/fail-monitor/argument.hpp b/fail-monitor/argument.hpp
new file mode 100644
index 0000000..2d08a2f
--- /dev/null
+++ b/fail-monitor/argument.hpp
@@ -0,0 +1,68 @@
+#pragma once
+
+#include <getopt.h>
+#include <map>
+#include <string>
+
+namespace phosphor
+{
+namespace unit
+{
+namespace failure
+{
+
+/** @brief Class - Encapsulates parsing command line options and
+ * populating arguments
+ */
+class ArgumentParser
+{
+ public:
+ ArgumentParser() = delete;
+ ~ArgumentParser() = default;
+ ArgumentParser(const ArgumentParser&) = delete;
+ ArgumentParser& operator=(const ArgumentParser&) = delete;
+ ArgumentParser(ArgumentParser&&) = default;
+ ArgumentParser& operator=(ArgumentParser&&) = default;
+
+ /** @brief Contructs Argument object
+ *
+ * @param argc - the main function's argc passed as is
+ * @param argv - the main function's argv passed as is
+ * @return Object constructed
+ */
+ ArgumentParser(int argc, char** argv);
+
+ /** @brief Given an option, returns its argument(optarg)
+ *
+ * @param opt - command line option string
+ *
+ * @return argument which is a standard optarg
+ */
+ const std::string& operator[](const std::string& opt);
+
+ /** @brief Displays usage
+ *
+ * @param argv - the main function's argv passed as is
+ */
+ static void usage(char** argv);
+
+ /** @brief Set to 'true' when an option is passed */
+ static const std::string trueString;
+
+ /** @brief Set to '' when an option is not passed */
+ static const std::string emptyString;
+
+ private:
+ /** @brief Option to argument mapping */
+ std::map<const std::string, std::string> arguments;
+
+ /** @brief Array of struct options as needed by getopt_long */
+ static const option options[];
+
+ /** @brief optstring as needed by getopt_long */
+ static const char* optionStr;
+};
+
+}
+}
+}
diff --git a/fail-monitor/main.cpp b/fail-monitor/main.cpp
index b13dc7f..1c47b98 100644
--- a/fail-monitor/main.cpp
+++ b/fail-monitor/main.cpp
@@ -20,11 +20,57 @@
* then it will either stop or start the target unit, depending
* on the command line arguments.
*/
+#include <iostream>
+#include <map>
+#include "argument.hpp"
+#include "monitor.hpp"
+using namespace phosphor::unit::failure;
+
+/**
+ * Prints usage and exits the program
+ *
+ * @param[in] err - the error message to print
+ * @param[in] argv - argv from main()
+ */
+void exitWithError(const char* err, char** argv)
+{
+ std::cerr << "ERROR: " << err << "\n";
+ ArgumentParser::usage(argv);
+ exit(EXIT_FAILURE);
+}
+
+static const std::map<std::string, Monitor::Action> actions =
+{
+ {"start", Monitor::Action::start},
+ {"stop", Monitor::Action::stop}
+};
int main(int argc, char** argv)
{
+ ArgumentParser args(argc, argv);
+ auto source = args["source"];
+ if (source == ArgumentParser::emptyString)
+ {
+ exitWithError("Source not specified", argv);
+ }
+
+ auto target = args["target"];
+ if (target == ArgumentParser::emptyString)
+ {
+ exitWithError("Target not specified", argv);
+ }
+
+ auto a = actions.find(args["action"]);
+ if (a == actions.end())
+ {
+ exitWithError("Missing or invalid action specified", argv);
+ }
+
+ Monitor monitor{source, target, a->second};
+
+ monitor.analyze();
return 0;
}