bug fix for command augment containers

CHANGELOG.MD

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [3.3.1-rc39] - 2025-01-30
+
+### Changed
+
+- Fixed a syncing bug that wasn't auto adding commands to new callbacks for command augmentation containers
+
 ## [3.3.1-rc38] - 2025-01-30
 
 ### Changed

VERSION

@@ -1 +1 @@
-3.3.1-rc38
+3.3.1-rc39

mythic-docker/src/VERSION

@@ -1 +1 @@
-3.3.1-rc38
+3.3.1-rc39

mythic-docker/src/rabbitmq/recv_pt_sync.go

@@ -403,7 +403,7 @@ func payloadTypeSync(in PayloadTypeSyncMessage) error {
 	if !in.ForcedSync {
 		go CreateGraphQLSpectatorAPITokenAndSendOnStartMessage(payloadtype.Name)
 		if in.PayloadType.AgentType == "command_augment" {
-			go updateAllCallbacksWithCommandAugments(in.PayloadType.Name)
+			go updateAllCallbacksWithCommandAugments()
 		}
 	}
 	return nil
@@ -1279,19 +1279,21 @@ func updatePayloadBuildSteps(in PayloadTypeSyncMessage, payloadtype databaseStru
 	}
 }
 
-func updateAllCallbacksWithCommandAugments(payloadTypeName string) {
+func updateAllCallbacksWithCommandAugments() {
 	callbacks := []databaseStructs.Callback{}
 	err := database.DB.Select(&callbacks, `SELECT
     	callback.id, callback.operator_id,
-    	payload.os "payload.os"
+    	payload.os "payload.os",
+    	payloadtype.Name "payload.payloadtype.name"
     	FROM callback
-    	JOIN payload on callback.registered_payload_id = payload.id`)
+    	JOIN payload on callback.registered_payload_id = payload.id
+    	JOIN payloadtype ON payload.payload_type_id = payloadtype.id`)
 	if err != nil {
 		logging.LogError(err, "failed to get callbacks for updating with newly synced command augmentation container")
 		return
 	}
 	for _, callback := range callbacks {
-		addCommandAugmentsToCallback(callback.ID, callback.Payload.Os, payloadTypeName, callback.OperatorID)
+		addCommandAugmentsToCallback(callback.ID, callback.Payload.Os, callback.Payload.Payloadtype.Name, callback.OperatorID)
 	}
 
 }