micropython/tests/extmod/vfs_lfs_corrupt.py

113 wiersze
2.5 KiB
Python

# Test for VfsLittle using a RAM device, testing error handling from corrupt block device
try:
import vfs
vfs.VfsLfs1
vfs.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)
fs = vfs_class(bdev)
with fs.open("f", "w") as f:
for i in range(100):
f.write("test")
return fs
def test(bdev, vfs_class):
print("test", vfs_class)
# statvfs
fs = create_vfs(bdev, vfs_class)
corrupt(bdev, 0)
corrupt(bdev, 1)
try:
print(fs.statvfs(""))
except OSError:
print("statvfs OSError")
# error during read
fs = create_vfs(bdev, vfs_class)
f = fs.open("f", "r")
bdev.ret = -5 # EIO
try:
f.read(10)
except OSError:
print("read OSError")
# error during write
fs = create_vfs(bdev, vfs_class)
f = fs.open("f", "a")
bdev.ret = -5 # EIO
try:
f.write("test")
except OSError:
print("write OSError")
# error during close
fs = create_vfs(bdev, vfs_class)
f = fs.open("f", "w")
f.write("test")
bdev.ret = -5 # EIO
try:
f.close()
except OSError:
print("close OSError")
# error during flush
fs = create_vfs(bdev, vfs_class)
f = fs.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, vfs.VfsLfs1)
test(bdev, vfs.VfsLfs2)