From 804e0b487912d44782267c84b703387662828aa3 Mon Sep 17 00:00:00 2001
From: Mikhail Klementev <jollheef@riseup.net>
Date: Sun, 25 Nov 2018 14:53:56 +0000
Subject: [PATCH] Move gdb listen address to params

---
 debug.go | 5 ++---
 main.go  | 5 ++++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/debug.go b/debug.go
index 99320bf..5cd35c7 100644
--- a/debug.go
+++ b/debug.go
@@ -81,7 +81,7 @@ func interactive(q *qemu.QemuSystem) (err error) {
 	}
 }
 
-func debugHandler(kcfg config.KernelConfig, workPath string, kernRegex string,
+func debugHandler(kcfg config.KernelConfig, workPath, kernRegex, gdb string,
 	dockerTimeout time.Duration) (err error) {
 
 	ka, err := config.ReadArtifactConfig(workPath + "/.out-of-tree.toml")
@@ -103,8 +103,7 @@ func debugHandler(kcfg config.KernelConfig, workPath string, kernRegex string,
 	if err != nil {
 		return
 	}
-	gdb := "tcp::1234" // TODO param
-	q.Debug(gdb)       // TODO param
+	q.Debug(gdb)
 	coloredGdbAddress := aurora.BgGreen(aurora.Black(gdb))
 	fmt.Printf("[*] gdb runned on %s\n", coloredGdbAddress)
 
diff --git a/main.go b/main.go
index acfcd09..1096df2 100644
--- a/main.go
+++ b/main.go
@@ -67,6 +67,8 @@ func main() {
 	debugCommand := app.Command("debug", "Kernel debug environment")
 	debugCommandFlag := debugCommand.Flag("kernel", "Regex (first match)")
 	debugKernel := debugCommandFlag.Required().String()
+	debugFlagGDB := debugCommand.Flag("gdb", "Set gdb listen address")
+	debugGDB := debugFlagGDB.Default("tcp::1234").String()
 
 	// Check for required commands
 	for _, cmd := range []string{"timeout", "docker", "qemu"} {
@@ -94,7 +96,8 @@ func main() {
 	case genExploitCommand.FullCommand():
 		err = genConfig(config.KernelExploit)
 	case debugCommand.FullCommand():
-		err = debugHandler(kcfg, *path, *debugKernel, *dockerTimeout)
+		err = debugHandler(kcfg, *path, *debugKernel, *debugGDB,
+			*dockerTimeout)
 	}
 
 	if err != nil {