feat: add UPPER_BOUND & LOWER_BOUND to all our new types

Author: tdelabro

crates/starknet-types-core/src/contract_address.rs

@@ -25,6 +25,11 @@ use crate::{
 pub struct ContractAddress(PatriciaKey);
 
 impl ContractAddress {
+    /// Lower inclusive bound
+    pub const LOWER_BOUND: Self = Self::from_hex_unchecked("0x0");
+    /// Upper non-inclusive bound
+    pub const UPPER_BOUND: Self = Self(PatriciaKey::UPPER_BOUND);
+
     pub const ZERO: Self = Self::from_hex_unchecked("0x0");
     pub const ONE: Self = Self::from_hex_unchecked("0x1");
     pub const TWO: Self = Self::from_hex_unchecked("0x2");
@@ -91,6 +96,11 @@ impl FromStr for ContractAddress {
 }
 
 impl ContractAddress {
+    /// Create a new [ContractAddress] from an hex encoded string without checking it is a valid value.
+    ///
+    /// Should NEVER be used on user inputs,
+    /// as it can cause erroneous execution if dynamically initialized with bad values.
+    /// Should mostly be used at compilation time on hardcoded static string.
     pub const fn from_hex_unchecked(s: &'static str) -> ContractAddress {
         let patricia_key = PatriciaKey::from_hex_unchecked(s);
 
@@ -110,8 +120,6 @@ mod test {
 
     #[test]
     fn basic_values() {
-        assert!(ContractAddress::try_from(Felt::ZERO).is_err());
-        assert!(ContractAddress::try_from(Felt::ONE).is_err());
         assert!(ContractAddress::try_from(PATRICIA_KEY_UPPER_BOUND).is_err());
 
         let felt = Felt::TWO;

crates/starknet-types-core/src/patricia_key.rs

@@ -25,6 +25,15 @@ pub const PATRICIA_KEY_UPPER_BOUND: Felt =
 )]
 pub struct PatriciaKey(Felt);
 
+impl PatriciaKey {
+    /// Lower inclusive bound
+    pub const LOWER_BOUND: Self = Self(Felt::ZERO);
+    /// Upper non-inclusive bound
+    pub const UPPER_BOUND: Self = Self::from_hex_unchecked(
+        "0x800000000000000000000000000000000000000000000000000000000000001",
+    );
+}
+
 impl core::fmt::Display for PatriciaKey {
     fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
         write!(f, "{}", self.0)
@@ -110,6 +119,11 @@ impl FromStr for PatriciaKey {
 }
 
 impl PatriciaKey {
+    /// Create a new [PatriciaKey] from an hex encoded string without checking it is a valid value.
+    ///
+    /// Should NEVER be used on user inputs,
+    /// as it can cause erroneous execution if dynamically initialized with bad values.
+    /// Should mostly be used at compilation time on hardcoded static string.
     pub const fn from_hex_unchecked(s: &'static str) -> PatriciaKey {
         let felt = Felt::from_hex_unwrap(s);
 

crates/starknet-types-core/src/regular_contract_address.rs

@@ -31,6 +31,13 @@ use crate::{
 )]
 pub struct RegularContractAddress(ContractAddress);
 
+impl RegularContractAddress {
+    /// Lower inclusive bound
+    pub const LOWER_BOUND: Self = Self::from_hex_unchecked("0x4");
+    /// Upper non-inclusive bound
+    pub const UPPER_BOUND: Self = Self(ContractAddress::UPPER_BOUND);
+}
+
 impl core::fmt::Display for RegularContractAddress {
     fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
         write!(f, "{}", self.0)
@@ -167,7 +174,7 @@ impl Felt {
     /// https://docs.starknet.io/learn/protocol/state#special-addresses
     /// https://github.com/starkware-libs/sequencer/blob/ecd4779abef7bf345938a69f18ef70b6239d3a50/crates/blockifier/resources/blockifier_versioned_constants_0_15_0.json#L92-L97
     pub fn is_regular_contract_address(&self) -> bool {
-        self > &Felt::THREE && self < &PATRICIA_KEY_UPPER_BOUND
+        self > &RegularContractAddress::LOWER_BOUND && self < &PATRICIA_KEY_UPPER_BOUND
     }
 }
 
@@ -218,6 +225,11 @@ impl FromStr for RegularContractAddress {
 }
 
 impl RegularContractAddress {
+    /// Create a new [RegularContractAddress] from an hex encoded string without checking it is a valid value.
+    ///
+    /// Should NEVER be used on user inputs,
+    /// as it can cause erroneous execution if dynamically initialized with bad values.
+    /// Should mostly be used at compilation time on hardcoded static string.
     pub const fn from_hex_unchecked(s: &'static str) -> RegularContractAddress {
         let contract_address = ContractAddress::from_hex_unchecked(s);