/* * ZeroTier One - Global Peer to Peer Ethernet * Copyright (C) 2011-2014 ZeroTier Networks LLC * * 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 . * * -- * * 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 #include #include #include #include #include #include "../node/Constants.hpp" #include "../node/EthernetTap.hpp" #include "../node/AtomicCounter.hpp" #include "../node/SharedPtr.hpp" #include "../node/Thread.hpp" #include "../node/Mutex.hpp" #include "../node/Condition.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 { friend class SharedPtr; public: struct TestFrame { TestFrame() : from(),to(),etherType(0),len(0) {} TestFrame(const MAC &f,const MAC &t,const void *d,unsigned int et,unsigned int l) : from(f), to(t), etherType(et), len(l) { memcpy(data,d,l); } MAC from; MAC to; unsigned int etherType; unsigned int len; char data[4096]; }; TestEthernetTap( TestEthernetTapFactory *parent, 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 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 &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; } // Get things that have been put() and empty queue inline void get(std::vector &v,bool clearQueue = true) { Mutex::Lock _l(_gq_m); v = _gq; if (clearQueue) _gq.clear(); } void threadMain() throw(); private: uint64_t _nwid; TestEthernetTapFactory *_parent; void (*_handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &); void *_arg; Thread _thread; std::string _dev; volatile bool _enabled; std::queue< TestFrame > _pq; Mutex _pq_m; Condition _pq_c; std::vector< TestFrame > _gq; Mutex _gq_m; AtomicCounter __refCount; }; } // namespace ZeroTier #endif