mirror of https://github.com/tidwall/redcon.git
357 lines
7.5 KiB
Go
357 lines
7.5 KiB
Go
package redcon
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"math/rand"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestRandomCommands fills a bunch of random commands and test various
|
|
// ways that the reader may receive data.
|
|
func TestRandomCommands(t *testing.T) {
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
// build random commands.
|
|
gcmds := make([][]string, 10000)
|
|
for i := 0; i < len(gcmds); i++ {
|
|
args := make([]string, (rand.Int()%50)+1) // 1-50 args
|
|
for j := 0; j < len(args); j++ {
|
|
n := rand.Int() % 10
|
|
if j == 0 {
|
|
n++
|
|
}
|
|
arg := make([]byte, n)
|
|
for k := 0; k < len(arg); k++ {
|
|
arg[k] = byte(rand.Int() % 0xFF)
|
|
}
|
|
args[j] = string(arg)
|
|
}
|
|
gcmds[i] = args
|
|
}
|
|
// create a list of a buffers
|
|
var bufs []string
|
|
|
|
// pipe valid RESP commands
|
|
for i := 0; i < len(gcmds); i++ {
|
|
args := gcmds[i]
|
|
msg := fmt.Sprintf("*%d\r\n", len(args))
|
|
for j := 0; j < len(args); j++ {
|
|
msg += fmt.Sprintf("$%d\r\n%s\r\n", len(args[j]), args[j])
|
|
}
|
|
bufs = append(bufs, msg)
|
|
}
|
|
bufs = append(bufs, "RESET THE INDEX\r\n")
|
|
|
|
// pipe valid plain commands
|
|
for i := 0; i < len(gcmds); i++ {
|
|
args := gcmds[i]
|
|
var msg string
|
|
for j := 0; j < len(args); j++ {
|
|
quotes := false
|
|
var narg []byte
|
|
arg := args[j]
|
|
if len(arg) == 0 {
|
|
quotes = true
|
|
}
|
|
for k := 0; k < len(arg); k++ {
|
|
switch arg[k] {
|
|
default:
|
|
narg = append(narg, arg[k])
|
|
case ' ':
|
|
quotes = true
|
|
narg = append(narg, arg[k])
|
|
case '\\', '"', '*':
|
|
quotes = true
|
|
narg = append(narg, '\\', arg[k])
|
|
case '\r':
|
|
quotes = true
|
|
narg = append(narg, '\\', 'r')
|
|
case '\n':
|
|
quotes = true
|
|
narg = append(narg, '\\', 'n')
|
|
}
|
|
}
|
|
msg += " "
|
|
if quotes {
|
|
msg += "\""
|
|
}
|
|
msg += string(narg)
|
|
if quotes {
|
|
msg += "\""
|
|
}
|
|
}
|
|
if msg != "" {
|
|
msg = msg[1:]
|
|
}
|
|
msg += "\r\n"
|
|
bufs = append(bufs, msg)
|
|
}
|
|
bufs = append(bufs, "RESET THE INDEX\r\n")
|
|
|
|
// pipe valid RESP commands in broken chunks
|
|
lmsg := ""
|
|
for i := 0; i < len(gcmds); i++ {
|
|
args := gcmds[i]
|
|
msg := fmt.Sprintf("*%d\r\n", len(args))
|
|
for j := 0; j < len(args); j++ {
|
|
msg += fmt.Sprintf("$%d\r\n%s\r\n", len(args[j]), args[j])
|
|
}
|
|
msg = lmsg + msg
|
|
if len(msg) > 0 {
|
|
lmsg = msg[len(msg)/2:]
|
|
msg = msg[:len(msg)/2]
|
|
}
|
|
bufs = append(bufs, msg)
|
|
}
|
|
bufs = append(bufs, lmsg)
|
|
bufs = append(bufs, "RESET THE INDEX\r\n")
|
|
|
|
// pipe valid RESP commands in large broken chunks
|
|
lmsg = ""
|
|
for i := 0; i < len(gcmds); i++ {
|
|
args := gcmds[i]
|
|
msg := fmt.Sprintf("*%d\r\n", len(args))
|
|
for j := 0; j < len(args); j++ {
|
|
msg += fmt.Sprintf("$%d\r\n%s\r\n", len(args[j]), args[j])
|
|
}
|
|
if len(lmsg) < 1500 {
|
|
lmsg += msg
|
|
continue
|
|
}
|
|
msg = lmsg + msg
|
|
if len(msg) > 0 {
|
|
lmsg = msg[len(msg)/2:]
|
|
msg = msg[:len(msg)/2]
|
|
}
|
|
bufs = append(bufs, msg)
|
|
}
|
|
bufs = append(bufs, lmsg)
|
|
bufs = append(bufs, "RESET THE INDEX\r\n")
|
|
|
|
// Pipe the buffers in a background routine
|
|
rd, wr := io.Pipe()
|
|
go func() {
|
|
defer wr.Close()
|
|
for _, msg := range bufs {
|
|
io.WriteString(wr, msg)
|
|
}
|
|
}()
|
|
defer rd.Close()
|
|
cnt := 0
|
|
idx := 0
|
|
start := time.Now()
|
|
r := newReader(rd, make([]byte, 256))
|
|
for {
|
|
cmds, err := r.ReadCommands()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
log.Fatal(err)
|
|
}
|
|
for _, cmd := range cmds {
|
|
if len(cmd) == 3 && string(cmd[0]) == "RESET" && string(cmd[1]) == "THE" && string(cmd[2]) == "INDEX" {
|
|
if idx != len(gcmds) {
|
|
t.Fatalf("did not process all commands")
|
|
}
|
|
idx = 0
|
|
break
|
|
}
|
|
if len(cmd) != len(gcmds[idx]) {
|
|
t.Fatalf("len not equal for index %d -- %d != %d", idx, len(cmd), len(gcmds[idx]))
|
|
}
|
|
for i := 0; i < len(cmd); i++ {
|
|
if i == 0 {
|
|
if len(cmd[i]) == len(gcmds[idx][i]) {
|
|
ok := true
|
|
for j := 0; j < len(cmd[i]); j++ {
|
|
c1, c2 := cmd[i][j], gcmds[idx][i][j]
|
|
if c1 >= 'A' && c1 <= 'Z' {
|
|
c1 += 32
|
|
}
|
|
if c2 >= 'A' && c2 <= 'Z' {
|
|
c2 += 32
|
|
}
|
|
if c1 != c2 {
|
|
ok = false
|
|
break
|
|
}
|
|
}
|
|
if ok {
|
|
continue
|
|
}
|
|
}
|
|
} else if string(cmd[i]) == string(gcmds[idx][i]) {
|
|
continue
|
|
}
|
|
t.Fatalf("not equal for index %d/%d", idx, i)
|
|
}
|
|
idx++
|
|
cnt++
|
|
}
|
|
}
|
|
if false {
|
|
dur := time.Now().Sub(start)
|
|
fmt.Printf("%d commands in %s - %.0f ops/sec\n", cnt, dur, float64(cnt)/(float64(dur)/float64(time.Second)))
|
|
}
|
|
}
|
|
func testHijack(t *testing.T, conn HijackedConn) {
|
|
conn.WriteString("HIJACKED")
|
|
if err := conn.Flush(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestServer(t *testing.T) {
|
|
s := NewServer(":12345",
|
|
func(conn Conn, cmds [][]string) {
|
|
for _, cmd := range cmds {
|
|
switch strings.ToLower(cmd[0]) {
|
|
default:
|
|
conn.WriteError("ERR unknown command '" + cmd[0] + "'")
|
|
case "ping":
|
|
conn.WriteString("PONG")
|
|
case "quit":
|
|
conn.WriteString("OK")
|
|
conn.Close()
|
|
case "hijack":
|
|
go testHijack(t, conn.Hijack())
|
|
case "int":
|
|
conn.WriteInt(100)
|
|
case "bulk":
|
|
conn.WriteBulk("bulk")
|
|
case "bulkbytes":
|
|
conn.WriteBulkBytes([]byte("bulkbytes"))
|
|
case "null":
|
|
conn.WriteNull()
|
|
case "err":
|
|
conn.WriteError("ERR error")
|
|
case "array":
|
|
conn.WriteArray(2)
|
|
conn.WriteInt(99)
|
|
conn.WriteString("Hi!")
|
|
}
|
|
}
|
|
},
|
|
func(conn Conn) bool {
|
|
//log.Printf("accept: %s", conn.RemoteAddr())
|
|
return true
|
|
},
|
|
func(conn Conn, err error) {
|
|
//log.Printf("closed: %s [%v]", conn.RemoteAddr(), err)
|
|
},
|
|
)
|
|
if err := s.Close(); err == nil {
|
|
t.Fatalf("expected an error, should not be able to close before serving")
|
|
}
|
|
go func() {
|
|
time.Sleep(time.Second / 4)
|
|
if err := ListenAndServe(":12345", nil, nil, nil); err == nil {
|
|
t.Fatalf("expected an error, should not be able to listen on the same port")
|
|
}
|
|
time.Sleep(time.Second / 4)
|
|
|
|
err := s.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = s.Close()
|
|
if err == nil {
|
|
t.Fatalf("expected an error")
|
|
}
|
|
}()
|
|
done := make(chan bool)
|
|
signal := make(chan error)
|
|
go func() {
|
|
defer func() {
|
|
done <- true
|
|
}()
|
|
err := <-signal
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c, err := net.Dial("tcp", ":12345")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer c.Close()
|
|
do := func(cmd string) (string, error) {
|
|
io.WriteString(c, cmd)
|
|
buf := make([]byte, 1024)
|
|
n, err := c.Read(buf)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(buf[:n]), nil
|
|
}
|
|
res, err := do("PING\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "+PONG\r\n" {
|
|
t.Fatal("expecting '+PONG\r\n', got '%v'", res)
|
|
}
|
|
res, err = do("BULK\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "$4\r\nbulk\r\n" {
|
|
t.Fatal("expecting bulk, got '%v'", res)
|
|
}
|
|
res, err = do("BULKBYTES\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "$9\r\nbulkbytes\r\n" {
|
|
t.Fatal("expecting bulkbytes, got '%v'", res)
|
|
}
|
|
res, err = do("INT\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != ":100\r\n" {
|
|
t.Fatal("expecting int, got '%v'", res)
|
|
}
|
|
res, err = do("NULL\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "$-1\r\n" {
|
|
t.Fatal("expecting nul, got '%v'", res)
|
|
}
|
|
res, err = do("ARRAY\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "*2\r\n:99\r\n+Hi!\r\n" {
|
|
t.Fatal("expecting array, got '%v'", res)
|
|
}
|
|
res, err = do("ERR\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "-ERR error\r\n" {
|
|
t.Fatal("expecting array, got '%v'", res)
|
|
}
|
|
res, err = do("HIJACK\r\n")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res != "+HIJACKED\r\n" {
|
|
t.Fatal("expecting string, got '%v'", res)
|
|
}
|
|
}()
|
|
go func() {
|
|
err := s.ListenServeAndSignal(signal)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
<-done
|
|
}
|