Merge pull request #1525 from BishopFox/fix-c2-db-duplication

Fix C2 URL Duplication in Database
BishopFox · Dec 22, 2023 · b4cf848 · b4cf848
2 parents a84f13e + de7d466
commit b4cf848
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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