mirror of
https://github.com/XRPLF/rippled.git
synced 2026-04-29 15:37:57 +00:00
136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
/*
|
|
* Copyright (c) 2011, Peter Thorson. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* * Neither the name of the WebSocket++ Project nor the
|
|
* names of its contributors may be used to endorse or promote products
|
|
* derived from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
|
|
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#ifndef WEBSOCKET_CLIENT_SESSION_HPP
|
|
#define WEBSOCKET_CLIENT_SESSION_HPP
|
|
|
|
#include <boost/asio.hpp>
|
|
#include <boost/bind.hpp>
|
|
|
|
#include <boost/shared_ptr.hpp>
|
|
#include <boost/enable_shared_from_this.hpp>
|
|
|
|
#if defined(WIN32)
|
|
#include <winsock2.h>
|
|
#else
|
|
#include <arpa/inet.h>
|
|
#endif
|
|
|
|
#include <algorithm>
|
|
#include <exception>
|
|
#include <iostream>
|
|
#include <map>
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace websocketpp {
|
|
class client_session;
|
|
typedef boost::shared_ptr<client_session> client_session_ptr;
|
|
}
|
|
|
|
#include "websocket_session.hpp"
|
|
#include "websocket_client.hpp"
|
|
|
|
using boost::asio::ip::tcp;
|
|
|
|
namespace websocketpp {
|
|
|
|
class client_session : public session {
|
|
public:
|
|
client_session (client_ptr c,
|
|
boost::asio::io_service& io_service,
|
|
connection_handler_ptr defc,
|
|
uint64_t buf_size);
|
|
|
|
/*** CLIENT INTERFACE ***/
|
|
|
|
// This function is called when a tcp connection has been established and
|
|
// the connection is ready to start the opening handshake.
|
|
void on_connect();
|
|
|
|
/*** HANDSHAKE INTERFACE ***/
|
|
|
|
void set_uri(const std::string& url);
|
|
|
|
bool get_secure() const;
|
|
std::string get_host() const;
|
|
uint16_t get_port() const;
|
|
|
|
// Set an HTTP header for the outgoing client handshake.
|
|
void set_header(const std::string& key,const std::string& val);
|
|
|
|
// adds a subprotocol. This will result in the appropriate
|
|
// Sec-WebSocket-Protocol header being sent with the opening connection.
|
|
// Values will be sent in the order they were added. Servers interpret this
|
|
// order as the preferred order.
|
|
void add_subprotocol(const std::string &val);
|
|
|
|
// Sets the origin value that will be sent to the server
|
|
void set_origin(const std::string &val);
|
|
|
|
// Adds an extension to the extension list. Extensions are sent in the
|
|
// order added
|
|
void add_extension(const std::string& val);
|
|
|
|
/*** SESSION INTERFACE ***/
|
|
// see session
|
|
bool is_server() const {return false;}
|
|
|
|
void log(const std::string& msg, uint16_t level) const;
|
|
void access_log(const std::string& msg, uint16_t level) const;
|
|
protected:
|
|
// Opening handshake processors and callbacks.
|
|
virtual void write_handshake();
|
|
virtual void handle_write_handshake(const boost::system::error_code& e);
|
|
virtual void read_handshake();
|
|
virtual void handle_read_handshake(const boost::system::error_code& e,
|
|
std::size_t bytes_transferred);
|
|
|
|
private:
|
|
|
|
protected:
|
|
ws_uri m_uri;
|
|
// url parts
|
|
bool m_secure;
|
|
std::string m_host;
|
|
uint16_t m_port;
|
|
|
|
// handshake stuff
|
|
std::string m_client_key;
|
|
|
|
// connection resources
|
|
client_ptr m_client;
|
|
private:
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif // WEBSOCKET_CLIENT_SESSION_HPP
|