Implement dynamic procedure at REST interface
[anna.git] / example / diameter / launcher / MyHandler.cpp
index e69de29..9086582 100644 (file)
@@ -0,0 +1,552 @@
+// ANNA - Anna is Not Nothingness Anymore                                                         //
+//                                                                                                //
+// (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo                         //
+//                                                                                                //
+// See project site at http://redmine.teslayout.com/projects/anna-suite                           //
+// See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
+
+
+// Standard
+#include <string>
+#include <sstream>
+#include <iomanip>
+#include <stdlib.h>
+
+// Project
+#include <anna/http/Request.hpp>
+#include <anna/core/functions.hpp>
+
+// Process
+#include <MyHandler.hpp>
+#include <Launcher.hpp>
+#include <EventOperation.hpp>
+
+void MyHandler::splitURI(const std::string &uri, std::string & operation, std::string & param1, std::string & param2) const {
+
+  std::string::size_type slash_pos = uri.find("/", 1u);
+  if (slash_pos == std::string::npos) {
+    operation = uri.substr(0);
+  }
+  else {
+    operation = uri.substr(0, slash_pos);
+    param1 = uri.substr(slash_pos + 1);
+
+    std::string::size_type slash2_pos = uri.find("/", slash_pos + 1);
+    if (slash2_pos != std::string::npos) {
+      param1 = uri.substr(slash_pos + 1, slash2_pos - slash_pos - 1);
+      param2 = uri.substr(slash2_pos + 1);
+    }
+  }
+}
+
+void MyHandler::sendResponse(anna::comm::ClientSocket& clientSocket, anna::http::Response *response)
+{
+  try {
+    clientSocket.send(*response);
+  } catch(Exception& ex) {
+    ex.trace();
+  }
+}
+
+void MyHandler::evRequest(anna::comm::ClientSocket& clientSocket, const anna::http::Request& request)
+throw(anna::RuntimeException) {
+
+  const anna::DataBlock& body = request.getBody();
+  anna::http::Method::Type::_v method = request.getMethod();
+  std::string uri = request.getURI();
+  bool isGET = (method == anna::http::Method::Type::Get);
+  bool isPOST = (method == anna::http::Method::Type::Post);
+  std::string body_content;
+  nlohmann::json json_body;
+  std::stringstream ss;
+
+  anna::http::Response* response = allocateResponse();
+  response->setStatusCode(200);
+
+  if (isGET) {
+    if(body.getSize() != 0)
+       response->setStatusCode(400); // bad request
+  }
+  else if (isPOST) {
+    if(body.getSize() != 0) {
+      body_content.assign(body.getData(), body.getSize()); // assign body content:
+
+      try {
+        json_body = nlohmann::json::parse(body_content);
+        LOGINFORMATION(
+          std::string msg("Json body received:\n\n");
+          msg += json_body.dump(4); // pretty print json body
+          anna::Logger::information(msg, ANNA_FILE_LOCATION);
+        );
+      }
+      catch (nlohmann::json::parse_error& e)
+      {
+        ss << "Json body parse error: " << e.what() << '\n'
+           << "exception id: " << e.id << '\n'
+           << "byte position of error: " << e.byte << std::endl;
+        anna::Logger::error(ss.str(), ANNA_FILE_LOCATION);
+      }
+    }
+  }
+  else {
+       response->setStatusCode(405); // method not allowed
+  }
+
+  //////////////////////
+  // Prepare response //
+  //////////////////////
+
+  // Headers:
+  http::Header* contentType =response->find(http::Header::Type::ContentType);
+  if(contentType == NULL)
+    contentType = response->createHeader(http::Header::Type::ContentType);
+  contentType->setValue("application/json");
+
+  // Body:
+  std::string response_content = "Internal error (check ADML traces): ";
+  bool success = false;
+  if (isGET) {
+    success = doGET(uri, response_content);
+  }
+  else if (isPOST) {
+    try {
+      success = doPOST(uri, json_body, response_content);
+    }
+    catch(anna::RuntimeException &ex) {
+      ex.trace();
+      LOGINFORMATION(anna::Logger::information("doPOST exception: TODO control in inner method. Check traces", ANNA_FILE_LOCATION));
+    }
+  }
+
+  ss << R"({ "success":")" << (success ? "true":"false")
+     << R"(", "response": )" << std::quoted(response_content);
+  ss << R"( })";
+
+  anna::DataBlock db_content(true);
+  db_content = ss.str();
+  response->setBody(db_content);
+
+  sendResponse(clientSocket, response);
+}
+
+bool MyHandler::doGET(const std::string &uri, std::string &response) {
+
+  bool result = false;
+
+  EventOperation eop(true /* is HTTP */);
+
+  // Snapshots
+  if (uri == "/show-oam") {
+    result = eop.show_oam(response);
+  }
+  else if (uri == "/show-stats") {
+    result = eop.show_stats(response);
+  }
+
+  return result;
+}
+
+bool MyHandler::doPOST(const std::string &uri, const nlohmann::json &j, std::string &response) {
+
+  bool result = false;
+
+  EventOperation eop(true /* is HTTP */);
+
+  std::string opType{}, param1{}, param2{};
+  splitURI(uri, opType, param1, param2);
+
+  // Node management
+  if (opType == "/node") {
+    auto it = j.find("name");
+    if (it != j.end() && it->is_string())
+      result = eop.node(response, *it);
+    else
+      response += "missing 'name' string field";
+  }
+  else if (opType == "/node-auto") {
+    result = eop.node_auto(response);
+  }
+
+  // Parsing operations
+  else if (opType == "/code") {
+    auto it = j.find("diameterJson");
+    if (it != j.end() && it->is_object())
+      result = eop.code(response, it->dump(4)); // get the object as string (always indentation = 4)
+    else
+      response += "missing 'diameterJson' object field";
+  }
+  else if (opType == "/decode") {
+    auto it = j.find("diameterHex");
+    if (it != j.end() && it->is_string())
+      result = eop.decode(response, *it);
+    else
+      response += "missing 'diameterHex' string field";
+  }
+  else if (opType == "/loadmsg") {
+    auto it = j.find("diameterJson");
+    if (it != j.end() && it->is_object())
+      result = eop.loadmsg(response, it->dump(4)); // get the object as string (always indentation = 4)
+    else
+      response += "missing 'diameterJson' object field";
+  }
+
+  // Hot changes
+  else if (opType == "/services") {
+    auto it = j.find("servicesJson");
+    if (it != j.end() && it->is_object()) {
+      result = eop.services(response, it->dump(4)); // get the object as string (always indentation = 4)
+    }
+    else
+      response += "missing 'servicesJson' object field";
+  }
+  else if (opType == "/diameterServerSessions") {
+    auto it = j.find("sessions");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.diameterServerSessions(response, it->get<int>());
+    else
+      response += "missing 'session' integer field";
+  }
+  else if (opType == "/change-dir") {
+    auto it = j.find("directory");
+    std::string directory = (it != j.end() && it->is_string()) ? *it : ""; // default is: restore initial directory
+    result = eop.change_dir(response, directory);
+  }
+
+  // Client sessions visibility
+  else if (opType == "/visibility") {
+    auto it = j.find("action");
+    if (it != j.end() && it->is_string()) {
+      std::string action = *it;
+
+      it = j.find("addressPort");
+      std::string addressPort = (it != j.end() && it->is_string()) ? *it : "";
+
+      it = j.find("socket");
+      int socket = (it != j.end() && it->is_number_integer()) ? it->get<int>() : -1;
+
+      result = eop.visibility(response, action, addressPort, socket);
+    }
+    else
+      response += "missing 'action' string field";
+  }
+
+  // Snapshots
+  else if (opType == "/collect") {
+    result = eop.collect(response);
+  }
+  else if (opType == "/context") {
+    auto it = j.find("targetFile");
+    std::string targetFile = (it != j.end() && it->is_string()) ? *it : "";
+    result = eop.context(response, targetFile);
+  }
+  else if (opType == "/forceCountersRecord") {
+    result = eop.forceCountersRecord(response);
+  }
+  else if (opType == "/log-statistics-samples") {
+    auto it = j.find("list");
+    std::string list = (it != j.end() && it->is_string()) ? *it : "list";
+    result = eop.log_statistics_samples(response, list);
+  }
+
+  // Flow operations
+  else if ((opType == "/sendmsg2e")||(opType == "/sendmsg2c")) {
+    auto itJ = j.find("diameterJson");
+    if (itJ != j.end() && itJ->is_object()) {
+      if (opType == "/sendmsg2e")
+        result = eop.sendmsg_hex_2e(response, itJ->dump(4), true); // get the object as string (always indentation = 4)
+      else
+        result = eop.sendmsg_hex_2c(response, itJ->dump(4), true); // get the object as string (always indentation = 4)
+    }
+    else
+      response += "missing 'diameterJson' object field";
+  }
+  else if ((opType == "/sendhex2e")||(opType == "/sendhex2c")) {
+    auto itH = j.find("diameterHex");
+    if (itH != j.end() && itH->is_string())
+      if (opType == "/sendhex2e")
+        result = eop.sendmsg_hex_2e(response, *itH, false);
+      else
+        result = eop.sendmsg_hex_2c(response, *itH, false);
+    else
+      response += "missing 'diameterHex' string field";
+  }
+  else if ((opType == "/answermsg2e")||(opType == "/answermsg2c")) {
+    auto itJ = j.find("diameterJson");
+    auto itA = j.find("action");
+    bool hasJ = (itJ != j.end() && itJ->is_object());
+    bool hasA = (itA != j.end() && itA->is_string());
+
+    if (hasJ != hasA) { // XOR
+      std::string action;
+      if (hasA) {
+        action = *itA;
+        if (action == "") action = "list"; // default if missing
+      }
+
+      if (opType == "/answermsg2e")
+        result = eop.answermsg_action_2e(response, (hasJ ? itJ->dump(4) : action), hasJ); // itJ: get the object as string (always indentation = 4)
+      else
+        result = eop.answermsg_action_2c(response, (hasJ ? itJ->dump(4) : action), hasJ); // itJ: get the object as string (always indentation = 4)
+    }
+    else
+      response += "missing 'diameterJson' object or 'action' string field (only one accepted)";
+  }
+
+  // FSM testing
+  // test_id__<command>
+  else if (opType == "/testid-description") {
+    auto it = j.find("description");
+    if (it != j.end() && it->is_string())
+      result = eop.test_id__description(response, atoi(param1.c_str()), *it);
+    else
+      response += "missing 'description' string field";
+  }
+  else if (opType == "/testid-ip-limit") {
+    auto it = j.find("amount");
+    int amount = (it != j.end() && it->is_number_integer()) ? it->get<int>() : 1;
+    result = eop.test_id__ip_limit(response, atoi(param1.c_str()), amount);
+  }
+  else if (opType == "/testid-timeout") {
+    auto it = j.find("msecs");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test_id__timeout(response, atoi(param1.c_str()), it->get<int>());
+    else
+      response += "missing 'msecs' integer field";
+  }
+  else if ((opType == "/testid-sendmsg2e")||(opType == "/testid-sendmsg2c")) {
+    auto it = j.find("diameterJson");
+    if (it != j.end() && it->is_object()) {
+
+      auto itS = j.find("stepNumber");
+      int stepNumber = (itS != j.end() && itS->is_number_integer()) ? itS->get<int>() : -1;
+
+      result = eop.test_id__sendmsg2e_2c(response, atoi(param1.c_str()),
+                                        (opType == "/testid-sendmsg2e"), it->dump(4), stepNumber); // get the object as string (always indentation = 4)
+    }
+    else
+      response += "missing 'diameterJson' object field";
+  }
+  else if (opType == "/testid-delay") {
+    auto it = j.find("msecs");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test_id__delay(response, atoi(param1.c_str()), it->get<int>());
+    else
+      response += "missing 'msecs' integer field";
+  }
+  else if (opType == "/testid-sh-command") {
+    auto it = j.find("script");
+    if (it != j.end() && it->is_string())
+      result = eop.test_id__sh_command(response, atoi(param1.c_str()), *it);
+    else
+      response += "missing 'script' string field";
+  }
+  else if ((opType == "/testid-waitfe-hex")||(opType == "/testid-waitfc-hex")) {
+    auto it = j.find("hex");
+    if (it != j.end() && it->is_string()) {
+
+      auto itS = j.find("strict");
+      bool strict = (itS != j.end() && itS->is_string()) ? (*itS == "true") : false;
+
+      result = eop.test_id__waitfefc_hex(response, atoi(param1.c_str()), (opType == "/testid-waitfe-hex"), *it, strict);
+    }
+    else
+      response += "missing 'hex' string field";
+  }
+  else if ((opType == "/testid-waitfe-msg")||(opType == "/testid-waitfc-msg")) {
+    auto it = j.find("diameterJson");
+    if (it != j.end() && it->is_object()) {
+
+      auto itS = j.find("strict");
+      bool strict = (itS != j.end() && itS->is_string()) ? (*itS == "true") : false;
+
+      result = eop.test_id__waitfefc_msg(response, atoi(param1.c_str()), (opType == "/testid-waitfe-msg"), it->dump(4), strict); // get the object as string (always indentation = 4)
+    }
+    else
+      response += "missing 'diameterJson' object field";
+  }
+  else if ((opType == "/testid-waitfe")||(opType == "/testid-waitfc")) {
+    auto it = j.find("condition");
+    if (it != j.end() && it->is_object()) {
+
+      // [code]|[bitR]|[hopByHop]|[applicationId]|[sessionId]|[resultCode]|[msisdn]|[imsi]|[serviceContextId]
+      auto it_code = it->find("code");
+      auto it_bitR = it->find("bitR");
+      auto it_hopByHop = it->find("hopByHop");
+      auto it_applicationId = it->find("applicationId");
+      auto it_sessionId = it->find("sessionId");
+      auto it_resultCode = it->find("resultCode");
+      auto it_msisdn = it->find("msisdn");
+      auto it_imsi = it->find("imsi");
+      auto it_serviceContextId = it->find("serviceContextId");
+
+      std::string p1 = (it_code != it->end() && it_code->is_string()) ? *it_code : "";
+      std::string p2 = (it_bitR != it->end() && it_bitR->is_string()) ? *it_bitR : "";
+      std::string p3 = (it_hopByHop != it->end() && it_hopByHop->is_string()) ? *it_hopByHop : "";
+      std::string p4 = (it_applicationId != it->end() && it_applicationId->is_string()) ? *it_applicationId : "";
+      std::string p5 = (it_sessionId != it->end() && it_sessionId->is_string()) ? *it_sessionId : "";
+      std::string p6 = (it_resultCode != it->end() && it_resultCode->is_string()) ? *it_resultCode : "";
+      std::string p7 = (it_msisdn != it->end() && it_msisdn->is_string()) ? *it_msisdn : "";
+      std::string p8 = (it_imsi != it->end() && it_imsi->is_string()) ? *it_imsi : "";
+      std::string p9 = (it_serviceContextId != it->end() && it_serviceContextId->is_string()) ? *it_serviceContextId : "";
+
+      result = eop.test_id__waitfefc(response, atoi(param1.c_str()), (opType == "/testid-waitfe"), p1, p2, p3, p4, p5, p6, p7, p8, p9);
+    }
+    else
+      response += "missing 'condition' object field";
+  }
+
+  // Testcases execution
+  // test__<command>
+  else if (opType == "/test-ttps") {
+    auto it = j.find("amount");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test__ttps(response, it->get<int>());
+    else
+      response += "missing 'amount' integer field";
+  }
+  else if (opType == "/test-next") {
+    auto it = j.find("syncAmount");
+    int syncAmount = (it != j.end() && it->is_number_integer()) ? it->get<int>() : 1;
+    result = eop.test__next(response, syncAmount);
+  }
+  else if (opType == "/test-ip-limit") {
+    auto it = j.find("amount");
+    int amount = (it != j.end() && it->is_number_integer()) ? it->get<int>() : -2; // default is: show current ip-limit and in-progress test cases amount
+    result = eop.test__ip_limit(response, amount);
+  }
+  else if (opType == "/test-goto") {
+    auto it = j.find("id");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test__goto(response, it->get<int>());
+    else
+      response += "missing 'id' integer field";
+  }
+  else if (opType == "/test-run") {
+    auto it = j.find("id");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test__run(response, it->get<int>());
+    else
+      response += "missing 'id' integer field";
+  }
+  else if (opType == "/test-look") {
+    auto it = j.find("id");
+    int id = (it != j.end() && it->is_number_integer()) ? it->get<int>() : -1; // default is: current
+    result = eop.test__look(response, id);
+  }
+  else if (opType == "/test-state") {
+    auto it = j.find("id");
+    int id = (it != j.end() && it->is_number_integer()) ? it->get<int>() : -1; // default is: current
+    result = eop.test__state(response, id);
+  }
+  else if (opType == "/test-interact") {
+    auto it = j.find("amount");
+    if (it != j.end() && it->is_number_integer()) {
+
+      auto itI = j.find("id");
+      int id = (itI != j.end() && itI->is_number_integer()) ? itI->get<int>() : -1; // default is: current
+
+      result = eop.test__interact(response, it->get<int>(), id);
+    }
+    else
+      response += "missing 'amount' integer field";
+  }
+  else if (opType == "/test-reset") {
+    auto it = j.find("type");
+    std::string type = (it != j.end() && it->is_string()) ? *it : "soft";
+
+    auto itI = j.find("id");
+    int id = (itI != j.end() && itI->is_number_integer()) ? itI->get<int>() : -1; // default is: apply to all the tests
+
+    if ((type == "soft") || (type == "hard")) {
+      result = eop.test__reset(response, (type == "soft"), id);
+    }
+    else
+      response += "invalid 'type' string field (allowed: [soft]|hard)";
+  }
+  else if (opType == "/test-repeats") {
+    auto it = j.find("amount");
+    if (it != j.end() && it->is_number_integer())
+      result = eop.test__repeats(response, it->get<int>());
+    else
+      response += "missing 'amount' integer field";
+  }
+  else if (opType == "/test-auto-reset") {
+    auto it = j.find("type");
+    if (it != j.end() && it->is_string()) {
+
+      if ((*it == "soft") || (*it == "hard")) {
+        result = eop.test__auto_reset(response, (*it == "soft"));
+      }
+      else
+        response += "invalid 'type' string field (allowed: soft|hard)";
+    }
+    else
+      response += "missing 'type' string field";
+  }
+  else if (opType == "/test-initialized") {
+    result = eop.test__initialized(response);
+  }
+  else if (opType == "/test-finished") {
+    result = eop.test__finished(response);
+  }
+  else if (opType == "/test-clear") {
+    result = eop.test__clear(response);
+  }
+  else if (opType == "/test-junit") {
+    auto it = j.find("targetFile");
+    std::string targetFile = (it != j.end() && it->is_string()) ? *it : "/tmp/junit.xml";
+    result = eop.test__junit(response, targetFile);
+  }
+  else if (opType == "/test-summary-counts") {
+    result = eop.test__summary_counts(response);
+  }
+  else if (opType == "/test-summary-states") {
+    result = eop.test__summary_states(response);
+  }
+  else if (opType == "/test-summary") {
+    result = eop.test__summary(response);
+  }
+  else if (opType == "/test-report") {
+    auto it = j.find("state");
+    std::string state = (it != j.end() && it->is_string()) ? *it : "all"; // initialized|in-progress|failed|success|[all]|none
+
+    auto itA = j.find("action");
+    std::string action = (itA != j.end() && itA->is_string()) ? *itA : "enable"; // default is: enable
+
+    if ((action == "enable") || (action == "disable")) {
+      result = eop.test__report(response, state, (action == "enable"));
+    }
+    else
+      response += "invalid 'action' string field (allowed: enable|disable)";
+  }
+  else if ((opType == "/test-report-hex")||(opType == "/test-dump-stdout")) {
+
+    auto itA = j.find("action");
+    std::string action = (itA != j.end() && itA->is_string()) ? *itA : "enable"; // default is: enable
+
+    if ((action == "enable") || (action == "disable")) {
+      bool enable = (action == "enable");
+
+      if (opType == "/test-report-hex")
+        result = eop.test__report_hex(response, enable);
+      else
+        result = eop.test__dump_stdout(response, enable);
+    }
+    else
+      response += "invalid 'action' string field (allowed: enable|disable)";
+  }
+
+  // Dynamic procedure
+  else if ((opType == "/dynamic")) {
+    auto it = j.find("arguments");
+    if (it != j.end() && it->is_object()) {
+      result = eop.test__dynamic(response, *it);
+    }
+    else
+      response += "missing 'arguments' object field";
+  }
+
+
+  return result;
+}
+