From a624a8104817329aa8771877ab85502fa4e47b1d Mon Sep 17 00:00:00 2001
From: Trial97 <alexandru.tripon@itsyscom.com>
Date: Tue, 14 Apr 2020 12:42:37 +0300
Subject: [PATCH] Correctly handle error when creating the diameter answer

---
 agents/diamagent.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/agents/diamagent.go b/agents/diamagent.go
index 00a76e9c3..92151056c 100644
--- a/agents/diamagent.go
+++ b/agents/diamagent.go
@@ -256,7 +256,8 @@ func (da *DiameterAgent) handleMessage(c diam.Conn, m *diam.Message) {
 				utils.DiameterAgent, err.Error(), m))
 		writeOnConn(c, diamErr)
 		return
-	} else if !processed {
+	}
+	if !processed {
 		utils.Logger.Warning(
 			fmt.Sprintf("<%s> no request processor enabled, ignoring message %s from %s",
 				utils.DiameterAgent, m, c.RemoteAddr()))
@@ -269,7 +270,8 @@ func (da *DiameterAgent) handleMessage(c diam.Conn, m *diam.Message) {
 		utils.Logger.Warning(
 			fmt.Sprintf("<%s> err: %s, replying to message: %+v",
 				utils.DiameterAgent, err.Error(), m))
-
+		writeOnConn(c, diamErr)
+		return
 	}
 	writeOnConn(c, a)
 }