Added Exclusive Proxy Lock for DETERMINISTIC SYM Nats.

* tweaked time periods as these type of connections take longer.
 * added NatHoleType() to ConnectStateBox logic - as its required for ProxyPort Determination.
 * added ConnectStateBox::getNetState() so we can work out if in ExclusiveNat mode.
 * added ExclusiveLock checks at Connection Initiation / Proxy Connection Auth & Connection Ends.
 * added Simulation of Symmetric / Exclusive Nat to UdpStunner.
 * exported ExclusiveProxyLock for GUI display.



git-svn-id: http://svn.code.sf.net/p/retroshare/code/branches/v0.5-netupgrade@4457 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
drbob 2011-07-17 13:48:37 +00:00
parent a5c77d02e7
commit 181ad06e33
9 changed files with 338 additions and 78 deletions

View File

@ -33,31 +33,35 @@
#define TESTING_PERIODS 1 #define TESTING_PERIODS 1
/* Have made the PROXY Attempts + MAX_TIME much larger,
* have have potential for this to take a while.
*/
#ifdef TESTING_PERIODS #ifdef TESTING_PERIODS
#define FAILED_WAIT_TIME (1800) // 5 minutes. #define FAILED_WAIT_TIME (1800) // 5 minutes.
#define TCP_WAIT_TIME (10) // 1/6 minutes. #define TCP_WAIT_TIME (10) // 1/6 minutes.
#define DIRECT_MAX_WAIT_TIME (30) // 1/6 minutes. #define DIRECT_MAX_WAIT_TIME (30) // 1/6 minutes.
#define PROXY_MAX_WAIT_TIME (30) // 1/6 minutes. #define PROXY_MAX_WAIT_TIME (120) // 1/6 minutes.
#define RELAY_MAX_WAIT_TIME (30) // 1/6 minutes. #define RELAY_MAX_WAIT_TIME (30) // 1/6 minutes.
#define REVERSE_WAIT_TIME (30) // 1/2 minutes. #define REVERSE_WAIT_TIME (30) // 1/2 minutes.
#define MAX_DIRECT_ATTEMPTS (3) #define MAX_DIRECT_ATTEMPTS (3)
#define MAX_PROXY_ATTEMPTS (3) #define MAX_PROXY_ATTEMPTS (10)
#define MAX_RELAY_ATTEMPTS (3) #define MAX_RELAY_ATTEMPTS (3)
#define MAX_DIRECT_FAILED_ATTEMPTS (1) #define MAX_DIRECT_FAILED_ATTEMPTS (1)
#define MAX_PROXY_FAILED_ATTEMPTS (1) #define MAX_PROXY_FAILED_ATTEMPTS (2)
#define MAX_RELAY_FAILED_ATTEMPTS (1) #define MAX_RELAY_FAILED_ATTEMPTS (1)
#else #else
#define FAILED_WAIT_TIME (1800) // 30 minutes. #define FAILED_WAIT_TIME (1800) // 30 minutes.
#define TCP_WAIT_TIME (60) // 1 minutes. #define TCP_WAIT_TIME (60) // 1 minutes.
#define DIRECT_MAX_WAIT_TIME (60) // 1 minutes. #define DIRECT_MAX_WAIT_TIME (60) // 1 minutes.
#define PROXY_MAX_WAIT_TIME (60) // 1 minutes. #define PROXY_MAX_WAIT_TIME (120) // 1 minutes.
#define RELAY_MAX_WAIT_TIME (60) // 1 minutes. #define RELAY_MAX_WAIT_TIME (60) // 1 minutes.
#define REVERSE_WAIT_TIME (300) // 5 minutes. #define REVERSE_WAIT_TIME (300) // 5 minutes.
#define MAX_DIRECT_ATTEMPTS (10) #define MAX_DIRECT_ATTEMPTS (10)
#define MAX_PROXY_ATTEMPTS (10) #define MAX_PROXY_ATTEMPTS (20)
#define MAX_RELAY_ATTEMPTS (10) #define MAX_RELAY_ATTEMPTS (10)
#define MAX_DIRECT_FAILED_ATTEMPTS (3) #define MAX_DIRECT_FAILED_ATTEMPTS (3)
@ -81,6 +85,11 @@ PeerConnectStateBox::PeerConnectStateBox()
} }
uint32_t PeerConnectStateBox::getNetState()
{
return mNetState;
}
std::string NetStateAsString(uint32_t netstate) std::string NetStateAsString(uint32_t netstate)
{ {
@ -230,14 +239,15 @@ std::string PeerConnectStateBox::connectState() const
std::ostringstream out; std::ostringstream out;
time_t now = time(NULL); time_t now = time(NULL);
out << str << "(" << mNoAttempts << "/" << mNoFailedAttempts << ") for " << now - mStateTS << " secs"; out << str << "(" << mNoAttempts << "/" << mNoFailedAttempts << ") for " << now - mStateTS << " secs";
out << " LA: " << mAttemptLength;
if ( (mState == CSB_CONNECTED) || (mState == CSB_DIRECT_ATTEMPT) || if ( (mState == CSB_CONNECTED) || (mState == CSB_DIRECT_ATTEMPT) ||
(mState == CSB_PROXY_ATTEMPT) || (mState == CSB_RELAY_ATTEMPT) ) (mState == CSB_PROXY_ATTEMPT) || (mState == CSB_RELAY_ATTEMPT) ||
(mState == CSB_FAILED_WAIT) )
{ {
// nothing here... as we are not dependent on the timer. out << " Last Attempt: " << mAttemptLength;
} }
else else
{ {
out << " LA: " << mAttemptLength;
out << " NextAttempt: " << mNextAttemptTS - now; out << " NextAttempt: " << mNextAttemptTS - now;
} }
@ -245,7 +255,7 @@ std::string PeerConnectStateBox::connectState() const
} }
uint32_t convertNetStateToInternal(uint32_t netmode, uint32_t nattype) uint32_t convertNetStateToInternal(uint32_t netmode, uint32_t nathole, uint32_t nattype)
{ {
uint32_t connNet = CSB_NETSTATE_UNKNOWN; uint32_t connNet = CSB_NETSTATE_UNKNOWN;
@ -253,6 +263,10 @@ uint32_t convertNetStateToInternal(uint32_t netmode, uint32_t nattype)
{ {
connNet = CSB_NETSTATE_FORWARD; connNet = CSB_NETSTATE_FORWARD;
} }
else if ((nathole != RSNET_NATHOLE_UNKNOWN) && (nathole != RSNET_NATHOLE_NONE))
{
connNet = CSB_NETSTATE_FORWARD;
}
else if (netmode == RSNET_NETWORK_BEHINDNAT) else if (netmode == RSNET_NETWORK_BEHINDNAT)
{ {
if ((nattype == RSNET_NATTYPE_RESTRICTED_CONE) || if ((nattype == RSNET_NATTYPE_RESTRICTED_CONE) ||
@ -281,15 +295,15 @@ bool shouldUseProxyPortInternal(uint32_t netstate)
return true; return true;
} }
bool PeerConnectStateBox::shouldUseProxyPort(uint32_t netmode, uint32_t nattype) bool PeerConnectStateBox::shouldUseProxyPort(uint32_t netmode, uint32_t nathole, uint32_t nattype)
{ {
uint32_t netstate = convertNetStateToInternal(netmode, nattype); uint32_t netstate = convertNetStateToInternal(netmode, nathole, nattype);
return shouldUseProxyPortInternal(netstate); return shouldUseProxyPortInternal(netstate);
} }
uint32_t PeerConnectStateBox::connectCb(uint32_t cbtype, uint32_t netmode, uint32_t nattype) uint32_t PeerConnectStateBox::connectCb(uint32_t cbtype, uint32_t netmode, uint32_t nathole, uint32_t nattype)
{ {
uint32_t netstate = convertNetStateToInternal(netmode, nattype); uint32_t netstate = convertNetStateToInternal(netmode, nathole, nattype);
std::cerr << "PeerConnectStateBox::connectCb("; std::cerr << "PeerConnectStateBox::connectCb(";
if (cbtype == CSB_CONNECT_DIRECT) if (cbtype == CSB_CONNECT_DIRECT)

View File

@ -84,11 +84,12 @@ class PeerConnectStateBox
public: public:
PeerConnectStateBox(); PeerConnectStateBox();
uint32_t connectCb(uint32_t cbtype, uint32_t netmode, uint32_t nattype); uint32_t connectCb(uint32_t cbtype, uint32_t netmode, uint32_t nathole, uint32_t nattype);
uint32_t updateCb(uint32_t updateType); uint32_t updateCb(uint32_t updateType);
bool shouldUseProxyPort(uint32_t netmode, uint32_t nattype); bool shouldUseProxyPort(uint32_t netmode, uint32_t nathole, uint32_t nattype);
uint32_t getNetState();
std::string connectState() const; std::string connectState() const;
std::string mPeerId; std::string mPeerId;

View File

@ -71,6 +71,8 @@ class DhtPeerDetails
time_t mPeerConnectTS; time_t mPeerConnectTS;
time_t mPeerConnectClosedTS; time_t mPeerConnectClosedTS;
bool mExclusiveProxyLock;
/* keeping the PeerCbMsg, as we will need it for debugging */ /* keeping the PeerCbMsg, as we will need it for debugging */
/* don't think this data is ever used for decisions??? */ /* don't think this data is ever used for decisions??? */
@ -109,6 +111,7 @@ class PeerAction
uint32_t mMode; uint32_t mMode;
uint32_t mPoint; uint32_t mPoint;
uint32_t mAnswer; uint32_t mAnswer;
uint32_t mDelayOrBandwidth;
}; };
@ -226,9 +229,12 @@ void ConnectCalloutRelay(const std::string &peerId, struct sockaddr_in srcaddr,
void Feedback_Connected(std::string pid); void Feedback_Connected(std::string pid);
void Feedback_ConnectionFailed(std::string pid); void Feedback_ConnectionFailed(std::string pid);
void UdpConnectionFailed_locked(DhtPeerDetails *dpd);
void Feedback_ConnectionClosed(std::string pid); void Feedback_ConnectionClosed(std::string pid);
void UdpConnectionFailed_locked(DhtPeerDetails *dpd);
void ReleaseProxyExclusiveMode_locked(DhtPeerDetails *dpd);
/*********************************************************************************************** /***********************************************************************************************
************************** Internal Accounting (p3bitdht_peers.cc) **************************** ************************** Internal Accounting (p3bitdht_peers.cc) ****************************

View File

@ -185,6 +185,8 @@ void convertDhtPeerDetailsToRsDhtNetPeer(RsDhtNetPeer &status, const DhtPeerDeta
status.mPeerReqState = details.mPeerReqState; status.mPeerReqState = details.mPeerReqState;
status.mExclusiveProxyLock = details.mExclusiveProxyLock;
status.mPeerConnectState = details.mPeerConnectState; status.mPeerConnectState = details.mPeerConnectState;
switch(details.mPeerConnectMode) switch(details.mPeerConnectMode)

View File

@ -313,7 +313,7 @@ int p3BitDht::OnlinePeerCallback_locked(const bdId *id, uint32_t status, DhtPeer
/* work out network state */ /* work out network state */
uint32_t connectFlags = dpd->mConnectLogic.connectCb(CSB_CONNECT_DIRECT, uint32_t connectFlags = dpd->mConnectLogic.connectCb(CSB_CONNECT_DIRECT,
mNetMgr->getNetworkMode(), mNetMgr->getNatTypeMode()); mNetMgr->getNetworkMode(), mNetMgr->getNatHoleMode(), mNetMgr->getNatTypeMode());
bool useProxyPort = (connectFlags & CSB_ACTION_PROXY_PORT); bool useProxyPort = (connectFlags & CSB_ACTION_PROXY_PORT);
switch(connectFlags & CSB_ACTION_MASK_MODE) switch(connectFlags & CSB_ACTION_MASK_MODE)
@ -414,7 +414,7 @@ int p3BitDht::UnreachablePeerCallback_locked(const bdId *id, uint32_t status, Dh
/* work out network state */ /* work out network state */
uint32_t connectFlags = dpd->mConnectLogic.connectCb(CSB_CONNECT_UNREACHABLE, uint32_t connectFlags = dpd->mConnectLogic.connectCb(CSB_CONNECT_UNREACHABLE,
mNetMgr->getNetworkMode(), mNetMgr->getNatTypeMode()); mNetMgr->getNetworkMode(), mNetMgr->getNatHoleMode(), mNetMgr->getNatTypeMode());
bool useProxyPort = (connectFlags & CSB_ACTION_PROXY_PORT); bool useProxyPort = (connectFlags & CSB_ACTION_PROXY_PORT);
switch(connectFlags & CSB_ACTION_MASK_MODE) switch(connectFlags & CSB_ACTION_MASK_MODE)
@ -580,6 +580,7 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
ca.mDestId = *destId; ca.mDestId = *destId;
ca.mPoint = point; ca.mPoint = point;
ca.mAnswer = connectionAllowed; ca.mAnswer = connectionAllowed;
ca.mDelayOrBandwidth = 0;
mActions.push_back(ca); mActions.push_back(ca);
} }
@ -676,6 +677,7 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
ca.mProxyId = *proxyId; ca.mProxyId = *proxyId;
ca.mSrcId = *srcId; ca.mSrcId = *srcId;
ca.mDestId = *destId; ca.mDestId = *destId;
ca.mDelayOrBandwidth = 0;
/* Check Proxy ExtAddress Status (but only if connection is Allowed) */ /* Check Proxy ExtAddress Status (but only if connection is Allowed) */
if ((connectionAllowed == BITDHT_CONNECT_ANSWER_OKAY) && if ((connectionAllowed == BITDHT_CONNECT_ANSWER_OKAY) &&
@ -690,6 +692,7 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
bool connectOk = false; bool connectOk = false;
bool proxyPort = false; bool proxyPort = false;
bool exclusivePort = false;
std::cerr << "dhtConnectionCallback(): Proxy... deciding which port to use."; std::cerr << "dhtConnectionCallback(): Proxy... deciding which port to use.";
std::cerr << std::endl; std::cerr << std::endl;
@ -697,7 +700,7 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
if (dpd) if (dpd)
{ {
proxyPort = dpd->mConnectLogic.shouldUseProxyPort( proxyPort = dpd->mConnectLogic.shouldUseProxyPort(
mNetMgr->getNetworkMode(), mNetMgr->getNatTypeMode()); mNetMgr->getNetworkMode(), mNetMgr->getNatHoleMode(), mNetMgr->getNatTypeMode());
dpd->mConnectLogic.storeProxyPortChoice(0, proxyPort); dpd->mConnectLogic.storeProxyPortChoice(0, proxyPort);
@ -705,6 +708,17 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
std::cerr << " UseProxyPort? " << proxyPort; std::cerr << " UseProxyPort? " << proxyPort;
std::cerr << std::endl; std::cerr << std::endl;
if (proxyPort)
{
exclusivePort = (CSB_NETSTATE_EXCLUSIVENAT == dpd->mConnectLogic.getNetState());
}
if (exclusivePort)
{
std::cerr << "dhtConnectionCallback: we Require Exclusive Proxy Port for connection";
std::cerr << std::endl;
}
connectOk = true; connectOk = true;
} }
else else
@ -725,7 +739,7 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
{ {
std::cerr << "dhtConnectionCallback() Proxy Connection Attempt to: "; std::cerr << "dhtConnectionCallback() Proxy Connection Attempt to: ";
bdStdPrintId(std::cerr, &(peerId)); bdStdPrintId(std::cerr, &(peerId));
std::cerr << " is OkGo as we have Stable Own External Proxy Address"; std::cerr << " is Ok as we have Stable Own External Proxy Address";
std::cerr << std::endl; std::cerr << std::endl;
if (point == BD_PROXY_CONNECTION_END_POINT) if (point == BD_PROXY_CONNECTION_END_POINT)
@ -739,13 +753,61 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
std::cerr << std::endl; std::cerr << std::endl;
} }
/* check if we require exclusive use of the proxy port */
if (exclusivePort)
{
std::cerr << "dhtConnectionCallback: Attempting to Grab ExclusiveLock of UdpStunner";
std::cerr << std::endl;
int delay = mProxyStunner->grabExclusiveMode();
if (delay > 0)
{
/* great we got it! */
ca.mDelayOrBandwidth = delay;
DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(peerId.id), RSDHT_PEERTYPE_FRIEND);
if (dpd)
{
dpd->mExclusiveProxyLock = true;
std::cerr << "dhtConnectionCallback: Success at grabbing ExclusiveLock of UdpStunner";
std::cerr << std::endl;
}
else
{
std::cerr << "PeerAction: Connect Proxy: ERROR Cannot find PeerStatus";
std::cerr << std::endl;
connectionAllowed = BITDHT_CONNECT_ERROR_TEMPUNAVAIL;
mProxyStunner->releaseExclusiveMode();
}
}
else
{
/* failed to get exclusive mode - must wait */
connectionAllowed = BITDHT_CONNECT_ERROR_TEMPUNAVAIL;
std::cerr << "dhtConnectionCallback: Failed to Grab ExclusiveLock, Returning TEMPUNAVAIL";
std::cerr << std::endl;
}
}
} }
else else
{ {
connectionAllowed = BITDHT_CONNECT_ERROR_UNREACHABLE; if (exclusivePort)
std::cerr << "dhtConnectionCallback() Proxy Connection"; {
std::cerr << " is Discarded, as Own External Proxy Address is Not Stable!"; connectionAllowed = BITDHT_CONNECT_ERROR_TEMPUNAVAIL;
std::cerr << std::endl; std::cerr << "dhtConnectionCallback() Proxy Connection";
std::cerr << " is Discarded, as External Proxy Address is Not Stable! (EXCLUSIVE MODE)";
std::cerr << std::endl;
}
else
{
connectionAllowed = BITDHT_CONNECT_ERROR_UNREACHABLE;
std::cerr << "dhtConnectionCallback() Proxy Connection";
std::cerr << " is Discarded, as Own External Proxy Address is Not Stable!";
std::cerr << std::endl;
}
} }
} }
else else
@ -853,6 +915,8 @@ int p3BitDht::ConnectCallback(const bdId *srcId, const bdId *proxyId, const bdId
{ {
if (errcode) if (errcode)
{ {
ReleaseProxyExclusiveMode_locked(dpd);
dpd->mPeerReqStatusMsg = "STOPPED: "; dpd->mPeerReqStatusMsg = "STOPPED: ";
dpd->mPeerReqStatusMsg += decodeConnectionError(errcode); dpd->mPeerReqStatusMsg += decodeConnectionError(errcode);
dpd->mPeerReqState = RSDHT_PEERREQ_STOPPED; dpd->mPeerReqState = RSDHT_PEERREQ_STOPPED;
@ -1032,23 +1096,22 @@ int p3BitDht::doActions()
std::cerr << " mode: " << action.mMode; std::cerr << " mode: " << action.mMode;
std::cerr << std::endl; std::cerr << std::endl;
bool connectionRequested = false; bool doConnectionRequest = false;
bool connectionReqFailed = false; bool connectionReqFailed = false;
bool grabbedExclusivePort = false;
// Parameters that will be used for the Connect Request.
struct sockaddr_in connAddr; // We zero this address. (DHT Layer handles most cases)
sockaddr_clear(&connAddr);
uint32_t connStart = 1; /* > 0 indicates GO (number indicates required startup delay) */
uint32_t connDelay = 0;
uint32_t failReason = CSB_UPDATE_MODE_UNAVAILABLE;
if ((action.mMode == BITDHT_CONNECT_MODE_DIRECT) || if ((action.mMode == BITDHT_CONNECT_MODE_DIRECT) ||
(action.mMode == BITDHT_CONNECT_MODE_RELAY)) (action.mMode == BITDHT_CONNECT_MODE_RELAY))
{ {
struct sockaddr_in laddr; // We zero this address. The DHT layer should be able to handle this! doConnectionRequest = true;
sockaddr_clear(&laddr);
uint32_t start = 1;
if (mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start))
{
connectionRequested = true;
}
else
{
connectionReqFailed = true;
}
} }
else if (action.mMode == BITDHT_CONNECT_MODE_PROXY) else if (action.mMode == BITDHT_CONNECT_MODE_PROXY)
{ {
@ -1056,6 +1119,7 @@ int p3BitDht::doActions()
uint8_t extStable = 0; uint8_t extStable = 0;
sockaddr_clear(&extaddr); sockaddr_clear(&extaddr);
bool proxyPort = true; bool proxyPort = true;
bool exclusivePort = false;
bool connectOk = false; bool connectOk = false;
std::cerr << "PeerAction: Proxy... deciding which port to use."; std::cerr << "PeerAction: Proxy... deciding which port to use.";
@ -1068,8 +1132,12 @@ int p3BitDht::doActions()
{ {
connectOk = true; connectOk = true;
proxyPort = dpd->mConnectLogic.getProxyPortChoice(); proxyPort = dpd->mConnectLogic.getProxyPortChoice();
if (proxyPort)
{
std::cerr << "PeerAction: Using ProxyPort. NetState indicates Exclusive Mode";
std::cerr << std::endl;
exclusivePort = (CSB_NETSTATE_EXCLUSIVENAT == dpd->mConnectLogic.getNetState());
}
} }
else else
{ {
@ -1092,24 +1160,54 @@ int p3BitDht::doActions()
std::cerr << " is OkGo as we have Stable Own External Proxy Address"; std::cerr << " is OkGo as we have Stable Own External Proxy Address";
std::cerr << std::endl; std::cerr << std::endl;
int start = 1; /* check if we require exclusive use of the proxy port */
if (mUdpBitDht->ConnectionRequest(&extaddr, &(action.mDestId.id), action.mMode, start)) if (exclusivePort)
{ {
connectionRequested = true; int delay = mProxyStunner->grabExclusiveMode();
if (delay > 0)
{
/* great we got it! */
connAddr = extaddr;
connDelay = delay;
doConnectionRequest = true;
grabbedExclusivePort = true;
}
else
{
/* failed to get exclusive mode - must wait */
connectionReqFailed = true;
failReason = CSB_UPDATE_RETRY_ATTEMPT;
}
} }
else else
{ {
connectionReqFailed = true; /* stable and non-exclusive - go for it */
connAddr = extaddr;
connStart = 1;
doConnectionRequest = true;
} }
} }
else else
{ {
std::cerr << "PeerAction: ERROR Proxy Connection Attempt to: "; std::cerr << "PeerAction: WARNING Proxy Connection Attempt to: ";
bdStdPrintId(std::cerr, &(action.mDestId)); bdStdPrintId(std::cerr, &(action.mDestId));
std::cerr << " is Discarded, as Own External Proxy Address is Not Stable!"; std::cerr << " is Discarded, as Own External Proxy Address is Not Stable!";
std::cerr << std::endl; std::cerr << std::endl;
connectionReqFailed = true; connectionReqFailed = true;
if (exclusivePort)
{
failReason = CSB_UPDATE_RETRY_ATTEMPT;
std::cerr << "PeerAction: As Exclusive Mode, Port Will stabilise => RETRY";
std::cerr << std::endl;
}
else
{
failReason = CSB_UPDATE_MODE_UNAVAILABLE;
std::cerr << "PeerAction: Not Exclusive Mode, => MODE UNAVAILABLE";
std::cerr << std::endl;
}
} }
} }
else else
@ -1119,38 +1217,48 @@ int p3BitDht::doActions()
std::cerr << " is Discarded, as Failed to get Own External Proxy Address."; std::cerr << " is Discarded, as Failed to get Own External Proxy Address.";
std::cerr << std::endl; std::cerr << std::endl;
connectionReqFailed = true;
failReason = CSB_UPDATE_RETRY_ATTEMPT;
}
}
if (doConnectionRequest)
{
// XXX TO DO.
//if (mUdpBitDht->ConnectionRequest(&connAddr, &(action.mDestId.id), action.mMode, connStart, connDelay))
if (mUdpBitDht->ConnectionRequest(&connAddr, &(action.mDestId.id), action.mMode, connStart))
{
RsStackMutex stack(dhtMtx); /********** LOCKED MUTEX ***************/ RsStackMutex stack(dhtMtx); /********** LOCKED MUTEX ***************/
std::cerr << "PeerAction: Connection Attempt to: ";
bdStdPrintId(std::cerr, &(action.mDestId));
std::cerr << " has gone ahead";
std::cerr << std::endl;
DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(action.mDestId.id), RSDHT_PEERTYPE_FRIEND); DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(action.mDestId.id), RSDHT_PEERTYPE_FRIEND);
if (dpd) if (dpd)
{ {
dpd->mConnectLogic.updateCb(CSB_UPDATE_FAILED_ATTEMPT); dpd->mPeerReqStatusMsg = "Connect Request";
dpd->mPeerReqState = RSDHT_PEERREQ_RUNNING;
dpd->mPeerReqMode = action.mMode;
dpd->mPeerReqTS = now;
if (grabbedExclusivePort)
{
dpd->mExclusiveProxyLock = true;
}
}
else
{
std::cerr << "PeerAction: Connect ERROR Cannot find PeerStatus";
std::cerr << std::endl;
} }
}
}
if (connectionRequested)
{
RsStackMutex stack(dhtMtx); /********** LOCKED MUTEX ***************/
std::cerr << "PeerAction: Connection Attempt to: ";
bdStdPrintId(std::cerr, &(action.mDestId));
std::cerr << " has gone ahead";
std::cerr << std::endl;
DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(action.mDestId.id), RSDHT_PEERTYPE_FRIEND);
if (dpd)
{
dpd->mPeerReqStatusMsg = "Connect Request";
dpd->mPeerReqState = RSDHT_PEERREQ_RUNNING;
dpd->mPeerReqMode = action.mMode;
dpd->mPeerReqTS = now;
} }
else else
{ {
std::cerr << "PeerAction: Connect ERROR Cannot find PeerStatus"; connectionReqFailed = true;
std::cerr << std::endl; failReason = CSB_UPDATE_MODE_UNAVAILABLE;
} }
} }
@ -1161,11 +1269,16 @@ int p3BitDht::doActions()
std::cerr << " is Discarded, as Mode is Unavailable"; std::cerr << " is Discarded, as Mode is Unavailable";
std::cerr << std::endl; std::cerr << std::endl;
if (grabbedExclusivePort)
{
mProxyStunner->releaseExclusiveMode();
}
RsStackMutex stack(dhtMtx); /********** LOCKED MUTEX ***************/ RsStackMutex stack(dhtMtx); /********** LOCKED MUTEX ***************/
DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(action.mDestId.id), RSDHT_PEERTYPE_FRIEND); DhtPeerDetails *dpd = findInternalDhtPeer_locked(&(action.mDestId.id), RSDHT_PEERTYPE_FRIEND);
if (dpd) if (dpd)
{ {
dpd->mConnectLogic.updateCb(CSB_UPDATE_MODE_UNAVAILABLE); dpd->mConnectLogic.updateCb(failReason);
dpd->mPeerReqStatusMsg = "Req Mode Unavailable"; dpd->mPeerReqStatusMsg = "Req Mode Unavailable";
dpd->mPeerReqState = RSDHT_PEERREQ_STOPPED; dpd->mPeerReqState = RSDHT_PEERREQ_STOPPED;
@ -1185,8 +1298,13 @@ int p3BitDht::doActions()
std::cerr << " and "; std::cerr << " and ";
bdStdPrintId(std::cerr, &(action.mDestId)); bdStdPrintId(std::cerr, &(action.mDestId));
std::cerr << " mode: " << action.mMode; std::cerr << " mode: " << action.mMode;
std::cerr << " delay/bandwidth: " << action.mDelayOrBandwidth;
std::cerr << std::endl; std::cerr << std::endl;
// XXX TODO
//mUdpBitDht->ConnectionAuth(&(action.mSrcId), &(action.mProxyId), &(action.mDestId),
// action.mMode, action.mPoint, action.mAnswer, action.mDelayOrBandwidth);
mUdpBitDht->ConnectionAuth(&(action.mSrcId), &(action.mProxyId), &(action.mDestId), mUdpBitDht->ConnectionAuth(&(action.mSrcId), &(action.mProxyId), &(action.mDestId),
action.mMode, action.mPoint, action.mAnswer); action.mMode, action.mPoint, action.mAnswer);
@ -1238,8 +1356,13 @@ int p3BitDht::doActions()
std::cerr << " and "; std::cerr << " and ";
bdStdPrintId(std::cerr, &(action.mDestId)); bdStdPrintId(std::cerr, &(action.mDestId));
std::cerr << " mode: " << action.mMode; std::cerr << " mode: " << action.mMode;
std::cerr << " delay/bandwidth: " << action.mDelayOrBandwidth;
std::cerr << std::endl; std::cerr << std::endl;
// XXX TO DO.
//initiateConnection(&(action.mSrcId), &(action.mProxyId), &(action.mDestId),
// action.mMode, action.mPoint, action.mAnswer, action.mDelayOrBandwidth);
initiateConnection(&(action.mSrcId), &(action.mProxyId), &(action.mDestId), initiateConnection(&(action.mSrcId), &(action.mProxyId), &(action.mDestId),
action.mMode, action.mPoint, action.mAnswer); action.mMode, action.mPoint, action.mAnswer);
} }
@ -1256,6 +1379,9 @@ int p3BitDht::doActions()
struct sockaddr_in laddr; struct sockaddr_in laddr;
sockaddr_clear(&laddr); sockaddr_clear(&laddr);
uint32_t start = 1; uint32_t start = 1;
// XXX TO DO.
//mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start, 0);
mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start); mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start);
} }
break; break;
@ -1271,6 +1397,9 @@ int p3BitDht::doActions()
struct sockaddr_in laddr; struct sockaddr_in laddr;
sockaddr_clear(&laddr); sockaddr_clear(&laddr);
uint32_t start = 0; uint32_t start = 0;
// XXX TO DO.
//mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start, 0);
mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start); mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start);
} }
break; break;
@ -1283,11 +1412,6 @@ int p3BitDht::doActions()
std::cerr << " mode: " << action.mMode; std::cerr << " mode: " << action.mMode;
std::cerr << std::endl; std::cerr << std::endl;
struct sockaddr_in laddr;
sockaddr_clear(&laddr);
uint32_t start = 0;
mUdpBitDht->ConnectionRequest(&laddr, &(action.mDestId.id), action.mMode, start);
std::string peerRsId; std::string peerRsId;
bool foundPeerId = false; bool foundPeerId = false;
{ {
@ -1883,6 +2007,8 @@ void p3BitDht::Feedback_Connected(std::string pid)
std::cerr << "p3BitDht::monitorConnections() ERROR Request not active, can't stop"; std::cerr << "p3BitDht::monitorConnections() ERROR Request not active, can't stop";
std::cerr << std::endl; std::cerr << std::endl;
} }
ReleaseProxyExclusiveMode_locked(dpd);
} }
void p3BitDht::Feedback_ConnectionFailed(std::string pid) void p3BitDht::Feedback_ConnectionFailed(std::string pid)
@ -1983,10 +2109,50 @@ void p3BitDht::UdpConnectionFailed_locked(DhtPeerDetails *dpd)
std::cerr << "p3BitDht::monitorConnections() ERROR Request not active, can't stop"; std::cerr << "p3BitDht::monitorConnections() ERROR Request not active, can't stop";
std::cerr << std::endl; std::cerr << std::endl;
} }
ReleaseProxyExclusiveMode_locked(dpd);
} }
void p3BitDht::ReleaseProxyExclusiveMode_locked(DhtPeerDetails *dpd)
{
#ifdef DEBUG_BITDHT_COMMON
std::cerr << "p3BitDht::ReleaseProxyExclusiveMode_locked()";
bdStdPrintNodeId(std::cerr, &(dpd->mDhtId.id));
std::cerr << std::endl;
#endif
if (dpd->mExclusiveProxyLock)
{
if (mProxyStunner->releaseExclusiveMode())
{
dpd->mExclusiveProxyLock = false;
std::cerr << "p3BitDht::ReleaseProxyExclusiveMode_locked() Lock released by Connection to peer: ";
bdStdPrintNodeId(std::cerr, &(dpd->mDhtId.id));
std::cerr << std::endl;
}
else
{
dpd->mExclusiveProxyLock = false;
std::cerr << "p3BitDht::ReleaseProxyExclusiveMode_locked() ERROR ProxyStunner is not Locked";
bdStdPrintNodeId(std::cerr, &(dpd->mDhtId.id));
std::cerr << std::endl;
}
}
else
{
#ifdef DEBUG_BITDHT_COMMON
std::cerr << "p3BitDht::ReleaseProxyExclusiveMode_locked() Don't Have a Lock";
std::cerr << std::endl;
#endif
}
}
void p3BitDht::ConnectionFeedback(std::string pid, int mode) void p3BitDht::ConnectionFeedback(std::string pid, int mode)
{ {

View File

@ -90,6 +90,8 @@ bool p3BitDht::findPeer(std::string pid)
//dpd->mPeerReqProxyId; //dpd->mPeerReqProxyId;
dpd->mPeerReqTS = time(NULL); dpd->mPeerReqTS = time(NULL);
dpd->mExclusiveProxyLock = false;
dpd->mPeerCbMsg = "No CB Yet"; dpd->mPeerCbMsg = "No CB Yet";
dpd->mPeerCbMode = 0; dpd->mPeerCbMode = 0;
dpd->mPeerCbPoint = 0; dpd->mPeerCbPoint = 0;

View File

@ -105,6 +105,8 @@ class RsDhtNetPeer
// connect mode // connect mode
uint32_t mPeerConnectMode; uint32_t mPeerConnectMode;
bool mExclusiveProxyLock;
std::string mPeerConnectProxyId; std::string mPeerConnectProxyId;
// Req Status. // Req Status.

View File

@ -23,10 +23,11 @@
* *
*/ */
#include "udpstunner.h" #include "tcponudp/udpstunner.h"
#include <iostream> #include <iostream>
#include <sstream> #include <sstream>
#include "util/rsrandom.h"
#include "util/rsprint.h" #include "util/rsprint.h"
static const int STUN_TTL = 64; static const int STUN_TTL = 64;
@ -56,6 +57,9 @@ UdpStunner::UdpStunner(UdpPublisher *pub)
{ {
#ifdef UDPSTUN_ALLOW_LOCALNET #ifdef UDPSTUN_ALLOW_LOCALNET
mAcceptLocalNet = false; mAcceptLocalNet = false;
mSimExclusiveNat = false;
mSimSymmetricNat = false;
mSimUnstableExt = false;
#endif #endif
@ -81,10 +85,29 @@ void UdpStunner::SetAcceptLocalNet()
mAcceptLocalNet = true; mAcceptLocalNet = true;
} }
// For Local Testing Only (Releases should have the #define disabled)
void UdpStunner::SimExclusiveNat()
{
RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/
mSimExclusiveNat = true;
mSimUnstableExt = true;
}
void UdpStunner::SimSymmetricNat()
{
RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/
mSimSymmetricNat = true;
mSimUnstableExt = true;
}
#endif #endif
int UdpStunner::setExclusiveMode() /* returns seconds since last send/recv */ int UdpStunner::grabExclusiveMode() /* returns seconds since last send/recv */
{ {
RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/ RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/
@ -127,7 +150,7 @@ int UdpStunner::setExclusiveMode() /* returns seconds since last send/recv */
return commsage; return commsage;
} }
int UdpStunner::cancelExclusiveMode() int UdpStunner::releaseExclusiveMode()
{ {
RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/ RsStackMutex stack(stunMtx); /********** LOCK MUTEX *********/
@ -143,6 +166,21 @@ int UdpStunner::cancelExclusiveMode()
time_t now = time(NULL); time_t now = time(NULL);
mExclusiveMode = false; mExclusiveMode = false;
#ifdef UDPSTUN_ALLOW_LOCALNET
/* if we are simulating an exclusive NAT, then immediately after we release - it'll become unstable.
* This is even harser than reality... so better test.
*
* In reality, it will only become unstable if we have tried a UDP connection.
*/
if (mSimExclusiveNat)
{
mSimUnstableExt = true;
}
#endif
#ifdef DEBUG_UDP_STUNNER_FILTER #ifdef DEBUG_UDP_STUNNER_FILTER
std::cerr << "UdpStunner::cancelExclusiveMode() Canceled. Was in ExclusiveMode for: " << now - mExclusiveModeTS; std::cerr << "UdpStunner::cancelExclusiveMode() Canceled. Was in ExclusiveMode for: " << now - mExclusiveModeTS;
std::cerr << " secs"; std::cerr << " secs";
@ -336,6 +374,7 @@ bool UdpStunner::externalAddr(struct sockaddr_in &external, uint8_t &stable)
std::cerr << std::endl; std::cerr << std::endl;
#endif #endif
return true; return true;
} }
#ifdef DEBUG_UDP_STUNNER #ifdef DEBUG_UDP_STUNNER
@ -780,6 +819,23 @@ bool UdpStunner::locked_recvdStun(const struct sockaddr_in &remote, const str
std::cerr << out.str() << std::endl; std::cerr << out.str() << std::endl;
#endif #endif
#ifdef UDPSTUN_ALLOW_LOCALNET
struct sockaddr_in fakeExtaddr = extaddr;
if (mSimUnstableExt)
{
std::cerr << "UdpStunner::locked_recvdStun() TEST SIM UNSTABLE EXT: Forcing Port to be wrong to sim an ExclusiveNat";
std::cerr << std::endl;
#define UNSTABLE_PORT_RANGE 100
fakeExtaddr.sin_port = htons(ntohs(fakeExtaddr.sin_port) - (UNSTABLE_PORT_RANGE / 2) + RSRandom::random_u32() % UNSTABLE_PORT_RANGE);
if (!mSimSymmetricNat)
{
mSimUnstableExt = false;
}
}
#endif
bool found = true; bool found = true;
std::list<TouStunPeer>::iterator it; std::list<TouStunPeer>::iterator it;
for(it = mStunList.begin(); it != mStunList.end(); it++) for(it = mStunList.begin(); it != mStunList.end(); it++)
@ -788,7 +844,11 @@ bool UdpStunner::locked_recvdStun(const struct sockaddr_in &remote, const str
(remote.sin_port == it->remote.sin_port)) (remote.sin_port == it->remote.sin_port))
{ {
it->failCount = 0; it->failCount = 0;
#ifdef UDPSTUN_ALLOW_LOCALNET
it->eaddr = fakeExtaddr;
#else
it->eaddr = extaddr; it->eaddr = extaddr;
#endif
it->response = true; it->response = true;
found = true; found = true;

View File

@ -84,10 +84,12 @@ virtual ~UdpStunner() { return; }
#ifdef UDPSTUN_ALLOW_LOCALNET #ifdef UDPSTUN_ALLOW_LOCALNET
// For Local Testing Mode. // For Local Testing Mode.
void SetAcceptLocalNet(); void SetAcceptLocalNet();
void SimExclusiveNat();
void SimSymmetricNat();
#endif #endif
int setExclusiveMode(); /* returns seconds since last send/recv */ int grabExclusiveMode(); /* returns seconds since last send/recv */
int cancelExclusiveMode(); int releaseExclusiveMode();
void setTargetStunPeriod(int32_t sec_per_stun); void setTargetStunPeriod(int32_t sec_per_stun);
@ -142,6 +144,10 @@ bool locked_checkExternalAddress();
#ifdef UDPSTUN_ALLOW_LOCALNET #ifdef UDPSTUN_ALLOW_LOCALNET
// For Local Testing Mode. // For Local Testing Mode.
bool mAcceptLocalNet; bool mAcceptLocalNet;
bool mSimUnstableExt;
bool mSimExclusiveNat;
bool mSimSymmetricNat;
#endif #endif
bool mPassiveStunMode; bool mPassiveStunMode;
@ -150,6 +156,7 @@ bool locked_checkExternalAddress();
bool mExclusiveMode; /* when this is switched on, the stunner stays silent (and extAddr is maintained) */ bool mExclusiveMode; /* when this is switched on, the stunner stays silent (and extAddr is maintained) */
time_t mExclusiveModeTS; time_t mExclusiveModeTS;
}; };
/* generic stun functions */ /* generic stun functions */