From 350320a32ce569d3e86352ba5a79ee969e9595af Mon Sep 17 00:00:00 2001 From: Justus Winter Date: Thu, 16 Apr 2020 11:58:32 +0200 Subject: Revert "openpgp: Make PacketParserResult a std::result::Result." This reverts commit 2e1eec5fe4157a391a13554ff7df3075cfe043cc. --- openpgp/src/cert/mod.rs | 4 +- openpgp/src/cert/parser/mod.rs | 6 +- openpgp/src/crypto/s2k.rs | 2 +- openpgp/src/packet/key.rs | 2 +- openpgp/src/packet/signature/mod.rs | 4 +- openpgp/src/packet/signature/subpacket.rs | 4 +- openpgp/src/packet_pile.rs | 18 +-- openpgp/src/parse.rs | 185 ++++++++++++++++++++--------- openpgp/src/parse/packet_parser_builder.rs | 12 +- openpgp/src/parse/packet_pile_parser.rs | 24 ++-- openpgp/src/parse/stream.rs | 18 +-- openpgp/src/serialize/stream.rs | 10 +- 12 files changed, 179 insertions(+), 110 deletions(-) (limited to 'openpgp/src') diff --git a/openpgp/src/cert/mod.rs b/openpgp/src/cert/mod.rs index 4ce0811b..2bd2e635 100644 --- a/openpgp/src/cert/mod.rs +++ b/openpgp/src/cert/mod.rs @@ -518,7 +518,7 @@ use super::*; /// ```rust /// # extern crate sequoia_openpgp as openpgp; /// # use openpgp::Result; -/// # use openpgp::parse::{Parse, PacketParser}; +/// # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; /// use openpgp::cert::prelude::*; /// /// # fn main() { f().unwrap(); } @@ -593,7 +593,7 @@ use super::*; /// # extern crate sequoia_openpgp as openpgp; /// # use std::convert::TryFrom; /// # use openpgp::Result; -/// # use openpgp::parse::{Parse, PacketParser}; +/// # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; /// use openpgp::Cert; /// /// # fn main() { f().unwrap(); } diff --git a/openpgp/src/cert/parser/mod.rs b/openpgp/src/cert/parser/mod.rs index da0194bc..50cc7e5a 100644 --- a/openpgp/src/cert/parser/mod.rs +++ b/openpgp/src/cert/parser/mod.rs @@ -394,7 +394,7 @@ enum PacketSource<'a, I: Iterator> { /// ```rust /// # extern crate sequoia_openpgp as openpgp; /// # use openpgp::Result; -/// # use openpgp::parse::{Parse, PacketParser}; +/// # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; /// use openpgp::cert::prelude::*; /// /// # fn main() { f().unwrap(); } @@ -441,7 +441,7 @@ impl<'a> From> for CertParser<'a, vec::IntoIter> /// Initializes a `CertParser` from a `PacketParser`. fn from(ppr: PacketParserResult<'a>) -> Self { let mut parser : Self = Default::default(); - if let Ok(pp) = ppr { + if let PacketParserResult::Some(pp) = ppr { parser.source = PacketSource::PacketParser(pp); } parser @@ -731,7 +731,7 @@ impl<'a, I: Iterator> Iterator for CertParser<'a, I> { match pp.next() { Ok((packet, ppr)) => { - if let Ok(pp) = ppr { + if let PacketParserResult::Some(pp) = ppr { self.source = PacketSource::PacketParser(pp); } diff --git a/openpgp/src/crypto/s2k.rs b/openpgp/src/crypto/s2k.rs index 40a26bda..8dff4693 100644 --- a/openpgp/src/crypto/s2k.rs +++ b/openpgp/src/crypto/s2k.rs @@ -416,7 +416,7 @@ mod tests { // Get the next packet. let (_, ppr) = pp.next().unwrap(); - assert!(ppr.is_err()); + assert!(ppr.is_none()); } } diff --git a/openpgp/src/packet/key.rs b/openpgp/src/packet/key.rs index 29409ace..d06ac07e 100644 --- a/openpgp/src/packet/key.rs +++ b/openpgp/src/packet/key.rs @@ -29,7 +29,7 @@ //! ```rust //! # extern crate sequoia_openpgp as openpgp; //! # use openpgp::Result; -//! # use openpgp::parse::{Parse, PacketParser}; +//! # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; //! # use openpgp::cert::prelude::*; //! use openpgp::packet::{Key, key}; //! diff --git a/openpgp/src/packet/signature/mod.rs b/openpgp/src/packet/signature/mod.rs index b294ae5a..66c0065a 100644 --- a/openpgp/src/packet/signature/mod.rs +++ b/openpgp/src/packet/signature/mod.rs @@ -1283,7 +1283,7 @@ mod test { use super::*; use crate::Cert; - use crate::parse::PacketParser; + use crate::parse::{PacketParserResult, PacketParser}; struct Test<'a> { key: &'a str, @@ -1369,7 +1369,7 @@ mod test { let mut good = 0; let mut ppr = PacketParser::from_bytes( crate::tests::message(test.data)).unwrap(); - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { if let Packet::Signature(ref sig) = pp.packet { let result = sig.verify(cert.primary_key().key()) .map(|_| true).unwrap_or(false); diff --git a/openpgp/src/packet/signature/subpacket.rs b/openpgp/src/packet/signature/subpacket.rs index ac000842..605d48d2 100644 --- a/openpgp/src/packet/signature/subpacket.rs +++ b/openpgp/src/packet/signature/subpacket.rs @@ -34,13 +34,13 @@ //! # extern crate sequoia_openpgp as openpgp; //! # use openpgp::Result; //! # use openpgp::Packet; -//! # use openpgp::parse::{Parse, PacketParser}; +//! # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; //! # //! # f(include_bytes!("../../../tests/data/messages/signed.gpg")); //! # //! # fn f(message_data: &[u8]) -> Result<()> { //! let mut ppr = PacketParser::from_bytes(message_data)?; -//! while let Ok(mut pp) = ppr { +//! while let PacketParserResult::Some(mut pp) = ppr { //! if let Packet::Signature(ref sig) = pp.packet { //! if let Some(fp) = sig.issuer_fingerprint() { //! eprintln!("Signature issued by: {}", fp.to_string()); diff --git a/openpgp/src/packet_pile.rs b/openpgp/src/packet_pile.rs index 4218a7bf..bc1bc9e3 100644 --- a/openpgp/src/packet_pile.rs +++ b/openpgp/src/packet_pile.rs @@ -528,7 +528,7 @@ impl<'a> TryFrom> for PacketPile { // ppo.recursion_depth and leave the rest of the message, but // it is hard to imagine that that is what the caller wants. // Instead of hiding that error, fail fast. - if let Ok(ref pp) = ppr { + if let PacketParserResult::Some(ref pp) = ppr { if pp.recursion_depth() != 0 { return Err(Error::InvalidOperation( format!("Expected top-level packet, \ @@ -542,7 +542,7 @@ impl<'a> TryFrom> for PacketPile { let mut last_position = 0; - if ppr.is_err() { + if ppr.is_none() { // Empty message. return Ok(PacketPile::from(Vec::new())); } @@ -593,7 +593,7 @@ impl<'a> TryFrom> for PacketPile { container.children_mut().unwrap().push(packet); - if ppr.is_err() { + if ppr.is_none() { break 'outer; } @@ -740,7 +740,7 @@ mod test { .try_into().unwrap(); let mut ppr = ppp.recurse().unwrap(); - while ppr.is_ok() { + while ppr.is_some() { ppr = ppp.recurse().unwrap(); } let pile = ppp.finish(); @@ -755,7 +755,7 @@ mod test { .try_into().unwrap(); let mut ppr = ppp.recurse().unwrap(); - while let Ok(pp) = ppr.as_mut() { + while let Some(pp) = ppr.as_mut() { eprintln!("{:?}", pp); ppr = ppp.recurse().unwrap(); } @@ -799,13 +799,13 @@ mod test { let mut count = 0; loop { - if let Ok(pp2) = ppr { + if let PacketParserResult::Some(pp2) = ppr { count += 1; let packet_depth = pp2.recursion_depth(); let pp2 = pp2.recurse().unwrap().1; assert_eq!(packet_depth, count - 1); - if pp2.is_ok() { + if pp2.is_some() { assert_eq!(pp2.as_ref().unwrap().recursion_depth(), count); } ppr = pp2; @@ -843,7 +843,7 @@ mod test { // recurse should now not recurse. Since there is nothing // following the compressed packet, ppr should be EOF. let (packet, ppr) = pp.next().unwrap(); - assert!(ppr.is_err()); + assert!(ppr.is_none()); // Get the rest of the content and put the initial byte that // we stole back. @@ -859,7 +859,7 @@ mod test { // And we're done... let ppr = pp.next().unwrap().1; - assert!(ppr.is_err()); + assert!(ppr.is_none()); } #[test] diff --git a/openpgp/src/parse.rs b/openpgp/src/parse.rs index fb556ad2..c8c42a29 100644 --- a/openpgp/src/parse.rs +++ b/openpgp/src/parse.rs @@ -212,13 +212,13 @@ macro_rules! impl_parse_generic_packet { pp.buffer_unread_content()?; match pp.next()? { - (Packet::$typ(o), Err(_)) + (Packet::$typ(o), PacketParserResult::EOF(_)) => Ok(o), - (p, Err(_)) => + (p, PacketParserResult::EOF(_)) => Err(Error::InvalidOperation( format!("Not a {} packet: {:?}", stringify!($typ), p)).into()), - (_, Ok(_)) => + (_, PacketParserResult::Some(_)) => Err(Error::InvalidOperation( "Excess data after packet".into()).into()), } @@ -1821,7 +1821,7 @@ fn one_pass_sig_test () { let mut one_pass_sigs = 0; let mut sigs = 0; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { if let Packet::OnePassSig(_) = pp.packet { one_pass_sigs += 1; } else if let Packet::Signature(ref sig) = pp.packet { @@ -2065,14 +2065,14 @@ impl<'a> Parse<'a, key::UnspecifiedKey> for key::UnspecifiedKey { pp.buffer_unread_content()?; match pp.next()? { - (Packet::PublicKey(o), Err(_)) => Ok(o.into()), - (Packet::PublicSubkey(o), Err(_)) => Ok(o.into()), - (Packet::SecretKey(o), Err(_)) => Ok(o.into()), - (Packet::SecretSubkey(o), Err(_)) => Ok(o.into()), - (p, Err(_)) => + (Packet::PublicKey(o), PacketParserResult::EOF(_)) => Ok(o.into()), + (Packet::PublicSubkey(o), PacketParserResult::EOF(_)) => Ok(o.into()), + (Packet::SecretKey(o), PacketParserResult::EOF(_)) => Ok(o.into()), + (Packet::SecretSubkey(o), PacketParserResult::EOF(_)) => Ok(o.into()), + (p, PacketParserResult::EOF(_)) => Err(Error::InvalidOperation( format!("Not a Key packet: {:?}", p)).into()), - (_, Ok(_)) => + (_, PacketParserResult::Some(_)) => Err(Error::InvalidOperation( "Excess data after packet".into()).into()), } @@ -2331,7 +2331,7 @@ fn compressed_data_parser_test () { } // And, we're done... - assert!(ppr.is_err()); + assert!(ppr.is_none()); } } @@ -2673,18 +2673,18 @@ impl<'a> Parse<'a, Packet> for Packet { ?.buffer_unread_content().build()?; let (p, ppr) = match ppr { - Ok(pp) => { + PacketParserResult::Some(pp) => { pp.next()? }, - Err(_) => + PacketParserResult::EOF(_) => return Err(Error::InvalidOperation( "Unexpected EOF".into()).into()), }; match (p, ppr) { - (p, Err(_)) => + (p, PacketParserResult::EOF(_)) => Ok(p), - (_, Ok(_)) => + (_, PacketParserResult::Some(_)) => Err(Error::InvalidOperation( "Excess data after packet".into()).into()), } @@ -2754,12 +2754,12 @@ impl PacketParserState { /// # extern crate sequoia_openpgp as openpgp; /// # use openpgp::Result; /// # use openpgp::Packet; -/// # use openpgp::parse::{Parse, PacketParser}; +/// # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; /// # let _ = f(include_bytes!("../tests/data/keys/public-key.gpg")); /// # /// # fn f(message_data: &[u8]) -> Result<()> { /// let mut ppr = PacketParser::from_bytes(message_data)?; -/// while let Ok(mut pp) = ppr { +/// while let PacketParserResult::Some(mut pp) = ppr { /// // Process the packet. /// /// if let Packet::Literal(_) = pp.packet { @@ -2916,20 +2916,89 @@ impl PacketParserEOF { /// We don't use an `Option`, because when we reach the end of the /// packet sequence, some information about the message needs to /// remain accessible. -pub type PacketParserResult<'a> = - std::result::Result, PacketParserEOF>; - -/// Like `Option::take`() for PacketParserResult. -/// -/// `self` is replaced with a `PacketParserEOF` with default values. -fn packet_parser_result_take<'a>(ppr: &mut PacketParserResult<'a>) - -> PacketParserResult<'a> { - mem::replace( - ppr, - Err(PacketParserEOF::new( - PacketParserState::new(Default::default())))) +#[derive(Debug)] +pub enum PacketParserResult<'a> { + /// A `PacketParser` for the next packet. + Some(PacketParser<'a>), + /// Information about a fully parsed packet sequence. + EOF(PacketParserEOF), } +impl<'a> PacketParserResult<'a> { + /// Like `Option::is_none`(). + pub fn is_none(&self) -> bool { + if let PacketParserResult::EOF(_) = self { + true + } else { + false + } + } + + /// An alias for `is_none`(). + pub fn is_eof(&self) -> bool { + Self::is_none(self) + } + + /// Like `Option::is_some`(). + pub fn is_some(&self) -> bool { + ! Self::is_none(self) + } + + /// Like `Option::expect`(). + pub fn expect(self, msg: &str) -> PacketParser<'a> { + if let PacketParserResult::Some(pp) = self { + return pp; + } else { + panic!("{}", msg); + } + } + + /// Like `Option::unwrap`(). + pub fn unwrap(self) -> PacketParser<'a> { + self.expect("called `PacketParserResult::unwrap()` on a \ + `PacketParserResult::PacketParserEOF` value") + } + + /// Like `Option::as_ref`(). + pub fn as_ref(&self) -> Option<&PacketParser<'a>> { + if let PacketParserResult::Some(ref pp) = self { + Some(pp) + } else { + None + } + } + + /// Like `Option::as_mut`(). + pub fn as_mut(&mut self) -> Option<&mut PacketParser<'a>> { + if let PacketParserResult::Some(ref mut pp) = self { + Some(pp) + } else { + None + } + } + + /// Like `Option::take`(). + /// + /// `self` is replaced with a `PacketParserEOF` with default + /// values. + pub fn take(&mut self) -> Self { + mem::replace( + self, + PacketParserResult::EOF( + PacketParserEOF::new( + PacketParserState::new(Default::default())))) + } + + /// Like `Option::map`(). + pub fn map(self, f: F) -> Option + where F: FnOnce(PacketParser<'a>) -> U + { + match self { + PacketParserResult::Some(x) => Some(f(x)), + PacketParserResult::EOF(_) => None, + } + } +} impl<'a> Parse<'a, PacketParserResult<'a>> for PacketParser<'a> { /// Starts parsing an OpenPGP message stored in a `std::io::Read` object. @@ -3464,7 +3533,7 @@ impl <'a> PacketParser<'a> { let mut eof = PacketParserEOF::new(state_); eof.last_path = self.last_path; return Ok((self.packet, - Err(eof))); + PacketParserResult::EOF(eof))); } else { self.state = state_; self.finish()?; @@ -3487,7 +3556,7 @@ impl <'a> PacketParser<'a> { pp.last_path = self.last_path; - return Ok((self.packet, Ok(pp))); + return Ok((self.packet, PacketParserResult::Some(pp))); } } } @@ -3557,7 +3626,7 @@ impl <'a> PacketParser<'a> { pp.last_path = last_path; return Ok((self.packet, - Ok(pp))); + PacketParserResult::Some(pp))); }, ParserResult::EOF(_) => { return Err(Error::MalformedPacket( @@ -3598,13 +3667,13 @@ impl <'a> PacketParser<'a> { /// # extern crate sequoia_openpgp as openpgp; /// # use openpgp::Result; /// # use openpgp::Packet; - /// # use openpgp::parse::{Parse, PacketParser}; + /// # use openpgp::parse::{Parse, PacketParserResult, PacketParser}; /// # use std::string::String; /// # f(include_bytes!("../tests/data/messages/literal-mode-t-partial-body.gpg")); /// # /// # fn f(message_data: &[u8]) -> Result<()> { /// let mut ppr = PacketParser::from_bytes(message_data)?; - /// while let Ok(mut pp) = ppr { + /// while let PacketParserResult::Some(mut pp) = ppr { /// // Process the packet. /// /// if let Packet::Literal(_) = pp.packet { @@ -3950,7 +4019,7 @@ fn packet_parser_reader_interface() { // Make sure we can still get the next packet (which in this case // is just EOF). let (packet, ppr) = pp.recurse().unwrap(); - assert!(ppr.is_err()); + assert!(ppr.is_none()); // Since we read all of the data, we expect content to be None. assert_eq!(packet.unprocessed_body().unwrap().len(), 0); } @@ -4284,11 +4353,11 @@ mod test { ppr = ppr.unwrap().recurse().unwrap().1; } - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { let tag = pp.packet.tag(); for t in keep.iter() { if *t == tag { - return Ok(pp); + return PacketParserResult::Some(pp); } } @@ -4332,7 +4401,7 @@ mod test { let mut ppr = consume_until( ppr, false, &[ Tag::SEIP, Tag::AED ][..], &[ Tag::SKESK, Tag::PKESK ][..] ); - if let Ok(ref mut pp) = ppr { + if let PacketParserResult::Some(ref mut pp) = ppr { let key = crate::fmt::from_hex(test.key_hex, false) .unwrap().into(); @@ -4344,7 +4413,7 @@ mod test { let mut ppr = consume_until( ppr, true, &[ Tag::Literal ][..], &[ Tag::OnePassSig, Tag::CompressedData ][..]); - if let Ok(ref mut pp) = ppr { + if let PacketParserResult::Some(ref mut pp) = ppr { if stream { let mut body = Vec::new(); loop { @@ -4373,20 +4442,20 @@ mod test { let ppr = consume_until( ppr, true, &[ Tag::MDC ][..], &[ Tag::Signature ][..]); - if let Ok( + if let PacketParserResult::Some( PacketParser { packet: Packet::MDC(ref mdc), .. }) = ppr { assert_eq!(mdc.computed_digest(), mdc.digest(), "MDC doesn't match"); } - if ppr.is_err() { + if ppr.is_none() { // AED packets don't have an MDC packet. continue; } let ppr = consume_until( ppr, true, &[][..], &[][..]); - assert!(ppr.is_err()); + assert!(ppr.is_none()); } } @@ -4400,7 +4469,7 @@ mod test { // Make sure we actually decrypted... let mut saw_literal = false; - while let Ok(mut pp) = ppr { + while let PacketParserResult::Some(mut pp) = ppr { assert!(pp.possible_message().is_ok()); match pp.packet { @@ -4419,7 +4488,7 @@ mod test { ppr = pp.recurse().unwrap().1; } assert!(saw_literal); - if let Err(eof) = ppr { + if let PacketParserResult::EOF(eof) = ppr { assert!(eof.is_message().is_ok()); } else { unreachable!(); @@ -4441,11 +4510,11 @@ mod test { .build() .expect(&format!("Error reading {:?}", test)); - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { assert!(pp.possible_keyring().is_ok()); ppr = pp.recurse().unwrap().1; } - if let Err(eof) = ppr { + if let PacketParserResult::EOF(eof) = ppr { assert!(eof.is_keyring().is_ok()); assert!(eof.is_cert().is_err()); } else { @@ -4466,12 +4535,12 @@ mod test { .build() .expect(&format!("Error reading {:?}", test)); - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { assert!(pp.possible_keyring().is_ok()); assert!(pp.possible_cert().is_ok()); ppr = pp.recurse().unwrap().1; } - if let Err(eof) = ppr { + if let PacketParserResult::EOF(eof) = ppr { assert!(eof.is_keyring().is_ok()); assert!(eof.is_cert().is_ok()); } else { @@ -4491,7 +4560,7 @@ mod test { .expect(&format!("Error reading {}", test.filename)[..]); let mut saw_literal = false; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { assert!(pp.possible_message().is_ok()); match pp.packet { @@ -4505,7 +4574,7 @@ mod test { ppr = pp.recurse().unwrap().1; } assert!(! saw_literal); - if let Err(eof) = ppr { + if let PacketParserResult::EOF(eof) = ppr { eprintln!("eof: {:?}; message: {:?}", eof, eof.is_message()); assert!(eof.is_message().is_ok()); } else { @@ -4530,7 +4599,7 @@ mod test { // We pop from the end. paths.reverse(); - while let Ok(mut pp) = ppr { + while let PacketParserResult::Some(mut pp) = ppr { let path = paths.pop().expect("Message longer than expect"); assert_eq!(path.0, pp.packet.tag()); assert_eq!(path.1, pp.path()); @@ -4557,7 +4626,7 @@ mod test { "Message shorter than expected (expecting: {:?})", paths); - if let Err(eof) = ppr { + if let PacketParserResult::EOF(eof) = ppr { assert_eq!(last_path, eof.last_path()); } else { panic!("Expect an EOF"); @@ -4581,7 +4650,7 @@ mod test { let mut userids = 0; let mut uas = 0; let mut unknown = 0; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { match pp.packet { Packet::Signature(_) => sigs += 1, Packet::PublicSubkey(_) => subkeys += 1, @@ -4613,7 +4682,7 @@ mod test { let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(packet_parser_builder::Dearmor::Disabled) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); // Prepend an invalid byte and make sure we fail. Note: we @@ -4640,14 +4709,14 @@ mod test { let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(packet_parser_builder::Dearmor::Disabled) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); // Now truncate the packet. let msg2 = &msg[..msg.len() - 1]; let ppr = PacketParserBuilder::from_bytes(msg2).unwrap() .dearmor(packet_parser_builder::Dearmor::Disabled) .build().unwrap(); - if let Ok(pp) = ppr { + if let PacketParserResult::Some(pp) = ppr { let err = pp.next().err().unwrap(); assert_match!(Some(&Error::MalformedPacket(_)) = err.downcast_ref()); @@ -4667,7 +4736,7 @@ mod test { // Make sure we can read it. let ppr = PacketParserBuilder::from_bytes(&buf).unwrap() .build().unwrap(); - if let Ok(pp) = ppr { + if let PacketParserResult::Some(pp) = ppr { assert_eq!(Packet::UserID("foobar".into()), pp.packet); } else { panic!("failed to parse userid"); @@ -4678,7 +4747,7 @@ mod test { let ppr = PacketParserBuilder::from_bytes(&buf).unwrap() .max_packet_size(5) .build().unwrap(); - if let Ok(pp) = ppr { + if let PacketParserResult::Some(pp) = ppr { if let Packet::Unknown(ref u) = pp.packet { assert_eq!(u.tag(), Tag::UserID); assert_match!(Some(&Error::PacketTooLarge(_, _, _)) diff --git a/openpgp/src/parse/packet_parser_builder.rs b/openpgp/src/parse/packet_parser_builder.rs index 93bf20eb..f22469bd 100644 --- a/openpgp/src/parse/packet_parser_builder.rs +++ b/openpgp/src/parse/packet_parser_builder.rs @@ -208,11 +208,11 @@ impl<'a> PacketParserBuilder<'a> { pp.state.message_validator.push(pp.packet.tag(), &[0]); pp.state.keyring_validator.push(pp.packet.tag()); pp.state.cert_validator.push(pp.packet.tag()); - Ok(Ok(pp)) + Ok(PacketParserResult::Some(pp)) }, ParserResult::EOF((_reader, state, _path)) => { // `bio` is empty. We're done. - Ok(Err(PacketParserEOF::new(state))) + Ok(PacketParserResult::EOF(PacketParserEOF::new(state))) } } } @@ -231,12 +231,12 @@ mod tests { let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(Dearmor::Disabled) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(Dearmor::Auto(Default::default())) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(Dearmor::Enabled(Default::default())) @@ -255,11 +255,11 @@ mod tests { let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(Dearmor::Auto(Default::default())) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); let ppr = PacketParserBuilder::from_bytes(msg).unwrap() .dearmor(Dearmor::Enabled(Default::default())) .build(); - assert_match!(Ok(Ok(ref _pp)) = ppr); + assert_match!(Ok(PacketParserResult::Some(ref _pp)) = ppr); } } diff --git a/openpgp/src/parse/packet_pile_parser.rs b/openpgp/src/parse/packet_pile_parser.rs index 170b451b..b0a7b97b 100644 --- a/openpgp/src/parse/packet_pile_parser.rs +++ b/openpgp/src/parse/packet_pile_parser.rs @@ -45,7 +45,7 @@ use buffered_reader::BufferedReader; /// # fn f(message_data: &[u8]) -> Result<()> { /// let mut ppp = PacketPileParser::from_bytes(message_data)?; /// let mut ppr = ppp.recurse()?; -/// while let Ok(pp) = ppr.as_ref() { +/// while let Some(pp) = ppr.as_ref() { /// eprintln!("{:?}", pp); /// ppr = ppp.recurse()?; /// } @@ -158,8 +158,8 @@ impl<'a> PacketPileParser<'a> { /// it returns the following packet. pub fn recurse(&mut self) -> Result<&mut PacketParserResult<'a>> { if self.returned_first { - match super::packet_parser_result_take(&mut self.ppr) { - Ok(pp) => { + match self.ppr.take() { + PacketParserResult::Some(pp) => { let recursion_depth = pp.recursion_depth(); let (packet, ppr) = pp.recurse()?; self.insert_packet( @@ -167,7 +167,7 @@ impl<'a> PacketPileParser<'a> { recursion_depth as isize); self.ppr = ppr; } - eof @ Err(_) => { + eof @ PacketParserResult::EOF(_) => { self.ppr = eof; } } @@ -188,8 +188,8 @@ impl<'a> PacketPileParser<'a> { /// container; it skips any packets that it may contain. pub fn next(&mut self) -> Result<&mut PacketParserResult<'a>> { if self.returned_first { - match super::packet_parser_result_take(&mut self.ppr) { - Ok(pp) => { + match self.ppr.take() { + PacketParserResult::Some(pp) => { let recursion_depth = pp.recursion_depth(); let (packet, ppr) = pp.next()?; self.insert_packet( @@ -197,7 +197,7 @@ impl<'a> PacketPileParser<'a> { recursion_depth as isize); self.ppr = ppr; }, - eof @ Err(_) => { + eof @ PacketParserResult::EOF(_) => { self.ppr = eof }, } @@ -213,7 +213,7 @@ impl<'a> PacketPileParser<'a> { /// A top-level packet has a recursion depth of 0. Packets in a /// top-level container have a recursion depth of 1. Etc. pub fn recursion_depth(&self) -> Option { - if let Ok(ref pp) = self.ppr { + if let PacketParserResult::Some(ref pp) = self.ppr { Some(pp.recursion_depth() as u8) } else { None @@ -222,7 +222,7 @@ impl<'a> PacketPileParser<'a> { /// Returns whether the message has been completely parsed. pub fn is_done(&self) -> bool { - self.ppr.is_err() + self.ppr.is_none() } /// Finishes parsing the message and returns the assembled @@ -245,7 +245,7 @@ fn test_recurse() -> Result<()> { let mut ppp = PacketPileParser::from_bytes(crate::tests::key("public-key.gpg"))?; let mut ppr = ppp.recurse().unwrap(); - while ppr.is_ok() { + while ppr.is_some() { count += 1; ppr = ppp.recurse().unwrap(); } @@ -259,7 +259,7 @@ fn test_next() -> Result<()> { let mut ppp = PacketPileParser::from_bytes(crate::tests::key("public-key.gpg"))?; let mut ppr = ppp.recurse().unwrap(); - while ppr.is_ok() { + while ppr.is_some() { count += 1; ppr = ppp.next().unwrap(); } @@ -282,7 +282,7 @@ fn message_parser_reader_interface() { crate::tests::message("compressed-data-algo-1.gpg")).unwrap(); let mut count = 0; let mut ppr = ppp.recurse().unwrap(); - while let Ok(pp) = ppr.as_mut() { + while let Some(pp) = ppr.as_mut() { count += 1; if let Packet::Literal(_) = pp.packet { assert_eq!(count, 2); diff --git a/openpgp/src/parse/stream.rs b/openpgp/src/parse/stream.rs index 3e034b27..9ae3a1ea 100644 --- a/openpgp/src/parse/stream.rs +++ b/openpgp/src/parse/stream.rs @@ -1092,7 +1092,7 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { let mut skesks: Vec = Vec::new(); let mut saw_content = false; - while let Ok(mut pp) = ppr { + while let PacketParserResult::Some(mut pp) = ppr { v.policy.packet(&pp.packet)?; v.helper.inspect(&pp)?; @@ -1165,7 +1165,7 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { v.structure.insert_missing_signature_group(); // Query keys. v.certs = v.helper.get_certs(&issuers)?; - v.oppr = Some(Ok(pp)); + v.oppr = Some(PacketParserResult::Some(pp)); v.finish_maybe()?; return Ok(v); @@ -1270,7 +1270,7 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { // // Note: once this call succeeds, you may not call it again. fn finish_maybe(&mut self) -> Result<()> { - if let Some(Ok(mut pp)) = self.oppr.take() { + if let Some(PacketParserResult::Some(mut pp)) = self.oppr.take() { // Check if we hit EOF. let data_len = pp.data(BUFFER_SIZE + 1)?.len(); if data_len <= BUFFER_SIZE { @@ -1279,9 +1279,9 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { self.cursor = 0; // Process the rest of the packets. - let mut ppr = Ok(pp); + let mut ppr = PacketParserResult::Some(pp); let mut first = true; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { // The literal data packet was already inspected. if first { assert_eq!(pp.packet.tag(), packet::Tag::Literal); @@ -1309,7 +1309,7 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { self.verify_signatures() } else { - self.oppr = Some(Ok(pp)); + self.oppr = Some(PacketParserResult::Some(pp)); Ok(()) } } else { @@ -1524,18 +1524,18 @@ impl<'a, H: VerificationHelper + DecryptionHelper> Decryptor<'a, H> { } // Read the data from the Literal data packet. - if let Some(Ok(mut pp)) = self.oppr.take() { + if let Some(PacketParserResult::Some(mut pp)) = self.oppr.take() { // Be careful to not read from the reserve. let data_len = pp.data(BUFFER_SIZE + buf.len())?.len(); if data_len <= BUFFER_SIZE { - self.oppr = Some(Ok(pp)); + self.oppr = Some(PacketParserResult::Some(pp)); self.finish_maybe()?; self.read_helper(buf) } else { let n = cmp::min(buf.len(), data_len - BUFFER_SIZE); let buf = &mut buf[..n]; let result = pp.read(buf); - self.oppr = Some(Ok(pp)); + self.oppr = Some(PacketParserResult::Some(pp)); Ok(result?) } } else { diff --git a/openpgp/src/serialize/stream.rs b/openpgp/src/serialize/stream.rs index 8b284274..856914b3 100644 --- a/openpgp/src/serialize/stream.rs +++ b/openpgp/src/serialize/stream.rs @@ -2386,7 +2386,7 @@ impl<'a> writer::Stackable<'a, Cookie> for Encryptor<'a> { mod test { use std::io::Read; use crate::{Packet, PacketPile, packet::CompressedData}; - use crate::parse::{Parse, PacketParser}; + use crate::parse::{Parse, PacketParserResult, PacketParser}; use super::*; use crate::types::DataFormat::Text as T; use crate::policy::Policy; @@ -2419,7 +2419,7 @@ mod test { // Make sure it is the only packet. let (_, ppr) = pp.recurse().unwrap(); - assert!(ppr.is_err()); + assert!(ppr.is_none()); } // Create some crazy nesting structures, serialize the messages, @@ -2594,7 +2594,7 @@ mod test { let mut ppr = PacketParser::from_bytes(&o).unwrap(); let mut good = 0; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { if let Packet::Signature(ref sig) = pp.packet { let key = keys.get(&sig.issuer_fingerprint().unwrap()) .unwrap(); @@ -2639,7 +2639,7 @@ mod test { for password in &passwords { let mut state = State::Start; let mut ppr = PacketParser::from_bytes(&o).unwrap(); - while let Ok(mut pp) = ppr { + while let PacketParserResult::Some(mut pp) = ppr { state = match state { // Look for the SKESK packet. State::Start => @@ -2916,7 +2916,7 @@ mod test { let mut ppr = PacketParser::from_bytes(&o).unwrap(); let mut good = 0; - while let Ok(pp) = ppr { + while let PacketParserResult::Some(pp) = ppr { if let Packet::Signature(ref sig) = pp.packet { assert_eq!(sig.signature_creation_time(), Some(timestamp)); sig.verify(ka.key()).unwrap(); -- cgit v1.2.3