Changeset View
Changeset View
Standalone View
Standalone View
src/seeder/bitcoin.cpp
Show All 20 Lines | |||||
void CSeederNode::BeginMessage(const char *pszCommand) { | void CSeederNode::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(); | ||||
// tfm::format(std::cout, "%s: SEND %s\n", ToString(you).c_str(), | // tfm::format(std::cout, "%s: SEND %s\n", ToString(you), | ||||
// pszCommand); | // pszCommand); | ||||
} | } | ||||
void CSeederNode::AbortMessage() { | void CSeederNode::AbortMessage() { | ||||
if (nHeaderStart == allones) { | if (nHeaderStart == allones) { | ||||
return; | return; | ||||
} | } | ||||
vSend.resize(nHeaderStart); | vSend.resize(nHeaderStart); | ||||
▲ Show 20 Lines • Show All 50 Lines • ▼ Show 20 Lines | void CSeederNode::PushVersion() { | ||||
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(); | ||||
} | } | ||||
PeerMessagingState CSeederNode::ProcessMessage(std::string strCommand, | PeerMessagingState CSeederNode::ProcessMessage(std::string strCommand, | ||||
CDataStream &recv) { | CDataStream &recv) { | ||||
// tfm::format(std::cout, "%s: RECV %s\n", ToString(you).c_str(), | // tfm::format(std::cout, "%s: RECV %s\n", ToString(you), | ||||
// strCommand.c_str()); | // strCommand); | ||||
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); | ||||
recv >> addrFrom >> nNonce; | recv >> addrFrom >> nNonce; | ||||
recv >> strSubVer; | recv >> strSubVer; | ||||
recv >> nStartingHeight; | recv >> nStartingHeight; | ||||
BeginMessage("verack"); | BeginMessage("verack"); | ||||
EndMessage(); | EndMessage(); | ||||
vSend.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | vSend.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | ||||
return PeerMessagingState::AwaitingMessages; | return PeerMessagingState::AwaitingMessages; | ||||
} | } | ||||
if (strCommand == "verack") { | if (strCommand == "verack") { | ||||
vRecv.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | vRecv.SetVersion(std::min(nVersion, PROTOCOL_VERSION)); | ||||
// tfm::format(std::cout, "\n%s: version %i\n", ToString(you).c_str(), | // tfm::format(std::cout, "\n%s: version %i\n", ToString(you), | ||||
// nVersion); | // 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; | ||||
} | } | ||||
return PeerMessagingState::AwaitingMessages; | return PeerMessagingState::AwaitingMessages; | ||||
} | } | ||||
if (strCommand == "addr" && vAddr) { | if (strCommand == "addr" && vAddr) { | ||||
std::vector<CAddress> vAddrNew; | std::vector<CAddress> vAddrNew; | ||||
recv >> vAddrNew; | recv >> vAddrNew; | ||||
// tfm::format(std::cout, "%s: got %i addresses\n", | // tfm::format(std::cout, "%s: got %i addresses\n", | ||||
// ToString(you).c_str(), | // ToString(you), | ||||
// (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) { | if (doneAfter == 0 || doneAfter > now + 1) { | ||||
doneAfter = now + 1; | doneAfter = now + 1; | ||||
} | } | ||||
} | } | ||||
while (it != vAddrNew.end()) { | while (it != vAddrNew.end()) { | ||||
CAddress &addr = *it; | CAddress &addr = *it; | ||||
// tfm::format(std::cout, "%s: got address %s\n", | // tfm::format(std::cout, "%s: got address %s\n", | ||||
// ToString(you).c_str(), | // ToString(you), | ||||
// addr.ToString().c_str(), (int)(vAddr->size())); | // addr.ToString(), (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); | ||||
} | } | ||||
// tfm::format(std::cout, "%s: added address %s (#%i)\n", | // tfm::format(std::cout, "%s: added address %s (#%i)\n", | ||||
// ToString(you).c_str(), | // ToString(you), | ||||
// addr.ToString().c_str(), (int)(vAddr->size())); | // addr.ToString(), (int)(vAddr->size())); | ||||
if (vAddr->size() > ADDR_SOFT_CAP) { | if (vAddr->size() > ADDR_SOFT_CAP) { | ||||
doneAfter = 1; | doneAfter = 1; | ||||
return PeerMessagingState::Finished; | return PeerMessagingState::Finished; | ||||
} | } | ||||
} | } | ||||
return PeerMessagingState::AwaitingMessages; | return PeerMessagingState::AwaitingMessages; | ||||
} | } | ||||
Show All 18 Lines | do { | ||||
} | } | ||||
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)) { | ||||
// tfm::format(std::cout, "%s: BAD (invalid header)\n", | // tfm::format(std::cout, "%s: BAD (invalid header)\n", | ||||
// ToString(you).c_str()); | // ToString(you)); | ||||
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) { | ||||
// tfm::format(std::cout, "%s: BAD (message too large)\n", | // tfm::format(std::cout, "%s: BAD (message too large)\n", | ||||
// ToString(you).c_str()); | // ToString(you)); | ||||
ban = 100000; | ban = 100000; | ||||
return true; | return true; | ||||
} | } | ||||
if (nMessageSize > vRecv.size()) { | if (nMessageSize > vRecv.size()) { | ||||
vRecv.insert(vRecv.begin(), vHeaderSave.begin(), vHeaderSave.end()); | vRecv.insert(vRecv.begin(), vHeaderSave.begin(), vHeaderSave.end()); | ||||
break; | break; | ||||
} | } | ||||
if (vRecv.GetVersion() >= 209) { | if (vRecv.GetVersion() >= 209) { | ||||
uint256 hash = Hash(vRecv.begin(), vRecv.begin() + nMessageSize); | uint256 hash = 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) == PeerMessagingState::Finished) { | if (ProcessMessage(strCommand, vMsg) == PeerMessagingState::Finished) { | ||||
return true; | return true; | ||||
} | } | ||||
// tfm::format(std::cout, "%s: done processing %s\n", | // tfm::format(std::cout, "%s: done processing %s\n", | ||||
// ToString(you).c_str(), | // ToString(you), | ||||
// strCommand.c_str()); | // strCommand); | ||||
} while (1); | } while (1); | ||||
return false; | return false; | ||||
} | } | ||||
CSeederNode::CSeederNode(const CService &ip, std::vector<CAddress> *vAddrIn) | CSeederNode::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), | ||||
nHeaderStart(-1), nMessageStart(-1), nVersion(0), vAddr(vAddrIn), ban(0), | nHeaderStart(-1), nMessageStart(-1), nVersion(0), vAddr(vAddrIn), ban(0), | ||||
doneAfter(0), you(ip, ServiceFlags(NODE_NETWORK | NODE_BITCOIN_CASH)) { | doneAfter(0), you(ip, ServiceFlags(NODE_NETWORK | NODE_BITCOIN_CASH)) { | ||||
Show All 28 Lines | if (you.IsValid()) { | ||||
} | } | ||||
// no proxy needed (none set for target network) | // no proxy needed (none set for target network) | ||||
connected = ConnectSocketDirectly(you, sock, nConnectTimeout, true); | connected = ConnectSocketDirectly(you, sock, nConnectTimeout, true); | ||||
} | } | ||||
} | } | ||||
if (!connected) { | if (!connected) { | ||||
// tfm::format(std::cout, "Cannot connect to %s\n", | // tfm::format(std::cout, "Cannot connect to %s\n", | ||||
// ToString(you).c_str()); | // ToString(you)); | ||||
CloseSocket(sock); | CloseSocket(sock); | ||||
return false; | return false; | ||||
} | } | ||||
PushVersion(); | PushVersion(); | ||||
Send(); | Send(); | ||||
bool res = true; | bool res = true; | ||||
Show All 26 Lines | while (now = time(nullptr), ban == 0 && | ||||
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) { | ||||
// tfm::format(std::cout, "%s: BAD (connection closed | // tfm::format(std::cout, "%s: BAD (connection closed | ||||
// prematurely)\n", | // prematurely)\n", | ||||
// ToString(you).c_str()); | // ToString(you)); | ||||
res = false; | res = false; | ||||
break; | break; | ||||
} else { | } else { | ||||
// tfm::format(std::cout, "%s: BAD (connection error)\n", | // tfm::format(std::cout, "%s: BAD (connection error)\n", | ||||
// ToString(you).c_str()); | // ToString(you)); | ||||
res = false; | res = false; | ||||
break; | break; | ||||
} | } | ||||
ProcessMessages(); | ProcessMessages(); | ||||
Send(); | Send(); | ||||
} | } | ||||
if (sock == INVALID_SOCKET) { | if (sock == INVALID_SOCKET) { | ||||
res = false; | res = false; | ||||
} | } | ||||
close(sock); | close(sock); | ||||
sock = INVALID_SOCKET; | sock = INVALID_SOCKET; | ||||
return (ban == 0) && res; | return (ban == 0) && res; | ||||
} | } |