virtio-balloon: Store the balloon size to support reboot

This commit store balloon size to MemoryConfig.
After reboot, virtio-balloon can use this size to inflate back to
the size before reboot.

Signed-off-by: Hui Zhu <teawater@antfin.com>
This commit is contained in:
Hui Zhu
2020-06-23 17:52:30 +08:00
committed by Rob Bradford
parent 50da100afd
commit 800220acbb
6 changed files with 24 additions and 12 deletions

View File

@@ -410,10 +410,11 @@ pub struct Balloon {
impl Balloon {
// Create a new virtio-balloon.
pub fn new(id: String) -> io::Result<Self> {
pub fn new(id: String, size: u64) -> io::Result<Self> {
let avail_features = 1u64 << VIRTIO_F_VERSION_1;
let config = VirtioBalloonConfig::default();
let mut config = VirtioBalloonConfig::default();
config.num_pages = (size >> PAGE_SHIFT) as u32;
Ok(Balloon {
id,