136 lines
3.3 KiB
Rust
136 lines
3.3 KiB
Rust
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)
|
|
}
|
|
|
|
//-----------------------------------------------
|