From e6bd5b06989964fb8f6dda425d727575a54f7f36 Mon Sep 17 00:00:00 2001
From: rustdesk <info@rustdesk.com>
Date: Sun, 7 Nov 2021 10:18:10 +0800
Subject: [PATCH] pb in version

---
 libs/hbb_common/protos/rendezvous.proto | 3 +++
 src/rendezvous_mediator.rs              | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/libs/hbb_common/protos/rendezvous.proto b/libs/hbb_common/protos/rendezvous.proto
index bd058ac4d..1a1671ce1 100644
--- a/libs/hbb_common/protos/rendezvous.proto
+++ b/libs/hbb_common/protos/rendezvous.proto
@@ -48,6 +48,7 @@ message PunchHoleSent {
   string id = 2;
   string relay_server = 3;
   NatType nat_type = 4;
+  string version = 5;
 }
 
 message RegisterPk {
@@ -110,6 +111,7 @@ message RelayResponse {
     bytes pk = 5;
   }
   string refuse_reason = 6;
+  string version = 7;
 }
 
 message SoftwareUpdate { string url = 1; }
@@ -128,6 +130,7 @@ message LocalAddr {
   bytes local_addr = 2;
   string relay_server = 3;
   string id = 4;
+  string version = 5;
 }
 
 message RendezvousMessage {
diff --git a/src/rendezvous_mediator.rs b/src/rendezvous_mediator.rs
index 13f0f605c..9e1554c6c 100644
--- a/src/rendezvous_mediator.rs
+++ b/src/rendezvous_mediator.rs
@@ -285,6 +285,7 @@ impl RendezvousMediator {
         let mut msg_out = Message::new();
         let mut rr = RelayResponse {
             socket_addr,
+            version: crate::VERSION.to_owned(),
             ..Default::default()
         };
         if initiate {
@@ -321,6 +322,7 @@ impl RendezvousMediator {
             socket_addr: AddrMangle::encode(peer_addr),
             local_addr: AddrMangle::encode(local_addr),
             relay_server,
+            version: crate::VERSION.to_owned(),
             ..Default::default()
         });
         let bytes = msg_out.write_to_bytes()?;
@@ -359,6 +361,7 @@ impl RendezvousMediator {
             id: Config::get_id(),
             relay_server,
             nat_type: nat_type.into(),
+            version: crate::VERSION.to_owned(),
             ..Default::default()
         });
         let bytes = msg_out.write_to_bytes()?;