thin-provisioning-tools/tests/common/thin.rs

use anyhow::Result;
use std::path::PathBuf;

use thinp::file_utils;
use thinp::io_engine::*;

use crate::common::fixture::*;
use crate::common::process::*;
use crate::common::target::*;
use crate::common::test_dir::TestDir;
use crate::common::thin_xml_generator::{write_xml, SingleThinS};

//-----------------------------------------------

pub fn mk_valid_xml(td: &mut TestDir) -> Result<PathBuf> {
    let xml = td.mk_path("meta.xml");
    let mut gen = SingleThinS::new(0, 1024, 2048, 2048);
    write_xml(&xml, &mut gen)?;
    Ok(xml)
}

pub fn mk_valid_md(td: &mut TestDir) -> Result<PathBuf> {
    let xml = td.mk_path("meta.xml");
    let md = td.mk_path("meta.bin");

    let mut gen = SingleThinS::new(0, 1024, 20480, 20480);
    write_xml(&xml, &mut gen)?;

    let _file = file_utils::create_sized_file(&md, 4096 * 4096);
    let args = ["-i", xml.to_str().unwrap(), "-o", md.to_str().unwrap()];
    run_ok(THIN_RESTORE, &args)?;

    Ok(md)
}

//-----------------------------------------------

// FIXME: replace mk_valid_md with this?
pub fn prep_metadata(td: &mut TestDir) -> Result<PathBuf> {
    let md = mk_zeroed_md(td)?;
    let args = [
        "-o",
        md.to_str().unwrap(),
        "--format",
        "--nr-data-blocks",
        "102400",
    ];
    run_ok(THIN_GENERATE_METADATA, &args)?;

    // Create a 2GB device
    let args = ["-o", md.to_str().unwrap(), "--create-thin", "1"];
    run_ok(THIN_GENERATE_METADATA, &args)?;
    let args = [
        "-o",
        md.to_str().unwrap(),
        "--dev-id",
        "1",
        "--size",
        "2097152",
        "--rw=randwrite",
        "--seq-nr=16",
    ];
    run_ok(THIN_GENERATE_MAPPINGS, &args)?;

    // Take a few snapshots.
    let mut snap_id = 2;
    for _i in 0..10 {
        // take a snapshot
        let args = [
            "-o",
            md.to_str().unwrap(),
            "--create-snap",
            &snap_id.to_string(),
            "--origin",
            "1",
        ];
        run_ok(THIN_GENERATE_METADATA, &args)?;

        // partially overwrite the origin (64MB)
        let args = [
            "-o",
            md.to_str().unwrap(),
            "--dev-id",
            "1",
            "--size",
            "2097152",
            "--io-size",
            "131072",
            "--rw=randwrite",
            "--seq-nr=16",
        ];
        run_ok(THIN_GENERATE_MAPPINGS, &args)?;
        snap_id += 1;
    }

    Ok(md)
}

pub fn set_needs_check(md: &PathBuf) -> Result<()> {
    let args = ["-o", md.to_str().unwrap(), "--set-needs-check"];
    run_ok(THIN_GENERATE_METADATA, &args)?;
    Ok(())
}

pub fn generate_metadata_leaks(
    md: &PathBuf,
    nr_blocks: u64,
    expected: u32,
    actual: u32,
) -> Result<()> {
    let args = [
        "-o",
        md.to_str().unwrap(),
        "--create-metadata-leaks",
        "--nr-blocks",
        &nr_blocks.to_string(),
        "--expected",
        &expected.to_string(),
        "--actual",
        &actual.to_string(),
    ];
    run_ok(THIN_GENERATE_DAMAGE, &args)?;

    Ok(())
}

pub fn get_needs_check(md: &PathBuf) -> Result<bool> {
    use thinp::thin::superblock::*;

    let engine = SyncIoEngine::new(&md, 1, false)?;
    let sb = read_superblock(&engine, SUPERBLOCK_LOCATION)?;
    Ok(sb.flags.needs_check)
}

//-----------------------------------------------