Skip to content

Commit

Permalink
Merge pull request redis#1570 from AdamKorcz/fuzz1
Browse files Browse the repository at this point in the history
Add fuzzer to integrate go-redis into OSS-fuzz
  • Loading branch information
vmihailenco authored Feb 4, 2021
2 parents f8cbf48 + 0d9368c commit 0abfa60
Showing 1 changed file with 46 additions and 0 deletions.
46 changes: 46 additions & 0 deletions fuzz.go
Original file line number Diff line number Diff line change
@@ -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
}

0 comments on commit 0abfa60

Please sign in to comment.