circuitpython/tests/extmod/vfs_lfs_corrupt.py

113 lines
2.5 KiB
Python

# Test for VfsLittle using a RAM device, testing error handling from corrupt block device
try:
import uos
uos.VfsLfs1
uos.VfsLfs2
except (ImportError, AttributeError):
print("SKIP")
raise SystemExit
class RAMBlockDevice:
ERASE_BLOCK_SIZE = 1024
def __init__(self, blocks):
self.data = bytearray(blocks * self.ERASE_BLOCK_SIZE)
self.ret = 0
def readblocks(self, block, buf, off):
addr = block * self.ERASE_BLOCK_SIZE + off
for i in range(len(buf)):
buf[i] = self.data[addr + i]
return self.ret
def writeblocks(self, block, buf, off):
addr = block * self.ERASE_BLOCK_SIZE + off
for i in range(len(buf)):
self.data[addr + i] = buf[i]
return self.ret
def ioctl(self, op, arg):
if op == 4: # block count
return len(self.data) // self.ERASE_BLOCK_SIZE
if op == 5: # block size
return self.ERASE_BLOCK_SIZE
if op == 6: # erase block
return 0
def corrupt(bdev, block):
addr = block * bdev.ERASE_BLOCK_SIZE
for i in range(bdev.ERASE_BLOCK_SIZE):
bdev.data[addr + i] = i & 0xFF
def create_vfs(bdev, vfs_class):
bdev.ret = 0
vfs_class.mkfs(bdev)
vfs = vfs_class(bdev)
with vfs.open("f", "w") as f:
for i in range(100):
f.write("test")
return vfs
def test(bdev, vfs_class):
print("test", vfs_class)
# statvfs
vfs = create_vfs(bdev, vfs_class)
corrupt(bdev, 0)
corrupt(bdev, 1)
try:
print(vfs.statvfs(""))
except OSError:
print("statvfs OSError")
# error during read
vfs = create_vfs(bdev, vfs_class)
f = vfs.open("f", "r")
bdev.ret = -5 # EIO
try:
f.read(10)
except OSError:
print("read OSError")
# error during write
vfs = create_vfs(bdev, vfs_class)
f = vfs.open("f", "a")
bdev.ret = -5 # EIO
try:
f.write("test")
except OSError:
print("write OSError")
# error during close
vfs = create_vfs(bdev, vfs_class)
f = vfs.open("f", "w")
f.write("test")
bdev.ret = -5 # EIO
try:
f.close()
except OSError:
print("close OSError")
# error during flush
vfs = create_vfs(bdev, vfs_class)
f = vfs.open("f", "w")
f.write("test")
bdev.ret = -5 # EIO
try:
f.flush()
except OSError:
print("flush OSError")
bdev.ret = 0
f.close()
bdev = RAMBlockDevice(30)
test(bdev, uos.VfsLfs1)
test(bdev, uos.VfsLfs2)