Skip to content

Commit

Permalink
fix(transport): Use Uri host if no domain for tls (#244)
Browse files Browse the repository at this point in the history
* fix(transport): Use Uri host if no domain for tls

* Fix warnings
  • Loading branch information
LucioFranco authored Jan 20, 2020
1 parent 5fc6762 commit 6de0b4d
Show file tree
Hide file tree
Showing 2 changed files with 71 additions and 8 deletions.
3 changes: 2 additions & 1 deletion tonic/src/transport/channel/tls.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
use crate::transport::{
service::TlsConnector,
tls::{Certificate, Identity},
Error,
};
use http::Uri;
use std::fmt;
Expand Down Expand Up @@ -81,7 +82,7 @@ impl ClientTlsConfig {

pub(crate) fn tls_connector(&self, uri: Uri) -> Result<TlsConnector, crate::Error> {
let domain = match &self.domain {
None => uri.to_string(),
None => uri.host().ok_or(Error::new_invalid_uri())?.to_string(),
Some(domain) => domain.clone(),
};
match &self.rustls_raw {
Expand Down
76 changes: 69 additions & 7 deletions tonic/src/transport/error.rs
Original file line number Diff line number Diff line change
@@ -1,23 +1,85 @@
use std::{error, fmt};
// TODO: remove this when we use errors throughout,
// there are some that are only used under the TLS feature
// these probably should be scoped to a `TLSError` enum.
#![allow(dead_code)]

use std::{error::Error as StdError, fmt};

type Source = Box<dyn StdError + Send + Sync + 'static>;

/// Error's that originate from the client or server;
pub struct Error {
inner: ErrorImpl,
}

struct ErrorImpl {
kind: Kind,
source: Option<Source>,
}

#[derive(Debug)]
pub struct Error(crate::Error);
pub(crate) enum Kind {
Transport,
InvalidUri,
}

impl Error {
pub(crate) fn new(kind: Kind) -> Self {
Self {
inner: ErrorImpl { kind, source: None },
}
}

pub(crate) fn with(mut self, source: impl Into<Source>) -> Self {
self.inner.source = Some(source.into());
self
}

pub(crate) fn from_source(source: impl Into<crate::Error>) -> Self {
Self(source.into())
Error::new(Kind::Transport).with(source)
}

pub(crate) fn new_invalid_uri() -> Self {
Error::new(Kind::InvalidUri)
}

fn description(&self) -> &str {
match &self.inner.kind {
Kind::Transport => "transport error",
Kind::InvalidUri => "invalid URI",
}
}
}

impl fmt::Debug for Error {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let mut f = f.debug_tuple("tonic::transport::Error");

f.field(&self.inner.kind);

if let Some(source) = &self.inner.source {
f.field(source);
}

f.finish()
}
}

impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.0.fmt(f)
if let Some(source) = &self.inner.source {
write!(f, "{}: {}", self.description(), source)
} else {
f.write_str(self.description())
}
}
}

impl error::Error for Error {
fn source(&self) -> Option<&(dyn error::Error + 'static)> {
self.0.source()
impl StdError for Error {
fn source(&self) -> Option<&(dyn StdError + 'static)> {
self.inner
.source
.as_ref()
.map(|source| &**source as &(dyn StdError + 'static))
}
}

0 comments on commit 6de0b4d

Please sign in to comment.