Skip to content

Commit

Permalink
Bump hyper version to 1.0
Browse files Browse the repository at this point in the history
Signed-off-by: csh <[email protected]>
  • Loading branch information
L-jasmine committed Jan 22, 2024
1 parent 2dc4198 commit 41e676b
Show file tree
Hide file tree
Showing 9 changed files with 631 additions and 101 deletions.
5 changes: 3 additions & 2 deletions .cargo/config.toml
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
[build]
target="wasm32-wasi"
target = "wasm32-wasi"
rustflags = "--cfg tokio_unstable"

[target.wasm32-wasi]
runner = "wasmedge"
runner = "wasmedge"
15 changes: 10 additions & 5 deletions client-https/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,14 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
hyper_wasi = { version = "0.15", features = ["full"]}
http-body-util = "0.1.0-rc.2"
tokio_wasi = { version = "1", features = ["rt", "macros", "net", "time", "io-util"]}
hyper = { version = "1", features = ["full"] }
tokio = { version = "1", features = ["rt", "macros", "net", "time", "io-util"] }
pretty_env_logger = "0.4.0"
wasmedge_rustls_api = { version = "0.1", features = [ "tokio_async" ] }
wasmedge_hyper_rustls = "0.1.0"

wasmedge_wasi_socket = "0.5"
pin-project = "1.1.3"
http-body-util = "0.1.0"

tokio-rustls = "0.25.0"
webpki-roots = "0.26.0"
rustls = "0.22.2"
161 changes: 148 additions & 13 deletions client-https/src/main.rs
Original file line number Diff line number Diff line change
@@ -1,30 +1,165 @@
use hyper::Client;
#![deny(warnings)]
#![warn(rust_2018_idioms)]

type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
// use tokio::io::{self, AsyncWriteExt as _};

use std::{
os::fd::{FromRawFd, IntoRawFd},
pin::Pin,
sync::Arc,
task::{Context, Poll},
};

use http_body_util::{BodyExt, Empty};
use hyper::{body::Bytes, Request};

use rustls::pki_types::ServerName;
use tokio::net::TcpStream;

type MainResult<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;

#[tokio::main(flavor = "current_thread")]
async fn main() {
async fn main() -> MainResult<()> {
pretty_env_logger::init();

let url = "https://httpbin.org/get?msg=WasmEdge"
.parse::<hyper::Uri>()
.unwrap();
fetch_https_url(url).await.unwrap();
fetch_https_url(url).await
}

use pin_project::pin_project;
use tokio_rustls::TlsConnector;

#[pin_project]
#[derive(Debug)]
struct TokioIo<T> {
#[pin]
inner: T,
}

impl<T> TokioIo<T> {
pub fn new(inner: T) -> Self {
Self { inner }
}

#[allow(dead_code)]
pub fn inner(self) -> T {
self.inner
}
}

impl<T> hyper::rt::Read for TokioIo<T>
where
T: tokio::io::AsyncRead,
{
fn poll_read(
self: std::pin::Pin<&mut Self>,
cx: &mut std::task::Context<'_>,
mut buf: hyper::rt::ReadBufCursor<'_>,
) -> std::task::Poll<Result<(), std::io::Error>> {
let n = unsafe {
let mut tbuf = tokio::io::ReadBuf::uninit(buf.as_mut());
match tokio::io::AsyncRead::poll_read(self.project().inner, cx, &mut tbuf) {
Poll::Ready(Ok(())) => tbuf.filled().len(),
other => return other,
}
};

unsafe {
buf.advance(n);
}
Poll::Ready(Ok(()))
}
}

async fn fetch_https_url(url: hyper::Uri) -> Result<()> {
let https = wasmedge_hyper_rustls::connector::new_https_connector(
wasmedge_rustls_api::ClientConfig::default(),
);
let client = Client::builder().build::<_, hyper::Body>(https);
impl<T> hyper::rt::Write for TokioIo<T>
where
T: tokio::io::AsyncWrite,
{
fn poll_write(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &[u8],
) -> Poll<Result<usize, std::io::Error>> {
tokio::io::AsyncWrite::poll_write(self.project().inner, cx, buf)
}

fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), std::io::Error>> {
tokio::io::AsyncWrite::poll_flush(self.project().inner, cx)
}

fn poll_shutdown(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
) -> Poll<Result<(), std::io::Error>> {
tokio::io::AsyncWrite::poll_shutdown(self.project().inner, cx)
}

fn is_write_vectored(&self) -> bool {
tokio::io::AsyncWrite::is_write_vectored(&self.inner)
}

fn poll_write_vectored(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
bufs: &[std::io::IoSlice<'_>],
) -> Poll<std::prelude::v1::Result<usize, std::io::Error>> {
tokio::io::AsyncWrite::poll_write_vectored(self.project().inner, cx, bufs)
}
}

async fn fetch_https_url(url: hyper::Uri) -> MainResult<()> {
let host = url.host().expect("uri has no host");
let port = url.port_u16().unwrap_or(443);
let addr = format!("{}:{}", host, port);

let mut root_store = rustls::RootCertStore::empty();
root_store.extend(webpki_roots::TLS_SERVER_ROOTS.iter().cloned());

let config = rustls::ClientConfig::builder()
.with_root_certificates(root_store)
.with_no_client_auth();

let connector = TlsConnector::from(Arc::new(config));
let stream = unsafe {
let fd = wasmedge_wasi_socket::TcpStream::connect(addr)?.into_raw_fd();
TcpStream::from_std(std::net::TcpStream::from_raw_fd(fd))?
};

let domain = ServerName::try_from(host.to_string()).unwrap();
let stream = connector.connect(domain, stream).await.unwrap();

let io = TokioIo::new(stream);

let (mut sender, conn) = hyper::client::conn::http1::handshake(io).await?;
tokio::task::spawn(async move {
if let Err(err) = conn.await {
println!("Connection failed: {:?}", err);
}
});

let authority = url.authority().unwrap().clone();

let req = Request::builder()
.uri(url)
.header(hyper::header::HOST, authority.as_str())
.body(Empty::<Bytes>::new())?;

let res = client.get(url).await?;
let mut res = sender.send_request(req).await?;

println!("Response: {}", res.status());
println!("Headers: {:#?}\n", res.headers());

let body = hyper::body::to_bytes(res.into_body()).await.unwrap();
println!("{}", String::from_utf8(body.into()).unwrap());
let mut resp_data = Vec::new();
while let Some(next) = res.frame().await {
let frame = next?;
if let Some(chunk) = frame.data_ref() {
resp_data.extend_from_slice(&chunk);
}
}

println!("\n\nDone!");
println!("{}", String::from_utf8_lossy(&resp_data));

Ok(())
}
13 changes: 5 additions & 8 deletions client/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,9 @@ version = "0.1.0"
edition = "2021"

[dependencies]
hyper_wasi = { version = "0.15", features = ["full"] }
tokio_wasi = { version = "1", features = [
"rt",
"macros",
"net",
"time",
"io-util",
] }
hyper = { version = "1", features = ["full"] }
tokio = { version = "1", features = ["rt", "macros", "net", "time", "io-util"] }
pretty_env_logger = "0.4.0"
wasmedge_wasi_socket = "0.5"
pin-project = "1.1.3"
http-body-util = "0.1.0"
Loading

0 comments on commit 41e676b

Please sign in to comment.