Sockets/HttpResponse.cpp

   1 /**
   2  **     \file HttpResponse.cpp
   3  **     \date  2007-10-05
   4  **     \author grymse@alhem.net
   5 **/
   6 /*
   7 Copyright (C) 2007  Anders Hedstrom
   8
   9 This program is free software; you can redistribute it and/or
  10 modify it under the terms of the GNU General Public License
  11 as published by the Free Software Foundation; either version 2
  12 of the License, or (at your option) any later version.
  13
  14 This program is distributed in the hope that it will be useful,
  15 but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 GNU General Public License for more details.
  18
  19 You should have received a copy of the GNU General Public License
  20 along with this program; if not, write to the Free Software
  21 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22 */
  23 #ifdef _MSC_VER
  24 #pragma warning(disable:4786)
  25 #endif
  26 #include <stdarg.h>
  27 #include <stdio.h>
  28
  29 #include "HttpResponse.h"
  30 #include "HttpRequest.h"
  31 #include "MemFile.h"
  32 #include "File.h"
  33 #include "Parse.h"
  34 #include "Utility.h"
  35
  36 #ifdef SOCKETS_NAMESPACE
  37 namespace SOCKETS_NAMESPACE {
  38 #endif
  39
  40 #ifdef _DEBUG
  41 #define DEB(x) x; fflush(stderr);
  42 #else
  43 #define DEB(x)
  44 #endif
  45
  46
  47 // --------------------------------------------------------------------------------------
  48 HttpResponse::HttpResponse(const std::string& version) : HttpTransaction()
  49 , m_http_version( version )
  50 , m_http_status_code(0)
  51 , m_file( new MemFile )
  52 {
  53 }
  54
  55
  56 // --------------------------------------------------------------------------------------
  57 HttpResponse::~HttpResponse()
  58 {
  59         delete m_file;
  60 }
  61
  62
  63 // --------------------------------------------------------------------------------------
  64 void HttpResponse::SetHttpVersion(const std::string& value)
  65 {
  66         m_http_version = value;
  67 }
  68
  69
  70 // --------------------------------------------------------------------------------------
  71 const std::string& HttpResponse::HttpVersion() const
  72 {
  73         return m_http_version;
  74 }
  75
  76
  77
  78 // --------------------------------------------------------------------------------------
  79 void HttpResponse::SetHttpStatusCode(int value)
  80 {
  81         m_http_status_code = value;
  82 }
  83
  84
  85 int HttpResponse::HttpStatusCode() const
  86 {
  87         return m_http_status_code;
  88 }
  89
  90
  91
  92 // --------------------------------------------------------------------------------------
  93 void HttpResponse::SetHttpStatusMsg(const std::string& value)
  94 {
  95         m_http_status_msg = value;
  96 }
  97
  98
  99 const std::string& HttpResponse::HttpStatusMsg() const
 100 {
 101         return m_http_status_msg;
 102 }
 103
 104
 105 // --------------------------------------------------------------------------------------
 106 void HttpResponse::SetCookie(const std::string& value)
 107 {
 108         Parse pa(value, "=");
 109         std::string name = pa.getword();
 110         m_cookie[Utility::ToLower(name)] = value;
 111 DEB(fprintf(stderr, "HttpResponse::Set-Cookie<%s>: %s\n", name.c_str(), value.c_str());)
 112 }
 113
 114
 115 const std::string HttpResponse::Cookie(const std::string& name) const
 116 {
 117         std::map<std::string, std::string>::const_iterator it = m_cookie.find(name);
 118 DEB(fprintf(stderr, "HttpResponse; get value of Cookie<%s>: ", name.c_str());)
 119         if (it != m_cookie.end())
 120         {
 121 DEB(fprintf(stderr, "%s\n", it -> second.c_str());)
 122                 return it -> second;
 123         }
 124 DEB(fprintf(stderr, "\n");)
 125         return "";
 126 }
 127
 128
 129 std::list<std::string> HttpResponse::CookieNames() const
 130 {
 131         std::list<std::string> vec;
 132         DEB(fprintf(stderr, "HttpResponse::CookieNames; ");)
 133         for (std::map<std::string, std::string>::const_iterator it = m_cookie.begin(); it != m_cookie.end(); it++)
 134         {
 135                 DEB(fprintf(stderr, " %s", it -> first.c_str());)
 136                 vec.push_back(it -> first);
 137         }
 138         DEB(fprintf(stderr, "\n");)
 139         return vec;
 140 }
 141
 142
 143
 144 // --------------------------------------------------------------------------------------
 145 void HttpResponse::Write( const std::string& str )
 146 {
 147         Write( str.c_str(), str.size() );
 148 }
 149
 150
 151 // --------------------------------------------------------------------------------------
 152 void HttpResponse::Write( const char *buf, size_t sz )
 153 {
 154         m_file -> fwrite( buf, 1, sz );
 155 }
 156
 157
 158 // --------------------------------------------------------------------------------------
 159 void HttpResponse::Writef( const char *format, ... )
 160 {
 161         va_list ap;
 162         va_start(ap, format);
 163         char tmp[10000];
 164         vsprintf(tmp, format, ap);
 165         va_end(ap);
 166         m_file -> fwrite( tmp, 1, strlen(tmp) );
 167 }
 168
 169
 170 // --------------------------------------------------------------------------------------
 171 void HttpResponse::SetFile( const std::string& path )
 172 {
 173         delete m_file;
 174         m_file = new File();
 175         m_file -> fopen( path, "rb" );
 176 }
 177
 178
 179 // --------------------------------------------------------------------------------------
 180 void HttpResponse::Reset()
 181 {
 182         HttpTransaction::Reset();
 183         m_http_version = "";
 184         m_http_status_code = 0;
 185         m_http_status_msg = "";
 186         while (!m_cookie.empty())
 187         {
 188                 m_cookie.erase(m_cookie.begin());
 189         }
 190         if (m_file)
 191         {
 192                 delete m_file;
 193                 m_file = new MemFile;
 194         }
 195 }
 196
 197
 198 #ifdef SOCKETS_NAMESPACE
 199 } // namespace SOCKETS_NAMESPACE {
 200 #endif
 201