hw/block/nvme: refactor the logic for zone write checks