Link: https://lore.kernel.org/r/20260217200002.683975158@linuxfoundation.org Tested-by: Florian Fainelli <florian.fainelli@broadcom.com> Tested-by: Takeshi Ogasawara <takeshi.ogasawara@futuring-girl.com> Tested-by: Peter Schneider <pschneider1968@googlemail.com> Tested-by: Jon Hunter <jonathanh@nvidia.com> Tested-by: Salvatore Bonaccorso <carnil@debian.org> Tested-by: Brett A C Sheffield <bacs@librecast.net> Tested-by: Mark Brown <broonie@kernel.org> Tested-by: Luna Jernberg <droidbittin@gmail.com> Tested-by: Ronald Warsow <rwarsow@gmx.de> Tested-by: Justin M. Forbes <jforbes@fedoraproject.org> Tested-by: Ron Economos <re@w6rz.net> Tested-by: Miguel Ojeda <ojeda@kernel.org> Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
30 lines
945 B
Rust
30 lines
945 B
Rust
// SPDX-License-Identifier: GPL-2.0
|
|
|
|
use crate::helpers::function_name;
|
|
use proc_macro::TokenStream;
|
|
|
|
/// Please see [`crate::export`] for documentation.
|
|
pub(crate) fn export(_attr: TokenStream, ts: TokenStream) -> TokenStream {
|
|
let Some(name) = function_name(ts.clone()) else {
|
|
return "::core::compile_error!(\"The #[export] attribute must be used on a function.\");"
|
|
.parse::<TokenStream>()
|
|
.unwrap();
|
|
};
|
|
|
|
// This verifies that the function has the same signature as the declaration generated by
|
|
// bindgen. It makes use of the fact that all branches of an if/else must have the same type.
|
|
let signature_check = quote!(
|
|
const _: () = {
|
|
if true {
|
|
::kernel::bindings::#name
|
|
} else {
|
|
#name
|
|
};
|
|
};
|
|
);
|
|
|
|
let no_mangle = quote!(#[no_mangle]);
|
|
|
|
TokenStream::from_iter([signature_check, no_mangle, ts])
|
|
}
|