Merge pull request #1570 from AdamKorcz/fuzz1

Add fuzzer to integrate go-redis into OSS-fuzz
This commit is contained in:
Vladimir Mihailenco 2021-02-04 17:01:12 +02:00 committed by GitHub
commit 0abfa60737
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 0 deletions

46
fuzz.go Normal file
View File

@ -0,0 +1,46 @@
// +build gofuzz
package redis
import (
"context"
"time"
)
var (
ctx = context.Background()
rdb *Client
)
func init() {
rdb = NewClient(&Options{
Addr: ":6379",
DialTimeout: 10 * time.Second,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
PoolSize: 10,
PoolTimeout: 10 * time.Second,
})
}
func Fuzz(data []byte) int {
arrayLen := len(data)
if arrayLen < 4 {
return -1
}
maxIter := int(uint(data[0]))
for i := 0; i < maxIter && i < arrayLen; i++ {
n := i % arrayLen
if n == 0 {
_ = rdb.Set(ctx, string(data[i:]), string(data[i:]), 0).Err()
} else if n == 1 {
_, _ = rdb.Get(ctx, string(data[i:])).Result()
} else if n == 2 {
_, _ = rdb.Incr(ctx, string(data[i:])).Result()
} else if n == 3 {
var cursor uint64
_, _, _ = rdb.Scan(ctx, cursor, string(data[i:]), 10).Result()
}
}
return 1
}