mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 06:38:20 +00:00
47500ceef4
Fixed most @load dependency issues in the process. The test is still failing in a "known" way due to hot.conn.bro and scan.bro. Adressess #545
60 lines
1.7 KiB
Text
60 lines
1.7 KiB
Text
##! Extracts the items from HTTP traffic, one per file. At this time only
|
|
##! the message body from the server can be extracted with this script.
|
|
|
|
@load ./main
|
|
@load ./file-ident
|
|
@load base/utils/files
|
|
|
|
module HTTP;
|
|
|
|
export {
|
|
## Pattern of file mime types to extract from HTTP entity bodies.
|
|
const extract_file_types = /NO_DEFAULT/ &redef;
|
|
|
|
## The on-disk prefix for files to be extracted from HTTP entity bodies.
|
|
const extraction_prefix = "http-item" &redef;
|
|
|
|
redef record Info += {
|
|
## This field can be set per-connection to determine if the entity body
|
|
## will be extracted. It must be set to T on or before the first
|
|
## entity_body_data event.
|
|
extracting_file: bool &default=F;
|
|
|
|
## This is the holder for the file handle as the file is being written
|
|
## to disk.
|
|
extraction_file: file &log &optional;
|
|
};
|
|
|
|
redef record State += {
|
|
entity_bodies: count &default=0;
|
|
};
|
|
}
|
|
|
|
event http_entity_data(c: connection, is_orig: bool, length: count, data: string) &priority=5
|
|
{
|
|
# Client body extraction is not currently supported in this script.
|
|
if ( is_orig || ! c$http$first_chunk ) return;
|
|
|
|
if ( c$http$first_chunk )
|
|
{
|
|
if ( c$http?$mime_type &&
|
|
extract_file_types in c$http$mime_type )
|
|
{
|
|
c$http$extracting_file = T;
|
|
local suffix = fmt("%s_%d.dat", is_orig ? "orig" : "resp", ++c$http_state$entity_bodies);
|
|
local fname = generate_extraction_filename(extraction_prefix, c, suffix);
|
|
|
|
c$http$extraction_file = open(fname);
|
|
enable_raw_output(c$http$extraction_file);
|
|
}
|
|
}
|
|
|
|
if ( c$http$extracting_file )
|
|
print c$http$extraction_file, data;
|
|
}
|
|
|
|
event http_end_entity(c: connection, is_orig: bool)
|
|
{
|
|
if ( c$http$extracting_file )
|
|
close(c$http$extraction_file);
|
|
}
|