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

GH-2034: Store module names and use them in lookups for ifdef

Browse source
This commit is contained in:
Tim Wojtulewicz 2022-08-02 12:19:36 -07:00 • committed by Tim Wojtulewicz
parent 686e740bbe
commit 1870d26684
6 changed files with 88 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/parse.y
View file

@ -156,6 +156,8 @@ static int func_hdr_cond_epoch = 0;
EnumType* cur_enum_type = nullptr;
static ID* cur_decl_type_id = nullptr;
std::set<std::string> module_names;
static void parse_new_enum(void)
{
// Starting a new enum definition.
@ -1269,6 +1271,7 @@ decl:
TOK_MODULE TOK_ID ';'
{
current_module = $2;
module_names.insert($2);
zeekygen_mgr->ModuleUsage(::filename, current_module);
}

16
src/scan.l
View file

@ -80,6 +80,8 @@ const char* last_tok_filename = 0;
const char* last_last_tok_filename = 0;
char last_tok[128];
extern std::set<std::string> module_names;
#define YY_USER_ACTION strncpy(last_tok, yytext, sizeof(last_tok) - 1); \
last_last_tok_filename = last_tok_filename; \
last_tok_filename = ::filename;
@ -796,7 +798,12 @@ void do_atifdef(const char* id)
const auto& i = zeek::detail::lookup_ID(id, zeek::detail::current_module.c_str());
if ( ! i )
// If the ID didn't exist, look to see if this is a module name instead.
bool found = (i != nullptr);
if ( ! found )
found = (module_names.count(id) != 0);
if ( ! found )
begin_ignoring();
}
@ -806,7 +813,12 @@ void do_atifndef(const char *id)
const auto& i = zeek::detail::lookup_ID(id, zeek::detail::current_module.c_str());
if ( i )
// If the ID didn't exist, look to see if this is a module name instead.
bool found = (i != nullptr);
if ( ! found )
found = (module_names.count(id) != 0);
if ( found )
begin_ignoring();
}

2
testing/btest/Baseline/language.at-ifdef/out
View file

@ -2,3 +2,5 @@
@ifdef (PASS)
@ifdef...@else (PASS)
@ifdef...@else (PASS)
@ifdef module name (PASS)
@ifdef child variable (PASS)

2
testing/btest/Baseline/language.at-ifndef/out
View file

@ -2,3 +2,5 @@
@ifndef (PASS)
@ifndef...@else (PASS)
@ifndef...@else (PASS)
@ifndef module name (PASS)
@ifndef child variable (PASS)

34
testing/btest/language/at-ifdef.zeek
View file

@ -1,7 +1,9 @@
# @TEST-EXEC: zeek -b %INPUT >out
# @TEST-EXEC: zeek -b bar.zeek main.zeek >out
# @TEST-EXEC: btest-diff out
# @TEST-EXEC: btest-diff .stderr
@TEST-START-FILE main.zeek
function test_case(msg: string, expect: bool)
{
print fmt("%s (%s)", msg, expect ? "PASS" : "FAIL");
@ -46,5 +48,35 @@ event zeek_init()
test_case( "@ifdef...@else", xyz == 1 );
xyz = 0;
@ifdef ( Bar )
xyz += 1;
@else
xyz += 2;
@endif
test_case( "@ifdef module name", xyz == 1 );
xyz = 0;
@ifdef ( Bar::exists )
xyz += 1;
@else
xyz += 2;
@endif
test_case( "@ifdef child variable", xyz == 1 );
}
@TEST-END-FILE
@TEST-START-FILE bar.zeek
module Bar;
export {
option exists = T;
}
@TEST-END-FILE

35
testing/btest/language/at-ifndef.zeek
View file

@ -1,7 +1,9 @@
# @TEST-EXEC: zeek -b %INPUT >out
# @TEST-EXEC: zeek -b bar.zeek main.zeek >out
# @TEST-EXEC: btest-diff out
# @TEST-EXEC: btest-diff .stderr
@TEST-START-FILE main.zeek
function test_case(msg: string, expect: bool)
{
print fmt("%s (%s)", msg, expect ? "PASS" : "FAIL");
@ -46,5 +48,36 @@ event zeek_init()
test_case( "@ifndef...@else", xyz == 2 );
xyz = 0;
@ifndef ( Bar )
xyz += 1;
@else
xyz += 2;
@endif
test_case( "@ifndef module name", xyz == 2 );
xyz = 0;
@ifndef ( Bar::exists )
xyz += 1;
@else
xyz += 2;
@endif
test_case( "@ifndef child variable", xyz == 2 );
}
@TEST-END-FILE
@TEST-START-FILE bar.zeek
module Bar;
export {
option exists = T;
}
@TEST-END-FILE