xref: /linux/rust/pin-init/internal/src/lib.rs (revision 531fdc2c341bded76817cef935c5fe97d88d643b)
1 // SPDX-License-Identifier: Apache-2.0 OR MIT
2 
3 // When fixdep scans this, it will find this string `CONFIG_RUSTC_VERSION_TEXT`
4 // and thus add a dependency on `include/config/RUSTC_VERSION_TEXT`, which is
5 // touched by Kconfig when the version string from the compiler changes.
6 
7 //! `pin-init` proc macros.
8 
9 // Documentation is done in the pin-init crate instead.
10 #![allow(missing_docs)]
11 
12 use proc_macro::TokenStream;
13 use syn::parse_macro_input;
14 
15 use crate::diagnostics::DiagCtxt;
16 
17 mod diagnostics;
18 mod init;
19 mod pin_data;
20 mod pinned_drop;
21 mod zeroable;
22 
23 #[proc_macro_attribute]
24 pub fn pin_data(args: TokenStream, input: TokenStream) -> TokenStream {
25     let args = parse_macro_input!(args);
26     let input = parse_macro_input!(input);
27     DiagCtxt::with(|dcx| pin_data::pin_data(args, input, dcx)).into()
28 }
29 
30 #[proc_macro_attribute]
31 pub fn pinned_drop(args: TokenStream, input: TokenStream) -> TokenStream {
32     let args = parse_macro_input!(args);
33     let input = parse_macro_input!(input);
34     DiagCtxt::with(|dcx| pinned_drop::pinned_drop(args, input, dcx)).into()
35 }
36 
37 #[proc_macro_derive(Zeroable)]
38 pub fn derive_zeroable(input: TokenStream) -> TokenStream {
39     let input = parse_macro_input!(input);
40     DiagCtxt::with(|dcx| zeroable::derive(input, dcx)).into()
41 }
42 
43 #[proc_macro_derive(MaybeZeroable)]
44 pub fn maybe_derive_zeroable(input: TokenStream) -> TokenStream {
45     let input = parse_macro_input!(input);
46     DiagCtxt::with(|dcx| zeroable::maybe_derive(input, dcx)).into()
47 }
48 #[proc_macro]
49 pub fn init(input: TokenStream) -> TokenStream {
50     let input = parse_macro_input!(input);
51     DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), false, dcx))
52         .into()
53 }
54 
55 #[proc_macro]
56 pub fn pin_init(input: TokenStream) -> TokenStream {
57     let input = parse_macro_input!(input);
58     DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), true, dcx)).into()
59 }
60