mirrors/crosvm
1
0
Fork 0
mirror of https://chromium.googlesource.com/crosvm/crosvm synced 2025-02-06 02:25:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

descriptor_utils: Add iter method

Browse source 
Add an `iter` method to the `Reader` struct so that callers can iterate
over the objects in place rather than having to store them in a separate
collection.

BUG=none
TEST=unit tests

Change-Id: I29671910a4244a8d7786ca2eb241416ae72b8c9f
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/crosvm/+/2093966
Tested-by: Chirantan Ekbote <chirantan@chromium.org>
Tested-by: kokoro <noreply+kokoro@google.com>
Reviewed-by: Daniel Verkamp <dverkamp@chromium.org>
Commit-Queue: Chirantan Ekbote <chirantan@chromium.org>
This commit is contained in:
Chirantan Ekbote 2020-03-09 18:09:51 +09:00 committed by Commit Bot
parent 8bb4faa662
commit 265967bcda
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
devices/src/virtio/descriptor_utils.rs
View file

@ -216,8 +216,8 @@ pub struct Reader<'a> {
buffer: DescriptorChainConsumer<'a>,
}
// An iterator over `DataInit` objects on readable descriptors in the descriptor chain.
struct ReaderIterator<'a, T: DataInit> {
/// An iterator over `DataInit` objects on readable descriptors in the descriptor chain.
pub struct ReaderIterator<'a, T: DataInit> {
reader: &'a mut Reader<'a>,
phantom: PhantomData<T>,
}
@ -283,10 +283,17 @@ impl<'a> Reader<'a> {
/// them as a collection. Returns an error if the size of the remaining data is indivisible by
/// the size of an object of type `T`.
pub fn collect<C: FromIterator<io::Result<T>>, T: DataInit>(&'a mut self) -> C {
C::from_iter(ReaderIterator {
C::from_iter(self.iter())
}
/// Creates an iterator for sequentially reading `DataInit` objects from the `Reader`. Unlike
/// `collect`, this doesn't consume all the remaining data in the `Reader` and doesn't require
/// the objects to be stored in a separate collection.
pub fn iter<T: DataInit>(&'a mut self) -> ReaderIterator<'a, T> {
ReaderIterator {
reader: self,
phantom: PhantomData,
})
}
}
/// Reads data from the descriptor chain buffer into a file descriptor.