/******************************************************************************* * libretroshare/src/dht: p3bitdht.h * * * * libretroshare: retroshare core library * * * * Copyright 2009-2010 by Robert Fernie * * * * This program is free software: you can redistribute it and/or modify * * it under the terms of the GNU Lesser 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 Lesser General Public License for more details. * * * * You should have received a copy of the GNU Lesser General Public License * * along with this program. If not, see . * * * *******************************************************************************/ #ifndef MRK_P3_BITDHT_H #define MRK_P3_BITDHT_H #include "pqi/pqiassist.h" #include "pqi/p3cfgmgr.h" #include "retroshare/rsdht.h" #include #include #include "pqi/pqinetwork.h" #include "pqi/pqimonitor.h" #include "util/rsthreads.h" #include "udp/udpstack.h" #include "udp/udpbitdht.h" #include "bitdht/bdiface.h" #include "dht/connectstatebox.h" class DhtPeerDetails { public: DhtPeerDetails(); RsDhtPeerType mPeerType; bdId mDhtId; RsPeerId mRsId; /* direct from the DHT! */ RsDhtPeerDht mDhtState; rstime_t mDhtUpdateTS; /* internal state */ PeerConnectStateBox mConnectLogic; /* Actual Connection Status */ RsDhtPeerConnectState mPeerConnectState; std::string mPeerConnectMsg; uint32_t mPeerConnectMode; bdId mPeerConnectPeerId; bdId mPeerConnectProxyId; struct sockaddr_in mPeerConnectAddr; uint32_t mPeerConnectPoint; rstime_t mPeerConnectUdpTS; rstime_t mPeerConnectTS; rstime_t mPeerConnectClosedTS; bool mExclusiveProxyLock; /* keeping the PeerCbMsg, as we will need it for debugging */ /* don't think this data is ever used for decisions??? */ /* Connection Request Status */ std::string mPeerReqStatusMsg; RsDhtPeerRequest mPeerReqState; uint32_t mPeerReqMode; bdId mPeerReqProxyId; rstime_t mPeerReqTS; /* Callback Info */ std::string mPeerCbMsg; uint32_t mPeerCbMode; uint32_t mPeerCbPoint; bdId mPeerCbProxyId; bdId mPeerCbDestId; rstime_t mPeerCbTS; }; #define PEERNET_ACTION_TYPE_CONNECT 1 #define PEERNET_ACTION_TYPE_AUTHORISE 2 #define PEERNET_ACTION_TYPE_START 3 #define PEERNET_ACTION_TYPE_RESTARTREQ 4 #define PEERNET_ACTION_TYPE_KILLREQ 5 #define PEERNET_ACTION_TYPE_TCPATTEMPT 6 class PeerAction { public: uint32_t mType; bdId mSrcId; bdId mProxyId; bdId mDestId; uint32_t mMode; uint32_t mPoint; uint32_t mAnswer; uint32_t mDelayOrBandwidth; }; /****** * Adding the ability to install alternative Handler * for monitoring/controlling Relay Connections outside of p3bitdht. ***/ class p3BitDhtRelayHandler { public: int (*mInstallRelay)(const bdId *srcId, const bdId *destId, uint32_t mode, uint32_t &bandwidth); int (*mLogFailedConnection)(const bdId *srcId, const bdId *destId, uint32_t mode, uint32_t errcode); }; class UdpRelayReceiver; #ifdef RS_USE_DHT_STUNNER class UdpStunner; #endif // RS_USE_DHT_STUNNER class p3NetMgr; class p3BitDht: public p3Config, public pqiNetAssistConnect, public RsDht { public: p3BitDht(const RsPeerId& id, pqiConnectCb *cb, p3NetMgr *nm, UdpStack *udpstack, std::string bootstrapfile, std::string bootstrapfilebak, const std::string &filteredipfile); virtual ~p3BitDht(); /*********************************************************************************************** ********** External RsDHT Interface (defined in libretroshare/src/retroshare/rsdht.h) ********* ************************************************************************************************/ virtual uint32_t getNetState(uint32_t type); virtual int getDhtPeers(int lvl, std::list &peers); virtual int getNetPeerList(std::list &peerIds); virtual int getNetPeerStatus(const RsPeerId& peerId, RsDhtNetPeer &status); virtual int getRelayEnds(std::list &relayEnds); virtual int getRelayProxies(std::list &relayProxies); //virtual int getNetFailedPeer(std::string peerId, PeerStatus &status); virtual std::string getUdpAddressString(); virtual void getDhtRates(float &read, float &write); virtual void getRelayRates(float &read, float &write, float &relay); virtual bool getOwnDhtId(std::string &ownDhtId); /*********************************************************************************************** ********** External RsDHT Interface (defined in libretroshare/src/retroshare/rsdht.h) ********* ************************************************************************************************/ #ifdef RS_USE_DHT_STUNNER void setupConnectBits(UdpStunner *dhtStunner, UdpStunner *proxyStunner, UdpRelayReceiver *relay); #else // RS_USE_DHT_STUNNER void setupConnectBits(UdpRelayReceiver *relay); #endif // RS_USE_DHT_STUNNER void setupPeerSharer(pqiNetAssistPeerShare *sharer); void modifyNodesPerBucket(uint16_t count); void start(); /* starts up the bitdht thread */ /* pqiNetAssist - external interface functions */ virtual int tick(); virtual void enable(bool on); virtual void shutdown(); /* blocking call */ virtual void restart(); virtual bool getEnabled(); virtual bool getActive(); virtual bool getNetworkStats(uint32_t &netsize, uint32_t &localnetsize); /* pqiNetAssistConnect - external interface functions */ /* add / remove peers */ virtual bool findPeer(const RsPeerId& id); virtual bool dropPeer(const RsPeerId& id); virtual int addBadPeer(const struct sockaddr_storage &addr, uint32_t reason, uint32_t flags, uint32_t age); virtual int addKnownPeer(const RsPeerId &pid, const struct sockaddr_storage &addr, uint32_t flags); //virtual int addFriend(const std::string pid); //virtual int addFriendOfFriend(const std::string pid); //virtual int addOther(const std::string pid); /* feedback on success failure of Connections */ virtual void ConnectionFeedback(const RsPeerId& pid, int state); /* extract current peer status */ virtual bool getPeerStatus(const RsPeerId& id, struct sockaddr_storage &laddr, struct sockaddr_storage &raddr, uint32_t &type, uint32_t &mode); virtual bool getExternalInterface(struct sockaddr_storage &raddr, uint32_t &mode); virtual bool isAddressBanned(const struct sockaddr_storage& raddr) ; virtual void getListOfBannedIps(std::list &lst) ; virtual bool setAttachMode(bool on); /* notifyPeer/setExtInterface/Bootstrap/Stun * hould all be removed from NetAssist? */ /* pqiNetAssistConnect - external interface functions */ /*********************************************************************************************** ****************************** Connections (p3bitdht_peernet.cc) ****************************** ************************************************************************************************/ /* Feedback from RS Upper Layers */ //virtual void ConnectionFeedback(std::string pid, int state); /* Callback functions - from bitdht */ int NodeCallback(const bdId *id, uint32_t peerflags); int PeerCallback(const bdId *id, uint32_t status); int ValueCallback(const bdNodeId *id, std::string key, uint32_t status); int ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId *destId, uint32_t mode, uint32_t point, uint32_t param, uint32_t cbtype, uint32_t errcode); int InfoCallback(const bdId *id, uint32_t type, uint32_t flags, std::string info); int OnlinePeerCallback_locked(const bdId *id, uint32_t status, DhtPeerDetails *dpd); int UnreachablePeerCallback_locked(const bdId *id, uint32_t status, DhtPeerDetails *dpd); //int tick(); int minuteTick(); int doActions(); int checkProxyAllowed(const bdId *srcId, const bdId *destId, int mode, uint32_t &bandwidth); int checkConnectionAllowed(const bdId *peerId, int mode); void initiateConnection(const bdId *srcId, const bdId *proxyId, const bdId *destId, uint32_t mode, uint32_t loc, uint32_t delayOrBandwidth); int installRelayConnection(const bdId *srcId, const bdId *destId, uint32_t &bandwidth); int removeRelayConnection(const bdId *srcId, const bdId *destId); void monitorConnections(); void ConnectCallout(const RsPeerId &peerId, struct sockaddr_in addr, uint32_t connectMode); void ConnectCalloutTCPAttempt(const RsPeerId &peerId, struct sockaddr_in addr); void ConnectCalloutDirectOrProxy(const RsPeerId &peerId, struct sockaddr_in raddr, uint32_t connectFlags, uint32_t delay); void ConnectCalloutRelay(const RsPeerId &peerId, struct sockaddr_in srcaddr, struct sockaddr_in proxyaddr, struct sockaddr_in destaddr, uint32_t connectMode, uint32_t bandwidth); void Feedback_Connected(const RsPeerId& pid); void Feedback_ConnectionFailed(const RsPeerId& pid); void Feedback_ConnectionClosed(const RsPeerId& pid); void UdpConnectionFailed_locked(DhtPeerDetails *dpd); void ReleaseProxyExclusiveMode_locked(DhtPeerDetails *dpd, bool addrChgLikely); /*** RELAY HANDLER CODE ***/ void installRelayHandler(p3BitDhtRelayHandler *); UdpRelayReceiver *getRelayReceiver(); int RelayHandler_InstallRelayConnection(const bdId *srcId, const bdId *destId, uint32_t mode, uint32_t &bandwidth); int RelayHandler_LogFailedProxyAttempt(const bdId *srcId, const bdId *destId, uint32_t mode, uint32_t errcode); /*********************************************************************************************** ******************** Relay Config Stuff (TEMP - MOSTLY, in p3bitdht_relay.cc) ***************** ********** External RsDHT Interface (defined in libretroshare/src/retroshare/rsdht.h) ********* ************************************************************************************************/ // Interface for controlling Relays & DHT Relay Mode virtual int getRelayServerList(std::list &ids); virtual int addRelayServer(std::string ids); virtual int removeRelayServer(std::string ids); virtual RsDhtRelayMode getRelayMode(); virtual int setRelayMode(RsDhtRelayMode mode); virtual int getRelayAllowance(RsDhtRelayClass classIdx, uint32_t &count, uint32_t &bandwidth); virtual int setRelayAllowance(RsDhtRelayClass classIdx, uint32_t count, uint32_t bandwidth); private: // Relay Handling Code / Variables (Mutex Protected). int setupRelayDefaults(); int pushRelayServers(); std::list mRelayServerList; RsDhtRelayMode mRelayMode; protected: /*****************************************************************/ /*********************** p3config ******************************/ /* Key Functions to be overloaded for Full Configuration */ virtual RsSerialiser *setupSerialiser(); virtual bool saveList(bool &cleanup, std::list&); virtual void saveDone(); virtual bool loadList(std::list& load); /*****************************************************************/ // DATA RATES: Variables (Mutex Protected). private: void updateDataRates(); void clearDataRates(); float mRelayReadRate; float mRelayWriteRate; float mRelayRelayRate; float mDhtReadRate; float mDhtWriteRate; rstime_t mLastDataRateUpdate; /*********************************************************************************************** ************************** Internal Accounting (p3bitdht_peers.cc) **************************** ************************************************************************************************/ public: int removePeer(const RsPeerId& pid); // Can be used externally too. int calculateNodeId(const RsPeerId& pid, bdNodeId *id); int addKnownNode(const bdId *id, uint32_t flags); private: DhtPeerDetails *addInternalPeer_locked(const RsPeerId& pid, RsDhtPeerType type); int removeInternalPeer_locked(const RsPeerId& pid); DhtPeerDetails *findInternalDhtPeer_locked(const bdNodeId *id, RsDhtPeerType type); DhtPeerDetails *findInternalRsPeer_locked(const RsPeerId &pid); bool havePeerTranslation_locked(const RsPeerId &pid); int lookupNodeId_locked(const RsPeerId& pid, bdNodeId *id); int lookupRsId_locked(const bdNodeId *id, RsPeerId &pid); int storeTranslation_locked(const RsPeerId& pid); int removeTranslation_locked(const RsPeerId& pid); UdpBitDht *mUdpBitDht; /* has own mutex, is static except for creation/destruction */ #ifdef RS_USE_DHT_STUNNER UdpStunner *mDhtStunner; UdpStunner *mProxyStunner; #endif // RS_USE_DHT_STUNNER UdpRelayReceiver *mRelay; p3NetMgr *mNetMgr; pqiNetAssistPeerShare *mPeerSharer; bdDhtFunctions *mDhtFns; RsMutex dhtMtx; p3BitDhtRelayHandler *mRelayHandler; RsPeerId mOwnRsId; bdNodeId mOwnDhtId; rstime_t mMinuteTS; /* translation maps */ std::map mTransToNodeId; std::map mTransToRsId; std::map mPeers; std::map mFailedPeers; /* Connection Action Queue */ std::list mActions; }; #endif /* MRK_P3_BITDHT_H */