mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-10 14:26:26 -05:00
f0236233f3
Fix JSON parsing state machine so Read([]byte) can accept byte array of any length. This fixes a infinite read with io.ReadFull().
99 lines
1.9 KiB
Go
99 lines
1.9 KiB
Go
package json_test
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
|
. "github.com/v2fly/v2ray-core/v5/infra/conf/json"
|
|
)
|
|
|
|
func TestReader(t *testing.T) {
|
|
data := []struct {
|
|
input string
|
|
output string
|
|
}{
|
|
{
|
|
`
|
|
content #comment 1
|
|
#comment 2
|
|
content 2`,
|
|
`
|
|
content
|
|
|
|
content 2`,
|
|
},
|
|
{`content`, `content`},
|
|
{" ", " "},
|
|
{`con/*abcd*/tent`, "content"},
|
|
{`
|
|
text // adlkhdf /*
|
|
//comment adfkj
|
|
text 2*/`, `
|
|
text
|
|
|
|
text 2*`},
|
|
{`"//"content`, `"//"content`},
|
|
{`abcd'//'abcd`, `abcd'//'abcd`},
|
|
{`"\""`, `"\""`},
|
|
{`\"/*abcd*/\"`, `\"\"`},
|
|
}
|
|
|
|
for _, testCase := range data {
|
|
reader := &Reader{
|
|
Reader: bytes.NewReader([]byte(testCase.input)),
|
|
}
|
|
|
|
actual := make([]byte, 1024)
|
|
n, err := reader.Read(actual)
|
|
common.Must(err)
|
|
if r := cmp.Diff(string(actual[:n]), testCase.output); r != "" {
|
|
t.Error(r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestReader1(t *testing.T) {
|
|
type dataStruct struct {
|
|
input string
|
|
output string
|
|
}
|
|
|
|
bufLen := 1
|
|
|
|
data := []dataStruct{
|
|
{"loooooooooooooooooooooooooooooooooooooooog", "loooooooooooooooooooooooooooooooooooooooog"},
|
|
{`{"t": "\/testlooooooooooooooooooooooooooooong"}`, `{"t": "\/testlooooooooooooooooooooooooooooong"}`},
|
|
{`{"t": "\/test"}`, `{"t": "\/test"}`},
|
|
{`"\// fake comment"`, `"\// fake comment"`},
|
|
{`"\/\/\/\/\/"`, `"\/\/\/\/\/"`},
|
|
{`/test/test`, `/test/test`},
|
|
}
|
|
|
|
for _, testCase := range data {
|
|
reader := &Reader{
|
|
Reader: bytes.NewReader([]byte(testCase.input)),
|
|
}
|
|
target := make([]byte, 0)
|
|
buf := make([]byte, bufLen)
|
|
var n int
|
|
var err error
|
|
for n, err = reader.Read(buf); err == nil; n, err = reader.Read(buf) {
|
|
if n > len(buf) {
|
|
t.Error("n: ", n)
|
|
}
|
|
target = append(target, buf[:n]...)
|
|
buf = make([]byte, bufLen)
|
|
}
|
|
if err != nil && err != io.EOF {
|
|
t.Error("error: ", err)
|
|
}
|
|
if string(target) != testCase.output {
|
|
t.Error("got ", string(target), " want ", testCase.output)
|
|
}
|
|
}
|
|
}
|