From 5b31042a69018426894e7e0a4f1b4ebd84255bf3 Mon Sep 17 00:00:00 2001
From: Arni Hod <arnon@starkware.co>
Date: Sun, 7 Jul 2024 15:09:53 +0300
Subject: [PATCH] chore: set config pointers for size validation

---
 crates/gateway/src/lib.rs      |  1 +
 crates/gateway/src/pointers.rs | 44 ++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 crates/gateway/src/pointers.rs

diff --git a/crates/gateway/src/lib.rs b/crates/gateway/src/lib.rs
index ec8052af8..5aeef30a0 100644
--- a/crates/gateway/src/lib.rs
+++ b/crates/gateway/src/lib.rs
@@ -4,6 +4,7 @@ mod compiler_version;
 pub mod config;
 pub mod errors;
 pub mod gateway;
+pub mod pointers;
 mod rpc_objects;
 mod rpc_state_reader;
 mod state_reader;
diff --git a/crates/gateway/src/pointers.rs b/crates/gateway/src/pointers.rs
new file mode 100644
index 000000000..50e26efc9
--- /dev/null
+++ b/crates/gateway/src/pointers.rs
@@ -0,0 +1,44 @@
+/// Deals with config pointers.
+use std::sync::OnceLock;
+
+use papyrus_config::dumping::ser_pointer_target_param;
+use papyrus_config::{ParamPath, SerializedParam};
+
+type ConfigPointers = Vec<((ParamPath, SerializedParam), Vec<ParamPath>)>;
+
+const MAX_BYTECODE_SIZE: usize = 81_920;
+const MAX_RAW_CLASS_SIZE: usize = 4_089_446; // (3.9 * 2_f32.pow(20)) as usize;
+
+// TODO(Arni): Use this to code dedup.
+pub fn config_pointers() -> ConfigPointers {
+    static CONFIG_POINTERS: OnceLock<ConfigPointers> = OnceLock::new();
+    CONFIG_POINTERS
+        .get_or_init(|| {
+            vec![
+                (
+                    ser_pointer_target_param(
+                        "max_bytecode_size",
+                        &MAX_BYTECODE_SIZE,
+                        "The maximum bytecode size allowed for a contract.",
+                    ),
+                    vec![
+                        "gateway_config.stateless_tx_validator_config.max_bytecode_size".to_owned(),
+                        "gateway_config.gateway_compiler_config.max_bytecode_size".to_owned(),
+                    ],
+                ),
+                (
+                    ser_pointer_target_param(
+                        "max_raw_class_size",
+                        &MAX_RAW_CLASS_SIZE,
+                        "The maximum raw class size allowed for a contract.",
+                    ),
+                    vec![
+                        "gateway_config.stateless_tx_validator_config.max_raw_class_size"
+                            .to_owned(),
+                        "gateway_config.gateway_compiler_config.max_raw_class_size".to_owned(),
+                    ],
+                ),
+            ]
+        })
+        .to_vec()
+}