refactor: move commands to cmd/

cmd/gen.go

@@ -0,0 +1,57 @@
+// Copyright 2018 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 cmd
+
+import (
+	"fmt"
+
+	"github.com/naoina/toml"
+
+	"code.dumpstack.io/tools/out-of-tree/config"
+	"code.dumpstack.io/tools/out-of-tree/distro"
+)
+
+type GenCmd struct {
+	Type string `enum:"module,exploit" required:"" help:"module/exploit"`
+}
+
+func (cmd *GenCmd) Run(g *Globals) (err error) {
+	switch cmd.Type {
+	case "module":
+		err = genConfig(config.KernelModule)
+	case "exploit":
+		err = genConfig(config.KernelExploit)
+	}
+	return
+}
+
+func genConfig(at config.ArtifactType) (err error) {
+	a := config.Artifact{
+		Name: "Put name here",
+		Type: at,
+	}
+	a.Targets = append(a.Targets, config.Target{
+		Distro: distro.Distro{ID: distro.Ubuntu, Release: "18.04"},
+		Kernel: config.Kernel{Regex: ".*"},
+	})
+	a.Targets = append(a.Targets, config.Target{
+		Distro: distro.Distro{ID: distro.Debian, Release: "8"},
+		Kernel: config.Kernel{Regex: ".*"},
+	})
+	a.Preload = append(a.Preload, config.PreloadModule{
+		Repo: "Repo name (e.g. https://github.com/openwall/lkrg)",
+	})
+	a.Patches = append(a.Patches, config.Patch{
+		Path: "/path/to/profiling.patch",
+	})
+
+	buf, err := toml.Marshal(&a)
+	if err != nil {
+		return
+	}
+
+	fmt.Print(string(buf))
+	return
+}