Merge pull request #13 from nmi/add_flag

flag: add arguments for kernel & initrd path

flag/flag.go

@@ -0,0 +1,18 @@
+package flag
+
+import (
+	"flag"
+)
+
+func ParseArgs(args []string) (string, string, error) {
+	kernel := flag.String("k", "./bzImage", "kernel image path")
+	initrd := flag.String("i", "./initrd", "initrd path")
+
+	flag.Parse()
+
+	if err := flag.CommandLine.Parse(args[1:]); err != nil {
+		return "", "", err
+	}
+
+	return *kernel, *initrd, nil
+}

flag/flag_test.go

@@ -0,0 +1,32 @@
+package flag_test
+
+import (
+	"testing"
+
+	"github.com/nmi/gokvm/flag"
+)
+
+func TestParseArg(t *testing.T) {
+	t.Parallel()
+
+	args := []string{
+		"gokvm",
+		"-i",
+		"initrd_path",
+		"-k",
+		"kernel_path",
+	}
+
+	kernel, initrd, err := flag.ParseArgs(args)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if kernel != "kernel_path" {
+		t.Fatal("invalid kernel image path")
+	}
+
+	if initrd != "initrd_path" {
+		t.Fatal("invalid initrd path")
+	}
+}

main.go

@@ -4,17 +4,23 @@ import (
 	"bufio"
 	"os"
 
+	"github.com/nmi/gokvm/flag"
 	"github.com/nmi/gokvm/machine"
 	"github.com/nmi/gokvm/term"
 )
 
 func main() {
+	kernelPath, initrdPath, err := flag.ParseArgs(os.Args)
+	if err != nil {
+		panic(err)
+	}
+
 	m, err := machine.New()
 	if err != nil {
 		panic(err)
 	}
 
-	if err := m.LoadLinux("./bzImage", "./initrd"); err != nil {
+	if err := m.LoadLinux(kernelPath, initrdPath); err != nil {
 		panic(err)
 	}