eutils_rs/proc/
kallsyms.rs

1use anyhow::Result;
2use std::collections::HashSet;
3use std::fs::File;
4use std::io::{self, BufRead};
5use once_cell::sync::Lazy;
6use std::sync::Mutex;
7
8pub static GLOBAL_KAKKSYMS: Lazy<Mutex<Kallsyms>> = Lazy::new(|| {
9    let ksyms = Kallsyms::new();
10    Mutex::new(ksyms)
11});
12
13#[derive(Debug, Default)]
14pub struct Kallsyms {
15    syms: Vec<(String, u64)>,
16    hs: HashSet<String>,
17}
18
19impl TryFrom<&str> for Kallsyms {
20    type Error = anyhow::Error;
21    fn try_from(path: &str) -> Result<Self> {
22        let mut ksyms = Kallsyms::new();
23        let file = File::open(path)?;
24        let lines = io::BufReader::new(file).lines();
25        for line in lines {
26            if let Ok(l) = line {
27                let mut iter = l.trim().split_whitespace();
28                if let Some(x) = iter.next() {
29                    iter.next();
30                    if let Some(y) = iter.next() {
31                        ksyms.insert(y.to_string(), u64::from_str_radix(x, 16)?);
32                    }
33                }
34            }
35        }
36        ksyms.sort();
37        log::debug!(
38            "Load ksyms done from {:?}, symbols length: {}",
39            path,
40            ksyms.get_ksyms_num()
41        );
42        Ok(ksyms)
43    }
44}
45
46impl Kallsyms {
47    pub fn new() -> Self {
48        Kallsyms {
49            syms: Vec::new(),
50            hs: HashSet::default(),
51        }
52    }
53
54    fn insert(&mut self, sym_name: String, sym_addr: u64) {
55        self.syms.push((sym_name.clone(), sym_addr));
56        self.hs.insert(sym_name);
57    }
58
59    fn get_ksyms_num(&self) -> usize {
60        self.syms.len()
61    }
62
63    fn sort(&mut self) {
64        self.syms.sort_by(|a, b| a.1.cmp(&b.1));
65    }
66
67    pub fn has_sym(&self, sym_name: &str) -> bool {
68        self.hs.contains(sym_name)
69    }
70
71    pub fn addr_to_sym(&self, addr: u64) -> String {
72        let mut start = 0;
73        let mut end = self.syms.len() - 1;
74        let mut mid;
75        let mut sym_addr;
76
77        while start < end {
78            mid = start + (end - start + 1) / 2;
79            sym_addr = self.syms[mid].1;
80
81            if sym_addr <= addr {
82                start = mid;
83            } else {
84                end = mid - 1;
85            }
86        }
87
88        if start == end && self.syms[start].1 <= addr {
89            let mut name = self.syms[start].0.clone();
90            name.push_str(&format!("+{}", addr - self.syms[start].1 - 1));
91            return name;
92        }
93
94        return String::from("Not Found");
95    }
96}
97
98#[cfg(test)]
99mod tests {
100    use super::*;
101    #[test]
102    fn test_ksyms_try_from() {
103        assert_eq!(Kallsyms::try_from("/proc/kallsyms").is_ok(), true);
104        assert_eq!(Kallsyms::try_from("/3124/2123").is_ok(), false);
105    }
106
107    #[test]
108    fn test_ksyms_has_sym() {
109        let ksym = Kallsyms::try_from("/proc/kallsyms").unwrap();
110        assert_eq!(ksym.has_sym("schedule"), true);
111        assert_eq!(ksym.has_sym("5813478"), false);
112    }
113}
eutils_rs/proc/kallsyms.rs

eutils_rs/proc/
kallsyms.rs
