You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.4 KiB
124 lines
3.4 KiB
/* |
|
* ZeroTier One - Network Virtualization Everywhere |
|
* Copyright (C) 2011-2015 ZeroTier, Inc. |
|
* |
|
* This program is free software: you can redistribute it and/or modify |
|
* it under the terms of the GNU General Public License as published by |
|
* the Free Software Foundation, either version 3 of the License, or |
|
* (at your option) any later version. |
|
* |
|
* This program 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 General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU General Public License |
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
* |
|
* -- |
|
* |
|
* ZeroTier may be used and distributed under the terms of the GPLv3, which |
|
* are available at: http://www.gnu.org/licenses/gpl-3.0.html |
|
* |
|
* If you would like to embed ZeroTier into a commercial application or |
|
* redistribute it in a modified binary form, please contact ZeroTier Networks |
|
* LLC. Start here: http://www.zerotier.com/ |
|
*/ |
|
|
|
#ifndef ZT_TESTETHERNETTAP_HPP |
|
#define ZT_TESTETHERNETTAP_HPP |
|
|
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include <string.h> |
|
|
|
#include <string> |
|
|
|
#include "../node/Constants.hpp" |
|
#include "../node/EthernetTap.hpp" |
|
#include "../node/Thread.hpp" |
|
#include "../node/Mutex.hpp" |
|
|
|
#include "MTQ.hpp" |
|
|
|
namespace ZeroTier { |
|
|
|
class TestEthernetTapFactory; |
|
|
|
/** |
|
* Dummy Ethernet tap |
|
* |
|
* This tap device prints the contents of packets it receives on stdout |
|
* and also prints outgoing packets when they are injected. It does not |
|
* connect to any real tap or other interface. It's useful for running |
|
* test networks. |
|
*/ |
|
class TestEthernetTap : public EthernetTap |
|
{ |
|
public: |
|
struct TestFrame |
|
{ |
|
TestFrame() : from(),to(),timestamp(0),etherType(0),len(0) {} |
|
TestFrame(const MAC &f,const MAC &t,const void *d,unsigned int et,unsigned int l) : |
|
from(f), |
|
to(t), |
|
timestamp(Utils::now()), |
|
etherType(et), |
|
len(l) |
|
{ |
|
memcpy(data,d,l); |
|
} |
|
|
|
MAC from; |
|
MAC to; |
|
uint64_t timestamp; |
|
unsigned int etherType; |
|
unsigned int len; |
|
char data[4096]; |
|
}; |
|
|
|
TestEthernetTap( |
|
const MAC &mac, |
|
unsigned int mtu, |
|
unsigned int metric, |
|
uint64_t nwid, |
|
const char *desiredDevice, |
|
const char *friendlyName, |
|
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &), |
|
void *arg); |
|
|
|
virtual ~TestEthernetTap(); |
|
|
|
virtual void setEnabled(bool en); |
|
virtual bool enabled() const; |
|
virtual bool addIP(const InetAddress &ip); |
|
virtual bool removeIP(const InetAddress &ip); |
|
virtual std::set<InetAddress> ips() const; |
|
virtual void put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len); |
|
virtual std::string deviceName() const; |
|
virtual void setFriendlyName(const char *friendlyName); |
|
virtual bool updateMulticastGroups(std::set<MulticastGroup> &groups); |
|
virtual bool injectPacketFromHost(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len); |
|
|
|
inline uint64_t nwid() const { return _nwid; } |
|
inline bool getNextReceivedFrame(TestFrame &v,unsigned long timeout) { return _gq.pop(v,timeout); } |
|
|
|
void threadMain() |
|
throw(); |
|
|
|
private: |
|
uint64_t _nwid; |
|
|
|
void (*_handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &); |
|
void *_arg; |
|
Thread _thread; |
|
std::string _dev; |
|
volatile bool _enabled; |
|
|
|
MTQ<TestFrame> _pq; |
|
MTQ<TestFrame> _gq; |
|
}; |
|
|
|
} // namespace ZeroTier |
|
|
|
#endif
|
|
|