mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 09:18:59 +00:00
b59cd2b48b
Return `accelerate: 10` with distributed queries if we do not have host details. This facilitates the host quickly joining all expected labels, as `platform` gated label queries will not be returned until the detail queries return with the platform. Fixes #1421.
82 lines
3.1 KiB
Go
82 lines
3.1 KiB
Go
package kolide
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type OsqueryService interface {
|
|
EnrollAgent(ctx context.Context, enrollSecret, hostIdentifier string) (nodeKey string, err error)
|
|
AuthenticateHost(ctx context.Context, nodeKey string) (host *Host, err error)
|
|
GetClientConfig(ctx context.Context) (config *OsqueryConfig, err error)
|
|
// GetDistributedQueries retrieves the distributed queries to run for
|
|
// the host in the provided context. These may be detail queries, label
|
|
// queries, or user-initiated distributed queries. A map from query
|
|
// name to query is returned. To enable the osquery "accelerated
|
|
// checkins" feature, a positive integer (number of seconds to activate
|
|
// for) should be returned. Returning 0 for this will not activate the
|
|
// feature.
|
|
GetDistributedQueries(ctx context.Context) (queries map[string]string, accelerate uint, err error)
|
|
SubmitDistributedQueryResults(ctx context.Context, results OsqueryDistributedQueryResults, statuses map[string]string) (err error)
|
|
SubmitStatusLogs(ctx context.Context, logs []OsqueryStatusLog) (err error)
|
|
SubmitResultLogs(ctx context.Context, logs []OsqueryResultLog) (err error)
|
|
}
|
|
|
|
type OsqueryDistributedQueryResults map[string][]map[string]string
|
|
|
|
type QueryContent struct {
|
|
Query string `json:"query"`
|
|
Description string `json:"description,omitempty"`
|
|
Interval uint `json:"interval"`
|
|
Platform *string `json:"platform,omitempty"`
|
|
Version *string `json:"version,omitempty"`
|
|
Snapshot *bool `json:"snapshot,omitempty"`
|
|
Removed *bool `json:"removed,omitempty"`
|
|
Shard *uint `json:"shard,omitempty"`
|
|
}
|
|
|
|
type Queries map[string]QueryContent
|
|
|
|
type PackContent struct {
|
|
Platform string `json:"platform,omitempty"`
|
|
Version string `json:"version,omitempty"`
|
|
Shard uint `json:"shard,omitempty"`
|
|
Discovery []string `json:"discovery,omitempty"`
|
|
Queries Queries `json:"queries"`
|
|
}
|
|
|
|
type Packs map[string]PackContent
|
|
|
|
type Decorators struct {
|
|
Load []string `json:"load,omitempty"`
|
|
Always []string `json:"always,omitempty"`
|
|
Interval map[string][]string `json:"interval,omitempty"`
|
|
}
|
|
|
|
type OsqueryConfig struct {
|
|
Options map[string]interface{} `json:"options"`
|
|
Decorators Decorators `json:"decorators,omitempty"`
|
|
Packs Packs `json:"packs,omitempty"`
|
|
}
|
|
|
|
type OsqueryResultLog struct {
|
|
Name string `json:"name"`
|
|
HostIdentifier string `json:"hostIdentifier"`
|
|
UnixTime string `json:"unixTime"`
|
|
CalendarTime string `json:"calendarTime"`
|
|
// Columns stores the columns of differential queries
|
|
Columns map[string]string `json:"columns,omitempty"`
|
|
// Snapshot stores the rows and columns of snapshot queries
|
|
Snapshot []map[string]string `json:"snapshot,omitempty"`
|
|
Action string `json:"action"`
|
|
Decorations map[string]string `json:"decorations"`
|
|
}
|
|
|
|
type OsqueryStatusLog struct {
|
|
Severity string `json:"severity"`
|
|
Filename string `json:"filename"`
|
|
Line string `json:"line"`
|
|
Message string `json:"message"`
|
|
Version string `json:"version"`
|
|
Decorations map[string]string `json:"decorations"`
|
|
}
|