mirror of
https://github.com/dashpay/dash.git
synced 2024-12-26 12:32:48 +01:00
Rework addnode behaviour
* Use CNode::addeName to track whether a connection to a name is already open * A new connection to a previously-connected by-name addednode is only opened when the previous one closes (even if the name starts resolving to something else) * At most one connection is opened per addednode (even if the name resolves to multiple) * Unify the code between ThreadOpenAddedNodeConnections and getaddednodeinfo * Information about open connections is always returned, and the dns argument becomes a dummy * An IP address and inbound/outbound is only reported for the (at most 1) open connection
This commit is contained in:
parent
be9711e597
commit
1111b80df8
113
src/net.cpp
113
src/net.cpp
@ -1616,6 +1616,58 @@ void ThreadOpenConnections()
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::vector<AddedNodeInfo> GetAddedNodeInfo()
|
||||||
|
{
|
||||||
|
std::vector<AddedNodeInfo> ret;
|
||||||
|
|
||||||
|
std::list<std::string> lAddresses(0);
|
||||||
|
{
|
||||||
|
LOCK(cs_vAddedNodes);
|
||||||
|
ret.reserve(vAddedNodes.size());
|
||||||
|
BOOST_FOREACH(const std::string& strAddNode, vAddedNodes)
|
||||||
|
lAddresses.push_back(strAddNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// Build a map of all already connected addresses (by IP:port and by name) to inbound/outbound and resolved CService
|
||||||
|
std::map<CService, bool> mapConnected;
|
||||||
|
std::map<std::string, std::pair<bool, CService>> mapConnectedByName;
|
||||||
|
{
|
||||||
|
LOCK(cs_vNodes);
|
||||||
|
for (const CNode* pnode : vNodes) {
|
||||||
|
if (pnode->addr.IsValid()) {
|
||||||
|
mapConnected[pnode->addr] = pnode->fInbound;
|
||||||
|
}
|
||||||
|
if (!pnode->addrName.empty()) {
|
||||||
|
mapConnectedByName[pnode->addrName] = std::make_pair(pnode->fInbound, static_cast<const CService&>(pnode->addr));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOST_FOREACH(const std::string& strAddNode, lAddresses) {
|
||||||
|
CService service(strAddNode, Params().GetDefaultPort());
|
||||||
|
if (service.IsValid()) {
|
||||||
|
// strAddNode is an IP:port
|
||||||
|
auto it = mapConnected.find(service);
|
||||||
|
if (it != mapConnected.end()) {
|
||||||
|
ret.push_back(AddedNodeInfo{strAddNode, service, true, it->second});
|
||||||
|
} else {
|
||||||
|
ret.push_back(AddedNodeInfo{strAddNode, CService(), false, false});
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// strAddNode is a name
|
||||||
|
auto it = mapConnectedByName.find(strAddNode);
|
||||||
|
if (it != mapConnectedByName.end()) {
|
||||||
|
ret.push_back(AddedNodeInfo{strAddNode, it->second.second, true, it->second.first});
|
||||||
|
} else {
|
||||||
|
ret.push_back(AddedNodeInfo{strAddNode, CService(), false, false});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
void ThreadOpenAddedConnections()
|
void ThreadOpenAddedConnections()
|
||||||
{
|
{
|
||||||
{
|
{
|
||||||
@ -1623,61 +1675,20 @@ void ThreadOpenAddedConnections()
|
|||||||
vAddedNodes = mapMultiArgs["-addnode"];
|
vAddedNodes = mapMultiArgs["-addnode"];
|
||||||
}
|
}
|
||||||
|
|
||||||
if (HaveNameProxy()) {
|
|
||||||
while(true) {
|
|
||||||
std::list<std::string> lAddresses(0);
|
|
||||||
{
|
|
||||||
LOCK(cs_vAddedNodes);
|
|
||||||
BOOST_FOREACH(const std::string& strAddNode, vAddedNodes)
|
|
||||||
lAddresses.push_back(strAddNode);
|
|
||||||
}
|
|
||||||
BOOST_FOREACH(const std::string& strAddNode, lAddresses) {
|
|
||||||
CAddress addr;
|
|
||||||
CSemaphoreGrant grant(*semOutbound);
|
|
||||||
OpenNetworkConnection(addr, false, &grant, strAddNode.c_str());
|
|
||||||
MilliSleep(500);
|
|
||||||
}
|
|
||||||
MilliSleep(120000); // Retry every 2 minutes
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (unsigned int i = 0; true; i++)
|
for (unsigned int i = 0; true; i++)
|
||||||
{
|
{
|
||||||
std::list<std::string> lAddresses(0);
|
std::vector<AddedNodeInfo> vInfo = GetAddedNodeInfo();
|
||||||
{
|
for (const AddedNodeInfo& info : vInfo) {
|
||||||
LOCK(cs_vAddedNodes);
|
if (!info.fConnected) {
|
||||||
BOOST_FOREACH(const std::string& strAddNode, vAddedNodes)
|
CSemaphoreGrant grant(*semOutbound);
|
||||||
lAddresses.push_back(strAddNode);
|
// If strAddedNode is an IP/port, decode it immediately, so
|
||||||
|
// OpenNetworkConnection can detect existing connections to that IP/port.
|
||||||
|
CService service(info.strAddedNode, Params().GetDefaultPort());
|
||||||
|
OpenNetworkConnection(CAddress(service, NODE_NONE), false, &grant, info.strAddedNode.c_str(), false);
|
||||||
|
MilliSleep(500);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
std::list<std::vector<CService> > lservAddressesToAdd(0);
|
|
||||||
BOOST_FOREACH(const std::string& strAddNode, lAddresses) {
|
|
||||||
std::vector<CService> vservNode(0);
|
|
||||||
if(Lookup(strAddNode.c_str(), vservNode, Params().GetDefaultPort(), fNameLookup, 0))
|
|
||||||
lservAddressesToAdd.push_back(vservNode);
|
|
||||||
}
|
|
||||||
// Attempt to connect to each IP for each addnode entry until at least one is successful per addnode entry
|
|
||||||
// (keeping in mind that addnode entries can have many IPs if fNameLookup)
|
|
||||||
{
|
|
||||||
LOCK(cs_vNodes);
|
|
||||||
BOOST_FOREACH(CNode* pnode, vNodes)
|
|
||||||
for (std::list<std::vector<CService> >::iterator it = lservAddressesToAdd.begin(); it != lservAddressesToAdd.end(); it++)
|
|
||||||
BOOST_FOREACH(const CService& addrNode, *(it))
|
|
||||||
if (pnode->addr == addrNode)
|
|
||||||
{
|
|
||||||
it = lservAddressesToAdd.erase(it);
|
|
||||||
it--;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
BOOST_FOREACH(std::vector<CService>& vserv, lservAddressesToAdd)
|
|
||||||
{
|
|
||||||
CSemaphoreGrant grant(*semOutbound);
|
|
||||||
/* We want -addnode to work even for nodes that don't provide all
|
|
||||||
* wanted services, so pass in nServices=NODE_NONE to CAddress. */
|
|
||||||
OpenNetworkConnection(CAddress(vserv[i % vserv.size()], NODE_NONE), false, &grant);
|
|
||||||
MilliSleep(500);
|
|
||||||
}
|
|
||||||
MilliSleep(120000); // Retry every 2 minutes
|
MilliSleep(120000); // Retry every 2 minutes
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
10
src/net.h
10
src/net.h
@ -818,4 +818,14 @@ public:
|
|||||||
/** Return a timestamp in the future (in microseconds) for exponentially distributed events. */
|
/** Return a timestamp in the future (in microseconds) for exponentially distributed events. */
|
||||||
int64_t PoissonNextSend(int64_t nNow, int average_interval_seconds);
|
int64_t PoissonNextSend(int64_t nNow, int average_interval_seconds);
|
||||||
|
|
||||||
|
struct AddedNodeInfo
|
||||||
|
{
|
||||||
|
std::string strAddedNode;
|
||||||
|
CService resolvedAddress;
|
||||||
|
bool fConnected;
|
||||||
|
bool fInbound;
|
||||||
|
};
|
||||||
|
|
||||||
|
std::vector<AddedNodeInfo> GetAddedNodeInfo();
|
||||||
|
|
||||||
#endif // BITCOIN_NET_H
|
#endif // BITCOIN_NET_H
|
||||||
|
@ -271,25 +271,22 @@ UniValue getaddednodeinfo(const UniValue& params, bool fHelp)
|
|||||||
{
|
{
|
||||||
if (fHelp || params.size() < 1 || params.size() > 2)
|
if (fHelp || params.size() < 1 || params.size() > 2)
|
||||||
throw runtime_error(
|
throw runtime_error(
|
||||||
"getaddednodeinfo dns ( \"node\" )\n"
|
"getaddednodeinfo dummy ( \"node\" )\n"
|
||||||
"\nReturns information about the given added node, or all added nodes\n"
|
"\nReturns information about the given added node, or all added nodes\n"
|
||||||
"(note that onetry addnodes are not listed here)\n"
|
"(note that onetry addnodes are not listed here)\n"
|
||||||
"If dns is false, only a list of added nodes will be provided,\n"
|
|
||||||
"otherwise connected information will also be available.\n"
|
|
||||||
"\nArguments:\n"
|
"\nArguments:\n"
|
||||||
"1. dns (boolean, required) If false, only a list of added nodes will be provided, otherwise connected information will also be available.\n"
|
"1. dummy (boolean, required) Kept for historical purposes but ignored\n"
|
||||||
"2. \"node\" (string, optional) If provided, return information about this specific node, otherwise all nodes are returned.\n"
|
"2. \"node\" (string, optional) If provided, return information about this specific node, otherwise all nodes are returned.\n"
|
||||||
"\nResult:\n"
|
"\nResult:\n"
|
||||||
"[\n"
|
"[\n"
|
||||||
" {\n"
|
" {\n"
|
||||||
" \"addednode\" : \"192.168.0.201\", (string) The node ip address\n"
|
" \"addednode\" : \"192.168.0.201\", (string) The node ip address or name (as provided to addnode)\n"
|
||||||
" \"connected\" : true|false, (boolean) If connected\n"
|
" \"connected\" : true|false, (boolean) If connected\n"
|
||||||
" \"addresses\" : [\n"
|
" \"addresses\" : [ (list of objects) Only when connected = true\n"
|
||||||
" {\n"
|
" {\n"
|
||||||
" \"address\" : \"192.168.0.201:8333\", (string) The bitcoin server host and port\n"
|
" \"address\" : \"192.168.0.201:8333\", (string) The bitcoin server IP and port we're connected to\n"
|
||||||
" \"connected\" : \"outbound\" (string) connection, inbound or outbound\n"
|
" \"connected\" : \"outbound\" (string) connection, inbound or outbound\n"
|
||||||
" }\n"
|
" }\n"
|
||||||
" ,...\n"
|
|
||||||
" ]\n"
|
" ]\n"
|
||||||
" }\n"
|
" }\n"
|
||||||
" ,...\n"
|
" ,...\n"
|
||||||
@ -300,83 +297,35 @@ UniValue getaddednodeinfo(const UniValue& params, bool fHelp)
|
|||||||
+ HelpExampleRpc("getaddednodeinfo", "true, \"192.168.0.201\"")
|
+ HelpExampleRpc("getaddednodeinfo", "true, \"192.168.0.201\"")
|
||||||
);
|
);
|
||||||
|
|
||||||
bool fDns = params[0].get_bool();
|
std::vector<AddedNodeInfo> vInfo = GetAddedNodeInfo();
|
||||||
|
|
||||||
list<string> laddedNodes(0);
|
if (params.size() == 2) {
|
||||||
if (params.size() == 1)
|
bool found = false;
|
||||||
{
|
for (const AddedNodeInfo& info : vInfo) {
|
||||||
LOCK(cs_vAddedNodes);
|
if (info.strAddedNode == params[1].get_str()) {
|
||||||
BOOST_FOREACH(const std::string& strAddNode, vAddedNodes)
|
vInfo.assign(1, info);
|
||||||
laddedNodes.push_back(strAddNode);
|
found = true;
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
string strNode = params[1].get_str();
|
|
||||||
LOCK(cs_vAddedNodes);
|
|
||||||
BOOST_FOREACH(const std::string& strAddNode, vAddedNodes) {
|
|
||||||
if (strAddNode == strNode)
|
|
||||||
{
|
|
||||||
laddedNodes.push_back(strAddNode);
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (laddedNodes.size() == 0)
|
if (!found) {
|
||||||
throw JSONRPCError(RPC_CLIENT_NODE_NOT_ADDED, "Error: Node has not been added.");
|
throw JSONRPCError(RPC_CLIENT_NODE_NOT_ADDED, "Error: Node has not been added.");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
UniValue ret(UniValue::VARR);
|
UniValue ret(UniValue::VARR);
|
||||||
if (!fDns)
|
|
||||||
{
|
|
||||||
BOOST_FOREACH (const std::string& strAddNode, laddedNodes) {
|
|
||||||
UniValue obj(UniValue::VOBJ);
|
|
||||||
obj.push_back(Pair("addednode", strAddNode));
|
|
||||||
ret.push_back(obj);
|
|
||||||
}
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
list<pair<string, vector<CService> > > laddedAddreses(0);
|
for (const AddedNodeInfo& info : vInfo) {
|
||||||
BOOST_FOREACH(const std::string& strAddNode, laddedNodes) {
|
|
||||||
vector<CService> vservNode(0);
|
|
||||||
if(Lookup(strAddNode.c_str(), vservNode, Params().GetDefaultPort(), fNameLookup, 0))
|
|
||||||
laddedAddreses.push_back(make_pair(strAddNode, vservNode));
|
|
||||||
else
|
|
||||||
{
|
|
||||||
UniValue obj(UniValue::VOBJ);
|
|
||||||
obj.push_back(Pair("addednode", strAddNode));
|
|
||||||
obj.push_back(Pair("connected", false));
|
|
||||||
UniValue addresses(UniValue::VARR);
|
|
||||||
obj.push_back(Pair("addresses", addresses));
|
|
||||||
ret.push_back(obj);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
LOCK(cs_vNodes);
|
|
||||||
for (list<pair<string, vector<CService> > >::iterator it = laddedAddreses.begin(); it != laddedAddreses.end(); it++)
|
|
||||||
{
|
|
||||||
UniValue obj(UniValue::VOBJ);
|
UniValue obj(UniValue::VOBJ);
|
||||||
obj.push_back(Pair("addednode", it->first));
|
obj.push_back(Pair("addednode", info.strAddedNode));
|
||||||
|
obj.push_back(Pair("connected", info.fConnected));
|
||||||
UniValue addresses(UniValue::VARR);
|
UniValue addresses(UniValue::VARR);
|
||||||
bool fConnected = false;
|
if (info.fConnected) {
|
||||||
BOOST_FOREACH(const CService& addrNode, it->second) {
|
UniValue address(UniValue::VOBJ);
|
||||||
bool fFound = false;
|
address.push_back(Pair("address", info.resolvedAddress.ToString()));
|
||||||
UniValue node(UniValue::VOBJ);
|
address.push_back(Pair("connected", info.fInbound ? "inbound" : "outbound"));
|
||||||
node.push_back(Pair("address", addrNode.ToString()));
|
addresses.push_back(address);
|
||||||
BOOST_FOREACH(CNode* pnode, vNodes) {
|
|
||||||
if (pnode->addr == addrNode)
|
|
||||||
{
|
|
||||||
fFound = true;
|
|
||||||
fConnected = true;
|
|
||||||
node.push_back(Pair("connected", pnode->fInbound ? "inbound" : "outbound"));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!fFound)
|
|
||||||
node.push_back(Pair("connected", "false"));
|
|
||||||
addresses.push_back(node);
|
|
||||||
}
|
}
|
||||||
obj.push_back(Pair("connected", fConnected));
|
|
||||||
obj.push_back(Pair("addresses", addresses));
|
obj.push_back(Pair("addresses", addresses));
|
||||||
ret.push_back(obj);
|
ret.push_back(obj);
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user