genesis 1
genesis 2
genesis 3
genesis 4
genesis 5
genesis 6 #include "headers.h"
genesis 7 #include "irc.h"
genesis 8 #include "net.h"
genesis 9 #include "strlcpy.h"
genesis 10
genesis 11 using namespace std;
genesis 12 using namespace boost;
genesis 13
genesis 14 int nGotIRCAddresses = 0;
genesis 15 bool fGotExternalIP = false;
genesis 16
genesis 17 void ThreadIRCSeed2(void* parg);
genesis 18
genesis 19
genesis 20
genesis 21
genesis 22 #pragma pack(push, 1)
genesis 23 struct ircaddr
genesis 24 {
genesis 25 int ip;
genesis 26 short port;
genesis 27 };
genesis 28 #pragma pack(pop)
genesis 29
genesis 30 string EncodeAddress(const CAddress& addr)
genesis 31 {
genesis 32 struct ircaddr tmp;
genesis 33 tmp.ip = addr.ip;
genesis 34 tmp.port = addr.port;
genesis 35
genesis 36 vector<unsigned char> vch(UBEGIN(tmp), UEND(tmp));
genesis 37 return string("u") + EncodeBase58Check(vch);
genesis 38 }
genesis 39
genesis 40 bool DecodeAddress(string str, CAddress& addr)
genesis 41 {
genesis 42 vector<unsigned char> vch;
genesis 43 if (!DecodeBase58Check(str.substr(1), vch))
genesis 44 return false;
genesis 45
genesis 46 struct ircaddr tmp;
genesis 47 if (vch.size() != sizeof(tmp))
genesis 48 return false;
genesis 49 memcpy(&tmp, &vch[0], sizeof(tmp));
genesis 50
genesis 51 addr = CAddress(tmp.ip, ntohs(tmp.port), NODE_NETWORK);
genesis 52 return true;
genesis 53 }
genesis 54
genesis 55
genesis 56
genesis 57
genesis 58
genesis 59
genesis 60 static bool Send(SOCKET hSocket, const char* pszSend)
genesis 61 {
genesis 62 if (strstr(pszSend, "PONG") != pszSend)
genesis 63 printf("IRC SENDING: %s\n", pszSend);
genesis 64 const char* psz = pszSend;
genesis 65 const char* pszEnd = psz + strlen(psz);
genesis 66 while (psz < pszEnd)
genesis 67 {
genesis 68 int ret = send(hSocket, psz, pszEnd - psz, MSG_NOSIGNAL);
genesis 69 if (ret < 0)
genesis 70 return false;
genesis 71 psz += ret;
genesis 72 }
genesis 73 return true;
genesis 74 }
genesis 75
genesis 76 bool RecvLine(SOCKET hSocket, string& strLine)
genesis 77 {
genesis 78 strLine = "";
genesis 79 loop
genesis 80 {
genesis 81 char c;
genesis 82 int nBytes = recv(hSocket, &c, 1, 0);
genesis 83 if (nBytes > 0)
genesis 84 {
genesis 85 if (c == '\n')
genesis 86 continue;
genesis 87 if (c == '\r')
genesis 88 return true;
genesis 89 strLine += c;
genesis 90 if (strLine.size() >= 9000)
genesis 91 return true;
genesis 92 }
genesis 93 else if (nBytes <= 0)
genesis 94 {
genesis 95 if (fShutdown)
genesis 96 return false;
genesis 97 if (nBytes < 0)
genesis 98 {
genesis 99 int nErr = WSAGetLastError();
genesis 100 if (nErr == WSAEMSGSIZE)
genesis 101 continue;
genesis 102 if (nErr == WSAEWOULDBLOCK || nErr == WSAEINTR || nErr == WSAEINPROGRESS)
genesis 103 {
genesis 104 Sleep(10);
genesis 105 continue;
genesis 106 }
genesis 107 }
genesis 108 if (!strLine.empty())
genesis 109 return true;
genesis 110 if (nBytes == 0)
genesis 111 {
genesis 112
genesis 113 printf("socket closed\n");
genesis 114 return false;
genesis 115 }
genesis 116 else
genesis 117 {
genesis 118
genesis 119 int nErr = WSAGetLastError();
genesis 120 printf("recv failed: %d\n", nErr);
genesis 121 return false;
genesis 122 }
genesis 123 }
genesis 124 }
genesis 125 }
genesis 126
genesis 127 bool RecvLineIRC(SOCKET hSocket, string& strLine)
genesis 128 {
genesis 129 loop
genesis 130 {
genesis 131 bool fRet = RecvLine(hSocket, strLine);
genesis 132 if (fRet)
genesis 133 {
genesis 134 if (fShutdown)
genesis 135 return false;
genesis 136 vector<string> vWords;
genesis 137 ParseString(strLine, ' ', vWords);
genesis 138 if (vWords.size() >= 1 && vWords[0] == "PING")
genesis 139 {
genesis 140 strLine[1] = 'O';
genesis 141 strLine += '\r';
genesis 142 Send(hSocket, strLine.c_str());
genesis 143 continue;
genesis 144 }
genesis 145 }
genesis 146 return fRet;
genesis 147 }
genesis 148 }
genesis 149
genesis 150 int RecvUntil(SOCKET hSocket, const char* psz1, const char* psz2=NULL, const char* psz3=NULL, const char* psz4=NULL)
genesis 151 {
genesis 152 loop
genesis 153 {
genesis 154 string strLine;
genesis 155 strLine.reserve(10000);
genesis 156 if (!RecvLineIRC(hSocket, strLine))
genesis 157 return 0;
genesis 158 printf("IRC %s\n", strLine.c_str());
genesis 159 if (psz1 && strLine.find(psz1) != -1)
genesis 160 return 1;
genesis 161 if (psz2 && strLine.find(psz2) != -1)
genesis 162 return 2;
genesis 163 if (psz3 && strLine.find(psz3) != -1)
genesis 164 return 3;
genesis 165 if (psz4 && strLine.find(psz4) != -1)
genesis 166 return 4;
genesis 167 }
genesis 168 }
genesis 169
genesis 170 bool Wait(int nSeconds)
genesis 171 {
genesis 172 if (fShutdown)
genesis 173 return false;
genesis 174 printf("IRC waiting %d seconds to reconnect\n", nSeconds);
genesis 175 for (int i = 0; i < nSeconds; i++)
genesis 176 {
genesis 177 if (fShutdown)
genesis 178 return false;
genesis 179 Sleep(1000);
genesis 180 }
genesis 181 return true;
genesis 182 }
genesis 183
genesis 184 bool RecvCodeLine(SOCKET hSocket, const char* psz1, string& strRet)
genesis 185 {
genesis 186 strRet.clear();
genesis 187 loop
genesis 188 {
genesis 189 string strLine;
genesis 190 if (!RecvLineIRC(hSocket, strLine))
genesis 191 return false;
genesis 192
genesis 193 vector<string> vWords;
genesis 194 ParseString(strLine, ' ', vWords);
genesis 195 if (vWords.size() < 2)
genesis 196 continue;
genesis 197
genesis 198 if (vWords[1] == psz1)
genesis 199 {
genesis 200 printf("IRC %s\n", strLine.c_str());
genesis 201 strRet = strLine;
genesis 202 return true;
genesis 203 }
genesis 204 }
genesis 205 }
genesis 206
genesis 207 bool GetIPFromIRC(SOCKET hSocket, string strMyName, unsigned int& ipRet)
genesis 208 {
genesis 209 Send(hSocket, strprintf("USERHOST %s\r", strMyName.c_str()).c_str());
genesis 210
genesis 211 string strLine;
genesis 212 if (!RecvCodeLine(hSocket, "302", strLine))
genesis 213 return false;
genesis 214
genesis 215 vector<string> vWords;
genesis 216 ParseString(strLine, ' ', vWords);
genesis 217 if (vWords.size() < 4)
genesis 218 return false;
genesis 219
genesis 220 string str = vWords[3];
genesis 221 if (str.rfind("@") == string::npos)
genesis 222 return false;
genesis 223 string strHost = str.substr(str.rfind("@")+1);
genesis 224
genesis 225
genesis 226
genesis 227 printf("GetIPFromIRC() got userhost %s\n", strHost.c_str());
genesis 228 if (fUseProxy)
genesis 229 return false;
genesis 230 CAddress addr(strHost, 0, true);
genesis 231 if (!addr.IsValid())
genesis 232 return false;
genesis 233 ipRet = addr.ip;
genesis 234
genesis 235 return true;
genesis 236 }
genesis 237
genesis 238
genesis 239
genesis 240 void ThreadIRCSeed(void* parg)
genesis 241 {
genesis 242 IMPLEMENT_RANDOMIZE_STACK(ThreadIRCSeed(parg));
genesis 243 try
genesis 244 {
genesis 245 ThreadIRCSeed2(parg);
genesis 246 }
genesis 247 catch (std::exception& e) {
genesis 248 PrintExceptionContinue(&e, "ThreadIRCSeed()");
genesis 249 } catch (...) {
genesis 250 PrintExceptionContinue(NULL, "ThreadIRCSeed()");
genesis 251 }
genesis 252 printf("ThreadIRCSeed exiting\n");
genesis 253 }
genesis 254
genesis 255 void ThreadIRCSeed2(void* parg)
genesis 256 {
genesis 257
genesis 258 if (mapArgs.count("-connect") || fNoListen)
genesis 259 return;
genesis 260
genesis 261 if (GetBoolArg("-noirc"))
genesis 262 return;
genesis 263 printf("ThreadIRCSeed started\n");
genesis 264 int nErrorWait = 10;
genesis 265 int nRetryWait = 10;
genesis 266 bool fNameInUse = false;
genesis 267
genesis 268 while (!fShutdown)
genesis 269 {
genesis 270 CAddress addrConnect("92.243.23.21", 6667);
genesis 271
genesis 272 SOCKET hSocket;
genesis 273 if (!ConnectSocket(addrConnect, hSocket))
genesis 274 {
genesis 275 printf("IRC connect failed\n");
genesis 276 nErrorWait = nErrorWait * 11 / 10;
genesis 277 if (Wait(nErrorWait += 60))
genesis 278 continue;
genesis 279 else
genesis 280 return;
genesis 281 }
genesis 282
genesis 283 if (!RecvUntil(hSocket, "Found your hostname", "using your IP address instead", "Couldn't look up your hostname", "ignoring hostname"))
genesis 284 {
genesis 285 closesocket(hSocket);
genesis 286 hSocket = INVALID_SOCKET;
genesis 287 nErrorWait = nErrorWait * 11 / 10;
genesis 288 if (Wait(nErrorWait += 60))
genesis 289 continue;
genesis 290 else
genesis 291 return;
genesis 292 }
genesis 293
genesis 294 string strMyName;
genesis 295 if (addrLocalHost.IsRoutable() && !fUseProxy && !fNameInUse)
genesis 296 strMyName = EncodeAddress(addrLocalHost);
genesis 297 else
genesis 298 strMyName = strprintf("x%u", GetRand(1000000000));
genesis 299
genesis 300 Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
genesis 301 Send(hSocket, strprintf("USER %s 8 * : %s\r", strMyName.c_str(), strMyName.c_str()).c_str());
genesis 302
genesis 303 int nRet = RecvUntil(hSocket, " 004 ", " 433 ");
genesis 304 if (nRet != 1)
genesis 305 {
genesis 306 closesocket(hSocket);
genesis 307 hSocket = INVALID_SOCKET;
genesis 308 if (nRet == 2)
genesis 309 {
genesis 310 printf("IRC name already in use\n");
genesis 311 fNameInUse = true;
genesis 312 Wait(10);
genesis 313 continue;
genesis 314 }
genesis 315 nErrorWait = nErrorWait * 11 / 10;
genesis 316 if (Wait(nErrorWait += 60))
genesis 317 continue;
genesis 318 else
genesis 319 return;
genesis 320 }
genesis 321 Sleep(500);
genesis 322
genesis 323
genesis 324 CAddress addrFromIRC;
genesis 325 if (GetIPFromIRC(hSocket, strMyName, addrFromIRC.ip))
genesis 326 {
genesis 327 printf("GetIPFromIRC() returned %s\n", addrFromIRC.ToStringIP().c_str());
genesis 328 if (!fUseProxy && addrFromIRC.IsRoutable())
genesis 329 {
genesis 330
genesis 331 fGotExternalIP = true;
genesis 332 addrLocalHost.ip = addrFromIRC.ip;
genesis 333 strMyName = EncodeAddress(addrLocalHost);
genesis 334 Send(hSocket, strprintf("NICK %s\r", strMyName.c_str()).c_str());
genesis 335 }
genesis 336 }
genesis 337
genesis 338 if (fTestNet) {
genesis 339 Send(hSocket, "JOIN #bitcoinTEST\r");
genesis 340 Send(hSocket, "WHO #bitcoinTEST\r");
genesis 341 } else {
genesis 342
genesis 343 int channel_number = GetRandInt(100);
genesis 344 Send(hSocket, strprintf("JOIN #bitcoin%02d\r", channel_number).c_str());
genesis 345 Send(hSocket, strprintf("WHO #bitcoin%02d\r", channel_number).c_str());
genesis 346 }
genesis 347
genesis 348 int64 nStart = GetTime();
genesis 349 string strLine;
genesis 350 strLine.reserve(10000);
genesis 351 while (!fShutdown && RecvLineIRC(hSocket, strLine))
genesis 352 {
genesis 353 if (strLine.empty() || strLine.size() > 900 || strLine[0] != ':')
genesis 354 continue;
genesis 355
genesis 356 vector<string> vWords;
genesis 357 ParseString(strLine, ' ', vWords);
genesis 358 if (vWords.size() < 2)
genesis 359 continue;
genesis 360
genesis 361 char pszName[10000];
genesis 362 pszName[0] = '\0';
genesis 363
genesis 364 if (vWords[1] == "352" && vWords.size() >= 8)
genesis 365 {
genesis 366
genesis 367
genesis 368 strlcpy(pszName, vWords[7].c_str(), sizeof(pszName));
genesis 369 printf("IRC got who\n");
genesis 370 }
genesis 371
genesis 372 if (vWords[1] == "JOIN" && vWords[0].size() > 1)
genesis 373 {
genesis 374
genesis 375 strlcpy(pszName, vWords[0].c_str() + 1, sizeof(pszName));
genesis 376 if (strchr(pszName, '!'))
genesis 377 *strchr(pszName, '!') = '\0';
genesis 378 printf("IRC got join\n");
genesis 379 }
genesis 380
genesis 381 if (pszName[0] == 'u')
genesis 382 {
genesis 383 CAddress addr;
genesis 384 if (DecodeAddress(pszName, addr))
genesis 385 {
genesis 386 addr.nTime = GetAdjustedTime();
genesis 387 if (AddAddress(addr, 51 * 60))
genesis 388 printf("IRC got new address: %s\n", addr.ToString().c_str());
genesis 389 nGotIRCAddresses++;
genesis 390 }
genesis 391 else
genesis 392 {
genesis 393 printf("IRC decode failed\n");
genesis 394 }
genesis 395 }
genesis 396 }
genesis 397 closesocket(hSocket);
genesis 398 hSocket = INVALID_SOCKET;
genesis 399
genesis 400 if (GetTime() - nStart > 20 * 60)
genesis 401 {
genesis 402 nErrorWait /= 3;
genesis 403 nRetryWait /= 3;
genesis 404 }
genesis 405
genesis 406 nRetryWait = nRetryWait * 11 / 10;
genesis 407 if (!Wait(nRetryWait += 60))
genesis 408 return;
genesis 409 }
genesis 410 }
genesis 411
genesis 412
genesis 413
genesis 414
genesis 415
genesis 416
genesis 417
genesis 418
genesis 419
genesis 420
genesis 421 #ifdef TEST
genesis 422 int main(int argc, char *argv[])
genesis 423 {
genesis 424 WSADATA wsadata;
genesis 425 if (WSAStartup(MAKEWORD(2,2), &wsadata) != NO_ERROR)
genesis 426 {
genesis 427 printf("Error at WSAStartup()\n");
genesis 428 return false;
genesis 429 }
genesis 430
genesis 431 ThreadIRCSeed(NULL);
genesis 432
genesis 433 WSACleanup();
genesis 434 return 0;
genesis 435 }
genesis 436 #endif