diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 17:39:49 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 17:39:49 +0000 |
commit | a0aa2307322cd47bbf416810ac0292925e03be87 (patch) | |
tree | 37076262a026c4b48c8a0e84f44ff9187556ca35 /rust/src/filecontainer.rs | |
parent | Initial commit. (diff) | |
download | suricata-a0aa2307322cd47bbf416810ac0292925e03be87.tar.xz suricata-a0aa2307322cd47bbf416810ac0292925e03be87.zip |
Adding upstream version 1:7.0.3.upstream/1%7.0.3
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'rust/src/filecontainer.rs')
-rw-r--r-- | rust/src/filecontainer.rs | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/rust/src/filecontainer.rs b/rust/src/filecontainer.rs new file mode 100644 index 0000000..3a8bde5 --- /dev/null +++ b/rust/src/filecontainer.rs @@ -0,0 +1,105 @@ +/* Copyright (C) 2017-2021 Open Information Security Foundation + * + * You can copy, redistribute or modify this Program under the terms of + * the GNU General Public License version 2 as published by the Free + * Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * version 2 along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + */ + +//! This module handles file container operations (open, append, close). + +use std::ptr; +use std::os::raw::{c_void}; + +use crate::core::*; + +// Defined in util-file.h +extern { + pub fn FileFlowFlagsToFlags(flow_file_flags: u16, flags: u8) -> u16; +} + +#[repr(C)] +#[derive(Debug)] +pub struct FileContainer { + head: * mut c_void, + tail: * mut c_void, +} + +impl Default for FileContainer { + fn default() -> Self { Self { + head: ptr::null_mut(), + tail: ptr::null_mut(), + }} +} + +impl FileContainer { + pub fn free(&mut self, cfg: &'static SuricataFileContext) { + SCLogDebug!("freeing self"); + if let Some(c) = unsafe {SC} { + (c.FileContainerRecycle)(self, cfg.files_sbcfg); + } + } + + pub fn file_open(&mut self, cfg: &'static SuricataFileContext, track_id: u32, name: &[u8], flags: u16) -> i32 { + match unsafe {SC} { + None => panic!("BUG no suricata_config"), + Some(c) => { + SCLogDebug!("FILE {:p} OPEN flags {:04X}", &self, flags); + + let res = (c.FileOpenFile)(self, cfg.files_sbcfg, track_id, + name.as_ptr(), name.len() as u16, + ptr::null(), 0u32, flags); + res + } + } + } + + pub fn file_append(&mut self, cfg: &'static SuricataFileContext, track_id: &u32, data: &[u8], is_gap: bool) -> i32 { + SCLogDebug!("FILECONTAINER: append {}", data.len()); + if data.is_empty() { + return 0 + } + match unsafe {SC} { + None => panic!("BUG no suricata_config"), + Some(c) => { + let res = match is_gap { + false => { + SCLogDebug!("appending file data"); + let r = (c.FileAppendData)(self, cfg.files_sbcfg, *track_id, + data.as_ptr(), data.len() as u32); + r + }, + true => { + SCLogDebug!("appending GAP"); + let r = (c.FileAppendGAP)(self, cfg.files_sbcfg, *track_id, + data.as_ptr(), data.len() as u32); + r + }, + }; + res + } + } + } + + pub fn file_close(&mut self, cfg: &'static SuricataFileContext, track_id: &u32, flags: u16) -> i32 { + SCLogDebug!("FILECONTAINER: CLOSEing"); + + match unsafe {SC} { + None => panic!("BUG no suricata_config"), + Some(c) => { + let res = (c.FileCloseFile)(self, cfg.files_sbcfg, *track_id, ptr::null(), 0u32, flags); + res + } + } + + } +} |