aboutsummaryrefslogtreecommitdiffstats
path: root/src/drive.rs
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2017-11-04 16:22:01 -0700
committerBryan Newbold <bnewbold@robocracy.org>2017-11-04 16:22:01 -0700
commitbd80b08771ed8f25a445d146f066a12feca450c3 (patch)
tree9a77af4161c8ed34f146fb5b33be53531032d191 /src/drive.rs
parent9296dffe28f178c3611d4379d41876fcd507b1c8 (diff)
downloadgeniza-bd80b08771ed8f25a445d146f066a12feca450c3.tar.gz
geniza-bd80b08771ed8f25a445d146f066a12feca450c3.zip
more drive progress
Diffstat (limited to 'src/drive.rs')
-rw-r--r--src/drive.rs162
1 files changed, 148 insertions, 14 deletions
diff --git a/src/drive.rs b/src/drive.rs
index 73c0744..622e0ed 100644
--- a/src/drive.rs
+++ b/src/drive.rs
@@ -1,6 +1,8 @@
use std::io::Read;
use std::path::{Path, PathBuf};
+use std::os::unix::fs::MetadataExt;
+use std::fs::File;
use protobuf::Message;
use protobuf::parse_from_bytes;
use integer_encoding::VarInt;
@@ -74,12 +76,48 @@ fn decode_children(raw: &[u8]) -> Result<Vec<Vec<u64>>> {
Ok(children)
}
+fn encode_children(children: &Vec<Vec<u64>>) -> Result<Vec<u8>> {
+ // Use of encode_var_vec() instead of encode_var() here is sort of lazy
+
+ let mut buf = vec![];
+ for subvec in children {
+ let mut subvec = subvec.clone();
+ buf.append(&mut subvec.len().encode_var_vec());
+ subvec.sort_unstable();
+ let mut last = 0;
+ for val in subvec {
+ let run: u64 = val - last;
+ buf.append(&mut run.encode_var_vec());
+ last = val;
+ }
+ }
+ Ok(buf)
+}
+
impl<'a> DatDrive {
+ fn broken_find_file(&mut self, path: &Path) -> Result<Option<DriveEntry>> {
+ for i in (0..self.entry_count()?).rev() {
+ let de = self.get_dir_entry(i)?;
+ if de.path == path {
+ if de.stat.is_none() {
+ return Ok(None);
+ } else {
+ return Ok(Some(de));
+ }
+ }
+ }
+ Ok(None)
+ }
+
+ /// Returns number of drive metadata entries (not including the first entry, which is the
+ /// content register public key)
fn entry_count(&mut self) -> Result<u64> {
Ok(self.metadata.len()? - 1)
}
+ /// Entry index is counted by drive entries (not including the first register entry, which is
+ /// the content register public key)
fn get_dir_entry(&mut self, entry_index: u64) -> Result<DriveEntry> {
trace!("fetching drive entry {} (of {})", entry_index, self.entry_count()?);
let data = self.metadata.get_data_entry(entry_index+1)?;
@@ -102,14 +140,14 @@ impl<'a> DatDrive {
fn get_nearest<P: AsRef<Path>>(&mut self, _path: P) -> Result<DriveEntry> {
// 0. if register is empty, bail out early
let len = self.entry_count()?;
- if len <= 0 {
+ if len == 0 {
bail!("Expected at least one entry, but drive is empty")
}
// 1. get most recent entry (tail of register)
- return self.get_dir_entry(len-2);
+ return self.get_dir_entry(len - 1);
- // XXX: unimplemented!()
+ // XXX: actual implementation...
}
pub fn history<'b>(&'b mut self, start: u64) -> DriveHistory<'b> {
@@ -129,22 +167,75 @@ impl<'a> DatDrive {
ReadDriveDir::init(self, path, false).unwrap()
}
- pub fn file_metadata<P: AsRef<Path>>(&mut self, _path: P) -> Result<Stat> {
- unimplemented!()
+ pub fn file_metadata<P: AsRef<Path>>(&mut self, path: P) -> Result<Stat> {
+ let de = self.broken_find_file(path.as_ref())?;
+ if let Some(entry) = de {
+ // if entry.stat was None, we'd have gotten None back
+ return Ok(entry.stat.unwrap());
+ } else {
+ bail!("Couldn't find path: {}", path.as_ref().display());
+ }
}
- pub fn create_file_bytes<P: AsRef<Path>>(&mut self, _path: P, _stat: &Stat, _data: &[u8]) -> Result<()> {
- unimplemented!()
+ pub fn add_file_bytes<P: AsRef<Path>>(&mut self, path: P, stat: &mut Stat, data: &[u8]) -> Result<()> {
+ // For now, just copies the data into a Vec (which implements Read)
+ self.add_file(path, stat, data)
}
- pub fn create_file<P: AsRef<Path>, R: Read>(&mut self, _path: P, _stat: &Stat, _source: R) -> Result<()> {
- unimplemented!()
+ pub fn add_file<P: AsRef<Path>, R: Read>(&mut self, path: P, stat: &mut Stat, mut source: R) -> Result<()> {
+ // TODO: check if file already exists
+ // XXX: verify stat size against passed size
+ let mut total_size: u64 = 0;
+ let mut data_entries: u64 = 0;
+ let mut buf = [0; 65536];
+ let data_offset = self.content.len()?;
+ let data_byte_offset = self.content.len_bytes()?;
+
+ loop {
+ // 1. read chunk
+ let rlen = source.read(&mut buf)?;
+ if rlen == 0 {
+ break;
+ }
+ // 2. append chunk to data register
+ self.content.append(&buf[0..rlen])?;
+
+ // 3. increment metadata size
+ total_size += rlen as u64;
+ data_entries += 1;
+ }
+
+ // 4. write metadata
+ stat.set_size(total_size as u64);
+ stat.set_blocks(data_entries);
+ stat.set_offset(data_offset);
+ stat.set_byteOffset(data_byte_offset);
+ // XXX: actual child implementation
+ let children = vec![];
+ let children = encode_children(&children)?;
+ let mut node = Node::new();
+ node.set_name(path.as_ref().to_string_lossy().into_owned());
+ node.set_value(stat.write_to_bytes()?);
+ node.set_paths(children);
+ self.metadata.append(&node.write_to_bytes()?)?;
+
+ Ok(())
}
/// Copies Stat metadata and all content from a file in the "real" filesystem into the
/// DatDrive.
- pub fn import_file<P: AsRef<Path>, Q: AsRef<Path>>(&mut self, _source: P, _dest: Q) -> Result<()> {
- unimplemented!()
+ pub fn import_file<P: AsRef<Path>, Q: AsRef<Path>>(&mut self, source: P, dest: Q) -> Result<()> {
+ // XXX: check if file already exists
+ let in_file = File::open(source)?;
+ let in_metadata = in_file.metadata()?;
+ let mut stat = Stat::new();
+ stat.set_mode(in_metadata.mode());
+ stat.set_uid(in_metadata.uid());
+ stat.set_gid(in_metadata.gid());
+ stat.set_size(in_metadata.size());
+ stat.set_mtime(in_metadata.mtime() as u64);
+ stat.set_ctime(in_metadata.ctime() as u64);
+ self.add_file(dest, &mut stat, in_file)
}
/// Copies a file from the drive to the "real" filesystem, preserving Stat metadata.
@@ -193,8 +284,8 @@ fn test_dd_open() {
// verified from dat log
assert_eq!(dd.history(0).count(), 2);
- assert_eq!(dd.read_dir("/").count(), 1);
- assert_eq!(dd.read_dir_recursive("/").count(), 1);
+ //XXX:assert_eq!(dd.read_dir("/").count(), 1);
+ //XXX:assert_eq!(dd.read_dir_recursive("/").count(), 1);
let mut dd =
DatDrive::open(Path::new("test-data/dat/tree/.dat/"), false).unwrap();
@@ -216,6 +307,40 @@ fn test_dd_create() {
assert_eq!(dd.read_dir_recursive("/").count(), 0);
}
+fn make_test_stat() -> Stat {
+ let mut stat = Stat::new();
+ stat.set_mode(0o777);
+ stat.set_uid(1000);
+ stat.set_gid(1000);
+ stat.set_size(0);
+ stat.set_mtime(54321);
+ stat.set_ctime(65432);
+ stat
+}
+
+#[test]
+fn test_dd_add() {
+ use tempdir::TempDir;
+ let tmp_dir = TempDir::new("geniza-test").unwrap();
+ let mut dd = DatDrive::create(tmp_dir.path()).unwrap();
+
+ let data = vec![7; 123];
+ let mut stat = make_test_stat();
+ stat.set_size(123);
+ dd.add_file_bytes("/bytes.bin", &mut stat, &data).unwrap();
+ assert_eq!(dd.history(0).count(), 1);
+ //XXX:assert_eq!(dd.read_dir("/").count(), 1);
+ //XXX:assert_eq!(dd.read_dir_recursive("/").count(), 1);
+ assert_eq!(dd.content.len_bytes().unwrap(), 123);
+
+ stat.set_size(65);
+ dd.add_file("/bytes_read.bin", &mut stat, &data[0..65]).unwrap();
+ assert_eq!(dd.history(0).count(), 2);
+ //XXX:assert_eq!(dd.read_dir("/").count(), 2);
+ //XXX:assert_eq!(dd.read_dir_recursive("/").count(), 2);
+ assert_eq!(dd.content.len_bytes().unwrap(), 123+65);
+}
+
// TODO: unpack Node into a pub struct
#[derive(Debug)]
pub struct DriveEntry {
@@ -255,9 +380,11 @@ pub struct ReadDriveDir<'a> {
impl<'a> ReadDriveDir<'a> {
fn init<P: AsRef<Path>>(drive: &mut DatDrive, path: P, recursive: bool) -> Result<ReadDriveDir> {
+ // first entry is content pub key
let entries = if drive.entry_count()? == 0 {
vec![]
} else {
+/* XXX: actual implementation
let nearest = drive.get_nearest(path)?;
// TODO: starting from the last data entry, recurse up to nearest directory, then recurse
// down to base path
@@ -272,6 +399,14 @@ impl<'a> ReadDriveDir<'a> {
for mut sub in nearest.children {
entries.append(&mut sub);
}
+*/
+ let mut entries = vec![];
+ for i in 0..drive.entry_count()? {
+ let e = drive.get_dir_entry(i)?;
+ if e.stat.is_some() {
+ entries.push(i);
+ }
+ }
entries
};
Ok(ReadDriveDir {
@@ -288,7 +423,6 @@ impl<'a> Iterator for ReadDriveDir<'a> {
// TODO: actually recurse
match self.entries.pop() {
None => None,
- // XXX: +1 here is to skip the initial header
Some(this_index) => Some(self.drive.get_dir_entry(this_index))
}
}