34 lines
920 B
Go
34 lines
920 B
Go
package extender
|
|
|
|
import "github.com/Adaptix-Framework/axc2"
|
|
|
|
func (ex *AdaptixExtender) ExAgentGenerate(agentName string, generateConfig adaptix.BuildProfile) ([]byte, string, error) {
|
|
module, err := ex.getAgentModule(agentName)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
agentProfile, err := module.GenerateProfiles(generateConfig)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
return module.BuildPayload(generateConfig, agentProfile)
|
|
}
|
|
|
|
func (ex *AdaptixExtender) ExAgentCreate(agentName string, beat []byte) (adaptix.AgentData, adaptix.ExtenderAgent, error) {
|
|
module, err := ex.getAgentModule(agentName)
|
|
if err != nil {
|
|
return adaptix.AgentData{}, nil, err
|
|
}
|
|
return module.CreateAgent(beat)
|
|
}
|
|
|
|
func (ex *AdaptixExtender) ExAgentGetExtender(agentName string) (adaptix.ExtenderAgent, error) {
|
|
module, err := ex.getAgentModule(agentName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return module.GetExtender(), nil
|
|
}
|