aboutsummaryrefslogtreecommitdiffstats
path: root/pkg/covermerger/file_line_merger.go
blob: 8a54bfa225f851fa419607dfe6b525b5cf29205b (plain)
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
// Copyright 2024 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 covermerger

import "github.com/google/syzkaller/pkg/log"

func makeFileLineCoverMerger(fvs FileVersions, base RepoCommit) FileCoverageMerger {
	baseFile := ""
	baseFileExists := false
	for repoCommit, fv := range fvs {
		if repoCommit == base {
			baseFile = fv
			baseFileExists = true
			break
		}
	}
	if !baseFileExists {
		return &DeletedFileLineMerger{}
	}
	a := &FileLineCoverMerger{
		MergeResult: &MergeResult{
			HitCounts:   make(map[int]int64),
			FileExists:  true,
			LineDetails: make(map[int][]*FileRecord),
		},
		baseFile:   baseFile,
		matchers:   make(map[RepoCommit]*LineToLineMatcher),
		lostFrames: map[RepoCommit]int64{},
	}
	for repoBranch, fv := range fvs {
		a.matchers[repoBranch] = makeLineToLineMatcher(fv, baseFile)
	}
	return a
}

type FileLineCoverMerger struct {
	*MergeResult
	baseFile   string
	matchers   map[RepoCommit]*LineToLineMatcher
	lostFrames map[RepoCommit]int64
}

func (a *FileLineCoverMerger) Add(record *FileRecord) {
	if record.StartLine < 0 {
		// This record doesn't have information about line coverage.
		// The best we sometimes have is the function name.
		return
	}
	if a.matchers[record.RepoCommit] == nil {
		if record.HitCount > 0 {
			a.lostFrames[record.RepoCommit]++
		}
		return
	}
	if targetLine := a.matchers[record.RepoCommit].SameLinePos(record.StartLine); targetLine != -1 {
		a.HitCounts[targetLine] += int64(record.HitCount)
		a.LineDetails[targetLine] = append(a.LineDetails[targetLine], record)
	}
}

func (a *FileLineCoverMerger) Result() *MergeResult {
	for repoBranch, lostFrames := range a.lostFrames {
		log.Logf(1, "\t[warn] lost %d frames from repoCommit(%s, %s)",
			lostFrames, repoBranch.Repo, repoBranch.Commit)
	}
	return a.MergeResult
}