reactos/syzkaller
1
0
Fork 0
mirror of https://github.com/reactos/syzkaller.git synced 2024-11-26 21:00:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pkg/cmdprof: add package

Browse Source 
cmdprof simplifies cpu/memory profiling for command line tools. Use as:

	flag.Parse()
	defer cmdprof.Install
This commit is contained in:
Dmitry Vyukov 2020-04-30 16:02:57 +02:00
parent cc90e4763e
commit 136082ab38
3 changed files with 64 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
pkg/cmdprof/cmdprof.go Normal file
View File

@ -0,0 +1,57 @@
// Copyright 2020 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 cmdprof simplifies cpu/memory profiling for command line tools. Use as:
// flag.Parse()
// defer cmdprof.Install()()
package cmdprof
import (
"flag"
"fmt"
"os"
"runtime"
"runtime/pprof"
)
var (
flagCPUProfile = flag.String("cpuprofile", "", "write CPU profile to this file")
flagMEMProfile = flag.String("memprofile", "", "write memory profile to this file")
)
func Install() func() {
res := func() {}
failf := func(msg string, args ...interface{}) {
fmt.Fprintf(os.Stderr, msg+"\n", args...)
os.Exit(1)
}
if *flagCPUProfile != "" {
f, err := os.Create(*flagCPUProfile)
if err != nil {
failf("failed to create cpuprofile file: %v", err)
}
if err := pprof.StartCPUProfile(f); err != nil {
failf("failed to start cpu profile: %v", err)
}
res = func() {
pprof.StopCPUProfile()
f.Close()
}
}
if *flagMEMProfile != "" {
prev := res
res = func() {
prev()
f, err := os.Create(*flagMEMProfile)
if err != nil {
failf("failed to create memprofile file: %v", err)
}
defer f.Close()
runtime.GC()
if err := pprof.WriteHeapProfile(f); err != nil {
failf("failed to write mem profile: %v", err)
}
}
}
return res
}

20
sys/syz-sysgen/sysgen.go
View File

@ -13,14 +13,13 @@ import (
"os"
"path/filepath"
"reflect"
"runtime"
"runtime/pprof"
"sort"
"strings"
"sync"
"text/template"
"github.com/google/syzkaller/pkg/ast"
"github.com/google/syzkaller/pkg/cmdprof"
"github.com/google/syzkaller/pkg/compiler"
"github.com/google/syzkaller/pkg/hash"
"github.com/google/syzkaller/pkg/osutil"
@ -29,10 +28,6 @@ import (
"github.com/google/syzkaller/sys/targets"
)
var (
flagMemProfile = flag.String("memprofile", "", "write a memory profile to the file")
)
type SyscallData struct {
Name string
CallName string
@ -64,6 +59,7 @@ type ExecutorData struct {
func main() {
flag.Parse()
defer cmdprof.Install()()
var OSList []string
for OS := range targets.List {
@ -175,18 +171,6 @@ func main() {
}
writeExecutorSyscalls(data)
if *flagMemProfile != "" {
f, err := os.Create(*flagMemProfile)
if err != nil {
failf("could not create memory profile: ", err)
}
runtime.GC() // get up-to-date statistics
if err := pprof.WriteHeapProfile(f); err != nil {
failf("could not write memory profile: ", err)
}
f.Close()
}
}
func generate(target *targets.Target, prg *compiler.Prog, consts map[string]uint64, out io.Writer) {

35
tools/syz-check/check.go
View File

@ -31,12 +31,12 @@ import (
"os"
"path/filepath"
"runtime"
"runtime/pprof"
"sort"
"strings"
"unsafe"
"github.com/google/syzkaller/pkg/ast"
"github.com/google/syzkaller/pkg/cmdprof"
"github.com/google/syzkaller/pkg/compiler"
"github.com/google/syzkaller/pkg/osutil"
"github.com/google/syzkaller/pkg/symbolizer"
@ -46,11 +46,9 @@ import (
func main() {
var (
flagOS = flag.String("os", runtime.GOOS, "OS")
flagCPUProfile = flag.String("cpuprofile", "", "write CPU profile to this file")
flagMEMProfile = flag.String("memprofile", "", "write memory profile to this file")
flagDWARF = flag.Bool("dwarf", true, "do checking based on DWARF")
flagNetlink = flag.Bool("netlink", true, "do checking of netlink policies")
flagOS = flag.String("os", runtime.GOOS, "OS")
flagDWARF = flag.Bool("dwarf", true, "do checking based on DWARF")
flagNetlink = flag.Bool("netlink", true, "do checking of netlink policies")
)
arches := map[string]*string{"amd64": nil, "386": nil, "arm64": nil, "arm": nil}
for arch := range arches {
@ -61,30 +59,7 @@ func main() {
os.Exit(1)
}
flag.Parse()
if *flagCPUProfile != "" {
f, err := os.Create(*flagCPUProfile)
if err != nil {
failf("failed to create cpuprofile file: %v", err)
}
defer f.Close()
if err := pprof.StartCPUProfile(f); err != nil {
failf("failed to start cpu profile: %v", err)
}
defer pprof.StopCPUProfile()
}
if *flagMEMProfile != "" {
defer func() {
f, err := os.Create(*flagMEMProfile)
if err != nil {
failf("failed to create memprofile file: %v", err)
}
defer f.Close()
runtime.GC()
if err := pprof.WriteHeapProfile(f); err != nil {
failf("failed to write mem profile: %v", err)
}
}()
}
defer cmdprof.Install()()
var warnings []Warn
for arch, obj := range arches {
if *obj == "" {