From: Paolo Bonzini <pbonzini@redhat.com>
Date: Fri, 19 Sep 2025 09:18:44 +0000 (+0200)
Subject: rust: migration: implement ToMigrationState as part of impl_vmstate_bitsized
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=7ee5875d423598ac55a0b55881d9a1ee5c3c7daf;p=thirdparty%2Fqemu.git

rust: migration: implement ToMigrationState as part of impl_vmstate_bitsized

This is most likely desirable, and is the easiest way to migrate
a bit-sized value without peeking at the innards of the bilge crate.

Reviewed-by: Zhao Liu <zhao1.liu@intel.com>
Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
---

diff --git a/rust/migration/src/vmstate.rs b/rust/migration/src/vmstate.rs
index 445fe7fbc0..42e5df8d81 100644
--- a/rust/migration/src/vmstate.rs
+++ b/rust/migration/src/vmstate.rs
@@ -296,6 +296,25 @@ macro_rules! impl_vmstate_bitsized {
                                           as ::bilge::prelude::Number>::UnderlyingType
                                          as $crate::vmstate::VMState>::VARRAY_FLAG;
         }
+
+        impl $crate::migratable::ToMigrationState for $type {
+            type Migrated = <<$type as ::bilge::prelude::Bitsized>::ArbitraryInt
+                                          as ::bilge::prelude::Number>::UnderlyingType;
+
+            fn snapshot_migration_state(&self, target: &mut Self::Migrated) -> Result<(), $crate::InvalidError> {
+                *target = Self::Migrated::from(*self);
+                Ok(())
+            }
+
+            fn restore_migrated_state_mut(
+                &mut self,
+                source: Self::Migrated,
+                version_id: u8,
+            ) -> Result<(), $crate::InvalidError> {
+                *self = Self::from(source);
+                Ok(())
+            }
+        }
     };
 }