use std::{ fs, io::Read, path::Path, sync::{atomic::AtomicBool, Arc},}; use anyhow::{anyhow, Result};use git_repository::{ hash::ObjectId, object, objs, odb, odb::{loose, pack, Write}, Progress,};use quick_error::quick_error; #[derive(PartialEq, Debug)]pub enum SafetyCheck { SkipFileChecksumVerification, SkipFileAndObjectChecksumVerification, SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError, All,} impl Default for SafetyCheck { fn default() -> Self { SafetyCheck::All }} impl SafetyCheck { pub fn variants() -> &'static [&'static str] { &[ "all", "skip-file-checksum", "skip-file-and-object-checksum", "skip-file-and-object-checksum-and-no-abort-on-decode", ] }} impl std::str::FromStr for SafetyCheck { type Err = String; fn from_str(s: &str) -> Result<Self, Self::Err> { Ok(match s { "skip-file-checksum" => SafetyCheck::SkipFileChecksumVerification, "skip-file-and-object-checksum" => SafetyCheck::SkipFileAndObjectChecksumVerification, "skip-file-and-object-checksum-and-no-abort-on-decode" => { SafetyCheck::SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError } "all" => SafetyCheck::All, _ => return Err(format!("Unknown value for safety check: '{}'", s)), }) }} impl From<SafetyCheck> for pack::index::traverse::SafetyCheck { fn from(v: SafetyCheck) -> Self { use pack::index::traverse::SafetyCheck::*; match v { SafetyCheck::All => All, SafetyCheck::SkipFileChecksumVerification => SkipFileChecksumVerification, SafetyCheck::SkipFileAndObjectChecksumVerification => SkipFileAndObjectChecksumVerification, SafetyCheck::SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError => { SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError } } }} quick_error! { #[derive(Debug)] enum Error { Io(err: std::io::Error) { display("An IO error occurred while writing an object") source(err) from() } OdbWrite(err: loose::write::Error) { display("An object could not be written to the database") source(err) from() } Write(err: Box<dyn std::error::Error + Send + Sync>, kind: object::Kind, id: ObjectId) { display("Failed to write {} object {}", kind, id) source(&**err) } Verify(err: objs::data::verify::Error) { display("Object didn't verify after right after writing it") source(err) from() } ObjectEncodeMismatch(kind: object::Kind, actual: ObjectId, expected: ObjectId) { display("{} object {} wasn't re-encoded without change - new hash is {}", kind, expected, actual) } WrittenFileMissing(id: ObjectId) { display("The recently written file for loose object {} could not be found", id) } WrittenFileCorrupt(err: loose::find::Error, id: ObjectId) { display("The recently written file for loose object {} cold not be read", id) source(err) } }} #[allow(clippy::large_enum_variant)]enum OutputWriter { Loose(loose::Store), Sink(odb::Sink),} impl git_repository::odb::Write for OutputWriter { type Error = Error; fn write_buf(&self, kind: object::Kind, from: &[u8]) -> Result<ObjectId, Self::Error> { match self { OutputWriter::Loose(db) => db.write_buf(kind, from).map_err(Into::into), OutputWriter::Sink(db) => db.write_buf(kind, from).map_err(Into::into), } } fn write_stream(&self, kind: object::Kind, size: u64, from: impl Read) -> Result<ObjectId, Self::Error> { match self { OutputWriter::Loose(db) => db.write_stream(kind, size, from).map_err(Into::into), OutputWriter::Sink(db) => db.write_stream(kind, size, from).map_err(Into::into), } }} impl OutputWriter { fn new(path: Option<impl AsRef<Path>>, compress: bool, object_hash: git_repository::hash::Kind) -> Self { match path { Some(path) => OutputWriter::Loose(loose::Store::at(path.as_ref(), object_hash)), None => OutputWriter::Sink(odb::sink(object_hash).compress(compress)), } }} #[derive(Default)]pub struct Context { pub thread_limit: Option<usize>, pub delete_pack: bool, pub sink_compress: bool, pub verify: bool, pub should_interrupt: Arc<AtomicBool>, pub object_hash: git_repository::hash::Kind,} pub fn pack_or_pack_index( pack_path: impl AsRef<Path>, object_path: Option<impl AsRef<Path>>, check: SafetyCheck, progress: impl Progress, Context { thread_limit, delete_pack, sink_compress, verify, should_interrupt, object_hash, }: Context,) -> Result<()> { use anyhow::Context; let path = pack_path.as_ref(); let bundle = pack::Bundle::at(path, object_hash).with_context(|| { format!( "Could not find .idx or .pack file from given file at '{}'", path.display() ) })?; if !object_path.as_ref().map(|p| p.as_ref().is_dir()).unwrap_or(true) { return Err(anyhow!( "The object directory at '{}' is inaccessible", object_path .expect("path present if no directory on disk") .as_ref() .display() )); } let algorithm = object_path .as_ref() .map(|_| pack::index::traverse::Algorithm::Lookup) .unwrap_or_else(|| { if sink_compress { pack::index::traverse::Algorithm::Lookup } else { pack::index::traverse::Algorithm::DeltaTreeLookup } }); let pack::index::traverse::Outcome{mut progress, ..} = bundle .index .traverse( &bundle.pack, progress, &should_interrupt, { let object_path = object_path.map(|p| p.as_ref().to_owned()); move || { let out = OutputWriter::new(object_path.clone(), sink_compress, object_hash); let loose_odb = verify.then(|| object_path.as_ref().map(|path| loose::Store::at(path, object_hash))).flatten(); let mut read_buf = Vec::new(); move |object_kind, buf, index_entry, progress| { let written_id = out.write_buf(object_kind, buf).map_err(|err| { Error::Write( Box::new(err) as Box<dyn std::error::Error + Send + Sync>, object_kind, index_entry.oid, ) })?; if written_id != index_entry.oid { if let object::Kind::Tree = object_kind { progress.info(format!( "The tree in pack named {} was written as {} due to modes 100664 and 100640 rewritten as 100644.", index_entry.oid, written_id )); } else { return Err(Error::ObjectEncodeMismatch(object_kind, index_entry.oid, written_id)); } } if let Some(verifier) = loose_odb.as_ref() { let obj = verifier .try_find(written_id, &mut read_buf) .map_err(|err| Error::WrittenFileCorrupt(err, written_id))? .ok_or(Error::WrittenFileMissing(written_id))?; obj.verify_checksum(written_id)?; } Ok(()) } } }, pack::index::traverse::Options { traversal: algorithm, thread_limit, check: check.into(), make_pack_lookup_cache: pack::cache::lru::StaticLinkedList::<64>::default, }, ) .with_context(|| "Failed to explode the entire pack - some loose objects may have been created nonetheless")?; let (index_path, data_path) = (bundle.index.path().to_owned(), bundle.pack.path().to_owned()); drop(bundle); if delete_pack { fs::remove_file(&index_path) .and_then(|_| fs::remove_file(&data_path)) .with_context(|| { format!( "Failed to delete pack index file at '{} or data file at '{}'", index_path.display(), data_path.display() ) })?; progress.info(format!( "Removed '{}' and '{}'", index_path.display(), data_path.display() )); } Ok(())}
use std::{
fs,
io::Read,
path::Path,
sync::{atomic::AtomicBool, Arc},
};
use anyhow::{anyhow, Result};
use git_repository::{
hash::ObjectId,
object, objs, odb,
odb::{loose, pack, Write},
Progress,
};
use quick_error::quick_error;
#[derive(PartialEq, Debug)]
pub enum SafetyCheck {
SkipFileChecksumVerification,
SkipFileAndObjectChecksumVerification,
SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError,
All,
}
impl Default for SafetyCheck {
fn default() -> Self {
SafetyCheck::All
}
}
impl SafetyCheck {
pub fn variants() -> &'static [&'static str] {
&[
"all",
"skip-file-checksum",
"skip-file-and-object-checksum",
"skip-file-and-object-checksum-and-no-abort-on-decode",
]
}
}
impl std::str::FromStr for SafetyCheck {
type Err = String;
fn from_str(s: &str) -> Result<Self, Self::Err> {
Ok(match s {
"skip-file-checksum" => SafetyCheck::SkipFileChecksumVerification,
"skip-file-and-object-checksum" => SafetyCheck::SkipFileAndObjectChecksumVerification,
"skip-file-and-object-checksum-and-no-abort-on-decode" => {
SafetyCheck::SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError
}
"all" => SafetyCheck::All,
_ => return Err(format!("Unknown value for safety check: '{}'", s)),
})
}
}
impl From<SafetyCheck> for pack::index::traverse::SafetyCheck {
fn from(v: SafetyCheck) -> Self {
use pack::index::traverse::SafetyCheck::*;
match v {
SafetyCheck::All => All,
SafetyCheck::SkipFileChecksumVerification => SkipFileChecksumVerification,
SafetyCheck::SkipFileAndObjectChecksumVerification => SkipFileAndObjectChecksumVerification,
SafetyCheck::SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError => {
SkipFileAndObjectChecksumVerificationAndNoAbortOnDecodeError
}
}
}
}
quick_error! {
#[derive(Debug)]
enum Error {
Io(err: std::io::Error) {
display("An IO error occurred while writing an object")
source(err)
from()
}
OdbWrite(err: loose::write::Error) {
display("An object could not be written to the database")
source(err)
from()
}
Write(err: Box<dyn std::error::Error + Send + Sync>, kind: object::Kind, id: ObjectId) {
display("Failed to write {} object {}", kind, id)
source(&**err)
}
Verify(err: objs::data::verify::Error) {
display("Object didn't verify after right after writing it")
source(err)
from()
}
ObjectEncodeMismatch(kind: object::Kind, actual: ObjectId, expected: ObjectId) {
display("{} object {} wasn't re-encoded without change - new hash is {}", kind, expected, actual)
}
WrittenFileMissing(id: ObjectId) {
display("The recently written file for loose object {} could not be found", id)
}
WrittenFileCorrupt(err: loose::find::Error, id: ObjectId) {
display("The recently written file for loose object {} cold not be read", id)
source(err)
}
}
}
#[allow(clippy::large_enum_variant)]
enum OutputWriter {
Loose(loose::Store),
Sink(odb::Sink),
}
impl git_repository::odb::Write for OutputWriter {
type Error = Error;
fn write_buf(&self, kind: object::Kind, from: &[u8]) -> Result<ObjectId, Self::Error> {
match self {
OutputWriter::Loose(db) => db.write_buf(kind, from).map_err(Into::into),
OutputWriter::Sink(db) => db.write_buf(kind, from).map_err(Into::into),
}
}
fn write_stream(&self, kind: object::Kind, size: u64, from: impl Read) -> Result<ObjectId, Self::Error> {
match self {
OutputWriter::Loose(db) => db.write_stream(kind, size, from).map_err(Into::into),
OutputWriter::Sink(db) => db.write_stream(kind, size, from).map_err(Into::into),
}
}
}
impl OutputWriter {
fn new(path: Option<impl AsRef<Path>>, compress: bool, object_hash: git_repository::hash::Kind) -> Self {
match path {
Some(path) => OutputWriter::Loose(loose::Store::at(path.as_ref(), object_hash)),
None => OutputWriter::Sink(odb::sink(object_hash).compress(compress)),
}
}
}
#[derive(Default)]
pub struct Context {
pub thread_limit: Option<usize>,
pub delete_pack: bool,
pub sink_compress: bool,
pub verify: bool,
pub should_interrupt: Arc<AtomicBool>,
pub object_hash: git_repository::hash::Kind,
}
pub fn pack_or_pack_index(
pack_path: impl AsRef<Path>,
object_path: Option<impl AsRef<Path>>,
check: SafetyCheck,
progress: impl Progress,
Context {
thread_limit,
delete_pack,
sink_compress,
verify,
should_interrupt,
object_hash,
}: Context,
) -> Result<()> {
use anyhow::Context;
let path = pack_path.as_ref();
let bundle = pack::Bundle::at(path, object_hash).with_context(|| {
format!(
"Could not find .idx or .pack file from given file at '{}'",
path.display()
)
})?;
if !object_path.as_ref().map(|p| p.as_ref().is_dir()).unwrap_or(true) {
return Err(anyhow!(
"The object directory at '{}' is inaccessible",
object_path
.expect("path present if no directory on disk")
.as_ref()
.display()
));
}
let algorithm = object_path
.as_ref()
.map(|_| pack::index::traverse::Algorithm::Lookup)
.unwrap_or_else(|| {
if sink_compress {
pack::index::traverse::Algorithm::Lookup
} else {
pack::index::traverse::Algorithm::DeltaTreeLookup
}
});
let pack::index::traverse::Outcome{mut progress, ..} = bundle
.index
.traverse(
&bundle.pack,
progress,
&should_interrupt,
{
let object_path = object_path.map(|p| p.as_ref().to_owned());
move || {
let out = OutputWriter::new(object_path.clone(), sink_compress, object_hash);
let loose_odb = verify.then(|| object_path.as_ref().map(|path| loose::Store::at(path, object_hash))).flatten();
let mut read_buf = Vec::new();
move |object_kind, buf, index_entry, progress| {
let written_id = out.write_buf(object_kind, buf).map_err(|err| {
Error::Write(
Box::new(err) as Box<dyn std::error::Error + Send + Sync>,
object_kind,
index_entry.oid,
)
})?;
if written_id != index_entry.oid {
if let object::Kind::Tree = object_kind {
progress.info(format!(
"The tree in pack named {} was written as {} due to modes 100664 and 100640 rewritten as 100644.",
index_entry.oid, written_id
));
} else {
return Err(Error::ObjectEncodeMismatch(object_kind, index_entry.oid, written_id));
}
}
if let Some(verifier) = loose_odb.as_ref() {
let obj = verifier
.try_find(written_id, &mut read_buf)
.map_err(|err| Error::WrittenFileCorrupt(err, written_id))?
.ok_or(Error::WrittenFileMissing(written_id))?;
obj.verify_checksum(written_id)?;
}
Ok(())
}
}
},
pack::index::traverse::Options {
traversal: algorithm,
thread_limit,
check: check.into(),
make_pack_lookup_cache: pack::cache::lru::StaticLinkedList::<64>::default,
},
)
.with_context(|| "Failed to explode the entire pack - some loose objects may have been created nonetheless")?;
let (index_path, data_path) = (bundle.index.path().to_owned(), bundle.pack.path().to_owned());
drop(bundle);
if delete_pack {
fs::remove_file(&index_path)
.and_then(|_| fs::remove_file(&data_path))
.with_context(|| {
format!(
"Failed to delete pack index file at '{} or data file at '{}'",
index_path.display(),
data_path.display()
)
})?;
progress.info(format!(
"Removed '{}' and '{}'",
index_path.display(),
data_path.display()
));
}
Ok(())
}