use crate::internal::streamname::{
self, DIGITAL_SIGNATURE_STREAM_NAME, DOCUMENT_SUMMARY_INFO_STREAM_NAME,
MSI_DIGITAL_SIGNATURE_EX_STREAM_NAME, SUMMARY_INFO_STREAM_NAME,
};
use cfb;
use std::io::{self, Read, Seek, SeekFrom, Write};
pub struct StreamReader<F> {
stream: cfb::Stream<F>,
}
impl<F> StreamReader<F> {
pub(crate) fn new(stream: cfb::Stream<F>) -> StreamReader<F> {
StreamReader { stream }
}
}
impl<F: Read + Seek> Read for StreamReader<F> {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
self.stream.read(buf)
}
}
impl<F: Read + Seek> Seek for StreamReader<F> {
fn seek(&mut self, from: SeekFrom) -> io::Result<u64> {
self.stream.seek(from)
}
}
pub struct StreamWriter<F> {
stream: cfb::Stream<F>,
}
impl<F> StreamWriter<F> {
pub(crate) fn new(stream: cfb::Stream<F>) -> StreamWriter<F> {
StreamWriter { stream }
}
}
impl<F: Read + Seek + Write> Write for StreamWriter<F> {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
self.stream.write(buf)
}
fn flush(&mut self) -> io::Result<()> {
self.stream.flush()
}
}
impl<F: Read + Seek + Write> Seek for StreamWriter<F> {
fn seek(&mut self, from: SeekFrom) -> io::Result<u64> {
self.stream.seek(from)
}
}
pub struct Streams<'a, F: 'a> {
entries: cfb::Entries<'a, F>,
}
impl<'a, F: 'a> Streams<'a, F> {
pub(crate) fn new(entries: cfb::Entries<'a, F>) -> Streams<'a, F> {
Streams { entries }
}
}
impl<'a, F: 'a> Iterator for Streams<'a, F> {
type Item = String;
fn next(&mut self) -> Option<String> {
loop {
let entry = self.entries.next()?;
if !entry.is_stream()
|| entry.name() == DIGITAL_SIGNATURE_STREAM_NAME
|| entry.name() == MSI_DIGITAL_SIGNATURE_EX_STREAM_NAME
|| entry.name() == SUMMARY_INFO_STREAM_NAME
|| entry.name() == DOCUMENT_SUMMARY_INFO_STREAM_NAME
{
continue;
}
let (name, is_table) = streamname::decode(entry.name());
if !is_table {
return Some(name);
}
}
}
}