1 |
/* ============================================================================ |
2 |
* Douglas Thrift's Search Engine License |
3 |
* |
4 |
* Copyright (C) 2002, Douglas Thrift. All Rights Reserved. |
5 |
* 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 |
// HttpHandler.h |
50 |
|
51 |
#ifndef _HttpHandler_h_ |
52 |
#define _HttpHandler_h_ |
53 |
|
54 |
#include "Search.h" |
55 |
#include "URL.h" |
56 |
|
57 |
// Lovely C Sockets! |
58 |
#ifdef _WIN32 |
59 |
|
60 |
// Windows Sockets |
61 |
#include <winsock2.h> |
62 |
#include <windows.h> |
63 |
|
64 |
#else |
65 |
|
66 |
// BSD Sockets |
67 |
#include <unistd.h> |
68 |
#include <sys/types.h> |
69 |
#include <sys/socket.h> |
70 |
#include <netinet/in.h> |
71 |
#include <netdb.h> |
72 |
|
73 |
#define INVALID_SOCKET -1 |
74 |
#define SOCKET_ERROR -1 |
75 |
|
76 |
typedef int SOCKET; |
77 |
|
78 |
inline int closesocket(SOCKET s) { return close(s); } |
79 |
|
80 |
#endif // _WIN32 |
81 |
|
82 |
class HttpHandler |
83 |
{ |
84 |
private: |
85 |
enum code {ok = 200, choices = 300, moved = 301, found = 302, notfound = |
86 |
404, internal = 500}; |
87 |
#ifdef _WIN32 |
88 |
WSADATA data; |
89 |
#endif // _WIN32 |
90 |
SOCKET http; |
91 |
char* buffer; |
92 |
unsigned begin; |
93 |
string page; |
94 |
string type; |
95 |
unsigned length; |
96 |
string location; |
97 |
bool chunked; |
98 |
void populate(); |
99 |
void putline(const string line = ""); |
100 |
string getline(); |
101 |
void error(const string& prefix, bool host = false); |
102 |
public: |
103 |
HttpHandler(); |
104 |
~HttpHandler(); |
105 |
bool handle(URL& url, bool head = false); |
106 |
HttpHandler& getline(string& line, char endline = '\n'); |
107 |
bool good(); |
108 |
void clear(); |
109 |
string contentType() { return type; } |
110 |
unsigned contentLength() { return length; } |
111 |
string redirect() { return location; } |
112 |
}; |
113 |
|
114 |
#endif // _HttpHandler_h_ |