Rooba
/
oss-fuzz
mirror of https://github.com/google/oss-fuzz.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Kafka]: Fix issue 51986 51996 (#8641) 

Fix two issues for kafka fuzzing
51986 (https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=51986)
51996 (https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=51996)
This commit is contained in:
Arthur Chan 2022-09-30 15:16:57 +01:00 committed by GitHub
parent 43c4d9a53a
commit 296b7b1b29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
projects/kafka/fuzz_consumer.py
View File

@ -26,7 +26,7 @@ def TestInput(data):
try:
c = Consumer({
'group.id': fdp.ConsumeString(10),
'group.id': fdp.ConsumeConsumeUnicodeNoSurrogates(10),
'socket.timeout.ms': fdp.ConsumeIntInRange(10,2000),
'session.timeout.ms': fdp.ConsumeIntInRange(10,2000),
'on_commit': dummy_callback})
@ -36,7 +36,7 @@ def TestInput(data):
try:
c.subscribe(
[fdp.ConsumeString(10)],
[fdp.ConsumeUnicodeNoSurrogates(10)],
on_assign=dummy_callback,
on_revoke=dummy_callback
)
@ -50,7 +50,7 @@ def TestInput(data):
partitions = list(
map(
lambda part: TopicPartition(fdp.ConsumeString(10).replace('\x00', ''), part),
lambda part: TopicPartition(fdp.ConsumeUnicodeNoSurrogates(10).replace('\x00', ''), part),
range(0, 100, 3)
)
)

8
projects/kafka/fuzz_producer.py
View File

@ -26,9 +26,9 @@ def TestInput(data):
'message.timeout.ms': fdp.ConsumeIntInRange(10,2000)
})
p.produce(fdp.ConsumeString(20).replace('\x00', ''))
p.produce(fdp.ConsumeUnicodeNoSurrogates(20).replace('\x00', ''))
p.produce(
fdp.ConsumeString(20).replace('\x00', ''),
fdp.ConsumeUnicodeNoSurrogates(20).replace('\x00', ''),
value=fdp.ConsumeString(20),
key=fdp.ConsumeString(20)
)
@ -37,8 +37,8 @@ def TestInput(data):
pass
p.produce(
topic=fdp.ConsumeString(20),
value=fdp.ConsumeString(20),
topic=fdp.ConsumeUnicodeNoSurrogates(20),
value=fdp.ConsumeUnicodeNoSurrogates(20),
partition=fdp.ConsumeIntInRange(1,10),
callback=on_delivery
)