1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// Copyright 2016 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package symbolizer
import (
"bufio"
"bytes"
"strconv"
"github.com/google/syzkaller/pkg/osutil"
)
type Symbol struct {
Addr uint64
Size int
}
// ReadSymbols returns list of text symbols in the binary bin.
func ReadSymbols(bin string) (map[string][]Symbol, error) {
cmd := osutil.Command("nm", "-Ptx", bin)
stdout, err := cmd.StdoutPipe()
if err != nil {
return nil, err
}
defer stdout.Close()
if err := cmd.Start(); err != nil {
return nil, err
}
defer cmd.Wait()
symbols := make(map[string][]Symbol)
s := bufio.NewScanner(stdout)
text := [][]byte{[]byte(" t "), []byte(" T ")}
for s.Scan() {
// A line looks as: "snb_uncore_msr_enable_box t ffffffff8104db90 0000000000000059"
ln := s.Bytes()
if !bytes.Contains(ln, text[0]) && !bytes.Contains(ln, text[1]) {
continue
}
sp1 := bytes.IndexByte(ln, ' ')
if sp1 == -1 {
continue
}
if !bytes.HasPrefix(ln[sp1:], text[0]) && !bytes.HasPrefix(ln[sp1:], text[1]) {
continue
}
sp2 := sp1 + len(text[0])
sp3 := bytes.IndexByte(ln[sp2:], ' ')
if sp3 == -1 {
continue
}
sp3 += sp2
addr, err := strconv.ParseUint(string(ln[sp2:sp3]), 16, 64)
if err != nil {
continue
}
size, err := strconv.ParseUint(string(ln[sp3+1:]), 16, 64)
if err != nil || size == 0 {
continue
}
name := string(ln[:sp1])
// Note: sizes reported by kernel do not match nm.
// Kernel probably subtracts address of this symbol from address of the next symbol.
// We could do the same, but for now we just round up size to 16.
symbols[name] = append(symbols[name], Symbol{addr, int(size+15) / 16 * 16})
}
if err := s.Err(); err != nil {
return nil, err
}
return symbols, nil
}
|
