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
| // Copyright 2018 Google Inc. All rights reserved.
| //
| // Licensed under the Apache License, Version 2.0 (the "License");
| // you may not use this file except in compliance with the License.
| // You may obtain a copy of the License at
| //
| // http://www.apache.org/licenses/LICENSE-2.0
| //
| // Unless required by applicable law or agreed to in writing, software
| // distributed under the License is distributed on an "AS IS" BASIS,
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
| // See the License for the specific language governing permissions and
| // limitations under the License.
|
| package symbol_inject
|
| import (
| "strconv"
| "testing"
| )
|
| func TestElfSymbolTable(t *testing.T) {
| testCases := []struct {
| file *mockElfFile
| symbol string
| offset, size uint64
| }{
| {
| file: elfSymbolTable1,
| symbol: "soong_build_number",
| offset: 0x1030,
| size: 128,
| },
| {
| file: elfSymbolTable2,
| symbol: "symbol1",
| offset: 0x1030,
| size: 128,
| },
| {
| file: elfSymbolTable2,
| symbol: "symbol2",
| offset: 0x10b0,
| size: 128,
| },
| }
|
| for i, testCase := range testCases {
| t.Run(strconv.Itoa(i), func(t *testing.T) {
| file, err := extractElfSymbols(testCase.file)
| if err != nil {
| t.Error(err.Error())
| }
| offset, size, err := findSymbol(file, testCase.symbol)
| if err != nil {
| t.Error(err.Error())
| }
| if offset != testCase.offset || size != testCase.size {
| t.Errorf("expected %x:%x, got %x:%x", testCase.offset, testCase.size, offset, size)
| }
| })
| }
| }
|
|
