mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 22:58:20 +00:00

This reverts commit fc267d010d
.
There were some diffs caused by this in external test suites I'm
unsure about, I'm going to go over optimizations more closely in
a different branch.
50 lines
1.4 KiB
Text
50 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];
|
|
|
|
local rval = fmt("%s %s %s", ANALYZER_FTP_DATA, c$start_time,
|
|
id_string(c$id));
|
|
|
|
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 rval;
|
|
else
|
|
# FTP server initiates dta channel.
|
|
if ( is_orig )
|
|
# Do care about FTP server data.
|
|
return rval;
|
|
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));
|
|
}
|