1
0
mirror of https://github.com/bspeice/qadapt synced 2024-11-13 01:28:08 -05:00

Start work on a procedural macro

Turns out the system still needs some work.
This commit is contained in:
Bradlee Speice 2018-11-06 23:04:26 -05:00
parent 16a0238dc5
commit 18d1c3e6d1
6 changed files with 50 additions and 2 deletions

View File

@ -12,4 +12,6 @@ repository = "https://github.com/bspeice/qadapt.git"
backtrace = "0.3"
libc = "0.2"
log = "0.4"
spin = "0.4"
spin = "0.4"
qadapt-macro = { path = "./qadapt-macro" }

3
qadapt-macro/.gitignore vendored Normal file
View File

@ -0,0 +1,3 @@
/target
**/*.rs.bk
Cargo.lock

9
qadapt-macro/Cargo.toml Normal file
View File

@ -0,0 +1,9 @@
[package]
name = "qadapt-macro"
version = "0.1.0"
authors = ["Bradlee Speice <bradlee@speice.io>"]
[lib]
proc-macro = true
[dependencies]

31
qadapt-macro/src/lib.rs Normal file
View File

@ -0,0 +1,31 @@
//! Helper macros to use with the QADAPT allocator system
//!
//! This crate is unusable on its own, but because `proc_macro` crates
//! can't export non-proc-macro symbols, we have to go through an extra step
//! to move these up.
//!
//! Ultimately, this does actually work because we don't need to actually use
//! references to the underlying functionality here, we just need to know
//! where they will ultimately end up at.
// TODO: This causes issues, but I can't track down why
// #![deny(missing_docs)]
extern crate proc_macro;
use proc_macro::TokenTree;
use proc_macro::TokenStream;
/// Set up the QADAPT allocator to trigger a panic if any allocations happen during
/// this function. Race conditions between threads are not checked, this macro will
/// only work as intended in a single-threaded or otherwise synchronized environment.
#[proc_macro_attribute]
pub fn allocate_panic(_attr: TokenStream, item: TokenStream) -> TokenStream {
let ret = item.clone();
let mut token_iter = item.into_iter();
match token_iter.next() {
Some(TokenTree::Ident(ref i)) if i.to_string() == "fn" => (),
_ => panic!("#[allocate_panic] macro can only be applied to functions")
}
ret
}

View File

@ -1,9 +1,13 @@
extern crate backtrace;
extern crate libc;
extern crate qadapt_macro;
#[macro_use]
extern crate log;
extern crate spin;
// Re-export the proc macros to use by other code
pub use qadapt_macro::*;
use backtrace::Backtrace;
use libc::c_void;
use libc::free;

View File

@ -1,5 +1,4 @@
#![feature(asm)]
extern crate qadapt;
use qadapt::QADAPT;