Server/src/httpmessage.cpp

   1 // vim: set noexpandtab tabstop=4 shiftwidth=4:
   2 //----------------------------------------------------------------------------
   3 //                                  PlaQui
   4 //----------------------------------------------------------------------------
   5 // This file is part of PlaQui.
   6 //
   7 // PlaQui is free software; you can redistribute it and/or modify it under the
   8 // terms of the GNU General Public License as published by the Free Software
   9 // Foundation; either version 2 of the License, or (at your option) any later
  10 // version.
  11 //
  12 // PlaQui is distributed in the hope that it will be useful, but WITHOUT ANY
  13 // WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14 // FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  15 // details.
  16 //
  17 // You should have received a copy of the GNU General Public License along
  18 // with PlaQui; if not, write to the Free Software Foundation, Inc., 59 Temple
  19 // Place, Suite 330, Boston, MA  02111-1307  USA
  20 //----------------------------------------------------------------------------
  21 // Creado:  dom oct 26 22:02:09 ART 2003
  22 // Autores: Leandro Lucarella <llucare@fi.uba.ar>
  23 //----------------------------------------------------------------------------
  24 //
  25 // $Id$
  26 //
  27
  28 #include "plaqui/server/string.h"
  29 #include "plaqui/server/httpmessage.h"
  30 #include <sstream>
  31 #include <cstdlib>
  32 #ifdef DEBUG
  33 #       include <iostream>
  34 #endif // DEBUG
  35
  36 using namespace std;
  37
  38 namespace PlaQui {
  39
  40 namespace Server {
  41
  42 HTTPMessage::~HTTPMessage(void) {
  43 #ifdef DEBUG
  44         cerr << __FILE__ << "(" << __LINE__ << ")"
  45                 << ": destructor." << endl;
  46 #endif // DEBUG
  47 }
  48
  49 HTTPMessage::HTTPMessage(const string& _body, const string& _version):
  50                 version(_version) {
  51 #ifdef DEBUG
  52         cerr << __FILE__ << "(" << __LINE__ << ")"
  53                 << ": version = " << version << " | body ("
  54                 << _body.length() << ") = " << _body << endl;
  55 #endif // DEBUG
  56         headers["Accept-Ranges"] = "bytes";
  57         set_body(_body);
  58 }
  59
  60 void HTTPMessage::set_body(const string& _body) {
  61         body = _body;
  62         headers["Content-Length"] = String().from(body.length());
  63 }
  64
  65 const string& HTTPMessage::get_body(void) const {
  66         return body;
  67 }
  68
  69 istream& operator>>(istream& is, HTTPMessage& m)
  70                 throw(HTTPError, sockerr, ios::failure) {
  71 #ifdef DEBUG
  72         cerr << __FILE__ << "(" << __LINE__ << ")"
  73                 << ": operator>>()" << endl;
  74 #endif // DEBUG
  75         char buf[BUFSIZ];
  76         while (is.getline(buf, BUFSIZ)) {
  77                 String sbuf(buf);
  78                 sbuf.trim();
  79                 if (sbuf.length()) {
  80                         stringstream ss(sbuf);
  81                         ss >> m.headers;
  82                 // Fin de las cabeceras.
  83                 } else {
  84                         // Hay Content-Length, entonces hay body (no respeta RFC AFAIK).
  85                         if (m.headers.find("Content-Length") != m.headers.end()) {
  86                                 streamsize size, readed;
  87                                 to(m.headers["Content-Length"], size);
  88                                 char* const buf2 = new char[size+1];
  89                                 is.read(buf2, size);
  90                                 if (is.gcount() == size) {
  91                                         // Agrego fin de string porque el readsome no lo hace.
  92                                         buf2[size] = '\0';
  93                                         m.set_body(buf2);
  94 #ifdef DEBUG
  95                                         cerr << __FILE__ << "(" << __LINE__ << ")"
  96                                                 << ": operator>>() raaaaw body: " << buf2 << endl;
  97 #endif // DEBUG
  98                                 } else {
  99                                         // TODO else dar error?
 100 #ifdef DEBUG
 101                                         cerr << __FILE__ << "(" << __LINE__ << ")"
 102                                                 << ": operator>>() ERROR: No se pudo leer el mensaje completo. Se leyeron sóolo "
 103                                                 << is.gcount() << " bytes de " << size << " que deberían haberse leído." << endl;
 104 #endif // DEBUG
 105                                 }
 106                                 delete []buf2;
 107                         }
 108                         // Después de una línea vacía, haya obtenido el body o no, sale del
 109                         // while.
 110                         break;
 111                 }
 112         }
 113         return is;
 114 }
 115
 116 ostream& operator<<(ostream& os, const HTTPMessage& m) {
 117 #ifdef DEBUG
 118         cerr << __FILE__ << "(" << __LINE__ << ")"
 119                 << ": operator<<()" << endl;
 120 #endif // DEBUG
 121         return os << m.headers << "\r\n" // Fin de cabeceras
 122                 << m.body;
 123 }
 124
 125 string HTTPMessage::reason(unsigned code) {
 126         switch (code) {
 127                 // TODO completar los códigos.
 128                 case OK:
 129                         return "OK";
 130                 case BAD_REQUEST:
 131                         return "Bad Request";
 132                 case NOT_FOUND:
 133                         return "Not Found";
 134                 case LENGTH_REQUIRED:
 135                         return "Length Required";
 136                 case INTERNAL_SERVER_ERROR:
 137                         return "Internal Server Error";
 138                 case NOT_IMPLEMENTED:
 139                         return "Not Implemented";
 140                 case HTTP_VERSION_NOT_SUPPORTED:
 141                         return "HTTP Version Not Supported";
 142                 default:
 143                         return "No Reason";
 144         }
 145 }
 146
 147 } // namespace Server
 148
 149 } // namespace PlaQui
 150