feature/scale v2 (#2289)

* scale v2 Signed-off-by: iceseer <[email protected]> --------- Signed-off-by: iceseer <[email protected]>
qdrvm · Dec 13, 2024 · 1d5d755 · 1d5d755
1 parent 6acdc45
commit 1d5d755
Show file tree

Hide file tree

Showing 19 changed files with 512 additions and 268 deletions.
diff --git a/core/authority_discovery/query/authority_peer_info.hpp b/core/authority_discovery/query/authority_peer_info.hpp
@@ -18,7 +18,7 @@ namespace kagome::authority_discovery {
 
     common::Buffer raw{};
     std::optional<TimestampScale> time{};
-    scale::PeerInfoSerializable peer{};
+    ::scale::PeerInfoSerializable peer{};
   };
 
 }  // namespace kagome::authority_discovery
diff --git a/core/authority_discovery/timestamp.hpp b/core/authority_discovery/timestamp.hpp
@@ -9,6 +9,6 @@
 #include "scale/big_fixed_integers.hpp"
 
 namespace kagome::authority_discovery {
-  using Timestamp = scale::uint128_t;
-  using TimestampScale = scale::Fixed<Timestamp>;
+  using Timestamp = ::scale::uint128_t;
+  using TimestampScale = ::scale::Fixed<Timestamp>;
 }  // namespace kagome::authority_discovery
diff --git a/core/crypto/type_hasher.hpp b/core/crypto/type_hasher.hpp
@@ -15,8 +15,11 @@ namespace kagome::crypto {
 
   template <typename H, typename... T>
   inline void hashTypes(H &hasher, common::Blob<H::kOutlen> &out, T &&...t) {
-    auto val = ::scale::encode(std::forward<T>(t)...).value();
-    hasher.update(val);
+    scale::encode(
+        [&](const uint8_t *const ptr, size_t count) {
+          hasher.update(std::span<const uint8_t>(ptr, count));
+        },
+        std::forward<T>(t)...);
     hasher.get_final(out);
   }
 

diff --git a/core/network/helpers/scale_message_read_writer.hpp b/core/network/helpers/scale_message_read_writer.hpp
@@ -66,7 +66,7 @@ namespace kagome::network {
     template <typename MsgType>
     void write(const MsgType &msg,
                libp2p::basic::Writer::WriteCallbackFunc cb) const {
-      auto encoded_msg_res = ::scale::encode(msg);
+      auto encoded_msg_res = scale::encode(msg);
       if (!encoded_msg_res) {
         return cb(encoded_msg_res.error());
       }

diff --git a/core/network/impl/sync_protocol_observer_impl.cpp b/core/network/impl/sync_protocol_observer_impl.cpp
@@ -196,7 +196,7 @@ namespace kagome::network {
             if (auto r = beefy_->getJustification(*number)) {
               if (auto &opt = r.value()) {
                 new_block.beefy_justification = primitives::Justification{
-                    common::Buffer{::scale::encode(*opt).value()},
+                    common::Buffer{scale::encode(*opt).value()},
                 };
               }
             }

diff --git a/core/network/types/collator_messages.hpp b/core/network/types/collator_messages.hpp
@@ -458,8 +458,7 @@ namespace kagome::network {
     auto commitments_hash =
         hasher.blake2b_256(scale::encode(receipt.commitments).value());
     return hasher.blake2b_256(
-        ::scale::encode(std::tie(receipt.descriptor, commitments_hash))
-            .value());
+        scale::encode(std::tie(receipt.descriptor, commitments_hash)).value());
   }
 
   inline CandidateHash candidateHash(const crypto::Hasher &hasher,

diff --git a/core/parachain/approval/knowledge.hpp b/core/parachain/approval/knowledge.hpp
@@ -11,13 +11,14 @@
 #include "common/visitor.hpp"
 #include "consensus/timeline/types.hpp"
 #include "outcome/outcome.hpp"
+#include "parachain/approval/approval.hpp"
 #include "parachain/approval/state.hpp"
 #include "parachain/types.hpp"
 
 template <>
 struct std::hash<scale::BitVec> {
   auto operator()(const scale::BitVec &v) const {
-    auto s = ::scale::encode(v).value();
+    auto s = scale::encode(v).value();
     return boost::hash_range(s.begin(), s.end());
   }
 };
@@ -86,7 +87,7 @@ namespace kagome::parachain::approval {
 
     // Generate the knowledge keys for querying if an approval is known by peer.
     static std::pair<MessageSubject, MessageKind> generate_approval_key(
-        const approval::IndirectSignedApprovalVoteV2 &approval) {
+        const IndirectSignedApprovalVoteV2 &approval) {
       return {
           std::make_tuple(approval.payload.payload.block_hash,
                           approval.payload.payload.candidate_indices,

diff --git a/core/parachain/pvf/pvf_impl.cpp b/core/parachain/pvf/pvf_impl.cpp
@@ -306,7 +306,7 @@ namespace kagome::parachain {
              receipt.descriptor.relay_parent,
              receipt.descriptor.para_id);
 
-    auto data_hash = hasher_->blake2b_256(::scale::encode(pvd).value());
+    auto data_hash = hasher_->blake2b_256(scale::encode(pvd).value());
     if (receipt.descriptor.persisted_data_hash != data_hash) {
       return cb(PvfError::PERSISTED_DATA_HASH);
     }

diff --git a/core/parachain/types.hpp b/core/parachain/types.hpp
@@ -93,7 +93,7 @@ namespace kagome::parachain {
 
     auto signable() {
       constexpr std::array<uint8_t, 4> kMagic{'V', 'C', 'P', 'C'};
-      return ::scale::encode(std::make_tuple(kMagic, *this)).value();
+      return scale::encode(std::make_tuple(kMagic, *this)).value();
     }
   };
 }  // namespace kagome::parachain
@@ -171,11 +171,11 @@ namespace kagome::network {
 
     common::Buffer signable() const {
       return common::Buffer{
-          ::scale::encode(relay_parent,
-                          para_id,
-                          persisted_data_hash,
-                          pov_hash,
-                          validation_code_hash)
+          scale::encode(relay_parent,
+                        para_id,
+                        persisted_data_hash,
+                        pov_hash,
+                        validation_code_hash)
               .value(),
       };
     }
@@ -193,7 +193,7 @@ namespace kagome::network {
     const parachain::Hash &hash(const crypto::Hasher &hasher) const {
       if (not hash_.has_value()) {
         hash_.emplace(hasher.blake2b_256(
-            ::scale::encode(std::tie(descriptor, commitments_hash)).value()));
+            scale::encode(std::tie(descriptor, commitments_hash)).value()));
       }
       return hash_.value();
     }

diff --git a/core/parachain/validator/signer.hpp b/core/parachain/validator/signer.hpp
@@ -48,7 +48,7 @@ namespace kagome::parachain {
     template <typename T>
     auto signable(const crypto::Hasher &hasher, const T &payload) const {
       auto &&signable = toSignable(hasher, payload);
-      return ::scale::encode(std::tie(signable, *this)).value();
+      return scale::encode(std::tie(signable, *this)).value();
     }
 
     /// Current session index.

diff --git a/core/parachain/validator/statement_distribution/types.hpp b/core/parachain/validator/statement_distribution/types.hpp
@@ -33,7 +33,7 @@ namespace kagome::parachain {
         statement,
         [&](const StatementWithPVDSeconded &val) {
           return hasher->blake2b_256(
-              ::scale::encode(val.committed_receipt.to_plain(*hasher)).value());
+              scale::encode(val.committed_receipt.to_plain(*hasher)).value());
         },
         [&](const StatementWithPVDValid &val) { return val.candidate_hash; });
   }

diff --git a/core/primitives/inherent_data.hpp b/core/primitives/inherent_data.hpp
@@ -50,7 +50,7 @@ namespace kagome::primitives {
                                   const T &inherent) {
       auto [it, inserted] = data.try_emplace(identifier, common::Buffer());
       if (inserted) {
-        it->second = common::Buffer(::scale::encode(inherent).value());
+        it->second = common::Buffer(scale::encode(inherent).value());
         return outcome::success();
       }
       return InherentDataError::IDENTIFIER_ALREADY_EXISTS;
@@ -62,7 +62,7 @@ namespace kagome::primitives {
      */
     template <typename T>
     void replaceData(InherentIdentifier identifier, const T &inherent) {
-      data[identifier] = common::Buffer(::scale::encode(inherent).value());
+      data[identifier] = common::Buffer(scale::encode(inherent).value());
     }
 
     /**

diff --git a/core/runtime/module_instance.hpp b/core/runtime/module_instance.hpp
@@ -18,6 +18,7 @@
 #include "outcome/outcome.hpp"
 #include "runtime/instance_environment.hpp"
 #include "runtime/ptr_size.hpp"
+#include "scale/kagome_scale.hpp"
 
 namespace kagome::runtime {
   class Module;
@@ -43,9 +44,9 @@ namespace kagome::runtime {
     template <typename... Args>
     static outcome::result<common::Buffer> encodeArgs(const Args &...args) {
       if constexpr (sizeof...(args) > 0) {
-        return common::map_result(::scale::encode(args...), [](auto &&vec) {
-          return common::Buffer{vec};
-        });
+        return common::map_result(
+            kagome::scale::encode(args...),
+            [](auto &&vec) { return common::Buffer{vec}; });
       }
       return outcome::success();
     }

diff --git a/core/scale/encoder/concepts.hpp b/core/scale/encoder/concepts.hpp
@@ -0,0 +1,21 @@
+/**
+ * Copyright Quadrivium LLC
+ * All Rights Reserved
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#pragma once
+
+#include <cstdint>
+#include <type_traits>
+
+namespace kagome::scale {
+  template <typename F>
+  concept Invocable = std::is_invocable_v<F, const uint8_t *const, std::size_t>;
+
+  template <typename T>
+  concept IsEnum = std::is_enum_v<std::decay_t<T>>;
+
+  template <typename T>
+  concept IsNotEnum = !std::is_enum_v<std::decay_t<T>>;
+}  // namespace kagome::scale