| 1 |
/* ============================================================================ |
| 2 |
* Douglas Thrift's Search Engine License |
| 3 |
* |
| 4 |
* Copyright (C) 2002-2004, 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 |
// $Id$ |
| 50 |
|
| 51 |
#ifndef _HttpHandler_hpp_ |
| 52 |
#define _HttpHandler_hpp_ |
| 53 |
|
| 54 |
#include "Search.hpp" |
| 55 |
#include "URL.hpp" |
| 56 |
|
| 57 |
// Lovely C Sockets! |
| 58 |
#ifdef _WIN32 |
| 59 |
// Windows Sockets |
| 60 |
#include <winsock2.h> |
| 61 |
#include <windows.h> |
| 62 |
|
| 63 |
inline int getpid() { return GetCurrentProcessId(); } |
| 64 |
#else |
| 65 |
// BSD Sockets |
| 66 |
typedef int SOCKET; |
| 67 |
|
| 68 |
#define INVALID_SOCKET -1 |
| 69 |
#define SOCKET_ERROR -1 |
| 70 |
#endif // _WIN32 |
| 71 |
|
| 72 |
#ifdef _OpenSSL_ |
| 73 |
#include <openssl/ssl.h> |
| 74 |
#include <openssl/err.h> |
| 75 |
#include <openssl/rand.h> |
| 76 |
#endif |
| 77 |
|
| 78 |
class HttpHandler |
| 79 |
{ |
| 80 |
private: |
| 81 |
enum Code { ok = 200, choices = 300, moved, found, notfound = 404, |
| 82 |
internal = 500 }; |
| 83 |
#ifdef _WIN32 |
| 84 |
WSADATA data; |
| 85 |
#endif // _WIN32 |
| 86 |
SOCKET http; |
| 87 |
#ifdef _OpenSSL_ |
| 88 |
SSL_CTX* ctx; |
| 89 |
SSL* ssl; |
| 90 |
bool tls; |
| 91 |
#endif |
| 92 |
char buffer[BUFSIZ + 1]; |
| 93 |
bool binary; |
| 94 |
stringstream page; |
| 95 |
string type; |
| 96 |
unsigned length; |
| 97 |
string location; |
| 98 |
bool chunked; |
| 99 |
void populate(); |
| 100 |
void putline(const string& line = ""); |
| 101 |
string getline(); |
| 102 |
void error(const string& prefix, bool host = false); |
| 103 |
#ifdef _OpenSSL_ |
| 104 |
void error(const string& prefix, int number); |
| 105 |
bool starttls(); |
| 106 |
#endif |
| 107 |
public: |
| 108 |
HttpHandler(); |
| 109 |
~HttpHandler(); |
| 110 |
bool handle(URL& url, const string& referer = "", bool head = false); |
| 111 |
iostream& pageStream() { return page; } |
| 112 |
istream& getline(string& line) { return std::getline(page, line); } |
| 113 |
istream& getline(string& line, char end) |
| 114 |
{ return std::getline(page, line, end); } |
| 115 |
bool good() const { return page.good(); } |
| 116 |
void clear(); |
| 117 |
const string& contentType() const { return type; } |
| 118 |
unsigned contentLength() const { return length; } |
| 119 |
const string& redirect() const { return location; } |
| 120 |
// friends: |
| 121 |
friend istream& operator>>(istream& is, Code& data) |
| 122 |
{ |
| 123 |
int number; |
| 124 |
|
| 125 |
is >> number; |
| 126 |
|
| 127 |
data = Code(number); |
| 128 |
|
| 129 |
return is; |
| 130 |
} |
| 131 |
}; |
| 132 |
|
| 133 |
#endif // _HttpHandler_hpp_ |
