gear_core_backend_codegen/
lib.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
// This file is part of Gear.
//
// Copyright (C) 2021-2023 Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

use host::{HostFn, HostFnMeta};
use proc_macro::TokenStream;
use syn::ItemFn;

mod host;

/// Apply host state wrapper to host functions.
///
/// Supported meta attributes:
/// * `fallible`             - if the host function executes fallible call.
/// * `wgas`                 - if the host function supports with-gas version.
/// * `cost`                 - RuntimeCosts definition, for example `#[host(cost = RuntimeCosts::Null)]`
/// * `err`                  - Structure definition with error code, for example `#[host(err = ErrorWithHash)]`
///
/// # Example
///
/// ```ignore
/// #[host(fallible, wgas, cost = RuntimeCosts::Reply(len))]
/// pub fn reply(
///     ctx: &mut R,
///     payload_ptr: u32,
///     len: u32,
///     value_ptr: u32,
///     delay: u32,
/// ) -> Result<(), R::Error> {
///     let read_payload = ctx.register_read(payload_ptr, len);
///     let value = ctx.register_and_read_value(value_ptr)?;
///     let payload = ctx.read(read_payload)?.try_into()?;
///
///     let state = ctx.host_state_mut();
///     state.ext.reply(ReplyPacket::new(payload, value), delay)
///         .map_err(Into::into)
/// }
/// ```
///
/// will generate
///
/// ```ignore
/// pub fn reply(ctx: &mut R,
///     payload_ptr: u32,
///     len: u32,
///     value_ptr: u32,
///     delay: u32
/// ) -> Result<(), R::Error> {
///     syscall_trace!("reply", payload_ptr, len, value_ptr, delay, err_mid_ptr);
///
///     ctx.run_fallible::<_, _, ErrorWithHash>(err_mid_ptr, RuntimeCosts::Reply(len), |ctx| {
///         // ...
///     })
/// }
///
/// pub fn reply_wgas(
///     ctx: &mut R,
///     payload_ptr: u32,
///     len: u32,
///     gas_limit: u64,
///     value_ptr: u32,
///     delay: u32
/// ) -> Result<(), R::Error> {
///     syscall_trace!("reply_wgas", payload_ptr, len, gas_limit, value_ptr, delay, err_mid_ptr);
///
///     ctx.run_fallible::<_, _, ErrorWithHash>(err_mid_ptr, RuntimeCosts::ReplyWGas(len), |ctx| {
///         // ...
///     })
/// }
/// ```
#[proc_macro_attribute]
pub fn host(meta: TokenStream, item: TokenStream) -> TokenStream {
    let meta: HostFnMeta = syn::parse_macro_input!(meta);
    let item: ItemFn = syn::parse_macro_input!(item);

    HostFn::new(meta, item).into()
}