package regexp
import (
"bytes"
"container/vector"
"io"
"os"
"strings"
"utf8"
)
var debug = false
var (
ErrInternal = os.NewError("internal error")
ErrUnmatchedLpar = os.NewError("unmatched '('")
ErrUnmatchedRpar = os.NewError("unmatched ')'")
ErrUnmatchedLbkt = os.NewError("unmatched '['")
ErrUnmatchedRbkt = os.NewError("unmatched ']'")
ErrBadRange = os.NewError("bad range in character class")
ErrExtraneousBackslash = os.NewError("extraneous backslash")
ErrBadClosure = os.NewError("repeated closure (**, ++, etc.)")
ErrBareClosure = os.NewError("closure applies to nothing")
ErrBadBackslash = os.NewError("illegal backslash escape")
)
type instr interface {
kind() int
next() instr
setNext(i instr)
index() int
setIndex(i int)
print()
}
type common struct {
_next instr
_index int
}
func (c *common) next() instr { return c._next }
func (c *common) setNext(i instr) { c._next = i }
func (c *common) index() int { return c._index }
func (c *common) setIndex(i int) { c._index = i }
type Regexp struct {
expr string
prefix string
prefixBytes []byte
inst *vector.Vector
start instr
prefixStart instr
nbra int
}
const (
_START = iota
_END
_BOT
_EOT
_CHAR
_CHARCLASS
_ANY
_NOTNL
_BRA
_EBRA
_ALT
_NOP
)
type _Start struct {
common
}
func (start *_Start) kind() int { return _START }
func (start *_Start) print() { print("start") }
type _End struct {
common
}
func (end *_End) kind() int { return _END }
func (end *_End) print() { print("end") }
type _Bot struct {
common
}
func (bot *_Bot) kind() int { return _BOT }
func (bot *_Bot) print() { print("bot") }
type _Eot struct {
common
}
func (eot *_Eot) kind() int { return _EOT }
func (eot *_Eot) print() { print("eot") }
type _Char struct {
common
char int
}
func (char *_Char) kind() int { return _CHAR }
func (char *_Char) print() { print("char ", string(char.char)) }
func newChar(char int) *_Char {
c := new(_Char)
c.char = char
return c
}
type _CharClass struct {
common
char int
negate bool
ranges *vector.IntVector
}
func (cclass *_CharClass) kind() int { return _CHARCLASS }
func (cclass *_CharClass) print() {
print("charclass")
if cclass.negate {
print(" (negated)")
}
for i := 0; i < cclass.ranges.Len(); i += 2 {
l := cclass.ranges.At(i)
r := cclass.ranges.At(i + 1)
if l == r {
print(" [", string(l), "]")
} else {
print(" [", string(l), "-", string(r), "]")
}
}
}
func (cclass *_CharClass) addRange(a, b int) {
cclass.ranges.Push(a)
cclass.ranges.Push(b)
}
func (cclass *_CharClass) matches(c int) bool {
for i := 0; i < cclass.ranges.Len(); i = i + 2 {
min := cclass.ranges.At(i)
max := cclass.ranges.At(i + 1)
if min <= c && c <= max {
return !cclass.negate
}
}
return cclass.negate
}
func newCharClass() *_CharClass {
c := new(_CharClass)
c.ranges = new(vector.IntVector)
return c
}
type _Any struct {
common
}
func (any *_Any) kind() int { return _ANY }
func (any *_Any) print() { print("any") }
type _NotNl struct {
common
}
func (notnl *_NotNl) kind() int { return _NOTNL }
func (notnl *_NotNl) print() { print("notnl") }
type _Bra struct {
common
n int
}
func (bra *_Bra) kind() int { return _BRA }
func (bra *_Bra) print() { print("bra", bra.n) }
type _Ebra struct {
common
n int
}
func (ebra *_Ebra) kind() int { return _EBRA }
func (ebra *_Ebra) print() { print("ebra ", ebra.n) }
type _Alt struct {
common
left instr
}
func (alt *_Alt) kind() int { return _ALT }
func (alt *_Alt) print() { print("alt(", alt.left.index(), ")") }
type _Nop struct {
common
}
func (nop *_Nop) kind() int { return _NOP }
func (nop *_Nop) print() { print("nop") }
func (re *Regexp) add(i instr) instr {
i.setIndex(re.inst.Len())
re.inst.Push(i)
return i
}
type parser struct {
re *Regexp
error os.Error
nlpar int
pos int
ch int
}
const endOfFile = -1
func (p *parser) c() int { return p.ch }
func (p *parser) nextc() int {
if p.pos >= len(p.re.expr) {
p.ch = endOfFile
} else {
c, w := utf8.DecodeRuneInString(p.re.expr[p.pos:])
p.ch = c
p.pos += w
}
return p.ch
}
func newParser(re *Regexp) *parser {
p := new(parser)
p.re = re
p.nextc()
return p
}
func special(c int) bool {
for _, r := range `\.+*?()|[]^$` {
if c == r {
return true
}
}
return false
}
func specialcclass(c int) bool {
for _, r := range `\-[]` {
if c == r {
return true
}
}
return false
}
func (p *parser) charClass() instr {
cc := newCharClass()
if p.c() == '^' {
cc.negate = true
p.nextc()
}
left := -1
for {
switch c := p.c(); c {
case ']', endOfFile:
if left >= 0 {
p.error = ErrBadRange
return nil
}
if cc.negate && cc.ranges.Len() == 2 &&
cc.ranges.At(0) == '\n' && cc.ranges.At(1) == '\n' {
nl := new(_NotNl)
p.re.add(nl)
return nl
}
if !cc.negate && cc.ranges.Len() == 2 && cc.ranges.At(0) == cc.ranges.At(1) {
c := newChar(cc.ranges.At(0))
p.re.add(c)
return c
}
p.re.add(cc)
return cc
case '-':
p.error = ErrBadRange
return nil
case '\\':
c = p.nextc()
switch {
case c == endOfFile:
p.error = ErrExtraneousBackslash
return nil
case c == 'n':
c = '\n'
case specialcclass(c):
default:
p.error = ErrBadBackslash
return nil
}
fallthrough
default:
p.nextc()
switch {
case left < 0:
if p.c() == '-' {
p.nextc()
left = c
} else {
cc.addRange(c, c)
}
case left <= c:
cc.addRange(left, c)
left = -1
default:
p.error = ErrBadRange
return nil
}
}
}
return nil
}
func (p *parser) term() (start, end instr) {
if p.error != nil {
return
}
switch c := p.c(); c {
case '|', endOfFile:
return nil, nil
case '*', '+':
p.error = ErrBareClosure
return
case ')':
if p.nlpar == 0 {
p.error = ErrUnmatchedRpar
return
}
return nil, nil
case ']':
p.error = ErrUnmatchedRbkt
return
case '^':
p.nextc()
start = p.re.add(new(_Bot))
return start, start
case '$':
p.nextc()
start = p.re.add(new(_Eot))
return start, start
case '.':
p.nextc()
start = p.re.add(new(_Any))
return start, start
case '[':
p.nextc()
start = p.charClass()
if p.error != nil {
return
}
if p.c() != ']' {
p.error = ErrUnmatchedLbkt
return
}
p.nextc()
return start, start
case '(':
p.nextc()
p.nlpar++
p.re.nbra++
nbra := p.re.nbra
start, end = p.regexp()
if p.c() != ')' {
p.error = ErrUnmatchedLpar
return
}
p.nlpar--
p.nextc()
bra := new(_Bra)
p.re.add(bra)
ebra := new(_Ebra)
p.re.add(ebra)
bra.n = nbra
ebra.n = nbra
if start == nil {
if end == nil {
p.error = ErrInternal
return
}
start = ebra
} else {
end.setNext(ebra)
}
bra.setNext(start)
return bra, ebra
case '\\':
c = p.nextc()
switch {
case c == endOfFile:
p.error = ErrExtraneousBackslash
return
case c == 'n':
c = '\n'
case special(c):
default:
p.error = ErrBadBackslash
return
}
fallthrough
default:
p.nextc()
start = newChar(c)
p.re.add(start)
return start, start
}
panic("unreachable")
}
func (p *parser) closure() (start, end instr) {
start, end = p.term()
if start == nil || p.error != nil {
return
}
switch p.c() {
case '*':
alt := new(_Alt)
p.re.add(alt)
end.setNext(alt)
alt.left = start
start = alt
end = alt
case '+':
alt := new(_Alt)
p.re.add(alt)
end.setNext(alt)
alt.left = start
end = alt
case '?':
alt := new(_Alt)
p.re.add(alt)
nop := new(_Nop)
p.re.add(nop)
alt.left = start
alt.setNext(nop)
end.setNext(nop)
start = alt
end = nop
default:
return
}
switch p.nextc() {
case '*', '+', '?':
p.error = ErrBadClosure
}
return
}
func (p *parser) concatenation() (start, end instr) {
for {
nstart, nend := p.closure()
if p.error != nil {
return
}
switch {
case nstart == nil:
if start == nil {
nop := p.re.add(new(_Nop))
return nop, nop
}
return
case start == nil:
start, end = nstart, nend
default:
end.setNext(nstart)
end = nend
}
}
panic("unreachable")
}
func (p *parser) regexp() (start, end instr) {
start, end = p.concatenation()
if p.error != nil {
return
}
for {
switch p.c() {
default:
return
case '|':
p.nextc()
nstart, nend := p.concatenation()
if p.error != nil {
return
}
alt := new(_Alt)
p.re.add(alt)
alt.left = start
alt.setNext(nstart)
nop := new(_Nop)
p.re.add(nop)
end.setNext(nop)
nend.setNext(nop)
start, end = alt, nop
}
}
panic("unreachable")
}
func unNop(i instr) instr {
for i.kind() == _NOP {
i = i.next()
}
return i
}
func (re *Regexp) eliminateNops() {
for i := 0; i < re.inst.Len(); i++ {
inst := re.inst.At(i).(instr)
if inst.kind() == _END {
continue
}
inst.setNext(unNop(inst.next()))
if inst.kind() == _ALT {
alt := inst.(*_Alt)
alt.left = unNop(alt.left)
}
}
}
func (re *Regexp) dump() {
print("prefix <", re.prefix, ">\n")
for i := 0; i < re.inst.Len(); i++ {
inst := re.inst.At(i).(instr)
print(inst.index(), ": ")
inst.print()
if inst.kind() != _END {
print(" -> ", inst.next().index())
}
print("\n")
}
}
func (re *Regexp) doParse() os.Error {
p := newParser(re)
start := new(_Start)
re.add(start)
s, e := p.regexp()
if p.error != nil {
return p.error
}
start.setNext(s)
re.start = start
e.setNext(re.add(new(_End)))
if debug {
re.dump()
println()
}
re.eliminateNops()
if debug {
re.dump()
println()
}
if p.error == nil {
re.setPrefix()
if debug {
re.dump()
println()
}
}
return p.error
}
func (re *Regexp) setPrefix() {
var b []byte
var utf = make([]byte, utf8.UTFMax)
i := re.inst.At(0).(instr).next().index()
Loop:
for i < re.inst.Len() {
inst := re.inst.At(i).(instr)
if inst.kind() != _CHAR {
break
}
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])
i = inst.next().index()
}
re.prefixStart = re.inst.At(i).(instr)
re.prefixBytes = b
re.prefix = string(b)
}
func Compile(str string) (regexp *Regexp, error os.Error) {
regexp = new(Regexp)
regexp.expr = str
regexp.inst = new(vector.Vector)
error = regexp.doParse()
return
}
func MustCompile(str string) *Regexp {
regexp, error := Compile(str)
if error != nil {
panicln(`regexp: compiling "`, str, `": `, error.String())
}
return regexp
}
func (re *Regexp) NumSubexp() int { return re.nbra }
type matchArena struct {
head *matchVec
len int
}
type matchVec struct {
m []int
ref int
next *matchVec
}
func (a *matchArena) new() *matchVec {
if a.head == nil {
const N = 10
block := make([]matchVec, N)
for i := 0; i < N; i++ {
b := &block[i]
b.next = a.head
a.head = b
}
}
m := a.head
a.head = m.next
m.ref = 0
if m.m == nil {
m.m = make([]int, a.len)
}
return m
}
func (a *matchArena) free(m *matchVec) {
m.ref--
if m.ref == 0 {
m.next = a.head
a.head = m
}
}
func (a *matchArena) copy(m *matchVec) *matchVec {
m1 := a.new()
copy(m1.m, m.m)
return m1
}
func (a *matchArena) noMatch() *matchVec {
m := a.new()
for i := range m.m {
m.m[i] = -1
}
m.ref = 1
return m
}
type state struct {
inst instr
match *matchVec
}
func (a *matchArena) addState(s []state, inst instr, match *matchVec, pos, end int) []state {
switch inst.kind() {
case _BOT:
if pos == 0 {
s = a.addState(s, inst.next(), match, pos, end)
}
return s
case _EOT:
if pos == end {
s = a.addState(s, inst.next(), match, pos, end)
}
return s
case _BRA:
n := inst.(*_Bra).n
match.m[2*n] = pos
s = a.addState(s, inst.next(), match, pos, end)
return s
case _EBRA:
n := inst.(*_Ebra).n
match.m[2*n+1] = pos
s = a.addState(s, inst.next(), match, pos, end)
return s
}
index := inst.index()
l := len(s)
for i := 0; i < l; i++ {
if s[i].inst.index() == index {
return s
}
}
if l == cap(s) {
s1 := make([]state, 2*l)[0:l]
copy(s1, s)
s = s1
}
s = s[0 : l+1]
s[l].inst = inst
s[l].match = match
match.ref++
if inst.kind() == _ALT {
s = a.addState(s, inst.(*_Alt).left, a.copy(match), pos, end)
s = a.addState(s, inst.next(), a.copy(match), pos, end)
}
return s
}
func (re *Regexp) doExecute(str string, bytestr []byte, pos int) []int {
var s [2][]state
s[0] = make([]state, 10)[0:0]
s[1] = make([]state, 10)[0:0]
in, out := 0, 1
var final state
found := false
end := len(str)
if bytestr != nil {
end = len(bytestr)
}
prefixed := false
if re.prefix != "" {
var advance int
if bytestr == nil {
advance = strings.Index(str[pos:], re.prefix)
} else {
advance = bytes.Index(bytestr[pos:], re.prefixBytes)
}
if advance == -1 {
return []int{}
}
pos += advance + len(re.prefix)
prefixed = true
}
arena := &matchArena{nil, 2 * (re.nbra + 1)}
for pos <= end {
if !found {
match := arena.noMatch()
match.m[0] = pos
if prefixed {
s[out] = arena.addState(s[out], re.prefixStart, match, pos, end)
prefixed = false
} else {
s[out] = arena.addState(s[out], re.start.next(), match, pos, end)
}
arena.free(match)
}
in, out = out, in
old := s[out]
for _, state := range old {
arena.free(state.match)
}
s[out] = old[0:0]
if found && len(s[in]) == 0 {
break
}
charwidth := 1
c := endOfFile
if pos < end {
if bytestr == nil {
c, charwidth = utf8.DecodeRuneInString(str[pos:end])
} else {
c, charwidth = utf8.DecodeRune(bytestr[pos:end])
}
}
pos += charwidth
for _, st := range s[in] {
switch st.inst.kind() {
case _BOT:
case _EOT:
case _CHAR:
if c == st.inst.(*_Char).char {
s[out] = arena.addState(s[out], st.inst.next(), st.match, pos, end)
}
case _CHARCLASS:
if st.inst.(*_CharClass).matches(c) {
s[out] = arena.addState(s[out], st.inst.next(), st.match, pos, end)
}
case _ANY:
if c != endOfFile {
s[out] = arena.addState(s[out], st.inst.next(), st.match, pos, end)
}
case _NOTNL:
if c != endOfFile && c != '\n' {
s[out] = arena.addState(s[out], st.inst.next(), st.match, pos, end)
}
case _BRA:
case _EBRA:
case _ALT:
case _END:
if !found ||
st.match.m[0] < final.match.m[0] ||
(st.match.m[0] == final.match.m[0] && pos-charwidth > final.match.m[1]) {
if final.match != nil {
arena.free(final.match)
}
final = st
final.match.ref++
final.match.m[1] = pos - charwidth
}
found = true
default:
st.inst.print()
panic("unknown instruction in execute")
}
}
}
if final.match == nil {
return nil
}
if re.prefix != "" && len(final.match.m) > 0 {
final.match.m[0] -= len(re.prefix)
}
return final.match.m
}
func (re *Regexp) ExecuteString(s string) (a []int) {
return re.doExecute(s, nil, 0)
}
func (re *Regexp) Execute(b []byte) (a []int) { return re.doExecute("", b, 0) }
func (re *Regexp) MatchString(s string) bool { return len(re.doExecute(s, nil, 0)) > 0 }
func (re *Regexp) Match(b []byte) bool { return len(re.doExecute("", b, 0)) > 0 }
func (re *Regexp) MatchStrings(s string) (a []string) {
r := re.doExecute(s, nil, 0)
if r == nil {
return nil
}
a = make([]string, len(r)/2)
for i := 0; i < len(r); i += 2 {
if r[i] != -1 {
a[i/2] = s[r[i]:r[i+1]]
}
}
return
}
func (re *Regexp) MatchSlices(b []byte) (a [][]byte) {
r := re.doExecute("", b, 0)
if r == nil {
return nil
}
a = make([][]byte, len(r)/2)
for i := 0; i < len(r); i += 2 {
if r[i] != -1 {
a[i/2] = b[r[i]:r[i+1]]
}
}
return
}
func MatchString(pattern string, s string) (matched bool, error os.Error) {
re, err := Compile(pattern)
if err != nil {
return false, err
}
return re.MatchString(s), nil
}
func Match(pattern string, b []byte) (matched bool, error os.Error) {
re, err := Compile(pattern)
if err != nil {
return false, err
}
return re.Match(b), nil
}
func (re *Regexp) ReplaceAllString(src, repl string) string {
lastMatchEnd := 0
searchPos := 0
buf := new(bytes.Buffer)
for searchPos <= len(src) {
a := re.doExecute(src, nil, searchPos)
if len(a) == 0 {
break
}
io.WriteString(buf, src[lastMatchEnd:a[0]])
if a[1] > lastMatchEnd || a[0] == 0 {
io.WriteString(buf, repl)
}
lastMatchEnd = a[1]
_, width := utf8.DecodeRuneInString(src[searchPos:])
if searchPos+width > a[1] {
searchPos += width
} else if searchPos+1 > a[1] {
searchPos++
} else {
searchPos = a[1]
}
}
io.WriteString(buf, src[lastMatchEnd:])
return buf.String()
}
func (re *Regexp) ReplaceAll(src, repl []byte) []byte {
lastMatchEnd := 0
searchPos := 0
buf := new(bytes.Buffer)
for searchPos <= len(src) {
a := re.doExecute("", src, searchPos)
if len(a) == 0 {
break
}
buf.Write(src[lastMatchEnd:a[0]])
if a[1] > lastMatchEnd || a[0] == 0 {
buf.Write(repl)
}
lastMatchEnd = a[1]
_, width := utf8.DecodeRune(src[searchPos:])
if searchPos+width > a[1] {
searchPos += width
} else if searchPos+1 > a[1] {
searchPos++
} else {
searchPos = a[1]
}
}
buf.Write(src[lastMatchEnd:])
return buf.Bytes()
}
func QuoteMeta(s string) string {
b := make([]byte, 2*len(s))
j := 0
for i := 0; i < len(s); i++ {
if special(int(s[i])) {
b[j] = '\\'
j++
}
b[j] = s[i]
j++
}
return string(b[0:j])
}
func (re *Regexp) allMatches(s string, b []byte, n int, deliver func(int, int)) {
var end int
if b == nil {
end = len(s)
} else {
end = len(b)
}
for pos, i, prevMatchEnd := 0, 0, -1; i < n && pos <= end; {
matches := re.doExecute(s, b, pos)
if len(matches) == 0 {
break
}
accept := true
if matches[1] == pos {
if matches[0] == prevMatchEnd {
accept = false
}
var width int
if b == nil {
_, width = utf8.DecodeRuneInString(s[pos:end])
} else {
_, width = utf8.DecodeRune(b[pos:end])
}
if width > 0 {
pos += width
} else {
pos = end + 1
}
} else {
pos = matches[1]
}
prevMatchEnd = matches[1]
if accept {
deliver(matches[0], matches[1])
i++
}
}
}
func (re *Regexp) AllMatches(b []byte, n int) [][]byte {
if n <= 0 {
n = len(b) + 1
}
result := make([][]byte, n)
i := 0
re.allMatches("", b, n, func(start, end int) {
result[i] = b[start:end]
i++
})
return result[0:i]
}
func (re *Regexp) AllMatchesString(s string, n int) []string {
if n <= 0 {
n = len(s) + 1
}
result := make([]string, n)
i := 0
re.allMatches(s, nil, n, func(start, end int) {
result[i] = s[start:end]
i++
})
return result[0:i]
}
func (re *Regexp) AllMatchesIter(b []byte, n int) <-chan []byte {
if n <= 0 {
n = len(b) + 1
}
c := make(chan []byte, 10)
go func() {
re.allMatches("", b, n, func(start, end int) { c <- b[start:end] })
close(c)
}()
return c
}
func (re *Regexp) AllMatchesStringIter(s string, n int) <-chan string {
if n <= 0 {
n = len(s) + 1
}
c := make(chan string, 10)
go func() {
re.allMatches(s, nil, n, func(start, end int) { c <- s[start:end] })
close(c)
}()
return c
}