103 lines
1.9 KiB
C++
103 lines
1.9 KiB
C++
|
|
// HTTPMessage.cpp
|
|
|
|
// Declares the cHTTPMessage class representing the common ancestor for HTTP request and response classes
|
|
|
|
#include "Globals.h"
|
|
#include "HTTPMessage.h"
|
|
|
|
|
|
|
|
|
|
|
|
// Disable MSVC warnings:
|
|
#if defined(_MSC_VER)
|
|
#pragma warning(push)
|
|
#pragma warning(disable:4355) // 'this' : used in base member initializer list
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// cHTTPMessage:
|
|
|
|
cHTTPMessage::cHTTPMessage(eKind a_Kind) :
|
|
m_Kind(a_Kind),
|
|
m_ContentLength(AString::npos)
|
|
{
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cHTTPMessage::AddHeader(const AString & a_Key, const AString & a_Value)
|
|
{
|
|
auto Key = StrToLower(a_Key);
|
|
auto itr = m_Headers.find(Key);
|
|
if (itr == m_Headers.end())
|
|
{
|
|
m_Headers[Key] = a_Value;
|
|
}
|
|
else
|
|
{
|
|
// The header-field key is specified multiple times, combine into comma-separated list (RFC 2616 @ 4.2)
|
|
itr->second.append(", ");
|
|
itr->second.append(a_Value);
|
|
}
|
|
|
|
// Special processing for well-known headers:
|
|
if (Key == "content-type")
|
|
{
|
|
m_ContentType = m_Headers[Key];
|
|
}
|
|
else if (Key == "content-length")
|
|
{
|
|
if (!StringToInteger(m_Headers[Key], m_ContentLength))
|
|
{
|
|
m_ContentLength = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// cHTTPResponse:
|
|
|
|
cHTTPResponse::cHTTPResponse(void) :
|
|
super(mkResponse)
|
|
{
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cHTTPResponse::AppendToData(AString & a_DataStream) const
|
|
{
|
|
a_DataStream.append("HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\nContent-Type: ");
|
|
a_DataStream.append(m_ContentType);
|
|
a_DataStream.append("\r\n");
|
|
for (auto itr = m_Headers.cbegin(), end = m_Headers.cend(); itr != end; ++itr)
|
|
{
|
|
if ((itr->first == "Content-Type") || (itr->first == "Content-Length"))
|
|
{
|
|
continue;
|
|
}
|
|
a_DataStream.append(itr->first);
|
|
a_DataStream.append(": ");
|
|
a_DataStream.append(itr->second);
|
|
a_DataStream.append("\r\n");
|
|
} // for itr - m_Headers[]
|
|
a_DataStream.append("\r\n");
|
|
}
|
|
|
|
|
|
|
|
|