X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2Fmain.cpp;h=280624cbc03f47d15e9e3f49e06f99c5b4f92f87;hb=5256cb543a9d35b610df19bd227cde977186aef0;hp=af3ceb0134cc7253201a25a1c97468a1e829aa20;hpb=b8c96c2917b9d29976a8771659d70bfb0881c133;p=anna.git diff --git a/example/diameter/launcher/main.cpp b/example/diameter/launcher/main.cpp index af3ceb0..280624c 100644 --- a/example/diameter/launcher/main.cpp +++ b/example/diameter/launcher/main.cpp @@ -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 #include #include @@ -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| Sends hex source file (pathfile) through configured entity."; + result += "\nsendhex2c| Sends hex source file (pathfile) to client."; + result += "\nsendhex| 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() : ""; // 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.";