source/http/Message.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 #include <anna/core/tracing/Logger.hpp>
  38 #include <anna/core/functions.hpp>
  39
  40 #include <anna/xml/Compiler.hpp>
  41
  42 #include <anna/http/Message.hpp>
  43 #include <anna/http/internal/defines.hpp>
  44 #include <anna/http/Header.hpp>
  45
  46 using namespace std;
  47 using namespace anna;
  48
  49 comm::Message* http::Message::setBody(const xml::Node* node)
  50 throw(RuntimeException) {
  51   comm::Message::setBody(node);
  52   http::Header* contentType = find(http::Header::Type::ContentType);
  53
  54   if(contentType == NULL)
  55     contentType = createHeader(http::Header::Type::ContentType);
  56
  57   contentType->setValue("text/xml");
  58   return this;
  59 }
  60
  61 http::Header* http::Message::find(const http::Header::Type::_v type)
  62 throw() {
  63   http::Header* result = NULL;
  64
  65   for(header_iterator ii = header_begin(), maxii = header_end(); ii != maxii; ii ++) {
  66     if(header(ii)->getType() == type) {
  67       result = header(ii);
  68       break;
  69     }
  70   }
  71
  72   return result;
  73 }
  74
  75 http::Header* http::Message::find(const char* name)
  76 throw() {
  77   Header* result = NULL;
  78   Header* w;
  79   const string* s;
  80
  81   for(header_iterator ii = header_begin(), maxii = header_end(); ii != maxii; ii ++) {
  82     w = header(ii);
  83
  84     if(w->getCategory() != Header::Category::Extension)
  85       continue;
  86
  87     if((s = w->getExtensionName()) == NULL)
  88       continue;
  89
  90     if(*s == name) {
  91       result = w;
  92       break;
  93     }
  94   }
  95
  96   return result;
  97 }
  98
  99 //-----------------------------------------------------------------------------------------------
 100 // Codifica el mensaje HTTP.
 101 //
 102 // Lo unico complicado a tener en cuenta es que el valor del content-length coincida con el
 103 // valor de la longitud real de 'body'.
 104 //-----------------------------------------------------------------------------------------------
 105 const DataBlock& http::Message::code()
 106 throw(RuntimeException) {
 107   Header* contentLength = const_cast <Header*>(find(Header::Type::ContentLength));
 108   const DataBlock& body = getBody();
 109
 110   if(contentLength != NULL) {
 111     if(body.isEmpty() == false) {
 112       if(contentLength->getIntegerValue() != body.getSize())
 113         contentLength->setValue(body.getSize());
 114     } else
 115       a_headers.release(contentLength);
 116   } else {
 117     if(body.isEmpty() == false) {
 118       contentLength = createHeader(Header::Type::ContentLength);
 119       contentLength->setValue(body.getSize());
 120     }
 121   }
 122
 123   a_codeBuffer->clear();
 124   codeLine(codeStartLine());
 125
 126   for(header_iterator ii = header_begin(), maxii = header_end(); ii != maxii; ii ++)
 127     codeLine(Message::header(ii)->code());
 128
 129   a_codeBuffer->append(endOfLine, sizeEndOfLine);
 130   a_codeBuffer->append(body);
 131   LOGDEBUG(
 132     string msg("anna:http::Message::code | Message: ");
 133     msg += anna::functions::asString(*a_codeBuffer);
 134     Logger::debug(msg, ANNA_FILE_LOCATION);
 135   );
 136   return *a_codeBuffer;
 137 }
 138
 139 void http::Message::codeLine(const std::string& line)
 140 throw(RuntimeException) {
 141   a_codeBuffer->append(line.c_str(), line.length());
 142   a_codeBuffer->append(endOfLine, sizeEndOfLine);
 143 }