84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
// Copyright 2019 Mikhail Klementev. All rights reserved.
|
|
// Use of this source code is governed by a AGPLv3 license
|
|
// (or later) that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"runtime"
|
|
"time"
|
|
|
|
"code.dumpstack.io/tools/out-of-tree/config"
|
|
)
|
|
|
|
type PackCmd struct {
|
|
Autogen bool `help:"kernel autogeneration"`
|
|
NoDownload bool `help:"do not download qemu image while kernel generation"`
|
|
ExploitRuns int64 `default:"4" help:"amount of runs of each exploit"`
|
|
KernelRuns int64 `default:"1" help:"amount of runs of each kernel"`
|
|
|
|
Tag string `help:"filter tag"`
|
|
|
|
Timeout time.Duration `help:"timeout after tool will not spawn new tests"`
|
|
}
|
|
|
|
func (cmd *PackCmd) Run(g *Globals) (err error) {
|
|
kcfg, err := config.ReadKernelConfig(g.Config.Kernels)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
db, err := openDatabase(g.Config.Database)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
stop := time.Time{} // never stop
|
|
if cmd.Timeout != 0 {
|
|
stop = time.Now().Add(cmd.Timeout)
|
|
}
|
|
|
|
threads := runtime.NumCPU()
|
|
|
|
tag := fmt.Sprintf("pack_run_%d", time.Now().Unix())
|
|
log.Println("Tag:", tag)
|
|
|
|
files, err := ioutil.ReadDir(g.WorkDir)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
for _, f := range files {
|
|
workPath := g.WorkDir + "/" + f.Name()
|
|
|
|
if !exists(workPath + "/.out-of-tree.toml") {
|
|
continue
|
|
}
|
|
|
|
if cmd.Autogen {
|
|
var perRegex int64 = 1
|
|
err = kernelAutogenHandler(workPath,
|
|
g.Config.Docker.Registry,
|
|
g.Config.Docker.Commands,
|
|
perRegex, false, !cmd.NoDownload)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
log.Println(f.Name())
|
|
|
|
pewHandler(kcfg, workPath, "", "", "", false, stop,
|
|
g.Config.Docker.Timeout.Duration,
|
|
g.Config.Qemu.Timeout.Duration,
|
|
cmd.KernelRuns, cmd.ExploitRuns, pathDevNull,
|
|
tag, threads, db, false)
|
|
}
|
|
|
|
return
|
|
}
|