From 9da15664755a963bc95ae107a64b6811e59674e2 Mon Sep 17 00:00:00 2001
From: SajjadPourali <sajjad@pourali.com>
Date: Sun, 7 Apr 2024 19:12:12 -0400
Subject: [PATCH] Bump version to 0.0.8

---
 Cargo.toml |  2 +-
 README.md  | 24 +++++++++++++-----------
 2 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index c7e0674..233b00e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -2,7 +2,7 @@
 authors = ['Narrowlink <opensource@narrowlink.com>']
 description = 'Asynchronous lightweight userspace implementation of TCP/IP stack for Tun device'
 name = "ipstack"
-version = "0.0.7"
+version = "0.0.8"
 edition = "2021"
 license = "Apache-2.0"
 repository = 'https://github.com/narrowlink/ipstack'
diff --git a/README.md b/README.md
index 75d579f..a32dccc 100644
--- a/README.md
+++ b/README.md
@@ -13,18 +13,19 @@ Unstable, under development.
 ### Usage
 
 ```rust
+use etherparse::{IcmpEchoHeader, Icmpv4Header};
+use ipstack::stream::IpStackStream;
 use std::net::{IpAddr, Ipv4Addr, SocketAddr};
+use tokio::net::TcpStream;
 use udp_stream::UdpStream;
-use tokio::io{AsyncRead, AsyncWrite};
-use etherparse::{IcmpEchoHeader, Icmpv4Header};
 
 #[tokio::main]
-async fn main(){
+async fn main() {
     const MTU: u16 = 1500;
     let ipv4 = Ipv4Addr::new(10, 0, 0, 1);
     let netmask = Ipv4Addr::new(255, 255, 255, 0);
     let mut config = tun2::Configuration::default();
-    config.address(ipv4).netmask(netmask).mtu(MTU as usize).up();
+    config.address(ipv4).netmask(netmask).mtu(MTU).up();
 
     #[cfg(target_os = "linux")]
     config.platform_config(|config| {
@@ -38,26 +39,27 @@ async fn main(){
 
     let mut ipstack_config = ipstack::IpStackConfig::default();
     ipstack_config.mtu(MTU);
-    let mut ip_stack = ipstack::IpStack::new(ipstack_config, tun2::create_as_async(&config).unwrap());
+    let mut ip_stack =
+        ipstack::IpStack::new(ipstack_config, tun2::create_as_async(&config).unwrap());
 
     while let Ok(stream) = ip_stack.accept().await {
         match stream {
             IpStackStream::Tcp(mut tcp) => {
                 let mut rhs = TcpStream::connect("1.1.1.1:80").await.unwrap();
                 tokio::spawn(async move {
-                    let _ = tokio::io::copy_bidirectional(& mut tcp, & mut rhs).await;
+                    let _ = tokio::io::copy_bidirectional(&mut tcp, &mut rhs).await;
                 });
             }
             IpStackStream::Udp(mut udp) => {
                 let addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(1, 1, 1, 1)), 53);
                 let mut rhs = UdpStream::connect(addr).await.unwrap();
                 tokio::spawn(async move {
-                    let _ = tokio::io::copy_bidirectional(& mut udp, & mut rhs).await;
+                    let _ = tokio::io::copy_bidirectional(&mut udp, &mut rhs).await;
                 });
             }
             IpStackStream::UnknownTransport(u) => {
-                if u.src_addr().is_ipv4() && u.ip_protocol() == 1 {
-                    let (icmp_header, req_payload) = Icmpv4Header::from_slice(u.payload())?;
+                if u.src_addr().is_ipv4() && u.ip_protocol() == etherparse::IpNumber(1) {
+                    let (icmp_header, req_payload) = Icmpv4Header::from_slice(u.payload()).unwrap();
                     if let etherparse::Icmpv4Type::EchoRequest(req) = icmp_header.icmp_type {
                         println!("ICMPv4 echo");
                         let echo = IcmpEchoHeader {
@@ -68,13 +70,13 @@ async fn main(){
                         resp.update_checksum(req_payload);
                         let mut payload = resp.to_bytes().to_vec();
                         payload.extend_from_slice(req_payload);
-                        u.send(payload)?;
+                        u.send(payload).unwrap();
                     } else {
                         println!("ICMPv4");
                     }
                     continue;
                 }
-                println!("unknown transport - Ip Protocol {}", u.ip_protocol());
+                println!("unknown transport - Ip Protocol {}", u.ip_protocol().0);
                 continue;
             }
             IpStackStream::UnknownNetwork(pkt) => {