From de7d466acbb75990d2b311f37225480fc044ef16 Mon Sep 17 00:00:00 2001
From: Raf <84349012+RafBishopFox@users.noreply.github.com>
Date: Fri, 22 Dec 2023 12:23:19 -0500
Subject: [PATCH] Fixing an issue where C2 URLs are duplicated every time a
 profile is used to generate an implant.

---
 server/db/models/implant.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/server/db/models/implant.go b/server/db/models/implant.go
index 33db497915..cccda8d286 100644
--- a/server/db/models/implant.go
+++ b/server/db/models/implant.go
@@ -349,9 +349,11 @@ type ImplantC2 struct {
 
 // BeforeCreate - GORM hook
 func (c2 *ImplantC2) BeforeCreate(tx *gorm.DB) (err error) {
-	c2.ID, err = uuid.NewV4()
-	if err != nil {
-		return err
+	if c2.ID == uuid.Nil {
+		c2.ID, err = uuid.NewV4()
+		if err != nil {
+			return err
+		}
 	}
 	c2.CreatedAt = time.Now()
 	return nil