btrfs: open code set_balance_control