fix bug in prefix code: must stop one character before any potential match of an empty string

Fixes issue 308.

R=rsc
CC=golang-dev
http://codereview.appspot.com/157142

1 files changed, 6 insertions, 3 deletions

diff --git a/src/pkg/regexp/regexp.go b/src/pkg/regexp/regexp.go
index 8f17954d7..014a9fdc7 100644
--- a/src/pkg/regexp/regexp.go
+++ b/src/pkg/regexp/regexp.go
@@ -633,15 +633,18 @@ func (re *Regexp) setPrefix() {
 	var utf = make([]byte, utf8.UTFMax);
 	// First instruction is start; skip that.
 	i := re.inst.At(0).(instr).next().index();
+Loop:
 	for i < re.inst.Len() {
 		inst := re.inst.At(i).(instr);
 		// stop if this is not a char
 		if inst.kind() != _CHAR {
 			break
 		}
-		// stop if this char starts a closure; liberal but easy test: is an ALT next?
-		if re.inst.At(inst.next().index()).(instr).kind() == _ALT {
-			break
+		// stop if this char can be followed by a match for an empty string,
+		// which includes closures, ^, and $.
+		switch re.inst.At(inst.next().index()).(instr).kind() {
+		case _BOT, _EOT, _ALT:
+			break Loop
 		}
 		n := utf8.EncodeRune(inst.(*_Char).char, utf);
 		b = bytes.Add(b, utf[0:n]);