Adding nil check to listener deletion to prevent panic

BishopFox · May 14, 2024 · 4082378 · 4082378
1 parent df65d7f
commit 4082378
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/db/helpers.go b/server/db/helpers.go
@@ -27,6 +27,7 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"errors"
+	"fmt"
 	"os"
 	"path/filepath"
 	"strings"
@@ -599,6 +600,9 @@ func DeleteListener(JobID uint32) error {
 	if result.Error != nil {
 		return result.Error
 	}
+	if listener == nil {
+		return fmt.Errorf("Job ID %d not found in database", JobID)
+	}
 	listenerID := listener.ID
 	switch listener.Type {
 	case constants.HttpStr, constants.HttpsStr: