Merge pull request #12 from os-checker/callee-source-code

Feat: add source code for callees

Author: zjp-CN

src/functions/mod.rs

@@ -33,7 +33,7 @@ pub fn analyze(tcx: TyCtxt, src_map: &SourceMap) -> Vec<SerFunction> {
     mono_items
         .iter()
         .filter_map(|item| Function::new(item, &callgraph, tcx, src_map, |x| !x.attrs.is_empty()))
-        .map(SerFunction::new)
+        .map(|fun| SerFunction::new(fun, tcx, src_map))
         .collect()
 }
 
@@ -51,6 +51,7 @@ pub struct Function {
     /// Raw function string, including name, signature, and body.
     func: String,
     /// Recursive fnction calls inside the body.
+    /// This set has insertion order, which is nice to check out call contexts.
     callees: IndexSet<Instance>,
 }
 

src/functions/serialization.rs

@@ -1,35 +1,54 @@
+use rustc_middle::ty::TyCtxt;
+use rustc_span::source_map::SourceMap;
 use serde::Serialize;
-use stable_mir::{CrateDef, DefId};
+use stable_mir::{CrateDef, DefId, mir::mono::Instance};
 
 /// A Rust funtion with its file source, attributes, and raw function content.
 #[derive(Debug, Serialize)]
 pub struct SerFunction {
     /// DefId in stable_mir.
-    pub def_id: String,
+    def_id: String,
     /// Every funtion must be declared in a specific file, even for those
     /// generated by macros.
-    pub file: String,
+    file: String,
     /// Attributes are attached the function, but it seems that attributes
     /// and function must be separated to query.
-    pub attrs: Vec<String>,
+    attrs: Vec<String>,
     /// Raw function string, including name, signature, and body.
-    pub func: String,
+    func: String,
     /// Recursive fnction calls inside the body.
-    pub callees: Vec<String>,
+    callees: Vec<Callee>,
 }
 
 impl SerFunction {
-    pub fn new(fun: super::Function) -> Self {
+    pub fn new(fun: super::Function, tcx: TyCtxt, src_map: &SourceMap) -> Self {
         SerFunction {
             def_id: format_def_id(&fun.def_id),
             file: fun.file,
             attrs: fun.attrs.iter().map(|a| a.as_str().to_owned()).collect(),
             func: fun.func,
-            callees: fun.callees.into_iter().map(|x| format_def_id(&x.def.def_id())).collect(),
+            callees: fun.callees.iter().map(|x| Callee::new(x, tcx, src_map)).collect(),
         }
     }
 }
 
 fn format_def_id(def_id: &DefId) -> String {
     format!("{def_id:?}")
 }
+
+#[derive(Debug, Serialize)]
+pub struct Callee {
+    def_id: String,
+    func: String,
+}
+
+impl Callee {
+    fn new(inst: &Instance, tcx: TyCtxt, src_map: &SourceMap) -> Self {
+        let def_id = format_def_id(&inst.def.def_id());
+        let func = inst
+            .body()
+            .map(|body| super::source_code_with(body.span, tcx, src_map))
+            .unwrap_or_default();
+        Callee { def_id, func }
+    }
+}