mirror of
https://github.com/RetroShare/RetroShare.git
synced 2025-01-30 09:13:40 -05:00
103 lines
2.6 KiB
C
103 lines
2.6 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 UdpSorter: public UdpReceiver
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
UdpSorter(struct sockaddr_in &local);
|
||
|
|
virtual ~UdpSorter() { return; }
|
||
|
|
|
||
|
|
/* add a TCPonUDP stream */
|
||
|
|
int addUdpPeer(UdpPeer *peer, const struct sockaddr_in &raddr);
|
||
|
|
bool addStunPeer(const struct sockaddr_in &remote, const char *peerid);
|
||
|
|
bool externalAddr(struct sockaddr_in &remote);
|
||
|
|
|
||
|
|
/* 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 isStunPacket(void *data, int size);
|
||
|
|
bool handleStunPkt(void *data, int size, struct sockaddr_in &from);
|
||
|
|
|
||
|
|
int doStun(struct sockaddr_in stun_addr);
|
||
|
|
bool response(void *stun_pkt, int size, struct sockaddr_in &addr);
|
||
|
|
|
||
|
|
void *generate_stun_reply(struct sockaddr_in *stun_addr, int *len);
|
||
|
|
bool generate_stun_pkt(void *stun_pkt, int *len);
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
UdpLayer *udpLayer;
|
||
|
|
|
||
|
|
RsMutex sortMtx; /* for all class data (below) */
|
||
|
|
|
||
|
|
struct sockaddr_in laddr; /* local addr */
|
||
|
|
struct sockaddr_in eaddr; /* external addr */
|
||
|
|
bool eaddrKnown;
|
||
|
|
|
||
|
|
std::list<struct sockaddr_in> stunPeers; /* potentials */
|
||
|
|
|
||
|
|
std::map<struct sockaddr_in, UdpPeer *> streams;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|