example/diameter/stackManagement/main.cpp

   1 // ANNA - Anna is Not Nothingness Anymore
   2 //
   3 // (c) Copyright 2005-2014 Eduardo Ramos Testillano & Francisco Ruiz Rayo
   4 //
   5 // http://redmine.teslayout.com/projects/anna-suite
   6 //
   7 // Redistribution and use in source and binary forms, with or without
   8 // modification, are permitted provided that the following conditions
   9 // are met:
  10 //
  11 //     * Redistributions of source code must retain the above copyright
  12 // notice, this list of conditions and the following disclaimer.
  13 //     * Redistributions in binary form must reproduce the above
  14 // copyright notice, this list of conditions and the following disclaimer
  15 // in the documentation and/or other materials provided with the
  16 // distribution.
  17 //     *  Neither the name of the copyright holder nor the names of its
  18 // contributors may be used to endorse or promote products derived from
  19 // this software without specific prior written permission.
  20 //
  21 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32 //
  33 // Authors: eduardo.ramos.testillano@gmail.com
  34 //          cisco.tierra@gmail.com
  35
  36
  37 // Standard
  38 #include <iostream>
  39 #include <fstream>
  40
  41 // STL
  42 #include <string>
  43
  44 #include <anna/core/functions.hpp>
  45 #include <anna/core/tracing/Logger.hpp>
  46 #include <anna/core/tracing/TraceWriter.hpp>
  47 #include <anna/core/RuntimeException.hpp>
  48 #include <anna/xml/xml.hpp>
  49 #include <anna/diameter/stack/Engine.hpp>
  50
  51
  52 using namespace anna;
  53 using namespace anna::diameter;
  54
  55
  56
  57 void _exit(const std::string & msg) {
  58   std::cout << std::endl << msg << std::endl;
  59   exit(-1);
  60 }
  61
  62
  63 int main(int argc, char** argv) {
  64   std::string exec = argv[0];
  65   std::string execBN = exec.substr(exec.find_last_of("/") + 1);
  66   std::string filetrace = execBN + ".trace";
  67   Logger::setLevel(Logger::Debug);
  68   Logger::initialize(execBN.c_str(), new TraceWriter(filetrace.c_str(), 2048000));
  69   stack::Engine & engine = stack::Engine::instantiate();
  70
  71   if(argc < 2) {
  72     std::string msg = anna::functions::asString("Usage: %s <list of '.xml' dictionaries overloaded>\n       e.g. '%s avps.xml commands.xml'", exec.c_str(), exec.c_str());
  73     _exit(msg);
  74   }
  75
  76   stack::Dictionary *dictionary;
  77   int index = 1;
  78
  79   try {
  80     dictionary = engine.createDictionary(0 /* general unique stack id */);
  81     dictionary->allowUpdates();
  82
  83     while(index < argc) {
  84       dictionary->load(argv[index]);
  85       index++;
  86     }
  87
  88     engine.removeStack(0);
  89     // Trace:
  90     LOGINFORMATION(Logger::information(engine.asString(), ANNA_FILE_LOCATION));
  91     LOGDEBUG(Logger::debug(dictionary->asString(), ANNA_FILE_LOCATION));
  92     // Output:
  93     std::ofstream out("./result.xml", std::ifstream::out);
  94     std::string buffer = dictionary->asXMLString();
  95     out.write(buffer.c_str(), buffer.size());
  96     out.close();
  97     std::cout << "Written 'result.xml'" << std::endl;
  98   } catch(anna::RuntimeException &ex) {
  99     ex.trace();
 100     std::cout << ex.getText() << std::endl;
 101   }
 102
 103   _exit("Open 'file.trace' in order to see the stacks loaded");
 104 }
 105