From 523000d75ac6c194ccaa3aec0d8b29ef07d3f504 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Thu, 6 Feb 2025 14:59:57 +0200
Subject: feat(zkvms/zkwasm): Proper handling of input strings and hashmaps

---
 zkvms/zkwasm/wrapper_macro/src/lib.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

(limited to 'zkvms/zkwasm/wrapper_macro')

diff --git a/zkvms/zkwasm/wrapper_macro/src/lib.rs b/zkvms/zkwasm/wrapper_macro/src/lib.rs
index e7a201c..76f49fc 100644
--- a/zkvms/zkwasm/wrapper_macro/src/lib.rs
+++ b/zkvms/zkwasm/wrapper_macro/src/lib.rs
@@ -64,10 +64,9 @@ fn return_string(readfn: &TokenTree) -> TokenStream {
     format!("
         {{
              let mut ret = Vec::new();
-             let mut current_char = read!({readfn} char);
-             while current_char != '\\0' {{
-                 ret.push(current_char);
-                 current_char = read!({readfn} char);
+             let size = read!({readfn} usize);
+             for _ in 0..size {{
+                 ret.push(read!({readfn} char));
              }}
              ret.into_iter().collect()
         }}
-- 
cgit v1.2.3