diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/bindings.rs | 13 | ||||
| -rw-r--r-- | src/lib.rs | 16 | ||||
| -rw-r--r-- | src/main.rs | 55 |
3 files changed, 59 insertions, 25 deletions
diff --git a/src/bindings.rs b/src/bindings.rs index 515caf0..6a5a03f 100644 --- a/src/bindings.rs +++ b/src/bindings.rs @@ -31,10 +31,11 @@ impl<T> ::std::fmt::Debug for __IncompleteArrayField<T> { } } -pub type UChar = ::std::os::raw::c_uchar; -pub type UShort = ::std::os::raw::c_ushort; -pub type UInt = ::std::os::raw::c_uint; -pub type ULong = ::std::os::raw::c_ulong; +pub type CUchar = ::std::os::raw::c_uchar; +pub type CUShort = ::std::os::raw::c_ushort; +pub type CUInt = ::std::os::raw::c_uint; +pub type CULong = ::std::os::raw::c_ulong; +pub type CLongLong = ::std::os::raw::c_longlong; extern "C" { pub fn ntohl(__netlong: u32) -> u32; @@ -79,7 +80,7 @@ fn bindgen_test_layout_maat_hit_path() { concat!("Alignment of ", stringify!(maat_hit_path)) ); assert_eq!( - unsafe { ::std::ptr::addr_of!((*ptr).Nth_scan) as usize - ptr as usize }, + unsafe { ::std::ptr::addr_of!((*ptr).nth_scan) as usize - ptr as usize }, 0usize, concat!( "Offset of field: ", @@ -683,4 +684,4 @@ extern "C" { instance: *mut maat, group: *mut maat_hit_group, ) -> ::std::os::raw::c_int; -}
\ No newline at end of file +} diff --git a/src/lib.rs b/src/lib.rs deleted file mode 100644 index 84f3eb8..0000000 --- a/src/lib.rs +++ /dev/null @@ -1,16 +0,0 @@ -pub mod bindings; - -pub fn add(left: usize, right: usize) -> usize { - left + right -} - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn it_works() { - let result = add(2, 2); - assert_eq!(result, 4); - } -} diff --git a/src/main.rs b/src/main.rs index ed9ff0a..5cba62d 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,7 +1,56 @@ -use maat_rust_binding::bindings; +pub mod bindings; + +use crate::bindings::MAAT_SCAN_HIT; +use std::ffi::CString; + +const ARRAY_SIZE: usize = 10; + +pub fn maat_hs_string_scan() { + let redis_ip = CString::new("127.0.0.1").unwrap(); + let redis_port = 6379; + let redis_db = 0; + let log_path = CString::new("./maat_framework_gtest.log").unwrap(); + let log_level = bindings::LOG_LEVEL_INFO; + let table_info_path = CString::new("./table_info.conf").unwrap(); -fn main() { unsafe { - let _opts = bindings::maat_options_new(); + let opts = bindings::maat_options_new(); + bindings::maat_options_set_redis(opts, redis_ip.as_ptr(), redis_port, redis_db); + bindings::maat_options_set_logger(opts, log_path.as_ptr(), log_level); + + let maat_inst = bindings::maat_new(opts, table_info_path.as_ptr()); + bindings::maat_options_free(opts); + + let table_name = CString::new("HTTP_URL").unwrap(); + let table_id = bindings::maat_get_table_id(maat_inst, table_name.as_ptr()); + assert_ne!(table_id, 0); + + let mut results: [bindings::CLongLong; ARRAY_SIZE] = [-1; ARRAY_SIZE]; + let mut n_hit_result: usize = 0; + let thread_id = 0; + let state = bindings::maat_state_new(maat_inst, thread_id); + let scan_data = CString::new("http://www.cyberessays.com/search_results.php?action=search&query=username,abckkk,1234567").unwrap(); + + let n_hit_result_ptr = &mut n_hit_result as *mut usize; + + let ret = bindings::maat_scan_string( + maat_inst, + table_id, + scan_data.as_ptr(), + scan_data.as_bytes().len(), + results.as_mut_ptr(), + ARRAY_SIZE, + n_hit_result_ptr, + state, + ); + + assert_eq!(ret, MAAT_SCAN_HIT); + assert_eq!(n_hit_result, 1); + assert_eq!(results[0], 125); + bindings::maat_state_free(state); + println!("hit compile_id:{}", results[0]); } } +fn main() { + crate::maat_hs_string_scan(); +} |