From b9500de0bab5fa57c35747911a6392cfbf5df5c2 Mon Sep 17 00:00:00 2001
From: Mikhail Klementev <jollheef@riseup.net>
Date: Fri, 23 Nov 2018 09:11:18 +0000
Subject: [PATCH] Implements skeleton module/exploit config generator

---
 gen.go  | 30 ++++++++++++++++++++++++++++++
 main.go | 10 ++++++++++
 2 files changed, 40 insertions(+)
 create mode 100644 gen.go

diff --git a/gen.go b/gen.go
new file mode 100644
index 0000000..2fa768f
--- /dev/null
+++ b/gen.go
@@ -0,0 +1,30 @@
+// 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 main
+
+import (
+	"fmt"
+
+	"github.com/jollheef/out-of-tree/config"
+	"github.com/naoina/toml"
+)
+
+func genConfig(at config.ArtifactType) (err error) {
+	a := config.Artifact{
+		Name: "Put name here",
+		Type: at,
+	}
+	a.SupportedKernels = append(a.SupportedKernels, config.KernelMask{
+		config.Ubuntu, "18.04", ".*",
+	})
+
+	buf, err := toml.Marshal(&a)
+	if err != nil {
+		return
+	}
+
+	fmt.Print(string(buf))
+	return
+}
diff --git a/main.go b/main.go
index cdf28eb..f6e1cc2 100644
--- a/main.go
+++ b/main.go
@@ -58,6 +58,12 @@ func main() {
 	kernelCommand := app.Command("kernel", "Manipulate kernels")
 	kernelListCommand := kernelCommand.Command("list", "List kernels")
 
+	genCommand := app.Command("gen", "Generate .out-of-tree.toml skeleton")
+	genModuleCommand := genCommand.Command("module",
+		"Generate .out-of-tree.toml skeleton for kernel module")
+	genExploitCommand := genCommand.Command("exploit",
+		"Generate .out-of-tree.toml skeleton for kernel exploit")
+
 	// Check for required commands
 	for _, cmd := range []string{"timeout", "docker", "qemu"} {
 		_, err := exec.Command("which", cmd).CombinedOutput()
@@ -79,6 +85,10 @@ func main() {
 			*pewTest, *pewGuess, *qemuTimeout, *dockerTimeout)
 	case kernelListCommand.FullCommand():
 		err = kernelListHandler(kcfg)
+	case genModuleCommand.FullCommand():
+		err = genConfig(config.KernelModule)
+	case genExploitCommand.FullCommand():
+		err = genConfig(config.KernelExploit)
 	}
 
 	if err != nil {