additional param on agent module req ctx

zarkones · Feb 18, 2024 · 6b512ee · 6b512ee
1 parent 1f5c2f6
commit 6b512ee
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/modules.go b/modules.go
@@ -12,15 +12,16 @@ import (
 )
 
 type AgentModuleReqCtx struct {
-	ModuleName string `json:"moduleName"` // Example: HELLO_WORLD.dll
+	AgentID    string `json:"agentId"`
+	ModuleName string `json:"moduleName"` // Example: HELLO_WORLD.DLL
 }
 
-func AgentDownloadModule(moduleName string, decryptionKey *rsa.PrivateKey) (module string, err error) {
+func AgentDownloadModule(agentID, moduleName string, decryptionKey *rsa.PrivateKey) (module string, err error) {
 	if TrustedPubKey == nil {
 		return "", ErrKeyIsNil
 	}
 
-	payload, err := json.Marshal(AgentModuleReqCtx{moduleName})
+	payload, err := json.Marshal(AgentModuleReqCtx{AgentID: agentID, ModuleName: moduleName})
 	if err != nil {
 		return "", err
 	}