rust/qemu-api-macros/src/lib.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

// Copyright 2024, Linaro Limited
// Author(s): Manos Pitsidianakis <manos.pitsidianakis@linaro.org>
// SPDX-License-Identifier: GPL-2.0-or-later

use proc_macro::TokenStream;
use quote::{format_ident, quote};
use syn::{parse_macro_input, DeriveInput};

#[proc_macro_derive(Object)]
pub fn derive_object(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);

    let name = input.ident;
    let module_static = format_ident!("__{}_LOAD_MODULE", name);

    let expanded = quote! {
        #[allow(non_upper_case_globals)]
        #[used]
        #[cfg_attr(
            not(any(target_vendor = "apple", target_os = "windows")),
            link_section = ".init_array"
        )]
        #[cfg_attr(target_vendor = "apple", link_section = "__DATA,__mod_init_func")]
        #[cfg_attr(target_os = "windows", link_section = ".CRT$XCU")]
        pub static #module_static: extern "C" fn() = {
            extern "C" fn __register() {
                unsafe {
                    ::qemu_api::bindings::type_register_static(&<#name as ::qemu_api::definitions::ObjectImpl>::TYPE_INFO);
                }
            }

            extern "C" fn __load() {
                unsafe {
                    ::qemu_api::bindings::register_module_init(
                        Some(__register),
                        ::qemu_api::bindings::module_init_type::MODULE_INIT_QOM
                    );
                }
            }

            __load
        };
    };

    TokenStream::from(expanded)
}