btrfs-progs: Introduce get_unit_mode_from_arg for common use