mirror of https://github.com/restic/restic.git
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
// +build debug
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"os"
|
|
"restic/errors"
|
|
"restic/repository"
|
|
|
|
"github.com/pkg/profile"
|
|
)
|
|
|
|
var (
|
|
listenMemoryProfile string
|
|
memProfilePath string
|
|
cpuProfilePath string
|
|
insecure bool
|
|
|
|
prof interface {
|
|
Stop()
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
f := cmdRoot.PersistentFlags()
|
|
f.StringVar(&listenMemoryProfile, "listen-profile", "", "listen on this `address:port` for memory profiling")
|
|
f.StringVar(&memProfilePath, "mem-profile", "", "write memory profile to `dir`")
|
|
f.StringVar(&cpuProfilePath, "cpu-profile", "", "write cpu profile to `dir`")
|
|
f.BoolVar(&insecure, "insecure-kdf", false, "use insecure KDF settings")
|
|
}
|
|
|
|
type fakeTestingTB struct{}
|
|
|
|
func (fakeTestingTB) Logf(msg string, args ...interface{}) {
|
|
fmt.Fprintf(os.Stderr, msg, args...)
|
|
}
|
|
|
|
func runDebug() error {
|
|
if listenMemoryProfile != "" {
|
|
fmt.Fprintf(os.Stderr, "running memory profile HTTP server on %v\n", listenMemoryProfile)
|
|
go func() {
|
|
err := http.ListenAndServe(listenMemoryProfile, nil)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "memory profile listen failed: %v\n", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
if memProfilePath != "" && cpuProfilePath != "" {
|
|
return errors.Fatal("only one profile (memory or CPU) may be activated at the same time")
|
|
}
|
|
|
|
if memProfilePath != "" {
|
|
prof = profile.Start(profile.Quiet, profile.MemProfile, profile.ProfilePath(memProfilePath))
|
|
} else if cpuProfilePath != "" {
|
|
prof = profile.Start(profile.Quiet, profile.CPUProfile, profile.ProfilePath(cpuProfilePath))
|
|
}
|
|
|
|
if insecure {
|
|
repository.TestUseLowSecurityKDFParameters(fakeTestingTB{})
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func shutdownDebug() {
|
|
if prof != nil {
|
|
prof.Stop()
|
|
}
|
|
}
|