최종완료보고 버전

hdssd/huslib/hsparser.h

@@ -0,0 +1,117 @@
+#ifndef HS_PARSER__H_
+#define HS_PARSER__H_
+
+#ifdef _WIN32
+
+#pragma once
+#pragma warning(disable:4996)
+
+#endif
+
+
+class HHsHeaderElement
+{
+public:
+	HHsHeaderElement(const char * pszName, bool bNameMewMemory, const char * pszValue, bool bValueNewMemory);
+	~HHsHeaderElement();
+
+public:
+	char * m_pszName;
+	size_t m_nNameLength;
+	bool m_bNameNewMemory;
+
+	char * m_pszValue;
+	size_t m_nValueLength;
+	bool m_bValueNewMemory;
+
+	HHsHeaderElement * m_pNext;
+};
+
+///////////////////////////////////////
+
+class HHsParser
+{
+public:
+	enum
+	{
+		HP_PARAMETER_ERROR = 0,
+		HP_NOT_HTTP_PACKET = 1,
+		HP_HEADER_REMAIN_PACKET = 2,
+		HP_BODY_REMAIN_PACKET = 3,
+		HP_BODYSIZE_UNKNOWN_PACKET = 4,
+		HP_COMPLETE_PACKET = 5
+	};
+
+	static char * SearchBodyPtr(char * pData, unsigned int nSize);
+
+	////////////
+
+	HHsParser(const char * pszProtocolName);
+	virtual ~HHsParser();
+
+	char * GetProtocolName() { return m_szProtocolName; }
+	char * GetMethod() { return m_pszMethod; }
+	
+	void SetRequestUri(const char * pszRequestUri);
+	char * GetRequestUri() { return m_pszRequestUri; }
+	
+	void SetVersion(const char * pszVersion);
+	char * GetVersion() { return m_pszVersion; }
+
+	void SetStatusCode(int nStatusCode) { m_nStatusCode = nStatusCode; }
+	int GetStatusCode() { return m_nStatusCode; }
+
+	void SetReasonPhrase(const char * pszReasonPhrase);
+	char * GetReasonPhrase() { return m_pszReasonPhrase; }
+
+	size_t GetHeaderSize() { return m_nHeaderSize; }
+	unsigned long GetContentLength() { return m_nContentLength; }
+
+	void SetBody(char * pData, unsigned int nSize);
+	char * GetBodyPtr() { return m_pBody; }
+	size_t GetBodySize() { return m_nBodySize; }
+
+	HHsHeaderElement * GetHeaderElement(const char * pszHeaderName, int nIndex = 0);
+	const char * GetHeaderValue(const char * pszHeaderName, int nIndex = 0);
+
+	bool AddHeader(const char * pszHeaderName, const char * pszHeaderValue);
+	bool SetHeader(const char * pszHeaderName, const char * pszHeaderValue, int nIndex = 0);	
+
+	void RemoveHeader(const char * pszHeaderName, const char * pszHeaderValue = NULL, int nIndex = 0);
+
+	size_t RefleshHeaderSize();
+
+	int Parse(char * pData, unsigned int nSize, char * pBodyPtr = NULL);
+	bool ToString(char * pBuffer, bool bOnlyHeader);
+
+protected:
+	bool OrganizeHeader(char * pszHeaderName, char * pszHeaderValue);
+
+private:
+	char m_szProtocolName[8];
+	size_t m_nProtocolNameLength;
+
+	char * m_pszMethod;
+	
+	char * m_pszRequestUri;
+	bool m_bRequestUriNewMemory;
+
+	char * m_pszVersion;
+
+	int m_nStatusCode;
+	
+	char * m_pszReasonPhrase;
+	bool m_bReasonPhraseNewMemory;
+
+	HHsHeaderElement * m_pElementHead;
+	HHsHeaderElement * m_pElementRear;
+
+	size_t m_nHeaderSize;
+	unsigned long m_nContentLength;
+	
+	char * m_pBody;
+	bool m_bBodyNewMemory;
+	size_t m_nBodySize;
+};
+
+#endif // HS_PARSER__H_