From 1ddd5203e58ac2f8ee0dd5e58b10bf8d3eb39be0 Mon Sep 17 00:00:00 2001
From: rustdesk <info@rustdesk.com>
Date: Wed, 20 Oct 2021 17:03:48 +0800
Subject: [PATCH] fix overflow

---
 src/common.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/common.rs b/src/common.rs
index f67751236..e5a7e4f99 100644
--- a/src/common.rs
+++ b/src/common.rs
@@ -1,6 +1,6 @@
 pub use arboard::Clipboard as ClipboardContext;
 use hbb_common::{
-    allow_err,
+    allow_err, bail,
     compress::{compress as compress_func, decompress},
     config::{Config, COMPRESS_LEVEL, RENDEZVOUS_TIMEOUT},
     log,
@@ -240,6 +240,10 @@ async fn test_nat_type_() -> ResultType<bool> {
     let rendezvous_server = get_rendezvous_server(100).await;
     let server1 = rendezvous_server;
     let mut server2 = server1;
+    if server1.port() == 0 { // offline
+        // avoid overflow crash
+        bail!("Offline");
+    }
     server2.set_port(server1.port() - 1);
     let mut msg_out = RendezvousMessage::new();
     let serial = Config::get_serial();