Profiling feature

forc-pkg/Cargo.toml

@@ -41,9 +41,14 @@ tracing = "0.1"
 url = { version = "2.2", features = ["serde"] }
 vec1 = "1.8.0"
 walkdir = "2"
+fuel-asm = { version = "0.57.0", features = ["serde"] }
+sway-ir = { version = "0.63.6", path = "../sway-ir" }
 
 [dev-dependencies]
 regex = "^1.10.2"
 
 [target.'cfg(not(target_os = "macos"))'.dependencies]
 sysinfo = "0.29"
+
+[features]
+profile = []

forc-pkg/src/pkg.rs

@@ -1785,6 +1785,7 @@ pub fn compile(
 
     // First, compile to an AST. We'll update the namespace and check for JSON ABI output.
     let ast_res = time_expr!(
+        pkg.name,
         "compile to ast",
         "compile_to_ast",
         sway_core::compile_to_ast(
@@ -1823,6 +1824,7 @@ pub fn compile(
     }
 
     let asm_res = time_expr!(
+        pkg.name,
         "compile ast to asm",
         "compile_ast_to_asm",
         sway_core::ast_to_asm(&handler, engines, &programs, &sway_build_config),
@@ -1837,6 +1839,7 @@ pub fn compile(
     let mut program_abi = match pkg.target {
         BuildTarget::Fuel => {
             let program_abi_res = time_expr!(
+                pkg.name,
                 "generate JSON ABI program",
                 "generate_json_abi",
                 fuel_abi::generate_program_abi(
@@ -1875,6 +1878,7 @@ pub fn compile(
             };
 
             let abi = time_expr!(
+                pkg.name,
                 "generate JSON ABI program",
                 "generate_json_abi",
                 evm_abi::generate_abi_program(typed_program, engines),
@@ -1899,15 +1903,16 @@ pub fn compile(
         .map(|finalized_entry| PkgEntry::from_finalized_entry(finalized_entry, engines))
         .collect::<anyhow::Result<_>>()?;
 
-    let asm = match asm_res {
+    let mut asm = match asm_res {
         Err(_) => return fail(handler),
         Ok(asm) => asm,
     };
 
     let bc_res = time_expr!(
+        pkg.name,
         "compile asm to bytecode",
         "compile_asm_to_bytecode",
-        sway_core::asm_to_bytecode(&handler, asm, source_map, engines.se(), &sway_build_config),
+        sway_core::asm_to_bytecode(&handler, &mut asm, source_map, engines.se(), &sway_build_config),
         Some(sway_build_config.clone()),
         metrics
     );
@@ -1957,9 +1962,62 @@ pub fn compile(
         warnings,
         metrics,
     };
+
+    #[cfg(feature = "profile")]
+    report_assembly_information(&asm, &compiled_package);
+
     Ok(compiled_package)
 }
 
+#[cfg(feature = "profile")]
+fn report_assembly_information(
+    compiled_asm: &sway_core::CompiledAsm,
+    compiled_package: &CompiledPackage,
+) {
+    let mut bytes = compiled_package.bytecode.bytes.clone();
+
+    let data_offset = u64::from_be_bytes(bytes.iter().skip(8).take(8).cloned().collect::<Vec<_>>().try_into().unwrap());
+    let data_section_size = bytes.len() as u64 - data_offset;
+
+    bytes.truncate(data_offset as usize);
+
+    let mut data_section_used = 0u64;
+
+    fn calculate_entry_size(entry: &sway_core::asm_generation::Entry) -> u64 {
+        let padding = match &entry.padding {
+            sway_ir::Padding::Left { target_size } => target_size,
+            sway_ir::Padding::Right { target_size } => target_size,
+        };
+
+        let value_size = match &entry.value {
+            sway_core::asm_generation::Datum::Byte(x) => std::mem::size_of_val(x) as u64,
+            sway_core::asm_generation::Datum::Word(x) => std::mem::size_of_val(x) as u64,
+            sway_core::asm_generation::Datum::ByteArray(x) => std::mem::size_of_val(x) as u64,
+            sway_core::asm_generation::Datum::Slice(slice) => slice.len() as u64,
+            sway_core::asm_generation::Datum::Collection(entries) => entries.iter().map(calculate_entry_size).sum(),
+        };
+
+        assert_eq!(*padding as u64, value_size);
+
+        value_size
+    }
+
+    for entry in &compiled_asm.0.data_section.value_pairs {
+        data_section_used += calculate_entry_size(entry);
+    }
+
+    let asm_information = sway_core::asm_generation::AsmInformation {
+        bytecode_size: bytes.len() as _,
+        data_section: sway_core::asm_generation::DataSectionInformation { 
+            size: data_section_size, 
+            used: data_section_used, 
+            value_pairs: compiled_asm.0.data_section.value_pairs.clone(),
+        }
+    };
+
+    println!("/forc-perf info {}", serde_json::to_string(&asm_information).unwrap());
+}
+
 impl PkgEntry {
     /// Returns whether this `PkgEntry` corresponds to a test.
     pub fn is_test(&self) -> bool {

forc/Cargo.toml

@@ -49,6 +49,7 @@ whoami = "1.1"
 
 [features]
 default = []
+profile = []
 test = []
 util = []
 uwu = ["uwuify"]

sway-core/Cargo.toml

@@ -49,9 +49,13 @@ thiserror = "1.0"
 tracing = "0.1"
 uint = "0.9"
 vec1 = "1.8.0"
+fuel-asm = "0.57.0"
 
 [target.'cfg(not(target_os = "macos"))'.dependencies]
 sysinfo = "0.29.0"
 
+[features]
+profile = []
+
 [lints.clippy]
 iter_over_hash_type = "deny"