dashboard/app: slightly relax command parsing

Users have misspelled test: multiple times.
Accept commands without the colon.

2 files changed, 107 insertions, 40 deletions

diff --git a/pkg/email/parser.go b/pkg/email/parser.go
index ce652c5da..2566ae1d5 100644
--- a/pkg/email/parser.go
+++ b/pkg/email/parser.go
@@ -25,12 +25,28 @@ type Email struct {
 	Subject     string
 	From        string
 	Cc          []string
-	Body        string // text/plain part
-	Patch       string // attached patch, if any
-	Command     string // command to bot (#syz is stripped)
-	CommandArgs string // arguments for the command
+	Body        string  // text/plain part
+	Patch       string  // attached patch, if any
+	Command     Command // command to bot
+	CommandArgs string  // arguments for the command
 }
 
+type Command int
+
+const (
+	CmdUnknown Command = iota
+	CmdNone
+	CmdUpstream
+	CmdFix
+	CmdDup
+	CmdUnDup
+	CmdTest
+	CmdInvalid
+	CmdUnCC
+
+	cmdTest5
+)
+
 const commandPrefix = "#syz "
 
 var groupsLinkRe = regexp.MustCompile("\nTo view this discussion on the web visit" +
@@ -87,7 +103,8 @@ func Parse(r io.Reader, ownEmails []string) (*Email, error) {
 		return nil, err
 	}
 	bodyStr := string(body)
-	patch, cmd, cmdArgs := "", "", ""
+	cmd := CmdNone
+	patch, cmdArgs := "", ""
 	if !fromMe {
 		for _, a := range attachments {
 			_, patch, _ = ParsePatch(string(a))
@@ -176,9 +193,10 @@ func CanonicalEmail(email string) string {
 // extractCommand extracts command to syzbot from email body.
 // Commands are of the following form:
 // ^#syz cmd args...
-func extractCommand(body []byte) (cmd, args string) {
+func extractCommand(body []byte) (cmd Command, args string) {
 	cmdPos := bytes.Index(append([]byte{'\n'}, body...), []byte("\n"+commandPrefix))
 	if cmdPos == -1 {
+		cmd = CmdNone
 		return
 	}
 	cmdPos += len(commandPrefix)
@@ -195,18 +213,41 @@ func extractCommand(body []byte) (cmd, args string) {
 	if cmdEnd1 := bytes.IndexByte(body[cmdPos:], ' '); cmdEnd1 != -1 && cmdEnd1 < cmdEnd {
 		cmdEnd = cmdEnd1
 	}
-	cmd = string(body[cmdPos : cmdPos+cmdEnd])
+	switch string(body[cmdPos : cmdPos+cmdEnd]) {
+	default:
+		cmd = CmdUnknown
+	case "":
+		cmd = CmdNone
+	case "upstream":
+		cmd = CmdUpstream
+	case "fix", "fix:":
+		cmd = CmdFix
+	case "dup", "dup:":
+		cmd = CmdDup
+	case "undup":
+		cmd = CmdUnDup
+	case "test", "test:":
+		cmd = CmdTest
+	case "invalid":
+		cmd = CmdInvalid
+	case "uncc", "uncc:":
+		cmd = CmdUnCC
+	case "test_5_arg_cmd":
+		cmd = cmdTest5
+	}
 	// Some email clients split text emails at 80 columns are the transformation is irrevesible.
 	// We try hard to restore what was there before.
 	// For "test:" command we know that there must be 2 tokens without spaces.
 	// For "fix:"/"dup:" we need a whole non-empty line of text.
 	switch cmd {
-	case "test:":
+	case CmdTest:
 		args = extractArgsTokens(body[cmdPos+cmdEnd:], 2)
-	case "test_5_arg_cmd":
+	case cmdTest5:
 		args = extractArgsTokens(body[cmdPos+cmdEnd:], 5)
-	case "fix:", "dup:":
+	case CmdFix, CmdDup:
 		args = extractArgsLine(body[cmdPos+cmdEnd:])
+	case CmdUnknown:
+		args = extractArgsLine(body[cmdPos:])
 	}
 	return
 }
diff --git a/pkg/email/parser_test.go b/pkg/email/parser_test.go
index 7df593642..274fcd9e7 100644
--- a/pkg/email/parser_test.go
+++ b/pkg/email/parser_test.go
@@ -17,14 +17,14 @@ func TestExtractCommand(t *testing.T) {
 		t.Run(fmt.Sprint(i), func(t *testing.T) {
 			cmd, args := extractCommand([]byte(test.body))
 			if cmd != test.cmd || !reflect.DeepEqual(args, test.args) {
-				t.Logf("expect: %q %q", test.cmd, test.args)
-				t.Logf("got   : %q %q", cmd, args)
+				t.Logf("expect: %v %q", test.cmd, test.args)
+				t.Logf("got   : %v %q", cmd, args)
 				t.Fail()
 			}
 			cmd, args = extractCommand([]byte(strings.Replace(test.body, "\n", "\r\n", -1)))
 			if cmd != test.cmd || !reflect.DeepEqual(args, test.args) {
-				t.Logf("expect: %q %q", test.cmd, test.args)
-				t.Logf("got   : %q %q", cmd, args)
+				t.Logf("expect: %v %q", test.cmd, test.args)
+				t.Logf("got   : %v %q", cmd, args)
 				t.Fail()
 			}
 		})
@@ -138,7 +138,7 @@ func TestParse(t *testing.T) {
 
 var extractCommandTests = []struct {
 	body string
-	cmd  string
+	cmd  Command
 	args string
 }{
 	{
@@ -146,17 +146,17 @@ var extractCommandTests = []struct {
 
 line1
 #syz  fix:  bar baz 	`,
-		cmd:  "fix:",
+		cmd:  CmdFix,
 		args: "bar baz",
 	},
 	{
 		body: `Hello,
 
 line1
-#syz fix:  bar  	 baz
+#syz fix  bar  	 baz
 line 2
 `,
-		cmd: "fix:",
+		cmd: CmdFix,
 		args: "bar  	 baz",
 	},
 	{
@@ -165,7 +165,7 @@ line1
 > #syz fix: bar   baz
 line 2
 `,
-		cmd:  "",
+		cmd:  CmdNone,
 		args: "",
 	},
 	// This is unfortunate case when a command is split by email client
@@ -175,15 +175,15 @@ line 2
 #syz test: git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git
 locking/core
 `,
-		cmd:  "test:",
+		cmd:  CmdTest,
 		args: "git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git locking/core",
 	},
 	{
 		body: `
-#syz test:
+#syz test
 git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git locking/core
 `,
-		cmd:  "test:",
+		cmd:  CmdTest,
 		args: "git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git locking/core",
 	},
 	{
@@ -193,7 +193,7 @@ git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git
 locking/core
 locking/core
 `,
-		cmd:  "test:",
+		cmd:  CmdTest,
 		args: "git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip.git locking/core",
 	},
 	{
@@ -205,14 +205,14 @@ locking/core
 arg4
 arg5
 `,
-		cmd:  "test_5_arg_cmd",
+		cmd:  cmdTest5,
 		args: "arg1 arg2 arg3 arg4 arg5",
 	},
 	{
 		body: `
 #syz test_5_arg_cmd arg1
 arg2`,
-		cmd:  "test_5_arg_cmd",
+		cmd:  cmdTest5,
 		args: "arg1 arg2",
 	},
 	{
@@ -220,7 +220,7 @@ arg2`,
 #syz test_5_arg_cmd arg1
 arg2
 `,
-		cmd:  "test_5_arg_cmd",
+		cmd:  cmdTest5,
 		args: "arg1 arg2",
 	},
 	{
@@ -230,7 +230,7 @@ arg2
 
  
 `,
-		cmd:  "test_5_arg_cmd",
+		cmd:  cmdTest5,
 		args: "arg1 arg2",
 	},
 	{
@@ -240,7 +240,7 @@ arg1 arg2 arg3
 arg4 arg5
  
 `,
-		cmd:  "fix:",
+		cmd:  CmdFix,
 		args: "arg1 arg2 arg3",
 	},
 	{
@@ -248,9 +248,34 @@ arg4 arg5
 #syz  fix: arg1 arg2 arg3
 arg4 arg5 
 `,
-		cmd:  "fix:",
+		cmd:  CmdFix,
 		args: "arg1 arg2 arg3",
 	},
+	{
+		body: `
+#syz dup: title goes here
+baz
+`,
+		cmd:  CmdDup,
+		args: "title goes here",
+	},
+	{
+		body: `
+#syz dup 
+title on the next line goes here  
+but not this one
+`,
+		cmd:  CmdDup,
+		args: "title on the next line goes here",
+	},
+	{
+		body: `
+#syz foo bar
+baz
+`,
+		cmd:  CmdUnknown,
+		args: "foo bar",
+	},
 }
 
 type ParseTest struct {
@@ -295,7 +320,7 @@ To post to this group, send email to syzkaller@googlegroups.com.
 To view this discussion on the web visit https://groups.google.com/d/msgid/syzkaller/abcdef@google.com.
 For more options, visit https://groups.google.com/d/optout.`,
 			Patch:       "",
-			Command:     "fix:",
+			Command:     CmdFix,
 			CommandArgs: "arg1 arg2 arg3",
 		}},
 
@@ -316,7 +341,8 @@ last line`,
 			Cc:        []string{"bob@example.com"},
 			Body: `text body
 last line`,
-			Patch: "",
+			Patch:   "",
+			Command: CmdNone,
 		}},
 
 	{`Date: Sun, 7 May 2017 19:54:00 -0700
@@ -339,7 +365,7 @@ text body
 second line
 last line`,
 			Patch:       "",
-			Command:     "invalid",
+			Command:     CmdInvalid,
 			CommandArgs: "",
 		}},
 
@@ -364,8 +390,8 @@ second line
 last line
 #syz command`,
 			Patch:       "",
-			Command:     "command",
-			CommandArgs: "",
+			Command:     CmdUnknown,
+			CommandArgs: "command",
 		}},
 
 	{`Date: Sun, 7 May 2017 19:54:00 -0700
@@ -413,7 +439,7 @@ IHQpKSB7CiAJCXNwaW5fdW5sb2NrKCZrY292LT5sb2NrKTsKIAkJcmV0dXJuOwo=
  		spin_unlock(&kcov->lock);
  		return;
 `,
-			Command:     "",
+			Command:     CmdNone,
 			CommandArgs: "",
 		}},
 
@@ -523,8 +549,8 @@ index 3d85747bd86e..a257b872a53d 100644
   error = vfs_statx(dfd, filename, flags, &stat, mask);
   if (error)
 `,
-			Command:     "test",
-			CommandArgs: "",
+			Command:     CmdTest,
+			CommandArgs: "commit 59372bbf3abd5b24a7f6f676a3968685c280f955",
 		}},
 
 	{`Sender: syzkaller-bugs@googlegroups.com
@@ -571,7 +597,7 @@ d
 
 #syz dup: BUG: unable to handle kernel NULL pointer dereference in corrupted
 `,
-		Command:     "dup:",
+		Command:     CmdDup,
 		CommandArgs: "BUG: unable to handle kernel NULL pointer dereference in corrupted",
 	}},
 
@@ -587,7 +613,7 @@ BUG: unable to handle kernel NULL pointer dereference in corrupted
 		Body: `#syz dup:
 BUG: unable to handle kernel NULL pointer dereference in corrupted
 `,
-		Command:     "dup:",
+		Command:     CmdDup,
 		CommandArgs: "BUG: unable to handle kernel NULL pointer dereference in corrupted",
 	}},
 
@@ -603,7 +629,7 @@ When freeing a lockf struct that already is part of a linked list, make sure to
 		Body: `#syz fix:
 When freeing a lockf struct that already is part of a linked list, make sure to
 `,
-		Command:     "fix:",
+		Command:     CmdFix,
 		CommandArgs: "When freeing a lockf struct that already is part of a linked list, make sure to",
 	}},
 }