cairo-lang-macro fix null-byte panics

commit-id:e5393b42
software-mansion · Nov 4, 2024 · b07cfb1 · b07cfb1
1 parent 702ddcf
commit b07cfb1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugins/cairo-lang-macro/src/types/conversion.rs b/plugins/cairo-lang-macro/src/types/conversion.rs
@@ -95,7 +95,9 @@ impl TokenStream {
     ///
     /// # Safety
     #[doc(hidden)]
-    pub fn into_stable(self) -> StableTokenStream {
+    pub fn into_stable(mut self) -> StableTokenStream {
+        // Remove 0 bytes so conversion will always work.
+        self.value.retain(|c| c != '\0');
         let cstr = CString::new(self.value).unwrap();
         StableTokenStream {
             value: cstr.into_raw(),