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

Prevent recursion of &on_change handlers.

Browse source 
This change prevents &on_change handlers for a table from running if an
&on_change handler for the same table is already running.
This commit is contained in:
Johanna Amann 2020-02-03 12:33:13 -08:00
parent 7166cb7373
commit 7f9f66fce9
5 changed files with 46 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/Attr.cc
View file

@ -14,7 +14,7 @@ const char* attr_name(attr_tag t)
"&read_expire", "&write_expire", "&create_expire",
"&raw_output", "&priority",
"&group", "&log", "&error_handler", "&type_column",
"(&tracked)", "&deprecated",
"(&tracked)", "&on_change", "&deprecated",
};
return attr_names[int(t)];
@ -405,7 +405,7 @@ void Attributes::CheckAttr(Attr* a)
{
if ( type->Tag() != TYPE_TABLE )
{
Error("expiration only applicable to tables");
Error("expiration only applicable to sets/tables");
break;
}

5
src/Val.cc
View file

@ -1933,7 +1933,7 @@ ListVal* TableVal::RecoverIndex(const HashKey* k) const
void TableVal::CallChangeFunc(const Val* index, Val* old_value, OnChangeType tpe)
{
if ( ! change_func || ! index )
if ( ! change_func || ! index || in_change_func )
return;
if ( ! table_type->IsSet() && ! old_value )
@ -1978,12 +1978,15 @@ void TableVal::CallChangeFunc(const Val* index, Val* old_value, OnChangeType tpe
if ( ! table_type->IsSet() )
vl.append(old_value->Ref());
in_change_func = true;
f->Call(&vl);
Unref(thefunc);
}
catch ( InterpreterException& e )
{
}
in_change_func = false;
}
Val* TableVal::Delete(const Val* index)

2
src/Val.h
View file

@ -865,6 +865,8 @@ protected:
PrefixTable* subnets;
Val* def_val;
Expr* change_func = nullptr;
// prevent recursion of change functions
bool in_change_func = false;
};
class RecordVal : public Val, public notifier::Modifiable {

8
testing/btest/Baseline/language.on_change-recurse/output Normal file
View file

@ -0,0 +1,8 @@
inserting
change_function, a, 1, 5, TABLE_ELEMENT_NEW
set_change, hi, TABLE_ELEMENT_NEW
changing
change_function, a, 1, 5, TABLE_ELEMENT_CHANGED
deleting
change_function, a, 1, 5, TABLE_ELEMENT_REMOVED
set_change, hi, TABLE_ELEMENT_REMOVED

30
testing/btest/language/on_change-recurse.test Normal file
View file

@ -0,0 +1,30 @@
# @TEST-EXEC: zeek %INPUT >output
# @TEST-EXEC: btest-diff output
module TestModule;
function change_function(t: table[string, int] of count, tpe: TableChange, idxa: string, idxb: int, val: count)
{
print "change_function", idxa, idxb, val, tpe;
t[idxa, idxb] = val;
}
function set_change(t: set[string], tpe: TableChange, idx: string)
{
print "set_change", idx, tpe;
}
global t: table[string, int] of count &on_change=change_function;
global s: set[string] &on_change=set_change;
event zeek_init()
{
print "inserting";
t["a", 1] = 5;
add s["hi"];
print "changing";
t["a", 1] = 2;
print "deleting";
delete t["a", 1];
delete s["hi"];
}