113 lines
3.2 KiB
C++
113 lines
3.2 KiB
C++
#ifndef CHTTPPROTOCOL_H
|
|
#define CHTTPPROTOCOL_H
|
|
|
|
#include "CString.h"
|
|
#include "BasicTypes.h"
|
|
#include "stdlist.h"
|
|
#include "stdmap.h"
|
|
#include "stdvector.h"
|
|
|
|
class IInputStream;
|
|
class IOutputStream;
|
|
|
|
class CHTTPRequest {
|
|
public:
|
|
typedef std::list<std::pair<CString, CString> > CHeaderList;
|
|
typedef std::map<CString, CHeaderList::iterator,
|
|
CStringUtil::CaselessCmp> CHeaderMap;
|
|
typedef CHeaderList::const_iterator const_iterator;
|
|
|
|
CHTTPRequest();
|
|
~CHTTPRequest();
|
|
|
|
// manipulators
|
|
|
|
// add a header by name. replaces existing header, if any.
|
|
// headers are sent in the order they're inserted. replacing
|
|
// a header does not change its original position in the order.
|
|
void insertHeader(const CString& name, const CString& value);
|
|
|
|
// append a header. equivalent to insertHeader() if the header
|
|
// doesn't exist, otherwise it appends a comma and the value to
|
|
// the existing header.
|
|
void appendHeader(const CString& name, const CString& value);
|
|
|
|
// remove a header by name. does nothing if no such header.
|
|
void eraseHeader(const CString& name);
|
|
|
|
// accessors
|
|
|
|
// returns true iff the header exists
|
|
bool isHeader(const CString& name) const;
|
|
|
|
// get a header by name. returns the empty string if no such header.
|
|
CString getHeader(const CString& name) const;
|
|
|
|
// get iterator over all headers in the order they were added
|
|
const_iterator begin() const { return m_headers.begin(); }
|
|
const_iterator end() const { return m_headers.end(); }
|
|
|
|
public:
|
|
// note -- these members are public for convenience
|
|
CString m_method;
|
|
CString m_uri;
|
|
SInt32 m_majorVersion;
|
|
SInt32 m_minorVersion;
|
|
CString m_body;
|
|
|
|
private:
|
|
CHeaderList m_headers;
|
|
CHeaderMap m_headerByName;
|
|
};
|
|
|
|
class CHTTPReply {
|
|
public:
|
|
typedef std::vector<std::pair<CString, CString> > CHeaderList;
|
|
|
|
SInt32 m_majorVersion;
|
|
SInt32 m_minorVersion;
|
|
SInt32 m_status;
|
|
CString m_reason;
|
|
CString m_method;
|
|
|
|
CHeaderList m_headers;
|
|
|
|
CString m_body;
|
|
};
|
|
|
|
class CHTTPProtocol {
|
|
public:
|
|
// read and parse an HTTP request. result is returned in a
|
|
// CHTTPRequest which the client must delete. throws an
|
|
// XHTTP if there was a parse error. throws an XIO exception
|
|
// if there was a read error. if maxSize is greater than
|
|
// zero and the request is larger than maxSize bytes then
|
|
// throws XHTTP(413).
|
|
static CHTTPRequest* readRequest(IInputStream*, UInt32 maxSize = 0);
|
|
|
|
// send an HTTP reply on the stream
|
|
static void reply(IOutputStream*, CHTTPReply&);
|
|
|
|
// parse a multipart/form-data body into its parts. returns true
|
|
// iff the entire body was correctly parsed.
|
|
// FIXME -- name/value pairs insufficient to save part headers
|
|
typedef std::map<CString, CString> CFormParts;
|
|
static bool parseFormData(const CHTTPRequest&,
|
|
CFormParts& parts);
|
|
|
|
private:
|
|
static CString readLine(IInputStream*, CString& tmpBuffer);
|
|
static CString readBlock(IInputStream*,
|
|
UInt32 numBytes, CString& tmpBuffer);
|
|
static CString readChunk(IInputStream*, CString& tmpBuffer,
|
|
UInt32* maxSize);
|
|
static void readHeaders(IInputStream*,
|
|
CHTTPRequest*, bool isFooter,
|
|
CString& tmpBuffer,
|
|
UInt32* maxSize);
|
|
|
|
static bool isValidToken(const CString&);
|
|
};
|
|
|
|
#endif
|