diff options
Diffstat (limited to 'zkvms/zkwasm/wrapper_macro/src/lib.rs')
| -rw-r--r-- | zkvms/zkwasm/wrapper_macro/src/lib.rs | 7 |
1 files changed, 3 insertions, 4 deletions
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() }} |