xref: /linux/rust/pin-init/internal/src/lib.rs (revision 23b0f90ba871f096474e1c27c3d14f455189d2d9)
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 #![cfg_attr(not(RUSTC_LINT_REASONS_IS_STABLE), feature(lint_reasons))]
10 // Documentation is done in the pin-init crate instead.
11 #![allow(missing_docs)]
12 
13 use proc_macro::TokenStream;
14 use syn::parse_macro_input;
15 
16 use crate::diagnostics::DiagCtxt;
17 
18 mod diagnostics;
19 mod init;
20 mod pin_data;
21 mod pinned_drop;
22 mod zeroable;
23 
24 #[proc_macro_attribute]
25 pub fn pin_data(args: TokenStream, input: TokenStream) -> TokenStream {
26     let args = parse_macro_input!(args);
27     let input = parse_macro_input!(input);
28     DiagCtxt::with(|dcx| pin_data::pin_data(args, input, dcx)).into()
29 }
30 
31 #[proc_macro_attribute]
32 pub fn pinned_drop(args: TokenStream, input: TokenStream) -> TokenStream {
33     let args = parse_macro_input!(args);
34     let input = parse_macro_input!(input);
35     DiagCtxt::with(|dcx| pinned_drop::pinned_drop(args, input, dcx)).into()
36 }
37 
38 #[proc_macro_derive(Zeroable)]
39 pub fn derive_zeroable(input: TokenStream) -> TokenStream {
40     let input = parse_macro_input!(input);
41     DiagCtxt::with(|dcx| zeroable::derive(input, dcx)).into()
42 }
43 
44 #[proc_macro_derive(MaybeZeroable)]
45 pub fn maybe_derive_zeroable(input: TokenStream) -> TokenStream {
46     let input = parse_macro_input!(input);
47     DiagCtxt::with(|dcx| zeroable::maybe_derive(input, dcx)).into()
48 }
49 #[proc_macro]
50 pub fn init(input: TokenStream) -> TokenStream {
51     let input = parse_macro_input!(input);
52     DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), false, dcx))
53         .into()
54 }
55 
56 #[proc_macro]
57 pub fn pin_init(input: TokenStream) -> TokenStream {
58     let input = parse_macro_input!(input);
59     DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), true, dcx)).into()
60 }
61