[PATCH 6.1 162/185] btrfs: zoned: factor out prepare_allocation_zoned()