connectivity

type Broker interface {
	InstallPolicy(ctx context.Context, policy PolicySet) error
	RemovePolicy(ctx context.Context, pluginInstanceID string) error
	MintConnectionGrant(ctx context.Context, req GrantRequest) (ConnectionGrant, error)
}

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

type CompileRequest struct {
	PluginInstanceID   string
	PluginID           string
	ActiveFingerprint  string
	PolicyRevision     uint64
	ManagementRevision uint64
	RevokeEpoch        uint64
	Manifest           manifest.Manifest
}

type ConnectionGrant struct {
	GrantID                 string                   `json:"grant_id"`
	PluginInstanceID        string                   `json:"plugin_instance_id"`
	ActiveFingerprint       string                   `json:"active_fingerprint"`
	ResourceScope           sessionctx.ResourceScope `json:"resource_scope"`
	PolicyRevision          uint64                   `json:"policy_revision"`
	ManagementRevision      uint64                   `json:"management_revision"`
	RevokeEpoch             uint64                   `json:"revoke_epoch"`
	ConnectorID             string                   `json:"connector_id"`
	Transport               Transport                `json:"transport"`
	Destination             Destination              `json:"destination"`
	RuntimeGenerationID     string                   `json:"runtime_generation_id,omitempty"`
	TargetClassifierVersion string                   `json:"target_classifier_version"`
	ExpiresAt               time.Time                `json:"expires_at"`
}

type ConnectorPolicy struct {
	ConnectorID  string        `json:"connector_id"`
	Transport    Transport     `json:"transport"`
	Scope        Scope         `json:"scope"`
	Destinations []Destination `json:"destinations"`
	SecretRef    string        `json:"secret_ref,omitempty"`
}

type Destination struct {
	Transport Transport `json:"transport"`
	Scheme    string    `json:"scheme,omitempty"`
	Host      string    `json:"host"`
	Port      int       `json:"port"`
}

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

type ExecutorOptions struct {
	Dialer           *net.Dialer
	LookupIPAddr     func(ctx context.Context, host string) ([]net.IPAddr, error)
	UDPRateLimiter   UDPRateLimiter
	MaxRequestBytes  int64
	MaxResponseBytes int64
	DefaultTimeout   time.Duration
	Now              func() time.Time
}

type GrantRequest struct {
	PluginInstanceID    string
	ActiveFingerprint   string
	ResourceScope       sessionctx.ResourceScope
	PolicyRevision      uint64
	ManagementRevision  uint64
	RevokeEpoch         uint64
	ConnectorID         string
	Transport           Transport
	Destination         string
	RuntimeGenerationID string
	Now                 time.Time `json:"-"`
	TTL                 time.Duration
}

type HTTPRequest struct {
	Grant            ConnectionGrant `json:"grant"`
	Method           string          `json:"method"`
	Path             string          `json:"path,omitempty"`
	Query            url.Values      `json:"query,omitempty"`
	Headers          http.Header     `json:"headers,omitempty"`
	Body             []byte          `json:"-"`
	MaxRequestBytes  int64           `json:"max_request_bytes,omitempty"`
	MaxResponseBytes int64           `json:"max_response_bytes,omitempty"`
	MaxChunkBytes    int64           `json:"max_chunk_bytes,omitempty"`
	Timeout          time.Duration   `json:"timeout,omitempty"`
	Now              time.Time       `json:"-"`
}

type HTTPResponse struct {
	StatusCode int         `json:"status_code"`
	Headers    http.Header `json:"headers,omitempty"`
	Body       []byte      `json:"-"`
}

type HTTPResponseChunk struct {
	Index int    `json:"index"`
	Data  []byte `json:"-"`
}

type HTTPStreamResponse struct {
	StatusCode int         `json:"status_code"`
	Headers    http.Header `json:"headers,omitempty"`
	BytesRead  int64       `json:"bytes_read"`
	ChunkCount int         `json:"chunk_count"`
}

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

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

type NetworkExecutor interface {
	DoHTTP(ctx context.Context, req HTTPRequest) (HTTPResponse, error)
	StreamHTTP(ctx context.Context, req HTTPRequest, onChunk func(HTTPResponseChunk) error) (HTTPStreamResponse, error)
	WebSocketRoundTrip(ctx context.Context, req WebSocketRoundTripRequest) (WebSocketRoundTripResponse, error)
	TCPRoundTrip(ctx context.Context, req TCPRoundTripRequest) (TCPRoundTripResponse, error)
	UDPRoundTrip(ctx context.Context, req UDPRoundTripRequest) (UDPRoundTripResponse, error)
}

type PolicySet struct {
	PluginInstanceID        string            `json:"plugin_instance_id"`
	PluginID                string            `json:"plugin_id"`
	ActiveFingerprint       string            `json:"active_fingerprint"`
	PolicyRevision          uint64            `json:"policy_revision"`
	ManagementRevision      uint64            `json:"management_revision"`
	RevokeEpoch             uint64            `json:"revoke_epoch"`
	TargetClassifierVersion string            `json:"target_classifier_version"`
	Connectors              []ConnectorPolicy `json:"connectors"`
}

type Scope string

type TCPRoundTripRequest struct {
	Grant           ConnectionGrant `json:"grant"`
	Payload         []byte          `json:"-"`
	MaxRequestBytes int64           `json:"max_request_bytes,omitempty"`
	MaxReadBytes    int64           `json:"max_read_bytes,omitempty"`
	Timeout         time.Duration   `json:"timeout,omitempty"`
	Now             time.Time       `json:"-"`
}

type TCPRoundTripResponse struct {
	Payload []byte `json:"-"`
}

type Transport string

type UDPRateLimit struct {
	MaxRoundTrips int
	Window        time.Duration
}

type UDPRateLimitKey struct {
	PluginInstanceID  string      `json:"plugin_instance_id"`
	ActiveFingerprint string      `json:"active_fingerprint"`
	ConnectorID       string      `json:"connector_id"`
	GrantID           string      `json:"grant_id"`
	Destination       Destination `json:"destination"`
}

type UDPRateLimiter interface {
	AllowUDPRoundTrip(now time.Time, key UDPRateLimitKey) bool
}

type UDPRateLimiterFunc func(now time.Time, key UDPRateLimitKey) bool

type UDPRoundTripRequest struct {
	Grant        ConnectionGrant `json:"grant"`
	Payload      []byte          `json:"-"`
	MaxReadBytes int64           `json:"max_read_bytes,omitempty"`
	Timeout      time.Duration   `json:"timeout,omitempty"`
	Now          time.Time       `json:"-"`
}

type UDPRoundTripResponse struct {
	Payload []byte `json:"-"`
}

type WebSocketMessageType string

type WebSocketRoundTripRequest struct {
	Grant            ConnectionGrant      `json:"grant"`
	Path             string               `json:"path,omitempty"`
	Headers          http.Header          `json:"headers,omitempty"`
	MessageType      WebSocketMessageType `json:"message_type,omitempty"`
	Payload          []byte               `json:"-"`
	MaxRequestBytes  int64                `json:"max_request_bytes,omitempty"`
	MaxResponseBytes int64                `json:"max_response_bytes,omitempty"`
	Timeout          time.Duration        `json:"timeout,omitempty"`
	Now              time.Time            `json:"-"`
}

type WebSocketRoundTripResponse struct {
	MessageType WebSocketMessageType `json:"message_type"`
	Payload     []byte               `json:"-"`
}