use crate::file::{File, COMMIT_DATA_ENTRY_SIZE, FAN_LEN, SIGNATURE};use bstr::ByteSlice;use byteorder::{BigEndian, ByteOrder};use filebuffer::FileBuffer;use git_object::SHA1_SIZE;use quick_error::quick_error;use std::{ convert::{TryFrom, TryInto}, ops::Range, path::Path,}; type ChunkId = [u8; 4]; quick_error! { #[derive(Debug)] pub enum Error { BaseGraphMismatch(from_header: u8, from_chunk: u32) { display( "Commit-graph {} chunk contains {} base graphs, but commit-graph file header claims {} base graphs", BASE_GRAPHS_LIST_CHUNK_ID.as_bstr(), from_chunk, from_header, ) } CommitCountMismatch(chunk1_id: ChunkId, chunk1_commits: u32, chunk2_id: ChunkId, chunk2_commits: u32) { display( "Commit-graph {:?} chunk contains {} commits, but {:?} chunk contains {} commits", chunk1_id.as_bstr(), chunk1_commits, chunk2_id.as_bstr(), chunk2_commits, ) } Corrupt(msg: String) { display("{}", msg) } DuplicateChunk(id: ChunkId) { display("Commit-graph file contains multiple {:?} chunks", id.as_bstr()) } // This error case is disabled, as git allows extra garbage in the extra edges list. // ExtraEdgesOverflow { // display("The last entry in commit-graph's extended edges list does is not marked as being terminal") // } InvalidChunkSize(id: ChunkId, msg: String) { display("Commit-graph chunk {:?} has invalid size: {}", id.as_bstr(), msg) } Io(err: std::io::Error, path: std::path::PathBuf) { display("Could not open commit-graph file at '{}'", path.display()) source(err) } MissingChunk(id: ChunkId) { display("Missing required chunk {:?}", id.as_bstr()) } UnsupportedHashVersion(version: u8) { display("Commit-graph file uses unsupported hash version: {}", version) } UnsupportedVersion(version: u8) { display("Unsupported commit-graph file version: {}", version) } }} const CHUNK_LOOKUP_SIZE: usize = 12;const HEADER_LEN: usize = 8;const MIN_FILE_SIZE: usize = HEADER_LEN + ((MIN_CHUNKS + 1) * CHUNK_LOOKUP_SIZE);const OID_LOOKUP_ENTRY_SIZE: usize = SHA1_SIZE; // Required chunks: OIDF, OIDL, CDATconst MIN_CHUNKS: usize = 3;const BASE_GRAPHS_LIST_CHUNK_ID: ChunkId = *b"BASE";const COMMIT_DATA_CHUNK_ID: ChunkId = *b"CDAT";const EXTENDED_EDGES_LIST_CHUNK_ID: ChunkId = *b"EDGE";const OID_FAN_CHUNK_ID: ChunkId = *b"OIDF";const OID_LOOKUP_CHUNK_ID: ChunkId = *b"OIDL";const SENTINEL_CHUNK_ID: ChunkId = [0u8; 4]; impl File { pub fn at(path: impl AsRef<Path>) -> Result<File, Error> { Self::try_from(path.as_ref()) }} impl TryFrom<&Path> for File { type Error = Error; fn try_from(path: &Path) -> Result<Self, Self::Error> { let data = FileBuffer::open(path).map_err(|e| Error::Io(e, path.to_owned()))?; let data_size = data.len(); if data_size < MIN_FILE_SIZE { return Err(Error::Corrupt( "Commit-graph file too small even for an empty graph".to_owned(), )); } let mut ofs = 0; if &data[ofs..ofs + SIGNATURE.len()] != SIGNATURE { return Err(Error::Corrupt( "Commit-graph file does not start with expected signature".to_owned(), )); } ofs += SIGNATURE.len(); match data[ofs] { 1 => (), x => { return Err(Error::UnsupportedVersion(x)); } }; ofs += 1; match data[ofs] { 1 => (), x => { return Err(Error::UnsupportedHashVersion(x)); } }; ofs += 1; let chunk_count = data[ofs]; // Can assert chunk_count >= MIN_CHUNKS here, but later OIDF+OIDL+CDAT presence checks make // it redundant. ofs += 1; let base_graph_count = data[ofs]; ofs += 1; let chunk_lookup_end = ofs + ((chunk_count as usize + 1) * CHUNK_LOOKUP_SIZE); if chunk_lookup_end > data_size { return Err(Error::Corrupt(format!( "Commit-graph file is too small to hold {} chunks", chunk_count ))); } let mut base_graphs_list_offset: Option<usize> = None; let mut commit_data_offset: Option<usize> = None; let mut commit_data_count = 0u32; let mut extra_edges_list_range: Option<Range<usize>> = None; let mut fan_offset: Option<usize> = None; let mut oid_lookup_offset: Option<usize> = None; let mut oid_lookup_count = 0u32; let mut chunk_id: ChunkId = data[ofs..ofs + 4].try_into().expect("ChunkId to accept 4 bytes"); ofs += 4; let mut chunk_offset: usize = BigEndian::read_u64(&data[ofs..ofs + 8]) .try_into() .expect("an offset small enough to fit a usize"); if chunk_offset < chunk_lookup_end { return Err(Error::Corrupt(format!( "Commit-graph chunk 0 has invalid offset {} (must be at least {})", chunk_offset, chunk_lookup_end ))); } ofs += 8; for _ in 0..chunk_count { let next_chunk_id: ChunkId = data[ofs..ofs + 4].try_into().expect("ChunkId to accept 4 bytes"); ofs += 4; let next_chunk_offset: usize = BigEndian::read_u64(&data[ofs..ofs + 8]) .try_into() .expect("an offset small enough to fit a usize"); ofs += 8; let chunk_size: usize = next_chunk_offset .checked_sub(chunk_offset) .ok_or_else(|| Error::InvalidChunkSize(chunk_id, "size is negative".to_string()))?; if next_chunk_offset >= data_size { return Err(Error::InvalidChunkSize( chunk_id, "chunk extends beyond end of file".to_string(), )); } match chunk_id { BASE_GRAPHS_LIST_CHUNK_ID => { if base_graphs_list_offset.is_some() { return Err(Error::DuplicateChunk(chunk_id)); } if chunk_size % SHA1_SIZE != 0 { return Err(Error::InvalidChunkSize( chunk_id, format!("chunk size {} is not a multiple of {}", chunk_size, SHA1_SIZE), )); } let chunk_base_graph_count = (chunk_size / SHA1_SIZE) as u32; if chunk_base_graph_count != base_graph_count as u32 { return Err(Error::BaseGraphMismatch(base_graph_count, chunk_base_graph_count)); } base_graphs_list_offset = Some(chunk_offset); } COMMIT_DATA_CHUNK_ID => { if commit_data_offset.is_some() { return Err(Error::DuplicateChunk(chunk_id)); } if chunk_size % COMMIT_DATA_ENTRY_SIZE != 0 { return Err(Error::InvalidChunkSize( chunk_id, format!( "chunk size {} is not a multiple of {}", chunk_size, COMMIT_DATA_ENTRY_SIZE ), )); } commit_data_offset = Some(chunk_offset); commit_data_count = (chunk_size / COMMIT_DATA_ENTRY_SIZE) as u32; } EXTENDED_EDGES_LIST_CHUNK_ID => { if extra_edges_list_range.is_some() { return Err(Error::DuplicateChunk(chunk_id)); } extra_edges_list_range = Some(Range { start: chunk_offset, end: next_chunk_offset, }) } OID_FAN_CHUNK_ID => { if fan_offset.is_some() { return Err(Error::DuplicateChunk(chunk_id)); } let expected_size = 4 * FAN_LEN; if chunk_size != expected_size { return Err(Error::InvalidChunkSize( chunk_id, format!("expected chunk length {}, got {}", expected_size, chunk_size), )); } fan_offset = Some(chunk_offset); } OID_LOOKUP_CHUNK_ID => { if oid_lookup_offset.is_some() { return Err(Error::DuplicateChunk(chunk_id)); } if chunk_size % OID_LOOKUP_ENTRY_SIZE != 0 { return Err(Error::InvalidChunkSize( chunk_id, format!( "chunk size {} is not a multiple of {}", chunk_size, OID_LOOKUP_ENTRY_SIZE ), )); } oid_lookup_offset = Some(chunk_offset); oid_lookup_count = (chunk_size / OID_LOOKUP_ENTRY_SIZE) as u32; // TODO(ST): Figure out how to handle this. Don't know what to do with the commented code. // git allows extra garbage in the extra edges list chunk? // if oid_lookup_count > 0 { // let last_edge = &data[next_chunk_offset - 4..next_chunk_offset]; // let last_edge = BigEndian::read_u32(last_edge); // if let ExtraEdge::Internal(_) = ExtraEdge::from_raw(last_edge) { // return Err(Error::ExtraEdgesListOverflow); // } // } } _ => {} } chunk_id = next_chunk_id; chunk_offset = next_chunk_offset; } if chunk_id != SENTINEL_CHUNK_ID { return Err(Error::Corrupt(format!( "Commit-graph file has invalid last chunk ID: {:?}", chunk_id.as_bstr() ))); } let fan_offset = fan_offset.ok_or_else(|| Error::MissingChunk(OID_FAN_CHUNK_ID))?; let oid_lookup_offset = oid_lookup_offset.ok_or_else(|| Error::MissingChunk(OID_LOOKUP_CHUNK_ID))?; let commit_data_offset = commit_data_offset.ok_or_else(|| Error::MissingChunk(COMMIT_DATA_CHUNK_ID))?; if base_graph_count > 0 && base_graphs_list_offset == None { return Err(Error::MissingChunk(BASE_GRAPHS_LIST_CHUNK_ID)); } let (fan, _) = read_fan(&data[fan_offset..]); if oid_lookup_count != fan[255] { return Err(Error::CommitCountMismatch( OID_FAN_CHUNK_ID, fan[255], OID_LOOKUP_CHUNK_ID, oid_lookup_count, )); } if commit_data_count != fan[255] { return Err(Error::CommitCountMismatch( OID_FAN_CHUNK_ID, fan[255], COMMIT_DATA_CHUNK_ID, commit_data_count, )); } Ok(File { base_graph_count, base_graphs_list_offset, commit_data_offset, data, extra_edges_list_range, fan, oid_lookup_offset, path: path.to_owned(), }) }} // Copied from git-odb/pack/index/init.rsfn read_fan(d: &[u8]) -> ([u32; FAN_LEN], usize) { let mut fan = [0; FAN_LEN]; for (c, f) in d.chunks(4).zip(fan.iter_mut()) { *f = BigEndian::read_u32(c); } (fan, FAN_LEN * 4)}
use crate::file::{File, COMMIT_DATA_ENTRY_SIZE, FAN_LEN, SIGNATURE};
use bstr::ByteSlice;
use byteorder::{BigEndian, ByteOrder};
use filebuffer::FileBuffer;
use git_object::SHA1_SIZE;
use quick_error::quick_error;
use std::{
convert::{TryFrom, TryInto},
ops::Range,
path::Path,
};
type ChunkId = [u8; 4];
quick_error! {
#[derive(Debug)]
pub enum Error {
BaseGraphMismatch(from_header: u8, from_chunk: u32) {
display(
"Commit-graph {} chunk contains {} base graphs, but commit-graph file header claims {} base graphs",
BASE_GRAPHS_LIST_CHUNK_ID.as_bstr(),
from_chunk,
from_header,
)
}
CommitCountMismatch(chunk1_id: ChunkId, chunk1_commits: u32, chunk2_id: ChunkId, chunk2_commits: u32) {
display(
"Commit-graph {:?} chunk contains {} commits, but {:?} chunk contains {} commits",
chunk1_id.as_bstr(),
chunk1_commits,
chunk2_id.as_bstr(),
chunk2_commits,
)
}
Corrupt(msg: String) {
display("{}", msg)
}
DuplicateChunk(id: ChunkId) {
display("Commit-graph file contains multiple {:?} chunks", id.as_bstr())
}
// This error case is disabled, as git allows extra garbage in the extra edges list.
// ExtraEdgesOverflow {
// display("The last entry in commit-graph's extended edges list does is not marked as being terminal")
// }
InvalidChunkSize(id: ChunkId, msg: String) {
display("Commit-graph chunk {:?} has invalid size: {}", id.as_bstr(), msg)
}
Io(err: std::io::Error, path: std::path::PathBuf) {
display("Could not open commit-graph file at '{}'", path.display())
source(err)
}
MissingChunk(id: ChunkId) {
display("Missing required chunk {:?}", id.as_bstr())
}
UnsupportedHashVersion(version: u8) {
display("Commit-graph file uses unsupported hash version: {}", version)
}
UnsupportedVersion(version: u8) {
display("Unsupported commit-graph file version: {}", version)
}
}
}
const CHUNK_LOOKUP_SIZE: usize = 12;
const HEADER_LEN: usize = 8;
const MIN_FILE_SIZE: usize = HEADER_LEN + ((MIN_CHUNKS + 1) * CHUNK_LOOKUP_SIZE);
const OID_LOOKUP_ENTRY_SIZE: usize = SHA1_SIZE;
// Required chunks: OIDF, OIDL, CDAT
const MIN_CHUNKS: usize = 3;
const BASE_GRAPHS_LIST_CHUNK_ID: ChunkId = *b"BASE";
const COMMIT_DATA_CHUNK_ID: ChunkId = *b"CDAT";
const EXTENDED_EDGES_LIST_CHUNK_ID: ChunkId = *b"EDGE";
const OID_FAN_CHUNK_ID: ChunkId = *b"OIDF";
const OID_LOOKUP_CHUNK_ID: ChunkId = *b"OIDL";
const SENTINEL_CHUNK_ID: ChunkId = [0u8; 4];
impl File {
pub fn at(path: impl AsRef<Path>) -> Result<File, Error> {
Self::try_from(path.as_ref())
}
}
impl TryFrom<&Path> for File {
type Error = Error;
fn try_from(path: &Path) -> Result<Self, Self::Error> {
let data = FileBuffer::open(path).map_err(|e| Error::Io(e, path.to_owned()))?;
let data_size = data.len();
if data_size < MIN_FILE_SIZE {
return Err(Error::Corrupt(
"Commit-graph file too small even for an empty graph".to_owned(),
));
}
let mut ofs = 0;
if &data[ofs..ofs + SIGNATURE.len()] != SIGNATURE {
return Err(Error::Corrupt(
"Commit-graph file does not start with expected signature".to_owned(),
));
}
ofs += SIGNATURE.len();
match data[ofs] {
1 => (),
x => {
return Err(Error::UnsupportedVersion(x));
}
};
ofs += 1;
match data[ofs] {
1 => (),
x => {
return Err(Error::UnsupportedHashVersion(x));
}
};
ofs += 1;
let chunk_count = data[ofs];
// Can assert chunk_count >= MIN_CHUNKS here, but later OIDF+OIDL+CDAT presence checks make
// it redundant.
ofs += 1;
let base_graph_count = data[ofs];
ofs += 1;
let chunk_lookup_end = ofs + ((chunk_count as usize + 1) * CHUNK_LOOKUP_SIZE);
if chunk_lookup_end > data_size {
return Err(Error::Corrupt(format!(
"Commit-graph file is too small to hold {} chunks",
chunk_count
)));
}
let mut base_graphs_list_offset: Option<usize> = None;
let mut commit_data_offset: Option<usize> = None;
let mut commit_data_count = 0u32;
let mut extra_edges_list_range: Option<Range<usize>> = None;
let mut fan_offset: Option<usize> = None;
let mut oid_lookup_offset: Option<usize> = None;
let mut oid_lookup_count = 0u32;
let mut chunk_id: ChunkId = data[ofs..ofs + 4].try_into().expect("ChunkId to accept 4 bytes");
ofs += 4;
let mut chunk_offset: usize = BigEndian::read_u64(&data[ofs..ofs + 8])
.try_into()
.expect("an offset small enough to fit a usize");
if chunk_offset < chunk_lookup_end {
return Err(Error::Corrupt(format!(
"Commit-graph chunk 0 has invalid offset {} (must be at least {})",
chunk_offset, chunk_lookup_end
)));
}
ofs += 8;
for _ in 0..chunk_count {
let next_chunk_id: ChunkId = data[ofs..ofs + 4].try_into().expect("ChunkId to accept 4 bytes");
ofs += 4;
let next_chunk_offset: usize = BigEndian::read_u64(&data[ofs..ofs + 8])
.try_into()
.expect("an offset small enough to fit a usize");
ofs += 8;
let chunk_size: usize = next_chunk_offset
.checked_sub(chunk_offset)
.ok_or_else(|| Error::InvalidChunkSize(chunk_id, "size is negative".to_string()))?;
if next_chunk_offset >= data_size {
return Err(Error::InvalidChunkSize(
chunk_id,
"chunk extends beyond end of file".to_string(),
));
}
match chunk_id {
BASE_GRAPHS_LIST_CHUNK_ID => {
if base_graphs_list_offset.is_some() {
return Err(Error::DuplicateChunk(chunk_id));
}
if chunk_size % SHA1_SIZE != 0 {
return Err(Error::InvalidChunkSize(
chunk_id,
format!("chunk size {} is not a multiple of {}", chunk_size, SHA1_SIZE),
));
}
let chunk_base_graph_count = (chunk_size / SHA1_SIZE) as u32;
if chunk_base_graph_count != base_graph_count as u32 {
return Err(Error::BaseGraphMismatch(base_graph_count, chunk_base_graph_count));
}
base_graphs_list_offset = Some(chunk_offset);
}
COMMIT_DATA_CHUNK_ID => {
if commit_data_offset.is_some() {
return Err(Error::DuplicateChunk(chunk_id));
}
if chunk_size % COMMIT_DATA_ENTRY_SIZE != 0 {
return Err(Error::InvalidChunkSize(
chunk_id,
format!(
"chunk size {} is not a multiple of {}",
chunk_size, COMMIT_DATA_ENTRY_SIZE
),
));
}
commit_data_offset = Some(chunk_offset);
commit_data_count = (chunk_size / COMMIT_DATA_ENTRY_SIZE) as u32;
}
EXTENDED_EDGES_LIST_CHUNK_ID => {
if extra_edges_list_range.is_some() {
return Err(Error::DuplicateChunk(chunk_id));
}
extra_edges_list_range = Some(Range {
start: chunk_offset,
end: next_chunk_offset,
})
}
OID_FAN_CHUNK_ID => {
if fan_offset.is_some() {
return Err(Error::DuplicateChunk(chunk_id));
}
let expected_size = 4 * FAN_LEN;
if chunk_size != expected_size {
return Err(Error::InvalidChunkSize(
chunk_id,
format!("expected chunk length {}, got {}", expected_size, chunk_size),
));
}
fan_offset = Some(chunk_offset);
}
OID_LOOKUP_CHUNK_ID => {
if oid_lookup_offset.is_some() {
return Err(Error::DuplicateChunk(chunk_id));
}
if chunk_size % OID_LOOKUP_ENTRY_SIZE != 0 {
return Err(Error::InvalidChunkSize(
chunk_id,
format!(
"chunk size {} is not a multiple of {}",
chunk_size, OID_LOOKUP_ENTRY_SIZE
),
));
}
oid_lookup_offset = Some(chunk_offset);
oid_lookup_count = (chunk_size / OID_LOOKUP_ENTRY_SIZE) as u32;
// TODO(ST): Figure out how to handle this. Don't know what to do with the commented code.
// git allows extra garbage in the extra edges list chunk?
// if oid_lookup_count > 0 {
// let last_edge = &data[next_chunk_offset - 4..next_chunk_offset];
// let last_edge = BigEndian::read_u32(last_edge);
// if let ExtraEdge::Internal(_) = ExtraEdge::from_raw(last_edge) {
// return Err(Error::ExtraEdgesListOverflow);
// }
// }
}
_ => {}
}
chunk_id = next_chunk_id;
chunk_offset = next_chunk_offset;
}
if chunk_id != SENTINEL_CHUNK_ID {
return Err(Error::Corrupt(format!(
"Commit-graph file has invalid last chunk ID: {:?}",
chunk_id.as_bstr()
)));
}
let fan_offset = fan_offset.ok_or_else(|| Error::MissingChunk(OID_FAN_CHUNK_ID))?;
let oid_lookup_offset = oid_lookup_offset.ok_or_else(|| Error::MissingChunk(OID_LOOKUP_CHUNK_ID))?;
let commit_data_offset = commit_data_offset.ok_or_else(|| Error::MissingChunk(COMMIT_DATA_CHUNK_ID))?;
if base_graph_count > 0 && base_graphs_list_offset == None {
return Err(Error::MissingChunk(BASE_GRAPHS_LIST_CHUNK_ID));
}
let (fan, _) = read_fan(&data[fan_offset..]);
if oid_lookup_count != fan[255] {
return Err(Error::CommitCountMismatch(
OID_FAN_CHUNK_ID,
fan[255],
OID_LOOKUP_CHUNK_ID,
oid_lookup_count,
));
}
if commit_data_count != fan[255] {
return Err(Error::CommitCountMismatch(
OID_FAN_CHUNK_ID,
fan[255],
COMMIT_DATA_CHUNK_ID,
commit_data_count,
));
}
Ok(File {
base_graph_count,
base_graphs_list_offset,
commit_data_offset,
data,
extra_edges_list_range,
fan,
oid_lookup_offset,
path: path.to_owned(),
})
}
}
// Copied from git-odb/pack/index/init.rs
fn read_fan(d: &[u8]) -> ([u32; FAN_LEN], usize) {
let mut fan = [0; FAN_LEN];
for (c, f) in d.chunks(4).zip(fan.iter_mut()) {
*f = BigEndian::read_u32(c);
}
(fan, FAN_LEN * 4)
}