source/diameter.comm/Transport.cpp

   1 // ANNA - Anna is Not 'N' 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 // Local
  38 #include <anna/diameter.comm/Transport.hpp>
  39
  40 #include <anna/core/tracing/Logger.hpp>
  41 #include <anna/core/tracing/TraceMethod.hpp>
  42 #include <anna/core/DataBlock.hpp>
  43 #include <anna/comm/Message.hpp>
  44
  45
  46
  47
  48 using namespace std;
  49 using namespace anna::diameter::comm;
  50
  51 anna::comm::TransportFactoryImpl <Transport> Transport::st_factory;
  52
  53 Transport::Transport() : anna::comm::Transport() {
  54   setInputMessage(new anna::comm::Message());
  55 }
  56
  57 Transport::~Transport() {
  58   delete getInputMessage();
  59 }
  60
  61 /*
  62  * La longitud se calcula con los primeros 4 bytes (el primer byte se ignora, pues es la version diameter)
  63  */
  64 int Transport::calculeSize(const anna::DataBlock& dataBlock)
  65 throw(anna::RuntimeException) {
  66   const int size = dataBlock.getSize();
  67
  68   if(size < 4)
  69     return -1;
  70
  71   const char* data = dataBlock.getData();
  72   int result = (((unsigned int)data[1] << 16) & 0xFF0000) +
  73                (((unsigned int)data[2] << 8)  & 0x00FF00) +
  74                (((unsigned int)data[3])    & 0x0000FF);
  75   return result;
  76 }
  77
  78 const anna::DataBlock& Transport::code(anna::comm::Message& message)
  79 throw(anna::RuntimeException) {
  80   LOGMETHOD(anna::TraceMethod ttmm(anna::Logger::Local7, "anna::diameter::comm::Transport", "code", ANNA_FILE_LOCATION));
  81   a_forCode = message.code();
  82   return a_forCode;
  83 }
  84
  85 //--------------------------------------------------------------------------------
  86 // Los 4 primeros bytes contienen la longitud del mensaje.
  87 // Cada ClientSocket tiene su anna::diameter::comm::Transport asociado
  88 //--------------------------------------------------------------------------------
  89 const anna::comm::Message* Transport::decode(const anna::DataBlock& input)
  90 throw(anna::RuntimeException) {
  91   anna::comm::Message* inputMessage = static_cast <anna::comm::Message*>(getInputMessage());
  92   inputMessage->setBody(input);
  93   return inputMessage;
  94 }