Changeset View
Changeset View
Standalone View
Standalone View
src/seeder/bitcoin.cpp
Show All 36 Lines | class CSeederNode { | ||||
void BeginMessage(const char *pszCommand) { | void BeginMessage(const char *pszCommand) { | ||||
if (nHeaderStart != allones) { | if (nHeaderStart != allones) { | ||||
AbortMessage(); | AbortMessage(); | ||||
} | } | ||||
nHeaderStart = vSend.size(); | nHeaderStart = vSend.size(); | ||||
vSend << CMessageHeader(netMagic, pszCommand, 0); | vSend << CMessageHeader(netMagic, pszCommand, 0); | ||||
nMessageStart = vSend.size(); | nMessageStart = vSend.size(); | ||||
// printf("%s: SEND %s\n", ToString(you).c_str(), pszCommand); | // fprintf(stdout, "%s: SEND %s\n", ToString(you).c_str(), pszCommand); | ||||
} | } | ||||
void AbortMessage() { | void AbortMessage() { | ||||
if (nHeaderStart == allones) { | if (nHeaderStart == allones) { | ||||
return; | return; | ||||
} | } | ||||
vSend.resize(nHeaderStart); | vSend.resize(nHeaderStart); | ||||
nHeaderStart = allones; | nHeaderStart = allones; | ||||
▲ Show 20 Lines • Show All 48 Lines • ▼ Show 20 Lines | void PushVersion() { | ||||
int nBestHeight = GetRequireHeight(); | int nBestHeight = GetRequireHeight(); | ||||
std::string ver = "/bitcoin-cash-seeder:0.15/"; | std::string ver = "/bitcoin-cash-seeder:0.15/"; | ||||
vSend << PROTOCOL_VERSION << nLocalServices << nTime << you << me | vSend << PROTOCOL_VERSION << nLocalServices << nTime << you << me | ||||
<< nLocalNonce << ver << nBestHeight; | << nLocalNonce << ver << nBestHeight; | ||||
EndMessage(); | EndMessage(); | ||||
} | } | ||||
void GotVersion() { | void GotVersion() { | ||||
// printf("\n%s: version %i\n", ToString(you).c_str(), nVersion); | // fprintf(stdout, "\n%s: version %i\n", ToString(you).c_str(), | ||||
// nVersion); | |||||
if (vAddr) { | if (vAddr) { | ||||
BeginMessage("getaddr"); | BeginMessage("getaddr"); | ||||
EndMessage(); | EndMessage(); | ||||
doneAfter = time(nullptr) + GetTimeout(); | doneAfter = time(nullptr) + GetTimeout(); | ||||
} else { | } else { | ||||
doneAfter = time(nullptr) + 1; | doneAfter = time(nullptr) + 1; | ||||
} | } | ||||
} | } | ||||
bool ProcessMessage(std::string strCommand, CDataStream &recv) { | bool ProcessMessage(std::string strCommand, CDataStream &recv) { | ||||
// printf("%s: RECV %s\n", ToString(you).c_str(), strCommand.c_str()); | // fprintf(stdout, "%s: RECV %s\n", ToString(you).c_str(), | ||||
// strCommand.c_str()); | |||||
if (strCommand == "version") { | if (strCommand == "version") { | ||||
int64_t nTime; | int64_t nTime; | ||||
CAddress addrMe; | CAddress addrMe; | ||||
CAddress addrFrom; | CAddress addrFrom; | ||||
uint64_t nNonce = 1; | uint64_t nNonce = 1; | ||||
uint64_t nServiceInt; | uint64_t nServiceInt; | ||||
recv >> nVersion >> nServiceInt >> nTime >> addrMe; | recv >> nVersion >> nServiceInt >> nTime >> addrMe; | ||||
you.nServices = ServiceFlags(nServiceInt); | you.nServices = ServiceFlags(nServiceInt); | ||||
Show All 18 Lines | bool ProcessMessage(std::string strCommand, CDataStream &recv) { | ||||
vRecv.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | vRecv.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | ||||
GotVersion(); | GotVersion(); | ||||
return false; | return false; | ||||
} | } | ||||
if (strCommand == "addr" && vAddr) { | if (strCommand == "addr" && vAddr) { | ||||
std::vector<CAddress> vAddrNew; | std::vector<CAddress> vAddrNew; | ||||
recv >> vAddrNew; | recv >> vAddrNew; | ||||
// printf("%s: got %i addresses\n", ToString(you).c_str(), | // fprintf(stdout, "%s: got %i addresses\n", ToString(you).c_str(), | ||||
// (int)vAddrNew.size()); | // (int)vAddrNew.size()); | ||||
int64_t now = time(nullptr); | int64_t now = time(nullptr); | ||||
std::vector<CAddress>::iterator it = vAddrNew.begin(); | std::vector<CAddress>::iterator it = vAddrNew.begin(); | ||||
if (vAddrNew.size() > 1) { | if (vAddrNew.size() > 1) { | ||||
if (doneAfter == 0 || doneAfter > now + 1) doneAfter = now + 1; | if (doneAfter == 0 || doneAfter > now + 1) doneAfter = now + 1; | ||||
} | } | ||||
while (it != vAddrNew.end()) { | while (it != vAddrNew.end()) { | ||||
CAddress &addr = *it; | CAddress &addr = *it; | ||||
// printf("%s: got address %s\n", ToString(you).c_str(), | // fprintf(stdout, "%s: got address %s\n", | ||||
// ToString(you).c_str(), | |||||
// addr.ToString().c_str(), (int)(vAddr->size())); | // addr.ToString().c_str(), (int)(vAddr->size())); | ||||
it++; | it++; | ||||
if (addr.nTime <= 100000000 || addr.nTime > now + 600) { | if (addr.nTime <= 100000000 || addr.nTime > now + 600) { | ||||
addr.nTime = now - 5 * 86400; | addr.nTime = now - 5 * 86400; | ||||
} | } | ||||
if (addr.nTime > now - 604800) { | if (addr.nTime > now - 604800) { | ||||
vAddr->push_back(addr); | vAddr->push_back(addr); | ||||
} | } | ||||
// printf("%s: added address %s (#%i)\n", ToString(you).c_str(), | // fprintf(stdout, "%s: added address %s (#%i)\n", | ||||
// ToString(you).c_str(), | |||||
// addr.ToString().c_str(), (int)(vAddr->size())); | // addr.ToString().c_str(), (int)(vAddr->size())); | ||||
if (vAddr->size() > 1000) { | if (vAddr->size() > 1000) { | ||||
doneAfter = 1; | doneAfter = 1; | ||||
return true; | return true; | ||||
} | } | ||||
} | } | ||||
return false; | return false; | ||||
} | } | ||||
Show All 18 Lines | bool ProcessMessages() { | ||||
break; | break; | ||||
} | } | ||||
vRecv.erase(vRecv.begin(), pstart); | vRecv.erase(vRecv.begin(), pstart); | ||||
std::vector<char> vHeaderSave(vRecv.begin(), | std::vector<char> vHeaderSave(vRecv.begin(), | ||||
vRecv.begin() + nHeaderSize); | vRecv.begin() + nHeaderSize); | ||||
CMessageHeader hdr(netMagic); | CMessageHeader hdr(netMagic); | ||||
vRecv >> hdr; | vRecv >> hdr; | ||||
if (!hdr.IsValidWithoutConfig(netMagic)) { | if (!hdr.IsValidWithoutConfig(netMagic)) { | ||||
// printf("%s: BAD (invalid header)\n", ToString(you).c_str()); | // fprintf(stdout, "%s: BAD (invalid header)\n", | ||||
// ToString(you).c_str()); | |||||
ban = 100000; | ban = 100000; | ||||
return true; | return true; | ||||
} | } | ||||
std::string strCommand = hdr.GetCommand(); | std::string strCommand = hdr.GetCommand(); | ||||
unsigned int nMessageSize = hdr.nMessageSize; | unsigned int nMessageSize = hdr.nMessageSize; | ||||
if (nMessageSize > MAX_SIZE) { | if (nMessageSize > MAX_SIZE) { | ||||
// printf("%s: BAD (message too large)\n", | // fprintf(stdout, "%s: BAD (message too large)\n", | ||||
// ToString(you).c_str()); | // ToString(you).c_str()); | ||||
ban = 100000; | ban = 100000; | ||||
return true; | return true; | ||||
} | } | ||||
if (nMessageSize > vRecv.size()) { | if (nMessageSize > vRecv.size()) { | ||||
vRecv.insert(vRecv.begin(), vHeaderSave.begin(), | vRecv.insert(vRecv.begin(), vHeaderSave.begin(), | ||||
vHeaderSave.end()); | vHeaderSave.end()); | ||||
break; | break; | ||||
} | } | ||||
if (vRecv.GetVersion() >= 209) { | if (vRecv.GetVersion() >= 209) { | ||||
uint256 hash = | uint256 hash = | ||||
Hash(vRecv.begin(), vRecv.begin() + nMessageSize); | Hash(vRecv.begin(), vRecv.begin() + nMessageSize); | ||||
if (memcmp(hash.begin(), hdr.pchChecksum, | if (memcmp(hash.begin(), hdr.pchChecksum, | ||||
CMessageHeader::CHECKSUM_SIZE) != 0) { | CMessageHeader::CHECKSUM_SIZE) != 0) { | ||||
continue; | continue; | ||||
} | } | ||||
} | } | ||||
CDataStream vMsg(vRecv.begin(), vRecv.begin() + nMessageSize, | CDataStream vMsg(vRecv.begin(), vRecv.begin() + nMessageSize, | ||||
vRecv.GetType(), vRecv.GetVersion()); | vRecv.GetType(), vRecv.GetVersion()); | ||||
vRecv.ignore(nMessageSize); | vRecv.ignore(nMessageSize); | ||||
if (ProcessMessage(strCommand, vMsg)) { | if (ProcessMessage(strCommand, vMsg)) { | ||||
return true; | return true; | ||||
} | } | ||||
// printf("%s: done processing %s\n", ToString(you).c_str(), | // fprintf(stdout, "%s: done processing %s\n", | ||||
// ToString(you).c_str(), | |||||
// strCommand.c_str()); | // strCommand.c_str()); | ||||
} while (1); | } while (1); | ||||
return false; | return false; | ||||
} | } | ||||
public: | public: | ||||
CSeederNode(const CService &ip, std::vector<CAddress> *vAddrIn) | CSeederNode(const CService &ip, std::vector<CAddress> *vAddrIn) | ||||
: sock(INVALID_SOCKET), vSend(SER_NETWORK, 0), vRecv(SER_NETWORK, 0), | : sock(INVALID_SOCKET), vSend(SER_NETWORK, 0), vRecv(SER_NETWORK, 0), | ||||
Show All 31 Lines | bool Run() { | ||||
} | } | ||||
// no proxy needed (none set for target network) | // no proxy needed (none set for target network) | ||||
connected = | connected = | ||||
ConnectSocketDirectly(you, sock, nConnectTimeout, true); | ConnectSocketDirectly(you, sock, nConnectTimeout, true); | ||||
} | } | ||||
} | } | ||||
if (!connected) { | if (!connected) { | ||||
// printf("Cannot connect to %s\n", ToString(you).c_str()); | // fprintf(stdout, "Cannot connect to %s\n", ToString(you).c_str()); | ||||
CloseSocket(sock); | CloseSocket(sock); | ||||
return false; | return false; | ||||
} | } | ||||
PushVersion(); | PushVersion(); | ||||
Send(); | Send(); | ||||
bool res = true; | bool res = true; | ||||
Show All 19 Lines | bool Run() { | ||||
break; | break; | ||||
} | } | ||||
int nBytes = recv(sock, pchBuf, sizeof(pchBuf), 0); | int nBytes = recv(sock, pchBuf, sizeof(pchBuf), 0); | ||||
int nPos = vRecv.size(); | int nPos = vRecv.size(); | ||||
if (nBytes > 0) { | if (nBytes > 0) { | ||||
vRecv.resize(nPos + nBytes); | vRecv.resize(nPos + nBytes); | ||||
memcpy(&vRecv[nPos], pchBuf, nBytes); | memcpy(&vRecv[nPos], pchBuf, nBytes); | ||||
} else if (nBytes == 0) { | } else if (nBytes == 0) { | ||||
// printf("%s: BAD (connection closed prematurely)\n", | // fprintf(stdout, "%s: BAD (connection closed prematurely)\n", | ||||
// ToString(you).c_str()); | // ToString(you).c_str()); | ||||
res = false; | res = false; | ||||
break; | break; | ||||
} else { | } else { | ||||
// printf("%s: BAD (connection error)\n", | // fprintf(stdout, "%s: BAD (connection error)\n", | ||||
// ToString(you).c_str()); | // ToString(you).c_str()); | ||||
res = false; | res = false; | ||||
break; | break; | ||||
} | } | ||||
ProcessMessages(); | ProcessMessages(); | ||||
Send(); | Send(); | ||||
} | } | ||||
if (sock == INVALID_SOCKET) res = false; | if (sock == INVALID_SOCKET) res = false; | ||||
Show All 20 Lines | try { | ||||
if (!ret) { | if (!ret) { | ||||
ban = node.GetBan(); | ban = node.GetBan(); | ||||
} else { | } else { | ||||
ban = 0; | ban = 0; | ||||
} | } | ||||
clientV = node.GetClientVersion(); | clientV = node.GetClientVersion(); | ||||
clientSV = node.GetClientSubVersion(); | clientSV = node.GetClientSubVersion(); | ||||
blocks = node.GetStartingHeight(); | blocks = node.GetStartingHeight(); | ||||
// printf("%s: %s!!!\n", cip.ToString().c_str(), ret ? "GOOD" : "BAD"); | // fprintf(stdout, "%s: %s!!!\n", cip.ToString().c_str(), ret ? "GOOD" : | ||||
// "BAD"); | |||||
return ret; | return ret; | ||||
} catch (std::ios_base::failure &e) { | } catch (std::ios_base::failure &e) { | ||||
ban = 0; | ban = 0; | ||||
return false; | return false; | ||||
} | } | ||||
} | } |