macro: fix 0E decode

binnvecs! for str arg

lib/examples/patchsomedll.rs

@@ -2,7 +2,6 @@ extern crate binnpatch;
 extern crate data_encoding;
 extern crate memmap;
 
-
 extern crate binnpatch_macro;
 
 use data_encoding::HEXUPPER;
@@ -11,15 +10,15 @@ use std::fs;
 use std::fs::{File, OpenOptions};
 use std::ops::{Deref, DerefMut};
 
-use binnpatch::{BytesPattern, FindPattern, ApplyPatch};
-use binnpatch_macro::binnvec;
+use binnpatch::{ApplyPatch, BytesPattern, FindPattern};
+use binnpatch_macro::binnvecs;
 
 fn main() {
     let src_file = "Some.dll";
     let dst_file = "Some.dll.patched";
-    let src_pattern =
-        "40 3E 1D ?? ?? 12 1C 7C 48 ?? 73 6F 02 22 ?? 61 19 4E 13 60 48 45 19 27 5B";
-    let replacement = binnvec!(06 5A 18 74 2D 62 12 6A 13 4A 2B 0E 6F 0F 36 7A 28 0A 37 67 0A 4B 01 73 0x14);
+    let src_pattern = "40 3E 1D ?? ?? 12 1C 7C 48 ?? 73 6F 02 22 ?? 61 19 4E 13 60 48 45 19 27 5B";
+    let replacement =
+        binnvecs!("06 5A 18 74 2D 62 12 6A 13 4A 2B 0E 6F 0F 36 7A 28 0A 37 67 0A 4B 01 73 14");
 
     let pattern = BytesPattern::from(src_pattern);
 

macro/src/lib.rs

@@ -1,30 +1,23 @@
 extern crate proc_macro;
 use proc_macro::{TokenStream, TokenTree};
 
-use proc_macro::{Punct, Literal, Spacing, Span, Ident, Group, Delimiter};
+use proc_macro::{Delimiter, Group, Ident, Literal, Punct, Spacing, Span};
 
-fn parse_str(s: &str) -> TokenStream {
-    s.parse().unwrap()
+macro_rules! forest {
+    ($($x:expr),+) => {[$( TokenTree::from($x), )+]};
 }
 
-#[proc_macro]
-pub fn binnvec(_item: TokenStream) -> TokenStream {
-    println!("- - - - -");
-    println!("{}", _item.to_string());
-    println!("- - - - -");
-    println!("{:?}", _item);
-    println!("- - - - -");
-    
+fn parse_hex_values(items: TokenStream) -> TokenStream {
     let mut new_tokens: Vec<TokenTree> = vec![];
     let comma: Punct = Punct::new(',', Spacing::Alone);
     let zero = Literal::u8_suffixed(0);
     let mut prev_separated: bool = true;
     let mut prev_is_mask: bool = false;
 
-    for token in _item {
+    for token in items {
         match token {
             TokenTree::Punct(ref x) if x.as_char() == ',' => {
-                new_tokens.push(token.into());
+                new_tokens.push(token);
                 prev_separated = true;
                 continue;
             }
@@ -47,8 +40,8 @@ pub fn binnvec(_item: TokenStream) -> TokenStream {
                 if !prev_separated {
                     new_tokens.push(comma.clone().into());
                 }
-                let lit_str = format!("0x{}u8", x.to_string());
-                let mut val:Literal = lit_str.as_str().parse().expect("parse literal from ident");
+                let lit_str = format!("0x{}u8", x);
+                let mut val: Literal = lit_str.as_str().parse().expect("parse literal from ident");
                 val.set_span(x.span());
                 new_tokens.push(val.into());
             }
@@ -59,33 +52,61 @@ pub fn binnvec(_item: TokenStream) -> TokenStream {
 
                 let in_str = x.to_string();
                 if in_str.starts_with("0x") {
-                    new_tokens.push(token.into());
+                    new_tokens.push(token);
                 } else {
                     let lit_str = format!("0x{}u8", in_str);
-                    let mut val:Literal = lit_str.as_str().parse().expect("parse literal");
+                    let mut val: Literal = lit_str.as_str().parse().expect("parse literal");
                     val.set_span(x.span());
                     new_tokens.push(val.into());
                 }
             }
-            _ => {
-                
-            }
+            _ => {}
         }
         prev_separated = false;
     }
+    let mut new_stream = TokenStream::new();
+    new_stream.extend(new_tokens);
+    new_stream
+}
 
-    let mut ts_new = TokenStream::new();
-    ts_new.extend(new_tokens);
+#[proc_macro]
+pub fn binnvec(_item: TokenStream) -> TokenStream {
+    let group_elements = parse_hex_values(_item);
 
-    let group = Group::new(Delimiter::Bracket, ts_new);
+    let group = Group::new(Delimiter::Bracket, group_elements);
     let mut result = TokenStream::new();
     let vec_ident = Ident::new("vec", Span::call_site());
     let macro_marc = Punct::new('!', Spacing::Joint);
-    result.extend([TokenTree::from(vec_ident), TokenTree::from(macro_marc)]);
-    result.extend([TokenTree::from(group)]);
-    
-    println!("result: \"{}\"", result.to_string());
+    result.extend(forest!(vec_ident, macro_marc, group));
 
     result
 }
 
+#[proc_macro]
+pub fn binnvecs(_item: TokenStream) -> TokenStream {
+    let first = _item.clone().into_iter().next();
+
+    match first {
+        Some(TokenTree::Literal(lit)) => {
+            let mut lit_str = lit.to_string();
+            if !(lit_str.starts_with('\"') && lit_str.ends_with('\"')) {
+                panic!("unexpected literal: \"{}\"", lit_str);
+            }
+            lit_str = lit_str.trim_matches('"').to_string();
+            let lit_str_copy = lit_str.clone();
+
+            (0u8..=9)
+                .map(|i| i.to_string())
+                .filter(|x| lit_str_copy.contains(format!("{}E", x).as_str()))
+                .for_each(|x| {
+                    lit_str =
+                        lit_str.replace(format!("{}E", x).as_str(), format!("0x{}E", x).as_str());
+                });
+            let stream: TokenStream = lit_str.parse().unwrap();
+            binnvec(stream)
+        }
+        _ => {
+            panic!("invalid item: \"{}\"", _item);
+        }
+    }
+}