1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22

use crate::Result;
use nom::bytes::complete::{take};
use crate::smb1::SMB1_ID;
use crate::smb2::SMB2_ID;
use std::convert::TryInto;

#[derive(Debug)]
pub enum SmbVersion {
    V1,
    V2,
    Unknown
}

pub fn check_smb_version(raw: &[u8]) -> Result<SmbVersion> {
    let magic = take(4usize)(raw)?.1;

    return Ok(match magic.try_into().unwrap() {
        SMB1_ID => SmbVersion::V1,
        SMB2_ID => SmbVersion::V2,
        _ => SmbVersion::Unknown
    });
}