| 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 |
string page; |
| 93 |
string type; |
| 94 |
unsigned length; |
| 95 |
string location; |
| 96 |
bool chunked; |
| 97 |
void populate(); |
| 98 |
void putline(const string line = ""); |
| 99 |
string getline(); |
| 100 |
void error(const string& prefix, bool host = false); |
| 101 |
public: |
| 102 |
HttpHandler(); |
| 103 |
~HttpHandler(); |
| 104 |
bool handle(URL& url, const string referer = "", bool head = false); |
| 105 |
HttpHandler& getline(string& line, char endline = '\n'); |
| 106 |
bool good() { return (page.length() > 0); } |
| 107 |
void clear(); |
| 108 |
string contentType() { return type; } |
| 109 |
unsigned contentLength() { return length; } |
| 110 |
string redirect() { return location; } |
| 111 |
}; |
| 112 |
|
| 113 |
#endif // _HttpHandler_h_ |
