refactoring

patcher/src/runtime_data.rs

@@ -0,0 +1,138 @@
+use androscalpel::{IdMethod, IdType};
+use std::collections::{HashMap, HashSet};
+use std::path::PathBuf;
+
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
+pub struct RuntimeData {
+    pub invoke_data: Vec<ReflectionInvokeData>,
+    pub class_new_inst_data: Vec<ReflectionClassNewInstData>,
+    pub cnstr_new_inst_data: Vec<ReflectionCnstrNewInstData>,
+    pub dyn_code_load: Vec<DynamicCodeLoadingData>,
+}
+
+impl RuntimeData {
+    /// List all the methods that made reflection calls.
+    pub fn get_method_referenced(&self) -> HashSet<IdMethod> {
+        self.invoke_data
+            .iter()
+            .map(|data| data.caller_method.clone())
+            .chain(
+                self.class_new_inst_data
+                    .iter()
+                    .map(|data| data.caller_method.clone())
+                    .chain(
+                        self.cnstr_new_inst_data
+                            .iter()
+                            .map(|data| data.caller_method.clone()),
+                    ),
+            )
+            .collect()
+    }
+
+    /// List all data collected from called to `java.lang.reflect.Method.invoke()` made by
+    /// `method`.
+    pub fn get_invoke_data_for(
+        &self,
+        method: &IdMethod,
+    ) -> HashMap<String, Vec<ReflectionInvokeData>> {
+        let mut data = HashMap::new();
+        for val in self
+            .invoke_data
+            .iter()
+            .filter(|data| &data.caller_method == method)
+        {
+            let key = format!("THESEUS_ADDR_{:08X}", val.addr);
+            let entry = data.entry(key).or_insert(vec![]);
+            entry.push(val.clone());
+        }
+        data
+    }
+    /// List all data collected from called to `java.lang.Class.newInstance()` made by
+    /// `method`.
+    pub fn get_class_new_instance_data_for(
+        &self,
+        method: &IdMethod,
+    ) -> HashMap<String, Vec<ReflectionClassNewInstData>> {
+        let mut data = HashMap::new();
+        for val in self
+            .class_new_inst_data
+            .iter()
+            .filter(|data| &data.caller_method == method)
+        {
+            let key = format!("THESEUS_ADDR_{:08X}", val.addr);
+            let entry = data.entry(key).or_insert(vec![]);
+            entry.push(val.clone());
+        }
+        data
+    }
+    /// List all data collected from called to `java.lang.reflect.Constructor.newInstance()` made by
+    /// `method`.
+    pub fn get_cnstr_new_instance_data_for(
+        &self,
+        method: &IdMethod,
+    ) -> HashMap<String, Vec<ReflectionCnstrNewInstData>> {
+        let mut data = HashMap::new();
+        for val in self
+            .cnstr_new_inst_data
+            .iter()
+            .filter(|data| &data.caller_method == method)
+        {
+            let key = format!("THESEUS_ADDR_{:08X}", val.addr);
+            let entry = data.entry(key).or_insert(vec![]);
+            entry.push(val.clone());
+        }
+        data
+    }
+}
+
+/// Structure storing the runtime information of a reflection call using
+/// `java.lang.reflect.Method.invoke()`.
+#[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
+pub struct ReflectionInvokeData {
+    /// The method called by `java.lang.reflect.Method.invoke()`
+    pub method: IdMethod,
+    /// The method calling `java.lang.reflect.Method.invoke()`
+    pub caller_method: IdMethod,
+    /// Address where the call to `java.lang.reflect.Method.invoke()` was made in `caller_method`.
+    pub addr: usize,
+    /// If the method is static (static method don't take 'this' as argument)
+    pub is_static: bool,
+    // TODO: type of invoke?
+}
+
+/// Structure storing the runtime information of a reflection instanciation using
+/// `java.lang.Class.newInstance()`.
+#[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
+pub struct ReflectionClassNewInstData {
+    /// The constructor called by `java.lang.Class.newInstance()`
+    pub constructor: IdMethod,
+    /// The method calling `java.lang.Class.newInstance()`
+    pub caller_method: IdMethod,
+    /// Address where the call to `java.lang.Class.newInstance()` was made in `caller_method`.
+    pub addr: usize,
+}
+
+/// Structure storing the runtime information of a reflection instanciation using
+/// `java.lang.reflect.Constructor.newInstance()`.
+#[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
+pub struct ReflectionCnstrNewInstData {
+    /// The constructor calleb by `java.lang.reflect.Constructor.newInstance()`
+    pub constructor: IdMethod,
+    /// The method calling `java.lang.reflect.Constructor.newInstance()`
+    pub caller_method: IdMethod,
+    /// Address where the call to `java.lang.Class.newInstance()` was made in `caller_method`.
+    pub addr: usize,
+}
+
+/// Structure storing the runtime information of a dynamic code loading.
+#[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
+pub struct DynamicCodeLoadingData {
+    /// The type of the class loader used to load the code.
+    pub classloader_class: IdType,
+    /// An identifier for the classloader, valid for one specific run of the application.
+    pub classloader: String,
+    /// The path to the files storing the .dex/.apk/other bytecode loaded.
+    pub files: Vec<PathBuf>,
+}