source/message.cpp

   1 /* $Id$
   2
   3 This file is part of libmsphttp
   4 Copyright © 2008  Mikkosoft Productions, Mikko Rasa
   5 Distributed under the LGPL
   6 */
   7
   8 #include <cstdlib>
   9 #include <msp/strings/formatter.h>
  10 #include <msp/strings/utils.h>
  11 #include "message.h"
  12
  13 using namespace std;
  14
  15 namespace Msp {
  16 namespace Http {
  17
  18 Message::Message():
  19         http_version(0x11),
  20         chunk_length(0),
  21         complete(false)
  22 { }
  23
  24 void Message::set_header(const string &hdr, const string &val)
  25 {
  26         headers[normalize_header_name(hdr)]=val;
  27 }
  28
  29 bool Message::has_header(const string &hdr) const
  30 {
  31         return headers.count(normalize_header_name(hdr));
  32 }
  33
  34 const string &Message::get_header(const string &hdr) const
  35 {
  36         HeaderMap::const_iterator i=headers.find(normalize_header_name(hdr));
  37         if(i==headers.end())
  38                 throw KeyError("Undefined header", hdr);
  39
  40         return i->second;
  41 }
  42
  43 void Message::add_content(const string &d)
  44 {
  45         content+=d;
  46         if(headers.count("Content-Type")==0)
  47                 set_header("Content-Type", "text/plain");
  48         set_header("Content-Length", lexical_cast(content.size()));
  49 }
  50
  51 void Message::set_user_data(const Variant &d)
  52 {
  53         user_data=d;
  54 }
  55
  56 unsigned Message::parse_content(const string &d)
  57 {
  58         if(complete)
  59                 return 0;
  60
  61         HeaderMap::const_iterator i=headers.find("Content-Length");
  62         if(i!=headers.end())
  63         {
  64                 unsigned needed=lexical_cast<unsigned>(i->second)-content.size();
  65                 unsigned len=min(needed, d.size());
  66
  67                 content.append(d, 0, len);
  68
  69                 if(len==needed)
  70                         complete=true;
  71
  72                 return len;
  73         }
  74
  75         i=headers.find("Transfer-Encoding");
  76         if(i!=headers.end() && strcasecmp(i->second, "chunked")==0)
  77         {
  78                 unsigned pos=0;
  79                 while(!complete && pos<d.size())
  80                 {
  81                         if(chunk_length==0)
  82                         {
  83                                 unsigned lf=d.find('\n', pos);
  84                                 if(lf==string::npos)
  85                                         return pos;
  86                                 chunk_length=lexical_cast<unsigned>(strip(d.substr(pos, lf-pos)), "x");
  87                                 if(chunk_length==0)
  88                                         complete=true;
  89                                 pos=lf+1;
  90                         }
  91                         else
  92                         {
  93                                 unsigned len=min(chunk_length, d.size()-pos);
  94                                 content.append(d, pos, len);
  95                                 chunk_length-=len;
  96                                 if((pos=d.find('\n', pos+len))!=string::npos)
  97                                         ++pos;
  98                         }
  99                 }
 100
 101                 return pos;
 102         }
 103
 104         complete=true;
 105         return 0;
 106 }
 107
 108 unsigned Message::parse_headers(const string &d)
 109 {
 110         unsigned start=0;
 111         while(1)
 112         {
 113                 unsigned lf=d.find('\n', start);
 114                 if(lf==string::npos)
 115                         throw InvalidParameterValue("Incomplete response");
 116                 if(lf==start || (d[start]=='\r' && lf==start+1))
 117                         return lf+1;
 118
 119                 unsigned colon=d.find(':', start);
 120                 if(colon>lf)
 121                         throw InvalidParameterValue("No colon in header");
 122
 123                 set_header(d.substr(start, colon-start), strip(d.substr(colon+1, lf-colon-1)));
 124
 125                 start=lf+1;
 126         }
 127 }
 128
 129 string Message::str_common() const
 130 {
 131         string result;
 132
 133         for(HeaderMap::const_iterator i=headers.begin(); i!=headers.end(); ++i)
 134                 if(i->first[0]!='-')
 135                         result+=format("%s: %s\r\n", i->first, i->second);
 136         result+="\r\n";
 137         result+=content;
 138
 139         return result;
 140 }
 141
 142 string Message::normalize_header_name(const string &hdr) const
 143 {
 144         string result=hdr;
 145         bool upper=true;
 146         for(string::iterator i=result.begin(); i!=result.end(); ++i)
 147         {
 148                 if(upper)
 149                 {
 150                         *i=toupper(*i);
 151                         upper=false;
 152                 }
 153                 else if(*i=='-')
 154                         upper=true;
 155                 else
 156                         *i=tolower(*i);
 157         }
 158         return result;
 159 }
 160
 161 } // namespace Http
 162 } // namespace Msp