rust: replace helper with built-in

Replace custom u32_as_bytes with built-in to_be_bytes

Issue: 5785
pull/15570/head
William Ling 6 months ago committed by Victor Julien
parent e7edc2a1e2
commit 30628e45bc

@ -697,14 +697,6 @@ impl SMBTree {
}
}
pub fn u32_as_bytes(i: u32) -> [u8;4] {
let o1: u8 = ((i >> 24) & 0xff) as u8;
let o2: u8 = ((i >> 16) & 0xff) as u8;
let o3: u8 = ((i >> 8) & 0xff) as u8;
let o4: u8 = (i & 0xff) as u8;
return [o1, o2, o3, o4]
}
#[derive(Debug)]
pub struct SMBState<> {
pub state_data: AppLayerStateData,

@ -303,7 +303,7 @@ fn smb1_request_record_one(state: &mut SMBState, r: &SmbRecord, command: u8, and
let tx_hdr = SMBCommonHdr::from1(r, SMBHDR_TYPE_GENERICTX);
let mut frankenfid = pd.fid.to_vec();
frankenfid.extend_from_slice(&u32_as_bytes(r.ssn_id));
frankenfid.extend_from_slice(&r.ssn_id.to_be_bytes());
let filename = match state.guid2name_cache.get(&frankenfid) {
Some(n) => n.to_vec(),
@ -338,7 +338,7 @@ fn smb1_request_record_one(state: &mut SMBState, r: &SmbRecord, command: u8, and
newname.retain(|&i|i != 0x00);
let mut frankenfid = pd.fid.to_vec();
frankenfid.extend_from_slice(&u32_as_bytes(r.ssn_id));
frankenfid.extend_from_slice(&r.ssn_id.to_be_bytes());
let oldname = match state.guid2name_cache.get(&frankenfid) {
Some(n) => n.to_vec(),
@ -403,7 +403,7 @@ fn smb1_request_record_one(state: &mut SMBState, r: &SmbRecord, command: u8, and
// store read fid,offset in map
let fid_key = SMBCommonHdr::from1(r, SMBHDR_TYPE_OFFSET);
let mut fid = rr.fid.to_vec();
fid.extend_from_slice(&u32_as_bytes(r.ssn_id));
fid.extend_from_slice(&r.ssn_id.to_be_bytes());
let fidoff = SMBFileGUIDOffset::new(fid, rr.offset);
state.read_offset_cache.put(fid_key, fidoff);
},
@ -533,7 +533,7 @@ fn smb1_request_record_one(state: &mut SMBState, r: &SmbRecord, command: u8, and
match parse_smb1_close_request_record(r.data) {
Ok((_, cd)) => {
let mut fid = cd.fid.to_vec();
fid.extend_from_slice(&u32_as_bytes(r.ssn_id));
fid.extend_from_slice(&r.ssn_id.to_be_bytes());
let _name = state.guid2name_cache.pop(&fid);
state.ssn2vec_cache.put(SMBCommonHdr::from1(r, SMBHDR_TYPE_GUID), fid.to_vec());
@ -728,7 +728,7 @@ fn smb1_response_record_one(state: &mut SMBState, r: &SmbRecord, command: u8, an
p.retain(|&i|i != 0x00);
let mut fid = cr.fid.to_vec();
fid.extend_from_slice(&u32_as_bytes(r.ssn_id));
fid.extend_from_slice(&r.ssn_id.to_be_bytes());
SCLogDebug!("SMB1_COMMAND_NT_CREATE_ANDX fid {:?}", fid);
SCLogDebug!("fid {:?} name {:?}", fid, p);
_ = state.guid2name_cache.put(fid, p);
@ -854,7 +854,7 @@ pub fn smb1_trans_request_record(state: &mut SMBState, r: &SmbRecord)
pipe.fid.to_vec());
let mut frankenfid = pipe.fid.to_vec();
frankenfid.extend_from_slice(&u32_as_bytes(r.ssn_id));
frankenfid.extend_from_slice(&r.ssn_id.to_be_bytes());
let (_filename, is_dcerpc) = state.get_service_for_guid(&frankenfid);
@ -891,7 +891,7 @@ pub fn smb1_trans_response_record(state: &mut SMBState, r: &SmbRecord)
SCLogDebug!("FID {:?}", fid);
let mut frankenfid = fid.to_vec();
frankenfid.extend_from_slice(&u32_as_bytes(r.ssn_id));
frankenfid.extend_from_slice(&r.ssn_id.to_be_bytes());
let (_filename, is_dcerpc) = state.get_service_for_guid(&frankenfid);
@ -943,7 +943,7 @@ pub fn smb1_write_request_record(state: &mut SMBState, r: &SmbRecord, andx_offse
return;
}
let mut file_fid = rd.fid.to_vec();
file_fid.extend_from_slice(&u32_as_bytes(r.ssn_id));
file_fid.extend_from_slice(&r.ssn_id.to_be_bytes());
SCLogDebug!("SMBv1 WRITE: FID {:?} offset {}",
file_fid, rd.offset);

Loading…
Cancel
Save