2013-07-10 02:45:36 +02:00
|
|
|
/*
|
|
|
|
* libjingle
|
|
|
|
* Copyright 2004--2005, Google Inc.
|
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
* modification, are permitted provided that the following conditions are met:
|
|
|
|
*
|
|
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
|
|
* this list of conditions and the following disclaimer.
|
|
|
|
* 2. 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.
|
|
|
|
* 3. The name of the author may not be used to endorse or promote products
|
|
|
|
* derived from this software without specific prior written permission.
|
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 THE AUTHOR 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 TALK_P2P_BASE_TCPPORT_H_
|
|
|
|
#define TALK_P2P_BASE_TCPPORT_H_
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
#include <list>
|
2014-07-29 19:36:52 +02:00
|
|
|
#include "webrtc/base/asyncpacketsocket.h"
|
2013-07-10 02:45:36 +02:00
|
|
|
#include "talk/p2p/base/port.h"
|
|
|
|
|
|
|
|
namespace cricket {
|
|
|
|
|
|
|
|
class TCPConnection;
|
|
|
|
|
|
|
|
// Communicates using a local TCP port.
|
|
|
|
//
|
|
|
|
// This class is designed to allow subclasses to take advantage of the
|
|
|
|
// connection management provided by this class. A subclass should take of all
|
|
|
|
// packet sending and preparation, but when a packet is received, it should
|
|
|
|
// call this TCPPort::OnReadPacket (3 arg) to dispatch to a connection.
|
|
|
|
class TCPPort : public Port {
|
|
|
|
public:
|
2014-07-29 19:36:52 +02:00
|
|
|
static TCPPort* Create(rtc::Thread* thread,
|
|
|
|
rtc::PacketSocketFactory* factory,
|
|
|
|
rtc::Network* network,
|
|
|
|
const rtc::IPAddress& ip,
|
2013-07-10 02:45:36 +02:00
|
|
|
int min_port, int max_port,
|
|
|
|
const std::string& username,
|
|
|
|
const std::string& password,
|
|
|
|
bool allow_listen) {
|
|
|
|
TCPPort* port = new TCPPort(thread, factory, network,
|
|
|
|
ip, min_port, max_port,
|
|
|
|
username, password, allow_listen);
|
|
|
|
if (!port->Init()) {
|
|
|
|
delete port;
|
|
|
|
port = NULL;
|
|
|
|
}
|
|
|
|
return port;
|
|
|
|
}
|
|
|
|
virtual ~TCPPort();
|
|
|
|
|
|
|
|
virtual Connection* CreateConnection(const Candidate& address,
|
|
|
|
CandidateOrigin origin);
|
|
|
|
|
|
|
|
virtual void PrepareAddress();
|
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
virtual int GetOption(rtc::Socket::Option opt, int* value);
|
|
|
|
virtual int SetOption(rtc::Socket::Option opt, int value);
|
2013-07-10 02:45:36 +02:00
|
|
|
virtual int GetError();
|
|
|
|
|
|
|
|
protected:
|
2014-07-29 19:36:52 +02:00
|
|
|
TCPPort(rtc::Thread* thread, rtc::PacketSocketFactory* factory,
|
|
|
|
rtc::Network* network, const rtc::IPAddress& ip,
|
2013-07-10 02:45:36 +02:00
|
|
|
int min_port, int max_port, const std::string& username,
|
|
|
|
const std::string& password, bool allow_listen);
|
|
|
|
bool Init();
|
|
|
|
|
|
|
|
// Handles sending using the local TCP socket.
|
|
|
|
virtual int SendTo(const void* data, size_t size,
|
2014-07-29 19:36:52 +02:00
|
|
|
const rtc::SocketAddress& addr,
|
|
|
|
const rtc::PacketOptions& options,
|
2013-09-23 22:34:45 +02:00
|
|
|
bool payload);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
|
|
|
// Accepts incoming TCP connection.
|
2014-07-29 19:36:52 +02:00
|
|
|
void OnNewConnection(rtc::AsyncPacketSocket* socket,
|
|
|
|
rtc::AsyncPacketSocket* new_socket);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
|
|
|
private:
|
|
|
|
struct Incoming {
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::SocketAddress addr;
|
|
|
|
rtc::AsyncPacketSocket* socket;
|
2013-07-10 02:45:36 +02:00
|
|
|
};
|
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::AsyncPacketSocket* GetIncoming(
|
|
|
|
const rtc::SocketAddress& addr, bool remove = false);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
|
|
|
// Receives packet signal from the local TCP Socket.
|
2014-07-29 19:36:52 +02:00
|
|
|
void OnReadPacket(rtc::AsyncPacketSocket* socket,
|
2013-07-10 02:45:36 +02:00
|
|
|
const char* data, size_t size,
|
2014-07-29 19:36:52 +02:00
|
|
|
const rtc::SocketAddress& remote_addr,
|
|
|
|
const rtc::PacketTime& packet_time);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
void OnReadyToSend(rtc::AsyncPacketSocket* socket);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
void OnAddressReady(rtc::AsyncPacketSocket* socket,
|
|
|
|
const rtc::SocketAddress& address);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
|
|
|
// TODO: Is this still needed?
|
|
|
|
bool incoming_only_;
|
|
|
|
bool allow_listen_;
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::AsyncPacketSocket* socket_;
|
2013-07-10 02:45:36 +02:00
|
|
|
int error_;
|
|
|
|
std::list<Incoming> incoming_;
|
|
|
|
|
|
|
|
friend class TCPConnection;
|
|
|
|
};
|
|
|
|
|
|
|
|
class TCPConnection : public Connection {
|
|
|
|
public:
|
|
|
|
// Connection is outgoing unless socket is specified
|
|
|
|
TCPConnection(TCPPort* port, const Candidate& candidate,
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::AsyncPacketSocket* socket = 0);
|
2013-07-10 02:45:36 +02:00
|
|
|
virtual ~TCPConnection();
|
|
|
|
|
2013-09-23 22:34:45 +02:00
|
|
|
virtual int Send(const void* data, size_t size,
|
2014-07-29 19:36:52 +02:00
|
|
|
const rtc::PacketOptions& options);
|
2013-07-10 02:45:36 +02:00
|
|
|
virtual int GetError();
|
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::AsyncPacketSocket* socket() { return socket_; }
|
2013-07-10 02:45:36 +02:00
|
|
|
|
|
|
|
private:
|
2014-07-29 19:36:52 +02:00
|
|
|
void OnConnect(rtc::AsyncPacketSocket* socket);
|
|
|
|
void OnClose(rtc::AsyncPacketSocket* socket, int error);
|
|
|
|
void OnReadPacket(rtc::AsyncPacketSocket* socket,
|
2013-07-10 02:45:36 +02:00
|
|
|
const char* data, size_t size,
|
2014-07-29 19:36:52 +02:00
|
|
|
const rtc::SocketAddress& remote_addr,
|
|
|
|
const rtc::PacketTime& packet_time);
|
|
|
|
void OnReadyToSend(rtc::AsyncPacketSocket* socket);
|
2013-07-10 02:45:36 +02:00
|
|
|
|
2014-07-29 19:36:52 +02:00
|
|
|
rtc::AsyncPacketSocket* socket_;
|
2013-07-10 02:45:36 +02:00
|
|
|
int error_;
|
|
|
|
|
|
|
|
friend class TCPPort;
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace cricket
|
|
|
|
|
|
|
|
#endif // TALK_P2P_BASE_TCPPORT_H_
|