Katyusha's blog

1
// Package ksym 提供内核符号解析：从 /proc/kallsyms 将地址映射为函数名，
2
// 用于把 off-CPU 阻塞栈符号化（如 futex_wait_queue / __mutex_lock_slowpath），
3
// 形成可读、可回溯的"线程堆栈聚集"证据。
4
package ksym
5

6
import (
7
  "bufio"
8
  "fmt"
9
  "os"
10
  "sort"
11
  "strconv"
12
  "strings"
13
)
14

15
// Table 是按地址升序排列的内核符号表。
16
type Table struct {
17
  addrs []uint64
18
  names []string
19
}
20

21
// Load 读取 /proc/kallsyms（需 root，且 kptr_restrict 允许）。
22
func Load() (*Table, error) {
23
  f, err := os.Open("/proc/kallsyms")
24
  if err != nil {
25
    return nil, err
26
  }
27
  defer f.Close()
28

29
  t := &Table{}
30
  sc := bufio.NewScanner(f)
31
  for sc.Scan() {
32
    fields := strings.Fields(sc.Text())
33
    if len(fields) < 3 {
34
      continue
35
    }
36
    addr, err := strconv.ParseUint(fields[0], 16, 64)
37
    if err != nil || addr == 0 {
38
      continue
39
    }
40
    t.addrs = append(t.addrs, addr)
41
    t.names = append(t.names, fields[2])
42
  }
43
  if err := sc.Err(); err != nil {
44
    return nil, err
45
  }
46
  if len(t.addrs) == 0 {
47
    return nil, fmt.Errorf("kallsyms 为空（可能 kptr_restrict 受限，请用 root 运行）")
48
  }
49

50
  idx := make([]int, len(t.addrs))
51
  for i := range idx {
52
    idx[i] = i
53
  }
54
  sort.Slice(idx, func(a, b int) bool { return t.addrs[idx[a]] < t.addrs[idx[b]] })
55
  sa := make([]uint64, len(t.addrs))
56
  sn := make([]string, len(t.names))
57
  for i, j := range idx {
58
    sa[i] = t.addrs[j]
59
    sn[i] = t.names[j]
60
  }
61
  t.addrs, t.names = sa, sn
62
  return t, nil
63
}
64

65
// Resolve 返回包含 addr 的最近符号名；找不到时返回十六进制地址。
66
func (t *Table) Resolve(addr uint64) string {
67
  if t == nil || len(t.addrs) == 0 {
68
    return fmt.Sprintf("0x%x", addr)
69
  }
70
  i := sort.Search(len(t.addrs), func(i int) bool { return t.addrs[i] > addr })
71
  if i == 0 {
72
    return fmt.Sprintf("0x%x", addr)
73
  }
74
  return t.names[i-1]
75
}