Merge branch 'keepnode' of https://github.com/TheBlueMatt/bitcoin
[novacoin.git] / src / net.cpp
1 // Copyright (c) 2009-2010 Satoshi Nakamoto
2 // Copyright (c) 2011 The Bitcoin developers
3 // Distributed under the MIT/X11 software license, see the accompanying
4 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
5
6 #include "headers.h"
7 #include "irc.h"
8 #include "db.h"
9 #include "net.h"
10 #include "init.h"
11 #include "strlcpy.h"
12
13 #ifdef WIN32
14 #include <string.h>
15 #endif
16
17 #ifdef USE_UPNP
18 #include <miniupnpc/miniwget.h>
19 #include <miniupnpc/miniupnpc.h>
20 #include <miniupnpc/upnpcommands.h>
21 #include <miniupnpc/upnperrors.h>
22 #endif
23
24 using namespace std;
25 using namespace boost;
26
27 static const int MAX_OUTBOUND_CONNECTIONS = 8;
28
29 void ThreadMessageHandler2(void* parg);
30 void ThreadSocketHandler2(void* parg);
31 void ThreadOpenConnections2(void* parg);
32 void ThreadOpenAddedConnections2(void* parg);
33 #ifdef USE_UPNP
34 void ThreadMapPort2(void* parg);
35 #endif
36 void ThreadDNSAddressSeed2(void* parg);
37 bool OpenNetworkConnection(const CAddress& addrConnect);
38
39
40
41
42
43 //
44 // Global state variables
45 //
46 bool fClient = false;
47 bool fAllowDNS = false;
48 uint64 nLocalServices = (fClient ? 0 : NODE_NETWORK);
49 CAddress addrLocalHost(CService("0.0.0.0", 0), nLocalServices);
50 static CNode* pnodeLocalHost = NULL;
51 uint64 nLocalHostNonce = 0;
52 array<int, 10> vnThreadsRunning;
53 static SOCKET hListenSocket = INVALID_SOCKET;
54
55 vector<CNode*> vNodes;
56 CCriticalSection cs_vNodes;
57 map<vector<unsigned char>, CAddress> mapAddresses;
58 CCriticalSection cs_mapAddresses;
59 map<CInv, CDataStream> mapRelay;
60 deque<pair<int64, CInv> > vRelayExpiration;
61 CCriticalSection cs_mapRelay;
62 map<CInv, int64> mapAlreadyAskedFor;
63
64
65 set<CNetAddr> setservAddNodeAddresses;
66 CCriticalSection cs_setservAddNodeAddresses;
67
68
69
70
71 unsigned short GetListenPort()
72 {
73     return (unsigned short)(GetArg("-port", GetDefaultPort()));
74 }
75
76 void CNode::PushGetBlocks(CBlockIndex* pindexBegin, uint256 hashEnd)
77 {
78     // Filter out duplicate requests
79     if (pindexBegin == pindexLastGetBlocksBegin && hashEnd == hashLastGetBlocksEnd)
80         return;
81     pindexLastGetBlocksBegin = pindexBegin;
82     hashLastGetBlocksEnd = hashEnd;
83
84     PushMessage("getblocks", CBlockLocator(pindexBegin), hashEnd);
85 }
86
87
88
89
90
91 bool GetMyExternalIP2(const CService& addrConnect, const char* pszGet, const char* pszKeyword, CNetAddr& ipRet)
92 {
93     SOCKET hSocket;
94     if (!ConnectSocket(addrConnect, hSocket))
95         return error("GetMyExternalIP() : connection to %s failed", addrConnect.ToString().c_str());
96
97     send(hSocket, pszGet, strlen(pszGet), MSG_NOSIGNAL);
98
99     string strLine;
100     while (RecvLine(hSocket, strLine))
101     {
102         if (strLine.empty()) // HTTP response is separated from headers by blank line
103         {
104             loop
105             {
106                 if (!RecvLine(hSocket, strLine))
107                 {
108                     closesocket(hSocket);
109                     return false;
110                 }
111                 if (pszKeyword == NULL)
112                     break;
113                 if (strLine.find(pszKeyword) != -1)
114                 {
115                     strLine = strLine.substr(strLine.find(pszKeyword) + strlen(pszKeyword));
116                     break;
117                 }
118             }
119             closesocket(hSocket);
120             if (strLine.find("<") != -1)
121                 strLine = strLine.substr(0, strLine.find("<"));
122             strLine = strLine.substr(strspn(strLine.c_str(), " \t\n\r"));
123             while (strLine.size() > 0 && isspace(strLine[strLine.size()-1]))
124                 strLine.resize(strLine.size()-1);
125             CService addr(strLine,0,true);
126             printf("GetMyExternalIP() received [%s] %s\n", strLine.c_str(), addr.ToString().c_str());
127             if (!addr.IsValid() || !addr.IsRoutable())
128                 return false;
129             ipRet.SetIP(addr);
130             return true;
131         }
132     }
133     closesocket(hSocket);
134     return error("GetMyExternalIP() : connection closed");
135 }
136
137 // We now get our external IP from the IRC server first and only use this as a backup
138 bool GetMyExternalIP(CNetAddr& ipRet)
139 {
140     CAddress addrConnect;
141     const char* pszGet;
142     const char* pszKeyword;
143
144     if (fUseProxy)
145         return false;
146
147     for (int nLookup = 0; nLookup <= 1; nLookup++)
148     for (int nHost = 1; nHost <= 2; nHost++)
149     {
150         // We should be phasing out our use of sites like these.  If we need
151         // replacements, we should ask for volunteers to put this simple
152         // php file on their webserver that prints the client IP:
153         //  <?php echo $_SERVER["REMOTE_ADDR"]; ?>
154         if (nHost == 1)
155         {
156             addrConnect = CAddress("91.198.22.70",80); // checkip.dyndns.org
157
158             if (nLookup == 1)
159             {
160                 CService addrIP("checkip.dyndns.org", 80, true);
161                 if (addrIP.IsValid())
162                     addrConnect = addrIP;
163             }
164
165             pszGet = "GET / HTTP/1.1\r\n"
166                      "Host: checkip.dyndns.org\r\n"
167                      "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)\r\n"
168                      "Connection: close\r\n"
169                      "\r\n";
170
171             pszKeyword = "Address:";
172         }
173         else if (nHost == 2)
174         {
175             addrConnect = CAddress("74.208.43.192", 80); // www.showmyip.com
176
177             if (nLookup == 1)
178             {
179                 CService addrIP("www.showmyip.com", 80, true);
180                 if (addrIP.IsValid())
181                     addrConnect = addrIP;
182             }
183
184             pszGet = "GET /simple/ HTTP/1.1\r\n"
185                      "Host: www.showmyip.com\r\n"
186                      "User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)\r\n"
187                      "Connection: close\r\n"
188                      "\r\n";
189
190             pszKeyword = NULL; // Returns just IP address
191         }
192
193         if (GetMyExternalIP2(addrConnect, pszGet, pszKeyword, ipRet))
194             return true;
195     }
196
197     return false;
198 }
199
200 void ThreadGetMyExternalIP(void* parg)
201 {
202     // Wait for IRC to get it first
203     if (!GetBoolArg("-noirc"))
204     {
205         for (int i = 0; i < 2 * 60; i++)
206         {
207             Sleep(1000);
208             if (fGotExternalIP || fShutdown)
209                 return;
210         }
211     }
212
213     // Fallback in case IRC fails to get it
214     if (GetMyExternalIP(addrLocalHost))
215     {
216         printf("GetMyExternalIP() returned %s\n", addrLocalHost.ToStringIP().c_str());
217         if (addrLocalHost.IsRoutable())
218         {
219             // If we already connected to a few before we had our IP, go back and addr them.
220             // setAddrKnown automatically filters any duplicate sends.
221             CAddress addr(addrLocalHost);
222             addr.nTime = GetAdjustedTime();
223             CRITICAL_BLOCK(cs_vNodes)
224                 BOOST_FOREACH(CNode* pnode, vNodes)
225                     pnode->PushAddress(addr);
226         }
227     }
228 }
229
230
231
232
233
234 bool AddAddress(CAddress addr, int64 nTimePenalty, CAddrDB *pAddrDB)
235 {
236     if (!addr.IsRoutable())
237         return false;
238     if ((CService)addr == (CService)addrLocalHost)
239         return false;
240     addr.nTime = max((int64)0, (int64)addr.nTime - nTimePenalty);
241     bool fUpdated = false;
242     bool fNew = false;
243     CAddress addrFound = addr;
244
245     CRITICAL_BLOCK(cs_mapAddresses)
246     {
247         map<vector<unsigned char>, CAddress>::iterator it = mapAddresses.find(addr.GetKey());
248         if (it == mapAddresses.end())
249         {
250             // New address
251             printf("AddAddress(%s)\n", addr.ToString().c_str());
252             mapAddresses.insert(make_pair(addr.GetKey(), addr));
253             fUpdated = true;
254             fNew = true;
255         }
256         else
257         {
258             addrFound = (*it).second;
259             if ((addrFound.nServices | addr.nServices) != addrFound.nServices)
260             {
261                 // Services have been added
262                 addrFound.nServices |= addr.nServices;
263                 fUpdated = true;
264             }
265             bool fCurrentlyOnline = (GetAdjustedTime() - addr.nTime < 24 * 60 * 60);
266             int64 nUpdateInterval = (fCurrentlyOnline ? 60 * 60 : 24 * 60 * 60);
267             if (addrFound.nTime < addr.nTime - nUpdateInterval)
268             {
269                 // Periodically update most recently seen time
270                 addrFound.nTime = addr.nTime;
271                 fUpdated = true;
272             }
273         }
274     }
275     // There is a nasty deadlock bug if this is done inside the cs_mapAddresses
276     // CRITICAL_BLOCK:
277     // Thread 1:  begin db transaction (locks inside-db-mutex)
278     //            then AddAddress (locks cs_mapAddresses)
279     // Thread 2:  AddAddress (locks cs_mapAddresses)
280     //             ... then db operation hangs waiting for inside-db-mutex
281     if (fUpdated)
282     {
283         if (pAddrDB)
284             pAddrDB->WriteAddress(addrFound);
285         else
286             CAddrDB().WriteAddress(addrFound);
287     }
288     return fNew;
289 }
290
291 void AddressCurrentlyConnected(const CService& addr)
292 {
293     CAddress *paddrFound = NULL;
294
295     CRITICAL_BLOCK(cs_mapAddresses)
296     {
297         // Only if it's been published already
298         map<vector<unsigned char>, CAddress>::iterator it = mapAddresses.find(addr.GetKey());
299         if (it != mapAddresses.end())
300             paddrFound = &(*it).second;
301     }
302
303     if (paddrFound)
304     {
305         int64 nUpdateInterval = 20 * 60;
306         if (paddrFound->nTime < GetAdjustedTime() - nUpdateInterval)
307         {
308             // Periodically update most recently seen time
309             paddrFound->nTime = GetAdjustedTime();
310             CAddrDB addrdb;
311             addrdb.WriteAddress(*paddrFound);
312         }
313     }
314 }
315
316
317
318
319
320 void AbandonRequests(void (*fn)(void*, CDataStream&), void* param1)
321 {
322     // If the dialog might get closed before the reply comes back,
323     // call this in the destructor so it doesn't get called after it's deleted.
324     CRITICAL_BLOCK(cs_vNodes)
325     {
326         BOOST_FOREACH(CNode* pnode, vNodes)
327         {
328             CRITICAL_BLOCK(pnode->cs_mapRequests)
329             {
330                 for (map<uint256, CRequestTracker>::iterator mi = pnode->mapRequests.begin(); mi != pnode->mapRequests.end();)
331                 {
332                     CRequestTracker& tracker = (*mi).second;
333                     if (tracker.fn == fn && tracker.param1 == param1)
334                         pnode->mapRequests.erase(mi++);
335                     else
336                         mi++;
337                 }
338             }
339         }
340     }
341 }
342
343
344
345
346
347
348
349 //
350 // Subscription methods for the broadcast and subscription system.
351 // Channel numbers are message numbers, i.e. MSG_TABLE and MSG_PRODUCT.
352 //
353 // The subscription system uses a meet-in-the-middle strategy.
354 // With 100,000 nodes, if senders broadcast to 1000 random nodes and receivers
355 // subscribe to 1000 random nodes, 99.995% (1 - 0.99^1000) of messages will get through.
356 //
357
358 bool AnySubscribed(unsigned int nChannel)
359 {
360     if (pnodeLocalHost->IsSubscribed(nChannel))
361         return true;
362     CRITICAL_BLOCK(cs_vNodes)
363         BOOST_FOREACH(CNode* pnode, vNodes)
364             if (pnode->IsSubscribed(nChannel))
365                 return true;
366     return false;
367 }
368
369 bool CNode::IsSubscribed(unsigned int nChannel)
370 {
371     if (nChannel >= vfSubscribe.size())
372         return false;
373     return vfSubscribe[nChannel];
374 }
375
376 void CNode::Subscribe(unsigned int nChannel, unsigned int nHops)
377 {
378     if (nChannel >= vfSubscribe.size())
379         return;
380
381     if (!AnySubscribed(nChannel))
382     {
383         // Relay subscribe
384         CRITICAL_BLOCK(cs_vNodes)
385             BOOST_FOREACH(CNode* pnode, vNodes)
386                 if (pnode != this)
387                     pnode->PushMessage("subscribe", nChannel, nHops);
388     }
389
390     vfSubscribe[nChannel] = true;
391 }
392
393 void CNode::CancelSubscribe(unsigned int nChannel)
394 {
395     if (nChannel >= vfSubscribe.size())
396         return;
397
398     // Prevent from relaying cancel if wasn't subscribed
399     if (!vfSubscribe[nChannel])
400         return;
401     vfSubscribe[nChannel] = false;
402
403     if (!AnySubscribed(nChannel))
404     {
405         // Relay subscription cancel
406         CRITICAL_BLOCK(cs_vNodes)
407             BOOST_FOREACH(CNode* pnode, vNodes)
408                 if (pnode != this)
409                     pnode->PushMessage("sub-cancel", nChannel);
410     }
411 }
412
413
414
415
416
417
418
419
420
421 CNode* FindNode(const CNetAddr& ip)
422 {
423     CRITICAL_BLOCK(cs_vNodes)
424     {
425         BOOST_FOREACH(CNode* pnode, vNodes)
426             if ((CNetAddr)pnode->addr == ip)
427                 return (pnode);
428     }
429     return NULL;
430 }
431
432 CNode* FindNode(const CService& addr)
433 {
434     CRITICAL_BLOCK(cs_vNodes)
435     {
436         BOOST_FOREACH(CNode* pnode, vNodes)
437             if ((CService)pnode->addr == addr)
438                 return (pnode);
439     }
440     return NULL;
441 }
442
443 CNode* ConnectNode(CAddress addrConnect, int64 nTimeout)
444 {
445     if ((CNetAddr)addrConnect == (CNetAddr)addrLocalHost)
446         return NULL;
447
448     // Look for an existing connection
449     CNode* pnode = FindNode((CService)addrConnect);
450     if (pnode)
451     {
452         if (nTimeout != 0)
453             pnode->AddRef(nTimeout);
454         else
455             pnode->AddRef();
456         return pnode;
457     }
458
459     /// debug print
460     printf("trying connection %s lastseen=%.1fhrs lasttry=%.1fhrs\n",
461         addrConnect.ToString().c_str(),
462         (double)(addrConnect.nTime - GetAdjustedTime())/3600.0,
463         (double)(addrConnect.nLastTry - GetAdjustedTime())/3600.0);
464
465     CRITICAL_BLOCK(cs_mapAddresses)
466         mapAddresses[addrConnect.GetKey()].nLastTry = GetAdjustedTime();
467
468     // Connect
469     SOCKET hSocket;
470     if (ConnectSocket(addrConnect, hSocket))
471     {
472         /// debug print
473         printf("connected %s\n", addrConnect.ToString().c_str());
474
475         // Set to nonblocking
476 #ifdef WIN32
477         u_long nOne = 1;
478         if (ioctlsocket(hSocket, FIONBIO, &nOne) == SOCKET_ERROR)
479             printf("ConnectSocket() : ioctlsocket nonblocking setting failed, error %d\n", WSAGetLastError());
480 #else
481         if (fcntl(hSocket, F_SETFL, O_NONBLOCK) == SOCKET_ERROR)
482             printf("ConnectSocket() : fcntl nonblocking setting failed, error %d\n", errno);
483 #endif
484
485         // Add node
486         CNode* pnode = new CNode(hSocket, addrConnect, false);
487         if (nTimeout != 0)
488             pnode->AddRef(nTimeout);
489         else
490             pnode->AddRef();
491         CRITICAL_BLOCK(cs_vNodes)
492             vNodes.push_back(pnode);
493
494         pnode->nTimeConnected = GetTime();
495         return pnode;
496     }
497     else
498     {
499         return NULL;
500     }
501 }
502
503 void CNode::CloseSocketDisconnect()
504 {
505     fDisconnect = true;
506     if (hSocket != INVALID_SOCKET)
507     {
508         if (fDebug)
509             printf("%s ", DateTimeStrFormat("%x %H:%M:%S", GetTime()).c_str());
510         printf("disconnecting node %s\n", addr.ToString().c_str());
511         closesocket(hSocket);
512         hSocket = INVALID_SOCKET;
513     }
514 }
515
516 void CNode::Cleanup()
517 {
518     // All of a nodes broadcasts and subscriptions are automatically torn down
519     // when it goes down, so a node has to stay up to keep its broadcast going.
520
521     // Cancel subscriptions
522     for (unsigned int nChannel = 0; nChannel < vfSubscribe.size(); nChannel++)
523         if (vfSubscribe[nChannel])
524             CancelSubscribe(nChannel);
525 }
526
527
528 void CNode::PushVersion()
529 {
530     /// when NTP implemented, change to just nTime = GetAdjustedTime()
531     int64 nTime = (fInbound ? GetAdjustedTime() : GetTime());
532     CAddress addrYou = (fUseProxy ? CAddress("0.0.0.0") : addr);
533     CAddress addrMe = (fUseProxy ? CAddress("0.0.0.0") : addrLocalHost);
534     RAND_bytes((unsigned char*)&nLocalHostNonce, sizeof(nLocalHostNonce));
535     PushMessage("version", PROTOCOL_VERSION, nLocalServices, nTime, addrYou, addrMe,
536                 nLocalHostNonce, FormatSubVersion(CLIENT_NAME, CLIENT_VERSION, std::vector<string>()), nBestHeight);
537 }
538
539
540
541
542
543 std::map<CNetAddr, int64> CNode::setBanned;
544 CCriticalSection CNode::cs_setBanned;
545
546 void CNode::ClearBanned()
547 {
548     setBanned.clear();
549 }
550
551 bool CNode::IsBanned(CNetAddr ip)
552 {
553     bool fResult = false;
554     CRITICAL_BLOCK(cs_setBanned)
555     {
556         std::map<CNetAddr, int64>::iterator i = setBanned.find(ip);
557         if (i != setBanned.end())
558         {
559             int64 t = (*i).second;
560             if (GetTime() < t)
561                 fResult = true;
562         }
563     }
564     return fResult;
565 }
566
567 bool CNode::Misbehaving(int howmuch)
568 {
569     if (addr.IsLocal())
570     {
571         printf("Warning: local node %s misbehaving\n", addr.ToString().c_str());
572         return false;
573     }
574
575     nMisbehavior += howmuch;
576     if (nMisbehavior >= GetArg("-banscore", 100))
577     {
578         int64 banTime = GetTime()+GetArg("-bantime", 60*60*24);  // Default 24-hour ban
579         CRITICAL_BLOCK(cs_setBanned)
580             if (setBanned[addr] < banTime)
581                 setBanned[addr] = banTime;
582         CloseSocketDisconnect();
583         printf("Disconnected %s for misbehavior (score=%d)\n", addr.ToString().c_str(), nMisbehavior);
584         return true;
585     }
586     return false;
587 }
588
589
590
591
592
593
594
595
596
597
598
599
600 void ThreadSocketHandler(void* parg)
601 {
602     IMPLEMENT_RANDOMIZE_STACK(ThreadSocketHandler(parg));
603     try
604     {
605         vnThreadsRunning[0]++;
606         ThreadSocketHandler2(parg);
607         vnThreadsRunning[0]--;
608     }
609     catch (std::exception& e) {
610         vnThreadsRunning[0]--;
611         PrintException(&e, "ThreadSocketHandler()");
612     } catch (...) {
613         vnThreadsRunning[0]--;
614         throw; // support pthread_cancel()
615     }
616     printf("ThreadSocketHandler exiting\n");
617 }
618
619 void ThreadSocketHandler2(void* parg)
620 {
621     printf("ThreadSocketHandler started\n");
622     list<CNode*> vNodesDisconnected;
623     int nPrevNodeCount = 0;
624
625     loop
626     {
627         //
628         // Disconnect nodes
629         //
630         CRITICAL_BLOCK(cs_vNodes)
631         {
632             // Disconnect unused nodes
633             vector<CNode*> vNodesCopy = vNodes;
634             BOOST_FOREACH(CNode* pnode, vNodesCopy)
635             {
636                 if (pnode->fDisconnect ||
637                     (pnode->GetRefCount() <= 0 && pnode->vRecv.empty() && pnode->vSend.empty()))
638                 {
639                     // remove from vNodes
640                     vNodes.erase(remove(vNodes.begin(), vNodes.end(), pnode), vNodes.end());
641
642                     // close socket and cleanup
643                     pnode->CloseSocketDisconnect();
644                     pnode->Cleanup();
645
646                     // hold in disconnected pool until all refs are released
647                     pnode->nReleaseTime = max(pnode->nReleaseTime, GetTime() + 15 * 60);
648                     if (pnode->fNetworkNode || pnode->fInbound)
649                         pnode->Release();
650                     vNodesDisconnected.push_back(pnode);
651                 }
652             }
653
654             // Delete disconnected nodes
655             list<CNode*> vNodesDisconnectedCopy = vNodesDisconnected;
656             BOOST_FOREACH(CNode* pnode, vNodesDisconnectedCopy)
657             {
658                 // wait until threads are done using it
659                 if (pnode->GetRefCount() <= 0)
660                 {
661                     bool fDelete = false;
662                     TRY_CRITICAL_BLOCK(pnode->cs_vSend)
663                      TRY_CRITICAL_BLOCK(pnode->cs_vRecv)
664                       TRY_CRITICAL_BLOCK(pnode->cs_mapRequests)
665                        TRY_CRITICAL_BLOCK(pnode->cs_inventory)
666                         fDelete = true;
667                     if (fDelete)
668                     {
669                         vNodesDisconnected.remove(pnode);
670                         delete pnode;
671                     }
672                 }
673             }
674         }
675         if (vNodes.size() != nPrevNodeCount)
676         {
677             nPrevNodeCount = vNodes.size();
678             MainFrameRepaint();
679         }
680
681
682         //
683         // Find which sockets have data to receive
684         //
685         struct timeval timeout;
686         timeout.tv_sec  = 0;
687         timeout.tv_usec = 50000; // frequency to poll pnode->vSend
688
689         fd_set fdsetRecv;
690         fd_set fdsetSend;
691         fd_set fdsetError;
692         FD_ZERO(&fdsetRecv);
693         FD_ZERO(&fdsetSend);
694         FD_ZERO(&fdsetError);
695         SOCKET hSocketMax = 0;
696
697         if(hListenSocket != INVALID_SOCKET)
698             FD_SET(hListenSocket, &fdsetRecv);
699         hSocketMax = max(hSocketMax, hListenSocket);
700         CRITICAL_BLOCK(cs_vNodes)
701         {
702             BOOST_FOREACH(CNode* pnode, vNodes)
703             {
704                 if (pnode->hSocket == INVALID_SOCKET)
705                     continue;
706                 FD_SET(pnode->hSocket, &fdsetRecv);
707                 FD_SET(pnode->hSocket, &fdsetError);
708                 hSocketMax = max(hSocketMax, pnode->hSocket);
709                 TRY_CRITICAL_BLOCK(pnode->cs_vSend)
710                     if (!pnode->vSend.empty())
711                         FD_SET(pnode->hSocket, &fdsetSend);
712             }
713         }
714
715         vnThreadsRunning[0]--;
716         int nSelect = select(hSocketMax + 1, &fdsetRecv, &fdsetSend, &fdsetError, &timeout);
717         vnThreadsRunning[0]++;
718         if (fShutdown)
719             return;
720         if (nSelect == SOCKET_ERROR)
721         {
722             int nErr = WSAGetLastError();
723             if (hSocketMax > -1)
724             {
725                 printf("socket select error %d\n", nErr);
726                 for (int i = 0; i <= hSocketMax; i++)
727                     FD_SET(i, &fdsetRecv);
728             }
729             FD_ZERO(&fdsetSend);
730             FD_ZERO(&fdsetError);
731             Sleep(timeout.tv_usec/1000);
732         }
733
734
735         //
736         // Accept new connections
737         //
738         if (hListenSocket != INVALID_SOCKET && FD_ISSET(hListenSocket, &fdsetRecv))
739         {
740             struct sockaddr_in sockaddr;
741             socklen_t len = sizeof(sockaddr);
742             SOCKET hSocket = accept(hListenSocket, (struct sockaddr*)&sockaddr, &len);
743             CAddress addr(sockaddr);
744             int nInbound = 0;
745
746             CRITICAL_BLOCK(cs_vNodes)
747                 BOOST_FOREACH(CNode* pnode, vNodes)
748                 if (pnode->fInbound)
749                     nInbound++;
750             if (hSocket == INVALID_SOCKET)
751             {
752                 if (WSAGetLastError() != WSAEWOULDBLOCK)
753                     printf("socket error accept failed: %d\n", WSAGetLastError());
754             }
755             else if (nInbound >= GetArg("-maxconnections", 125) - MAX_OUTBOUND_CONNECTIONS)
756             {
757                 CRITICAL_BLOCK(cs_setservAddNodeAddresses)
758                     if (!setservAddNodeAddresses.count(addr))
759                         closesocket(hSocket);
760             }
761             else if (CNode::IsBanned(addr))
762             {
763                 printf("connetion from %s dropped (banned)\n", addr.ToString().c_str());
764                 closesocket(hSocket);
765             }
766             else
767             {
768                 printf("accepted connection %s\n", addr.ToString().c_str());
769                 CNode* pnode = new CNode(hSocket, addr, true);
770                 pnode->AddRef();
771                 CRITICAL_BLOCK(cs_vNodes)
772                     vNodes.push_back(pnode);
773             }
774         }
775
776
777         //
778         // Service each socket
779         //
780         vector<CNode*> vNodesCopy;
781         CRITICAL_BLOCK(cs_vNodes)
782         {
783             vNodesCopy = vNodes;
784             BOOST_FOREACH(CNode* pnode, vNodesCopy)
785                 pnode->AddRef();
786         }
787         BOOST_FOREACH(CNode* pnode, vNodesCopy)
788         {
789             if (fShutdown)
790                 return;
791
792             //
793             // Receive
794             //
795             if (pnode->hSocket == INVALID_SOCKET)
796                 continue;
797             if (FD_ISSET(pnode->hSocket, &fdsetRecv) || FD_ISSET(pnode->hSocket, &fdsetError))
798             {
799                 TRY_CRITICAL_BLOCK(pnode->cs_vRecv)
800                 {
801                     CDataStream& vRecv = pnode->vRecv;
802                     unsigned int nPos = vRecv.size();
803
804                     if (nPos > ReceiveBufferSize()) {
805                         if (!pnode->fDisconnect)
806                             printf("socket recv flood control disconnect (%d bytes)\n", vRecv.size());
807                         pnode->CloseSocketDisconnect();
808                     }
809                     else {
810                         // typical socket buffer is 8K-64K
811                         char pchBuf[0x10000];
812                         int nBytes = recv(pnode->hSocket, pchBuf, sizeof(pchBuf), MSG_DONTWAIT);
813                         if (nBytes > 0)
814                         {
815                             vRecv.resize(nPos + nBytes);
816                             memcpy(&vRecv[nPos], pchBuf, nBytes);
817                             pnode->nLastRecv = GetTime();
818                         }
819                         else if (nBytes == 0)
820                         {
821                             // socket closed gracefully
822                             if (!pnode->fDisconnect)
823                                 printf("socket closed\n");
824                             pnode->CloseSocketDisconnect();
825                         }
826                         else if (nBytes < 0)
827                         {
828                             // error
829                             int nErr = WSAGetLastError();
830                             if (nErr != WSAEWOULDBLOCK && nErr != WSAEMSGSIZE && nErr != WSAEINTR && nErr != WSAEINPROGRESS)
831                             {
832                                 if (!pnode->fDisconnect)
833                                     printf("socket recv error %d\n", nErr);
834                                 pnode->CloseSocketDisconnect();
835                             }
836                         }
837                     }
838                 }
839             }
840
841             //
842             // Send
843             //
844             if (pnode->hSocket == INVALID_SOCKET)
845                 continue;
846             if (FD_ISSET(pnode->hSocket, &fdsetSend))
847             {
848                 TRY_CRITICAL_BLOCK(pnode->cs_vSend)
849                 {
850                     CDataStream& vSend = pnode->vSend;
851                     if (!vSend.empty())
852                     {
853                         int nBytes = send(pnode->hSocket, &vSend[0], vSend.size(), MSG_NOSIGNAL | MSG_DONTWAIT);
854                         if (nBytes > 0)
855                         {
856                             vSend.erase(vSend.begin(), vSend.begin() + nBytes);
857                             pnode->nLastSend = GetTime();
858                         }
859                         else if (nBytes < 0)
860                         {
861                             // error
862                             int nErr = WSAGetLastError();
863                             if (nErr != WSAEWOULDBLOCK && nErr != WSAEMSGSIZE && nErr != WSAEINTR && nErr != WSAEINPROGRESS)
864                             {
865                                 printf("socket send error %d\n", nErr);
866                                 pnode->CloseSocketDisconnect();
867                             }
868                         }
869                         if (vSend.size() > SendBufferSize()) {
870                             if (!pnode->fDisconnect)
871                                 printf("socket send flood control disconnect (%d bytes)\n", vSend.size());
872                             pnode->CloseSocketDisconnect();
873                         }
874                     }
875                 }
876             }
877
878             //
879             // Inactivity checking
880             //
881             if (pnode->vSend.empty())
882                 pnode->nLastSendEmpty = GetTime();
883             if (GetTime() - pnode->nTimeConnected > 60)
884             {
885                 if (pnode->nLastRecv == 0 || pnode->nLastSend == 0)
886                 {
887                     printf("socket no message in first 60 seconds, %d %d\n", pnode->nLastRecv != 0, pnode->nLastSend != 0);
888                     pnode->fDisconnect = true;
889                 }
890                 else if (GetTime() - pnode->nLastSend > 90*60 && GetTime() - pnode->nLastSendEmpty > 90*60)
891                 {
892                     printf("socket not sending\n");
893                     pnode->fDisconnect = true;
894                 }
895                 else if (GetTime() - pnode->nLastRecv > 90*60)
896                 {
897                     printf("socket inactivity timeout\n");
898                     pnode->fDisconnect = true;
899                 }
900             }
901         }
902         CRITICAL_BLOCK(cs_vNodes)
903         {
904             BOOST_FOREACH(CNode* pnode, vNodesCopy)
905                 pnode->Release();
906         }
907
908         Sleep(10);
909     }
910 }
911
912
913
914
915
916
917
918
919
920 #ifdef USE_UPNP
921 void ThreadMapPort(void* parg)
922 {
923     IMPLEMENT_RANDOMIZE_STACK(ThreadMapPort(parg));
924     try
925     {
926         vnThreadsRunning[5]++;
927         ThreadMapPort2(parg);
928         vnThreadsRunning[5]--;
929     }
930     catch (std::exception& e) {
931         vnThreadsRunning[5]--;
932         PrintException(&e, "ThreadMapPort()");
933     } catch (...) {
934         vnThreadsRunning[5]--;
935         PrintException(NULL, "ThreadMapPort()");
936     }
937     printf("ThreadMapPort exiting\n");
938 }
939
940 void ThreadMapPort2(void* parg)
941 {
942     printf("ThreadMapPort started\n");
943
944     char port[6];
945     sprintf(port, "%d", GetListenPort());
946
947     const char * multicastif = 0;
948     const char * minissdpdpath = 0;
949     struct UPNPDev * devlist = 0;
950     char lanaddr[64];
951
952 #ifndef UPNPDISCOVER_SUCCESS
953     /* miniupnpc 1.5 */
954     devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0);
955 #else
956     /* miniupnpc 1.6 */
957     int error = 0;
958     devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0, 0, &error);
959 #endif
960
961     struct UPNPUrls urls;
962     struct IGDdatas data;
963     int r;
964
965     r = UPNP_GetValidIGD(devlist, &urls, &data, lanaddr, sizeof(lanaddr));
966     if (r == 1)
967     {
968         string strDesc = "Bitcoin " + FormatFullVersion();
969 #ifndef UPNPDISCOVER_SUCCESS
970     /* miniupnpc 1.5 */
971         r = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype,
972                                 port, port, lanaddr, strDesc.c_str(), "TCP", 0);
973 #else
974     /* miniupnpc 1.6 */
975         r = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype,
976                                 port, port, lanaddr, strDesc.c_str(), "TCP", 0, "0");
977 #endif
978
979         if(r!=UPNPCOMMAND_SUCCESS)
980             printf("AddPortMapping(%s, %s, %s) failed with code %d (%s)\n",
981                 port, port, lanaddr, r, strupnperror(r));
982         else
983             printf("UPnP Port Mapping successful.\n");
984         loop {
985             if (fShutdown || !fUseUPnP)
986             {
987                 r = UPNP_DeletePortMapping(urls.controlURL, data.first.servicetype, port, "TCP", 0);
988                 printf("UPNP_DeletePortMapping() returned : %d\n", r);
989                 freeUPNPDevlist(devlist); devlist = 0;
990                 FreeUPNPUrls(&urls);
991                 return;
992             }
993             Sleep(2000);
994         }
995     } else {
996         printf("No valid UPnP IGDs found\n");
997         freeUPNPDevlist(devlist); devlist = 0;
998         if (r != 0)
999             FreeUPNPUrls(&urls);
1000         loop {
1001             if (fShutdown || !fUseUPnP)
1002                 return;
1003             Sleep(2000);
1004         }
1005     }
1006 }
1007
1008 void MapPort(bool fMapPort)
1009 {
1010     if (fUseUPnP != fMapPort)
1011     {
1012         fUseUPnP = fMapPort;
1013         WriteSetting("fUseUPnP", fUseUPnP);
1014     }
1015     if (fUseUPnP && vnThreadsRunning[5] < 1)
1016     {
1017         if (!CreateThread(ThreadMapPort, NULL))
1018             printf("Error: ThreadMapPort(ThreadMapPort) failed\n");
1019     }
1020 }
1021 #else
1022 void MapPort(bool /* unused fMapPort */)
1023 {
1024     // Intentionally left blank.
1025 }
1026 #endif
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037 static const char *strDNSSeed[] = {
1038     "bitseed.xf2.org",
1039     "dnsseed.bluematt.me",
1040     "seed.bitcoin.sipa.be",
1041     "dnsseed.bitcoin.dashjr.org",
1042 };
1043
1044 void ThreadDNSAddressSeed(void* parg)
1045 {
1046     IMPLEMENT_RANDOMIZE_STACK(ThreadDNSAddressSeed(parg));
1047     try
1048     {
1049         vnThreadsRunning[6]++;
1050         ThreadDNSAddressSeed2(parg);
1051         vnThreadsRunning[6]--;
1052     }
1053     catch (std::exception& e) {
1054         vnThreadsRunning[6]--;
1055         PrintException(&e, "ThreadDNSAddressSeed()");
1056     } catch (...) {
1057         vnThreadsRunning[6]--;
1058         throw; // support pthread_cancel()
1059     }
1060     printf("ThreadDNSAddressSeed exiting\n");
1061 }
1062
1063 void ThreadDNSAddressSeed2(void* parg)
1064 {
1065     printf("ThreadDNSAddressSeed started\n");
1066     int found = 0;
1067
1068     if (!fTestNet)
1069     {
1070         printf("Loading addresses from DNS seeds (could take a while)\n");
1071
1072         for (int seed_idx = 0; seed_idx < ARRAYLEN(strDNSSeed); seed_idx++) {
1073             vector<CNetAddr> vaddr;
1074             if (LookupHost(strDNSSeed[seed_idx], vaddr))
1075             {
1076                 CAddrDB addrDB;
1077                 addrDB.TxnBegin();
1078                 BOOST_FOREACH (CNetAddr& ip, vaddr)
1079                 {
1080                     if (ip.IsRoutable())
1081                     {
1082                         CAddress addr(CService(ip, GetDefaultPort()), NODE_NETWORK);
1083                         addr.nTime = 0;
1084                         AddAddress(addr, 0, &addrDB);
1085                         found++;
1086                     }
1087                 }
1088                 addrDB.TxnCommit();  // Save addresses (it's ok if this fails)
1089             }
1090         }
1091     }
1092
1093     printf("%d addresses found from DNS seeds\n", found);
1094 }
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107 unsigned int pnSeed[] =
1108 {
1109     0x6884ac63, 0x3ffecead, 0x2919b953, 0x0942fe50, 0x7a1d922e, 0xcdd6734a, 0x953a5bb6, 0x2c46922e,
1110     0xe2a5f143, 0xaa39103a, 0xa06afa5c, 0x135ffd59, 0xe8e82863, 0xf61ef029, 0xf75f042e, 0x2b363532,
1111     0x29b2df42, 0x16b1f64e, 0xd46e281b, 0x5280bf58, 0x60372229, 0x1be58e4f, 0xa8496f45, 0x1fb1a057,
1112     0x756b3844, 0x3bb79445, 0x0b375518, 0xcccb0102, 0xb682bf2e, 0x46431c02, 0x3a81073a, 0xa3771f1f,
1113     0x213a121f, 0x85dc2c1b, 0x56b4323b, 0xb34e8945, 0x3c40b33d, 0xfa276418, 0x1f818d29, 0xebe1e344,
1114     0xf6160a18, 0xf4fa384a, 0x34b09558, 0xb882b543, 0xe3ce2253, 0x6abf56d8, 0xe91b1155, 0x688ee6ad,
1115     0x2efc6058, 0x4792cd47, 0x0c32f757, 0x4c813a46, 0x8c93644a, 0x37507444, 0x813ad218, 0xdac06d4a,
1116     0xe4c63e4b, 0x21a1ea3c, 0x8d88556f, 0x30e9173a, 0x041f681b, 0xdc77ba50, 0xc0072753, 0xceddd44f,
1117     0x052d1743, 0xe3c77a4a, 0x13981c3a, 0x5685d918, 0x3c0e4e70, 0x3e56fb54, 0xb676ae0c, 0xac93c859,
1118     0x22279f43, 0x975a4542, 0xe527f071, 0xea162f2e, 0x3c65a32e, 0x5be5713b, 0x961ec418, 0xb202922e,
1119     0x5ef7be50, 0xce49f53e, 0x05803b47, 0x8463b055, 0x78576153, 0x3ec2ae3a, 0x4bbd7118, 0xafcee043,
1120     0x56a3e8ba, 0x6174de4d, 0x8d01ba4b, 0xc9af564e, 0xdbc9c547, 0xa627474d, 0xdada9244, 0xd3b3083a,
1121     0x523e071f, 0xd6b96f18, 0xbd527c46, 0xdf2bbb4d, 0xd37b4a4b, 0x3a6a2158, 0xc064b055, 0x18a8e055,
1122     0xec4dae3b, 0x0540416c, 0x475b4fbe, 0x064803b2, 0x48e9f062, 0x2898524b, 0xd315ff43, 0xf786d247,
1123     0xc7ea2f3e, 0xc087f043, 0xc163354b, 0x8250284d, 0xed300029, 0xbf36e05c, 0x8eb3ae4c, 0xe7aa623e,
1124     0x7ced0274, 0xdd362c1b, 0x362b995a, 0xca26b629, 0x3fc41618, 0xb97b364e, 0xa05b8729, 0x0f5e3c43,
1125     0xdf942618, 0x6aeb9b5b, 0xbf04762e, 0xfaaeb118, 0x87579958, 0x76520044, 0xc2660c5b, 0x628b201b,
1126     0xf193932e, 0x1c0ad045, 0xff908346, 0x8da9d4da, 0xed201c1f, 0xa47a2b1b, 0x330007d4, 0x8ba1ed47,
1127     0xb2f02d44, 0x7db62c1b, 0x781c454b, 0xc0300029, 0xb7062a45, 0x88b52e3a, 0x78dd6b63, 0x1cb9b718,
1128     0x5d358e47, 0x59912c3b, 0x79607544, 0x5197f759, 0xc023be48, 0xd1013743, 0x0f354057, 0x8e3aac3b,
1129     0x4114693e, 0x22316318, 0xe27dda50, 0x878eac3b, 0x4948a21f, 0x5db7f24c, 0x8ccb6157, 0x26a5de18,
1130     0x0a11bd43, 0x27bb1e41, 0x60a7a951, 0x3e16b35e, 0x07888b53, 0x5648a853, 0x0149fe50, 0xd070a34f,
1131     0x6454c96d, 0xd6e54758, 0xa96dc152, 0x65447861, 0xf6bdf95e, 0x10400202, 0x2c29d483, 0x18174732,
1132     0x1d840618, 0x12e61818, 0x089d3f3c, 0x917e931f, 0xd1b0c90e, 0x25bd3c42, 0xeb05775b, 0x7d550c59,
1133     0x6cfacb01, 0xe4224444, 0xa41dd943, 0x0f5aa643, 0x5e33731b, 0x81036d50, 0x6f46a0d1, 0x7731be43,
1134     0x14840e18, 0xf1e8d059, 0x661d2b1f, 0x40a3201b, 0x9407b843, 0xedf0254d, 0x7bd1a5bc, 0x073dbe51,
1135     0xe864a97b, 0x2efd947b, 0xb9ca0e45, 0x4e2113ad, 0xcc305731, 0xd39ca63c, 0x733df918, 0xda172b1f,
1136     0xaa03b34d, 0x7230fd4d, 0xf1ce6e3a, 0x2e9fab43, 0xa4010750, 0xa928bd18, 0x6809be42, 0xb19de348,
1137     0xff956270, 0x0d795f51, 0xd2dec247, 0x6df5774b, 0xbac11f79, 0xdfb05c75, 0x887683d8, 0xa1e83632,
1138     0x2c0f7671, 0x28bcb65d, 0xac2a7545, 0x3eebfc60, 0x304ad7c4, 0xa215a462, 0xc86f0f58, 0xcfb92ebe,
1139     0x5e23ed82, 0xf506184b, 0xec0f19b7, 0x060c59ad, 0x86ee3174, 0x85380774, 0xa199a562, 0x02b507ae,
1140     0x33eb2163, 0xf2112b1f, 0xb702ba50, 0x131b9618, 0x90ccd04a, 0x08f3273b, 0xecb61718, 0x64b8b44d,
1141     0x182bf4dc, 0xc7b68286, 0x6e318d5f, 0xfdb03654, 0xb3272e54, 0xe014ad4b, 0x274e4a31, 0x7806375c,
1142     0xbc34a748, 0x1b5ad94a, 0x6b54d10e, 0x73e2ae6e, 0x5529d483, 0x8455a76d, 0x99c13f47, 0x1d811741,
1143     0xa9782a78, 0x0b00464d, 0x7266ea50, 0x532dab46, 0x33e1413e, 0x780d0c18, 0x0fb0854e, 0x03370155,
1144     0x2693042e, 0xfa3d824a, 0x2bb1681b, 0x37ea2a18, 0x7fb8414b, 0x32e0713b, 0xacf38d3f, 0xa282716f,
1145     0xb1a09d7b, 0xa04b764b, 0x83c94d18, 0x05ee4c6d, 0x0e795f51, 0x46984352, 0xf80fc247, 0x3fccb946,
1146     0xd7ae244b, 0x0a8e0a4c, 0x57b141bc, 0x3647bed1, 0x1431b052, 0x803a8bbb, 0xfc69056b, 0xf5991862,
1147     0x14963b2e, 0xd35d5dda, 0xc6c73574, 0xc8f1405b, 0x0ca4224d, 0xecd36071, 0xa9461754, 0xe7a0ed72,
1148     0x559e8346, 0x1c9beec1, 0xc786ea4a, 0x9561b44d, 0x9788074d, 0x1a69934f, 0x23c5614c, 0x07c79d4b,
1149     0xc7ee52db, 0xc72df351, 0xcb135e44, 0xa0988346, 0xc211fc4c, 0x87dec34b, 0x1381074d, 0x04a65cb7,
1150     0x4409083a, 0x4a407a4c, 0x92b8d37d, 0xacf50b4d, 0xa58aa5bc, 0x448f801f, 0x9c83762e, 0x6fd5734a,
1151     0xfe2d454b, 0x84144c55, 0x05190e4c, 0xb2151448, 0x63867a3e, 0x16099018, 0x9c010d3c, 0x962d8f3d,
1152     0xd51ee453, 0x9d86801f, 0x68e87b47, 0x6bf7bb73, 0x5fc7910e, 0x10d90118, 0x3db04442, 0x729d3e4b,
1153     0xc397d842, 0x57bb15ad, 0x72f31f4e, 0xc9380043, 0x2bb24e18, 0xd9b8ab50, 0xb786801f, 0xf4dc4847,
1154     0x85f4bb51, 0x4435995b, 0x5ba07e40, 0x2c57392e, 0x3628124b, 0x9839b64b, 0x6fe8b24d, 0xaddce847,
1155     0x75260e45, 0x0c572a43, 0xfea21902, 0xb9f9742e, 0x5a70d443, 0x8fc5910e, 0x868d4744, 0x56245e02,
1156     0xd7eb5f02, 0x35c12c1b, 0x4373034b, 0x8786554c, 0xa6facf18, 0x4b11a31f, 0x3570664e, 0x5a64bc42,
1157     0x0b03983f, 0x8f457e4c, 0x0fd874c3, 0xb6cf31b2, 0x2bbc2d4e, 0x146ca5b2, 0x9d00b150, 0x048a4153,
1158     0xca4dcd43, 0xc1607cca, 0x8234cf57, 0x9c7daead, 0x3dc07658, 0xea5c6e4c, 0xf1a0084e, 0x16d2ee53,
1159     0x1b849418, 0xfe913a47, 0x1e988f62, 0x208b644c, 0xc55ee980, 0xbdbce747, 0xf59a384e, 0x0f56091b,
1160     0x7417b745, 0x0c37344e, 0x2c62ab47, 0xf8533a4d, 0x8030084d, 0x76b93c4b, 0xda6ea0ad, 0x3c54f618,
1161     0x63b0de1f, 0x7370d858, 0x1a70bb4c, 0xdda63b2e, 0x60b2ba50, 0x1ba7d048, 0xbe1b2c1b, 0xabea5747,
1162     0x29ad2e4d, 0xe8cd7642, 0x66c80e18, 0x138bf34a, 0xc6145e44, 0x2586794c, 0x07bc5478, 0x0da0b14d,
1163     0x8f95354e, 0x9eb11c62, 0xa1545e46, 0x2e7a2602, 0x408c9c3d, 0x59065d55, 0xf51d1a4c, 0x3bbc6a4e,
1164     0xc71b2a2e, 0xcdaaa545, 0x17d659d0, 0x5202e7ad, 0xf1b68445, 0x93375961, 0xbd88a043, 0x066ad655,
1165     0x890f6318, 0x7b7dca47, 0x99bdd662, 0x3bb4fc53, 0x1231efdc, 0xc0a99444, 0x96bbea47, 0x61ed8748,
1166     0x27dfa73b, 0x8d4d1754, 0x3460042e, 0x551f0c4c, 0x8d0e0718, 0x162ddc53, 0x53231718, 0x1ecd65d0,
1167     0x944d28bc, 0x3b79d058, 0xaff97fbc, 0x4860006c, 0xc101c90e, 0xace41743, 0xa5975d4c, 0x5cc2703e,
1168     0xb55a4450, 0x02d18840, 0xee2765ae, 0xd6012fd5, 0x24c94d7d, 0x8c6eec47, 0x7520ba5d, 0x9e15e460,
1169     0x8510b04c, 0x75ec3847, 0x1dfa6661, 0xe172b3ad, 0x5744c90e, 0x52a0a152, 0x8d6fad18, 0x67b74b6d,
1170     0x93a089b2, 0x0f3ac5d5, 0xe5de1855, 0x43d25747, 0x4bad804a, 0x55b408d8, 0x60a36441, 0xf553e860,
1171     0xdb2fa2c8, 0x03152b32, 0xdd27a7d5, 0x3116a8b8, 0x0a1d708c, 0xeee2f13c, 0x6acf436f, 0xce6eb4ca,
1172     0x101cd3d9, 0x1c48a6b8, 0xe57d6f44, 0x93dcf562,
1173 };
1174
1175
1176
1177 void ThreadOpenConnections(void* parg)
1178 {
1179     IMPLEMENT_RANDOMIZE_STACK(ThreadOpenConnections(parg));
1180     try
1181     {
1182         vnThreadsRunning[1]++;
1183         ThreadOpenConnections2(parg);
1184         vnThreadsRunning[1]--;
1185     }
1186     catch (std::exception& e) {
1187         vnThreadsRunning[1]--;
1188         PrintException(&e, "ThreadOpenConnections()");
1189     } catch (...) {
1190         vnThreadsRunning[1]--;
1191         PrintException(NULL, "ThreadOpenConnections()");
1192     }
1193     printf("ThreadOpenConnections exiting\n");
1194 }
1195
1196 void ThreadOpenConnections2(void* parg)
1197 {
1198     printf("ThreadOpenConnections started\n");
1199
1200     // Connect to specific addresses
1201     if (mapArgs.count("-connect"))
1202     {
1203         for (int64 nLoop = 0;; nLoop++)
1204         {
1205             BOOST_FOREACH(string strAddr, mapMultiArgs["-connect"])
1206             {
1207                 CAddress addr(strAddr, fAllowDNS);
1208                 if (addr.IsValid())
1209                     OpenNetworkConnection(addr);
1210                 for (int i = 0; i < 10 && i < nLoop; i++)
1211                 {
1212                     Sleep(500);
1213                     if (fShutdown)
1214                         return;
1215                 }
1216             }
1217         }
1218     }
1219
1220     // Initiate network connections
1221     int64 nStart = GetTime();
1222     loop
1223     {
1224         // Limit outbound connections
1225         vnThreadsRunning[1]--;
1226         Sleep(500);
1227         loop
1228         {
1229             int nOutbound = 0;
1230             CRITICAL_BLOCK(cs_vNodes)
1231                 BOOST_FOREACH(CNode* pnode, vNodes)
1232                     if (!pnode->fInbound)
1233                         nOutbound++;
1234             int nMaxOutboundConnections = MAX_OUTBOUND_CONNECTIONS;
1235             nMaxOutboundConnections = min(nMaxOutboundConnections, (int)GetArg("-maxconnections", 125));
1236             if (nOutbound < nMaxOutboundConnections)
1237                 break;
1238             Sleep(2000);
1239             if (fShutdown)
1240                 return;
1241         }
1242         vnThreadsRunning[1]++;
1243         if (fShutdown)
1244             return;
1245
1246         bool fAddSeeds = false;
1247
1248         CRITICAL_BLOCK(cs_mapAddresses)
1249         {
1250             // Add seed nodes if IRC isn't working
1251             bool fTOR = (fUseProxy && addrProxy.GetPort() == 9050);
1252             if (mapAddresses.empty() && (GetTime() - nStart > 60 || fTOR) && !fTestNet)
1253                 fAddSeeds = true;
1254         }
1255
1256         if (fAddSeeds)
1257         {
1258             for (int i = 0; i < ARRAYLEN(pnSeed); i++)
1259             {
1260                 // It'll only connect to one or two seed nodes because once it connects,
1261                 // it'll get a pile of addresses with newer timestamps.
1262                 // Seed nodes are given a random 'last seen time' of between one and two
1263                 // weeks ago.
1264                 const int64 nOneWeek = 7*24*60*60;
1265                 struct in_addr ip;
1266                 memcpy(&ip, &pnSeed[i], sizeof(ip));
1267                 CAddress addr(CService(ip, GetDefaultPort()));
1268                 addr.nTime = GetTime()-GetRand(nOneWeek)-nOneWeek;
1269                 AddAddress(addr);
1270             }
1271         }
1272
1273         //
1274         // Choose an address to connect to based on most recently seen
1275         //
1276         CAddress addrConnect;
1277         int64 nBest = std::numeric_limits<int64>::min();
1278
1279         // Only connect to one address per a.b.?.? range.
1280         // Do this here so we don't have to critsect vNodes inside mapAddresses critsect.
1281         set<vector<unsigned char> > setConnected;
1282         CRITICAL_BLOCK(cs_vNodes)
1283             BOOST_FOREACH(CNode* pnode, vNodes)
1284                 setConnected.insert(pnode->addr.GetGroup());
1285
1286         int64 nANow = GetAdjustedTime();
1287
1288         CRITICAL_BLOCK(cs_mapAddresses)
1289         {
1290             BOOST_FOREACH(const PAIRTYPE(vector<unsigned char>, CAddress)& item, mapAddresses)
1291             {
1292                 const CAddress& addr = item.second;
1293                 if (!addr.IsIPv4() || !addr.IsValid() || setConnected.count(addr.GetGroup()))
1294                     continue;
1295                 int64 nSinceLastSeen = nANow - addr.nTime;
1296                 int64 nSinceLastTry = nANow - addr.nLastTry;
1297
1298                 // Randomize the order in a deterministic way, putting the standard port first
1299                 int64 nRandomizer = (uint64)(nStart * 4951 + addr.nLastTry * 9567851 + addr.GetHash()) % (2 * 60 * 60);
1300                 if (addr.GetPort() != GetDefaultPort())
1301                     nRandomizer += 2 * 60 * 60;
1302
1303                 // Last seen  Base retry frequency
1304                 //   <1 hour   10 min
1305                 //    1 hour    1 hour
1306                 //    4 hours   2 hours
1307                 //   24 hours   5 hours
1308                 //   48 hours   7 hours
1309                 //    7 days   13 hours
1310                 //   30 days   27 hours
1311                 //   90 days   46 hours
1312                 //  365 days   93 hours
1313                 int64 nDelay = (int64)(3600.0 * sqrt(fabs((double)nSinceLastSeen) / 3600.0) + nRandomizer);
1314
1315                 // Fast reconnect for one hour after last seen
1316                 if (nSinceLastSeen < 60 * 60)
1317                     nDelay = 10 * 60;
1318
1319                 // Limit retry frequency
1320                 if (nSinceLastTry < nDelay)
1321                     continue;
1322
1323                 // If we have IRC, we'll be notified when they first come online,
1324                 // and again every 24 hours by the refresh broadcast.
1325                 if (nGotIRCAddresses > 0 && vNodes.size() >= 2 && nSinceLastSeen > 24 * 60 * 60)
1326                     continue;
1327
1328                 // Only try the old stuff if we don't have enough connections
1329                 if (vNodes.size() >= 8 && nSinceLastSeen > 24 * 60 * 60)
1330                     continue;
1331
1332                 // If multiple addresses are ready, prioritize by time since
1333                 // last seen and time since last tried.
1334                 int64 nScore = min(nSinceLastTry, (int64)24 * 60 * 60) - nSinceLastSeen - nRandomizer;
1335                 if (nScore > nBest)
1336                 {
1337                     nBest = nScore;
1338                     addrConnect = addr;
1339                 }
1340             }
1341         }
1342
1343         if (addrConnect.IsValid())
1344             OpenNetworkConnection(addrConnect);
1345     }
1346 }
1347
1348 void ThreadOpenAddedConnections(void* parg)
1349 {
1350     IMPLEMENT_RANDOMIZE_STACK(ThreadOpenAddedConnections(parg));
1351     try
1352     {
1353         vnThreadsRunning[7]++;
1354         ThreadOpenAddedConnections2(parg);
1355         vnThreadsRunning[7]--;
1356     }
1357     catch (std::exception& e) {
1358         vnThreadsRunning[7]--;
1359         PrintException(&e, "ThreadOpenAddedConnections()");
1360     } catch (...) {
1361         vnThreadsRunning[7]--;
1362         PrintException(NULL, "ThreadOpenAddedConnections()");
1363     }
1364     printf("ThreadOpenAddedConnections exiting\n");
1365 }
1366
1367 void ThreadOpenAddedConnections2(void* parg)
1368 {
1369     printf("ThreadOpenAddedConnections started\n");
1370
1371     if (mapArgs.count("-addnode") == 0)
1372         return;
1373
1374     vector<vector<CService> > vservAddressesToAdd(0);
1375     BOOST_FOREACH(string& strAddNode, mapMultiArgs["-addnode"])
1376     {
1377         vector<CService> vservNode(0);
1378         if(Lookup(strAddNode.c_str(), vservNode, GetDefaultPort(), fAllowDNS, 0))
1379         {
1380             vservAddressesToAdd.push_back(vservNode);
1381             CRITICAL_BLOCK(cs_setservAddNodeAddresses)
1382                 BOOST_FOREACH(CService& serv, vservNode)
1383                     setservAddNodeAddresses.insert(serv);
1384         }
1385     }
1386     loop
1387     {
1388         vector<vector<CService> > vservConnectAddresses = vservAddressesToAdd;
1389         // Attempt to connect to each IP for each addnode entry until at least one is successful per addnode entry
1390         // (keeping in mind that addnode entries can have many IPs if fAllowDNS)
1391         CRITICAL_BLOCK(cs_vNodes)
1392             BOOST_FOREACH(CNode* pnode, vNodes)
1393                 for (vector<vector<CService> >::iterator it = vservConnectAddresses.begin(); it != vservConnectAddresses.end(); it++)
1394                     BOOST_FOREACH(CService& addrNode, *(it))
1395                         if (pnode->addr == addrNode)
1396                         {
1397                             it = vservConnectAddresses.erase(it);
1398                             it--;
1399                             break;
1400                         }
1401         BOOST_FOREACH(vector<CService>& vserv, vservConnectAddresses)
1402         {
1403             OpenNetworkConnection(CAddress(*(vserv.begin())));
1404             Sleep(500);
1405             if (fShutdown)
1406                 return;
1407         }
1408         if (fShutdown)
1409             return;
1410         vnThreadsRunning[7]--;
1411         Sleep(120000); // Retry every 2 minutes
1412         vnThreadsRunning[7]++;
1413         if (fShutdown)
1414             return;
1415     }
1416 }
1417
1418 bool OpenNetworkConnection(const CAddress& addrConnect)
1419 {
1420     //
1421     // Initiate outbound network connection
1422     //
1423     if (fShutdown)
1424         return false;
1425     if ((CNetAddr)addrConnect == (CNetAddr)addrLocalHost || !addrConnect.IsIPv4() ||
1426         FindNode((CNetAddr)addrConnect) || CNode::IsBanned(addrConnect))
1427         return false;
1428
1429     vnThreadsRunning[1]--;
1430     CNode* pnode = ConnectNode(addrConnect);
1431     vnThreadsRunning[1]++;
1432     if (fShutdown)
1433         return false;
1434     if (!pnode)
1435         return false;
1436     pnode->fNetworkNode = true;
1437
1438     return true;
1439 }
1440
1441
1442
1443
1444
1445
1446
1447
1448 void ThreadMessageHandler(void* parg)
1449 {
1450     IMPLEMENT_RANDOMIZE_STACK(ThreadMessageHandler(parg));
1451     try
1452     {
1453         vnThreadsRunning[2]++;
1454         ThreadMessageHandler2(parg);
1455         vnThreadsRunning[2]--;
1456     }
1457     catch (std::exception& e) {
1458         vnThreadsRunning[2]--;
1459         PrintException(&e, "ThreadMessageHandler()");
1460     } catch (...) {
1461         vnThreadsRunning[2]--;
1462         PrintException(NULL, "ThreadMessageHandler()");
1463     }
1464     printf("ThreadMessageHandler exiting\n");
1465 }
1466
1467 void ThreadMessageHandler2(void* parg)
1468 {
1469     printf("ThreadMessageHandler started\n");
1470     SetThreadPriority(THREAD_PRIORITY_BELOW_NORMAL);
1471     while (!fShutdown)
1472     {
1473         vector<CNode*> vNodesCopy;
1474         CRITICAL_BLOCK(cs_vNodes)
1475         {
1476             vNodesCopy = vNodes;
1477             BOOST_FOREACH(CNode* pnode, vNodesCopy)
1478                 pnode->AddRef();
1479         }
1480
1481         // Poll the connected nodes for messages
1482         CNode* pnodeTrickle = NULL;
1483         if (!vNodesCopy.empty())
1484             pnodeTrickle = vNodesCopy[GetRand(vNodesCopy.size())];
1485         BOOST_FOREACH(CNode* pnode, vNodesCopy)
1486         {
1487             // Receive messages
1488             TRY_CRITICAL_BLOCK(pnode->cs_vRecv)
1489                 ProcessMessages(pnode);
1490             if (fShutdown)
1491                 return;
1492
1493             // Send messages
1494             TRY_CRITICAL_BLOCK(pnode->cs_vSend)
1495                 SendMessages(pnode, pnode == pnodeTrickle);
1496             if (fShutdown)
1497                 return;
1498         }
1499
1500         CRITICAL_BLOCK(cs_vNodes)
1501         {
1502             BOOST_FOREACH(CNode* pnode, vNodesCopy)
1503                 pnode->Release();
1504         }
1505
1506         // Wait and allow messages to bunch up.
1507         // Reduce vnThreadsRunning so StopNode has permission to exit while
1508         // we're sleeping, but we must always check fShutdown after doing this.
1509         vnThreadsRunning[2]--;
1510         Sleep(100);
1511         if (fRequestShutdown)
1512             Shutdown(NULL);
1513         vnThreadsRunning[2]++;
1514         if (fShutdown)
1515             return;
1516     }
1517 }
1518
1519
1520
1521
1522
1523
1524 bool BindListenPort(string& strError)
1525 {
1526     strError = "";
1527     int nOne = 1;
1528     addrLocalHost.SetPort(GetListenPort());
1529
1530 #ifdef WIN32
1531     // Initialize Windows Sockets
1532     WSADATA wsadata;
1533     int ret = WSAStartup(MAKEWORD(2,2), &wsadata);
1534     if (ret != NO_ERROR)
1535     {
1536         strError = strprintf("Error: TCP/IP socket library failed to start (WSAStartup returned error %d)", ret);
1537         printf("%s\n", strError.c_str());
1538         return false;
1539     }
1540 #endif
1541
1542     // Create socket for listening for incoming connections
1543     hListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
1544     if (hListenSocket == INVALID_SOCKET)
1545     {
1546         strError = strprintf("Error: Couldn't open socket for incoming connections (socket returned error %d)", WSAGetLastError());
1547         printf("%s\n", strError.c_str());
1548         return false;
1549     }
1550
1551 #ifdef SO_NOSIGPIPE
1552     // Different way of disabling SIGPIPE on BSD
1553     setsockopt(hListenSocket, SOL_SOCKET, SO_NOSIGPIPE, (void*)&nOne, sizeof(int));
1554 #endif
1555
1556 #ifndef WIN32
1557     // Allow binding if the port is still in TIME_WAIT state after
1558     // the program was closed and restarted.  Not an issue on windows.
1559     setsockopt(hListenSocket, SOL_SOCKET, SO_REUSEADDR, (void*)&nOne, sizeof(int));
1560 #endif
1561
1562 #ifdef WIN32
1563     // Set to nonblocking, incoming connections will also inherit this
1564     if (ioctlsocket(hListenSocket, FIONBIO, (u_long*)&nOne) == SOCKET_ERROR)
1565 #else
1566     if (fcntl(hListenSocket, F_SETFL, O_NONBLOCK) == SOCKET_ERROR)
1567 #endif
1568     {
1569         strError = strprintf("Error: Couldn't set properties on socket for incoming connections (error %d)", WSAGetLastError());
1570         printf("%s\n", strError.c_str());
1571         return false;
1572     }
1573
1574     // The sockaddr_in structure specifies the address family,
1575     // IP address, and port for the socket that is being bound
1576     struct sockaddr_in sockaddr;
1577     memset(&sockaddr, 0, sizeof(sockaddr));
1578     sockaddr.sin_family = AF_INET;
1579     sockaddr.sin_addr.s_addr = INADDR_ANY; // bind to all IPs on this computer
1580     sockaddr.sin_port = htons(GetListenPort());
1581     if (::bind(hListenSocket, (struct sockaddr*)&sockaddr, sizeof(sockaddr)) == SOCKET_ERROR)
1582     {
1583         int nErr = WSAGetLastError();
1584         if (nErr == WSAEADDRINUSE)
1585             strError = strprintf(_("Unable to bind to port %d on this computer.  Bitcoin is probably already running."), ntohs(sockaddr.sin_port));
1586         else
1587             strError = strprintf("Error: Unable to bind to port %d on this computer (bind returned error %d)", ntohs(sockaddr.sin_port), nErr);
1588         printf("%s\n", strError.c_str());
1589         return false;
1590     }
1591     printf("Bound to port %d\n", ntohs(sockaddr.sin_port));
1592
1593     // Listen for incoming connections
1594     if (listen(hListenSocket, SOMAXCONN) == SOCKET_ERROR)
1595     {
1596         strError = strprintf("Error: Listening for incoming connections failed (listen returned error %d)", WSAGetLastError());
1597         printf("%s\n", strError.c_str());
1598         return false;
1599     }
1600
1601     return true;
1602 }
1603
1604 void StartNode(void* parg)
1605 {
1606     if (pnodeLocalHost == NULL)
1607         pnodeLocalHost = new CNode(INVALID_SOCKET, CAddress(CService("127.0.0.1", 0), nLocalServices));
1608
1609 #ifdef WIN32
1610     // Get local host ip
1611     char pszHostName[1000] = "";
1612     if (gethostname(pszHostName, sizeof(pszHostName)) != SOCKET_ERROR)
1613     {
1614         vector<CNetAddr> vaddr;
1615         if (LookupHost(pszHostName, vaddr))
1616             BOOST_FOREACH (const CNetAddr &addr, vaddr)
1617                 if (!addr.IsLocal())
1618                 {
1619                     addrLocalHost.SetIP(addr);
1620                     break;
1621                 }
1622     }
1623 #else
1624     // Get local host ip
1625     struct ifaddrs* myaddrs;
1626     if (getifaddrs(&myaddrs) == 0)
1627     {
1628         for (struct ifaddrs* ifa = myaddrs; ifa != NULL; ifa = ifa->ifa_next)
1629         {
1630             if (ifa->ifa_addr == NULL) continue;
1631             if ((ifa->ifa_flags & IFF_UP) == 0) continue;
1632             if (strcmp(ifa->ifa_name, "lo") == 0) continue;
1633             if (strcmp(ifa->ifa_name, "lo0") == 0) continue;
1634             char pszIP[100];
1635             if (ifa->ifa_addr->sa_family == AF_INET)
1636             {
1637                 struct sockaddr_in* s4 = (struct sockaddr_in*)(ifa->ifa_addr);
1638                 if (inet_ntop(ifa->ifa_addr->sa_family, (void*)&(s4->sin_addr), pszIP, sizeof(pszIP)) != NULL)
1639                     printf("ipv4 %s: %s\n", ifa->ifa_name, pszIP);
1640
1641                 // Take the first IP that isn't loopback 127.x.x.x
1642                 CAddress addr(CService(s4->sin_addr, GetListenPort()), nLocalServices);
1643                 if (addr.IsValid() && !addr.IsLocal())
1644                 {
1645                     addrLocalHost = addr;
1646                     break;
1647                 }
1648             }
1649             else if (ifa->ifa_addr->sa_family == AF_INET6)
1650             {
1651                 struct sockaddr_in6* s6 = (struct sockaddr_in6*)(ifa->ifa_addr);
1652                 if (inet_ntop(ifa->ifa_addr->sa_family, (void*)&(s6->sin6_addr), pszIP, sizeof(pszIP)) != NULL)
1653                     printf("ipv6 %s: %s\n", ifa->ifa_name, pszIP);
1654             }
1655         }
1656         freeifaddrs(myaddrs);
1657     }
1658 #endif
1659     printf("addrLocalHost = %s\n", addrLocalHost.ToString().c_str());
1660
1661     if (fUseProxy || mapArgs.count("-connect") || fNoListen)
1662     {
1663         // Proxies can't take incoming connections
1664         addrLocalHost.SetIP(CNetAddr("0.0.0.0"));
1665         printf("addrLocalHost = %s\n", addrLocalHost.ToString().c_str());
1666     }
1667     else
1668     {
1669         CreateThread(ThreadGetMyExternalIP, NULL);
1670     }
1671
1672     //
1673     // Start threads
1674     //
1675
1676     if (GetBoolArg("-nodnsseed"))
1677         printf("DNS seeding disabled\n");
1678     else
1679         if (!CreateThread(ThreadDNSAddressSeed, NULL))
1680             printf("Error: CreateThread(ThreadDNSAddressSeed) failed\n");
1681
1682     // Map ports with UPnP
1683     if (fHaveUPnP)
1684         MapPort(fUseUPnP);
1685
1686     // Get addresses from IRC and advertise ours
1687     if (!CreateThread(ThreadIRCSeed, NULL))
1688         printf("Error: CreateThread(ThreadIRCSeed) failed\n");
1689
1690     // Send and receive from sockets, accept connections
1691     if (!CreateThread(ThreadSocketHandler, NULL))
1692         printf("Error: CreateThread(ThreadSocketHandler) failed\n");
1693
1694     // Initiate outbound connections from -addnode
1695     if (!CreateThread(ThreadOpenAddedConnections, NULL))
1696         printf("Error: CreateThread(ThreadOpenAddedConnections) failed\n");
1697
1698     // Initiate outbound connections
1699     if (!CreateThread(ThreadOpenConnections, NULL))
1700         printf("Error: CreateThread(ThreadOpenConnections) failed\n");
1701
1702     // Process messages
1703     if (!CreateThread(ThreadMessageHandler, NULL))
1704         printf("Error: CreateThread(ThreadMessageHandler) failed\n");
1705
1706     // Generate coins in the background
1707     GenerateBitcoins(fGenerateBitcoins, pwalletMain);
1708 }
1709
1710 bool StopNode()
1711 {
1712     printf("StopNode()\n");
1713     fShutdown = true;
1714     nTransactionsUpdated++;
1715     int64 nStart = GetTime();
1716     while (vnThreadsRunning[0] > 0 || vnThreadsRunning[2] > 0 || vnThreadsRunning[3] > 0 || vnThreadsRunning[4] > 0
1717         || (fHaveUPnP && vnThreadsRunning[5] > 0) || vnThreadsRunning[6] > 0 || vnThreadsRunning[7] > 0
1718     )
1719     {
1720         if (GetTime() - nStart > 20)
1721             break;
1722         Sleep(20);
1723     }
1724     if (vnThreadsRunning[0] > 0) printf("ThreadSocketHandler still running\n");
1725     if (vnThreadsRunning[1] > 0) printf("ThreadOpenConnections still running\n");
1726     if (vnThreadsRunning[2] > 0) printf("ThreadMessageHandler still running\n");
1727     if (vnThreadsRunning[3] > 0) printf("ThreadBitcoinMiner still running\n");
1728     if (vnThreadsRunning[4] > 0) printf("ThreadRPCServer still running\n");
1729     if (fHaveUPnP && vnThreadsRunning[5] > 0) printf("ThreadMapPort still running\n");
1730     if (vnThreadsRunning[6] > 0) printf("ThreadDNSAddressSeed still running\n");
1731     if (vnThreadsRunning[7] > 0) printf("ThreadOpenAddedConnections still running\n");
1732     while (vnThreadsRunning[2] > 0 || vnThreadsRunning[4] > 0)
1733         Sleep(20);
1734     Sleep(50);
1735
1736     return true;
1737 }
1738
1739 class CNetCleanup
1740 {
1741 public:
1742     CNetCleanup()
1743     {
1744     }
1745     ~CNetCleanup()
1746     {
1747         // Close sockets
1748         BOOST_FOREACH(CNode* pnode, vNodes)
1749             if (pnode->hSocket != INVALID_SOCKET)
1750                 closesocket(pnode->hSocket);
1751         if (hListenSocket != INVALID_SOCKET)
1752             if (closesocket(hListenSocket) == SOCKET_ERROR)
1753                 printf("closesocket(hListenSocket) failed with error %d\n", WSAGetLastError());
1754
1755 #ifdef WIN32
1756         // Shutdown Windows Sockets
1757         WSACleanup();
1758 #endif
1759     }
1760 }
1761 instance_of_cnetcleanup;