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     try
 196     {
 197         ThreadIRCSeed2(parg);
 198     }
 199     catch (std::exception& e) {
 200         PrintExceptionContinue(&e, "ThreadIRCSeed()");
 201     } catch (...) {
 202         PrintExceptionContinue(NULL, "ThreadIRCSeed()");
 203     }
 204     printf("ThreadIRCSeed exited\n");
 205 }
 206
 207 void ThreadIRCSeed2(void* parg)
 208 {
 209     // Don't connect to IRC if we won't use IPv4 connections.
 210     if (IsLimited(NET_IPV4))
 211         return;
 212
 213     // ... or if we won't make outbound connections and won't accept inbound ones.
 214     if (mapArgs.count("-connect") && fNoListen)
 215         return;
 216
 217     // ... or if IRC is not enabled.
 218     if (!GetBoolArg("-irc", true))
 219         return;
 220
 221     printf("ThreadIRCSeed started\n");
 222     int nErrorWait = 10;
 223     int nRetryWait = 10;
 224     int nNameRetry = 0;
 225
 226     while (!fShutdown)
 227     {
 228         CService addrConnect("92.243.23.21", 6667); // irc.lfnet.org
 229
 230         CService addrIRC("irc.lfnet.org", 6667, true);
 231         if (addrIRC.IsValid())
 232             addrConnect = addrIRC;
 233
 234         SOCKET hSocket;
 235         if (!ConnectSocket(addrConnect, hSocket))
 236         {
 237             printf("IRC connect failed\n");
 238             nErrorWait = nErrorWait * 11 / 10;
 239             if (Wait(nErrorWait += 60))
 240                 continue;
 241             else
 242                 return;
 243         }
 244
 245         if (!RecvUntil(hSocket, "Found your hostname", "using your IP address instead", "Couldn't look up your hostname", "ignoring hostname"))
 246         {
 247             closesocket(hSocket);
 248             hSocket = INVALID_SOCKET;
 249             nErrorWait = nErrorWait * 11 / 10;
 250             if (Wait(nErrorWait += 60))
 251                 continue;
 252             else
 253                 return;
 254         }
 255
 256         CNetAddr addrIPv4("1.2.3.4"); // arbitrary IPv4 address to make GetLocal prefer IPv4 addresses
 257         CService addrLocal;
 258         string strMyName;
 259         // Don't use our IP as our nick if we're not listening
 260         // or if it keeps failing because the nick is already in use.
 261         if (!fNoListen && GetLocal(addrLocal, &addrIPv4) && nNameRetry<3)
 262             strMyName = EncodeAddress(GetLocalAddress(&addrConnect));
 263         if (strMyName == "")
 264             strMyName = strprintf("x%" PRIu64 "", GetRand(1000000000));
 265
 266         Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
 267         Send(hSocket, strprintf("USER %s 8 * : %s\r", strMyName.c_str(), strMyName.c_str()).c_str());
 268
 269         int nRet = RecvUntil(hSocket, " 004 ", " 433 ");
 270         if (nRet != 1)
 271         {
 272             closesocket(hSocket);
 273             hSocket = INVALID_SOCKET;
 274             if (nRet == 2)
 275             {
 276                 printf("IRC name already in use\n");
 277                 nNameRetry++;
 278                 Wait(10);
 279                 continue;
 280             }
 281             nErrorWait = nErrorWait * 11 / 10;
 282             if (Wait(nErrorWait += 60))
 283                 continue;
 284             else
 285                 return;
 286         }
 287         nNameRetry = 0;
 288         Sleep(500);
 289
 290         // Get our external IP from the IRC server and re-nick before joining the channel
 291         CNetAddr addrFromIRC;
 292         if (GetIPFromIRC(hSocket, strMyName, addrFromIRC))
 293         {
 294             printf("GetIPFromIRC() returned %s\n", addrFromIRC.ToString().c_str());
 295             // Don't use our IP as our nick if we're not listening
 296             if (!fNoListen && addrFromIRC.IsRoutable())
 297             {
 298                 // IRC lets you to re-nick
 299                 AddLocal(addrFromIRC, LOCAL_IRC);
 300                 strMyName = EncodeAddress(GetLocalAddress(&addrConnect));
 301                 Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
 302             }
 303         }
 304
 305         if (fTestNet) {
 306             Send(hSocket, "JOIN #novacoinTEST2\r");
 307             Send(hSocket, "WHO #novacoinTEST2\r");
 308         } else {
 309             // randomly join #novacoin00-#novacoin05
 310             // int channel_number = GetRandInt(5);
 311
 312             // Channel number is always 0 for initial release
 313             int channel_number = 0;
 314             Send(hSocket, strprintf("JOIN #novacoin%02d\r", channel_number).c_str());
 315             Send(hSocket, strprintf("WHO #novacoin%02d\r", channel_number).c_str());
 316         }
 317
 318         int64_t nStart = GetTime();
 319         string strLine;
 320         strLine.reserve(10000);
 321         while (!fShutdown && RecvLineIRC(hSocket, strLine))
 322         {
 323             if (strLine.empty() || strLine.size() > 900 || strLine[0] != ':')
 324                 continue;
 325
 326             vector<string> vWords;
 327             ParseString(strLine, ' ', vWords);
 328             if (vWords.size() < 2)
 329                 continue;
 330
 331             std::string strName;
 332
 333             if (vWords[1] == "352" && vWords.size() >= 8)
 334             {
 335                 // index 7 is limited to 16 characters
 336                 // could get full length name at index 10, but would be different from join messages
 337                 strName = vWords[7].c_str();
 338                 printf("IRC got who\n");
 339             }
 340
 341             if (vWords[1] == "JOIN" && vWords[0].size() > 1)
 342             {
 343                 // :username!username@50000007.F000000B.90000002.IP JOIN :#channelname
 344                 strName = vWords[0].substr(1, vWords[0].find('!', 1) - 1);
 345                 printf("IRC got join\n");
 346             }
 347
 348             if (boost::algorithm::starts_with(strName, "u"))
 349             {
 350                 CAddress addr;
 351                 if (DecodeAddress(strName, addr))
 352                 {
 353                     addr.nTime = GetAdjustedTime();
 354                     if (addrman.Add(addr, addrConnect, 51 * 60))
 355                         printf("IRC got new address: %s\n", addr.ToString().c_str());
 356                     nGotIRCAddresses++;
 357                 }
 358                 else
 359                 {
 360                     printf("IRC decode failed\n");
 361                 }
 362             }
 363         }
 364         closesocket(hSocket);
 365         hSocket = INVALID_SOCKET;
 366
 367         if (GetTime() - nStart > 20 * 60)
 368         {
 369             nErrorWait /= 3;
 370             nRetryWait /= 3;
 371         }
 372
 373         nRetryWait = nRetryWait * 11 / 10;
 374         if (!Wait(nRetryWait += 60))
 375             return;
 376     }
 377 }
 378
 379
 380
 381
 382
 383
 384
 385
 386
 387
 388 #ifdef TEST
 389 int main(int argc, char *argv[])
 390 {
 391     WSADATA wsadata;
 392     if (WSAStartup(MAKEWORD(2,2), &wsadata) != NO_ERROR)
 393     {
 394         printf("Error at WSAStartup()\n");
 395         return false;
 396     }
 397
 398     ThreadIRCSeed(NULL);
 399
 400     WSACleanup();
 401     return 0;
 402 }
 403 #endif