10a65e412SManos Pitsidianakis // Copyright 2024, Linaro Limited 20a65e412SManos Pitsidianakis // Author(s): Manos Pitsidianakis <manos.pitsidianakis@linaro.org> 30a65e412SManos Pitsidianakis // SPDX-License-Identifier: GPL-2.0-or-later 40a65e412SManos Pitsidianakis 50a65e412SManos Pitsidianakis //! Helper macros to declare migration state for device models. 60a65e412SManos Pitsidianakis //! 700f89716SPaolo Bonzini //! This module includes four families of macros: 80d43ddaeSPaolo Bonzini //! 90d43ddaeSPaolo Bonzini //! * [`vmstate_unused!`](crate::vmstate_unused) and 100d43ddaeSPaolo Bonzini //! [`vmstate_of!`](crate::vmstate_of), which are used to express the 110d43ddaeSPaolo Bonzini //! migration format for a struct. This is based on the [`VMState`] trait, 120d43ddaeSPaolo Bonzini //! which is defined by all migrateable types. 130d43ddaeSPaolo Bonzini //! 1400f89716SPaolo Bonzini //! * [`impl_vmstate_forward`](crate::impl_vmstate_forward) and 1500f89716SPaolo Bonzini //! [`impl_vmstate_bitsized`](crate::impl_vmstate_bitsized), which help with 1600f89716SPaolo Bonzini //! the definition of the [`VMState`] trait (respectively for transparent 1700f89716SPaolo Bonzini //! structs and for `bilge`-defined types) 1800f89716SPaolo Bonzini //! 190d43ddaeSPaolo Bonzini //! * helper macros to declare a device model state struct, in particular 200a65e412SManos Pitsidianakis //! [`vmstate_subsections`](crate::vmstate_subsections) and 210d43ddaeSPaolo Bonzini //! [`vmstate_fields`](crate::vmstate_fields). 220d43ddaeSPaolo Bonzini //! 230d43ddaeSPaolo Bonzini //! * direct equivalents to the C macros declared in 249d489949SPaolo Bonzini //! `include/migration/vmstate.h`. These are not type-safe and only provide 259d489949SPaolo Bonzini //! functionality that is missing from `vmstate_of!`. 260a65e412SManos Pitsidianakis 2780aa3045SPaolo Bonzini use core::{marker::PhantomData, mem, ptr::NonNull}; 28*b1310037SZhao Liu use std::os::raw::{c_int, c_void}; 290d43ddaeSPaolo Bonzini 300d43ddaeSPaolo Bonzini pub use crate::bindings::{VMStateDescription, VMStateField}; 31*b1310037SZhao Liu use crate::{ 32*b1310037SZhao Liu bindings::VMStateFlags, callbacks::FnCall, prelude::*, qom::Owned, zeroable::Zeroable, 33*b1310037SZhao Liu }; 340d43ddaeSPaolo Bonzini 350d43ddaeSPaolo Bonzini /// This macro is used to call a function with a generic argument bound 360d43ddaeSPaolo Bonzini /// to the type of a field. The function must take a 370d43ddaeSPaolo Bonzini /// [`PhantomData`]`<T>` argument; `T` is the type of 380d43ddaeSPaolo Bonzini /// field `$field` in the `$typ` type. 390d43ddaeSPaolo Bonzini /// 400d43ddaeSPaolo Bonzini /// # Examples 410d43ddaeSPaolo Bonzini /// 420d43ddaeSPaolo Bonzini /// ``` 430d43ddaeSPaolo Bonzini /// # use qemu_api::call_func_with_field; 440d43ddaeSPaolo Bonzini /// # use core::marker::PhantomData; 450d43ddaeSPaolo Bonzini /// const fn size_of_field<T>(_: PhantomData<T>) -> usize { 460d43ddaeSPaolo Bonzini /// std::mem::size_of::<T>() 470d43ddaeSPaolo Bonzini /// } 480d43ddaeSPaolo Bonzini /// 490d43ddaeSPaolo Bonzini /// struct Foo { 500d43ddaeSPaolo Bonzini /// x: u16, 510d43ddaeSPaolo Bonzini /// }; 520d43ddaeSPaolo Bonzini /// // calls size_of_field::<u16>() 530d43ddaeSPaolo Bonzini /// assert_eq!(call_func_with_field!(size_of_field, Foo, x), 2); 540d43ddaeSPaolo Bonzini /// ``` 550d43ddaeSPaolo Bonzini #[macro_export] 560d43ddaeSPaolo Bonzini macro_rules! call_func_with_field { 570d43ddaeSPaolo Bonzini // Based on the answer by user steffahn (Frank Steffahn) at 580d43ddaeSPaolo Bonzini // https://users.rust-lang.org/t/inferring-type-of-field/122857 590d43ddaeSPaolo Bonzini // and used under MIT license 600d43ddaeSPaolo Bonzini ($func:expr, $typ:ty, $($field:tt).+) => { 610d43ddaeSPaolo Bonzini $func(loop { 620d43ddaeSPaolo Bonzini #![allow(unreachable_code)] 630d43ddaeSPaolo Bonzini const fn phantom__<T>(_: &T) -> ::core::marker::PhantomData<T> { ::core::marker::PhantomData } 640d43ddaeSPaolo Bonzini // Unreachable code is exempt from checks on uninitialized values. 650d43ddaeSPaolo Bonzini // Use that trick to infer the type of this PhantomData. 660d43ddaeSPaolo Bonzini break ::core::marker::PhantomData; 670d43ddaeSPaolo Bonzini break phantom__(&{ let value__: $typ; value__.$($field).+ }); 680d43ddaeSPaolo Bonzini }) 690d43ddaeSPaolo Bonzini }; 700d43ddaeSPaolo Bonzini } 710d43ddaeSPaolo Bonzini 72f2cb78bdSPaolo Bonzini /// Workaround for lack of `const_refs_static`: references to global variables 73f2cb78bdSPaolo Bonzini /// can be included in a `static`, but not in a `const`; unfortunately, this 74f2cb78bdSPaolo Bonzini /// is exactly what would go in the `VMStateField`'s `info` member. 75f2cb78bdSPaolo Bonzini /// 76f2cb78bdSPaolo Bonzini /// This enum contains the contents of the `VMStateField`'s `info` member, 77f2cb78bdSPaolo Bonzini /// but as an `enum` instead of a pointer. 78f2cb78bdSPaolo Bonzini #[allow(non_camel_case_types)] 79f2cb78bdSPaolo Bonzini pub enum VMStateFieldType { 80f2cb78bdSPaolo Bonzini null, 81f2cb78bdSPaolo Bonzini vmstate_info_bool, 82f2cb78bdSPaolo Bonzini vmstate_info_int8, 83f2cb78bdSPaolo Bonzini vmstate_info_int16, 84f2cb78bdSPaolo Bonzini vmstate_info_int32, 85f2cb78bdSPaolo Bonzini vmstate_info_int64, 86f2cb78bdSPaolo Bonzini vmstate_info_uint8, 87f2cb78bdSPaolo Bonzini vmstate_info_uint16, 88f2cb78bdSPaolo Bonzini vmstate_info_uint32, 89f2cb78bdSPaolo Bonzini vmstate_info_uint64, 90f2cb78bdSPaolo Bonzini vmstate_info_timer, 91f2cb78bdSPaolo Bonzini } 92f2cb78bdSPaolo Bonzini 93f2cb78bdSPaolo Bonzini /// Workaround for lack of `const_refs_static`. Converts a `VMStateFieldType` 94f2cb78bdSPaolo Bonzini /// to a `*const VMStateInfo`, for inclusion in a `VMStateField`. 95f2cb78bdSPaolo Bonzini #[macro_export] 96f2cb78bdSPaolo Bonzini macro_rules! info_enum_to_ref { 97f2cb78bdSPaolo Bonzini ($e:expr) => { 98f2cb78bdSPaolo Bonzini unsafe { 99f2cb78bdSPaolo Bonzini match $e { 100f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::null => ::core::ptr::null(), 101f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_bool => { 102f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_bool) 103f2cb78bdSPaolo Bonzini } 104f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_int8 => { 105f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_int8) 106f2cb78bdSPaolo Bonzini } 107f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_int16 => { 108f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_int16) 109f2cb78bdSPaolo Bonzini } 110f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_int32 => { 111f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_int32) 112f2cb78bdSPaolo Bonzini } 113f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_int64 => { 114f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_int64) 115f2cb78bdSPaolo Bonzini } 116f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_uint8 => { 117f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_uint8) 118f2cb78bdSPaolo Bonzini } 119f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_uint16 => { 120f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_uint16) 121f2cb78bdSPaolo Bonzini } 122f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_uint32 => { 123f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_uint32) 124f2cb78bdSPaolo Bonzini } 125f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_uint64 => { 126f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_uint64) 127f2cb78bdSPaolo Bonzini } 128f2cb78bdSPaolo Bonzini $crate::vmstate::VMStateFieldType::vmstate_info_timer => { 129f2cb78bdSPaolo Bonzini ::core::ptr::addr_of!($crate::bindings::vmstate_info_timer) 130f2cb78bdSPaolo Bonzini } 131f2cb78bdSPaolo Bonzini } 132f2cb78bdSPaolo Bonzini } 133f2cb78bdSPaolo Bonzini }; 134f2cb78bdSPaolo Bonzini } 135f2cb78bdSPaolo Bonzini 1360d43ddaeSPaolo Bonzini /// A trait for types that can be included in a device's migration stream. It 1370d43ddaeSPaolo Bonzini /// provides the base contents of a `VMStateField` (minus the name and offset). 1380d43ddaeSPaolo Bonzini /// 1390d43ddaeSPaolo Bonzini /// # Safety 1400d43ddaeSPaolo Bonzini /// 1410d43ddaeSPaolo Bonzini /// The contents of this trait go straight into structs that are parsed by C 14200f89716SPaolo Bonzini /// code and used to introspect into other structs. Generally, you don't need 14300f89716SPaolo Bonzini /// to implement it except via macros that do it for you, such as 14400f89716SPaolo Bonzini /// `impl_vmstate_bitsized!`. 1450d43ddaeSPaolo Bonzini pub unsafe trait VMState { 146f2cb78bdSPaolo Bonzini /// The `info` member of a `VMStateField` is a pointer and as such cannot 147f2cb78bdSPaolo Bonzini /// yet be included in the [`BASE`](VMState::BASE) associated constant; 148f2cb78bdSPaolo Bonzini /// this is only allowed by Rust 1.83.0 and newer. For now, include the 149f2cb78bdSPaolo Bonzini /// member as an enum which is stored in a separate constant. 150f2cb78bdSPaolo Bonzini const SCALAR_TYPE: VMStateFieldType = VMStateFieldType::null; 151f2cb78bdSPaolo Bonzini 1520d43ddaeSPaolo Bonzini /// The base contents of a `VMStateField` (minus the name and offset) for 1530d43ddaeSPaolo Bonzini /// the type that is implementing the trait. 1540d43ddaeSPaolo Bonzini const BASE: VMStateField; 1555b024b4eSPaolo Bonzini 1565b024b4eSPaolo Bonzini /// A flag that is added to another field's `VMStateField` to specify the 1575b024b4eSPaolo Bonzini /// length's type in a variable-sized array. If this is not a supported 1585b024b4eSPaolo Bonzini /// type for the length (i.e. if it is not `u8`, `u16`, `u32`), using it 1595b024b4eSPaolo Bonzini /// in a call to [`vmstate_of!`](crate::vmstate_of) will cause a 1605b024b4eSPaolo Bonzini /// compile-time error. 1615b024b4eSPaolo Bonzini const VARRAY_FLAG: VMStateFlags = { 1625b024b4eSPaolo Bonzini panic!("invalid type for variable-sized array"); 1635b024b4eSPaolo Bonzini }; 1640d43ddaeSPaolo Bonzini } 1650d43ddaeSPaolo Bonzini 166f2cb78bdSPaolo Bonzini /// Internal utility function to retrieve a type's `VMStateFieldType`; 167f2cb78bdSPaolo Bonzini /// used by [`vmstate_of!`](crate::vmstate_of). 168f2cb78bdSPaolo Bonzini pub const fn vmstate_scalar_type<T: VMState>(_: PhantomData<T>) -> VMStateFieldType { 169f2cb78bdSPaolo Bonzini T::SCALAR_TYPE 170f2cb78bdSPaolo Bonzini } 171f2cb78bdSPaolo Bonzini 1720d43ddaeSPaolo Bonzini /// Internal utility function to retrieve a type's `VMStateField`; 1730d43ddaeSPaolo Bonzini /// used by [`vmstate_of!`](crate::vmstate_of). 1740d43ddaeSPaolo Bonzini pub const fn vmstate_base<T: VMState>(_: PhantomData<T>) -> VMStateField { 1750d43ddaeSPaolo Bonzini T::BASE 1760d43ddaeSPaolo Bonzini } 1770d43ddaeSPaolo Bonzini 1785b024b4eSPaolo Bonzini /// Internal utility function to retrieve a type's `VMStateFlags` when it 1795b024b4eSPaolo Bonzini /// is used as the element count of a `VMSTATE_VARRAY`; used by 1805b024b4eSPaolo Bonzini /// [`vmstate_of!`](crate::vmstate_of). 1815b024b4eSPaolo Bonzini pub const fn vmstate_varray_flag<T: VMState>(_: PhantomData<T>) -> VMStateFlags { 1825b024b4eSPaolo Bonzini T::VARRAY_FLAG 1835b024b4eSPaolo Bonzini } 1845b024b4eSPaolo Bonzini 1850d43ddaeSPaolo Bonzini /// Return the `VMStateField` for a field of a struct. The field must be 1860d43ddaeSPaolo Bonzini /// visible in the current scope. 1870d43ddaeSPaolo Bonzini /// 188f2cb78bdSPaolo Bonzini /// Only a limited set of types is supported out of the box: 189f2cb78bdSPaolo Bonzini /// * scalar types (integer and `bool`) 190f2cb78bdSPaolo Bonzini /// * the C struct `QEMUTimer` 191f2cb78bdSPaolo Bonzini /// * a transparent wrapper for any of the above (`Cell`, `UnsafeCell`, 192ee7d3aecSPaolo Bonzini /// [`BqlCell`], [`BqlRefCell`] 193f2cb78bdSPaolo Bonzini /// * a raw pointer to any of the above 194ee7d3aecSPaolo Bonzini /// * a `NonNull` pointer, a `Box` or an [`Owned`] for any of the above 195f2cb78bdSPaolo Bonzini /// * an array of any of the above 196f2cb78bdSPaolo Bonzini /// 1970d43ddaeSPaolo Bonzini /// In order to support other types, the trait `VMState` must be implemented 19800f89716SPaolo Bonzini /// for them. The macros 19900f89716SPaolo Bonzini /// [`impl_vmstate_bitsized!`](crate::impl_vmstate_bitsized) 20000f89716SPaolo Bonzini /// and [`impl_vmstate_forward!`](crate::impl_vmstate_forward) help with this. 2010d43ddaeSPaolo Bonzini #[macro_export] 2020d43ddaeSPaolo Bonzini macro_rules! vmstate_of { 2035b024b4eSPaolo Bonzini ($struct_name:ty, $field_name:ident $([0 .. $num:ident $(* $factor:expr)?])? $(,)?) => { 2040d43ddaeSPaolo Bonzini $crate::bindings::VMStateField { 2050d43ddaeSPaolo Bonzini name: ::core::concat!(::core::stringify!($field_name), "\0") 2060d43ddaeSPaolo Bonzini .as_bytes() 2070d43ddaeSPaolo Bonzini .as_ptr() as *const ::std::os::raw::c_char, 2080d43ddaeSPaolo Bonzini offset: $crate::offset_of!($struct_name, $field_name), 2096ca5c3beSZhao Liu $(num_offset: $crate::offset_of!($struct_name, $num),)? 210f2cb78bdSPaolo Bonzini // The calls to `call_func_with_field!` are the magic that 211f2cb78bdSPaolo Bonzini // computes most of the VMStateField from the type of the field. 212f2cb78bdSPaolo Bonzini info: $crate::info_enum_to_ref!($crate::call_func_with_field!( 213f2cb78bdSPaolo Bonzini $crate::vmstate::vmstate_scalar_type, 214f2cb78bdSPaolo Bonzini $struct_name, 215f2cb78bdSPaolo Bonzini $field_name 216f2cb78bdSPaolo Bonzini )), 2170d43ddaeSPaolo Bonzini ..$crate::call_func_with_field!( 2180d43ddaeSPaolo Bonzini $crate::vmstate::vmstate_base, 2190d43ddaeSPaolo Bonzini $struct_name, 2200d43ddaeSPaolo Bonzini $field_name 2215b024b4eSPaolo Bonzini )$(.with_varray_flag($crate::call_func_with_field!( 2225b024b4eSPaolo Bonzini $crate::vmstate::vmstate_varray_flag, 2235b024b4eSPaolo Bonzini $struct_name, 2245b024b4eSPaolo Bonzini $num)) 2255b024b4eSPaolo Bonzini $(.with_varray_multiply($factor))?)? 2260d43ddaeSPaolo Bonzini } 2270d43ddaeSPaolo Bonzini }; 2280d43ddaeSPaolo Bonzini } 2290d43ddaeSPaolo Bonzini 23080aa3045SPaolo Bonzini impl VMStateFlags { 23180aa3045SPaolo Bonzini const VMS_VARRAY_FLAGS: VMStateFlags = VMStateFlags( 23280aa3045SPaolo Bonzini VMStateFlags::VMS_VARRAY_INT32.0 23380aa3045SPaolo Bonzini | VMStateFlags::VMS_VARRAY_UINT8.0 23480aa3045SPaolo Bonzini | VMStateFlags::VMS_VARRAY_UINT16.0 23580aa3045SPaolo Bonzini | VMStateFlags::VMS_VARRAY_UINT32.0, 23680aa3045SPaolo Bonzini ); 23780aa3045SPaolo Bonzini } 23880aa3045SPaolo Bonzini 2390d43ddaeSPaolo Bonzini // Add a couple builder-style methods to VMStateField, allowing 2400d43ddaeSPaolo Bonzini // easy derivation of VMStateField constants from other types. 2410d43ddaeSPaolo Bonzini impl VMStateField { 2420d43ddaeSPaolo Bonzini #[must_use] 2430d43ddaeSPaolo Bonzini pub const fn with_version_id(mut self, version_id: i32) -> Self { 2440d43ddaeSPaolo Bonzini assert!(version_id >= 0); 2450d43ddaeSPaolo Bonzini self.version_id = version_id; 2460d43ddaeSPaolo Bonzini self 2470d43ddaeSPaolo Bonzini } 24880aa3045SPaolo Bonzini 24980aa3045SPaolo Bonzini #[must_use] 25080aa3045SPaolo Bonzini pub const fn with_array_flag(mut self, num: usize) -> Self { 25180aa3045SPaolo Bonzini assert!(num <= 0x7FFF_FFFFusize); 25280aa3045SPaolo Bonzini assert!((self.flags.0 & VMStateFlags::VMS_ARRAY.0) == 0); 25380aa3045SPaolo Bonzini assert!((self.flags.0 & VMStateFlags::VMS_VARRAY_FLAGS.0) == 0); 25480aa3045SPaolo Bonzini if (self.flags.0 & VMStateFlags::VMS_POINTER.0) != 0 { 25580aa3045SPaolo Bonzini self.flags = VMStateFlags(self.flags.0 & !VMStateFlags::VMS_POINTER.0); 25680aa3045SPaolo Bonzini self.flags = VMStateFlags(self.flags.0 | VMStateFlags::VMS_ARRAY_OF_POINTER.0); 25720797069SZhao Liu // VMS_ARRAY_OF_POINTER flag stores the size of pointer. 25820797069SZhao Liu // FIXME: *const, *mut, NonNull and Box<> have the same size as usize. 25920797069SZhao Liu // Resize if more smart pointers are supported. 26020797069SZhao Liu self.size = std::mem::size_of::<usize>(); 26180aa3045SPaolo Bonzini } 26280aa3045SPaolo Bonzini self.flags = VMStateFlags(self.flags.0 & !VMStateFlags::VMS_SINGLE.0); 26380aa3045SPaolo Bonzini self.flags = VMStateFlags(self.flags.0 | VMStateFlags::VMS_ARRAY.0); 26480aa3045SPaolo Bonzini self.num = num as i32; 26580aa3045SPaolo Bonzini self 26680aa3045SPaolo Bonzini } 26780aa3045SPaolo Bonzini 26880aa3045SPaolo Bonzini #[must_use] 26980aa3045SPaolo Bonzini pub const fn with_pointer_flag(mut self) -> Self { 27080aa3045SPaolo Bonzini assert!((self.flags.0 & VMStateFlags::VMS_POINTER.0) == 0); 27180aa3045SPaolo Bonzini self.flags = VMStateFlags(self.flags.0 | VMStateFlags::VMS_POINTER.0); 27280aa3045SPaolo Bonzini self 27380aa3045SPaolo Bonzini } 2745b024b4eSPaolo Bonzini 2755b024b4eSPaolo Bonzini #[must_use] 2765006e39cSZhao Liu pub const fn with_varray_flag_unchecked(mut self, flag: VMStateFlags) -> VMStateField { 2775b024b4eSPaolo Bonzini self.flags = VMStateFlags(self.flags.0 & !VMStateFlags::VMS_ARRAY.0); 2785b024b4eSPaolo Bonzini self.flags = VMStateFlags(self.flags.0 | flag.0); 279c3d80af5SZhao Liu self.num = 0; // varray uses num_offset instead of num. 2805b024b4eSPaolo Bonzini self 2815b024b4eSPaolo Bonzini } 2825b024b4eSPaolo Bonzini 2835b024b4eSPaolo Bonzini #[must_use] 2845006e39cSZhao Liu #[allow(unused_mut)] 2855006e39cSZhao Liu pub const fn with_varray_flag(mut self, flag: VMStateFlags) -> VMStateField { 2865006e39cSZhao Liu assert!((self.flags.0 & VMStateFlags::VMS_ARRAY.0) != 0); 2875006e39cSZhao Liu self.with_varray_flag_unchecked(flag) 2885006e39cSZhao Liu } 2895006e39cSZhao Liu 2905006e39cSZhao Liu #[must_use] 2915b024b4eSPaolo Bonzini pub const fn with_varray_multiply(mut self, num: u32) -> VMStateField { 2925b024b4eSPaolo Bonzini assert!(num <= 0x7FFF_FFFFu32); 2935b024b4eSPaolo Bonzini self.flags = VMStateFlags(self.flags.0 | VMStateFlags::VMS_MULTIPLY_ELEMENTS.0); 2945b024b4eSPaolo Bonzini self.num = num as i32; 2955b024b4eSPaolo Bonzini self 2965b024b4eSPaolo Bonzini } 29780aa3045SPaolo Bonzini } 29880aa3045SPaolo Bonzini 29900f89716SPaolo Bonzini /// This macro can be used (by just passing it a type) to forward the `VMState` 30000f89716SPaolo Bonzini /// trait to the first field of a tuple. This is a workaround for lack of 30100f89716SPaolo Bonzini /// support of nested [`offset_of`](core::mem::offset_of) until Rust 1.82.0. 30200f89716SPaolo Bonzini /// 30300f89716SPaolo Bonzini /// # Examples 30400f89716SPaolo Bonzini /// 30500f89716SPaolo Bonzini /// ``` 30616534af5SPaolo Bonzini /// # use qemu_api::impl_vmstate_forward; 30700f89716SPaolo Bonzini /// pub struct Fifo([u8; 16]); 30800f89716SPaolo Bonzini /// impl_vmstate_forward!(Fifo); 30900f89716SPaolo Bonzini /// ``` 31000f89716SPaolo Bonzini #[macro_export] 31100f89716SPaolo Bonzini macro_rules! impl_vmstate_forward { 31200f89716SPaolo Bonzini // This is similar to impl_vmstate_transparent below, but it 31300f89716SPaolo Bonzini // uses the same trick as vmstate_of! to obtain the type of 31400f89716SPaolo Bonzini // the first field of the tuple 31500f89716SPaolo Bonzini ($tuple:ty) => { 31600f89716SPaolo Bonzini unsafe impl $crate::vmstate::VMState for $tuple { 31700f89716SPaolo Bonzini const SCALAR_TYPE: $crate::vmstate::VMStateFieldType = 31800f89716SPaolo Bonzini $crate::call_func_with_field!($crate::vmstate::vmstate_scalar_type, $tuple, 0); 31900f89716SPaolo Bonzini const BASE: $crate::bindings::VMStateField = 32000f89716SPaolo Bonzini $crate::call_func_with_field!($crate::vmstate::vmstate_base, $tuple, 0); 32100f89716SPaolo Bonzini } 32200f89716SPaolo Bonzini }; 32300f89716SPaolo Bonzini } 32400f89716SPaolo Bonzini 32580aa3045SPaolo Bonzini // Transparent wrappers: just use the internal type 32680aa3045SPaolo Bonzini 32780aa3045SPaolo Bonzini macro_rules! impl_vmstate_transparent { 32880aa3045SPaolo Bonzini ($type:ty where $base:tt: VMState $($where:tt)*) => { 32980aa3045SPaolo Bonzini unsafe impl<$base> VMState for $type where $base: VMState $($where)* { 330f2cb78bdSPaolo Bonzini const SCALAR_TYPE: VMStateFieldType = <$base as VMState>::SCALAR_TYPE; 33180aa3045SPaolo Bonzini const BASE: VMStateField = VMStateField { 33280aa3045SPaolo Bonzini size: mem::size_of::<$type>(), 33380aa3045SPaolo Bonzini ..<$base as VMState>::BASE 33480aa3045SPaolo Bonzini }; 3355b024b4eSPaolo Bonzini const VARRAY_FLAG: VMStateFlags = <$base as VMState>::VARRAY_FLAG; 33680aa3045SPaolo Bonzini } 33780aa3045SPaolo Bonzini }; 33880aa3045SPaolo Bonzini } 33980aa3045SPaolo Bonzini 34080aa3045SPaolo Bonzini impl_vmstate_transparent!(std::cell::Cell<T> where T: VMState); 34180aa3045SPaolo Bonzini impl_vmstate_transparent!(std::cell::UnsafeCell<T> where T: VMState); 342d7f5ae8bSPaolo Bonzini impl_vmstate_transparent!(std::pin::Pin<T> where T: VMState); 34380aa3045SPaolo Bonzini impl_vmstate_transparent!(crate::cell::BqlCell<T> where T: VMState); 34480aa3045SPaolo Bonzini impl_vmstate_transparent!(crate::cell::BqlRefCell<T> where T: VMState); 3453baf82e0SZhao Liu impl_vmstate_transparent!(crate::cell::Opaque<T> where T: VMState); 34680aa3045SPaolo Bonzini 34700f89716SPaolo Bonzini #[macro_export] 34800f89716SPaolo Bonzini macro_rules! impl_vmstate_bitsized { 34900f89716SPaolo Bonzini ($type:ty) => { 35000f89716SPaolo Bonzini unsafe impl $crate::vmstate::VMState for $type { 35100f89716SPaolo Bonzini const SCALAR_TYPE: $crate::vmstate::VMStateFieldType = 35200f89716SPaolo Bonzini <<<$type as ::bilge::prelude::Bitsized>::ArbitraryInt 35300f89716SPaolo Bonzini as ::bilge::prelude::Number>::UnderlyingType 35400f89716SPaolo Bonzini as $crate::vmstate::VMState>::SCALAR_TYPE; 35500f89716SPaolo Bonzini const BASE: $crate::bindings::VMStateField = 35600f89716SPaolo Bonzini <<<$type as ::bilge::prelude::Bitsized>::ArbitraryInt 35700f89716SPaolo Bonzini as ::bilge::prelude::Number>::UnderlyingType 35800f89716SPaolo Bonzini as $crate::vmstate::VMState>::BASE; 35900f89716SPaolo Bonzini const VARRAY_FLAG: $crate::bindings::VMStateFlags = 36000f89716SPaolo Bonzini <<<$type as ::bilge::prelude::Bitsized>::ArbitraryInt 36100f89716SPaolo Bonzini as ::bilge::prelude::Number>::UnderlyingType 36200f89716SPaolo Bonzini as $crate::vmstate::VMState>::VARRAY_FLAG; 36300f89716SPaolo Bonzini } 36400f89716SPaolo Bonzini }; 36500f89716SPaolo Bonzini } 36600f89716SPaolo Bonzini 367f2cb78bdSPaolo Bonzini // Scalar types using predefined VMStateInfos 368f2cb78bdSPaolo Bonzini 369f2cb78bdSPaolo Bonzini macro_rules! impl_vmstate_scalar { 370f2cb78bdSPaolo Bonzini ($info:ident, $type:ty$(, $varray_flag:ident)?) => { 371f2cb78bdSPaolo Bonzini unsafe impl VMState for $type { 372f2cb78bdSPaolo Bonzini const SCALAR_TYPE: VMStateFieldType = VMStateFieldType::$info; 373f2cb78bdSPaolo Bonzini const BASE: VMStateField = VMStateField { 374f2cb78bdSPaolo Bonzini size: mem::size_of::<$type>(), 375f2cb78bdSPaolo Bonzini flags: VMStateFlags::VMS_SINGLE, 376f2cb78bdSPaolo Bonzini ..Zeroable::ZERO 377f2cb78bdSPaolo Bonzini }; 378f2cb78bdSPaolo Bonzini $(const VARRAY_FLAG: VMStateFlags = VMStateFlags::$varray_flag;)? 379f2cb78bdSPaolo Bonzini } 380f2cb78bdSPaolo Bonzini }; 381f2cb78bdSPaolo Bonzini } 382f2cb78bdSPaolo Bonzini 383f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_bool, bool); 384f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_int8, i8); 385f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_int16, i16); 386f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_int32, i32); 387f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_int64, i64); 388f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_uint8, u8, VMS_VARRAY_UINT8); 389f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_uint16, u16, VMS_VARRAY_UINT16); 390f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_uint32, u32, VMS_VARRAY_UINT32); 391f2cb78bdSPaolo Bonzini impl_vmstate_scalar!(vmstate_info_uint64, u64); 3923baf82e0SZhao Liu impl_vmstate_scalar!(vmstate_info_timer, crate::timer::Timer); 393f2cb78bdSPaolo Bonzini 39480aa3045SPaolo Bonzini // Pointer types using the underlying type's VMState plus VMS_POINTER 39580aa3045SPaolo Bonzini // Note that references are not supported, though references to cells 39680aa3045SPaolo Bonzini // could be allowed. 39780aa3045SPaolo Bonzini 39880aa3045SPaolo Bonzini macro_rules! impl_vmstate_pointer { 39980aa3045SPaolo Bonzini ($type:ty where $base:tt: VMState $($where:tt)*) => { 40080aa3045SPaolo Bonzini unsafe impl<$base> VMState for $type where $base: VMState $($where)* { 401f2cb78bdSPaolo Bonzini const SCALAR_TYPE: VMStateFieldType = <T as VMState>::SCALAR_TYPE; 40280aa3045SPaolo Bonzini const BASE: VMStateField = <$base as VMState>::BASE.with_pointer_flag(); 40380aa3045SPaolo Bonzini } 40480aa3045SPaolo Bonzini }; 40580aa3045SPaolo Bonzini } 40680aa3045SPaolo Bonzini 40780aa3045SPaolo Bonzini impl_vmstate_pointer!(*const T where T: VMState); 40880aa3045SPaolo Bonzini impl_vmstate_pointer!(*mut T where T: VMState); 40980aa3045SPaolo Bonzini impl_vmstate_pointer!(NonNull<T> where T: VMState); 41080aa3045SPaolo Bonzini 41180aa3045SPaolo Bonzini // Unlike C pointers, Box is always non-null therefore there is no need 41280aa3045SPaolo Bonzini // to specify VMS_ALLOC. 41380aa3045SPaolo Bonzini impl_vmstate_pointer!(Box<T> where T: VMState); 4140fcccf3fSPaolo Bonzini impl_vmstate_pointer!(Owned<T> where T: VMState + ObjectType); 41580aa3045SPaolo Bonzini 41680aa3045SPaolo Bonzini // Arrays using the underlying type's VMState plus 41780aa3045SPaolo Bonzini // VMS_ARRAY/VMS_ARRAY_OF_POINTER 41880aa3045SPaolo Bonzini 41980aa3045SPaolo Bonzini unsafe impl<T: VMState, const N: usize> VMState for [T; N] { 420f2cb78bdSPaolo Bonzini const SCALAR_TYPE: VMStateFieldType = <T as VMState>::SCALAR_TYPE; 42180aa3045SPaolo Bonzini const BASE: VMStateField = <T as VMState>::BASE.with_array_flag(N); 4220d43ddaeSPaolo Bonzini } 423716d89f9SPaolo Bonzini 4240a65e412SManos Pitsidianakis #[doc(alias = "VMSTATE_UNUSED")] 4250a65e412SManos Pitsidianakis #[macro_export] 4260a65e412SManos Pitsidianakis macro_rules! vmstate_unused { 4270a65e412SManos Pitsidianakis ($size:expr) => {{ 4280a65e412SManos Pitsidianakis $crate::bindings::VMStateField { 4299d489949SPaolo Bonzini name: $crate::c_str!("unused").as_ptr(), 4300a65e412SManos Pitsidianakis size: $size, 4319d489949SPaolo Bonzini info: unsafe { ::core::ptr::addr_of!($crate::bindings::vmstate_info_unused_buffer) }, 4329d489949SPaolo Bonzini flags: $crate::bindings::VMStateFlags::VMS_BUFFER, 4339d489949SPaolo Bonzini ..$crate::zeroable::Zeroable::ZERO 4340a65e412SManos Pitsidianakis } 4350a65e412SManos Pitsidianakis }}; 4360a65e412SManos Pitsidianakis } 4370a65e412SManos Pitsidianakis 4389a2ba488SPaolo Bonzini // FIXME: including the `vmsd` field in a `const` is not possible without 4399a2ba488SPaolo Bonzini // the const_refs_static feature (stabilized in Rust 1.83.0). Without it, 4409a2ba488SPaolo Bonzini // it is not possible to use VMS_STRUCT in a transparent manner using 4419a2ba488SPaolo Bonzini // `vmstate_of!`. While VMSTATE_CLOCK can at least try to be type-safe, 4429a2ba488SPaolo Bonzini // VMSTATE_STRUCT includes $type only for documentation purposes; it 4439a2ba488SPaolo Bonzini // is checked against $field_name and $struct_name, but not against $vmsd 4449a2ba488SPaolo Bonzini // which is what really would matter. 4459a2ba488SPaolo Bonzini #[doc(alias = "VMSTATE_STRUCT")] 4469a2ba488SPaolo Bonzini #[macro_export] 4479a2ba488SPaolo Bonzini macro_rules! vmstate_struct { 4489a2ba488SPaolo Bonzini ($struct_name:ty, $field_name:ident $([0 .. $num:ident $(* $factor:expr)?])?, $vmsd:expr, $type:ty $(,)?) => { 4499a2ba488SPaolo Bonzini $crate::bindings::VMStateField { 4509a2ba488SPaolo Bonzini name: ::core::concat!(::core::stringify!($field_name), "\0") 4519a2ba488SPaolo Bonzini .as_bytes() 4529a2ba488SPaolo Bonzini .as_ptr() as *const ::std::os::raw::c_char, 4536ca5c3beSZhao Liu $(num_offset: $crate::offset_of!($struct_name, $num),)? 4549a2ba488SPaolo Bonzini offset: { 45561825825SZhao Liu $crate::assert_field_type!($struct_name, $field_name, $type $(, num = $num)?); 4569a2ba488SPaolo Bonzini $crate::offset_of!($struct_name, $field_name) 4579a2ba488SPaolo Bonzini }, 4589a2ba488SPaolo Bonzini size: ::core::mem::size_of::<$type>(), 4599a2ba488SPaolo Bonzini flags: $crate::bindings::VMStateFlags::VMS_STRUCT, 460ea8a7cebSZhao Liu vmsd: $vmsd, 46142c814b1SZhao Liu ..$crate::zeroable::Zeroable::ZERO 4625006e39cSZhao Liu } $(.with_varray_flag_unchecked( 46342c814b1SZhao Liu $crate::call_func_with_field!( 4649a2ba488SPaolo Bonzini $crate::vmstate::vmstate_varray_flag, 4659a2ba488SPaolo Bonzini $struct_name, 46642c814b1SZhao Liu $num 46742c814b1SZhao Liu ) 46842c814b1SZhao Liu ) 4699a2ba488SPaolo Bonzini $(.with_varray_multiply($factor))?)? 4709a2ba488SPaolo Bonzini }; 4719a2ba488SPaolo Bonzini } 4729a2ba488SPaolo Bonzini 4730a65e412SManos Pitsidianakis #[doc(alias = "VMSTATE_CLOCK")] 4740a65e412SManos Pitsidianakis #[macro_export] 4750a65e412SManos Pitsidianakis macro_rules! vmstate_clock { 47624f0e8d8SPaolo Bonzini ($struct_name:ty, $field_name:ident) => {{ 4779d489949SPaolo Bonzini $crate::bindings::VMStateField { 4789d489949SPaolo Bonzini name: ::core::concat!(::core::stringify!($field_name), "\0") 4799d489949SPaolo Bonzini .as_bytes() 4809d489949SPaolo Bonzini .as_ptr() as *const ::std::os::raw::c_char, 4819d489949SPaolo Bonzini offset: { 4829d489949SPaolo Bonzini $crate::assert_field_type!( 4830a65e412SManos Pitsidianakis $struct_name, 4849d489949SPaolo Bonzini $field_name, 485fc22d650SPaolo Bonzini $crate::qom::Owned<$crate::qdev::Clock> 4869d489949SPaolo Bonzini ); 4879d489949SPaolo Bonzini $crate::offset_of!($struct_name, $field_name) 4889d489949SPaolo Bonzini }, 489201ef001SPaolo Bonzini size: ::core::mem::size_of::<*const $crate::qdev::Clock>(), 4909d489949SPaolo Bonzini flags: VMStateFlags(VMStateFlags::VMS_STRUCT.0 | VMStateFlags::VMS_POINTER.0), 4919d489949SPaolo Bonzini vmsd: unsafe { ::core::ptr::addr_of!($crate::bindings::vmstate_clock) }, 4929d489949SPaolo Bonzini ..$crate::zeroable::Zeroable::ZERO 4930a65e412SManos Pitsidianakis } 4940a65e412SManos Pitsidianakis }}; 4950a65e412SManos Pitsidianakis } 4960a65e412SManos Pitsidianakis 4970a65e412SManos Pitsidianakis /// Helper macro to declare a list of 4980a65e412SManos Pitsidianakis /// ([`VMStateField`](`crate::bindings::VMStateField`)) into a static and return 4990a65e412SManos Pitsidianakis /// a pointer to the array of values it created. 5000a65e412SManos Pitsidianakis #[macro_export] 5010a65e412SManos Pitsidianakis macro_rules! vmstate_fields { 5020a65e412SManos Pitsidianakis ($($field:expr),*$(,)*) => {{ 5030a65e412SManos Pitsidianakis static _FIELDS: &[$crate::bindings::VMStateField] = &[ 5040a65e412SManos Pitsidianakis $($field),*, 5050a65e412SManos Pitsidianakis $crate::bindings::VMStateField { 5062537f830SPaolo Bonzini flags: $crate::bindings::VMStateFlags::VMS_END, 5072537f830SPaolo Bonzini ..$crate::zeroable::Zeroable::ZERO 5080a65e412SManos Pitsidianakis } 5090a65e412SManos Pitsidianakis ]; 5100a65e412SManos Pitsidianakis _FIELDS.as_ptr() 5110a65e412SManos Pitsidianakis }} 5120a65e412SManos Pitsidianakis } 5130a65e412SManos Pitsidianakis 514*b1310037SZhao Liu pub extern "C" fn rust_vms_test_field_exists<T, F: for<'a> FnCall<(&'a T, u8), bool>>( 515*b1310037SZhao Liu opaque: *mut c_void, 516*b1310037SZhao Liu version_id: c_int, 517*b1310037SZhao Liu ) -> bool { 518*b1310037SZhao Liu let owner: &T = unsafe { &*(opaque.cast::<T>()) }; 519*b1310037SZhao Liu let version: u8 = version_id.try_into().unwrap(); 520*b1310037SZhao Liu // SAFETY: the opaque was passed as a reference to `T`. 521*b1310037SZhao Liu F::call((owner, version)) 522*b1310037SZhao Liu } 523*b1310037SZhao Liu 524*b1310037SZhao Liu pub type VMSFieldExistCb = unsafe extern "C" fn( 525*b1310037SZhao Liu opaque: *mut std::os::raw::c_void, 526*b1310037SZhao Liu version_id: std::os::raw::c_int, 527*b1310037SZhao Liu ) -> bool; 528*b1310037SZhao Liu 529*b1310037SZhao Liu #[doc(alias = "VMSTATE_VALIDATE")] 530*b1310037SZhao Liu #[macro_export] 531*b1310037SZhao Liu macro_rules! vmstate_validate { 532*b1310037SZhao Liu ($struct_name:ty, $test_name:expr, $test_fn:expr $(,)?) => { 533*b1310037SZhao Liu $crate::bindings::VMStateField { 534*b1310037SZhao Liu name: ::std::ffi::CStr::as_ptr($test_name), 535*b1310037SZhao Liu field_exists: { 536*b1310037SZhao Liu const fn test_cb_builder__< 537*b1310037SZhao Liu T, 538*b1310037SZhao Liu F: for<'a> $crate::callbacks::FnCall<(&'a T, u8), bool>, 539*b1310037SZhao Liu >( 540*b1310037SZhao Liu _phantom: ::core::marker::PhantomData<F>, 541*b1310037SZhao Liu ) -> $crate::vmstate::VMSFieldExistCb { 542*b1310037SZhao Liu let _: () = F::ASSERT_IS_SOME; 543*b1310037SZhao Liu $crate::vmstate::rust_vms_test_field_exists::<T, F> 544*b1310037SZhao Liu } 545*b1310037SZhao Liu 546*b1310037SZhao Liu const fn phantom__<T>(_: &T) -> ::core::marker::PhantomData<T> { 547*b1310037SZhao Liu ::core::marker::PhantomData 548*b1310037SZhao Liu } 549*b1310037SZhao Liu Some(test_cb_builder__::<$struct_name, _>(phantom__(&$test_fn))) 550*b1310037SZhao Liu }, 551*b1310037SZhao Liu flags: $crate::bindings::VMStateFlags( 552*b1310037SZhao Liu $crate::bindings::VMStateFlags::VMS_MUST_EXIST.0 553*b1310037SZhao Liu | $crate::bindings::VMStateFlags::VMS_ARRAY.0, 554*b1310037SZhao Liu ), 555*b1310037SZhao Liu num: 0, // 0 elements: no data, only run test_fn callback 556*b1310037SZhao Liu ..$crate::zeroable::Zeroable::ZERO 557*b1310037SZhao Liu } 558*b1310037SZhao Liu }; 559*b1310037SZhao Liu } 560*b1310037SZhao Liu 5610a65e412SManos Pitsidianakis /// A transparent wrapper type for the `subsections` field of 562716d89f9SPaolo Bonzini /// [`VMStateDescription`]. 5630a65e412SManos Pitsidianakis /// 5640a65e412SManos Pitsidianakis /// This is necessary to be able to declare subsection descriptions as statics, 5650a65e412SManos Pitsidianakis /// because the only way to implement `Sync` for a foreign type (and `*const` 5660a65e412SManos Pitsidianakis /// pointers are foreign types in Rust) is to create a wrapper struct and 5670a65e412SManos Pitsidianakis /// `unsafe impl Sync` for it. 5680a65e412SManos Pitsidianakis /// 5690a65e412SManos Pitsidianakis /// This struct is used in the 5700a65e412SManos Pitsidianakis /// [`vm_state_subsections`](crate::vmstate_subsections) macro implementation. 5710a65e412SManos Pitsidianakis #[repr(transparent)] 5720a65e412SManos Pitsidianakis pub struct VMStateSubsectionsWrapper(pub &'static [*const crate::bindings::VMStateDescription]); 5730a65e412SManos Pitsidianakis 5740a65e412SManos Pitsidianakis unsafe impl Sync for VMStateSubsectionsWrapper {} 5750a65e412SManos Pitsidianakis 576716d89f9SPaolo Bonzini /// Helper macro to declare a list of subsections ([`VMStateDescription`]) 577716d89f9SPaolo Bonzini /// into a static and return a pointer to the array of pointers it created. 5780a65e412SManos Pitsidianakis #[macro_export] 5790a65e412SManos Pitsidianakis macro_rules! vmstate_subsections { 5800a65e412SManos Pitsidianakis ($($subsection:expr),*$(,)*) => {{ 5810a65e412SManos Pitsidianakis static _SUBSECTIONS: $crate::vmstate::VMStateSubsectionsWrapper = $crate::vmstate::VMStateSubsectionsWrapper(&[ 5820a65e412SManos Pitsidianakis $({ 5830a65e412SManos Pitsidianakis static _SUBSECTION: $crate::bindings::VMStateDescription = $subsection; 5840a65e412SManos Pitsidianakis ::core::ptr::addr_of!(_SUBSECTION) 5850a65e412SManos Pitsidianakis }),*, 5860a65e412SManos Pitsidianakis ::core::ptr::null() 5870a65e412SManos Pitsidianakis ]); 5880a65e412SManos Pitsidianakis _SUBSECTIONS.0.as_ptr() 5890a65e412SManos Pitsidianakis }} 5900a65e412SManos Pitsidianakis } 591