xref: /linux/rust/pin-init/internal/src/lib.rs (revision 560f6d13c33f9f06ca34c14dc7c0a045d949c4a0)
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 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