You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.3 KiB
66 lines
1.3 KiB
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
rs "github.com/tendermint/tendermint/rpc/jsonrpc/server"
|
|
"github.com/tendermint/tendermint/rpc/jsonrpc/types"
|
|
)
|
|
|
|
var rpcFuncMap = map[string]*rs.RPCFunc{
|
|
"c": rs.NewRPCFunc(func(ctx context.Context, s string, i int) (string, error) {
|
|
return "foo", nil
|
|
}, "s", "i"),
|
|
}
|
|
var mux *http.ServeMux
|
|
|
|
func init() {
|
|
mux = http.NewServeMux()
|
|
lgr := log.MustNewDefaultLogger(log.LogFormatPlain, log.LogLevelInfo)
|
|
rs.RegisterRPCFuncs(mux, rpcFuncMap, lgr)
|
|
}
|
|
|
|
func Fuzz(data []byte) int {
|
|
if len(data) == 0 {
|
|
return -1
|
|
}
|
|
|
|
req, _ := http.NewRequest("POST", "http://localhost/", bytes.NewReader(data))
|
|
rec := httptest.NewRecorder()
|
|
mux.ServeHTTP(rec, req)
|
|
res := rec.Result()
|
|
blob, err := io.ReadAll(res.Body)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if err := res.Body.Close(); err != nil {
|
|
panic(err)
|
|
}
|
|
if len(blob) == 0 {
|
|
return 1
|
|
}
|
|
|
|
if outputJSONIsSlice(blob) {
|
|
recv := []types.RPCResponse{}
|
|
if err := json.Unmarshal(blob, &recv); err != nil {
|
|
panic(err)
|
|
}
|
|
return 1
|
|
}
|
|
recv := &types.RPCResponse{}
|
|
if err := json.Unmarshal(blob, recv); err != nil {
|
|
panic(err)
|
|
}
|
|
return 1
|
|
}
|
|
|
|
func outputJSONIsSlice(input []byte) bool {
|
|
slice := []interface{}{}
|
|
return json.Unmarshal(input, &slice) == nil
|
|
}
|