infohash.h

// Copyright (c) 2014-2026 Savoir-faire Linux Inc.
// SPDX-License-Identifier: MIT
#pragma once
 
#include "def.h"
#include "rng.h"
 
#include <msgpack.hpp>
#include <fmt/core.h>
 
#ifndef _WIN32
#include <netinet/in.h>
#include <netdb.h>
#ifdef __ANDROID__
typedef uint16_t in_port_t;
#endif
#else
#include <iso646.h>
#include <ws2tcpip.h>
typedef uint16_t sa_family_t;
typedef uint16_t in_port_t;
#endif
 
#include <iostream>
#include <iomanip>
#include <array>
#include <vector>
#include <string_view>
#include <algorithm>
#include <stdexcept>
#include <sstream>
 
#include <cstring>
#include <cstddef>
 
namespace dht {
 
using byte = uint8_t;
 
namespace crypto {
OPENDHT_PUBLIC void hash(const uint8_t* data, size_t data_length, uint8_t* hash, size_t hash_length);
}

template<size_t N>
class Hash
{
public:
    using T = std::array<uint8_t, N>;
    typedef typename T::iterator iterator;
    typedef typename T::const_iterator const_iterator;
 
    constexpr Hash() noexcept
        : data_ {}
    {}
    constexpr Hash(const uint8_t* h, size_t data_len)
        : data_(fromBytes(h, data_len))
    {}
    explicit constexpr Hash(std::string_view hex)
        : data_(fromString(hex))
    {}
 
    Hash(const msgpack::object& o) { msgpack_unpack(o); }
 
    static constexpr size_t size() noexcept { return N; }
    constexpr const uint8_t* data() const { return data_.data(); }
    uint8_t* data() { return data_.data(); }
    iterator begin() { return data_.begin(); }
    const_iterator cbegin() const { return data_.cbegin(); }
    iterator end() { return data_.end(); }
    const_iterator cend() const { return data_.cend(); }
 
    static constexpr inline Hash zero() noexcept { return Hash {}; }
 
    constexpr bool operator==(const Hash& h) const
    {
#if __cplusplus >= 202002L
        return data_ == h.data_;
#else
        for (size_t i = 0; i < N; ++i) {
            if (data_[i] != h.data_[i])
                return false;
        }
        return true;
#endif
    }
    constexpr bool operator!=(const Hash& h) const { return !(*this == h); }
 
    constexpr bool operator<(const Hash& o) const
    {
        for (unsigned i = 0; i < N; i++) {
            if (data_[i] != o.data_[i])
                return data_[i] < o.data_[i];
        }
        return false;
    }
 
    constexpr Hash operator^(const Hash& o) const
    {
        Hash result;
        for (auto i = 0u; i < N; i++) {
            result[i] = data_[i] ^ o.data_[i];
        }
        return result;
    }
 
    explicit constexpr operator bool() const
    {
#if __cplusplus >= 202002L
        auto a = reinterpret_cast<const uint32_t*>(data_.data());
        auto b = reinterpret_cast<const uint32_t*>(data_.data() + N);
        for (; a != b; a++) {
            if (*a)
                return true;
        }
        return false;
#else
        for (size_t i = 0; i < N; ++i) {
            if (data_[i] != 0)
                return true;
        }
        return false;
#endif
    }
 
    uint8_t& operator[](size_t index) { return data_[index]; }
    constexpr const uint8_t& operator[](size_t index) const { return data_[index]; }

    constexpr int lowbit() const
    {
        int i = N - 1;
        for (; i >= 0; i--)
            if (data_[i] != 0)
                break;
        if (i < 0)
            return -1;
        int j = 7;
        for (; j >= 0; j--)
            if ((data_[i] & (0x80 >> j)) != 0)
                break;
        return 8 * i + j;
    }
 
    static inline int cmp(const Hash& id1, const Hash& id2)
    {
        return std::memcmp(id1.data_.data(), id2.data_.data(), N);
    }

    static constexpr inline unsigned commonBits(const Hash& id1, const Hash& id2)
    {
        unsigned i = 0;
        for (; i < N; i++) {
            if (id1.data_[i] != id2.data_[i])
                break;
        }
 
        if (i == N)
            return 8 * N;
 
        uint8_t x = id1.data_[i] ^ id2.data_[i];
 
        unsigned j = 0;
        while ((x & 0x80) == 0) {
            x <<= 1;
            j++;
        }
 
        return 8 * i + j;
    }

    constexpr int xorCmp(const Hash& id1, const Hash& id2) const
    {
        for (unsigned i = 0; i < N; i++) {
            if (id1.data_[i] == id2.data_[i])
                continue;
            uint8_t xor1 = id1.data_[i] ^ data_[i];
            uint8_t xor2 = id2.data_[i] ^ data_[i];
            return (xor1 < xor2) ? -1 : 1;
        }
        return 0;
    }
 
    constexpr bool getBit(unsigned nbit) const
    {
        auto& num = *(data_.cbegin() + (nbit / 8));
        unsigned bit = 7 - (nbit % 8);
        return (num >> bit) & 1;
    }
 
    void setBit(unsigned nbit, bool b)
    {
        auto& num = data_[nbit / 8];
        unsigned bit = 7 - (nbit % 8);
        num ^= (-b ^ num) & (1 << bit);
    }
 
    double toFloat() const
    {
        using D = size_t;
        double v = 0.;
        for (size_t i = 0; i < std::min<size_t>(N, sizeof(D) - 1); i++)
            v += *(data_.cbegin() + i) / (double) ((D) 1 << 8 * (i + 1));
        return v;
    }
 
    static inline Hash get(std::string_view data) { return get((const uint8_t*) data.data(), data.size()); }
 
    static inline Hash get(const std::vector<uint8_t>& data) { return get(data.data(), data.size()); }
 
    template<size_t H>
    static Hash get(const Hash<H>& o)
    {
        return get(o.data(), o.size());
    }

    static Hash get(const uint8_t* data, size_t data_len)
    {
        Hash ret;
        crypto::hash(data, data_len, ret.data(), N);
        return ret;
    }
 
    static Hash getRandom();
 
    template<typename Rd>
    static Hash getRandom(Rd&);
 
    template<size_t M>
    friend std::ostream& operator<<(std::ostream& s, const Hash<M>& h);
 
    template<size_t M>
    friend std::istream& operator>>(std::istream& s, Hash<M>& h);

    std::string_view to_view() const { return std::string_view(to_c_str(), N * 2); }
    const char* to_c_str() const;
 
    std::string toString() const;
 
    template<typename Packer>
    void msgpack_pack(Packer& pk) const
    {
        pk.pack_bin(N);
        pk.pack_bin_body((char*) data_.data(), N);
    }
 
    void msgpack_unpack(msgpack::object o)
    {
        if (o.type != msgpack::type::BIN or o.via.bin.size != N)
            throw msgpack::type_error();
        std::copy_n(o.via.bin.ptr, N, data_.data());
    }
 
private:
    T data_;
    static constexpr T fromBytes(const uint8_t*, size_t);
    static constexpr T fromString(std::string_view);
};
 
#define HASH_LEN 20u
using InfoHash = Hash<HASH_LEN>;
using h256 = Hash<32>;
using PkId = h256;
 
template<size_t N>
std::ostream&
operator<<(std::ostream& s, const Hash<N>& h)
{
    s.write(h.to_c_str(), N * 2);
    return s;
}
 
template<size_t N>
std::istream&
operator>>(std::istream& s, Hash<N>& h)
{
    std::array<char, h.size() * 2> dat;
    s.exceptions(std::istream::eofbit | std::istream::failbit);
    s.read(&(*dat.begin()), dat.size());
    h = Hash<N>(std::string_view(dat.data(), dat.size()));
    return s;
}
 
template<size_t N>
constexpr typename Hash<N>::T
Hash<N>::fromBytes(const uint8_t* in, size_t size)
{
    T data {};
    if (size < N)
        return data;
    for (size_t i = 0; i < N; ++i)
        data[i] = in[i];
    return data;
}
 
template<size_t N>
constexpr typename Hash<N>::T
Hash<N>::fromString(std::string_view in)
{
    T data {};
    if (in.size() < 2 * N)
        return data;
 
    constexpr auto hex2bin = [](char c, bool& valid) -> uint8_t {
        if (c >= 'a' and c <= 'f')
            return 10 + c - 'a';
        else if (c >= 'A' and c <= 'F')
            return 10 + c - 'A';
        else if (c >= '0' and c <= '9')
            return c - '0';
        else {
            valid = false;
            return 0;
        }
    };
    bool valid = true;
    for (size_t i = 0; i < N && valid; i++) {
        uint8_t high = hex2bin(in[2 * i], valid);
        uint8_t low = hex2bin(in[2 * i + 1], valid);
        data[i] = (high << 4) | low;
    }
    if (!valid)
        return {};
    return data;
}
 
template<size_t N>
Hash<N>
Hash<N>::getRandom()
{
    Hash h;
    std::random_device rdev;
    std::uniform_int_distribution<uint32_t> rand_int;
    auto a = reinterpret_cast<uint32_t*>(h.data());
    auto b = reinterpret_cast<uint32_t*>(h.data() + h.size());
    std::generate(a, b, std::bind(rand_int, std::ref(rdev)));
    return h;
}
 
template<size_t N>
template<typename Rd>
Hash<N>
Hash<N>::getRandom(Rd& rdev)
{
    Hash h;
    std::uniform_int_distribution<uint32_t> rand_int;
    auto a = reinterpret_cast<uint32_t*>(h.data());
    auto b = reinterpret_cast<uint32_t*>(h.data() + h.size());
    std::generate(a, b, std::bind(rand_int, std::ref(rdev)));
    return h;
}
 
struct alignas(std::max_align_t) HexMap : public std::array<std::array<char, 2>, 256>
{
    constexpr HexMap()
        : std::array<std::array<char, 2>, 256>()
    {
        for (size_t i = 0; i < size(); i++) {
            auto& e = (*this)[i];
            e[0] = hex_digits[(i >> 4) & 0x0F];
            e[1] = hex_digits[i & 0x0F];
        }
    }
 
private:
    static constexpr const char* hex_digits = "0123456789abcdef";
};
 
inline constexpr HexMap hex_map {};
 
inline std::string
toHex(const uint8_t* data, size_t size)
{
    std::string ret(size * 2, '\0');
    for (size_t i = 0; i < size; i++) {
        auto b = ret.data() + i * 2;
        const auto& m = hex_map[data[i]];
        *((uint16_t*) b) = *((uint16_t*) &m);
    }
    return ret;
}
 
inline std::string
toHex(const std::vector<uint8_t>& data)
{
    return toHex(data.data(), data.size());
}
 
template<size_t N>
constexpr std::array<char, N * 2>
toHexArray(const Hash<N>& h)
{
    std::array<char, N * 2> ret {};
    for (size_t i = 0; i < N; i++) {
        const auto& m = hex_map[h[i]];
        ret[i * 2] = m[0];
        ret[i * 2 + 1] = m[1];
    }
    return ret;
}
 
template<size_t N>
const char*
Hash<N>::to_c_str() const
{
    alignas(std::max_align_t) thread_local std::array<char, N * 2 + 1> buf;
    for (size_t i = 0; i < N; i++) {
        auto b = buf.data() + i * 2;
        const auto& m = hex_map[data_[i]];
        *((uint16_t*) b) = *((uint16_t*) &m);
    }
    buf[N * 2] = '\0';
    return buf.data();
}
 
template<size_t N>
std::string
Hash<N>::toString() const
{
    return std::string(to_c_str(), N * 2);
}
 
} // namespace dht
 
template<size_t N>
struct fmt::formatter<dht::Hash<N>> : formatter<string_view>
{
    constexpr auto format(const dht::Hash<N>& c, format_context& ctx) const
    {
        return formatter<string_view>::format(c.to_view(), ctx);
    }
};