mirror of
https://github.com/ZeroTier/ZeroTierOne
synced 2025-07-10 15:23:06 -07:00
Some work on simulated net...
This commit is contained in:
parent
ae228ade8c
commit
0a195e7bc0
5 changed files with 291 additions and 28 deletions
|
@ -28,43 +28,98 @@
|
|||
#ifndef ZT_SIMNETSOCKETMANAGER_HPP
|
||||
#define ZT_SIMNETSOCKETMANAGER_HPP
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include <map>
|
||||
#include <stdexcept>
|
||||
#include <utility>
|
||||
#include <queue>
|
||||
|
||||
#include "Constants.hpp"
|
||||
#include "../node/SocketManager.hpp"
|
||||
#include "../node/Mutex.hpp"
|
||||
#include "../node/Condition.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
class SimNet;
|
||||
|
||||
/**
|
||||
* Socket I/O implementation
|
||||
* Socket manager for an IP endpoint in a simulated network
|
||||
*/
|
||||
class SimNetSocketManager : public SocketManager
|
||||
{
|
||||
friend class SimNet;
|
||||
|
||||
public:
|
||||
struct TransferStats
|
||||
{
|
||||
TransferStats() : received(0),sent(0) {}
|
||||
unsigned long long received;
|
||||
unsigned long long sent;
|
||||
};
|
||||
|
||||
SimNetSocketManager();
|
||||
virtual ~SimNetSocketManager();
|
||||
|
||||
virtual bool send(
|
||||
const InetAddress &to,
|
||||
bool tcp,
|
||||
bool autoConnectTcp,
|
||||
const void *msg,
|
||||
unsigned int msglen);
|
||||
/**
|
||||
* @return IP address of this simulated endpoint
|
||||
*/
|
||||
inline const InetAddress &address() const { return _address; }
|
||||
|
||||
virtual void poll(
|
||||
unsigned long timeout,
|
||||
void (*handler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),
|
||||
void *arg);
|
||||
/**
|
||||
* @return Local endpoint stats
|
||||
*/
|
||||
inline const TransferStats &totals() const { return _totals; }
|
||||
|
||||
/**
|
||||
* @param peer Peer IP address
|
||||
* @return Transfer stats for this peer
|
||||
*/
|
||||
inline TransferStats stats(const InetAddress &peer) const
|
||||
{
|
||||
Mutex::Lock _l(_stats_m);
|
||||
return _stats[peer];
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Network to which this endpoint belongs
|
||||
*/
|
||||
inline SimNet *net() const { return _sn; }
|
||||
|
||||
/**
|
||||
* Enqueue data from another endpoint to be picked up on next poll()
|
||||
*
|
||||
* @param from Originating endpoint address
|
||||
* @param data Data
|
||||
* @param len Length of data in bytes
|
||||
*/
|
||||
inline void enqueue(const InetAddress &from,const void *data,unsigned int len)
|
||||
{
|
||||
{
|
||||
Mutex::Lock _l(_inbox_m);
|
||||
_inbox.push(std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> >(from,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN>(data,len)));
|
||||
}
|
||||
_waitCond.signal();
|
||||
}
|
||||
|
||||
virtual bool send(const InetAddress &to,bool tcp,bool autoConnectTcp,const void *msg,unsigned int msglen);
|
||||
virtual void poll(unsigned long timeout,void (*handler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),void *arg);
|
||||
virtual void whack();
|
||||
|
||||
virtual void closeTcpSockets();
|
||||
|
||||
private:
|
||||
// These are set by SimNet after object creation
|
||||
SimNet *_sn;
|
||||
InetAddress _address;
|
||||
|
||||
SharedPtr<Socket> _mySocket;
|
||||
TransferStats _totals;
|
||||
|
||||
std::queue< std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> > > _inbox;
|
||||
Mutex _inbox_m;
|
||||
|
||||
std::map< InetAddress,TransferStats > _stats;
|
||||
Mutex _stats_m;
|
||||
|
||||
Condition _waitCond;
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue