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

NetControl: make new broker plugin options accessible

Browse source
This commit is contained in:
Johanna Amann 2016-03-14 21:10:12 -07:00
parent 0bac1faa09
commit 06414e5c40
2 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

30
scripts/base/frameworks/netcontrol/plugins/broker.bro
View file

@ -9,14 +9,9 @@ module NetControl;
@load base/frameworks/broker @load base/frameworks/broker
export { export {
## Instantiates the broker plugin.
global create_broker: function(host: addr, host_port: port, topic: string, can_expire: bool &default=F) : PluginState;
type BrokerConfig: record { type BrokerConfig: record {
## The broker topic used to send events to ## The broker topic used to send events to
topic: string &optional; topic: string &optional;
## The ID of this broker instance - for the mapping to PluginStates
id: count &optional;
## Broker host to connect to ## Broker host to connect to
host: addr &optional; host: addr &optional;
## Broker port to connect to ## Broker port to connect to
@ -37,8 +32,13 @@ export {
check_pred: function(p: PluginState, r: Rule): bool &optional; check_pred: function(p: PluginState, r: Rule): bool &optional;
}; };
## Instantiates the broker plugin.
global create_broker: function(config: BrokerConfig, can_expire: bool) : PluginState;
redef record PluginState += { redef record PluginState += {
broker_config: BrokerConfig &optional; broker_config: BrokerConfig &optional;
## The ID of this broker instance - for the mapping to PluginStates
broker_id: count &optional;
}; };
global broker_add_rule: event(id: count, r: Rule); global broker_add_rule: event(id: count, r: Rule);
@ -133,7 +133,7 @@ function broker_add_rule_fun(p: PluginState, r: Rule) : bool
if ( ! broker_check_rule(p, r) ) if ( ! broker_check_rule(p, r) )
return F; return F;
BrokerComm::event(p$broker_config$topic, BrokerComm::event_args(broker_add_rule, p$broker_config$id, r)); BrokerComm::event(p$broker_config$topic, BrokerComm::event_args(broker_add_rule, p$broker_id, r));
return T; return T;
} }
@ -142,7 +142,7 @@ function broker_remove_rule_fun(p: PluginState, r: Rule) : bool
if ( ! broker_check_rule(p, r) ) if ( ! broker_check_rule(p, r) )
return F; return F;
BrokerComm::event(p$broker_config$topic, BrokerComm::event_args(broker_remove_rule, p$broker_config$id, r)); BrokerComm::event(p$broker_config$topic, BrokerComm::event_args(broker_remove_rule, p$broker_id, r));
return T; return T;
} }
@ -178,23 +178,23 @@ global broker_plugin_can_expire = Plugin(
$init = broker_init $init = broker_init
); );
function create_broker(host: addr, host_port: port, topic: string, can_expire: bool &default=F) : PluginState function create_broker(config: BrokerConfig, can_expire: bool) : PluginState
{ {
if ( topic in netcontrol_broker_topics ) if ( config$topic in netcontrol_broker_topics )
Reporter::warning(fmt("Topic %s was added to NetControl broker plugin twice. Possible duplication of commands", topic)); Reporter::warning(fmt("Topic %s was added to NetControl broker plugin twice. Possible duplication of commands", config$topic));
else else
add netcontrol_broker_topics[topic]; add netcontrol_broker_topics[config$topic];
local plugin = broker_plugin; local plugin = broker_plugin;
if ( can_expire ) if ( can_expire )
plugin = broker_plugin_can_expire; plugin = broker_plugin_can_expire;
local p = PluginState($plugin=plugin, $broker_config=BrokerConfig($host=host, $bport=host_port, $topic=topic, $id=netcontrol_broker_current_id)); local p = PluginState($plugin=plugin, $broker_id=netcontrol_broker_current_id, $broker_config=config);
if ( [host_port, cat(host)] in netcontrol_broker_peers ) if ( [config$bport, cat(config$host)] in netcontrol_broker_peers )
Reporter::warning(fmt("Peer %s:%s was added to NetControl broker plugin twice.", host, host_port)); Reporter::warning(fmt("Peer %s:%s was added to NetControl broker plugin twice.", config$host, config$bport));
else else
netcontrol_broker_peers[host_port, cat(host)] = p; netcontrol_broker_peers[config$bport, cat(config$host)] = p;
netcontrol_broker_id[netcontrol_broker_current_id] = p; netcontrol_broker_id[netcontrol_broker_current_id] = p;
++netcontrol_broker_current_id; ++netcontrol_broker_current_id;

2
testing/btest/scripts/base/frameworks/netcontrol/broker.bro
View file

@ -18,7 +18,7 @@ redef exit_only_after_terminate = T;
event NetControl::init() event NetControl::init()
{ {
suspend_processing(); suspend_processing();
local netcontrol_broker = NetControl::create_broker(127.0.0.1, broker_port, "bro/event/netcontroltest", T); local netcontrol_broker = NetControl::create_broker(NetControl::BrokerConfig($host=127.0.0.1, $bport=broker_port, $topic="bro/event/netcontroltest"), T);
NetControl::activate(netcontrol_broker, 0); NetControl::activate(netcontrol_broker, 0);
} }