From 0d4a2a751ef30a0fd141c91cadb44988a5c649be Mon Sep 17 00:00:00 2001
From: "Lawrence, Rendall" <rlawrence@sot-te.ch>
Date: Sun, 24 Apr 2022 21:37:57 +0300
Subject: [PATCH] (tested) fix wait group race in udp frontend

---
 frontend/udp/frontend.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/frontend/udp/frontend.go b/frontend/udp/frontend.go
index 565c33e..e67ecce 100644
--- a/frontend/udp/frontend.go
+++ b/frontend/udp/frontend.go
@@ -133,6 +133,7 @@ func NewFrontend(logic frontend.TrackerLogic, c conf.MapConfig) (*Frontend, erro
 		return nil, err
 	}
 
+	f.wg.Add(1)
 	go func() {
 		if err := f.serve(); err != nil {
 			log.Fatal("failed while serving udp", log.Err(err))
@@ -175,8 +176,6 @@ func (t *Frontend) listen() error {
 // until Stop() is called or an error is returned.
 func (t *Frontend) serve() error {
 	pool := bytepool.New(2048)
-
-	t.wg.Add(1)
 	defer t.wg.Done()
 
 	for {