ppp-stream's Introduction

ppp-stream

This is a simple library that adds helper functions to AsyncRead streams for HAProxy Protocol extraction. The Proxy Protocol parsing is implemented by the great ppp library, many thanks to @misalcedo and all the other contributors.

Also, big thanks to @simao for his contribution to this project.

ppp-stream's People

Contributors

Stargazers

Watchers

ppp-stream's Issues

Implementing AsyncWrite

Hi,

I am trying to implement AsyncWrite for PPPRefStream, so I can run the following code:

    let mut local = stream.remote_addr_unpin().await?;

    let mut remote = TcpStream::connect("myip:22").await.context("could not connect to remote")?;

    tokio::io::copy_bidirectional(&mut local, &mut remote).await.context("copying streams")?;

I have the following implementation:

impl <'a, T> AsyncWrite for PPPRefStream<'a, T> where T: AsyncWrite {
    fn poll_write(
        self: Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
        buf: &[u8],
    ) -> std::task::Poll<std::result::Result<usize, std::io::Error>> {
        let this = self.get_mut();
        this.inner.as_mut().poll_write(cx, buf)
    }

    fn poll_flush(self: Pin<&mut Self>, cx: &mut std::task::Context<'_>) -> std::task::Poll<std::result::Result<(), std::io::Error>> {
        let this = self.get_mut();
        this.inner.as_mut().poll_flush(cx)
    }

    fn poll_shutdown(self: Pin<&mut Self>, cx: &mut std::task::Context<'_>) -> std::task::Poll<std::result::Result<(), std::io::Error>> {
        let this = self.get_mut();
        this.inner.as_mut().poll_shutdown(cx)
    }
}

This works sometimes. It always works with http, proxying the request, but with ssh sometimes I get errors when trying to connect, like:

ssh client error:
Bad packet length 1231976033.

rust app error:
thread 'tokio-runtime-worker' panicked at 'called `Result::unwrap()` on an `Err` value: copying streams

Caused by:
    Connection reset by peer (os error 104)', src/main.rs:55:34

Am I missing something in my implementation?

Thanks

Recommend Projects