Handle v2 errors

DanGould · DanGould · commit b46dac1489f1 · 2023-10-02T11:26:09.000-04:00
diff --git a/payjoin-cli/src/app.rs b/payjoin-cli/src/app.rs
@@ -92,13 +92,15 @@ impl App {
         &self,
         client: &reqwest::Client,
         enroll_context: &mut EnrollContext,
-    ) -> Result<UncheckedProposal, reqwest::Error> {
+    ) -> Result<UncheckedProposal> {
         loop {
-            let (enroll_body, context) = enroll_context.enroll_body();
+            let (enroll_body, context) = enroll_context.enroll_body()?;
             let ohttp_response =
                 client.post(&self.config.ohttp_proxy).body(enroll_body).send().await?;
             let ohttp_response = ohttp_response.bytes().await?;
-            let proposal = enroll_context.parse_proposal(ohttp_response.as_ref(), context).unwrap();
+            let proposal = enroll_context
+                .parse_proposal(ohttp_response.as_ref(), context)
+                .map_err(|e| anyhow!("parse error {}", e))?;
             match proposal {
                 Some(proposal) => return Ok(proposal),
                 None => tokio::time::sleep(std::time::Duration::from_secs(5)).await,
@@ -238,8 +240,11 @@ impl App {
             .map_err(|e| anyhow!("Failed to process UncheckedProposal {}", e))?;
 
         let receive_endpoint = format!("{}/{}", self.config.pj_endpoint, context.receive_subdir());
-        let (body, ohttp_ctx) =
-            payjoin_proposal.extract_v2_req(&self.config.ohttp_config, &receive_endpoint);
+        let ohttp_config =
+            bitcoin::base64::decode_config(&self.config.ohttp_config, base64::URL_SAFE)?;
+        let (body, ohttp_ctx) = payjoin_proposal
+            .extract_v2_req(&ohttp_config, &receive_endpoint)
+            .map_err(|e| anyhow!("v2 req extraction failed {}", e))?;
         let res = client
             .post(&self.config.ohttp_proxy)
             .body(body)
diff --git a/payjoin-relay/src/main.rs b/payjoin-relay/src/main.rs
@@ -68,8 +68,8 @@ fn init_ohttp() -> Result<ohttp::Server> {
         &[SymmetricSuite::new(Kdf::HkdfSha256, Aead::ChaCha20Poly1305)];
 
     // create or read from file
-    let server_config = ohttp::KeyConfig::new(KEY_ID, KEM, Vec::from(SYMMETRIC)).unwrap();
-    let encoded_config = server_config.encode().unwrap();
+    let server_config = ohttp::KeyConfig::new(KEY_ID, KEM, Vec::from(SYMMETRIC))?;
+    let encoded_config = server_config.encode()?;
     let b64_config = payjoin::bitcoin::base64::encode_config(
         &encoded_config,
         payjoin::bitcoin::base64::Config::new(
@@ -83,43 +83,53 @@ fn init_ohttp() -> Result<ohttp::Server> {
 
 async fn handle_ohttp(
     enc_request: Bytes,
-    mut target: Router,
+    target: Router,
     ohttp: Arc<ohttp::Server>,
 ) -> (StatusCode, Vec<u8>) {
+    match handle_ohttp_inner(enc_request, target, ohttp).await {
+        Ok(res) => res,
+        Err(e) => {
+            tracing::error!("ohttp error: {:?}", e);
+            (StatusCode::INTERNAL_SERVER_ERROR, vec![])
+        }
+    }
+}
+
+async fn handle_ohttp_inner(
+    enc_request: Bytes,
+    mut target: Router,
+    ohttp: Arc<ohttp::Server>,
+) -> Result<(StatusCode, Vec<u8>)> {
     use axum::body::Body;
     use http::Uri;
     use tower_service::Service;
 
-    // decapsulate
-    let (bhttp_req, res_ctx) = ohttp.decapsulate(&enc_request).unwrap();
+    let (bhttp_req, res_ctx) = ohttp.decapsulate(&enc_request)?;
     let mut cursor = std::io::Cursor::new(bhttp_req);
-    let req = bhttp::Message::read_bhttp(&mut cursor).unwrap();
-    // let parsed_request: httparse::Request = httparse::Request::new(&mut vec![]).parse(cursor).unwrap();
-    // // handle request
-    // Request::new
+    let req = bhttp::Message::read_bhttp(&mut cursor)?;
     let uri = Uri::builder()
-        .scheme(req.control().scheme().unwrap())
-        .authority(req.control().authority().unwrap())
-        .path_and_query(req.control().path().unwrap())
-        .build()
-        .unwrap();
+        .scheme(req.control().scheme().unwrap_or_default())
+        .authority(req.control().authority().unwrap_or_default())
+        .path_and_query(req.control().path().unwrap_or_default())
+        .build()?;
     let body = req.content().to_vec();
-    let mut request = Request::builder().uri(uri).method(req.control().method().unwrap());
+    let mut request =
+        Request::builder().uri(uri).method(req.control().method().unwrap_or_default());
     for header in req.header().fields() {
         request = request.header(header.name(), header.value())
     }
-    let request = request.body(Body::from(body)).unwrap();
+    let request = request.body(Body::from(body))?;
 
-    let response = target.call(request).await.unwrap();
+    let response = target.call(request).await?;
 
     let (parts, body) = response.into_parts();
     let mut bhttp_res = bhttp::Message::response(parts.status.as_u16());
-    let full_body = hyper::body::to_bytes(body).await.unwrap();
+    let full_body = hyper::body::to_bytes(body).await?;
     bhttp_res.write_content(&full_body);
     let mut bhttp_bytes = Vec::new();
-    bhttp_res.write_bhttp(bhttp::Mode::KnownLength, &mut bhttp_bytes).unwrap();
-    let ohttp_res = res_ctx.encapsulate(&bhttp_bytes).unwrap();
-    (StatusCode::OK, ohttp_res)
+    bhttp_res.write_bhttp(bhttp::Mode::KnownLength, &mut bhttp_bytes)?;
+    let ohttp_res = res_ctx.encapsulate(&bhttp_bytes)?;
+    Ok((StatusCode::OK, ohttp_res))
 }
 
 fn ohttp_config(server: &ohttp::Server) -> Result<String> {
diff --git a/payjoin/src/receive/error.rs b/payjoin/src/receive/error.rs
@@ -7,13 +7,18 @@ pub enum Error {
     BadRequest(RequestError),
     // To be returned as HTTP 500
     Server(Box<dyn error::Error>),
+    // V2 d/encapsulation failed
+    #[cfg(feature = "v2")]
+    V2(crate::v2::Error),
 }
 
 impl fmt::Display for Error {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         match &self {
             Self::BadRequest(e) => e.fmt(f),
             Self::Server(e) => write!(f, "Internal Server Error: {}", e),
+            #[cfg(feature = "v2")]
+            Self::V2(e) => e.fmt(f),
         }
     }
 }
@@ -23,6 +28,8 @@ impl error::Error for Error {
         match &self {
             Self::BadRequest(_) => None,
             Self::Server(e) => Some(e.as_ref()),
+            #[cfg(feature = "v2")]
+            Self::V2(e) => Some(e),
         }
     }
 }
@@ -31,6 +38,15 @@ impl From<RequestError> for Error {
     fn from(e: RequestError) -> Self { Error::BadRequest(e) }
 }
 
+impl From<InternalRequestError> for Error {
+    fn from(e: InternalRequestError) -> Self { Error::BadRequest(e.into()) }
+}
+
+impl From<crate::v2::Error> for Error {
+    #[cfg(feature = "v2")]
+    fn from(e: crate::v2::Error) -> Self { Error::V2(e) }
+}
+
 /// Error that may occur when the request from sender is malformed.
 ///
 /// This is currently opaque type because we aren't sure which variants will stay.
diff --git a/payjoin/src/receive/mod.rs b/payjoin/src/receive/mod.rs
@@ -285,6 +285,7 @@ use url::Url;
 use crate::input_type::InputType;
 use crate::optional_parameters::Params;
 use crate::psbt::PsbtExt;
+use crate::v2;
 
 pub trait Headers {
     fn get_header(&self, key: &str) -> Option<&str>;
@@ -329,30 +330,28 @@ impl EnrollContext {
         format!("{}/{}", self.subdirectory(), crate::v2::RECEIVE)
     }
 
-    pub fn enroll_body(&mut self) -> (Vec<u8>, ohttp::ClientResponse) {
+    pub fn enroll_body(&mut self) -> Result<(Vec<u8>, ohttp::ClientResponse), crate::v2::Error> {
         let receive_endpoint = self.receive_subdir();
         log::debug!("{}{}", self.relay_url.as_str(), receive_endpoint);
-        let (ohttp_req, ctx) = crate::v2::ohttp_encapsulate(
+        crate::v2::ohttp_encapsulate(
             &self.ohttp_config,
             "GET",
             format!("{}{}", self.relay_url.as_str(), receive_endpoint).as_str(),
             None,
-        );
-
-        (ohttp_req, ctx)
+        )
     }
 
     pub fn parse_proposal(
         &self,
         encrypted_proposal: &[u8],
         context: ohttp::ClientResponse,
-    ) -> Result<Option<UncheckedProposal>, RequestError> {
-        let response = crate::v2::ohttp_decapsulate(context, &encrypted_proposal);
+    ) -> Result<Option<UncheckedProposal>, Error> {
+        let response = crate::v2::ohttp_decapsulate(context, &encrypted_proposal)?;
         if response.is_empty() {
             log::debug!("response is empty");
             return Ok(None);
         }
-        let (proposal, e) = crate::v2::decrypt_message_a(&response, self.s.secret_key());
+        let (proposal, e) = crate::v2::decrypt_message_a(&response, self.s.secret_key())?;
         let mut proposal = serde_json::from_slice::<UncheckedProposal>(&proposal)
             .map_err(InternalRequestError::Json)?;
         proposal.psbt = proposal.psbt.validate().map_err(InternalRequestError::InconsistentPsbt)?;
@@ -684,20 +683,26 @@ impl PayjoinProposal {
     #[cfg(feature = "v2")]
     pub fn extract_v2_req(
         &self,
-        ohttp_config: &str,
+        ohttp_config: &Vec<u8>,
         receive_endpoint: &str,
-    ) -> (Vec<u8>, ohttp::ClientResponse) {
+    ) -> Result<(Vec<u8>, ohttp::ClientResponse), Error> {
         let e = self.v2_context.unwrap(); // TODO make v2 only
         let mut payjoin_bytes = self.payjoin_psbt.serialize();
-        let body = crate::v2::encrypt_message_b(&mut payjoin_bytes, e);
-        let ohttp_config = bitcoin::base64::decode_config(ohttp_config, base64::URL_SAFE).unwrap();
+        let body = crate::v2::encrypt_message_b(&mut payjoin_bytes, e)?;
         dbg!(receive_endpoint);
-        crate::v2::ohttp_encapsulate(&ohttp_config, "POST", receive_endpoint, Some(&body))
+        let (req, ctx) =
+            crate::v2::ohttp_encapsulate(&ohttp_config, "POST", receive_endpoint, Some(&body))?;
+        Ok((req, ctx))
     }
 
-    pub fn deserialize_res(&self, res: Vec<u8>, ohttp_context: ohttp::ClientResponse) -> Vec<u8> {
+    pub fn deserialize_res(
+        &self,
+        res: Vec<u8>,
+        ohttp_context: ohttp::ClientResponse,
+    ) -> Result<Vec<u8>, Error> {
         // display success or failure
-        crate::v2::ohttp_decapsulate(ohttp_context, &res)
+        let res = crate::v2::ohttp_decapsulate(ohttp_context, &res)?;
+        Ok(res)
     }
 }
 
diff --git a/payjoin/src/send/error.rs b/payjoin/src/send/error.rs
@@ -16,13 +16,22 @@ pub struct ValidationError {
 
 #[derive(Debug)]
 pub(crate) enum InternalValidationError {
-    Psbt(bitcoin::psbt::PsbtParseError),
+    PsbtParse(bitcoin::psbt::PsbtParseError),
     Io(std::io::Error),
     InvalidInputType(InputTypeError),
     InvalidProposedInput(crate::psbt::PrevTxOutError),
-    VersionsDontMatch { proposed: i32, original: i32 },
-    LockTimesDontMatch { proposed: LockTime, original: LockTime },
-    SenderTxinSequenceChanged { proposed: Sequence, original: Sequence },
+    VersionsDontMatch {
+        proposed: i32,
+        original: i32,
+    },
+    LockTimesDontMatch {
+        proposed: LockTime,
+        original: LockTime,
+    },
+    SenderTxinSequenceChanged {
+        proposed: Sequence,
+        original: Sequence,
+    },
     SenderTxinContainsNonWitnessUtxo,
     SenderTxinContainsWitnessUtxo,
     SenderTxinContainsFinalScriptSig,
@@ -32,7 +41,10 @@ pub(crate) enum InternalValidationError {
     ReceiverTxinNotFinalized,
     ReceiverTxinMissingUtxoInfo,
     MixedSequence,
-    MixedInputTypes { proposed: InputType, original: InputType },
+    MixedInputTypes {
+        proposed: InputType,
+        original: InputType,
+    },
     MissingOrShuffledInputs,
     TxOutContainsKeyPaths,
     FeeContributionExceedsMaximum,
@@ -44,6 +56,10 @@ pub(crate) enum InternalValidationError {
     PayeeTookContributedFee,
     FeeContributionPaysOutputSizeIncrease,
     FeeRateBelowMinimum,
+    #[cfg(feature = "v2")]
+    V2(crate::v2::Error),
+    #[cfg(feature = "v2")]
+    Psbt(bitcoin::psbt::Error),
 }
 
 impl From<InternalValidationError> for ValidationError {
@@ -58,7 +74,7 @@ impl fmt::Display for ValidationError {
         use InternalValidationError::*;
 
         match &self.internal {
-            Psbt(e) => write!(f, "couldn't decode PSBT: {}", e),
+            PsbtParse(e) => write!(f, "couldn't decode PSBT: {}", e),
             Io(e) => write!(f, "couldn't read PSBT: {}", e),
             InvalidInputType(e) => write!(f, "invalid transaction input type: {}", e),
             InvalidProposedInput(e) => write!(f, "invalid proposed transaction input: {}", e),
@@ -86,6 +102,10 @@ impl fmt::Display for ValidationError {
             PayeeTookContributedFee => write!(f, "payee tried to take fee contribution for himself"),
             FeeContributionPaysOutputSizeIncrease => write!(f, "fee contribution pays for additional outputs"),
             FeeRateBelowMinimum =>  write!(f, "the fee rate of proposed transaction is below minimum"),
+            #[cfg(feature = "v2")]
+            V2(e) => write!(f, "v2 error: {}", e),
+            #[cfg(feature = "v2")]
+            Psbt(e) => write!(f, "psbt error: {}", e),
         }
     }
 }
@@ -95,7 +115,7 @@ impl std::error::Error for ValidationError {
         use InternalValidationError::*;
 
         match &self.internal {
-            Psbt(error) => Some(error),
+            PsbtParse(error) => Some(error),
             Io(error) => Some(error),
             InvalidInputType(error) => Some(error),
             InvalidProposedInput(error) => Some(error),
@@ -123,6 +143,10 @@ impl std::error::Error for ValidationError {
             PayeeTookContributedFee => None,
             FeeContributionPaysOutputSizeIncrease => None,
             FeeRateBelowMinimum => None,
+            #[cfg(feature = "v2")]
+            V2(error) => Some(error),
+            #[cfg(feature = "v2")]
+            Psbt(error) => Some(error),
         }
     }
 }
@@ -152,6 +176,8 @@ pub(crate) enum InternalCreateRequestError {
     UriDoesNotSupportPayjoin,
     PrevTxOut(crate::psbt::PrevTxOutError),
     InputType(crate::input_type::InputTypeError),
+    #[cfg(feature = "v2")]
+    V2(crate::v2::Error),
 }
 
 impl fmt::Display for CreateRequestError {
@@ -174,6 +200,8 @@ impl fmt::Display for CreateRequestError {
             UriDoesNotSupportPayjoin => write!(f, "the URI does not support payjoin"),
             PrevTxOut(e) => write!(f, "invalid previous transaction output: {}", e),
             InputType(e) => write!(f, "invalid input type: {}", e),
+            #[cfg(feature = "v2")]
+            V2(e) => write!(f, "v2 error: {}", e),
         }
     }
 }
@@ -198,6 +226,8 @@ impl std::error::Error for CreateRequestError {
             UriDoesNotSupportPayjoin => None,
             PrevTxOut(error) => Some(error),
             InputType(error) => Some(error),
+            #[cfg(feature = "v2")]
+            V2(error) => Some(error),
         }
     }
 }
diff --git a/payjoin/src/send/mod.rs b/payjoin/src/send/mod.rs
diff --git a/payjoin/src/v2.rs b/payjoin/src/v2.rs

Original file line number	Diff line number	Diff line change
`@@ -7,13 +7,18 @@ pub enum Error {`
`7`	`7`	`BadRequest(RequestError),`
`8`	`8`	`// To be returned as HTTP 500`
`9`	`9`	`Server(Box<dyn error::Error>),`
	`10`	`+ // V2 d/encapsulation failed`
	`11`	`+ #[cfg(feature = "v2")]`
	`12`	`+ V2(crate::v2::Error),`
`10`	`13`	`}`
`11`	`14`
`12`	`15`	`impl fmt::Display for Error {`
`13`	`16`	`fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {`
`14`	`17`	`match &self {`
`15`	`18`	`Self::BadRequest(e) => e.fmt(f),`
`16`	`19`	`Self::Server(e) => write!(f, "Internal Server Error: {}", e),`
	`20`	`+ #[cfg(feature = "v2")]`
	`21`	`+ Self::V2(e) => e.fmt(f),`
`17`	`22`	`}`
`18`	`23`	`}`
`19`	`24`	`}`
`@@ -23,6 +28,8 @@ impl error::Error for Error {`
`23`	`28`	`match &self {`
`24`	`29`	`Self::BadRequest(_) => None,`
`25`	`30`	`Self::Server(e) => Some(e.as_ref()),`
	`31`	`+ #[cfg(feature = "v2")]`
	`32`	`+ Self::V2(e) => Some(e),`
`26`	`33`	`}`
`27`	`34`	`}`
`28`	`35`	`}`
`@@ -31,6 +38,15 @@ impl From<RequestError> for Error {`
`31`	`38`	`fn from(e: RequestError) -> Self { Error::BadRequest(e) }`
`32`	`39`	`}`
`33`	`40`
	`41`	`+impl From<InternalRequestError> for Error {`
	`42`	`+ fn from(e: InternalRequestError) -> Self { Error::BadRequest(e.into()) }`
	`43`	`+}`
	`44`	`+`
	`45`	`+impl From<crate::v2::Error> for Error {`
	`46`	`+ #[cfg(feature = "v2")]`
	`47`	`+ fn from(e: crate::v2::Error) -> Self { Error::V2(e) }`
	`48`	`+}`
	`49`	`+`
`34`	`50`	`/// Error that may occur when the request from sender is malformed.`
`35`	`51`	`///`
`36`	`52`	`/// This is currently opaque type because we aren't sure which variants will stay.`