wasm extractor - fix report parameters

framework/meta-lib/Cargo.toml

@@ -29,6 +29,7 @@ hex = "0.4"
 wasmparser = "0.216"
 wasmprinter = "0.216"
 semver = "1.0.20"
+wat = "1.217.0"
 
 [dependencies.multiversx-sc]
 version = "=0.53.0"

framework/meta-lib/src/code_report_json.rs

@@ -24,7 +24,7 @@ impl CodeReportJson {
             path: report.path.clone(),
             size,
             has_allocator: report.has_allocator,
-            has_panic: report.has_panic.clone(),
+            has_panic: report.has_panic.to_string(),
         }
     }
 }

framework/meta-lib/src/tools.rs

@@ -3,6 +3,7 @@ mod git_describe;
 pub(crate) mod report_creator;
 pub mod twiggy;
 mod wasm_extractor;
+mod wasm_extractor_test;
 mod wasm_opt;
 mod wasm_to_wat;
 

framework/meta-lib/src/tools/report_creator.rs

@@ -1,10 +1,28 @@
-pub const WITH_MESSAGE: &str = "with message";
-pub const WITHOUT_MESSAGE: &str = "without message";
+use std::fmt::Display;
+
+#[derive(Default)]
+pub enum PanicMessage {
+    #[default]
+    None,
+    WithoutMessage,
+    WithMessage,
+}
+
+impl Display for PanicMessage {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        let panic_status = match self {
+            PanicMessage::None => "None",
+            PanicMessage::WithoutMessage => "without message",
+            PanicMessage::WithMessage => "with message",
+        };
+        write!(f, "{}", panic_status)
+    }
+}
 
 pub struct ReportCreator {
     pub path: String,
     pub has_allocator: bool,
-    pub has_panic: String,
+    pub has_panic: PanicMessage,
 }
 
 impl ReportCreator {}

framework/meta-lib/src/tools/wasm_extractor.rs

@@ -1,17 +1,17 @@
 use colored::Colorize;
 use std::fs;
 use wasmparser::{
-    BinaryReaderError, DataSectionReader, FunctionBody, ImportSectionReader, Parser, Payload,
+    BinaryReaderError, DataSectionReader, FunctionBody, ImportSectionReader, Operator, Parser,
+    Payload,
 };
 
 use crate::ei::EIVersion;
 
-use super::report_creator::{ReportCreator, WITHOUT_MESSAGE, WITH_MESSAGE};
+use super::report_creator::{PanicMessage, ReportCreator};
 
 const PANIC_WITH_MESSAGE: &[u8; 16] = b"panic occurred: ";
 const PANIC_WITHOUT_MESSAGE: &[u8; 14] = b"panic occurred";
 const ERROR_FAIL_ALLOCATOR: &[u8; 27] = b"memory allocation forbidden";
-const MEMORY_GROW_OPCODE: u8 = 0x40;
 
 pub struct WasmInfo {
     pub imports: Vec<String>,
@@ -39,7 +39,7 @@ impl WasmInfo {
     }
 }
 
-fn populate_wasm_info(
+pub(crate) fn populate_wasm_info(
     path: String,
     wasm_data: Vec<u8>,
     extract_imports_enabled: bool,
@@ -49,25 +49,35 @@ fn populate_wasm_info(
     let mut allocator_trigger = false;
     let mut ei_check = false;
     let mut memory_grow_flag = false;
-    let mut has_panic = "none";
+    let mut has_panic: PanicMessage = PanicMessage::default();
 
     let parser = Parser::new(0);
     for payload in parser.parse_all(&wasm_data) {
         match payload? {
             Payload::ImportSection(import_section) => {
                 imports = extract_imports(import_section, extract_imports_enabled);
-                ei_check = is_ei_valid(imports.clone(), check_ei);
+                ei_check |= is_ei_valid(imports.clone(), check_ei);
             },
             Payload::DataSection(data_section) => {
-                allocator_trigger = is_fail_allocator_triggered(data_section.clone());
-                if is_panic_with_message_triggered(data_section.clone()) {
-                    has_panic = WITH_MESSAGE;
-                } else if is_panic_without_message_triggered(data_section) {
-                    has_panic = WITHOUT_MESSAGE;
+                allocator_trigger |= is_fail_allocator_triggered(data_section.clone());
+                match has_panic {
+                    PanicMessage::None => {
+                        if is_panic_with_message_triggered(data_section.clone()) {
+                            has_panic = PanicMessage::WithMessage;
+                        } else if is_panic_without_message_triggered(data_section) {
+                            has_panic = PanicMessage::WithoutMessage;
+                        }
+                    },
+                    PanicMessage::WithoutMessage => {
+                        if is_panic_with_message_triggered(data_section.clone()) {
+                            has_panic = PanicMessage::WithMessage;
+                        }
+                    },
+                    PanicMessage::WithMessage => continue,
                 }
             },
             Payload::CodeSectionEntry(code_section) => {
-                memory_grow_flag = is_mem_grow(code_section);
+                memory_grow_flag |= is_mem_grow(code_section);
             },
             _ => (),
         }
@@ -76,7 +86,7 @@ fn populate_wasm_info(
     let report = ReportCreator {
         path,
         has_allocator: allocator_trigger,
-        has_panic: has_panic.to_string(),
+        has_panic,
     };
 
     Ok(WasmInfo {
@@ -173,11 +183,15 @@ fn is_ei_valid(imports: Vec<String>, check_ei: &Option<EIVersion>) -> bool {
 }
 
 fn is_mem_grow(code_section: FunctionBody) -> bool {
-    let mut code = code_section.get_binary_reader();
-    while code.bytes_remaining() > 0 {
-        if code.read_u8().unwrap() == MEMORY_GROW_OPCODE {
+    let mut instructions_reader = code_section
+        .get_operators_reader()
+        .expect("Failed to get operators reader");
+
+    while let Ok(op) = instructions_reader.read() {
+        if let Operator::MemoryGrow { mem: _ } = op {
             return true;
         }
     }
+
     false
 }