addrman.h

Go to the documentation of this file.

// Copyright (c) 2012 Pieter Wuille
// Copyright (c) 2012-2016 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_ADDRMAN_H
#define BITCOIN_ADDRMAN_H
 
#include <netaddress.h>
#include <protocol.h>
#include <streams.h>
#include <util/time.h>
 
#include <cstdint>
#include <memory>
#include <optional>
#include <utility>
#include <vector>
 
class InvalidAddrManVersionError : public std::ios_base::failure {
public:
    InvalidAddrManVersionError(std::string msg) : std::ios_base::failure(msg) {}
};
 
class AddrManImpl;
 
static constexpr int32_t DEFAULT_ADDRMAN_CONSISTENCY_CHECKS{0};
 
class AddrMan {
    const std::unique_ptr<AddrManImpl> m_impl;
 
public:
    AddrMan(std::vector<bool> asmap, int32_t consistency_check_ratio);
 
    ~AddrMan();
 
    template <typename Stream> void Serialize(Stream &s_) const;
 
    template <typename Stream> void Unserialize(Stream &s_);
 
    size_t size() const;
 
    bool Add(const std::vector<CAddress> &vAddr, const CNetAddr &source,
             std::chrono::seconds time_penalty = 0s);
 
    void Good(const CService &addr, bool test_before_evict = true,
              NodeSeconds time = Now<NodeSeconds>());
 
    void Attempt(const CService &addr, bool fCountFailure,
                 NodeSeconds time = Now<NodeSeconds>());
 
    void ResolveCollisions();
 
    std::pair<CAddress, NodeSeconds> SelectTriedCollision();
 
    std::pair<CAddress, NodeSeconds> Select(bool newOnly = false) const;
 
    std::vector<CAddress> GetAddr(size_t max_addresses, size_t max_pct,
                                  std::optional<Network> network) const;
 
    void Connected(const CService &addr, NodeSeconds time = Now<NodeSeconds>());
 
    void SetServices(const CService &addr, ServiceFlags nServices);
 
    const std::vector<bool> &GetAsmap() const;
 
    void Clear();
 
    void MakeDeterministic();
 
    friend class AddrManTest;
    friend class AddrManCorrupted;
};
 
#endif // BITCOIN_ADDRMAN_H