Merge pull request #71 from narrowlink/DropIpstackHandle

Drop task handle in IpStack

Author: SajjadPourali

src/lib.rs

@@ -83,7 +83,7 @@ impl IpStackConfig {
 
 pub struct IpStack {
     accept_receiver: UnboundedReceiver<IpStackStream>,
-    _handle: JoinHandle<Result<()>>, // Just hold the task handle
+    handle: JoinHandle<Result<()>>,
 }
 
 impl IpStack {
@@ -94,7 +94,7 @@ impl IpStack {
         let (accept_sender, accept_receiver) = mpsc::unbounded_channel::<IpStackStream>();
         IpStack {
             accept_receiver,
-            _handle: run(config, device, accept_sender),
+            handle: run(config, device, accept_sender),
         }
     }
 
@@ -103,6 +103,12 @@ impl IpStack {
     }
 }
 
+impl Drop for IpStack {
+    fn drop(&mut self) {
+        self.handle.abort();
+    }
+}
+
 fn run<Device: AsyncRead + AsyncWrite + Unpin + Send + 'static>(
     config: IpStackConfig,
     mut device: Device,