data_socket.h - mozsearch

Enable keyboard shortcuts

/*

 *  Copyright 2011 The WebRTC Project Authors. All rights reserved.

 *  Use of this source code is governed by a BSD-style license

 *  that can be found in the LICENSE file in the root of the source

 *  tree. An additional intellectual property rights grant can be found

 *  in the file PATENTS.  All contributing project authors may

 *  be found in the AUTHORS file in the root of the source tree.

*/

#ifndef EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_

#define EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_

#include <string>

#include "rtc_base/ip_address.h"

#include "rtc_base/net_helpers.h"

#ifdef WIN32

typedef int socklen_t;

typedef SOCKET NativeSocket;

#else

#include <sys/select.h>

#define closesocket close

typedef int NativeSocket;

#ifndef SOCKET_ERROR

#define SOCKET_ERROR (-1)

#endif

#ifndef INVALID_SOCKET

#define INVALID_SOCKET static_cast<NativeSocket>(-1)

#endif

#endif

class SocketBase {

 public:

  SocketBase() : socket_(INVALID_SOCKET) {}

  explicit SocketBase(NativeSocket socket) : socket_(socket) {}

  SocketBase(SocketBase& other) = delete;

  SocketBase& operator=(const SocketBase& other) = delete;

  ~SocketBase() { Close(); }

  NativeSocket socket() const { return socket_; }

  bool valid() const { return socket_ != INVALID_SOCKET; }

  bool Create();

  void Close();

 protected:

  NativeSocket socket_;

};

// Represents an HTTP server socket.

class DataSocket : public SocketBase {

 public:

  enum RequestMethod {

    INVALID,

    GET,

    POST,

    OPTIONS,

};

  explicit DataSocket(NativeSocket socket)

      : SocketBase(socket), method_(INVALID), content_length_(0) {}

  ~DataSocket() {}

  static const char kCrossOriginAllowHeaders[];

  bool headers_received() const { return method_ != INVALID; }

  RequestMethod method() const { return method_; }

  const std::string& request_path() const { return request_path_; }

  std::string request_arguments() const;

  const std::string& data() const { return data_; }

  const std::string& content_type() const { return content_type_; }

  size_t content_length() const { return content_length_; }

  bool request_received() const {

    return headers_received() && (method_ != POST || data_received());

  bool data_received() const {

    return method_ != POST || data_.length() >= content_length_;

  // Checks if the request path (minus arguments) matches a given path.

  bool PathEquals(const char* path) const;

  // Called when we have received some data from clients.

  // Returns false if an error occurred.

  bool OnDataAvailable(bool* close_socket);

  // Send a raw buffer of bytes.

  bool Send(const std::string& data) const;

  // Send an HTTP response.  The `status` should start with a valid HTTP

  // response code, followed by a string.  E.g. "200 OK".

  // If `connection_close` is set to true, an extra "Connection: close" HTTP

  // header will be included.  `content_type` is the mime content type, not

  // including the "Content-Type: " string.

  // `extra_headers` should be either empty or a list of headers where each

  // header terminates with "\r\n".

  // `data` is the body of the message.  It's length will be specified via

  // a "Content-Length" header.

  bool Send(const std::string& status,

            bool connection_close,

            const std::string& content_type,

            const std::string& extra_headers,

            const std::string& data) const;

  // Clears all held state and prepares the socket for receiving a new request.

  void Clear();

 protected:

  // A fairly relaxed HTTP header parser.  Parses the method, path and

  // content length (POST only) of a request.

  // Returns true if a valid request was received and no errors occurred.

  bool ParseHeaders();

  // Figures out whether the request is a GET or POST and what path is

  // being requested.

  bool ParseMethodAndPath(const char* begin, size_t len);

  // Determines the length of the body and it's mime type.

  bool ParseContentLengthAndType(const char* headers, size_t length);

 protected:

  RequestMethod method_;

  size_t content_length_;

  std::string content_type_;

  std::string request_path_;

  std::string request_headers_;

  std::string data_;

};

// The server socket.  Accepts connections and generates DataSocket instances

// for each new connection.

class ListeningSocket : public SocketBase {

 public:

  ListeningSocket() {}

  bool Listen(unsigned short port);

  DataSocket* Accept() const;

};

#endif  // EXAMPLES_PEERCONNECTION_SERVER_DATA_SOCKET_H_