include/openbmc_dbus_rest.hpp

#include <crow/app.h>

#include <tinyxml2.h>
#include <dbus_singleton.hpp>
#include <boost/algorithm/string.hpp>
#include <boost/container/flat_set.hpp>

namespace crow {
namespace openbmc_mapper {

void introspect_objects(crow::response &res, std::string process_name,
                        std::string path,
                        std::shared_ptr<nlohmann::json> transaction) {
  crow::connections::system_bus->async_method_call(
      [
        &res, transaction, process_name{std::move(process_name)},
        object_path{std::move(path)}
      ](const boost::system::error_code ec, const std::string &introspect_xml) {
        if (ec) {
          CROW_LOG_ERROR << "Introspect call failed with error: "
                         << ec.message() << " on process: " << process_name
                         << " path: " << object_path << "\n";

        } else {
          transaction->push_back({{"path", object_path}});

          tinyxml2::XMLDocument doc;

          doc.Parse(introspect_xml.c_str());
          tinyxml2::XMLNode *pRoot = doc.FirstChildElement("node");
          if (pRoot == nullptr) {
            CROW_LOG_ERROR << "XML document failed to parse " << process_name
                           << " " << object_path << "\n";

          } else {
            tinyxml2::XMLElement *node = pRoot->FirstChildElement("node");
            while (node != nullptr) {
              std::string child_path = node->Attribute("name");
              std::string newpath;
              if (object_path != "/") {
                newpath += object_path;
              }
              newpath += "/" + child_path;
              // introspect the subobjects as well
              introspect_objects(res, process_name, newpath, transaction);

              node = node->NextSiblingElement("node");
            }
          }
        }
        // if we're the last outstanding caller, finish the request
        if (transaction.use_count() == 1) {
          res.json_value = {{"status", "ok"},
                            {"bus_name", process_name},
                            {"objects", *transaction}};
          res.end();
        }
      },
      process_name, path, "org.freedesktop.DBus.Introspectable", "Introspect");
}

// A smattering of common types to unpack.  TODO(ed) this should really iterate
// the sdbusplus object directly and build the json response
using DbusRestVariantType = sdbusplus::message::variant<
    std::vector<std::tuple<std::string, std::string, std::string>>, std::string,
    int64_t, uint64_t, double, int32_t, uint32_t, int16_t, uint16_t, uint8_t,
    bool>;

using ManagedObjectType = std::vector<std::pair<
    sdbusplus::message::object_path,
    boost::container::flat_map<
        std::string,
        boost::container::flat_map<std::string, DbusRestVariantType>>>>;

void get_managed_objects_for_enumerate(
    const std::string &object_name, const std::string &connection_name,
    crow::response &res, std::shared_ptr<nlohmann::json> transaction) {
  crow::connections::system_bus->async_method_call(
      [&res, transaction](const boost::system::error_code ec,
                          const ManagedObjectType &objects) {
        if (ec) {
          CROW_LOG_ERROR << ec;
        } else {
          nlohmann::json &data_json = *transaction;

          for (auto &object_path : objects) {
            CROW_LOG_DEBUG << "Reading object "
                           << static_cast<const std::string&>(object_path.first);
            nlohmann::json &object_json =
                data_json[static_cast<const std::string&>(object_path.first)];
            if (object_json.is_null()) {
              object_json = nlohmann::json::object();
            }
            for (const auto &interface : object_path.second) {
              for (const auto &property : interface.second) {
                nlohmann::json &property_json = object_json[property.first];
                mapbox::util::apply_visitor(
                    [&property_json](auto &&val) { property_json = val; },
                    property.second);

                // dbus-rest represents booleans as 1 or 0, implement to match
                // TODO(ed) see if dbus-rest should be changed
                const bool *property_bool =
                    property_json.get_ptr<const bool *>();
                if (property_bool != nullptr) {
                  property_json = *property_bool ? 1 : 0;
                }
              }
            }
          }
        }

        if (transaction.use_count() == 1) {
          res.json_value = {{"message", "200 OK"},
                            {"status", "ok"},
                            {"data", std::move(*transaction)}};
          res.end();
        }
      },
      connection_name, object_name, "org.freedesktop.DBus.ObjectManager",
      "GetManagedObjects");
}

using GetSubTreeType = std::vector<
    std::pair<std::string,
              std::vector<std::pair<std::string, std::vector<std::string>>>>>;

void handle_enumerate(crow::response &res, const std::string &object_path) {
  crow::connections::system_bus->async_method_call(
      [&res, object_path{std::string(object_path)} ](
          const boost::system::error_code ec,
          const GetSubTreeType &object_names) {
        if (ec) {
          res.result(boost::beast::http::status::internal_server_error);
          res.end();
          return;
        }

        boost::container::flat_set<std::string> connections;

        for (const auto &object : object_names) {
          for (const auto &connection : object.second) {
            connections.insert(connection.first);
          }
        }

        if (connections.size() <= 0) {
          res.result(boost::beast::http::status::not_found);
          res.end();
          return;
        }
        auto transaction =
            std::make_shared<nlohmann::json>(nlohmann::json::object());
        for (const std::string &connection : connections) {
          get_managed_objects_for_enumerate(object_path, connection, res,
                                            transaction);
        }

      },
      "xyz.openbmc_project.ObjectMapper", "/xyz/openbmc_project/object_mapper",
      "xyz.openbmc_project.ObjectMapper", "GetSubTree", object_path, (int32_t)0,
      std::array<std::string, 0>());
}

template <typename... Middlewares>
void request_routes(Crow<Middlewares...> &app) {
  CROW_ROUTE(app, "/bus/")
      .methods("GET"_method)([](const crow::request &req, crow::response &res) {
        res.json_value = {{"busses", {{{"name", "system"}}}}, {"status", "ok"}};
      });

  CROW_ROUTE(app, "/bus/system/")
      .methods("GET"_method)([](const crow::request &req, crow::response &res) {

        auto myCallback = [&res](const boost::system::error_code ec,
                                 std::vector<std::string> &names) {
          if (ec) {
            res.result(boost::beast::http::status::internal_server_error);
          } else {
            std::sort(names.begin(), names.end());
            nlohmann::json j{{"status", "ok"}};
            auto &objects_sub = j["objects"];
            for (auto &name : names) {
              objects_sub.push_back({{"name", name}});
            }
            res.json_value = std::move(j);
          }
          res.end();
        };
        crow::connections::system_bus->async_method_call(
            std::move(myCallback), "org.freedesktop.DBus", "/",
            "org.freedesktop.DBus", "ListNames");
      });

  CROW_ROUTE(app, "/list/")
      .methods("GET"_method)([](const crow::request &req, crow::response &res) {
        crow::connections::system_bus->async_method_call(
            [&res](const boost::system::error_code ec,
                   const std::vector<std::string> &object_paths) {
              if (ec) {
                res.result(boost::beast::http::status::internal_server_error);
              } else {
                res.json_value = {{"status", "ok"},
                                  {"message", "200 OK"},
                                  {"data", std::move(object_paths)}};
              }
              res.end();
            },
            "xyz.openbmc_project.ObjectMapper",
            "/xyz/openbmc_project/object_mapper",
            "xyz.openbmc_project.ObjectMapper", "GetSubTreePaths", "",
            static_cast<int32_t>(99), std::array<std::string, 0>());
      });

  CROW_ROUTE(app, "/xyz/<path>")
      .methods("GET"_method,
               "PUT"_method)([](const crow::request &req, crow::response &res,
                                const std::string &path) {
        std::shared_ptr<nlohmann::json> transaction =
            std::make_shared<nlohmann::json>(nlohmann::json::object());
        using GetObjectType =
            std::vector<std::pair<std::string, std::vector<std::string>>>;
        std::string object_path;
        std::string dest_property;
        std::string property_set_value;
        size_t attr_position = path.find("/attr/");
        if (attr_position == path.npos) {
          object_path = "/xyz/" + path;
        } else {
          object_path = "/xyz/" + path.substr(0, attr_position);
          dest_property =
              path.substr((attr_position + strlen("/attr/")), path.length());
          auto request_dbus_data =
              nlohmann::json::parse(req.body, nullptr, false);
          if (request_dbus_data.is_discarded()) {
            res.result(boost::beast::http::status::unauthorized);
            res.end();
            return;
          }

          auto property_value_it = request_dbus_data.find("data");
          if (property_value_it == request_dbus_data.end()) {
            res.result(boost::beast::http::status::unauthorized);
            res.end();
            return;
          }

          property_set_value = property_value_it->get<const std::string>();
          if (property_set_value.empty()) {
            res.result(boost::beast::http::status::unauthorized);
            res.end();
            return;
          }
        }

        if (boost::ends_with(object_path, "/enumerate")) {
          object_path.erase(object_path.end() - 10, object_path.end());
          handle_enumerate(res, object_path);
          return;
        }

        crow::connections::system_bus->async_method_call(
            [
              &res, &req, object_path{std::move(object_path)},
              dest_property{std::move(dest_property)},
              property_set_value{std::move(property_set_value)}, transaction
            ](const boost::system::error_code ec,
              const GetObjectType &object_names) {
              if (ec) {
                res.result(boost::beast::http::status::internal_server_error);
                res.end();
                return;
              }
              if (object_names.size() != 1) {
                res.result(boost::beast::http::status::not_found);
                res.end();
                return;
              }
              if (req.method() == "GET"_method) {
                for (auto &interface : object_names[0].second) {
                  crow::connections::system_bus->async_method_call(
                      [&res, transaction](
                          const boost::system::error_code ec,
                          const std::vector<std::pair<
                              std::string, DbusRestVariantType>> &properties) {
                        if (ec) {
                          CROW_LOG_ERROR << "Bad dbus request error: " << ec;
                        } else {
                          for (auto &property : properties) {
                            nlohmann::json &it = (*transaction)[property.first];
                            mapbox::util::apply_visitor(
                                [&it](auto &&val) { it = val; },
                                property.second);
                          }
                        }
                        if (transaction.use_count() == 1) {
                          res.json_value = {{"status", "ok"},
                                            {"message", "200 OK"},
                                            {"data", *transaction}};

                          res.end();
                        }
                      },
                      object_names[0].first, object_path,
                      "org.freedesktop.DBus.Properties", "GetAll", interface);
                }
              } else if (req.method() == "PUT"_method) {
                for (auto &interface : object_names[0].second) {
                  crow::connections::system_bus->async_method_call(
                      [
                            &, interface{std::move(interface)},
                            object_names{std::move(object_names)},
                            object_path{std::move(object_path)},
                            dest_property{std::move(dest_property)},
                            property_set_value{std::move(property_set_value)},
                            transaction
                      ](const boost::system::error_code ec,
                        const boost::container::flat_map<
                            std::string, DbusRestVariantType> &properties) {
                        if (ec) {
                          CROW_LOG_ERROR << "Bad dbus request error: " << ec;
                        } else {
                          auto it = properties.find(dest_property);
                          if (it != properties.end()) {
                            // find the matched property in the interface
                            DbusRestVariantType property_value(
                                property_set_value);
                            // create the dbus variant for dbus call
                            crow::connections::system_bus->async_method_call(
                                [transaction](
                                    const boost::system::error_code ec) {
                                  // use the method "Set" to set the property
                                  // value
                                  if (ec) {
                                    CROW_LOG_ERROR << "Bad dbus request error: "
                                                   << ec;
                                  }
                                  // find the matched property and send the
                                  // response
                                  *transaction = {{"status", "ok"},
                                                  {"message", "200 OK"},
                                                  {"data", nullptr}};

                                },
                                object_names[0].first, object_path,
                                "org.freedesktop.DBus.Properties", "Set",
                                interface, dest_property, property_value);
                          }
                        }
                        // if we are the last caller, finish the transaction
                        if (transaction.use_count() == 1) {
                          // if nobody filled in the property, all calls either
                          // errored, or failed
                          if (transaction == nullptr) {
                            res.result(boost::beast::http::status::forbidden);
                            res.json_value = {{"status", "error"},
                                              {"message", "403 Forbidden"},
                                              {"data",
                                               {{"message",
                                                 "The specified property "
                                                 "cannot be created: " +
                                                     dest_property}}}};

                          } else {
                            res.json_value = *transaction;
                          }

                          res.end();
                          return;
                        }
                      },
                      object_names[0].first, object_path,
                      "org.freedesktop.DBus.Properties", "GetAll", interface);
                }
              }
            },
            "xyz.openbmc_project.ObjectMapper",
            "/xyz/openbmc_project/object_mapper",
            "xyz.openbmc_project.ObjectMapper", "GetObject", object_path,
            std::array<std::string, 0>());
      });

  CROW_ROUTE(app, "/bus/system/<str>/")
      .methods("GET"_method)([](const crow::request &req, crow::response &res,
                                const std::string &connection) {
        std::shared_ptr<nlohmann::json> transaction;
        introspect_objects(res, connection, "/", transaction);
      });

  CROW_ROUTE(app, "/bus/system/<str>/<path>")
      .methods("GET"_method)([](const crow::request &req, crow::response &res,
                                const std::string &process_name,
                                const std::string &requested_path) {

        std::vector<std::string> strs;
        boost::split(strs, requested_path, boost::is_any_of("/"));
        std::string object_path;
        std::string interface_name;
        std::string method_name;
        auto it = strs.begin();
        if (it == strs.end()) {
          object_path = "/";
        }
        while (it != strs.end()) {
          // Check if segment contains ".".  If it does, it must be an
          // interface
          if ((*it).find(".") != std::string::npos) {
            break;
            // THis check is neccesary as the trailing slash gets parsed as
            // part of our <path> specifier above, which causes the normal
            // trailing backslash redirector to fail.
          } else if (!it->empty()) {
            object_path += "/" + *it;
          }
          it++;
        }
        if (it != strs.end()) {
          interface_name = *it;
          it++;

          // after interface, we might have a method name
          if (it != strs.end()) {
            method_name = *it;
            it++;
          }
        }
        if (it != strs.end()) {
          // if there is more levels past the method name, something went
          // wrong, throw an error
          res.result(boost::beast::http::status::not_found);
          res.end();
          return;
        }
        if (interface_name.empty()) {
          crow::connections::system_bus->async_method_call(
              [
                    &, process_name{std::move(process_name)},
                    object_path{std::move(object_path)}
              ](const boost::system::error_code ec,
                const std::string &introspect_xml) {
                if (ec) {
                  CROW_LOG_ERROR
                      << "Introspect call failed with error: " << ec.message()
                      << " on process: " << process_name
                      << " path: " << object_path << "\n";

                } else {
                  tinyxml2::XMLDocument doc;

                  doc.Parse(introspect_xml.c_str());
                  tinyxml2::XMLNode *pRoot = doc.FirstChildElement("node");
                  if (pRoot == nullptr) {
                    CROW_LOG_ERROR << "XML document failed to parse "
                                   << process_name << " " << object_path
                                   << "\n";
                    res.json_value = {{"status", "XML parse error"}};
                    res.result(
                        boost::beast::http::status::internal_server_error);
                  } else {
                    nlohmann::json interfaces_array = nlohmann::json::array();
                    tinyxml2::XMLElement *interface =
                        pRoot->FirstChildElement("interface");

                    while (interface != nullptr) {
                      std::string iface_name = interface->Attribute("name");
                      interfaces_array.push_back({{"name", iface_name}});

                      interface = interface->NextSiblingElement("interface");
                    }
                    res.json_value = {{"status", "ok"},
                                      {"bus_name", process_name},
                                      {"interfaces", interfaces_array},
                                      {"object_path", object_path}};
                  }
                }
                res.end();
              },
              process_name, object_path, "org.freedesktop.DBus.Introspectable",
              "Introspect");
        } else {
          crow::connections::system_bus->async_method_call(
              [
                    &, process_name{std::move(process_name)},
                    interface_name{std::move(interface_name)},
                    object_path{std::move(object_path)}
              ](const boost::system::error_code ec,
                const std::string &introspect_xml) {
                if (ec) {
                  CROW_LOG_ERROR
                      << "Introspect call failed with error: " << ec.message()
                      << " on process: " << process_name
                      << " path: " << object_path << "\n";

                } else {
                  tinyxml2::XMLDocument doc;

                  doc.Parse(introspect_xml.c_str());
                  tinyxml2::XMLNode *pRoot = doc.FirstChildElement("node");
                  if (pRoot == nullptr) {
                    CROW_LOG_ERROR << "XML document failed to parse "
                                   << process_name << " " << object_path
                                   << "\n";
                    res.result(
                        boost::beast::http::status::internal_server_error);

                  } else {
                    tinyxml2::XMLElement *node =
                        pRoot->FirstChildElement("node");

                    // if we know we're the only call, build the json directly
                    nlohmann::json methods_array = nlohmann::json::array();
                    nlohmann::json signals_array = nlohmann::json::array();
                    tinyxml2::XMLElement *interface =
                        pRoot->FirstChildElement("interface");

                    while (interface != nullptr) {
                      std::string iface_name = interface->Attribute("name");

                      if (iface_name == interface_name) {
                        tinyxml2::XMLElement *methods =
                            interface->FirstChildElement("method");
                        while (methods != nullptr) {
                          nlohmann::json args_array = nlohmann::json::array();
                          tinyxml2::XMLElement *arg =
                              methods->FirstChildElement("arg");
                          while (arg != nullptr) {
                            args_array.push_back(
                                {{"name", arg->Attribute("name")},
                                 {"type", arg->Attribute("type")},
                                 {"direction", arg->Attribute("direction")}});
                            arg = arg->NextSiblingElement("arg");
                          }
                          methods_array.push_back(
                              {{"name", methods->Attribute("name")},
                               {"uri", "/bus/system/" + process_name +
                                           object_path + "/" + interface_name +
                                           "/" + methods->Attribute("name")},
                               {"args", args_array}});
                          methods = methods->NextSiblingElement("method");
                        }
                        tinyxml2::XMLElement *signals =
                            interface->FirstChildElement("signal");
                        while (signals != nullptr) {
                          nlohmann::json args_array = nlohmann::json::array();

                          tinyxml2::XMLElement *arg =
                              signals->FirstChildElement("arg");
                          while (arg != nullptr) {
                            std::string name = arg->Attribute("name");
                            std::string type = arg->Attribute("type");
                            args_array.push_back({
                                {"name", name},
                                {"type", type},
                            });
                            arg = arg->NextSiblingElement("arg");
                          }
                          signals_array.push_back(
                              {{"name", signals->Attribute("name")},
                               {"args", args_array}});
                          signals = signals->NextSiblingElement("signal");
                        }

                        nlohmann::json j{
                            {"status", "ok"},
                            {"bus_name", process_name},
                            {"interface", interface_name},
                            {"methods", methods_array},
                            {"object_path", object_path},
                            {"properties", nlohmann::json::object()},
                            {"signals", signals_array}};

                        res.write(j.dump());
                        break;
                      }

                      interface = interface->NextSiblingElement("interface");
                    }
                    if (interface == nullptr) {
                      // if we got to the end of the list and never found a
                      // match, throw 404
                      res.result(boost::beast::http::status::not_found);
                    }
                  }
                }
                res.end();
              },
              process_name, object_path, "org.freedesktop.DBus.Introspectable",
              "Introspect");
        }

      });
}  // namespace openbmc_mapper
}  // namespace openbmc_mapper
}  // namespace crow