From 4eacea7d571560c754f516bf79f6072900804407 Mon Sep 17 00:00:00 2001
From: emanuele-f <black.silver@hotmail.it>
Date: Mon, 10 Jun 2019 00:32:38 +0200
Subject: [PATCH] Little improvement

---
 edge_utils.c | 4 +++-
 sn.c         | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/edge_utils.c b/edge_utils.c
index d93a785..8fe7032 100644
--- a/edge_utils.c
+++ b/edge_utils.c
@@ -1503,6 +1503,9 @@ static void readFromIPSocket(n2n_edge_t * eee, int in_sock) {
 	     */
 	    traceEvent(TRACE_DEBUG, "Got P2P register");
 	    find_and_remove_peer(&eee->pending_peers, reg.srcMac);
+
+	    /* NOTE: only ACK to peers */
+	    send_register_ack(eee, orig_sender, &reg);
 	  }
 
 	  traceEvent(TRACE_INFO, "Rx REGISTER src=%s dst=%s from peer %s (%s)",
@@ -1512,7 +1515,6 @@ static void readFromIPSocket(n2n_edge_t * eee, int in_sock) {
 		     sock_to_cstr(sockbuf2, orig_sender));
 
 	  check_peer_registration_needed(eee, from_supernode, reg.srcMac, orig_sender);
-	  send_register_ack(eee, orig_sender, &reg);
 	  break;
       }
       case MSG_TYPE_REGISTER_ACK:
diff --git a/sn.c b/sn.c
index e49e8d5..0e41826 100644
--- a/sn.c
+++ b/sn.c
@@ -706,6 +706,8 @@ static int process_udp(n2n_sn_t * sss,
         traceEvent( TRACE_DEBUG, "Ignoring QUERY_PEER for unknown edge %s",
                     macaddr_str( mac_buf, query.targetMac ) );
     }
+
+    break;
   }
   default:
     /* Not a known message type */