proxmox-backup/src/tape/helpers/emulate_tape_reader.rs

use std::io::Read;

use proxmox::tools::io::ReadExt;

use crate::tape::{
    BlockRead,
    BlockReadStatus,
    file_formats::PROXMOX_TAPE_BLOCK_SIZE,
};

/// Emulate tape read behavior on a normal Reader
///
/// Tapes reads are always return one whole block PROXMOX_TAPE_BLOCK_SIZE.
pub struct EmulateTapeReader<R: Read> {
    reader: R,
    got_eof: bool,
}

impl <R: Read> EmulateTapeReader<R> {

    pub fn new(reader: R) -> Self {
        Self { reader, got_eof: false }
    }
}

impl <R: Read> BlockRead for EmulateTapeReader<R> {
    fn read_block(&mut self, buffer: &mut [u8]) -> Result<BlockReadStatus, std::io::Error> {
        if self.got_eof {
            proxmox::io_bail!("detected read after EOF!");
        }
        match self.reader.read_exact_or_eof(buffer)? {
            false => {
                self.got_eof = true;
                Ok(BlockReadStatus::EndOfFile)
            }
            true => {
                // test buffer len after EOF test (to allow EOF test with small buffers in BufferedReader)
                if buffer.len() != PROXMOX_TAPE_BLOCK_SIZE {
                    proxmox::io_bail!("EmulateTapeReader: read_block with wrong block size ({} != {})",
                                      buffer.len(), PROXMOX_TAPE_BLOCK_SIZE);
                }
                Ok(BlockReadStatus::Ok(buffer.len()))
            }
        }
    }
}