virtio: balloon: separate out common code between remove and freeze functions