Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-02 06:38:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

"-O gen-C++" support for "assert" statements

Browse source
This commit is contained in:
Vern Paxson 2023-08-16 17:07:21 -07:00
parent 4928e074d4
commit 81a9745fb3
5 changed files with 146 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/script_opt/CPP/Stmts.cc
View file

@ -78,6 +78,10 @@ void CPPCompile::GenStmt(const Stmt* s)
GenForStmt(s->AsForStmt());
break;
case STMT_ASSERT:
GenAssertStmt(s->AsAssertStmt());
break;
case STMT_NEXT:
Emit("continue;");
break;
@ -567,4 +571,33 @@ void CPPCompile::GenForOverString(const ExprPtr& str, const IDPList* loop_vars)
Emit("%s = std::move(sv__CPP);", IDName(lv0));
}
void CPPCompile::GenAssertStmt(const AssertStmt* a)
{
auto& cond = a->Cond();
auto& msg = a->Msg();
Emit("{ // begin a new scope for internal \"assert\" variables");
Emit("static auto assertion_result_hook = id::find_func(\"assertion_result\");");
Emit("bool run_result_hook = assertion_result_hook && "
"assertion_result_hook->HasEnabledBodies();");
Emit("auto assert_result = %s;", GenExpr(cond, GEN_NATIVE));
Emit("if ( ! assert_result || run_result_hook )");
StartBlock();
if ( msg )
Emit("auto msg_val = %s;", GenExpr(msg, GEN_VAL_PTR));
else
Emit("auto msg_val = zeek::val_mgr->EmptyString();");
auto loc = a->GetLocationInfo();
Emit("static Location loc(\"%s\", %s, %s, %s, %s);", loc->filename,
std::to_string(loc->first_line), std::to_string(loc->last_line),
std::to_string(loc->first_column), std::to_string(loc->last_column));
Emit("report_assert(assert_result, \"%s\", msg_val, &loc);",
CPPEscape(a->CondDesc().c_str()).c_str());
EndBlock();
Emit("} // end of \"assert\" scope");
}
} // zeek::detail