| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * libjingle | |
| 3 * Copyright 2011 Google Inc. | |
| 4 * | |
| 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 * 2. Redistributions in binary form must reproduce the above copyright notice, | |
| 11 * this list of conditions and the following disclaimer in the documentation | |
| 12 * and/or other materials provided with the distribution. | |
| 13 * 3. The name of the author may not be used to endorse or promote products | |
| 14 * derived from this software without specific prior written permission. | |
| 15 * | |
| 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED | |
| 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
| 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO | |
| 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
| 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
| 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
| 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
| 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
| 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 26 */ | |
| 27 | |
| 28 #ifndef TALK_EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_ | |
| 29 #define TALK_EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_ | |
| 30 #pragma once | |
| 31 | |
| 32 #ifdef WIN32 | |
| 33 #include <winsock2.h> | |
| 34 typedef int socklen_t; | |
| 35 typedef SOCKET NativeSocket; | |
| 36 #else | |
| 37 #include <netinet/in.h> | |
| 38 #include <sys/select.h> | |
| 39 #include <sys/socket.h> | |
| 40 #define closesocket close | |
| 41 typedef int NativeSocket; | |
| 42 | |
| 43 #ifndef SOCKET_ERROR | |
| 44 #define SOCKET_ERROR (-1) | |
| 45 #endif | |
| 46 | |
| 47 #ifndef INVALID_SOCKET | |
| 48 #define INVALID_SOCKET static_cast<NativeSocket>(-1) | |
| 49 #endif | |
| 50 #endif | |
| 51 | |
| 52 #include <string> | |
| 53 | |
| 54 class SocketBase { | |
| 55 public: | |
| 56 SocketBase() : socket_(INVALID_SOCKET) { } | |
| 57 explicit SocketBase(NativeSocket socket) : socket_(socket) { } | |
| 58 ~SocketBase() { Close(); } | |
| 59 | |
| 60 NativeSocket socket() const { return socket_; } | |
| 61 bool valid() const { return socket_ != INVALID_SOCKET; } | |
| 62 | |
| 63 bool Create(); | |
| 64 void Close(); | |
| 65 | |
| 66 protected: | |
| 67 NativeSocket socket_; | |
| 68 }; | |
| 69 | |
| 70 // Represents an HTTP server socket. | |
| 71 class DataSocket : public SocketBase { | |
| 72 public: | |
| 73 enum RequestMethod { | |
| 74 INVALID, | |
| 75 GET, | |
| 76 POST, | |
| 77 OPTIONS, | |
| 78 }; | |
| 79 | |
| 80 explicit DataSocket(NativeSocket socket) | |
| 81 : SocketBase(socket), | |
| 82 method_(INVALID), | |
| 83 content_length_(0) { | |
| 84 } | |
| 85 | |
| 86 ~DataSocket() { | |
| 87 } | |
| 88 | |
| 89 static const char kCrossOriginAllowHeaders[]; | |
| 90 | |
| 91 bool headers_received() const { return method_ != INVALID; } | |
| 92 | |
| 93 RequestMethod method() const { return method_; } | |
| 94 | |
| 95 const std::string& request_path() const { return request_path_; } | |
| 96 std::string request_arguments() const; | |
| 97 | |
| 98 const std::string& data() const { return data_; } | |
| 99 | |
| 100 const std::string& content_type() const { return content_type_; } | |
| 101 | |
| 102 size_t content_length() const { return content_length_; } | |
| 103 | |
| 104 bool request_received() const { | |
| 105 return headers_received() && (method_ != POST || data_received()); | |
| 106 } | |
| 107 | |
| 108 bool data_received() const { | |
| 109 return method_ != POST || data_.length() >= content_length_; | |
| 110 } | |
| 111 | |
| 112 // Checks if the request path (minus arguments) matches a given path. | |
| 113 bool PathEquals(const char* path) const; | |
| 114 | |
| 115 // Called when we have received some data from clients. | |
| 116 // Returns false if an error occurred. | |
| 117 bool OnDataAvailable(bool* close_socket); | |
| 118 | |
| 119 // Send a raw buffer of bytes. | |
| 120 bool Send(const std::string& data) const; | |
| 121 | |
| 122 // Send an HTTP response. The |status| should start with a valid HTTP | |
| 123 // response code, followed by a string. E.g. "200 OK". | |
| 124 // If |connection_close| is set to true, an extra "Connection: close" HTTP | |
| 125 // header will be included. |content_type| is the mime content type, not | |
| 126 // including the "Content-Type: " string. | |
| 127 // |extra_headers| should be either empty or a list of headers where each | |
| 128 // header terminates with "\r\n". | |
| 129 // |data| is the body of the message. It's length will be specified via | |
| 130 // a "Content-Length" header. | |
| 131 bool Send(const std::string& status, bool connection_close, | |
| 132 const std::string& content_type, | |
| 133 const std::string& extra_headers, const std::string& data) const; | |
| 134 | |
| 135 // Clears all held state and prepares the socket for receiving a new request. | |
| 136 void Clear(); | |
| 137 | |
| 138 protected: | |
| 139 // A fairly relaxed HTTP header parser. Parses the method, path and | |
| 140 // content length (POST only) of a request. | |
| 141 // Returns true if a valid request was received and no errors occurred. | |
| 142 bool ParseHeaders(); | |
| 143 | |
| 144 // Figures out whether the request is a GET or POST and what path is | |
| 145 // being requested. | |
| 146 bool ParseMethodAndPath(const char* begin, size_t len); | |
| 147 | |
| 148 // Determines the length of the body and it's mime type. | |
| 149 bool ParseContentLengthAndType(const char* headers, size_t length); | |
| 150 | |
| 151 protected: | |
| 152 RequestMethod method_; | |
| 153 size_t content_length_; | |
| 154 std::string content_type_; | |
| 155 std::string request_path_; | |
| 156 std::string request_headers_; | |
| 157 std::string data_; | |
| 158 }; | |
| 159 | |
| 160 // The server socket. Accepts connections and generates DataSocket instances | |
| 161 // for each new connection. | |
| 162 class ListeningSocket : public SocketBase { | |
| 163 public: | |
| 164 ListeningSocket() {} | |
| 165 | |
| 166 bool Listen(unsigned short port); | |
| 167 DataSocket* Accept() const; | |
| 168 }; | |
| 169 | |
| 170 #endif // TALK_EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_ | |
| OLD | NEW |