oozextract 0.4.1

use crate::algorithm::Algorithm;
use crate::core::error::{ErrorContext, Res, ResultBuilder, WithContext};
use crate::core::pointer::{Pointer, PointerDest};
use crate::core::{pointer, Core};
use std::collections::VecDeque;

#[derive(Debug)]
pub struct Mermaid;

impl Algorithm for Mermaid {
    fn process(
        &self,
        core: &mut Core,
        mode: usize,
        src: Pointer<{ PointerDest::INPUT }>,
        src_used: usize,
        dst_start: Pointer<{ PointerDest::OUTPUT }>,
        dst: Pointer<{ PointerDest::OUTPUT }>,
        dst_size: usize,
    ) -> Res<()> {
        let offset = dst - dst_start;
        let mut lz = MermaidLzTable::default();
        lz.read_lz_table(core, mode, src, src + src_used, dst, dst_size, offset)?;
        lz.process_lz_runs(core, mode, src + src_used, dst, dst_size, offset)
    }
}

#[derive(Default, Copy, Clone)]
enum Chunk {
    #[default]
    Stream1,
    Stream2,
}

/// Mermaid/Selkie decompression also happens in two phases, just like in Kraken,
/// but the match copier works differently.
/// Both Mermaid and Selkie use the same on-disk format, only the compressor
/// differs.
#[derive(Default)]
struct MermaidLzTable {
    // Flag stream. Format of flags:
    // Read flagbyte from |cmd_stream|
    // If flagbyte >= 24:
    //   flagbyte & 0x80 == 0 : Read from |off16_stream| into |recent_offs|.
    //                   != 0 : Don't read offset.
    //   flagbyte & 7 = Number of literals to copy first from |lit_stream|.
    //   (flagbyte >> 3) & 0xF = Number of bytes to copy from |recent_offs|.
    //
    //  If flagbyte == 0 :
    //    Read byte L from |length_stream|
    //    If L > 251: L += 4 * Read word from |length_stream|
    //    L += 64
    //    Copy L bytes from |lit_stream|.
    //
    //  If flagbyte == 1 :
    //    Read byte L from |length_stream|
    //    If L > 251: L += 4 * Read word from |length_stream|
    //    L += 91
    //    Copy L bytes from match pointed by next offset from |off16_stream|
    //
    //  If flagbyte == 2 :
    //    Read byte L from |length_stream|
    //    If L > 251: L += 4 * Read word from |length_stream|
    //    L += 29
    //    Copy L bytes from match pointed by next offset from |off32_stream|,
    //    relative to start of block.
    //    Then prefetch |off32_stream[3]|
    //
    //  If flagbyte > 2:
    //    L = flagbyte + 5
    //    Copy L bytes from match pointed by next offset from |off32_stream|,
    //    relative to start of block.
    //    Then prefetch |off32_stream[3]|
    cmd_stream: Pointer<{ PointerDest::TEMP }>,
    cmd_stream_end: Pointer<{ PointerDest::TEMP }>,

    /// Length stream
    length_stream: Pointer<{ PointerDest::INPUT }>,

    /// Literal stream
    lit_stream: Pointer<{ PointerDest::TEMP }>,
    lit_stream_end: Pointer<{ PointerDest::TEMP }>,

    /// Near offsets
    off16_stream: VecDeque<u16>,

    /// Stream selector for current chunk
    off32_stream: Chunk,

    /// Holds the offsets for the two chunks
    off32_stream_1: Vec<u32>,
    off32_stream_2: Vec<u32>,

    /// Flag offsets for next 64k chunk.
    cmd_stream_2_offs: usize,
    cmd_stream_2_offs_end: usize,
}

impl ErrorContext for MermaidLzTable {}

impl MermaidLzTable {
    pub(crate) fn process_lz_runs(
        &mut self,
        core: &mut Core,
        mode: usize,
        src_end: Pointer<{ PointerDest::INPUT }>,
        mut dst: Pointer<{ PointerDest::OUTPUT }>,
        mut dst_size: usize,
        offset: usize,
    ) -> Res<()> {
        let mut saved_dist = -8;

        for iteration in 0..2 {
            let mut dst_size_cur = dst_size;
            if dst_size_cur > 0x10000 {
                dst_size_cur = 0x10000;
            }

            if iteration == 0 {
                self.off32_stream = Chunk::Stream1;
                self.cmd_stream_end = self.cmd_stream + self.cmd_stream_2_offs;
            } else {
                self.off32_stream = Chunk::Stream2;
                self.cmd_stream_end = self.cmd_stream + self.cmd_stream_2_offs_end;
                self.cmd_stream += self.cmd_stream_2_offs;
            }
            let startoff = if (offset == 0) && (iteration == 0) {
                8
            } else {
                0
            };

            if mode == 0 {
                self.process::<true>(core, dst, dst_size_cur, src_end, &mut saved_dist, startoff)
                    .at(self)?;
            } else {
                self.process::<false>(core, dst, dst_size_cur, src_end, &mut saved_dist, startoff)
                    .at(self)?;
            }
            assert!(!self.length_stream.is_null());

            dst += dst_size_cur;
            dst_size -= dst_size_cur;
            if dst_size == 0 {
                break;
            }
        }

        self.assert_eq(self.length_stream, src_end)?;
        Ok(())
    }

    fn off32(&self) -> &Vec<u32> {
        match self.off32_stream {
            Chunk::Stream1 => &self.off32_stream_1,
            Chunk::Stream2 => &self.off32_stream_2,
        }
    }

    fn process<const ADD_MODE: bool>(
        &mut self,
        core: &mut Core,
        mut dst: Pointer<{ PointerDest::OUTPUT }>,
        dst_size: usize,
        src_end: Pointer<{ PointerDest::INPUT }>,
        saved_dist: &mut i32,
        startoff: i32,
    ) -> Res<()> {
        let dst_end = dst + dst_size;
        let mut cmd_stream = self.cmd_stream;
        let cmd_stream_end = self.cmd_stream_end;
        let mut length_stream = self.length_stream;
        let mut lit_stream = self.lit_stream;
        let lit_stream_end = self.lit_stream_end;
        let mut off32_stream = 0;
        let off32_stream_end = self.off32().len();
        let mut recent_offs = *saved_dist;
        let mut offs_ptr;
        let mut length;
        let dst_begin = dst;

        dst += startoff;

        while cmd_stream < cmd_stream_end {
            let cmd = core.get_byte(cmd_stream).at(self)? as usize;
            cmd_stream += 1;
            if cmd >= 24 {
                let litlen = cmd & 7;
                if ADD_MODE {
                    core.copy_64_add(dst, lit_stream, dst + recent_offs, litlen)
                        .at(self)?;
                } else {
                    core.repeat_copy_64(dst, lit_stream, litlen).at(self)?;
                }
                dst += litlen;
                lit_stream += litlen;
                if (cmd >> 7) == 0 {
                    recent_offs = -(self.off16_stream.pop_front().unwrap() as i32);
                }
                offs_ptr = dst + recent_offs;
                core.repeat_copy_64(dst, offs_ptr, (cmd >> 3) & 0xF)
                    .at(self)?;
                dst += (cmd >> 3) & 0xF;
            } else if cmd > 2 {
                length = cmd + 5;

                assert_ne!(off32_stream, off32_stream_end);
                offs_ptr = dst_begin - self.off32()[off32_stream];
                off32_stream += 1;
                recent_offs = offs_ptr.index as i32 - dst.index as i32;

                assert!((dst_end - dst) >= length);
                core.repeat_copy_64(dst, offs_ptr, length).at(self)?;
                dst += length;
                if let Some(&offset) = self.off32().get(off32_stream + 3) {
                    core.prefetch(dst_begin - offset);
                }
            } else if cmd == 0 {
                self.assert_lt(length_stream, src_end)?;
                length = core.get_byte(length_stream).at(self)? as usize;
                if length > 251 {
                    assert!((src_end - length_stream) >= 3);
                    length += core.get_le_bytes(length_stream + 1, 2).at(core)? * 4;
                    length_stream += 2;
                }
                length_stream += 1;

                length += 64;
                assert!((dst_end - dst) >= length);
                assert!((lit_stream_end - lit_stream) >= length);
                if ADD_MODE {
                    core.copy_64_add(dst, lit_stream, dst + recent_offs, length)
                        .at(self)?;
                } else {
                    core.repeat_copy_64(dst, lit_stream, length).at(self)?;
                }
                dst += length;
                lit_stream += length;
            } else if cmd == 1 {
                self.assert_lt(length_stream, src_end)?;
                length = core.get_byte(length_stream).at(self)? as usize;
                if length > 251 {
                    assert!((src_end - length_stream) >= 3);
                    length += core.get_le_bytes(length_stream + 1, 2).at(core)? * 4;
                    length_stream += 2;
                }
                length_stream += 1;
                length += 91;

                offs_ptr = dst
                    - self
                        .off16_stream
                        .pop_front()
                        .msg_of(&"offset_stream_empty")? as usize;
                recent_offs = offs_ptr.index as i32 - dst.index as i32;
                core.repeat_copy_64(dst, offs_ptr, length).at(self)?;
                dst += length;
            } else {
                /* flag == 2 */
                self.assert_lt(length_stream, src_end)?;
                length = core.get_byte(length_stream).at(self)? as usize;
                if length > 251 {
                    assert!((src_end - length_stream) >= 3);
                    length += core.get_le_bytes(length_stream + 1, 2).at(core)? * 4;
                    length_stream += 2;
                }
                length_stream += 1;
                length += 29;
                assert_ne!(off32_stream, off32_stream_end);
                offs_ptr = dst_begin - self.off32()[off32_stream];
                off32_stream += 1;
                recent_offs = offs_ptr.index as i32 - dst.index as i32;
                core.repeat_copy_64(dst, offs_ptr, length).at(self)?;
                dst += length;
                if let Some(&offset) = self.off32().get(off32_stream + 3) {
                    core.prefetch(dst_begin - offset);
                }
            }
        }

        length = dst_end - dst;
        if ADD_MODE {
            core.copy_64_add(dst, lit_stream, dst + recent_offs, length)
                .at(self)?;
        } else {
            core.repeat_copy_64(dst, lit_stream, length).at(self)?;
        }
        lit_stream += length;

        *saved_dist = recent_offs;
        self.length_stream = length_stream;
        self.lit_stream = lit_stream;
        Ok(())
    }
}

impl MermaidLzTable {
    fn read_lz_table(
        &mut self,
        core: &mut Core,
        mode: usize,
        mut src: Pointer<{ PointerDest::INPUT }>,
        src_end: Pointer<{ PointerDest::INPUT }>,
        mut dst: Pointer<{ PointerDest::OUTPUT }>,
        dst_size: usize,
        offset: usize,
    ) -> Res<()> {
        let mut out;
        let mut decode_count = 0;
        let mut off32_size_2;
        let mut off32_size_1;
        let mut scratch = pointer::tmp(0);

        assert!(mode <= 1, "{}", mode);
        assert!((src_end - src) >= 10);

        if offset == 0 {
            core.copy_bytes(dst, src, 8).at(self)?;
            dst += 8;
            src += 8;
        }

        // Decode lit stream
        out = scratch;
        src += core
            .decode_bytes(
                &mut out,
                src,
                src_end,
                &mut decode_count,
                dst_size,
                false,
                pointer::scratch(0),
            )
            .at(self)?;
        self.lit_stream = out;
        self.lit_stream_end = out + decode_count;
        scratch += decode_count;

        // Decode flag stream
        out = scratch;
        src += core
            .decode_bytes(
                &mut out,
                src,
                src_end,
                &mut decode_count,
                dst_size,
                false,
                pointer::scratch(0),
            )
            .at(self)?;
        self.cmd_stream = out;
        self.cmd_stream_end = out + decode_count;
        scratch += decode_count;

        self.cmd_stream_2_offs_end = decode_count;
        if dst_size <= 0x10000 {
            self.cmd_stream_2_offs = decode_count;
        } else {
            assert!((src_end - src) >= 2);
            self.cmd_stream_2_offs = core.get_le_bytes(src, 2).at(core)?;
            src += 2;
            assert!(self.cmd_stream_2_offs <= self.cmd_stream_2_offs_end);
        }

        assert!((src_end - src) >= 2);

        let off16_count = core.get_le_bytes(src, 2).at(core)?;
        src += 2;
        if off16_count == 0xffff {
            // off16 is entropy coded
            let mut off16_lo;
            let mut off16_hi;
            let mut off16_lo_count = 0;
            let mut off16_hi_count = 0;
            off16_hi = scratch;
            src += core
                .decode_bytes(
                    &mut off16_hi,
                    src,
                    src_end,
                    &mut off16_hi_count,
                    dst_size >> 1,
                    false,
                    pointer::scratch(0),
                )
                .at(self)?;
            scratch += off16_hi_count;

            off16_lo = scratch;
            src += core
                .decode_bytes(
                    &mut off16_lo,
                    src,
                    src_end,
                    &mut off16_lo_count,
                    dst_size >> 1,
                    false,
                    pointer::scratch(0),
                )
                .at(self)?;
            scratch += off16_lo_count;

            self.assert_eq(off16_lo_count, off16_hi_count)?;
            self.off16_stream.reserve(off16_lo_count);
            for i in 0..off16_lo_count {
                let off16 = core.get_byte(off16_lo + i).at(self)? as u16
                    + core.get_byte(off16_hi + i).at(self)? as u16 * 256;
                self.off16_stream.push_back(off16)
            }
        } else {
            self.off16_stream = core
                .get_slice(src, off16_count * 2)
                .at(self)?
                .chunks(2)
                .map(|c| u16::from_le_bytes(c.try_into().unwrap()))
                .collect();
            src += off16_count * 2;
        }

        assert!((src_end - src) >= 3);
        let tmp = core.get_le_bytes(src, 3).at(core)?;
        src += 3;

        if tmp != 0 {
            off32_size_1 = tmp >> 12;
            off32_size_2 = tmp & 0xFFF;
            if off32_size_1 == 4095 {
                assert!((src_end - src) >= 2);
                off32_size_1 = core.get_le_bytes(src, 2).at(core)?;
                src += 2;
            }
            if off32_size_2 == 4095 {
                assert!((src_end - src) >= 2);
                off32_size_2 = core.get_le_bytes(src, 2).at(core)?;
                src += 2;
            }

            self.off32_stream_1.reserve(off32_size_1);
            self.off32_stream_2.reserve(off32_size_2);

            src += self
                .decode_far_offsets(core, src, src_end, true, off32_size_1, offset)
                .at(self)?;

            src += self
                .decode_far_offsets(core, src, src_end, false, off32_size_2, offset + 0x10000)
                .at(self)?;
        }
        self.length_stream = src;

        Ok(())
    }

    fn decode_far_offsets(
        &mut self,
        core: &mut Core,
        src: Pointer<{ PointerDest::INPUT }>,
        src_end: Pointer<{ PointerDest::INPUT }>,
        stream1: bool,
        output_size: usize,
        offset: usize,
    ) -> Res<usize> {
        let mut src_cur = src;

        if offset < (0xC00000 - 1) {
            for _ in 0..output_size {
                assert!((src_end - src_cur) >= 3);
                let off = core.get_le_bytes(src_cur, 3).at(core)?;
                src_cur += 3;
                assert!(off <= offset);
                if stream1 {
                    self.off32_stream_1.push(off as u32)
                } else {
                    self.off32_stream_2.push(off as u32)
                }
            }
            Ok(src_cur - src)
        } else {
            for _ in 0..output_size {
                assert!((src_end - src_cur) >= 3);
                let mut off = core.get_le_bytes(src_cur, 3).at(core)?;
                src_cur += 3;

                if off >= 0xc00000 {
                    assert_ne!(src_cur, src_end);
                    off += (core.get_byte(src_cur).at(self)? as usize) << 22;
                    src_cur += 1;
                }
                assert!(off <= offset);
                if stream1 {
                    self.off32_stream_1.push(off as u32)
                } else {
                    self.off32_stream_2.push(off as u32)
                }
            }
            Ok(src_cur - src)
        }
    }
}