agentstructs

type BrowserScript struct {
	ScriptPath     string `json:"-"`
	Author         string `json:"author"`
	ScriptContents string `json:"script"`
}

type BuildParameter struct {
	Name              string                     `json:"name"`
	Description       string                     `json:"description"`
	Required          bool                       `json:"required"`
	VerifierRegex     string                     `json:"verifier_regex"`
	DefaultValue      interface{}                `json:"default_value"`
	ParameterType     BuildParameterType         `json:"parameter_type"`
	FormatString      string                     `json:"format_string"`
	Randomize         bool                       `json:"randomize"`
	IsCryptoType      bool                       `json:"crypto_type"`
	Choices           []string                   `json:"choices"`
	DictionaryChoices []BuildParameterDictionary `json:"dictionary_choices"`
}

type BuildParameterDictionary C2ParameterDictionary

type BuildParameterType = string

type BuildStep struct {
	Name        string `json:"step_name"`
	Description string `json:"step_description"`
}

type C2ParameterDictionary struct {
	Name         string `json:"name"`
	DefaultValue string `json:"default_value"`
	DefaultShow  bool   `json:"default_show"`
}

type Command struct {
	Name                           string                              `json:"name"`
	NeedsAdminPermissions          bool                                `json:"needs_admin_permissions"`
	HelpString                     string                              `json:"help_string"`
	Description                    string                              `json:"description"`
	Version                        uint32                              `json:"version"`
	SupportedUIFeatures            []string                            `json:"supported_ui_features"`
	Author                         string                              `json:"author"`
	MitreAttackMappings            []string                            `json:"attack"`
	ScriptOnlyCommand              bool                                `json:"script_only"`
	CommandAttributes              CommandAttribute                    `json:"attributes"`
	CommandParameters              []CommandParameter                  `json:"parameters"`
	AssociatedBrowserScript        *BrowserScript                      `json:"browserscript,omitempty"`
	TaskFunctionOPSECPre           PtTaskFunctionOPSECPre              `json:"-"`
	TaskFunctionCreateTasking      PtTaskFunctionCreateTasking         `json:"-"`
	TaskFunctionProcessResponse    PtTaskFunctionProcessResponse       `json:"-"`
	TaskFunctionOPSECPost          PtTaskFunctionOPSECPost             `json:"-"`
	TaskFunctionParseArgString     PtTaskFunctionParseArgString        `json:"-"`
	TaskFunctionParseArgDictionary PtTaskFunctionParseArgDictionary    `json:"-"`
	TaskCompletionFunctions        map[string]PTTaskCompletionFunction `json:"-"`
}

type CommandAttribute struct {
	CommandIsInjectableIntoProcess                  bool              `json:"spawn_and_injectable"`
	SupportedOS                                     []string          `json:"supported_os"`
	CommandIsBuiltin                                bool              `json:"builtin"`
	CommandIsSuggested                              bool              `json:"suggested_command"`
	CommandCanOnlyBeLoadedLater                     bool              `json:"load_only"`
	FilterCommandAvailabilityByAgentBuildParameters map[string]string `json:"filter_by_build_parameter"`
	AdditionalAttributes                            map[string]string `json:"additional_items"`
}

type CommandParameter struct {
	Name                                    string                        `json:"name"`
	ModalDisplayName                        string                        `json:"display_name"`
	CLIName                                 string                        `json:"cli_name"`
	ParameterType                           CommandParameterType          `json:"parameter_type"`
	Description                             string                        `json:"description"`
	Choices                                 []string                      `json:"choices"`
	DefaultValue                            interface{}                   `json:"default_value"`
	SupportedAgents                         []string                      `json:"supported_agents"`
	SupportedAgentBuildParameters           map[string]string             `json:"supported_agent_build_parameters"`
	ChoicesAreAllCommands                   bool                          `json:"choices_are_all_commands"`
	ChoicesAreLoadedCommands                bool                          `json:"choices_are_loaded_commands"`
	FilterCommandChoicesByCommandAttributes map[string]string             `json:"choice_filter_by_command_attributes"`
	DynamicQueryFunction                    PTTaskingDynamicQueryFunction `json:"dynamic_query_function"`
	ParameterGroupInformation               []ParameterGroupInfo          `json:"parameter_group_info"`
	// contains filtered or unexported fields
}

type CommandParameterType = string

type FileBrowserTask struct {
	Path     string `json:"path" mapstructure:"path"`
	FullPath string `json:"full_path" mapstructure:"full_path"`
	Filename string `json:"file" mapstructure:"file"`
	Host     string `json:"host" mapstructure:"host"`
}

type OPSEC_ROLE string

type PAYLOAD_BUILD_STATUS = string

type PTRPCDynamicQueryFunctionMessage struct {
	Command       string `json:"command" binding:"required"`
	ParameterName string `json:"parameter_name" binding:"required"`
	PayloadType   string `json:"payload_type" binding:"required"`
	Callback      int    `json:"callback" binding:"required"`
}

type PTRPCDynamicQueryFunctionMessageResponse struct {
	Success bool     `json:"success"`
	Error   string   `json:"error"`
	Choices []string `json:"choices"`
}

type PTRPCOtherServiceRPCMessage struct {
	Name                 string                 `json:"service_name"` //required
	RPCFunction          string                 `json:"service_function"`
	RPCFunctionArguments map[string]interface{} `json:"service_arguments"`
}

type PTRPCOtherServiceRPCMessageResponse struct {
	Success bool                   `json:"success"`
	Error   string                 `json:"error"`
	Result  map[string]interface{} `json:"result"`
}

type PTRPCReSyncMessage struct {
	Name string `json:"payload_type"`
}

type PTRPCReSyncMessageResponse struct {
	Success bool   `json:"success"`
	Error   string `json:"error"`
}

type PTTTaskOPSECPreTaskMessageResponse struct {
	TaskID             int        `json:"task_id"`
	Success            bool       `json:"success"`
	Error              string     `json:"error"`
	OpsecPreBlocked    bool       `json:"opsec_pre_blocked"`
	OpsecPreMessage    string     `json:"opsec_pre_message"`
	OpsecPreBypassed   *bool      `json:"opsec_pre_bypassed,omitempty"`
	OpsecPreBypassRole OPSEC_ROLE `json:"opsec_pre_bypass_role"`
}

type PTTaskCompletionFunction func(*PTTaskMessageAllData, *PTTaskMessageAllData, *SubtaskGroupName) PTTaskCompletionFunctionMessageResponse

type PTTaskCompletionFunctionMessage struct {
	TaskData               *PTTaskMessageAllData `json:"task"`
	SubtaskData            *PTTaskMessageAllData `json:"subtask,omitempty"`
	SubtaskGroup           *SubtaskGroupName     `json:"subtask_group_name,omitempty"`
	CompletionFunctionName string                `json:"function_name"`
}

type PTTaskCompletionFunctionMessageResponse struct {
	TaskID                 int     `json:"task_id"`
	ParentTaskId           int     `json:"parent_task_id"`
	Success                bool    `json:"success"`
	Error                  string  `json:"error"`
	TaskStatus             *string `json:"task_status,omitempty"`
	DisplayParams          *string `json:"display_params,omitempty"`
	Stdout                 *string `json:"stdout,omitempty"`
	Stderr                 *string `json:"stderr,omitempty"`
	Completed              *bool   `json:"completed,omitempty"`
	TokenID                *int    `json:"token_id,omitempty"`
	CompletionFunctionName *string `json:"completion_function_name,omitempty"`
	Params                 *string `json:"params,omitempty"`
	ParameterGroupName     *string `json:"parameter_group_name,omitempty"`
}

type PTTaskCreateTaskingMessageResponse struct {
	// TaskID - the task associated with the create tasking function - this will be automatically filled in for you
	TaskID int `json:"task_id"`
	// Success - indicate if the create tasking function ran successfully or not
	Success bool `json:"success"`
	// Error - if you want to provide an error message about some error you hit while executing the create tasking
	Error string `json:"error"`
	// CommandName - if you want to change the associated command name that's sent down to the agent
	CommandName *string `json:"command_name,omitempty"`
	// TaskStatus - if you want to manually set the task status to be something other than default
	TaskStatus *string `json:"task_status,omitempty"`
	// DisplayParams - if you want to change the display parameters for your task to be something other than the default JSON
	DisplayParams *string `json:"display_params,omitempty"`
	// Stdout - Provide any task-based stdout
	Stdout *string `json:"stdout,omitempty"`
	// Stderr - Provide any task-based stderr
	Stderr *string `json:"stderr,omitempty"`
	// Completed - identify if the task is already completed and shouldn't be sent down to the agent
	Completed *bool `json:"completed,omitempty"`
	// TokenID - identifier for the token id associated with this task - normally doesn't need to be set unless you're changing it
	TokenID *uint64 `json:"token_id,omitempty"`
	// CompletionFunctionName - name of the completion function to call from the Command's TaskCompletionFunctions dictionary
	CompletionFunctionName *string `json:"completion_function_name,omitempty"`
	// ParameterGroupName - manually give the parameter group name instead of having Mythic determine it
	ParameterGroupName string `json:"parameter_group_name"`
}

type PTTaskMessageAllData struct {
	// Task - Read-only data about the task
	Task PTTaskMessageTaskData `json:"task"`
	// Callback - Read-only data about the callback
	Callback PTTaskMessageCallbackData `json:"callback"`
	// BuildParameters - Read-only data about the build parameters
	BuildParameters []PayloadConfigurationBuildParameter `json:"build_parameters"`
	// Commands - Read-only data about the commands built into the callback
	Commands []string `json:"commands"`
	// Payload - Read-only data about the backing payload for this task
	Payload PTTaskMessagePayloadData `json:"payload"`
	// C2Profiles - Read-only data about the c2 profiles and their values for this callback
	C2Profiles []PayloadConfigurationC2Profile `json:"c2info"`
	// PayloadType - Read-only the name of the payload type associated with this task
	PayloadType string `json:"payload_type"`
	// Args - Read-Write argument data for adding/removing/modifying args associated with this task instance.
	// Mainly for create tasking function to augment parameters
	Args PTTaskMessageArgsData
}

type PTTaskMessageArgsData struct {
	// contains filtered or unexported fields
}

type PTTaskMessageCallbackData struct {
	ID                  int    `json:"id"`
	AgentCallbackID     string `json:"agent_callback_id"`
	InitCallback        string `json:"init_callback"`
	LastCheckin         string `json:"last_checkin"`
	User                string `json:"user"`
	Host                string `json:"host"`
	PID                 int    `json:"pid"`
	Ip                  string `json:"ip"`
	ExternalIp          string `json:"external_ip"`
	ProcessName         string `json:"process_name"`
	Description         string `json:"description"`
	OperatorID          int    `json:"operator_id"`
	Active              bool   `json:"active"`
	RegisteredPayloadID int    `json:"registered_payload_id"`
	IntegrityLevel      int    `json:"integrity_level"`
	Locked              bool   `json:"locked"`
	OperationID         int    `json:"operation_id"`
	CryptoType          string `json:"crypto_type"`
	DecKey              []byte `json:"dec_key"`
	EncKey              []byte `json:"enc_key"`
	Os                  string `json:"os"`
	Architecture        string `json:"architecture"`
	Domain              string `json:"domain"`
	ExtraInfo           string `json:"extra_info"`
	SleepInfo           string `json:"sleep_info"`
}

type PTTaskMessagePayloadData struct {
	Os          string `json:"os"`
	UuID        string `json:"uuid"`
	PayloadType string `json:"payload_type"`
}

type PTTaskMessageTaskData struct {
	ID                                 int    `json:"id"`
	AgentTaskID                        string `json:"agent_task_id"`
	CommandName                        string `json:"command_name"`
	Params                             string `json:"params"`
	Timestamp                          string `json:"timestamp"`
	CallbackID                         int    `json:"callback_id"`
	Status                             string `json:"status"`
	OriginalParams                     string `json:"original_params"`
	DisplayParams                      string `json:"display_params"`
	Comment                            string `json:"comment"`
	Stdout                             string `json:"stdout"`
	Stderr                             string `json:"stderr"`
	Completed                          bool   `json:"completed"`
	OperatorUsername                   string `json:"operator_username"`
	OpsecPreBlocked                    bool   `json:"opsec_pre_blocked"`
	OpsecPreMessage                    string `json:"opsec_pre_message"`
	OpsecPreBypassed                   bool   `json:"opsec_pre_bypassed"`
	OpsecPreBypassRole                 string `json:"opsec_pre_bypass_role"`
	OpsecPostBlocked                   bool   `json:"opsec_post_blocked"`
	OpsecPostMessage                   string `json:"opsec_post_message"`
	OpsecPostBypassed                  bool   `json:"opsec_post_bypassed"`
	OpsecPostBypassRole                string `json:"opsec_post_bypass_role"`
	ParentTaskID                       int    `json:"parent_task_id"`
	SubtaskCallbackFunction            string `json:"subtask_callback_function"`
	SubtaskCallbackFunctionCompleted   bool   `json:"subtask_callback_function_completed"`
	GroupCallbackFunction              string `json:"group_callback_function"`
	GroupCallbackFunctionCompleted     bool   `json:"group_callback_function_completed"`
	CompletedCallbackFunction          string `json:"completed_callback_function"`
	CompletedCallbackFunctionCompleted bool   `json:"completed_callback_function_completed"`
	SubtaskGroupName                   string `json:"subtask_group_name"`
	TaskingLocation                    string `json:"tasking_location"`
	ParameterGroupName                 string `json:"parameter_group_name"`
	TokenID                            int    `json:"token_id"`
}

type PTTaskOPSECPostTaskMessageResponse struct {
	TaskID              int        `json:"task_id"`
	Success             bool       `json:"success"`
	Error               string     `json:"error"`
	OpsecPostBlocked    bool       `json:"opsec_post_blocked"`
	OpsecPostMessage    string     `json:"opsec_post_message"`
	OpsecPostBypassed   *bool      `json:"opsec_post_bypassed,omitempty"`
	OpsecPostBypassRole OPSEC_ROLE `json:"opsec_post_bypass_role"`
}

type PTTaskProcessResponseMessageResponse struct {
	TaskID  int    `json:"task_id"`
	Success bool   `json:"success"`
	Error   string `json:"error"`
}

type PTTaskingDynamicQueryFunction func(PTRPCDynamicQueryFunctionMessage) []string

type PT_TASK_FUNCTION_STATUS = string

type ParameterGroupInfo struct {
	ParameterIsRequired   bool              `json:"required"`
	GroupName             string            `json:"group_name"`
	UIModalPosition       uint32            `json:"ui_position"`
	AdditionalInformation map[string]string `json:"additional_info"`
}

type PayloadBuildC2Profile struct {
	Name  string `json:"name"`
	IsP2P bool   `json:"is_p2p"`
	// parameter name: parameter value
	// Parameters - this is an interface of parameter name -> parameter value from the associated C2 profile.
	// The types for the various parameter names can be found by looking at the build parameters in the Mythic UI.
	Parameters map[string]interface{} `json:"parameters"`
}

type PayloadBuildC2ProfileMessage struct {
	PayloadUUID     string                 `json:"uuid"`
	Parameters      map[string]interface{} `json:"parameters"`
	BuildParameters map[string]interface{} `json:"build_parameters"`
	SelectedOS      string                 `json:"selected_os"`
	PayloadType     string                 `json:"payload_type"`
}

type PayloadBuildC2ProfileMessageResponse struct {
	PayloadUUID  string  `json:"uuid"`
	Status       string  `json:"status"`
	Payload      *[]byte `json:"payload,omitempty"`
	BuildStdErr  string  `json:"build_stderr"`
	BuildStdOut  string  `json:"build_stdout"`
	BuildMessage string  `json:"build_message"`
}

type PayloadBuildMessage struct {
	// PayloadType - the name of the payload type for the build
	PayloadType string `json:"payload_type"`
	// CommandList - the list of commands the user selected to include in the build
	CommandList []string `json:"commands"`
	// build param name : build value
	// BuildParameters - map of param name -> build value from the user for the build parameters defined
	BuildParameters map[string]interface{} `json:"build_parameters"`
	// C2Profiles - list of C2 profiles selected to include in the payload and their associated parameters
	C2Profiles []PayloadBuildC2Profile `json:"c2profiles"`
	// WrappedPayload - bytes of the wrapped payload if one exists
	WrappedPayload *[]byte `json:"wrapped_payload,omitempty"`
	// SelectedOS - the operating system the user selected when building the agent
	SelectedOS string `json:"selected_os"`
	// PayloadUUID - the Mythic generated UUID for this payload instance
	PayloadUUID string `json:"uuid"`
	// PayloadFileUUID - The Mythic generated File UUID associated with this payload
	PayloadFileUUID string `json:"payload_file_uuid"`
}

type PayloadBuildResponse struct {
	// PayloadUUID - The UUID associated with this payload
	PayloadUUID string `json:"uuid"`
	// Success - was this build process successful or not
	Success bool `json:"success"`
	// Payload - the raw bytes of the payload that was compiled/created
	Payload *[]byte `json:"payload,omitempty"`
	// UpdatedCommandList - if you want to adjust the list of commands in this payload from what the user provided,
	// provide the updated list of command names here
	UpdatedCommandList *[]string `json:"updated_command_list,omitempty"`
	// BuildStdErr - build stderr message to associate with the build
	BuildStdErr string `json:"build_stderr"`
	// BuildStdOut - build stdout message to associate with the build
	BuildStdOut string `json:"build_stdout"`
	// BuildMessage - general message to associate with the build. Usually not as verbose as the stdout/stderr.
	BuildMessage string `json:"build_message"`
}

type PayloadConfiguration struct {
	Description        string                                `json:"tag"`
	PayloadType        string                                `json:"payload_type" binding:"required"`
	C2Profiles         *[]PayloadConfigurationC2Profile      `json:"c2_profiles,omitempty"`
	BuildParameters    *[]PayloadConfigurationBuildParameter `json:"build_parameters,omitempty"`
	Commands           []string                              `json:"commands"`
	SelectedOS         string                                `json:"selected_os" binding:"required"`
	Filename           string                                `json:"filename" binding:"required"`
	WrappedPayloadUUID string                                `json:"wrapped_payload"`
}

type PayloadConfigurationBuildParameter struct {
	Name  string      `json:"name" binding:"required"`
	Value interface{} `json:"value" binding:"required"`
}

type PayloadConfigurationC2Profile struct {
	Name       string                 `json:"c2_profile"`
	Parameters map[string]interface{} `json:"c2_profile_parameters"`
}

type PayloadType struct {
	Name                                   string                                                                                   `json:"name"`
	FileExtension                          string                                                                                   `json:"file_extension"`
	Author                                 string                                                                                   `json:"author"`
	SupportedOS                            []string                                                                                 `json:"supported_os"`
	Wrapper                                bool                                                                                     `json:"wrapper"`
	CanBeWrappedByTheFollowingPayloadTypes []string                                                                                 `json:"supported_wrapper_payload_types"`
	SupportsDynamicLoading                 bool                                                                                     `json:"supports_dynamic_load"`
	Description                            string                                                                                   `json:"description"`
	SupportedC2Profiles                    []string                                                                                 `json:"supported_c2_profiles"`
	TranslationContainerName               string                                                                                   `json:"translation_container_name"`
	MythicEncryptsData                     bool                                                                                     `json:"mythic_encrypts"`
	BuildParameters                        []BuildParameter                                                                         `json:"build_parameters"`
	BuildSteps                             []BuildStep                                                                              `json:"build_steps"`
	AgentIcon                              *[]byte                                                                                  `json:"agent_icon"` // automatically filled in based on Name
	CustomRPCFunctions                     map[string]func(message PTRPCOtherServiceRPCMessage) PTRPCOtherServiceRPCMessageResponse `json:"-"`
}

type PayloadTypeSyncMessage struct {
	PayloadType      PayloadType `json:"payload_type"`
	CommandList      []Command   `json:"commands"`
	ContainerVersion string      `json:"container_version"`
}

type PayloadTypeSyncMessageResponse struct {
	Success bool   `json:"success"`
	Error   string `json:"error"`
}

type PtTaskFunctionCreateTasking func(*PTTaskMessageAllData) PTTaskCreateTaskingMessageResponse

type PtTaskFunctionOPSECPost func(*PTTaskMessageAllData) PTTaskOPSECPostTaskMessageResponse

type PtTaskFunctionOPSECPre func(*PTTaskMessageAllData) PTTTaskOPSECPreTaskMessageResponse

type PtTaskFunctionParseArgDictionary func(args *PTTaskMessageArgsData, input map[string]interface{}) error

type PtTaskFunctionParseArgString func(args *PTTaskMessageArgsData, input string) error

type PtTaskFunctionProcessResponse func(PtTaskProcessResponseMessage) PTTaskProcessResponseMessageResponse

type PtTaskProcessResponseMessage struct {
	TaskData *PTTaskMessageAllData `json:"task"`
	Response interface{}           `json:"response"`
}

type RabbitmqDirectMethod struct {
	RabbitmqRoutingKey         string
	RabbitmqProcessingFunction func([]byte)
}

type RabbitmqRPCMethod struct {
	RabbitmqRoutingKey         string
	RabbitmqProcessingFunction func([]byte) interface{}
}

type SubtaskGroupName = string