mirror of
https://github.com/zeek/zeek.git
synced 2025-10-17 14:08:20 +00:00
35 lines
851 B
Text
35 lines
851 B
Text
# Source this script in addition to protocols/conn/community-id
|
|
# to add Community ID to notices.
|
|
|
|
# Only support loading this if the main script is also loaded.
|
|
@load base/protocols/conn
|
|
@load base/frameworks/notice
|
|
|
|
@ifdef ( CommunityID::seed )
|
|
|
|
module CommunityID::Notice;
|
|
|
|
export {
|
|
# Turn notice support on/off at runtime. When disabled,
|
|
# this still leaves the `community_id` string in the notice
|
|
# log, just unset.
|
|
option enabled: bool = T;
|
|
|
|
redef record Notice::Info += {
|
|
community_id: string &optional &log;
|
|
};
|
|
}
|
|
|
|
hook Notice::notice(n: Notice::Info)
|
|
{
|
|
if ( CommunityID::Notice::enabled && n?$conn && n$conn?$conn )
|
|
{
|
|
local info = n$conn$conn;
|
|
# This is set during new_connection(), so it should
|
|
# always be there, but better safe than sorry.
|
|
if ( info?$community_id )
|
|
n$community_id = info$community_id;
|
|
}
|
|
}
|
|
|
|
@endif
|