openharmony/third_party_rust_proc-macro-error
1
0
Fork 0
mirror of https://github.com/openharmony/third_party_rust_proc-macro-error.git synced 2026-07-19 16:33:33 -04:00
Code Issues Packages Projects Releases Wiki Activity

Support nested #[proc_macro_error]

Browse Source
This commit is contained in:
CreepySkeleton
2020-03-25 21:43:02 +03:00
parent 6f9b1801b8
commit b8552c8f16
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lib.rs
+4 -4
View File
@@ -327,11 +327,11 @@ pub fn entry_point<F>(f: F, proc_macro_hack: bool) -> proc_macro::TokenStream
where
F: FnOnce() -> proc_macro::TokenStream + UnwindSafe,
{
ENTERED_ENTRY_POINT.with(|flag| flag.set(true));
ENTERED_ENTRY_POINT.with(|flag| flag.set(flag.get() + 1));
let caught = catch_unwind(f);
let dummy = dummy::cleanup();
let err_storage = imp::cleanup();
ENTERED_ENTRY_POINT.with(|flag| flag.set(false));
ENTERED_ENTRY_POINT.with(|flag| flag.set(flag.get() - 1));
let mut appendix = TokenStream::new();
if proc_macro_hack {
@@ -365,13 +365,13 @@ fn abort_now() -> ! {
}
thread_local! {
static ENTERED_ENTRY_POINT: Cell<bool> = Cell::new(false);
static ENTERED_ENTRY_POINT: Cell<usize> = Cell::new(0);
}
struct AbortNow;
fn check_correctness() {
if !ENTERED_ENTRY_POINT.with(|flag| flag.get()) {
if ENTERED_ENTRY_POINT.with(|flag| flag.get()) == 0 {
panic!(
"proc-macro-error API cannot be used outside of `entry_point` invocation, \
perhaps you forgot to annotate your #[proc_macro] function with `#[proc_macro_error]"