convert instruction to an editable format WIP

androscalpel/src/code.rs

@@ -4,11 +4,11 @@ use std::collections::HashSet;
 
 use pyo3::prelude::*;
 
-use crate::{DexString, IdField, IdMethod, IdMethodType, IdType, MethodHandle};
+use crate::{DexString, IdField, IdMethod, IdMethodType, IdType, Instruction, MethodHandle};
 
 // TODO: make this easy to edit/manipulate, maybe move to Method
 
-type TmpHandlerType = (Vec<(IdType, u32)>, Option<u32>);
+// type TmpHandlerType = (Vec<(IdType, u32)>, Option<u32>);
 
 /// The code run by a method.
 #[pyclass]
@@ -33,12 +33,12 @@ pub struct Code {
     // TODO: implement OPcode
     /// The instructions.
     #[pyo3(get, set)]
-    pub insns: Vec<u16>,
+    pub insns: Vec<Instruction>,
     // TODO: currently unusable, juste a mapping ty TryItem
     // TODO: maybe implement as custom OPcode to make me easy to modify?
-    /// Try blocks
-    #[pyo3(get, set)]
-    pub tries: Vec<(u32, u16, TmpHandlerType)>,
+    // /// Try blocks
+    // #[pyo3(get, set)]
+    // pub tries: Vec<(u32, u16, TmpHandlerType)>,
     // TODO: currently unusable, juste a mapping ty TryItem
     // TODO: maybe implement as custom OPcode to make me easy to modify?
 }
@@ -62,6 +62,8 @@ impl Code {
 
     /// Return all strings referenced in the codes.
     pub fn get_all_strings(&self) -> HashSet<DexString> {
+        todo!()
+        /*
         let mut strings = HashSet::new();
         for (_, _, (list, _)) in &self.tries {
             for (ty, _) in list {
@@ -69,10 +71,13 @@ impl Code {
             }
         }
         strings
+        */
     }
 
     /// Return all types referenced in the codes.
     pub fn get_all_types(&self) -> HashSet<IdType> {
+        todo!()
+        /*
         let mut types = HashSet::new();
         for (_, _, (list, _)) in &self.tries {
             for (ty, _) in list {
@@ -80,6 +85,7 @@ impl Code {
             }
         }
         types
+        */
     }
 
     /// Return all prototypes referenced in the codes.

androscalpel/src/lib.rs

@@ -24,7 +24,7 @@ pub use dex_id::*;
 pub use dex_string::*;
 pub use dex_writer::*;
 pub use field::*;
-//pub use instructions::*;
+pub use instructions::{CallSite, Instruction};
 pub use method::*;
 pub use method_handle::*;
 pub use scalar::*;