evmmax: Poseidon hash amd Falcon primitive nttfw implementation in EVMMAX [DO NOT MERGE]

include/evmmax/evmmax.hpp

@@ -3,11 +3,94 @@
 // SPDX-License-Identifier: Apache-2.0
 #pragma once
 
+#include <evmc/evmc.hpp>
 #include <intx/intx.hpp>
+#include <memory>
 
 namespace evmmax
 {
 
+using intx::uint256;
+
+struct EXMMAXModStateInterface;
+
+/// Ephemeral EVMMAX (EVM Modular Arithmetic Extensions) state
+class EVMMAXState
+{
+    struct OpcodesGasCost
+    {
+    private:
+        int64_t addmodx;
+        int64_t mulmodx;
+
+    public:
+        explicit OpcodesGasCost(int64_t _addmodx, int64_t _mulmodx)
+          : addmodx(_addmodx), mulmodx(_mulmodx)
+        {}
+
+        int64_t calc_mul_gas_cost(uint8_t count) const { return (mulmodx * count + 999) / 1000; }
+        int64_t calc_add_gas_cost(uint8_t count) const { return (addmodx * count + 999) / 1000; }
+    };
+
+    OpcodesGasCost current_gas_cost = OpcodesGasCost{0, 0};
+
+    std::unique_ptr<EXMMAXModStateInterface> active_mod;  ///< Current active modulus
+
+public:
+    /// Create new modulus and activates it. In case the modulus already exists, activates it.
+    /// Deducts gas accordingly.
+    ///
+    /// \param gas_left Amount of gas before calling. Is modified by `setmodx`
+    /// \param mod_ptr Modulus big endian value memory pointer
+    /// \param mod_size Modulus size in bytes
+    /// \param vals_used Number of needed value slots
+    /// \return Status code.
+    [[nodiscard]] evmc_status_code setmodx(
+        int64_t& gas_left, const uint8_t* mod_ptr, size_t mod_size, size_t alloc_count) noexcept;
+
+    /// Loads EVMMAX values into EVM memory. Deducts gas accordingly.
+    /// Converts to the Montgomery form
+    [[nodiscard]] evmc_status_code loadx(
+        int64_t& gas_left, uint8_t* out_ptr, size_t val_idx, size_t num_vals) noexcept;
+
+    /// Stores EVM memory into EVMMAX value slots. Deducts gas accordingly.
+    /// Converts from the Montgomery form
+    [[nodiscard]] evmc_status_code storex(
+        int64_t& gas_left, const uint8_t* in_ptr, size_t dst_val_idx, size_t num_vals) noexcept;
+
+    /// Computes modular addition. Deducts gas accordingly. Operates on active modulus.
+    ///
+    /// for i in range(count):
+    ///    active_ctx.registers[dst_idx+i*dst_stride] =
+    ///    operation(active_ctx.registers[x_idx+i*x_stride], active_ctx.registers[y_idx+i*y_stride])
+    [[nodiscard]] evmc_status_code addmodx(int64_t& gas_left, size_t dst_idx, size_t dst_stride,
+        size_t x_idx, size_t x_stride, size_t y_idx, size_t y_stride, size_t count) noexcept;
+
+    /// Computes modular subtraction. Deducts gas accordingly. Operates on active modulus.
+    [[nodiscard]] evmc_status_code submodx(int64_t& gas_left, size_t dst_idx, size_t dst_stride,
+        size_t x_idx, size_t x_stride, size_t y_idx, size_t y_stride, size_t count) noexcept;
+
+    /// Computes modular multiplication. Deducts gas accordingly. Operates on active modulus.
+    [[nodiscard]] evmc_status_code mulmodx(int64_t& gas_left, size_t dst_idx, size_t dst_stride,
+        size_t x_idx, size_t x_stride, size_t y_idx, size_t y_stride, size_t count) noexcept;
+
+    /// Checks that there exists an active modulus
+    [[nodiscard]] bool is_activated() const noexcept;
+
+    /// Returns active modulus size.
+    /// Size in bytes needed to represent modulus.
+    [[nodiscard]] size_t active_mod_value_size() const noexcept;
+
+    void print_state(std::ostream& out) const noexcept;
+
+    void clear() noexcept;
+
+    EVMMAXState& operator=(EVMMAXState&&) noexcept;
+    explicit EVMMAXState() noexcept;
+    explicit EVMMAXState(EVMMAXState&&) noexcept;
+    virtual ~EVMMAXState();
+};
+
 /// The modular arithmetic operations for EVMMAX (EVM Modular Arithmetic Extensions).
 template <typename UintT>
 class ModArith

lib/evmmax/CMakeLists.txt

@@ -2,17 +2,14 @@
 # Copyright 2023 The evmone Authors.
 # SPDX-License-Identifier: Apache-2.0
 
-add_library(evmmax INTERFACE)
+add_library(evmmax STATIC)
 add_library(evmone::evmmax ALIAS evmmax)
-target_compile_features(evmmax INTERFACE cxx_std_20)
-target_include_directories(evmmax INTERFACE ${PROJECT_SOURCE_DIR}/include)
-target_link_libraries(evmmax INTERFACE intx::intx)
+target_compile_features(evmmax PUBLIC cxx_std_20)
+target_include_directories(evmmax PUBLIC ${PROJECT_SOURCE_DIR}/include)
+target_link_libraries(evmmax PUBLIC intx::intx PRIVATE evmc::evmc_cpp)
 
-if(CMAKE_VERSION VERSION_GREATER_EQUAL 3.19)
-    # We want to add the header file to the library for IDEs.
-    # However, cmake 3.18 does not support PRIVATE scope for INTERFACE libraries.
-    target_sources(
-        evmmax PRIVATE
-        ${PROJECT_SOURCE_DIR}/include/evmmax/evmmax.hpp
-    )
-endif()
+target_sources(
+    evmmax PRIVATE
+    ${PROJECT_SOURCE_DIR}/include/evmmax/evmmax.hpp
+    evmmax.cpp
+)