feat(apollo_starknet_os_program): define and implement static ProgramHash

Author: dorimedini-starkware

Cargo.lock

@@ -15,7 +15,10 @@ workspace = true
 
 [dependencies]
 cairo-vm.workspace = true
+serde = { workspace = true, features = ["derive"] }
 serde_json.workspace = true
+starknet-types-core.workspace = true
+thiserror.workspace = true
 
 [build-dependencies]
 apollo_infra_utils.workspace = true

crates/apollo_starknet_os_program/Cargo.toml

@@ -4,6 +4,10 @@ use std::sync::LazyLock;
 
 use cairo_vm::types::program::Program;
 
+use crate::program_hash::ProgramHash;
+
+pub mod program_hash;
+
 #[cfg(feature = "dump_source_files")]
 pub static CAIRO_FILES_MAP: LazyLock<HashMap<String, String>> = LazyLock::new(|| {
     serde_json::from_str(include_str!(concat!(env!("OUT_DIR"), "/cairo_files_map.json")))
@@ -17,3 +21,8 @@ pub static OS_PROGRAM: LazyLock<Program> = LazyLock::new(|| {
     )
     .expect("Failed to load the OS bytes.")
 });
+
+pub static PROGRAM_HASH: LazyLock<ProgramHash> = LazyLock::new(|| {
+    serde_json::from_str(include_str!("program_hash.json"))
+        .expect("Failed to deserialize program_hash.json.")
+});

crates/apollo_starknet_os_program/src/lib.rs

@@ -0,0 +1,3 @@
+{
+  "os": "0x1a2e53ac2b85c63107ef7232a6b96ee3d7b7a3fc25e0ba6e223e01a79e3739b"
+}

crates/apollo_starknet_os_program/src/program_hash.json

@@ -0,0 +1,71 @@
+use cairo_vm::types::builtin_name::BuiltinName;
+use cairo_vm::types::errors::program_errors::ProgramError;
+use serde::{Deserialize, Serialize};
+use starknet_types_core::felt::Felt;
+use starknet_types_core::hash::{Pedersen, StarkHash};
+
+use crate::OS_PROGRAM;
+
+#[cfg(test)]
+#[path = "program_hash_test.rs"]
+pub mod test;
+
+#[derive(thiserror::Error, Debug)]
+pub enum ProgramHashError {
+    #[error("Builtin name is too long: {0}.")]
+    BuiltinNameTooLong(BuiltinName),
+    #[error(transparent)]
+    Program(#[from] ProgramError),
+    #[error("Program data contains unexpected relocatable.")]
+    UnexpectedRelocatable,
+}
+
+#[derive(Deserialize, Serialize)]
+pub struct ProgramHash {
+    os: Felt,
+}
+
+const BOOTLOADER_VERSION: u8 = 0;
+
+fn pedersen_hash_chain(data: Vec<Felt>) -> Felt {
+    let length = Felt::from(data.len());
+    vec![length]
+        .into_iter()
+        .chain(data)
+        .rev()
+        .reduce(|x, y| Pedersen::hash(&y, &x))
+        .expect("Hash data chain is not empty.")
+}
+
+pub fn compute_os_program_hash() -> Result<Felt, ProgramHashError> {
+    let builtins = OS_PROGRAM
+        .iter_builtins()
+        .map(|builtin| {
+            let builtin_bytes = builtin.to_str().to_string().into_bytes();
+            if builtin_bytes.len() > 32 {
+                Err(ProgramHashError::BuiltinNameTooLong(*builtin))
+            } else {
+                let mut padded_builtin_bytes = [0].repeat(32 - builtin_bytes.len());
+                padded_builtin_bytes.extend(builtin_bytes);
+                Ok(Felt::from_bytes_be(
+                    padded_builtin_bytes
+                        .as_slice()
+                        .try_into()
+                        .expect("Padded bytes are 32 bytes long."),
+                ))
+            }
+        })
+        .collect::<Result<Vec<Felt>, _>>()?;
+    let program_header = vec![
+        Felt::from(BOOTLOADER_VERSION),
+        Felt::from(OS_PROGRAM.get_stripped_program()?.main),
+        Felt::from(builtins.len()),
+    ];
+    let data = OS_PROGRAM
+        .iter_data()
+        .map(|data| data.get_int().ok_or(ProgramHashError::UnexpectedRelocatable))
+        .collect::<Result<Vec<Felt>, _>>()?;
+
+    let data_chain: Vec<Felt> = program_header.into_iter().chain(builtins).chain(data).collect();
+    Ok(pedersen_hash_chain(data_chain))
+}

crates/apollo_starknet_os_program/src/program_hash.rs

@@ -0,0 +1,7 @@
+use crate::program_hash::compute_os_program_hash;
+use crate::PROGRAM_HASH;
+
+#[test]
+fn test_program_hash() {
+    assert_eq!(compute_os_program_hash().unwrap(), PROGRAM_HASH.os)
+}