Skip to main content

pin_init_internal/
lib.rs

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
13use proc_macro::TokenStream;
14use syn::parse_macro_input;
15
16use crate::diagnostics::DiagCtxt;
17
18mod diagnostics;
19mod init;
20mod pin_data;
21mod pinned_drop;
22mod zeroable;
23
24#[proc_macro_attribute]
25pub 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]
32pub 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)]
39pub 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)]
45pub 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]
50pub 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]
57pub 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}