mirror of
https://github.com/9001/copyparty.git
synced 2025-08-17 09:02:15 -06:00
improve fuse-fuzzer
This commit is contained in:
parent
b6a3c52d67
commit
765af31b83
|
@ -1,6 +1,7 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
import time
|
||||||
|
|
||||||
"""
|
"""
|
||||||
mkdir -p /dev/shm/fusefuzz/{r,v}
|
mkdir -p /dev/shm/fusefuzz/{r,v}
|
||||||
|
@ -10,15 +11,36 @@ PYTHONPATH=.. python3 -m copyparty -v /dev/shm/fusefuzz/r::r -i 127.0.0.1
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def chk(fsz, rsz, ofs0, shift, ofs, rf, vf):
|
||||||
|
if ofs != rf.tell():
|
||||||
|
rf.seek(ofs)
|
||||||
|
vf.seek(ofs)
|
||||||
|
|
||||||
|
rb = rf.read(rsz)
|
||||||
|
vb = vf.read(rsz)
|
||||||
|
|
||||||
|
print(f"fsz {fsz} rsz {rsz} ofs {ofs0} shift {shift} ofs {ofs} = {len(rb)}")
|
||||||
|
|
||||||
|
if rb != vb:
|
||||||
|
for n, buf in enumerate([rb, vb]):
|
||||||
|
with open("buf." + str(n), "wb") as f:
|
||||||
|
f.write(buf)
|
||||||
|
|
||||||
|
raise Exception(f"{len(rb)} != {len(vb)}")
|
||||||
|
|
||||||
|
return rb, vb
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
v = "v"
|
v = "v"
|
||||||
for n in range(5):
|
for n in range(5):
|
||||||
with open(f"r/{n}", "wb") as f:
|
with open(f"r/{n}", "wb") as f:
|
||||||
f.write(b"h" * n)
|
f.write(b"h" * n)
|
||||||
|
|
||||||
|
rand = os.urandom(7919) # prime
|
||||||
for fsz in range(1024 * 1024 * 2 - 3, 1024 * 1024 * 2 + 3):
|
for fsz in range(1024 * 1024 * 2 - 3, 1024 * 1024 * 2 + 3):
|
||||||
with open("r/f", "wb", fsz) as f:
|
with open("r/f", "wb", fsz) as f:
|
||||||
f.write(b"\xab" * fsz)
|
f.write((rand * int(fsz / len(rand) + 1))[:fsz])
|
||||||
|
|
||||||
for rsz in range(64 * 1024 - 2, 64 * 1024 + 2):
|
for rsz in range(64 * 1024 - 2, 64 * 1024 + 2):
|
||||||
ofslist = [0, 1, 2]
|
ofslist = [0, 1, 2]
|
||||||
|
@ -48,19 +70,7 @@ def main():
|
||||||
|
|
||||||
prev_ofs = ofs
|
prev_ofs = ofs
|
||||||
|
|
||||||
if ofs != rf.tell():
|
rb, vb = chk(fsz, rsz, ofs0, shift, ofs, rf, vf)
|
||||||
rf.seek(ofs)
|
|
||||||
vf.seek(ofs)
|
|
||||||
|
|
||||||
rb = rf.read(rsz)
|
|
||||||
vb = vf.read(rsz)
|
|
||||||
|
|
||||||
print(
|
|
||||||
f"fsz {fsz} rsz {rsz} ofs {ofs0} shift {shift} ofs {ofs} = {len(rb)}"
|
|
||||||
)
|
|
||||||
|
|
||||||
if rb != vb:
|
|
||||||
raise Exception(f"{len(rb)} != {len(vb)}")
|
|
||||||
|
|
||||||
if not rb:
|
if not rb:
|
||||||
break
|
break
|
||||||
|
@ -78,28 +88,13 @@ def main():
|
||||||
if ofs < 0 or ofs > fsz:
|
if ofs < 0 or ofs > fsz:
|
||||||
break
|
break
|
||||||
|
|
||||||
if ofs != rf.tell():
|
rb, vb = chk(fsz, rsz, ofs0, shift, ofs, rf, vf)
|
||||||
rf.seek(ofs)
|
|
||||||
vf.seek(ofs)
|
|
||||||
|
|
||||||
rb = rf.read(rsz)
|
|
||||||
vb = vf.read(rsz)
|
|
||||||
|
|
||||||
print(
|
|
||||||
f"fsz {fsz} rsz {rsz} ofs {ofs0} shift {shift} ofs {ofs} = {len(rb)}"
|
|
||||||
)
|
|
||||||
|
|
||||||
if rb != vb:
|
|
||||||
raise Exception(f"{len(rb)} != {len(vb)}")
|
|
||||||
|
|
||||||
ofs -= rsz
|
ofs -= rsz
|
||||||
|
|
||||||
|
# bumping fsz, sleep away the dentry cache in cppf
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
|
||||||
"""
|
|
||||||
f() { cat virt/{1,2,AAAAAAAAAAAAA}; echo; dd if=virt/f bs=${1} | cmp r/f; }
|
|
||||||
901120
|
|
||||||
"""
|
|
||||||
|
|
Loading…
Reference in a new issue