adaptix_struct.go

package adaptix

const (
	OS_UNKNOWN = 0
	OS_WINDOWS = 1
	OS_LINUX   = 2
	OS_MAC     = 3

	TASK_TYPE_LOCAL      = 0
	TASK_TYPE_TASK       = 1
	TASK_TYPE_BROWSER    = 2
	TASK_TYPE_JOB        = 3
	TASK_TYPE_TUNNEL     = 4
	TASK_TYPE_PROXY_DATA = 5

	MESSAGE_INFO    = 5
	MESSAGE_ERROR   = 6
	MESSAGE_SUCCESS = 7

	BUILD_LOG_NONE    = 0
	BUILD_LOG_INFO    = 1
	BUILD_LOG_ERROR   = 2
	BUILD_LOG_SUCCESS = 3

	TRANSFER_STATE_RUNNING  = 1
	TRANSFER_STATE_STOPPED  = 2
	TRANSFER_STATE_FINISHED = 3
	TRANSFER_STATE_CANCELED = 4

	TRANSFER_KIND_FILE   = 0
	TRANSFER_KIND_MEMORY = 1

	TUNNEL_TYPE_SOCKS4      = 1
	TUNNEL_TYPE_SOCKS5      = 2
	TUNNEL_TYPE_SOCKS5_AUTH = 3
	TUNNEL_TYPE_LOCAL_PORT  = 4
	TUNNEL_TYPE_REVERSE     = 5
	TUNNEL_TYPE_SOCKS_BIND  = 6

	TUNNEL_BIND_LISTENING = 2
	TUNNEL_BIND_ACCEPTED  = 3

	ADDRESS_TYPE_IPV4   = 1
	ADDRESS_TYPE_DOMAIN = 3
	ADDRESS_TYPE_IPV6   = 4

	SOCKS5_SUCCESS                 byte = 0
	SOCKS5_SERVER_FAILURE          byte = 1
	SOCKS5_NOT_ALLOWED_RULESET     byte = 2
	SOCKS5_NETWORK_UNREACHABLE     byte = 3
	SOCKS5_HOST_UNREACHABLE        byte = 4
	SOCKS5_CONNECTION_REFUSED      byte = 5
	SOCKS5_TTL_EXPIRED             byte = 6
	SOCKS5_COMMAND_NOT_SUPPORTED   byte = 7
	SOCKS5_ADDR_TYPE_NOT_SUPPORTED byte = 8
)

// --- Plugin interfaces ---

type PluginService interface {
	Call(operator string, function string, args string)
	CallRPC(operator string, function string, args string) (resultJSON string, err error)
}

type PluginListener interface {
	Create(name, config string, customData []byte) (ExtenderListener, ListenerData, []byte, error)
	Call(operator string, listenerName string, function string, args string)
}

type ExtenderListener interface {
	Start() error
	Edit(config string) (ListenerData, []byte, error)
	Stop() error
	GetProfile() ([]byte, error)
	InternalHandler(data []byte) (int64, error)
}

type PluginAgent interface {
	GenerateProfiles(profile BuildProfile) ([][]byte, error)
	BuildPayload(profile BuildProfile, agentProfiles [][]byte) ([]byte, string, error)
	CreateAgent(beat []byte) (AgentData, AgentFunctions, error)
	AgentRestore(agentData AgentData) AgentFunctions
	Call(operator string, agentId int64, function string, args string)
}

type AgentFunctions struct {
	CreateCommand func(AgentData, map[string]any) (TaskData, ConsoleMessageData, error)
	PackTasks     func(AgentData, []TaskData) ([]byte, error)
	ProcessData   func(AgentData, []byte) error
	Encrypt       func([]byte, []byte) ([]byte, error)
	Decrypt       func([]byte, []byte) ([]byte, error)
	PivotPackData func(string, []byte) (TaskData, error)
	Delivery      DeliveryFunc
	TunnelCB      TunnelCallbacks
	TerminalCB    TerminalCallbacks
}

// --- Callbacks ---

type TunnelCallbacks struct {
	ConnectTCP func(channelId int64, tunnelType, addressType int, address string, port int) TaskData
	ConnectUDP func(channelId int64, tunnelType, addressType int, address string, port int) TaskData
	WriteTCP   func(channelId int64, data []byte) TaskData
	WriteUDP   func(channelId int64, data []byte) TaskData
	Pause      func(channelId int64) TaskData
	Resume     func(channelId int64) TaskData
	Close      func(channelId int64) TaskData
	Reverse    func(tunnelId int64, port int) TaskData
	BindTCP    func(channelId int64, addressType int, address string, port int) TaskData
}

type TerminalCallbacks struct {
	Start func(terminalId int64, program string, sizeH, sizeW, oemCP int) TaskData
	Write func(terminalId int64, oemCP int, data []byte) TaskData
	Close func(terminalId int64) TaskData
}

// --- Data structures ---

type ListenerData struct {
	Name       string `json:"l_name"`
	RegName    string `json:"l_reg_name"`
	Protocol   string `json:"l_protocol"`
	Type       string `json:"l_type"`
	BindHost   string `json:"l_bind_host"`
	BindPort   string `json:"l_bind_port"`
	AgentAddr  string `json:"l_agent_addr"`
	CreateTime int64  `json:"a_create_time"`
	Status     string `json:"l_status"`
	Tags       string `json:"l_tags"`
	Data       string `json:"l_data"`
	Watermark  string `json:"l_watermark"`
}

type AgentData struct {
	Crc          string `json:"a_crc"`
	Id           int64  `json:"a_id"`
	UID          []byte `json:"a_uid"`
	Name         string `json:"a_name"`
	SessionKey   []byte `json:"a_session_key"`
	Listener     string `json:"a_listener"`
	Async        bool   `json:"a_async"`
	ExternalIP   string `json:"a_external_ip"`
	InternalIP   string `json:"a_internal_ip"`
	GmtOffset    int    `json:"a_gmt_offset"`
	Sleep        uint   `json:"a_sleep"`
	Jitter       uint   `json:"a_jitter"`
	Pid          string `json:"a_pid"`
	Tid          string `json:"a_tid"`
	Arch         string `json:"a_arch"`
	Elevated     bool   `json:"a_elevated"`
	Process      string `json:"a_process"`
	Os           int    `json:"a_os"`
	OsDesc       string `json:"a_os_desc"`
	Domain       string `json:"a_domain"`
	Computer     string `json:"a_computer"`
	Username     string `json:"a_username"`
	Impersonated string `json:"a_impersonated"`
	OemCP        int    `json:"a_oemcp"`
	ACP          int    `json:"a_acp"`
	CreateTime   int64  `json:"a_create_time"`
	LastTick     int    `json:"a_last_tick"`
	KillDate     int    `json:"a_killdate"`
	WorkingTime  int    `json:"a_workingtime"`
	Tags         string `json:"a_tags"`
	Mark         string `json:"a_mark"`
	Color        string `json:"a_color"`
	TargetId     int64  `json:"a_target"`
	CustomData   []byte `json:"a_custom_data"`
	CmdGroups    string `json:"a_cmd_groups_json,omitempty"`
}

type TaskDataTunnel struct {
	ChannelId int64
	Data      TaskData
}

type TaskData struct {
	Type        int    `json:"t_type"`
	TaskId      int64  `json:"t_task_id"`
	AgentId     int64  `json:"t_agent_id"`
	Client      string `json:"t_client"`
	HookId      string `json:"t_hook_id"`
	HandlerId   string `json:"t_handler_id"`
	User        string `json:"t_user"`
	Computer    string `json:"t_computer"`
	StartDate   int64  `json:"t_start_date"`
	FinishDate  int64  `json:"t_finish_date"`
	Data        []byte `json:"t_data"`
	CommandLine string `json:"t_command_line"`
	MessageType int    `json:"t_message_type"`
	Message     string `json:"t_message"`
	ClearText   string `json:"t_clear_text"`
	Completed   bool   `json:"t_completed"`
	Priority    uint   `json:"t_priority"`
	Sync        bool   `json:"t_sync"`

	Repeat         bool `json:"-"`
	DispatchBudget int  `json:"-"`

	OnDispatch func(ts any, task *TaskData) `json:"-"`
	OnComplete func(ts any, task *TaskData) `json:"-"`
}

type ConsoleMessageData struct {
	Message string `json:"m_message"`
	Status  int    `json:"m_status"`
	Text    string `json:"m_text"`
}

type ListingFileDataWin struct {
	IsDir    bool   `json:"b_is_dir"`
	Size     int64  `json:"b_size"`
	Date     int64  `json:"b_date"`
	Filename string `json:"b_filename"`
}

type ListingFileDataUnix struct {
	IsDir    bool   `json:"b_is_dir"`
	Mode     string `json:"b_mode"`
	User     string `json:"b_user"`
	Group    string `json:"b_group"`
	Size     int64  `json:"b_size"`
	Date     string `json:"b_date"`
	Filename string `json:"b_filename"`
}

type ListingProcessDataWin struct {
	Pid         uint   `json:"b_pid"`
	Ppid        uint   `json:"b_ppid"`
	SessionId   uint   `json:"b_session_id"`
	Arch        string `json:"b_arch"`
	Context     string `json:"b_context"`
	ProcessName string `json:"b_process_name"`
}

type ListingProcessDataUnix struct {
	Pid         uint   `json:"b_pid"`
	Ppid        uint   `json:"b_ppid"`
	TTY         string `json:"b_tty"`
	Context     string `json:"b_context"`
	ProcessName string `json:"b_process_name"`
}

type ListingDrivesDataWin struct {
	Name string `json:"b_name"`
	Type string `json:"b_type"`
}

type ChatData struct {
	Username string `json:"c_username"`
	Message  string `json:"c_message"`
	Date     int64  `json:"c_date"`
}

type TransferData struct {
	FileId       int64  `json:"t_file_id"`
	AgentId      int64  `json:"t_agent_id"`
	AgentName    string `json:"t_agent_name"`
	User         string `json:"t_user"`
	Computer     string `json:"t_computer"`
	RemotePath   string `json:"t_remote_path"`
	LocalPath    string `json:"t_local_path"`
	TotalSize    int64  `json:"t_total_size"`
	Progress     int64  `json:"t_progress"`
	ReadOffset   int64  `json:"-"`
	Date         int64  `json:"t_date"`
	State        int    `json:"t_state"`
	Tag          string `json:"t_tag"`
	Cancellable  bool   `json:"t_cancellable"`
	Kind         int    `json:"t_kind"`
	ArtifactName string `json:"t_artifact_name,omitempty"`
	ArtifactType string `json:"t_artifact_type,omitempty"`
}

type ScreenData struct {
	ScreenId  int64  `json:"s_screen_id"`
	AgentId   int64  `json:"s_agent_id"`
	User      string `json:"s_user"`
	Computer  string `json:"s_computer"`
	LocalPath string `json:"s_local_path"`
	Note      string `json:"s_note"`
	Date      int64  `json:"s_date"`
	Content   []byte `json:"s_content"`
}

type TunnelData struct {
	TunnelId  int64  `json:"p_tunnel_id"`
	AgentId   int64  `json:"p_agent_id"`
	Computer  string `json:"p_computer"`
	Username  string `json:"p_username"`
	Process   string `json:"p_process"`
	Type      string `json:"p_type"`
	Info      string `json:"p_info"`
	Interface string `json:"p_interface"`
	Port      string `json:"p_port"`
	Client    string `json:"p_client"`
	Fhost     string `json:"p_fhost"`
	Fport     string `json:"p_fport"`
	AuthUser  string `json:"p_auth_user"`
	AuthPass  string `json:"p_auth_pass"`
	Date      int64  `json:"p_date"`
	Active    bool   `json:"p_active"`
}

type PivotData struct {
	PivotId       string `json:"p_pivot_id"`
	PivotName     string `json:"p_pivot_name"`
	ParentAgentId int64  `json:"p_parent_agent_id"`
	ChildAgentId  int64  `json:"p_child_agent_id"`
}

type GroupData struct {
	GroupId       int64   `json:"group_id"`
	ParentGroupId int64   `json:"parent_group_id"`
	GroupName     string  `json:"group_name"`
	Scope         string  `json:"scope"`
	Members       []int64 `json:"members"`
}

type CredsData struct {
	CredId   int64  `json:"c_creds_id"`
	Username string `json:"c_username"`
	Password string `json:"c_password"`
	Realm    string `json:"c_realm"`
	Type     string `json:"c_type"`
	Tag      string `json:"c_tag"`
	Date     int64  `json:"c_date"`
	Storage  string `json:"c_storage"`
	AgentId  int64  `json:"c_agent_id"`
	Host     string `json:"c_host"`
}

type TargetData struct {
	TargetId int64   `json:"t_target_id"`
	Computer string  `json:"t_computer"`
	Domain   string  `json:"t_domain"`
	Address  string  `json:"t_address"`
	Os       int     `json:"t_os"`
	OsDesk   string  `json:"t_os_desk"`
	Tag      string  `json:"t_tag"`
	Info     string  `json:"t_info"`
	Date     int64   `json:"t_date"`
	Alive    bool    `json:"t_alive"`
	Agents   []int64 `json:"t_agents"`
}

type TransportProfile struct {
	Watermark string `json:"watermark"`
	Profile   []byte `json:"profile"`
}

type BuildProfile struct {
	BuilderId        string             `json:"build_id"`
	AgentConfig      string             `json:"agent_params"`
	ListenerProfiles []TransportProfile `json:"listener_profiles"`
}

type PayloadData struct {
	PayloadId  int64    `json:"p_id"`
	Name       string   `json:"p_name"`
	AgentType  string   `json:"p_type"`
	Artifact   string   `json:"p_artifact"`
	Arch       string   `json:"p_arch"`
	Listeners  []string `json:"p_listeners"`
	Size       int64    `json:"p_size"`
	Sha1       string   `json:"p_sha1"`
	Sha256     string   `json:"p_sha256"`
	Md5        string   `json:"p_md5"`
	Creator    string   `json:"p_creator"`
	Created    int64    `json:"p_date"`
	Hidden     bool     `json:"p_hidden"`
	LocalPath  string   `json:"p_local_path,omitempty"`
	ConfigJson string   `json:"p_config,omitempty"`
	BuildId    string   `json:"p_build_id,omitempty"`
	Watermark  string   `json:"p_watermark,omitempty"`
	Filename   string   `json:"p_filename"`
	Notes      string   `json:"p_notes,omitempty"`
	Tag        string   `json:"p_tag,omitempty"`
	Uid        string   `json:"p_uid,omitempty"`
	Color      string   `json:"p_color,omitempty"`
	Missing    bool     `json:"p_missing,omitempty"`
}

// --- Logging ---

type LogStatus int

const (
	LogStatusDebug   LogStatus = 0
	LogStatusInfo    LogStatus = 1
	LogStatusSuccess LogStatus = 2
	LogStatusWarn    LogStatus = 3
	LogStatusError   LogStatus = 4
)

type LogEntry struct {
	Id       int64     `json:"id"`
	Time     int64     `json:"time"`
	Status   LogStatus `json:"status"`
	Level    int       `json:"level"`
	Source   string    `json:"source"`
	Category string    `json:"category,omitempty"`
	Message  string    `json:"message"`
}

func (e LogEntry) LogSourceKey() string {
	if e.Category == "" {
		return e.Source
	}
	if e.Source == "" {
		return e.Category
	}
	return e.Source + "::" + e.Category
}