mirror of
https://github.com/RetroShare/RetroShare.git
synced 2025-01-28 16:27:01 -05:00
2bf94b909a
* Stun code now runs continually - to check external network state. * Udpsorter controls DHT stun is on/off. (via p3ConnectMgr) * added code to enable threads to join/restart * enabled NetRestart for UDP and TCP. * tweaked networking code for faster startup (now ~30 seconds - can still be improved). * tweaked debug messages for testing networking * Added test for checking external IP address determination. git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@1492 b45a01b8-16f6-495d-af2f-9b41ad6348cc
162 lines
3.9 KiB
C++
162 lines
3.9 KiB
C++
/*
|
|
* libretroshare/src/tcponudp: udpsorter.h
|
|
*
|
|
* TCP-on-UDP (tou) network interface for RetroShare.
|
|
*
|
|
* Copyright 2007-2008 by Robert Fernie.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License Version 2 as published by the Free Software Foundation.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
|
|
* USA.
|
|
*
|
|
* Please report all bugs and problems to "retroshare@lunamutt.com".
|
|
*
|
|
*/
|
|
|
|
|
|
#ifndef TOU_UDP_SORTER_H
|
|
#define TOU_UDP_SORTER_H
|
|
|
|
/* universal networking functions */
|
|
#include "tou_net.h"
|
|
|
|
#include <iosfwd>
|
|
#include <map>
|
|
|
|
#include "udplayer.h"
|
|
/* UdpSorter ..... filters the UDP packets.
|
|
*/
|
|
|
|
class UdpPeer
|
|
{
|
|
public:
|
|
virtual void recvPkt(void *data, int size) = 0;
|
|
};
|
|
|
|
|
|
|
|
class TouStunPeer
|
|
{
|
|
public:
|
|
TouStunPeer()
|
|
:response(false), lastsend(0), failCount(0)
|
|
{
|
|
eaddr.sin_addr.s_addr = 0;
|
|
eaddr.sin_port = 0;
|
|
return;
|
|
}
|
|
|
|
TouStunPeer(std::string id_in, const struct sockaddr_in &addr)
|
|
:id(id_in), remote(addr), response(false), lastsend(0), failCount(0)
|
|
{
|
|
eaddr.sin_addr.s_addr = 0;
|
|
eaddr.sin_port = 0;
|
|
return;
|
|
}
|
|
|
|
std::string id;
|
|
struct sockaddr_in remote, eaddr;
|
|
bool response;
|
|
time_t lastsend;
|
|
uint32_t failCount;
|
|
};
|
|
|
|
|
|
class UdpSorter: public UdpReceiver
|
|
{
|
|
public:
|
|
|
|
UdpSorter(struct sockaddr_in &local);
|
|
virtual ~UdpSorter() { return; }
|
|
|
|
bool resetAddress(struct sockaddr_in &local);
|
|
|
|
/* add a TCPonUDP stream */
|
|
int addUdpPeer(UdpPeer *peer, const struct sockaddr_in &raddr);
|
|
int removeUdpPeer(UdpPeer *peer);
|
|
|
|
bool setStunKeepAlive(uint32_t required);
|
|
bool addStunPeer(const struct sockaddr_in &remote, const char *peerid);
|
|
bool getStunPeer(int idx, std::string &id,
|
|
struct sockaddr_in &remote, struct sockaddr_in &eaddr,
|
|
uint32_t &failCount, time_t &lastSend);
|
|
|
|
bool checkStunKeepAlive();
|
|
bool needStunPeers();
|
|
|
|
bool externalAddr(struct sockaddr_in &remote, uint8_t &stable);
|
|
|
|
/* Packet IO */
|
|
/* pass-through send packets */
|
|
int sendPkt(void *data, int size, struct sockaddr_in &to, int ttl);
|
|
/* callback for recved data (overloaded from UdpReceiver) */
|
|
virtual void recvPkt(void *data, int size, struct sockaddr_in &from);
|
|
|
|
int status(std::ostream &out);
|
|
|
|
/* setup connections */
|
|
int openSocket();
|
|
|
|
/* monitoring / updates */
|
|
int okay();
|
|
int tick();
|
|
|
|
int close();
|
|
|
|
private:
|
|
|
|
/* STUN handling */
|
|
bool locked_handleStunPkt(void *data, int size, struct sockaddr_in &from);
|
|
|
|
int doStun(struct sockaddr_in stun_addr);
|
|
|
|
|
|
/* stun keepAlive */
|
|
bool locked_printStunList();
|
|
bool locked_recvdStun(const struct sockaddr_in &remote, const struct sockaddr_in &extaddr);
|
|
bool locked_checkExternalAddress();
|
|
|
|
bool storeStunPeer(const struct sockaddr_in &remote, const char *peerid, bool sent);
|
|
|
|
UdpLayer *udpLayer;
|
|
|
|
RsMutex sortMtx; /* for all class data (below) */
|
|
|
|
struct sockaddr_in laddr; /* local addr */
|
|
|
|
struct sockaddr_in eaddr; /* external addr */
|
|
bool eaddrKnown;
|
|
bool eaddrStable; /* if true then usable. if false -> Symmettric NAT */
|
|
time_t eaddrTime;
|
|
|
|
bool mStunKeepAlive;
|
|
time_t mStunLastRecv;
|
|
time_t mStunLastSend;
|
|
|
|
std::list<TouStunPeer> mStunList; /* potentials */
|
|
|
|
std::map<struct sockaddr_in, UdpPeer *> streams;
|
|
|
|
|
|
|
|
};
|
|
|
|
/* generic stun functions */
|
|
|
|
bool UdpStun_isStunPacket(void *data, int size);
|
|
bool UdpStun_response(void *stun_pkt, int size, struct sockaddr_in &addr);
|
|
void *UdpStun_generate_stun_reply(struct sockaddr_in *stun_addr, int *len);
|
|
bool UdpStun_generate_stun_pkt(void *stun_pkt, int *len);
|
|
|
|
#endif
|