mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 14:48:21 +00:00
fc267d010d
When a file handle is needed and the last event in the queue is also a get_file_handle event with the same arguments, instead of queueing a new event, just remember to cache/re-use the resulting handle from the previous event. This depends on get_file_handle handlers not changing global state that is also used to derive the file handle string.
32 lines
930 B
Text
32 lines
930 B
Text
@load ./main
|
|
@load ./entities
|
|
@load base/utils/conn-ids
|
|
@load base/frameworks/file-analysis/main
|
|
|
|
module SMTP;
|
|
|
|
export {
|
|
## Determines whether the default :bro:see:`get_file_handle` handler
|
|
## is used to return file handles to the file analysis framework.
|
|
## Redefine to true in order to provide a custom handler which overrides
|
|
## the default for SMTP.
|
|
const disable_default_file_handle_provider: bool = F &redef;
|
|
|
|
## Default file handle provider for SMTP.
|
|
function get_file_handle(c: connection, is_orig: bool): string
|
|
{
|
|
if ( ! c?$smtp ) return "";
|
|
|
|
return cat(ANALYZER_SMTP, " ", c$start_time, " ",
|
|
c$smtp$trans_depth, " ", c$smtp_state$mime_level);
|
|
}
|
|
}
|
|
|
|
module GLOBAL;
|
|
|
|
event get_file_handle(tag: AnalyzerTag, c: connection, is_orig: bool)
|
|
{
|
|
if ( tag != ANALYZER_SMTP ) return;
|
|
if ( SMTP::disable_default_file_handle_provider ) return;
|
|
return_file_handle(SMTP::get_file_handle(c, is_orig));
|
|
}
|