nodedb-wal 0.6.2

// SPDX-License-Identifier: Apache-2.0

//! WAL recovery: scan an existing WAL file to determine the last committed LSN
//! and file offset, enabling safe reopening for continued writes.
//!
//! ## Recovery algorithm
//!
//! 1. Open the WAL file read-only.
//! 2. Scan forward, validating each record (magic, checksum).
//! 3. Stop at first corruption.
//! 4. Classify that stop: an unfsynced torn tail bounds the committed prefix,
//!    but damage with intact higher-LSN records behind it is a hole and fails
//!    recovery rather than silently dropping those records.
//! 5. Return the last valid LSN and the byte offset past the last valid record.
//!
//! ## Invariants
//!
//! - Recovery is deterministic: same file → same result.
//! - Recovery is idempotent: running twice gives the same answer.
//! - Truncated/torn writes are not errors — they're the boundary of committed data.
//! - Corruption that hides committed records IS an error — see [`crate::torn_tail`].

use std::path::Path;

use crate::error::Result;
use crate::reader::WalReader;

/// Result of scanning a WAL file for recovery.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct RecoveryInfo {
    /// Last valid LSN found in the WAL. 0 if the WAL is empty.
    pub last_lsn: u64,

    /// Number of valid records found.
    pub record_count: u64,

    /// Byte offset past the last valid record (where new writes should begin).
    pub end_offset: u64,
}

impl RecoveryInfo {
    /// The next LSN to assign for new writes.
    pub fn next_lsn(&self) -> u64 {
        self.last_lsn + 1
    }
}

/// Scan a WAL file and recover the committed prefix.
///
/// Returns `RecoveryInfo` describing the state of the WAL, or an error
/// if the file cannot be opened.
pub fn recover(path: &Path) -> Result<RecoveryInfo> {
    if !path.exists() {
        return Ok(RecoveryInfo {
            last_lsn: 0,
            record_count: 0,
            end_offset: 0,
        });
    }

    // Structural scan only: recovery reads framing (LSN, lengths, offsets) and
    // never a payload. It runs from `WalWriter::open`, which resumes a segment
    // for appending and holds no key ring, so demanding keys here would make an
    // encrypted WAL impossible to reopen for writing.
    let mut reader = WalReader::open_raw(path)?;
    let mut last_lsn = 0u64;
    let mut record_count = 0u64;

    loop {
        match reader.next_record() {
            Ok(Some(record)) => {
                last_lsn = record.header.lsn;
                record_count += 1;
            }
            Ok(None) => {
                // End of committed prefix (EOF or corruption).
                break;
            }
            // Nothing past an unknown required record can be read, and a version
            // gap is the one error that needs the segment it was found in: the
            // reader sees one header at a time and holds no path. Every other
            // error passes through unchanged.
            Err(error) => return Err(error.with_segment_path(path)),
        }
    }

    // Refuse to call a hole "the end of the log".
    crate::torn_tail::verify_committed_prefix(path, reader.stop_reason(), last_lsn)?;

    Ok(RecoveryInfo {
        last_lsn,
        record_count,
        // The reader's own accounting is authoritative: it covers alignment
        // padding that trails the final record and torn records rebuilt from
        // the double-write buffer, neither of which can be reconstructed from
        // the returned records' headers.
        end_offset: reader.committed_end(),
    })
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::error::WalError;
    use crate::record::RecordType;
    use crate::writer::WalWriter;

    #[test]
    fn recover_empty_wal() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("empty.wal");

        // Create empty WAL.
        {
            let mut writer = WalWriter::open_without_direct_io(&path).unwrap();
            writer.sync().unwrap();
        }

        let info = recover(&path).unwrap();
        assert_eq!(info.last_lsn, 0);
        assert_eq!(info.record_count, 0);
        assert_eq!(info.next_lsn(), 1);
    }

    #[test]
    fn recover_nonexistent_file() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("nonexistent.wal");

        let info = recover(&path).unwrap();
        assert_eq!(info.last_lsn, 0);
        assert_eq!(info.record_count, 0);
    }

    #[test]
    fn recover_with_records() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("test.wal");

        {
            let mut writer = WalWriter::open_without_direct_io(&path).unwrap();
            writer
                .append(RecordType::Put as u32, 1, 0, 0, b"first")
                .unwrap();
            writer
                .append(RecordType::Put as u32, 1, 0, 0, b"second")
                .unwrap();
            writer
                .append(RecordType::Delete as u32, 2, 1, 0, b"third")
                .unwrap();
            writer.sync().unwrap();
        }

        let info = recover(&path).unwrap();
        assert_eq!(info.last_lsn, 3);
        assert_eq!(info.record_count, 3);
        assert_eq!(info.next_lsn(), 4);
        assert!(info.end_offset > 0);
    }

    #[test]
    fn recover_truncated_wal() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("truncated.wal");

        {
            let mut writer = WalWriter::open_without_direct_io(&path).unwrap();
            writer
                .append(RecordType::Put as u32, 1, 0, 0, b"good")
                .unwrap();
            writer
                .append(RecordType::Put as u32, 1, 0, 0, b"also-good")
                .unwrap();
            writer.sync().unwrap();
        }

        // Append garbage (simulating torn write).
        {
            use std::io::Write;
            let mut file = std::fs::OpenOptions::new()
                .append(true)
                .open(&path)
                .unwrap();
            file.write_all(b"GARBAGE_TORN_WRITE_PARTIAL").unwrap();
        }

        let info = recover(&path).unwrap();
        assert_eq!(info.last_lsn, 2);
        assert_eq!(info.record_count, 2);
        assert_eq!(info.next_lsn(), 3);
    }

    /// Where each of `count` records starts, and the file's bytes.
    fn segment_with_records(path: &Path, count: u8) -> (Vec<u64>, Vec<u8>) {
        {
            let mut writer = WalWriter::open_without_direct_io(path).unwrap();
            for i in 0..count {
                writer
                    .append(RecordType::Put as u32, 1, 0, 0, &[i; 8])
                    .unwrap();
            }
            writer.sync().unwrap();
        }
        let mut reader = WalReader::open_raw(path).unwrap();
        let mut starts = Vec::new();
        loop {
            let start = reader.committed_end();
            if reader.next_record().unwrap().is_none() {
                break;
            }
            starts.push(start);
        }
        (starts, std::fs::read(path).unwrap())
    }

    #[test]
    fn a_lone_first_record_in_another_format_is_a_format_gap() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("lone.wal");
        let (starts, mut bytes) = segment_with_records(&path, 1);
        let at = usize::try_from(starts[0]).unwrap();
        crate::record::foreign_version::reseal_with_version(
            &mut bytes,
            at,
            crate::record::WAL_FORMAT_VERSION + 1,
        );
        std::fs::write(&path, &bytes).unwrap();

        match recover(&path) {
            Err(WalError::SegmentFormatVersion {
                path: named,
                version,
                supported,
            }) => {
                assert_eq!(named, path.display().to_string());
                assert_eq!(version, crate::record::WAL_FORMAT_VERSION + 1);
                assert_eq!(supported, crate::record::WAL_FORMAT_VERSION);
            }
            other => panic!("expected SegmentFormatVersion, got {other:?}"),
        }
    }

    /// A flipped bit in the first record's version field is damage to this
    /// build's record, not another build's format. The intact records behind
    /// it make the damage a hole, so recovery refuses it as mid-file
    /// corruption instead of reporting a format gap.
    #[test]
    fn a_flipped_first_version_bit_is_damage_not_a_format_gap() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("flipped.wal");
        let (starts, mut bytes) = segment_with_records(&path, 3);
        // The version field is bytes 4..6 of a record header.
        let at = usize::try_from(starts[0]).unwrap() + 4;
        let flipped = crate::record::WAL_FORMAT_VERSION ^ 0x0100;
        bytes[at..at + 2].copy_from_slice(&flipped.to_le_bytes());
        std::fs::write(&path, &bytes).unwrap();

        match recover(&path) {
            Err(WalError::MidFileCorruption {
                offset, resync_lsn, ..
            }) => {
                assert_eq!(offset, starts[0]);
                assert_eq!(resync_lsn, 2);
            }
            other => panic!("expected MidFileCorruption, got {other:?}"),
        }
    }
}