trunk/Search/HttpHandler.h

/* ============================================================================
 * Douglas Thrift's Search Engine License
 *
 * Copyright (C) 2002-2003, Douglas Thrift. All Rights Reserved.
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * 3. The end-user documentation included with the redistribution, if any, must
 *    include the following acknowledgment:
 *
 *       "This product includes software developed by Douglas Thrift
 *       (http://computers.douglasthrift.net/searchengine/)."
 *
 *    Alternately, this acknowledgment may appear in the software itself, if
 *    and wherever such third-party acknowledgments normally appear.
 *
 * 4. The names "Douglas Thrift" and "Douglas Thrift's Search Engine" must not
 *    be used to endorse or promote products derived from this software without
 *    specific prior written permission.  For written permission, please visit
 *    http://www.douglasthrift.net/contact.cgi for contact information.
 *
 * 5. Products derived from this software may not be called "Douglas Thrift's
 *    Search Engine", nor may "Douglas Thrift's Search Engine" appear in their
 *    name, without prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * ============================================================================
 */
// Douglas Thrift's Search Engine HTTP Handler
//
// Douglas Thrift
//
// $Id: HttpHandler.h,v 1.19 2003/07/20 04:00:25 douglas Exp $

#ifndef _HttpHandler_h_
#define _HttpHandler_h_

#include "Search.h"
#include "URL.h"

// Lovely C Sockets!
#ifdef _WIN32
// Windows Sockets
#include <winsock2.h>
#include <windows.h>

inline int getpid() { return GetCurrentProcessId(); }
#else
// BSD Sockets
typedef int SOCKET;

#define INVALID_SOCKET -1
#define SOCKET_ERROR -1
#endif // _WIN32

#ifdef _OpenSSL_
#include <openssl/ssl.h>
#include <openssl/err.h>
#include <openssl/rand.h>
#endif

class HttpHandler
{
private:
        enum code {ok = 200, choices = 300, moved = 301, found = 302, notfound =
                404, internal = 500};
#ifdef _WIN32
        WSADATA data;
#endif // _WIN32
        SOCKET http;
#ifdef _OpenSSL_
        SSL_CTX* ctx;
        SSL* ssl;
        bool tls;
#endif
        char* buffer;
        bool binary;
        stringstream page;
        string type;
        unsigned length;
        string location;
        bool chunked;
        void populate();
        void putline(const string line = "");
        string getline();
        void error(const string& prefix, bool host = false);
#ifdef _OpenSSL_
        void error(const string& prefix, int number);
        bool starttls();
#endif
// friends:
    friend istream& operator>>(istream& is, code& data);
public:
        HttpHandler();
        ~HttpHandler();
        bool handle(URL& url, const string referer = "", bool head = false);
        iostream& pageStream() { return page; }
        istream& getline(string& line) { return std::getline(page, line); }
        istream& getline(string& line, char end) { return std::getline(page, line,
                end); }
        bool good() { return page.good(); }
        void clear();
        string contentType() { return type; }
        unsigned contentLength() { return length; }
        string redirect() { return location; }
};

#endif // _HttpHandler_h_
Revision:	212
Committed:	2003-07-19T21:00:25-07:00 (21 years, 11 months ago) by douglas
Content type:	text/x-c
File size:	4210 byte(s)
Log Message:	Replaced some strtoul functions with istringstream stuff, and fixed Outputer stuff.
#	User	Rev	Content
1	douglas	1	/* ============================================================================
2			* Douglas Thrift's Search Engine License
3			*
4	douglas	28	* Copyright (C) 2002-2003, Douglas Thrift. All Rights Reserved.
5	douglas	1	* Redistribution and use in source and binary forms, with or without
6			* modification, are permitted provided that the following conditions are met:
7			*
8			* 1. Redistributions of source code must retain the above copyright notice,
9			* this list of conditions and the following disclaimer.
10			*
11			* 2. Redistributions in binary form must reproduce the above copyright notice,
12			* this list of conditions and the following disclaimer in the documentation
13			* and/or other materials provided with the distribution.
14			*
15			* 3. The end-user documentation included with the redistribution, if any, must
16			* include the following acknowledgment:
17			*
18			* "This product includes software developed by Douglas Thrift
19			* (http://computers.douglasthrift.net/searchengine/)."
20			*
21			* Alternately, this acknowledgment may appear in the software itself, if
22			* and wherever such third-party acknowledgments normally appear.
23			*
24			* 4. The names "Douglas Thrift" and "Douglas Thrift's Search Engine" must not
25			* be used to endorse or promote products derived from this software without
26			* specific prior written permission. For written permission, please visit
27			* http://www.douglasthrift.net/contact.cgi for contact information.
28			*
29			* 5. Products derived from this software may not be called "Douglas Thrift's
30			* Search Engine", nor may "Douglas Thrift's Search Engine" appear in their
31			* name, without prior written permission.
32			*
33			* THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
34			* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
35			* FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
36			* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
37			* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
38			* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
39			* OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
40			* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
41			* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
42			* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
43			* ============================================================================
44			*/
45			// Douglas Thrift's Search Engine HTTP Handler
46			//
47			// Douglas Thrift
48			//
49	douglas	212	// $Id: HttpHandler.h,v 1.19 2003/07/20 04:00:25 douglas Exp $
50	douglas	1
51			#ifndef _HttpHandler_h_
52			#define _HttpHandler_h_
53
54			#include "Search.h"
55			#include "URL.h"
56
57	douglas	13	// Lovely C Sockets!
58			#ifdef _WIN32
59			// Windows Sockets
60			#include <winsock2.h>
61			#include <windows.h>
62	douglas	211
63			inline int getpid() { return GetCurrentProcessId(); }
64	douglas	13	#else
65			// BSD Sockets
66			typedef int SOCKET;
67	douglas	211
68			#define INVALID_SOCKET -1
69			#define SOCKET_ERROR -1
70	douglas	13	#endif // _WIN32
71
72	douglas	208	#ifdef _OpenSSL_
73			#include <openssl/ssl.h>
74			#include <openssl/err.h>
75			#include <openssl/rand.h>
76			#endif
77
78	douglas	1	class HttpHandler
79			{
80			private:
81	douglas	17	enum code {ok = 200, choices = 300, moved = 301, found = 302, notfound =
82			404, internal = 500};
83	douglas	14	#ifdef _WIN32
84			WSADATA data;
85			#endif // _WIN32
86	douglas	13	SOCKET http;
87	douglas	201	#ifdef _OpenSSL_
88	douglas	208	SSL_CTX* ctx;
89			SSL* ssl;
90	douglas	201	bool tls;
91			#endif
92	douglas	14	char* buffer;
93	douglas	208	bool binary;
94			stringstream page;
95	douglas	13	string type;
96			unsigned length;
97	douglas	17	string location;
98	douglas	18	bool chunked;
99	douglas	19	void populate();
100	douglas	14	void putline(const string line = "");
101	douglas	17	string getline();
102	douglas	18	void error(const string& prefix, bool host = false);
103	douglas	208	#ifdef _OpenSSL_
104			void error(const string& prefix, int number);
105			bool starttls();
106			#endif
107	douglas	212	// friends:
108			friend istream& operator>>(istream& is, code& data);
109	douglas	1	public:
110			HttpHandler();
111			~HttpHandler();
112	douglas	25	bool handle(URL& url, const string referer = "", bool head = false);
113	douglas	211	iostream& pageStream() { return page; }
114	douglas	208	istream& getline(string& line) { return std::getline(page, line); }
115			istream& getline(string& line, char end) { return std::getline(page, line,
116			end); }
117			bool good() { return page.good(); }
118	douglas	1	void clear();
119	douglas	13	string contentType() { return type; }
120			unsigned contentLength() { return length; }
121	douglas	17	string redirect() { return location; }
122	douglas	1	};
123
124			#endif // _HttpHandler_h_