bool OverlayManagerImpl::isPeerPreferred(Peer::pointer peer) { std::string pstr = peer->toString(); if (mPreferredPeers.find(pstr) != mPreferredPeers.end()) { CLOG(DEBUG, "Overlay") << "Peer " << pstr << " is preferred"; return true; } if (peer->isAuthenticated()) { std::string kstr = PubKeyUtils::toStrKey(peer->getPeerID()); std::vector<std::string> const& pk = mApp.getConfig().PREFERRED_PEER_KEYS; if (std::find(pk.begin(), pk.end(), kstr) != pk.end()) { CLOG(DEBUG, "Overlay") << "Peer key " << mApp.getConfig().toStrKey(peer->getPeerID()) << " is preferred"; return true; } } CLOG(DEBUG, "Overlay") << "Peer " << pstr << " is not preferred"; return false; }
void Tracker::tryNextPeer() { // will be called by some timer or when we get a // response saying they don't have it Peer::pointer peer; CLOG(TRACE, "Overlay") << "tryNextPeer " << hexAbbrev(mItemHash) << " last: " << (mLastAskedPeer ? mLastAskedPeer->toString() : "<none>"); // if we don't have a list of peers to ask and we're not // currently asking peers, build a new list if (mPeersToAsk.empty() && !mLastAskedPeer) { std::set<std::shared_ptr<Peer>> peersWithEnvelope; for (auto const& e : mWaitingEnvelopes) { auto const& s = mApp.getOverlayManager().getPeersKnows(e.first); peersWithEnvelope.insert(s.begin(), s.end()); } // move the peers that have the envelope to the back, // to be processed first for (auto const& p : mApp.getOverlayManager().getRandomPeers()) { if (peersWithEnvelope.find(p) != peersWithEnvelope.end()) { mPeersToAsk.emplace_back(p); } else { mPeersToAsk.emplace_front(p); } } mNumListRebuild++; CLOG(TRACE, "Overlay") << "tryNextPeer " << hexAbbrev(mItemHash) << " attempt " << mNumListRebuild << " reset to #" << mPeersToAsk.size(); mTryNextPeerReset.Mark(); } while (!peer && !mPeersToAsk.empty()) { peer = mPeersToAsk.back(); if (!peer->isAuthenticated()) { peer.reset(); } mPeersToAsk.pop_back(); } std::chrono::milliseconds nextTry; if (!peer) { // we have asked all our peers // clear mLastAskedPeer so that we rebuild a new list mLastAskedPeer.reset(); if (mNumListRebuild > MAX_REBUILD_FETCH_LIST) { nextTry = MS_TO_WAIT_FOR_FETCH_REPLY * MAX_REBUILD_FETCH_LIST; } else { nextTry = MS_TO_WAIT_FOR_FETCH_REPLY * mNumListRebuild; } } else { mLastAskedPeer = peer; CLOG(TRACE, "Overlay") << "Asking for " << hexAbbrev(mItemHash) << " to " << peer->toString(); mTryNextPeer.Mark(); mAskPeer(peer, mItemHash); nextTry = MS_TO_WAIT_FOR_FETCH_REPLY; } mTimer.expires_from_now(nextTry); mTimer.async_wait([this]() { this->tryNextPeer(); }, VirtualTimer::onFailureNoop); }