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.
49 lines
1.4 KiB
Text
49 lines
1.4 KiB
Text
@load ./main
|
|
@load base/utils/conn-ids
|
|
@load base/frameworks/file-analysis/main
|
|
|
|
module FTP;
|
|
|
|
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 FTP.
|
|
const disable_default_file_handle_provider: bool = F &redef;
|
|
|
|
## Default file handle provider for FTP.
|
|
function get_file_handle(c: connection, is_orig: bool): string
|
|
{
|
|
if ( [c$id$resp_h, c$id$resp_p] !in ftp_data_expected ) return "";
|
|
|
|
local info: FTP::Info = ftp_data_expected[c$id$resp_h, c$id$resp_p];
|
|
|
|
if ( info$passive )
|
|
# FTP client initiates data channel.
|
|
if ( is_orig )
|
|
# Don't care about FTP client data.
|
|
return "";
|
|
else
|
|
# Do care about FTP server data.
|
|
return cat(ANALYZER_FTP_DATA, " ", c$start_time, " ",
|
|
id_string(c$id));
|
|
else
|
|
# FTP server initiates dta channel.
|
|
if ( is_orig )
|
|
# Do care about FTP server data.
|
|
return cat(ANALYZER_FTP_DATA, " ", c$start_time, " ",
|
|
id_string(c$id));
|
|
else
|
|
# Don't care about FTP client data.
|
|
return "";
|
|
}
|
|
}
|
|
|
|
module GLOBAL;
|
|
|
|
event get_file_handle(tag: AnalyzerTag, c: connection, is_orig: bool)
|
|
{
|
|
if ( tag != ANALYZER_FTP_DATA ) return;
|
|
if ( FTP::disable_default_file_handle_provider ) return;
|
|
return_file_handle(FTP::get_file_handle(c, is_orig));
|
|
}
|