src/irc.cpp

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Copyright (c) 2009-2012 The Bitcoin developers
   3 // Distributed under the MIT/X11 software license, see the accompanying
   4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5
   6 #include "irc.h"
   7 #include "base58.h"
   8 #include "net.h"
   9
  10 #include <boost/algorithm/string/predicate.hpp> // for startswith() and endswith()
  11
  12 using namespace std;
  13 using namespace boost;
  14
  15 int nGotIRCAddresses = 0;
  16
  17 void ThreadIRCSeed2(void* parg);
  18
  19
  20
  21
  22 #pragma pack(push, 1)
  23 struct ircaddr
  24 {
  25     struct in_addr ip;
  26     unsigned short port;
  27 };
  28 #pragma pack(pop)
  29
  30 string EncodeAddress(const CService& addr)
  31 {
  32     struct ircaddr tmp;
  33     if (addr.GetInAddr(&tmp.ip))
  34     {
  35         tmp.port = htons(addr.GetPort());
  36
  37         vector<unsigned char> vch(UBEGIN(tmp), UEND(tmp));
  38         return string("u") + EncodeBase58Check(vch);
  39     }
  40     return "";
  41 }
  42
  43 bool DecodeAddress(string str, CService& addr)
  44 {
  45     vector<unsigned char> vch;
  46     if (!DecodeBase58Check(str.substr(1), vch))
  47         return false;
  48
  49     struct ircaddr tmp;
  50     if (vch.size() != sizeof(tmp))
  51         return false;
  52     memcpy(&tmp, &vch[0], sizeof(tmp));
  53
  54     addr = CService(tmp.ip, ntohs(tmp.port));
  55     return true;
  56 }
  57
  58
  59
  60
  61
  62
  63 static bool Send(SOCKET hSocket, const char* pszSend)
  64 {
  65     if (strstr(pszSend, "PONG") != pszSend)
  66         printf("IRC SENDING: %s\n", pszSend);
  67     const char* psz = pszSend;
  68     const char* pszEnd = psz + strlen(psz);
  69     while (psz < pszEnd)
  70     {
  71         int ret = send(hSocket, psz, pszEnd - psz, MSG_NOSIGNAL);
  72         if (ret < 0)
  73             return false;
  74         psz += ret;
  75     }
  76     return true;
  77 }
  78
  79 bool RecvLineIRC(SOCKET hSocket, string& strLine)
  80 {
  81     while (true)
  82     {
  83         bool fRet = RecvLine(hSocket, strLine);
  84         if (fRet)
  85         {
  86             if (fShutdown)
  87                 return false;
  88             vector<string> vWords;
  89             ParseString(strLine, ' ', vWords);
  90             if (vWords.size() >= 1 && vWords[0] == "PING")
  91             {
  92                 strLine[1] = 'O';
  93                 strLine += '\r';
  94                 Send(hSocket, strLine.c_str());
  95                 continue;
  96             }
  97         }
  98         return fRet;
  99     }
 100 }
 101
 102 int RecvUntil(SOCKET hSocket, const char* psz1, const char* psz2=NULL, const char* psz3=NULL, const char* psz4=NULL)
 103 {
 104     while (true)
 105     {
 106         string strLine;
 107         strLine.reserve(10000);
 108         if (!RecvLineIRC(hSocket, strLine))
 109             return 0;
 110         printf("IRC %s\n", strLine.c_str());
 111         if (psz1 && strLine.find(psz1) != string::npos)
 112             return 1;
 113         if (psz2 && strLine.find(psz2) != string::npos)
 114             return 2;
 115         if (psz3 && strLine.find(psz3) != string::npos)
 116             return 3;
 117         if (psz4 && strLine.find(psz4) != string::npos)
 118             return 4;
 119     }
 120 }
 121
 122 bool Wait(int nSeconds)
 123 {
 124     if (fShutdown)
 125         return false;
 126     printf("IRC waiting %d seconds to reconnect\n", nSeconds);
 127     for (int i = 0; i < nSeconds; i++)
 128     {
 129         if (fShutdown)
 130             return false;
 131         Sleep(1000);
 132     }
 133     return true;
 134 }
 135
 136 bool RecvCodeLine(SOCKET hSocket, const char* psz1, string& strRet)
 137 {
 138     strRet.clear();
 139     while (true)
 140     {
 141         string strLine;
 142         if (!RecvLineIRC(hSocket, strLine))
 143             return false;
 144
 145         vector<string> vWords;
 146         ParseString(strLine, ' ', vWords);
 147         if (vWords.size() < 2)
 148             continue;
 149
 150         if (vWords[1] == psz1)
 151         {
 152             printf("IRC %s\n", strLine.c_str());
 153             strRet = strLine;
 154             return true;
 155         }
 156     }
 157 }
 158
 159 bool GetIPFromIRC(SOCKET hSocket, string strMyName, CNetAddr& ipRet)
 160 {
 161     Send(hSocket, strprintf("USERHOST %s\r", strMyName.c_str()).c_str());
 162
 163     string strLine;
 164     if (!RecvCodeLine(hSocket, "302", strLine))
 165         return false;
 166
 167     vector<string> vWords;
 168     ParseString(strLine, ' ', vWords);
 169     if (vWords.size() < 4)
 170         return false;
 171
 172     string str = vWords[3];
 173     if (str.rfind("@") == string::npos)
 174         return false;
 175     string strHost = str.substr(str.rfind("@")+1);
 176
 177     // Hybrid IRC used by lfnet always returns IP when you userhost yourself,
 178     // but in case another IRC is ever used this should work.
 179     printf("GetIPFromIRC() got userhost %s\n", strHost.c_str());
 180     CNetAddr addr(strHost, true);
 181     if (!addr.IsValid())
 182         return false;
 183     ipRet = addr;
 184
 185     return true;
 186 }
 187
 188
 189
 190 void ThreadIRCSeed(void* parg)
 191 {
 192     // Make this thread recognisable as the IRC seeding thread
 193     RenameThread("novacoin-ircseed");
 194
 195     printf("ThreadIRCSeed started\n");
 196
 197     try
 198     {
 199         ThreadIRCSeed2(parg);
 200     }
 201     catch (std::exception& e) {
 202         PrintExceptionContinue(&e, "ThreadIRCSeed()");
 203     } catch (...) {
 204         PrintExceptionContinue(NULL, "ThreadIRCSeed()");
 205     }
 206     printf("ThreadIRCSeed exited\n");
 207 }
 208
 209 void ThreadIRCSeed2(void* parg)
 210 {
 211     // Don't connect to IRC if we won't use IPv4 connections.
 212     if (IsLimited(NET_IPV4))
 213         return;
 214
 215     // ... or if we won't make outbound connections and won't accept inbound ones.
 216     if (mapArgs.count("-connect") && fNoListen)
 217         return;
 218
 219     // ... or if IRC is not enabled.
 220     if (!GetBoolArg("-irc", true))
 221         return;
 222
 223     printf("ThreadIRCSeed trying to connect...\n");
 224
 225     int nErrorWait = 10;
 226     int nRetryWait = 10;
 227     int nNameRetry = 0;
 228
 229     while (!fShutdown)
 230     {
 231         const uint16_t nIrcPort = 6667;
 232         CService addrConnect("92.243.23.21", nIrcPort); // irc.lfnet.org
 233
 234         CService addrIRC("irc.lfnet.org", nIrcPort, true);
 235         if (addrIRC.IsValid())
 236             addrConnect = addrIRC;
 237
 238         SOCKET hSocket;
 239         if (!ConnectSocket(addrConnect, hSocket))
 240         {
 241             printf("IRC connect failed\n");
 242             nErrorWait = nErrorWait * 11 / 10;
 243             if (Wait(nErrorWait += 60))
 244                 continue;
 245             else
 246                 return;
 247         }
 248
 249         if (!RecvUntil(hSocket, "Found your hostname", "using your IP address instead", "Couldn't look up your hostname", "ignoring hostname"))
 250         {
 251             CloseSocket(hSocket);
 252             nErrorWait = nErrorWait * 11 / 10;
 253             if (Wait(nErrorWait += 60))
 254                 continue;
 255             else
 256                 return;
 257         }
 258
 259         CNetAddr addrIPv4("1.2.3.4"); // arbitrary IPv4 address to make GetLocal prefer IPv4 addresses
 260         CService addrLocal;
 261         string strMyName;
 262         // Don't use our IP as our nick if we're not listening
 263         // or if it keeps failing because the nick is already in use.
 264         if (!fNoListen && GetLocal(addrLocal, &addrIPv4) && nNameRetry<3)
 265             strMyName = EncodeAddress(GetLocalAddress(&addrConnect));
 266         if (strMyName == "")
 267             strMyName = strprintf("x%" PRIu64 "", GetRand(1000000000));
 268
 269         Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
 270         Send(hSocket, strprintf("USER %s 8 * : %s\r", strMyName.c_str(), strMyName.c_str()).c_str());
 271
 272         int nRet = RecvUntil(hSocket, " 004 ", " 433 ");
 273         if (nRet != 1)
 274         {
 275             CloseSocket(hSocket);
 276             if (nRet == 2)
 277             {
 278                 printf("IRC name already in use\n");
 279                 nNameRetry++;
 280                 Wait(10);
 281                 continue;
 282             }
 283             nErrorWait = nErrorWait * 11 / 10;
 284             if (Wait(nErrorWait += 60))
 285                 continue;
 286             else
 287                 return;
 288         }
 289         nNameRetry = 0;
 290         Sleep(500);
 291
 292         // Get our external IP from the IRC server and re-nick before joining the channel
 293         CNetAddr addrFromIRC;
 294         if (GetIPFromIRC(hSocket, strMyName, addrFromIRC))
 295         {
 296             printf("GetIPFromIRC() returned %s\n", addrFromIRC.ToString().c_str());
 297             // Don't use our IP as our nick if we're not listening
 298             if (!fNoListen && addrFromIRC.IsRoutable())
 299             {
 300                 // IRC lets you to re-nick
 301                 AddLocal(addrFromIRC, LOCAL_IRC);
 302                 strMyName = EncodeAddress(GetLocalAddress(&addrConnect));
 303                 Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
 304             }
 305         }
 306
 307         if (fTestNet) {
 308             Send(hSocket, "JOIN #novacoinTEST2\r");
 309             Send(hSocket, "WHO #novacoinTEST2\r");
 310         } else {
 311             // randomly join #novacoin00-#novacoin05
 312             // int channel_number = GetRandInt(5);
 313
 314             // Channel number is always 0 for initial release
 315             int channel_number = 0;
 316             Send(hSocket, strprintf("JOIN #novacoin%02d\r", channel_number).c_str());
 317             Send(hSocket, strprintf("WHO #novacoin%02d\r", channel_number).c_str());
 318         }
 319
 320         int64_t nStart = GetTime();
 321         string strLine;
 322         strLine.reserve(10000);
 323         while (!fShutdown && RecvLineIRC(hSocket, strLine))
 324         {
 325             if (strLine.empty() || strLine.size() > 900 || strLine[0] != ':')
 326                 continue;
 327
 328             vector<string> vWords;
 329             ParseString(strLine, ' ', vWords);
 330             if (vWords.size() < 2)
 331                 continue;
 332
 333             std::string strName;
 334
 335             if (vWords[1] == "352" && vWords.size() >= 8)
 336             {
 337                 // index 7 is limited to 16 characters
 338                 // could get full length name at index 10, but would be different from join messages
 339                 strName = vWords[7].c_str();
 340                 printf("IRC got who\n");
 341             }
 342
 343             if (vWords[1] == "JOIN" && vWords[0].size() > 1)
 344             {
 345                 // :username!username@50000007.F000000B.90000002.IP JOIN :#channelname
 346                 strName = vWords[0].substr(1, vWords[0].find('!', 1) - 1);
 347                 printf("IRC got join\n");
 348             }
 349
 350             if (boost::algorithm::starts_with(strName, "u"))
 351             {
 352                 CAddress addr;
 353                 if (DecodeAddress(strName, addr))
 354                 {
 355                     addr.nTime = GetAdjustedTime();
 356                     if (addrman.Add(addr, addrConnect, 51 * 60))
 357                         printf("IRC got new address: %s\n", addr.ToString().c_str());
 358                     nGotIRCAddresses++;
 359                 }
 360                 else
 361                 {
 362                     printf("IRC decode failed\n");
 363                 }
 364             }
 365         }
 366         CloseSocket(hSocket);
 367
 368         if (GetTime() - nStart > 20 * 60)
 369         {
 370             nErrorWait /= 3;
 371             nRetryWait /= 3;
 372         }
 373
 374         nRetryWait = nRetryWait * 11 / 10;
 375         if (!Wait(nRetryWait += 60))
 376             return;
 377     }
 378 }
 379
 380
 381
 382
 383
 384
 385
 386
 387
 388
 389 #ifdef TEST
 390 int main(int argc, char *argv[])
 391 {
 392     WSADATA wsadata;
 393     if (WSAStartup(MAKEWORD(2,2), &wsadata) != NO_ERROR)
 394     {
 395         printf("Error at WSAStartup()\n");
 396         return false;
 397     }
 398
 399     ThreadIRCSeed(NULL);
 400
 401     WSACleanup();
 402     return 0;
 403 }
 404 #endif