1 |
douglas |
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 |
douglas |
13 |
// 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 |
douglas |
1 |
class HttpHandler |
83 |
|
|
{ |
84 |
|
|
private: |
85 |
douglas |
17 |
enum code {ok = 200, choices = 300, moved = 301, found = 302, notfound = |
86 |
|
|
404, internal = 500}; |
87 |
douglas |
14 |
#ifdef _WIN32 |
88 |
|
|
WSADATA data; |
89 |
|
|
#endif // _WIN32 |
90 |
douglas |
13 |
SOCKET http; |
91 |
douglas |
14 |
char* buffer; |
92 |
douglas |
18 |
unsigned begin; |
93 |
douglas |
1 |
string page; |
94 |
douglas |
13 |
string type; |
95 |
|
|
unsigned length; |
96 |
douglas |
17 |
string location; |
97 |
douglas |
18 |
bool chunked; |
98 |
douglas |
19 |
void populate(); |
99 |
douglas |
14 |
void putline(const string line = ""); |
100 |
douglas |
17 |
string getline(); |
101 |
douglas |
18 |
void error(const string& prefix, bool host = false); |
102 |
douglas |
1 |
public: |
103 |
|
|
HttpHandler(); |
104 |
|
|
~HttpHandler(); |
105 |
douglas |
14 |
bool handle(URL& url, bool head = false); |
106 |
douglas |
1 |
HttpHandler& getline(string& line, char endline = '\n'); |
107 |
|
|
bool good(); |
108 |
|
|
void clear(); |
109 |
douglas |
13 |
string contentType() { return type; } |
110 |
|
|
unsigned contentLength() { return length; } |
111 |
douglas |
17 |
string redirect() { return location; } |
112 |
douglas |
1 |
}; |
113 |
|
|
|
114 |
|
|
#endif // _HttpHandler_h_ |