STPathSet.h

#pragma once
 
#include <xrpl/basics/CountedObject.h>
#include <xrpl/basics/UnorderedContainers.h>
#include <xrpl/beast/utility/instrumentation.h>
#include <xrpl/json/json_value.h>
#include <xrpl/protocol/AccountID.h>
#include <xrpl/protocol/PathAsset.h>
#include <xrpl/protocol/SField.h>
#include <xrpl/protocol/STBase.h>
#include <xrpl/protocol/Serializer.h>
#include <xrpl/protocol/UintTypes.h>
 
#include <cstddef>
#include <cstdint>
#include <memory>
#include <optional>
#include <utility>
#include <vector>
 
namespace xrpl {
 
class STPathElement final : public CountedObject<STPathElement>
{
    unsigned int type_;
    AccountID accountID_;
    PathAsset assetID_;
    AccountID issuerID_;
 
    bool isOffer_;
    std::size_t hashValue_;
 
public:
    // Bitwise values (typeCurrency | typeMPT)
    // NOLINTNEXTLINE(cppcoreguidelines-use-enum-class)
    enum Type {
        TypeNone = 0x00,
        TypeAccount = 0x01,   // Rippling through an account (vs taking an offer).
        TypeCurrency = 0x10,  // Currency follows.
        TypeIssuer = 0x20,    // Issuer follows.
        TypeMpt = 0x40,       // MPT follows.
        TypeBoundary = 0xFF,  // Boundary between alternate paths.
        TypeAsset = TypeCurrency | TypeMpt,
        TypeAll = TypeAccount | TypeCurrency | TypeIssuer | TypeMpt,
        // Combination of all types.
    };
 
    STPathElement();
    STPathElement(STPathElement const&) = default;
    STPathElement&
    operator=(STPathElement const&) = default;
 
    STPathElement(
        std::optional<AccountID> const& account,
        std::optional<PathAsset> const& asset,
        std::optional<AccountID> const& issuer);
 
    STPathElement(
        AccountID const& account,
        PathAsset const& asset,
        AccountID const& issuer,
        bool forceAsset = false);
 
    STPathElement(
        unsigned int uType,
        AccountID const& account,
        PathAsset const& asset,
        AccountID const& issuer);
 
    [[nodiscard]] std::uint32_t
    getNodeType() const;
 
    [[nodiscard]] bool
    isOffer() const;
 
    [[nodiscard]] bool
    isAccount() const;
 
    [[nodiscard]] bool
    hasIssuer() const;
 
    [[nodiscard]] bool
    hasCurrency() const;
 
    [[nodiscard]] bool
    hasMPT() const;
 
    [[nodiscard]] bool
    hasAsset() const;
 
    [[nodiscard]] bool
    isNone() const;
 
    // Nodes are either an account ID or a offer prefix. Offer prefixs denote a
    // class of offers.
    [[nodiscard]] AccountID const&
    getAccountID() const;
 
    [[nodiscard]] PathAsset const&
    getPathAsset() const;
 
    [[nodiscard]] Currency const&
    getCurrency() const;
 
    [[nodiscard]] MPTID const&
    getMPTID() const;
 
    [[nodiscard]] AccountID const&
    getIssuerID() const;
 
    [[nodiscard]] bool
    isType(Type const& pe) const;
 
    bool
    operator==(STPathElement const& t) const;
 
private:
    static std::size_t
    getHash(STPathElement const& element);
};
 
template <class Hasher>
void
hash_append(Hasher& h, STPathElement const& e) noexcept
{
    using beast::hash_append;
    hash_append(h, (e.getNodeType() & STPathElement::TypeAccount) != 0u);
    hash_append(h, e.getAccountID());
    hash_append(h, e.getPathAsset());
    hash_append(h, e.getIssuerID());
}
 
class STPath final : public CountedObject<STPath>
{
    std::vector<STPathElement> path_;
 
public:
    STPath() = default;
 
    STPath(std::vector<STPathElement> p);
 
    [[nodiscard]] std::vector<STPathElement>::size_type
    size() const;
 
    [[nodiscard]] bool
    empty() const;
 
    void
    pushBack(STPathElement const& e);
 
    template <typename... Args>
    void
    emplaceBack(Args&&... args);
 
    [[nodiscard]] bool
    hasSeen(AccountID const& account, PathAsset const& asset, AccountID const& issuer) const;
 
    [[nodiscard]] json::Value getJson(JsonOptions) const;
 
    [[nodiscard]] std::vector<STPathElement>::const_iterator
    begin() const;
 
    [[nodiscard]] std::vector<STPathElement>::const_iterator
    end() const;
 
    bool
    operator==(STPath const& t) const;
 
    [[nodiscard]] std::vector<STPathElement>::const_reference
    back() const;
 
    [[nodiscard]] std::vector<STPathElement>::const_reference
    front() const;
 
    STPathElement&
    operator[](int i);
 
    STPathElement const&
    operator[](int i) const;
 
    void
    reserve(size_t s);
};
 
template <class Hasher>
void
hash_append(Hasher& h, STPath const& p) noexcept
{
    using beast::hash_append;
    for (auto const& e : p)
    {
        hash_append(h, e);
    }
}
 
//------------------------------------------------------------------------------
 
// A set of zero or more payment paths
class STPathSet final : public STBase, public CountedObject<STPathSet>
{
    std::vector<STPath> value_;

    std::unique_ptr<HardenedHashSet<STPath>> seen_;
 
public:
    struct DeduplicationTag
    {
    };
 
    STPathSet() = default;

    explicit STPathSet(DeduplicationTag);
 
    STPathSet(SField const& n);
    STPathSet(SerialIter& sit, SField const& name);
    STPathSet(STPathSet const& other);
    STPathSet(STPathSet&&) = default;
 
    STPathSet&
    operator=(STPathSet const& other);
    STPathSet&
    operator=(STPathSet&&) = default;
 
    ~STPathSet() override = default;
 
    void
    add(Serializer& s) const override;
 
    [[nodiscard]] json::Value getJson(JsonOptions) const override;
 
    [[nodiscard]] SerializedTypeID
    getSType() const override;

    bool
    assembleAdd(STPath const& base, STPathElement const& tail);
 
    [[nodiscard]] bool
    isEquivalent(STBase const& t) const override;
 
    [[nodiscard]] bool
    isDefault() const override;
 
    // std::vector like interface:
    std::vector<STPath>::const_reference
    operator[](std::vector<STPath>::size_type n) const;
 
    [[nodiscard]] std::vector<STPath>::const_iterator
    begin() const;
 
    [[nodiscard]] std::vector<STPath>::const_iterator
    end() const;
 
    [[nodiscard]] std::vector<STPath>::size_type
    size() const;
 
    [[nodiscard]] bool
    empty() const;

    bool
    pushBack(STPath const& e);

    template <typename... Args>
    bool
    emplaceBack(Args&&... args);
 
private:
    STBase*
    copy(std::size_t n, void* buf) const override;
    STBase*
    move(std::size_t n, void* buf) override;

    template <typename Append>
    bool
    appendUnique(Append&& append);
 
    friend class detail::STVar;
};
 
// ------------ STPathElement ------------
 
inline STPathElement::STPathElement() : type_(TypeNone), isOffer_(true)
{
    // hashValue_ is derived from the whole object, so it is computed in the body
    // once every other member is initialized (as in the other constructors).
    // NOLINTNEXTLINE(cppcoreguidelines-prefer-member-initializer)
    hashValue_ = getHash(*this);
}
 
inline STPathElement::STPathElement(
    std::optional<AccountID> const& account,
    std::optional<PathAsset> const& asset,
    std::optional<AccountID> const& issuer)
    : type_(TypeNone)
{
    if (!account)
    {
        isOffer_ = true;
    }
    else
    {
        isOffer_ = false;
        accountID_ = *account;
        type_ |= TypeAccount;
        XRPL_ASSERT(
            accountID_ != noAccount(), "xrpl::STPathElement::STPathElement : account is set");
    }
 
    if (asset)
    {
        assetID_ = *asset;
        type_ |= assetID_.holds<Currency>() ? TypeCurrency : TypeMpt;
    }
 
    if (issuer)
    {
        issuerID_ = *issuer;
        type_ |= TypeIssuer;
        XRPL_ASSERT(issuerID_ != noAccount(), "xrpl::STPathElement::STPathElement : issuer is set");
    }
 
    hashValue_ = getHash(*this);
}
 
inline STPathElement::STPathElement(
    AccountID const& account,
    PathAsset const& asset,
    AccountID const& issuer,
    bool forceAsset)
    : type_(TypeNone)
    , accountID_(account)
    , assetID_(asset)
    , issuerID_(issuer)
    , isOffer_(isXRP(accountID_))
{
    if (!isOffer_)
        type_ |= TypeAccount;
 
    if (forceAsset || !isXRP(assetID_))
        type_ |= asset.holds<Currency>() ? TypeCurrency : TypeMpt;
 
    if (!isXRP(issuer))
        type_ |= TypeIssuer;
 
    hashValue_ = getHash(*this);
}
 
inline STPathElement::STPathElement(
    unsigned int uType,
    AccountID const& account,
    PathAsset const& asset,
    AccountID const& issuer)
    : type_(uType)
    , accountID_(account)
    , assetID_(asset)
    , issuerID_(issuer)
    , isOffer_(isXRP(accountID_))
{
    assetID_.visit(
        [&](Currency const&) { type_ = type_ & (~Type::TypeMpt); },
        [&](MPTID const&) { type_ = type_ & (~Type::TypeCurrency); });
    // hashValue_ must be computed after type_ is adjusted above, so this cannot
    // be a member initializer.
    // NOLINTNEXTLINE(cppcoreguidelines-prefer-member-initializer)
    hashValue_ = getHash(*this);
}
 
inline std::uint32_t
STPathElement::getNodeType() const
{
    return type_;
}
 
inline bool
STPathElement::isOffer() const
{
    return isOffer_;
}
 
inline bool
STPathElement::isAccount() const
{
    return !isOffer();
}
 
inline bool
STPathElement::isType(Type const& pe) const
{
    return (type_ & pe) != 0u;
}
 
inline bool
STPathElement::hasIssuer() const
{
    return isType(STPathElement::TypeIssuer);
}
 
inline bool
STPathElement::hasCurrency() const
{
    return isType(STPathElement::TypeCurrency);
}
 
inline bool
STPathElement::hasMPT() const
{
    return isType(STPathElement::TypeMpt);
}
 
inline bool
STPathElement::hasAsset() const
{
    return isType(STPathElement::TypeAsset);
}
 
inline bool
STPathElement::isNone() const
{
    return getNodeType() == STPathElement::TypeNone;
}
 
// Nodes are either an account ID or a offer prefix. Offer prefixs denote a
// class of offers.
inline AccountID const&
STPathElement::getAccountID() const
{
    return accountID_;
}
 
inline PathAsset const&
STPathElement::getPathAsset() const
{
    return assetID_;
}
 
inline Currency const&
STPathElement::getCurrency() const
{
    return assetID_.get<Currency>();
}
 
inline MPTID const&
STPathElement::getMPTID() const
{
    return assetID_.get<MPTID>();
}
 
inline AccountID const&
STPathElement::getIssuerID() const
{
    return issuerID_;
}
 
inline bool
STPathElement::operator==(STPathElement const& t) const
{
    return (type_ & TypeAccount) == (t.type_ & TypeAccount) && hashValue_ == t.hashValue_ &&
        accountID_ == t.accountID_ && assetID_ == t.assetID_ && issuerID_ == t.issuerID_;
}
 
// ------------ STPath ------------
 
inline STPath::STPath(std::vector<STPathElement> p) : path_(std::move(p))
{
}
 
inline std::vector<STPathElement>::size_type
STPath::size() const
{
    return path_.size();
}
 
inline bool
STPath::empty() const
{
    return path_.empty();
}
 
inline void
STPath::pushBack(STPathElement const& e)
{
    path_.push_back(e);
}
 
template <typename... Args>
inline void
STPath::emplaceBack(Args&&... args)
{
    path_.emplace_back(std::forward<Args>(args)...);
}
 
inline std::vector<STPathElement>::const_iterator
STPath::begin() const
{
    return path_.begin();
}
 
inline std::vector<STPathElement>::const_iterator
STPath::end() const
{
    return path_.end();
}
 
inline bool
STPath::operator==(STPath const& t) const
{
    return path_ == t.path_;
}
 
inline std::vector<STPathElement>::const_reference
STPath::back() const
{
    return path_.back();
}
 
inline std::vector<STPathElement>::const_reference
STPath::front() const
{
    return path_.front();
}
 
inline STPathElement&
STPath::operator[](int i)
{
    return path_[i];
}
 
inline STPathElement const&
STPath::operator[](int i) const
{
    return path_[i];
}
 
inline void
STPath::reserve(size_t s)
{
    path_.reserve(s);
}
 
// ------------ STPathSet ------------
 
inline STPathSet::STPathSet(SField const& n) : STBase(n)
{
}
 
// std::vector like interface:
inline std::vector<STPath>::const_reference
STPathSet::operator[](std::vector<STPath>::size_type n) const
{
    return value_[n];
}
 
inline std::vector<STPath>::const_iterator
STPathSet::begin() const
{
    return value_.begin();
}
 
inline std::vector<STPath>::const_iterator
STPathSet::end() const
{
    return value_.end();
}
 
inline std::vector<STPath>::size_type
STPathSet::size() const
{
    return value_.size();
}
 
inline bool
STPathSet::empty() const
{
    return value_.empty();
}
 
template <typename Append>
inline bool
STPathSet::appendUnique(Append&& append)
{
    // Append to the vector first, so that a failed allocation there leaves both
    // containers untouched rather than leaving the index holding a path the
    // vector does not.
    append(value_);
 
    if (seen_ == nullptr)
    {
        return true;
    }
 
    try
    {
        if (!seen_->insert(value_.back()).second)
        {
            // Already present: roll back the append.
            value_.pop_back();
            return false;
        }
    }
    catch (...)
    {
        // The index insert failed, so roll back the append to keep the vector
        // and the index consistent.
        value_.pop_back();
        throw;
    }
    return true;
}
 
inline bool
STPathSet::pushBack(STPath const& e)
{
    return appendUnique([&](auto& value) { value.push_back(e); });
}
 
template <typename... Args>
inline bool
STPathSet::emplaceBack(Args&&... args)
{
    return appendUnique([&](auto& value) { value.emplace_back(std::forward<Args>(args)...); });
}
 
}  // namespace xrpl