Suuports clang compiler
[anna.git] / example / diameter / launcher / main.cpp
index af3ceb0..280624c 100644 (file)
@@ -2,7 +2,7 @@
 //
 // (c) Copyright 2005-2014 Eduardo Ramos Testillano & Francisco Ruiz Rayo
 //
-// https://bitbucket.org/testillano/anna
+// http://redmine.teslayout.com/projects/anna-suite
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions
@@ -14,7 +14,7 @@
 // copyright notice, this list of conditions and the following disclaimer
 // in the documentation and/or other materials provided with the
 // distribution.
-//     * Neither the name of Google Inc. nor the names of its
+//     *  Neither the name of the copyright holder nor the names of its
 // contributors may be used to endorse or promote products derived from
 // this software without specific prior written permission.
 //
@@ -33,6 +33,7 @@
 // Authors: eduardo.ramos.testillano@gmail.com
 //          cisco.tierra@gmail.com
 
+
 #include <fstream>
 #include <iostream>
 #include <time.h>
@@ -336,6 +337,9 @@ public:
    std::string programmedAnswers2e() const throw();
    std::string programmedAnswers2c() const throw();
 
+   // helpers
+   bool getDataBlockFromHexFile(const std::string &pathfile, anna::DataBlock &db) const throw();
+
    // Burst feature
    int clearBurst() throw(); // returns removed
    int loadBurstMessage(const anna::DataBlock & db) throw(anna::RuntimeException);
@@ -351,6 +355,23 @@ public:
    std::string gotoBurst(int order) throw();
 };
 
+bool Launcher::getDataBlockFromHexFile(const std::string &pathfile, anna::DataBlock &db) const throw() {
+
+  // Get hex string
+  static char buffer[2048];
+  std::ifstream infile(pathfile.c_str(), std::ifstream::in);
+  if(infile.is_open()) {
+    infile >> buffer;
+    std::string hexString(buffer, strlen(buffer));
+    anna::functions::fromHexString(hexString, db);
+    // Close file
+    infile.close();
+    return true;
+  }
+
+  return false;
+}
+
 int Launcher::clearBurst() throw() {
    int size = a_burstMessages.size();
 
@@ -431,7 +452,7 @@ int Launcher::pushBurst(int loadAmount) throw() {
    }
 
    a_burstActive = true;
-   register int count;
+   int count;
 
    for (count = 0; count < loadAmount; count++)
       if (!sendBurstMessage()) break;
@@ -455,7 +476,7 @@ int Launcher::sendBurst(int loadAmount) throw() {
       return -2;
    }
 
-   register int count;
+   int count;
 
    for (count = 0; count < loadAmount; count++)
       if (!sendBurstMessage(true /* anyway */)) break;
@@ -496,7 +517,7 @@ bool Launcher::sendBurstMessage(bool anyway) throw() {
          if (a_burstRepeat) {
             a_burstCycle++;
 
-            if (burstLogEnabled()) writeBurstLogFile(anna::functions::asString(("\nCompleted burst cycle. Starting again (repeat mode) on cycle %d.\n", a_burstCycle)));
+            if (burstLogEnabled()) writeBurstLogFile(anna::functions::asString("\nCompleted burst cycle. Starting again (repeat mode) on cycle %d.\n", a_burstCycle));
          } else {
             if (burstLogEnabled()) writeBurstLogFile("\nCompleted burst cycle. Burst finished (repeat mode disabled).\n");
 
@@ -736,6 +757,17 @@ std::string Launcher::help() const throw() {
    result += "\nanswerxml|[source_file]    Same as 'answerxml2c'.";
    result += "\n                           List programmed answers if no parameter provided.";
    result += "\n";
+   result += "\nSend operations are available using hexadecimal content (hex formatted files) which also allow to test";
+   result += "\nspecial scenarios (protocol errors):";
+   result += "\n";
+   result += "\nsendhex2e|<source_file>    Sends hex source file (pathfile) through configured entity.";
+   result += "\nsendhex2c|<source_file>    Sends hex source file (pathfile) to client.";
+   result += "\nsendhex|<source_file>      Same as 'sendhex2e'.";
+   result += "\n";
+   result += "\nAnswer programming in hexadecimal is not really neccessary (you could use send primitives) and also";
+   result += "\nis intended to be used with decoded messages in order to replace things like hop by hop, end to end,";
+   result += "\nsubscriber id, session id, etc.";
+   result += "\n";
    result += "\nIf a request is received, answer map (built with 'answerxml<[2c] or 2e>' operations) will be";
    result += "\n checked to find a corresponding programmed answer to be replied(*). If no ocurrence is found,";
    result += "\n or answer message was received, the message is forwarded to the other side (entity or client),";
@@ -744,7 +776,7 @@ std::string Launcher::help() const throw() {
    result += "\n";
    result += "\n(*) sequence values (hop-by-hop and end-to-end), Session-Id and Subscription-Id avps, are mirrored";
    result += "\n    to the peer which sent the request. If user wants to test a specific answer without changing it,";
-   result += "\n    use sendxml operations better than programming.";
+   result += "\n    use sendxml/sendhex operations better than programming.";
    result += "\n";
    result += "\nBalance ('-balance' command line parameter) could be used to forward server socket receptions through";
    result += "\n entity servers by mean a round-robin algorithm. Both diameter server socket and entity targets should";
@@ -1644,6 +1676,10 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
    LOGDEBUG(anna::Logger::debug(operation, ANNA_FILE_LOCATION));
    response_content = "Operation processed with exception. See traces\n"; // supposed
    std::string result = "";
+   anna::DataBlock db_aux(true);
+
+   ///////////////////////////////////////////////////////////////////
+   // Simple operations without arguments:
 
    // Help:
    if (operation == "help") {
@@ -1661,61 +1697,69 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       response_content = "All process counters & statistic information have been reset\n";
       return;
    }
+   ///////////////////////////////////////////////////////////////////
 
    // Tokenize operation
    Tokenizer params;
    params.apply(operation, "|");
    int numParams = params.size() - 1;
-   //LOGDEBUG(anna::Logger::debug(anna::functions::asString("Number of operation parameters: %d", numParams), ANNA_FILE_LOCATION));
 
+   // No operation has more than 2 arguments ...
    if (numParams > 2) {
       LOGWARNING(anna::Logger::warning(help(), ANNA_FILE_LOCATION));
       throw anna::RuntimeException("Wrong body content format on HTTP Request", ANNA_FILE_LOCATION);
    }
 
+   // Get the operation type:
    Tokenizer::const_iterator tok_iter = params.begin();
    std::string opType = Tokenizer::data(tok_iter);
-   std::string param1, param2;
 
-   if (numParams >= 1) { tok_iter++; param1 = Tokenizer::data(tok_iter); }
+   // Check the number of parameters:
+   bool wrongBody = false;
+   if ( ((opType == "code")||(opType == "decode")) && (numParams != 2) ) wrongBody = true;
+   if ( ((opType == "sendxml") || (opType == "sendxml2e") || (opType == "sendhex") || (opType == "sendhex2e")) && (numParams != 1) ) wrongBody = true;
+   if ( (opType == "burst") && (numParams < 1) ) wrongBody = true;
+   if ( ((opType == "sendxml2c")||(opType == "sendhex2c")||(opType == "loadxml")||(opType == "diameterServerSessions")) && (numParams != 1) ) wrongBody = true;
+   if (wrongBody) {
+     // Launch exception
+     std::string msg = "Wrong body content format on HTTP Request for '";
+     msg += opType;
+     msg += "' operation (missing parameter/s)";
+     throw anna::RuntimeException(msg, ANNA_FILE_LOCATION);
+   }
 
+   // All seems ok:
+   std::string param1, param2;
+   if (numParams >= 1) { tok_iter++; param1 = Tokenizer::data(tok_iter); }
    if (numParams == 2) { tok_iter++; param2 = Tokenizer::data(tok_iter); }
 
-   if (opType == "code") {
-      if (numParams != 2)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'code' operation (missing parameter/s)", ANNA_FILE_LOCATION);
 
+   // Operations:
+   if (opType == "code") {
       G_codecMsg.loadXML(param1);
       std::string hexString = anna::functions::asHexString(G_codecMsg.code());
       // write to outfile
       ofstream outfile(param2.c_str(), ifstream::out);
       outfile.write(hexString.c_str(), hexString.size());
       outfile.close();
-   } else if (opType == "decode") {
-      if (numParams != 2)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'decode' operation (missing parameter/s)", ANNA_FILE_LOCATION);
 
-      char buffer[2048];
-      ifstream infile(param1.c_str(), ifstream::in);
-      infile >> buffer;
-      std::string hexString(buffer, strlen(buffer));
-      anna::DataBlock db(true);
-      anna::functions::fromHexString(hexString, db);
+   } else if (opType == "decode") {
+      // Get DataBlock from file with hex content:
+      if (!getDataBlockFromHexFile(param1, db_aux))
+         throw anna::RuntimeException("Error reading hex file provided", ANNA_FILE_LOCATION);
 
       // Decode
-      try { G_codecMsg.decode(db); } catch (anna::RuntimeException &ex) { ex.trace(); }
+      try { G_codecMsg.decode(db_aux); } catch (anna::RuntimeException &ex) { ex.trace(); }
 
       std::string xmlString = G_codecMsg.asXMLString();
       // write to outfile
       ofstream outfile(param2.c_str(), ifstream::out);
       outfile.write(xmlString.c_str(), xmlString.size());
       outfile.close();
-      infile.close();
+
    } else if ((opType == "hide") || (opType == "show") || (opType == "hidden") || (opType == "shown")) {
       anna::diameter::comm::Entity *entity = getEntity();
-
       if (!entity) throw anna::RuntimeException("No entity configured to send messages", ANNA_FILE_LOCATION);
-
       if (param1 != "") {
          if (param2 != "") {
             std::string key = param1;
@@ -1723,11 +1767,8 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
             key += param2;
 
             if (opType == "hide") getMyDiameterEngine()->findClientSession(key)->hide();
-
             if (opType == "show") getMyDiameterEngine()->findClientSession(key)->show();
-
             if (opType == "hidden") result = getMyDiameterEngine()->findClientSession(key)->hidden() ? "true" : "false";
-
             if (opType == "shown") result = getMyDiameterEngine()->findClientSession(key)->shown() ? "true" : "false";
          } else {
             std::string address;
@@ -1735,35 +1776,34 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
             anna::functions::getAddressAndPortFromSocketLiteral(param1, address, port);
 
             if (opType == "hide") getMyDiameterEngine()->findServer(address, port)->hide();
-
             if (opType == "show") getMyDiameterEngine()->findServer(address, port)->show();
-
             if (opType == "hidden") result = getMyDiameterEngine()->findServer(address, port)->hidden() ? "true" : "false";
-
             if (opType == "shown") result = getMyDiameterEngine()->findServer(address, port)->shown() ? "true" : "false";
          }
       } else {
          if (opType == "hide") entity->hide();
-
          if (opType == "show") entity->show();
-
          if (opType == "hidden") result = entity->hidden() ? "true" : "false";
-
          if (opType == "shown") result = entity->shown() ? "true" : "false";
       }
-   } else if ((opType == "sendxml") || (opType == "sendxml2e")) {
-      if (numParams != 1)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'sendxml/sendxml2e' operation (missing parameter)", ANNA_FILE_LOCATION);
+   } else if ((opType == "sendxml") || (opType == "sendxml2e") || (opType == "sendhex") || (opType == "sendhex2e")) {
 
       anna::diameter::comm::Entity *entity = getEntity();
-
       if (!entity) throw anna::RuntimeException("No entity configured to send the message", ANNA_FILE_LOCATION);
 
-      G_codecMsg.loadXML(param1);
-      G_commMsgSent2e.clearBody();
-      try { G_codecMsg.valid(); } catch (anna::RuntimeException &ex) { ex.trace(); } // at least we need to see validation errors although it will continue sending (see validation mode configured in launcher)
+      if ((opType == "sendxml") || (opType == "sendxml2e")) {
+        G_codecMsg.loadXML(param1);
+        G_commMsgSent2e.clearBody();
+        try { G_codecMsg.valid(); } catch (anna::RuntimeException &ex) { ex.trace(); } // at least we need to see validation errors although it will continue sending (see validation mode configured in launcher)
+        G_commMsgSent2e.setBody(G_codecMsg.code());
+      }
+      else {
+        // Get DataBlock from file with hex content:
+        if (!getDataBlockFromHexFile(param1, db_aux))
+          throw anna::RuntimeException("Error reading hex file provided", ANNA_FILE_LOCATION);
+        G_commMsgSent2e.setBody(db_aux);
+      }
 
-      G_commMsgSent2e.setBody(G_codecMsg.code());
       bool success = entity->send(G_commMsgSent2e, cl.exists("balance"));
 
       // Detailed log:
@@ -1774,11 +1814,7 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
          writeLogFile(G_codecMsg, (success ? "sent2e" : "send2eError"), detail);
       }
    } else if ((opType == "burst")) {
-      if (numParams < 1)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'burst' operation (missing action parameter)", ANNA_FILE_LOCATION);
-
       anna::diameter::comm::Entity *entity = getEntity();
-
       if (!entity) throw anna::RuntimeException("No entity configured to use burst feature", ANNA_FILE_LOCATION);
 
       // burst|clear                     clears all loaded burst messages.
@@ -1880,19 +1916,24 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       } else {
          throw anna::RuntimeException("Wrong body content format on HTTP Request for 'burst' operation (unexpected action parameter). See help", ANNA_FILE_LOCATION);
       }
-   } else if (opType == "sendxml2c") {
-      if (numParams != 1)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'sendxml2c' operation (missing parameter)", ANNA_FILE_LOCATION);
 
+   } else if ((opType == "sendxml2c") || (opType == "sendhex2c")) {
       anna::diameter::comm::LocalServer *localServer = getDiameterLocalServer();
-
       if (!localServer) throw anna::RuntimeException("No local server configured to send the message", ANNA_FILE_LOCATION);
 
-      G_codecMsg.loadXML(param1);
-      G_commMsgSent2c.clearBody();
-      try { G_codecMsg.valid(); } catch (anna::RuntimeException &ex) { ex.trace(); } // at least we need to see validation errors although it will continue sending (see validation mode configured in launcher)
+      if (opType == "sendxml2c") {
+        G_codecMsg.loadXML(param1);
+        G_commMsgSent2c.clearBody();
+        try { G_codecMsg.valid(); } catch (anna::RuntimeException &ex) { ex.trace(); } // at least we need to see validation errors although it will continue sending (see validation mode configured in launcher)
+        G_commMsgSent2c.setBody(G_codecMsg.code());
+      }
+      else {
+        // Get DataBlock from file with hex content:
+        if (!getDataBlockFromHexFile(param1, db_aux))
+          throw anna::RuntimeException("Error reading hex file provided", ANNA_FILE_LOCATION);
+        G_commMsgSent2c.setBody(db_aux);
+      }
 
-      G_commMsgSent2c.setBody(G_codecMsg.code());
       bool success = localServer->send(G_commMsgSent2c);
 
       // Detailed log:
@@ -1901,16 +1942,13 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
          std::string detail = usedServerSession ? usedServerSession->asString() : "<null server session>"; // esto no deberia ocurrir
          writeLogFile(G_codecMsg, (success ? "sent2c" : "send2cError"), detail);
       }
-   } else if (opType == "loadxml") {
-      if (numParams != 1)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'loadxml' operation (missing parameter)", ANNA_FILE_LOCATION);
 
+   } else if (opType == "loadxml") {
       G_codecMsg.loadXML(param1);
       std::string xmlString = G_codecMsg.asXMLString();
       std::cout << xmlString << std::endl;
+
    } else if (opType == "diameterServerSessions") {
-      if (numParams != 1)
-         throw anna::RuntimeException("Wrong body content format on HTTP Request for 'diameterServerSessions' operation (missing parameter)", ANNA_FILE_LOCATION);
 
       int diameterServerSessions = atoi(param1.c_str());
 
@@ -1949,7 +1987,7 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
          response_content = "Programmed answers dumped on stdout\n";
          return;
       }
-   } else if ((opType == "answerxml2e")) {
+   } else if (opType == "answerxml2e") {
       anna::diameter::comm::Entity *entity = getEntity();
 
       if (!entity)
@@ -2013,7 +2051,7 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       response_content += "Result: ";
       response_content += result;
       response_content += "\n";
-   } else if ((opType == "sendxml") || (opType == "sendxml2e")) {
+   } else if ((opType == "sendxml") || (opType == "sendxml2e") || (opType == "sendhex") || (opType == "sendhex2e")) {
       response_content += "Message '";
       response_content += param1;
       response_content += "' sent to entity.";
@@ -2024,7 +2062,7 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       response_content += "' executed. ";
       response_content += result;
       response_content += "\n";
-   } else if (opType == "sendxml2c") {
+   } else if ((opType == "sendxml2c")||(opType == "sendhex2c")) {
       response_content += "Message '";
       response_content += param1;
       response_content += "' sent to client.";
@@ -2039,7 +2077,7 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       response_content += param1;
       response_content += "' programmed.";
       response_content += "\n";
-   } else if ((opType == "answerxml2e")) {
+   } else if (opType == "answerxml2e") {
       response_content += "Answer to entity '";
       response_content += param1;
       response_content += "' programmed.";