use std::path::Path;
use crate::Result;
use crate::extract::{FiveTuple, FiveTupleKey};
use crate::pcap::PcapFlowSource;
use crate::session::SessionEvent;
use crate::smb::session::SmbParser;
use crate::smb::types::SmbMessage;
pub fn messages_from_pcap<P: AsRef<Path>>(
path: P,
) -> Result<impl Iterator<Item = (FiveTupleKey, SmbMessage)>> {
let source = PcapFlowSource::open(path)?;
Ok(source
.sessions(FiveTuple::bidirectional(), SmbParser::default())
.filter_map(|evt| evt.ok())
.filter_map(|evt| match evt {
SessionEvent::Application { key, message, .. } => Some((key, message)),
_ => None,
}))
}