From ebb357245a9d0bd5ec2c1656458a80e14fa45503 Mon Sep 17 00:00:00 2001
From: Logan oos Even <46396513+Logan007@users.noreply.github.com>
Date: Sat, 2 Oct 2021 23:17:45 +0545
Subject: [PATCH] added a flag field to QUERY_PEER (#835)

---
 include/n2n_typedefs.h | 1 +
 src/edge_utils.c       | 2 +-
 src/wire.c             | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/include/n2n_typedefs.h b/include/n2n_typedefs.h
index 31d5d9d..7ba234e 100644
--- a/include/n2n_typedefs.h
+++ b/include/n2n_typedefs.h
@@ -421,6 +421,7 @@ typedef struct n2n_PEER_INFO {
 
 
 typedef struct n2n_QUERY_PEER {
+    uint16_t                      aflags;
     n2n_mac_t                     srcMac;
     n2n_sock_t                    sock;
     n2n_mac_t                     targetMac;
diff --git a/src/edge_utils.c b/src/edge_utils.c
index ce87743..552a755 100644
--- a/src/edge_utils.c
+++ b/src/edge_utils.c
@@ -1148,7 +1148,7 @@ void send_query_peer (n2n_edge_t * eee,
     uint8_t pktbuf[N2N_PKT_BUF_SIZE];
     size_t idx;
     n2n_common_t cmn = {0};
-    n2n_QUERY_PEER_t query = {{0}};
+    n2n_QUERY_PEER_t query = {0};
     struct peer_info *peer, *tmp;
     int n_o_pings = 0;
     int n_o_top_sn = 0;
diff --git a/src/wire.c b/src/wire.c
index f061952..bac76e3 100644
--- a/src/wire.c
+++ b/src/wire.c
@@ -705,6 +705,7 @@ int encode_QUERY_PEER (uint8_t * base,
     retval += encode_common(base, idx, common);
     retval += encode_mac(base, idx, pkt->srcMac);
     retval += encode_mac(base, idx, pkt->targetMac);
+    retval += encode_uint16(base, idx, pkt->aflags);
 
     return retval;
 }
@@ -720,6 +721,7 @@ int decode_QUERY_PEER (n2n_QUERY_PEER_t * pkt,
 
     retval += decode_mac(pkt->srcMac, base, rem, idx);
     retval += decode_mac(pkt->targetMac, base, rem, idx);
+    retval += decode_uint16(&(pkt->aflags), base, rem, idx);
 
     return retval;
 }