mirror of
https://github.com/reactos/syzkaller.git
synced 2024-11-24 11:59:58 +00:00
c85e1c5be9
Better coverage reports with hierarchical coverage information, number of programs covering each line, handling of partially covered lines, links to programs covering lines. Fixes #682
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
// 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 main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/google/syzkaller/pkg/cover"
|
|
"github.com/google/syzkaller/pkg/osutil"
|
|
)
|
|
|
|
var (
|
|
initCoverOnce sync.Once
|
|
initCoverError error
|
|
initCoverVMOffset uint32
|
|
reportGenerator *cover.ReportGenerator
|
|
)
|
|
|
|
func initCover(kernelObj, kernelObjName, kernelSrc, kernelBuildSrc, arch, OS string) error {
|
|
initCoverOnce.Do(func() {
|
|
if kernelObj == "" {
|
|
initCoverError = fmt.Errorf("kernel_obj is not specified")
|
|
return
|
|
}
|
|
vmlinux := filepath.Join(kernelObj, kernelObjName)
|
|
reportGenerator, initCoverError = cover.MakeReportGenerator(vmlinux, kernelSrc, kernelBuildSrc, arch)
|
|
if initCoverError != nil {
|
|
return
|
|
}
|
|
initCoverVMOffset, initCoverError = getVMOffset(vmlinux, OS)
|
|
})
|
|
return initCoverError
|
|
}
|
|
|
|
func coverToPCs(cov []uint32, arch string) []uint64 {
|
|
pcs := make([]uint64, 0, len(cov))
|
|
for _, pc := range cov {
|
|
fullPC := cover.RestorePC(pc, initCoverVMOffset)
|
|
prevPC := cover.PreviousInstructionPC(arch, fullPC)
|
|
pcs = append(pcs, prevPC)
|
|
}
|
|
return pcs
|
|
}
|
|
|
|
func getVMOffset(vmlinux, OS string) (uint32, error) {
|
|
if OS == "freebsd" {
|
|
return 0xffffffff, nil
|
|
}
|
|
out, err := osutil.RunCmd(time.Hour, "", "readelf", "-SW", vmlinux)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
s := bufio.NewScanner(bytes.NewReader(out))
|
|
var addr uint32
|
|
for s.Scan() {
|
|
ln := s.Text()
|
|
pieces := strings.Fields(ln)
|
|
for i := 0; i < len(pieces); i++ {
|
|
if pieces[i] != "PROGBITS" {
|
|
continue
|
|
}
|
|
v, err := strconv.ParseUint("0x"+pieces[i+1], 0, 64)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("failed to parse addr in readelf output: %v", err)
|
|
}
|
|
if v == 0 {
|
|
continue
|
|
}
|
|
v32 := (uint32)(v >> 32)
|
|
if addr == 0 {
|
|
addr = v32
|
|
}
|
|
if addr != v32 {
|
|
return 0, fmt.Errorf("different section offsets in a single binary")
|
|
}
|
|
}
|
|
}
|
|
return addr, nil
|
|
}
|