[memdisk] Add drive shift limit logic