RetroShare/libretroshare/src/dht/p3bitdht.h

/*
 * libretroshare/src/dht: p3bitdht.h
 *
 * BitDht interface for RetroShare.
 *
 * Copyright 2009-2010 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 MRK_P3_BITDHT_H
#define MRK_P3_BITDHT_H

#include "pqi/pqiassist.h"
#include "retroshare/rsdht.h"

#include <string>
#include <map>
#include "pqi/pqinetwork.h"
#include "pqi/pqimonitor.h"
#include "util/rsthreads.h"

#include "udp/udpstack.h"
#include "udp/udpbitdht.h"
#include "bitdht/bdiface.h"


class DhtPeerDetails
{
	public:

	bdId    mDhtId;
	std::string mRsId;

};

class UdpRelayReceiver;
class UdpStunner;


class p3BitDht: public pqiNetAssistConnect, public RsDht
{
	public:
	p3BitDht(std::string id, pqiConnectCb *cb, 
		UdpStack *udpstack, std::string bootstrapfile);


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<RsDhtPeer> &peers);
virtual int      getNetPeerList(std::list<std::string> &peerIds);
virtual int      getNetPeerStatus(std::string peerId, RsDhtNetPeer &status);

virtual int     getRelayEnds(std::list<RsDhtRelayEnd> &relayEnds);
virtual int     getRelayProxies(std::list<RsDhtRelayProxy> &relayProxies);

//virtual int      getNetFailedPeer(std::string peerId, PeerStatus &status);

/***********************************************************************************************
 ********** External RsDHT Interface (defined in libretroshare/src/retroshare/rsdht.h) *********
************************************************************************************************/


	void	setupConnectBits(UdpStunner *dhtStunner, UdpStunner *proxyStunner, UdpRelayReceiver  *relay);




void	start(); /* starts up the bitdht thread */

	/* pqiNetAssist - external interface functions */
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(std::string id);
virtual bool 	dropPeer(std::string id);

	/* extract current peer status */
virtual bool 	getPeerStatus(std::string id, 
			struct sockaddr_in &laddr, struct sockaddr_in &raddr, 
					uint32_t &type, uint32_t &mode);

virtual bool 	getExternalInterface(struct sockaddr_in &raddr, 
					uint32_t &mode);

	/* notifyPeer/setExtInterface/Bootstrap/Stun 
	 * hould all be removed from NetAssist?
	 */

	/* pqiNetAssistConnect - external interface functions */


	/* 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);

	private:
	/* translation stuff */
	int calculateNodeId(const std::string pid, bdNodeId *id);

	int lookupNodeId(const std::string pid, bdNodeId *id);
	int lookupRsId(const bdNodeId *id, std::string &pid);
	int storeTranslation(const std::string pid);
	int removeTranslation(const std::string pid);

	UdpBitDht *mUdpBitDht; /* has own mutex, is static except for creation/destruction */
	UdpStunner *mDhtStunner;
	UdpStunner *mProxyStunner;
	UdpRelayReceiver   *mRelay;

	RsMutex dhtMtx;
	/* translation maps */
        std::map<std::string, bdNodeId> mTransToNodeId;
        std::map<bdNodeId, std::string> mTransToRsId;

	std::map<std::string, DhtPeerDetails> mPeers;
};

#endif /* MRK_P3_BITDHT_H */