HTTPServerConnection.cpp

Go to the documentation of this file.

 
// HTTPConnection.cpp
 
// Implements the cHTTPServerConnection class representing a single persistent connection in the HTTP server.
 
#include "Globals.h"
#include "HTTPServerConnection.h"
#include "HTTPMessage.h"
#include "HTTPServer.h"
 
 
 
 
 
cHTTPServerConnection::cHTTPServerConnection(cHTTPServer & a_HTTPServer) :
    m_HTTPServer(a_HTTPServer),
    m_Parser(*this)
{
}
 
 
 
 
 
cHTTPServerConnection::~cHTTPServerConnection() = default;
 
 
 
 
 
void cHTTPServerConnection::SendStatusAndReason(int a_StatusCode, const AString & a_Response)
{
    SendData(fmt::format(FMT_STRING("HTTP/1.1 {} {}\r\n"), a_StatusCode, a_Response));
    SendData(fmt::format(FMT_STRING("Content-Length: {}\r\n\r\n"), a_Response));
    SendData(a_Response.data(), a_Response.size());
    m_CurrentRequest.reset();
    m_Parser.Reset();
}
 
 
 
 
 
void cHTTPServerConnection::SendNeedAuth(const AString & a_Realm)
{
    SendData(fmt::format(FMT_STRING("HTTP/1.1 401 Unauthorized\r\nWWW-Authenticate: Basic realm=\"{}\"\r\nContent-Length: 0\r\n\r\n"), a_Realm));
    m_CurrentRequest.reset();
    m_Parser.Reset();
}
 
 
 
 
 
void cHTTPServerConnection::Send(const cHTTPOutgoingResponse & a_Response)
{
    ASSERT(m_CurrentRequest != nullptr);
    AString toSend;
    a_Response.AppendToData(toSend);
    SendData(toSend);
}
 
 
 
 
 
void cHTTPServerConnection::Send(const void * a_Data, size_t a_Size)
{
    ASSERT(m_CurrentRequest != nullptr);
    // We're sending in Chunked transfer encoding
    SendData(fmt::format("{0:x}\r\n", a_Size));
    SendData(a_Data, a_Size);
    SendData("\r\n");
}
 
 
 
 
 
void cHTTPServerConnection::FinishResponse(void)
{
    ASSERT(m_CurrentRequest != nullptr);
    SendData("0\r\n\r\n");
    m_CurrentRequest.reset();
    m_Parser.Reset();
}
 
 
 
 
 
void cHTTPServerConnection::Terminate(void)
{
    if (m_CurrentRequest != nullptr)
    {
        m_HTTPServer.RequestFinished(*this, *m_CurrentRequest);
    }
    m_Link->Close();  // Terminate the connection
    m_Link.reset();
}
 
 
 
 
 
void cHTTPServerConnection::OnLinkCreated(cTCPLinkPtr a_Link)
{
    ASSERT(m_Link == nullptr);
    m_Link = a_Link;
}
 
 
 
 
 
void cHTTPServerConnection::OnReceivedData(const char * a_Data, size_t a_Size)
{
    ASSERT(m_Link != nullptr);
 
    m_Parser.Parse(a_Data, a_Size);
}
 
 
 
 
 
void cHTTPServerConnection::OnRemoteClosed(void)
{
    if (m_CurrentRequest != nullptr)
    {
        m_HTTPServer.RequestFinished(*this, *m_CurrentRequest);
    }
    m_Link.reset();
}
 
 
 
 
 
void cHTTPServerConnection::OnError(int a_ErrorCode, const AString & a_ErrorMsg)
{
    OnRemoteClosed();
}
 
 
 
 
 
void cHTTPServerConnection::OnError(const AString & a_ErrorDescription)
{
    Terminate();  // HTTP data malformed, disconnect
}
 
 
 
 
 
void cHTTPServerConnection::OnFirstLine(const AString & a_FirstLine)
{
    const auto Split = StringSplit(a_FirstLine, " ");
    if (Split.size() < 2)
    {
        // Invalid request line. We need at least the Method and URL:
        Terminate();
        return;
    }
 
    // Create a new request object for this request:
    m_CurrentRequest = std::make_unique<cHTTPIncomingRequest>(Split[0], Split[1]);
}
 
 
 
 
 
void cHTTPServerConnection::OnHeaderLine(const AString & a_Key, const AString & a_Value)
{
    if (m_CurrentRequest == nullptr)
    {
        return;
    }
    m_CurrentRequest->AddHeader(a_Key, a_Value);
}
 
 
 
 
 
void cHTTPServerConnection::OnHeadersFinished(void)
{
    if (m_CurrentRequest == nullptr)
    {
        return;
    }
    m_HTTPServer.NewRequest(*this, *m_CurrentRequest);
}
 
 
 
 
 
void cHTTPServerConnection::OnBodyData(const void * a_Data, size_t a_Size)
{
    if (m_CurrentRequest == nullptr)
    {
        return;
    }
    m_HTTPServer.RequestBody(*this, *m_CurrentRequest, a_Data, a_Size);
}
 
 
 
 
 
void cHTTPServerConnection::OnBodyFinished(void)
{
    // Process the request:
    if (m_CurrentRequest != nullptr)
    {
        m_HTTPServer.RequestFinished(*this, *m_CurrentRequest);
    }
 
    // ...and reset:
    m_CurrentRequest.reset();
    m_Parser.Reset();
}
 
 
 
 
 
void cHTTPServerConnection::SendData(const void * a_Data, size_t a_Size)
{
    m_Link->Send(a_Data, a_Size);
}