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
|
// Copyright 2015 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 manager
import (
"fmt"
"sync"
"github.com/google/syzkaller/pkg/cover"
"github.com/google/syzkaller/pkg/cover/backend"
"github.com/google/syzkaller/pkg/log"
"github.com/google/syzkaller/pkg/mgrconfig"
"github.com/google/syzkaller/pkg/vminfo"
)
type ReportGeneratorWrapper struct {
cfg *mgrconfig.Config
modules []*vminfo.KernelModule
mu sync.Mutex
initialized bool
cached *cover.ReportGenerator
}
func ReportGeneratorCache(cfg *mgrconfig.Config) *ReportGeneratorWrapper {
return &ReportGeneratorWrapper{cfg: cfg}
}
func (w *ReportGeneratorWrapper) Get() (*cover.ReportGenerator, error) {
w.mu.Lock()
defer w.mu.Unlock()
if !w.initialized {
return nil, fmt.Errorf("report generator creation before Init() is called")
}
if w.cached == nil {
log.Logf(0, "initializing coverage information...")
rg, err := cover.MakeReportGenerator(w.cfg, w.modules)
if err != nil {
return nil, err
}
w.cached = rg
}
return w.cached, nil
}
func (w *ReportGeneratorWrapper) Init(modules []*vminfo.KernelModule) {
w.mu.Lock()
defer w.mu.Unlock()
if w.initialized {
panic("Init() called twice")
}
w.initialized = true
w.modules = modules
}
func (w *ReportGeneratorWrapper) Reset() {
w.mu.Lock()
defer w.mu.Unlock()
w.cached = nil
}
func CoverToPCs(cfg *mgrconfig.Config, cov []uint64) []uint64 {
pcs := make([]uint64, 0, len(cov))
for _, pc := range cov {
prev := backend.PreviousInstructionPC(cfg.SysTarget, cfg.Type, pc)
pcs = append(pcs, prev)
}
return pcs
}
|
