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 // https://bitbucket.org/testillano/anna
   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 Google Inc. 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    Logger::setLevel(Logger::Debug);
  65    Logger::initialize("stackManagement", new TraceWriter("file.trace", 2048000));
  66    stack::Engine & engine = stack::Engine::instantiate();
  67    std::string exec = argv[0];
  68    std::string param = argv[1] ? argv[1] : "";
  69
  70    if (param == "") {
  71       std::string msg = anna::functions::asString("Use: %s <list of xml dictionaries overloaded>,\n     i.e. '%s avps.xml commands.xml'", exec.c_str(), exec.c_str());
  72       _exit(msg);
  73    }
  74
  75    int index = 1;
  76    const char *xmlFile = argv[index];
  77    stack::Dictionary *dictionary;
  78
  79    try {
  80       dictionary = engine.createDictionary(0 /* general unique stack id */);
  81       dictionary->allowUpdates();
  82
  83       while (xmlFile) {
  84          dictionary->load(xmlFile);
  85          xmlFile = argv[++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