Update bytes written when manually incrementing offset of virtio chain

This commit is contained in:
Bruce Leidl 2019-09-20 16:03:36 -04:00
parent 98a1e9361e
commit 626b6583db

View File

@ -128,13 +128,22 @@ impl Chain {
/// load next descriptor if `current` descriptor
/// has been fully consumed.
///
pub fn inc_offset(&mut self, sz: usize) {
fn _inc_offset(&mut self, sz: usize) {
self.offset += sz;
if self.offset >= self.current_size() {
self.load_next_descriptor();
}
}
pub fn inc_offset(&mut self, sz: usize, write: bool) {
if write {
assert!(!self.is_current_readable());
self.wlen += sz;
}
self._inc_offset(sz)
}
///
/// Read from the `current` readable descriptor and return
/// the number of bytes read.
@ -151,7 +160,7 @@ impl Chain {
let nread = self.with_current_descriptor(0, |desc| {
desc.read_from(&self.memory, self.offset, bytes)
});
self.inc_offset(nread);
self._inc_offset(nread);
nread
}
@ -170,7 +179,7 @@ impl Chain {
let sz = self.with_current_descriptor(0, |desc| {
desc.write_to(&self.memory, self.offset, bytes)
});
self.inc_offset(sz);
self._inc_offset(sz);
sz
}
@ -192,6 +201,10 @@ impl Chain {
pub fn current_write_address(&mut self, size: usize) -> Option<u64> {
self.skip_readable();
self.current_address(size)
}
pub fn current_address(&mut self, size: usize) -> Option<u64> {
self.with_current_descriptor(None, |desc| {
if desc.len as usize - self.offset < size {
None
@ -220,7 +233,7 @@ impl Chain {
desc.write_from_reader(&self.memory, self.offset,r, size)
});
if let Ok(nread) = res {
self.inc_offset(nread);
self._inc_offset(nread);
self.wlen += nread;
}
res