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
|
/*
* MRustC - Rust Compiler
* - By John Hodge (Mutabah/thePowersGang)
*
* expand/assert.cpp
* - assert! built-in macro (1.29)
*/
#include <synext_macro.hpp>
#include <synext.hpp> // for Expand_BareExpr
#include <parse/interpolated_fragment.hpp>
#include "../parse/ttstream.hpp"
#include "../parse/common.hpp"
#include "../parse/parseerror.hpp"
class CExpander_assert:
public ExpandProcMacro
{
::std::unique_ptr<TokenStream> expand(const Span& sp, const ::AST::Crate& crate, const TokenTree& tt, AST::Module& mod) override
{
Token tok;
auto lex = TTStream(sp, tt);
lex.parse_state().module = &mod;
// assertion condition
auto n = Parse_Expr0(lex);
ASSERT_BUG(sp, n, "No expression returned");
::std::vector<TokenTree> toks;
toks.push_back( Token(TOK_RWORD_IF) );
toks.push_back( Token(TOK_EXCLAM) );
GET_TOK(tok, lex);
if( tok == TOK_COMMA )
{
toks.push_back( Token(InterpolatedFragment(InterpolatedFragment::EXPR, n.release())) );
toks.push_back( Token(TOK_BRACE_OPEN) );
// User-provided message
toks.push_back( Token(TOK_IDENT, RcString::new_interned("panic")) );
toks.push_back( Token(TOK_EXCLAM) );
toks.push_back( Token(TOK_PAREN_OPEN) );
while(lex.lookahead(0) != TOK_EOF )
{
toks.push_back( Token(InterpolatedFragment(InterpolatedFragment::EXPR, Parse_Expr0(lex).release())) );
if( lex.lookahead(0) != TOK_COMMA )
break;
GET_CHECK_TOK(tok, lex, TOK_COMMA);
toks.push_back( Token(TOK_COMMA) );
}
GET_CHECK_TOK(tok, lex, TOK_EOF);
toks.push_back( Token(TOK_PAREN_CLOSE) );
}
else if( tok == TOK_EOF )
{
::std::stringstream ss;
ss << "assertion failed: ";
n->print(ss);
toks.push_back( Token(InterpolatedFragment(InterpolatedFragment::EXPR, n.release())) );
toks.push_back( Token(TOK_BRACE_OPEN) );
// Auto-generated message
toks.push_back( Token(TOK_IDENT, RcString::new_interned("panic")) );
toks.push_back( Token(TOK_EXCLAM) );
toks.push_back( Token(TOK_PAREN_OPEN) );
toks.push_back( Token(TOK_STRING, ss.str()) );
toks.push_back( Token(TOK_PAREN_CLOSE) );
}
else
{
throw ParseError::Unexpected(lex, tok, {TOK_COMMA, TOK_EOF});
}
toks.push_back( Token(TOK_BRACE_CLOSE) );
return box$( TTStreamO(sp, TokenTree(Ident::Hygiene::new_scope(), mv$(toks))) );
}
};
void Expand_init_assert()
{
if( TARGETVER_1_29 )
{
Register_Synext_Macro("assert", ::std::unique_ptr<ExpandProcMacro>(new CExpander_assert));
}
}
|
