2016-09-26 18:48:55 +00:00
|
|
|
package service
|
2016-08-28 03:59:17 +00:00
|
|
|
|
|
|
|
import (
|
2016-11-09 17:19:07 +00:00
|
|
|
"encoding/json"
|
2016-08-28 03:59:17 +00:00
|
|
|
"net/http"
|
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
"github.com/go-kit/kit/endpoint"
|
2016-08-28 03:59:17 +00:00
|
|
|
kitlog "github.com/go-kit/kit/log"
|
|
|
|
kithttp "github.com/go-kit/kit/transport/http"
|
|
|
|
"github.com/gorilla/mux"
|
2016-09-26 18:48:55 +00:00
|
|
|
"github.com/kolide/kolide-ose/server/kolide"
|
2016-09-03 17:25:16 +00:00
|
|
|
"golang.org/x/net/context"
|
2016-08-28 03:59:17 +00:00
|
|
|
)
|
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
// KolideEndpoints is a collection of RPC endpoints implemented by the Kolide API.
|
|
|
|
type KolideEndpoints struct {
|
2016-11-16 21:07:50 +00:00
|
|
|
Login endpoint.Endpoint
|
|
|
|
Logout endpoint.Endpoint
|
|
|
|
ForgotPassword endpoint.Endpoint
|
|
|
|
ResetPassword endpoint.Endpoint
|
|
|
|
Me endpoint.Endpoint
|
2016-12-14 18:11:43 +00:00
|
|
|
ChangePassword endpoint.Endpoint
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateUser endpoint.Endpoint
|
|
|
|
GetUser endpoint.Endpoint
|
|
|
|
ListUsers endpoint.Endpoint
|
|
|
|
ModifyUser endpoint.Endpoint
|
|
|
|
GetSessionsForUserInfo endpoint.Endpoint
|
|
|
|
DeleteSessionsForUser endpoint.Endpoint
|
|
|
|
GetSessionInfo endpoint.Endpoint
|
|
|
|
DeleteSession endpoint.Endpoint
|
|
|
|
GetAppConfig endpoint.Endpoint
|
|
|
|
ModifyAppConfig endpoint.Endpoint
|
|
|
|
CreateInvite endpoint.Endpoint
|
|
|
|
ListInvites endpoint.Endpoint
|
|
|
|
DeleteInvite endpoint.Endpoint
|
|
|
|
GetQuery endpoint.Endpoint
|
|
|
|
ListQueries endpoint.Endpoint
|
|
|
|
CreateQuery endpoint.Endpoint
|
|
|
|
ModifyQuery endpoint.Endpoint
|
|
|
|
DeleteQuery endpoint.Endpoint
|
2016-12-09 17:12:45 +00:00
|
|
|
DeleteQueries endpoint.Endpoint
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateDistributedQueryCampaign endpoint.Endpoint
|
|
|
|
GetPack endpoint.Endpoint
|
|
|
|
ListPacks endpoint.Endpoint
|
|
|
|
CreatePack endpoint.Endpoint
|
|
|
|
ModifyPack endpoint.Endpoint
|
|
|
|
DeletePack endpoint.Endpoint
|
2016-12-13 22:22:05 +00:00
|
|
|
ScheduleQueries endpoint.Endpoint
|
|
|
|
GetScheduledQueriesInPack endpoint.Endpoint
|
|
|
|
GetScheduledQuery endpoint.Endpoint
|
|
|
|
ModifyScheduledQuery endpoint.Endpoint
|
|
|
|
DeleteScheduledQuery endpoint.Endpoint
|
2016-11-16 21:07:50 +00:00
|
|
|
EnrollAgent endpoint.Endpoint
|
|
|
|
GetClientConfig endpoint.Endpoint
|
|
|
|
GetDistributedQueries endpoint.Endpoint
|
|
|
|
SubmitDistributedQueryResults endpoint.Endpoint
|
|
|
|
SubmitLogs endpoint.Endpoint
|
|
|
|
GetLabel endpoint.Endpoint
|
|
|
|
ListLabels endpoint.Endpoint
|
|
|
|
CreateLabel endpoint.Endpoint
|
|
|
|
DeleteLabel endpoint.Endpoint
|
|
|
|
AddLabelToPack endpoint.Endpoint
|
|
|
|
GetLabelsForPack endpoint.Endpoint
|
|
|
|
DeleteLabelFromPack endpoint.Endpoint
|
|
|
|
GetHost endpoint.Endpoint
|
|
|
|
DeleteHost endpoint.Endpoint
|
|
|
|
ListHosts endpoint.Endpoint
|
|
|
|
SearchTargets endpoint.Endpoint
|
2016-09-26 17:14:39 +00:00
|
|
|
}
|
2016-09-04 05:13:42 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
// MakeKolideServerEndpoints creates the Kolide API endpoints.
|
|
|
|
func MakeKolideServerEndpoints(svc kolide.Service, jwtKey string) KolideEndpoints {
|
|
|
|
return KolideEndpoints{
|
2016-09-29 04:21:39 +00:00
|
|
|
Login: makeLoginEndpoint(svc),
|
|
|
|
Logout: makeLogoutEndpoint(svc),
|
|
|
|
ForgotPassword: makeForgotPasswordEndpoint(svc),
|
|
|
|
ResetPassword: makeResetPasswordEndpoint(svc),
|
|
|
|
CreateUser: makeCreateUserEndpoint(svc),
|
|
|
|
|
|
|
|
// Authenticated user endpoints
|
2016-12-13 16:57:49 +00:00
|
|
|
// Each of these endpoints should have exactly one
|
|
|
|
// authorization check around the make.*Endpoint method. At a
|
|
|
|
// minimum, canPerformActions. Some endpoints use
|
|
|
|
// stricter/different checks and should NOT also use
|
|
|
|
// canPerformActions (these other checks should also call
|
|
|
|
// canPerformActions if that is appropriate).
|
|
|
|
Me: authenticatedUser(jwtKey, svc, canPerformActions(makeGetSessionUserEndpoint(svc))),
|
2016-12-14 18:11:43 +00:00
|
|
|
ChangePassword: authenticatedUser(jwtKey, svc, canPerformActions(makeChangePasswordEndpoint(svc))),
|
2016-11-16 21:07:50 +00:00
|
|
|
GetUser: authenticatedUser(jwtKey, svc, canReadUser(makeGetUserEndpoint(svc))),
|
|
|
|
ListUsers: authenticatedUser(jwtKey, svc, canPerformActions(makeListUsersEndpoint(svc))),
|
|
|
|
ModifyUser: authenticatedUser(jwtKey, svc, validateModifyUserRequest(makeModifyUserEndpoint(svc))),
|
|
|
|
GetSessionsForUserInfo: authenticatedUser(jwtKey, svc, canReadUser(makeGetInfoAboutSessionsForUserEndpoint(svc))),
|
|
|
|
DeleteSessionsForUser: authenticatedUser(jwtKey, svc, canModifyUser(makeDeleteSessionsForUserEndpoint(svc))),
|
|
|
|
GetSessionInfo: authenticatedUser(jwtKey, svc, mustBeAdmin(makeGetInfoAboutSessionEndpoint(svc))),
|
|
|
|
DeleteSession: authenticatedUser(jwtKey, svc, mustBeAdmin(makeDeleteSessionEndpoint(svc))),
|
2016-12-13 16:57:49 +00:00
|
|
|
GetAppConfig: authenticatedUser(jwtKey, svc, canPerformActions(makeGetAppConfigEndpoint(svc))),
|
2016-11-16 21:07:50 +00:00
|
|
|
ModifyAppConfig: authenticatedUser(jwtKey, svc, mustBeAdmin(makeModifyAppConfigRequest(svc))),
|
|
|
|
CreateInvite: authenticatedUser(jwtKey, svc, mustBeAdmin(makeCreateInviteEndpoint(svc))),
|
|
|
|
ListInvites: authenticatedUser(jwtKey, svc, mustBeAdmin(makeListInvitesEndpoint(svc))),
|
|
|
|
DeleteInvite: authenticatedUser(jwtKey, svc, mustBeAdmin(makeDeleteInviteEndpoint(svc))),
|
2016-12-13 22:22:05 +00:00
|
|
|
GetQuery: authenticatedUser(jwtKey, svc, makeGetQueryEndpoint(svc)),
|
|
|
|
ListQueries: authenticatedUser(jwtKey, svc, makeListQueriesEndpoint(svc)),
|
|
|
|
CreateQuery: authenticatedUser(jwtKey, svc, makeCreateQueryEndpoint(svc)),
|
|
|
|
ModifyQuery: authenticatedUser(jwtKey, svc, makeModifyQueryEndpoint(svc)),
|
|
|
|
DeleteQuery: authenticatedUser(jwtKey, svc, makeDeleteQueryEndpoint(svc)),
|
|
|
|
DeleteQueries: authenticatedUser(jwtKey, svc, makeDeleteQueriesEndpoint(svc)),
|
|
|
|
CreateDistributedQueryCampaign: authenticatedUser(jwtKey, svc, makeCreateDistributedQueryCampaignEndpoint(svc)),
|
|
|
|
GetPack: authenticatedUser(jwtKey, svc, makeGetPackEndpoint(svc)),
|
|
|
|
ListPacks: authenticatedUser(jwtKey, svc, makeListPacksEndpoint(svc)),
|
|
|
|
CreatePack: authenticatedUser(jwtKey, svc, makeCreatePackEndpoint(svc)),
|
|
|
|
ModifyPack: authenticatedUser(jwtKey, svc, makeModifyPackEndpoint(svc)),
|
|
|
|
DeletePack: authenticatedUser(jwtKey, svc, makeDeletePackEndpoint(svc)),
|
|
|
|
ScheduleQueries: authenticatedUser(jwtKey, svc, makeScheduleQueriesEndpoint(svc)),
|
|
|
|
GetScheduledQueriesInPack: authenticatedUser(jwtKey, svc, makeGetScheduledQueriesInPackEndpoint(svc)),
|
|
|
|
GetScheduledQuery: authenticatedUser(jwtKey, svc, makeGetScheduledQueryEndpoint(svc)),
|
|
|
|
ModifyScheduledQuery: authenticatedUser(jwtKey, svc, makeModifyScheduledQueryEndpoint(svc)),
|
|
|
|
DeleteScheduledQuery: authenticatedUser(jwtKey, svc, makeDeleteScheduledQueryEndpoint(svc)),
|
|
|
|
GetHost: authenticatedUser(jwtKey, svc, makeGetHostEndpoint(svc)),
|
|
|
|
ListHosts: authenticatedUser(jwtKey, svc, makeListHostsEndpoint(svc)),
|
|
|
|
DeleteHost: authenticatedUser(jwtKey, svc, makeDeleteHostEndpoint(svc)),
|
|
|
|
GetLabel: authenticatedUser(jwtKey, svc, makeGetLabelEndpoint(svc)),
|
|
|
|
ListLabels: authenticatedUser(jwtKey, svc, makeListLabelsEndpoint(svc)),
|
|
|
|
CreateLabel: authenticatedUser(jwtKey, svc, makeCreateLabelEndpoint(svc)),
|
|
|
|
DeleteLabel: authenticatedUser(jwtKey, svc, makeDeleteLabelEndpoint(svc)),
|
|
|
|
AddLabelToPack: authenticatedUser(jwtKey, svc, makeAddLabelToPackEndpoint(svc)),
|
|
|
|
GetLabelsForPack: authenticatedUser(jwtKey, svc, makeGetLabelsForPackEndpoint(svc)),
|
|
|
|
DeleteLabelFromPack: authenticatedUser(jwtKey, svc, makeDeleteLabelFromPackEndpoint(svc)),
|
|
|
|
SearchTargets: authenticatedUser(jwtKey, svc, makeSearchTargetsEndpoint(svc)),
|
2016-09-29 04:21:39 +00:00
|
|
|
|
|
|
|
// Osquery endpoints
|
|
|
|
EnrollAgent: makeEnrollAgentEndpoint(svc),
|
|
|
|
GetClientConfig: authenticatedHost(svc, makeGetClientConfigEndpoint(svc)),
|
|
|
|
GetDistributedQueries: authenticatedHost(svc, makeGetDistributedQueriesEndpoint(svc)),
|
|
|
|
SubmitDistributedQueryResults: authenticatedHost(svc, makeSubmitDistributedQueryResultsEndpoint(svc)),
|
|
|
|
SubmitLogs: authenticatedHost(svc, makeSubmitLogsEndpoint(svc)),
|
2016-09-26 17:14:39 +00:00
|
|
|
}
|
|
|
|
}
|
2016-09-04 05:13:42 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
type kolideHandlers struct {
|
2016-11-16 21:07:50 +00:00
|
|
|
Login *kithttp.Server
|
|
|
|
Logout *kithttp.Server
|
|
|
|
ForgotPassword *kithttp.Server
|
|
|
|
ResetPassword *kithttp.Server
|
|
|
|
Me *kithttp.Server
|
2016-12-14 18:11:43 +00:00
|
|
|
ChangePassword *kithttp.Server
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateUser *kithttp.Server
|
|
|
|
GetUser *kithttp.Server
|
|
|
|
ListUsers *kithttp.Server
|
|
|
|
ModifyUser *kithttp.Server
|
|
|
|
GetSessionsForUserInfo *kithttp.Server
|
|
|
|
DeleteSessionsForUser *kithttp.Server
|
|
|
|
GetSessionInfo *kithttp.Server
|
|
|
|
DeleteSession *kithttp.Server
|
|
|
|
GetAppConfig *kithttp.Server
|
|
|
|
ModifyAppConfig *kithttp.Server
|
|
|
|
CreateInvite *kithttp.Server
|
|
|
|
ListInvites *kithttp.Server
|
|
|
|
DeleteInvite *kithttp.Server
|
|
|
|
GetQuery *kithttp.Server
|
|
|
|
ListQueries *kithttp.Server
|
|
|
|
CreateQuery *kithttp.Server
|
|
|
|
ModifyQuery *kithttp.Server
|
|
|
|
DeleteQuery *kithttp.Server
|
2016-12-09 17:12:45 +00:00
|
|
|
DeleteQueries *kithttp.Server
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateDistributedQueryCampaign *kithttp.Server
|
|
|
|
GetPack *kithttp.Server
|
|
|
|
ListPacks *kithttp.Server
|
|
|
|
CreatePack *kithttp.Server
|
|
|
|
ModifyPack *kithttp.Server
|
|
|
|
DeletePack *kithttp.Server
|
2016-12-13 22:22:05 +00:00
|
|
|
ScheduleQueries *kithttp.Server
|
|
|
|
GetScheduledQueriesInPack *kithttp.Server
|
|
|
|
GetScheduledQuery *kithttp.Server
|
|
|
|
ModifyScheduledQuery *kithttp.Server
|
|
|
|
DeleteScheduledQuery *kithttp.Server
|
2016-11-16 21:07:50 +00:00
|
|
|
EnrollAgent *kithttp.Server
|
|
|
|
GetClientConfig *kithttp.Server
|
|
|
|
GetDistributedQueries *kithttp.Server
|
|
|
|
SubmitDistributedQueryResults *kithttp.Server
|
|
|
|
SubmitLogs *kithttp.Server
|
|
|
|
GetLabel *kithttp.Server
|
|
|
|
ListLabels *kithttp.Server
|
|
|
|
CreateLabel *kithttp.Server
|
|
|
|
DeleteLabel *kithttp.Server
|
|
|
|
AddLabelToPack *kithttp.Server
|
|
|
|
GetLabelsForPack *kithttp.Server
|
|
|
|
DeleteLabelFromPack *kithttp.Server
|
|
|
|
GetHost *kithttp.Server
|
|
|
|
DeleteHost *kithttp.Server
|
|
|
|
ListHosts *kithttp.Server
|
|
|
|
SearchTargets *kithttp.Server
|
2016-09-26 17:14:39 +00:00
|
|
|
}
|
2016-09-04 05:13:42 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
func makeKolideKitHandlers(ctx context.Context, e KolideEndpoints, opts []kithttp.ServerOption) kolideHandlers {
|
|
|
|
newServer := func(e endpoint.Endpoint, decodeFn kithttp.DecodeRequestFunc) *kithttp.Server {
|
|
|
|
return kithttp.NewServer(ctx, e, decodeFn, encodeResponse, opts...)
|
|
|
|
}
|
|
|
|
return kolideHandlers{
|
2016-11-16 21:07:50 +00:00
|
|
|
Login: newServer(e.Login, decodeLoginRequest),
|
|
|
|
Logout: newServer(e.Logout, decodeNoParamsRequest),
|
|
|
|
ForgotPassword: newServer(e.ForgotPassword, decodeForgotPasswordRequest),
|
|
|
|
ResetPassword: newServer(e.ResetPassword, decodeResetPasswordRequest),
|
|
|
|
Me: newServer(e.Me, decodeNoParamsRequest),
|
2016-12-14 18:11:43 +00:00
|
|
|
ChangePassword: newServer(e.ChangePassword, decodeChangePasswordRequest),
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateUser: newServer(e.CreateUser, decodeCreateUserRequest),
|
|
|
|
GetUser: newServer(e.GetUser, decodeGetUserRequest),
|
|
|
|
ListUsers: newServer(e.ListUsers, decodeListUsersRequest),
|
|
|
|
ModifyUser: newServer(e.ModifyUser, decodeModifyUserRequest),
|
|
|
|
GetSessionsForUserInfo: newServer(e.GetSessionsForUserInfo, decodeGetInfoAboutSessionsForUserRequest),
|
|
|
|
DeleteSessionsForUser: newServer(e.DeleteSessionsForUser, decodeDeleteSessionsForUserRequest),
|
|
|
|
GetSessionInfo: newServer(e.GetSessionInfo, decodeGetInfoAboutSessionRequest),
|
|
|
|
DeleteSession: newServer(e.DeleteSession, decodeDeleteSessionRequest),
|
|
|
|
GetAppConfig: newServer(e.GetAppConfig, decodeNoParamsRequest),
|
|
|
|
ModifyAppConfig: newServer(e.ModifyAppConfig, decodeModifyAppConfigRequest),
|
|
|
|
CreateInvite: newServer(e.CreateInvite, decodeCreateInviteRequest),
|
|
|
|
ListInvites: newServer(e.ListInvites, decodeListInvitesRequest),
|
|
|
|
DeleteInvite: newServer(e.DeleteInvite, decodeDeleteInviteRequest),
|
|
|
|
GetQuery: newServer(e.GetQuery, decodeGetQueryRequest),
|
|
|
|
ListQueries: newServer(e.ListQueries, decodeListQueriesRequest),
|
|
|
|
CreateQuery: newServer(e.CreateQuery, decodeCreateQueryRequest),
|
|
|
|
ModifyQuery: newServer(e.ModifyQuery, decodeModifyQueryRequest),
|
|
|
|
DeleteQuery: newServer(e.DeleteQuery, decodeDeleteQueryRequest),
|
2016-12-09 17:12:45 +00:00
|
|
|
DeleteQueries: newServer(e.DeleteQueries, decodeDeleteQueriesRequest),
|
2016-11-16 21:07:50 +00:00
|
|
|
CreateDistributedQueryCampaign: newServer(e.CreateDistributedQueryCampaign, decodeCreateDistributedQueryCampaignRequest),
|
2016-09-29 04:21:39 +00:00
|
|
|
GetPack: newServer(e.GetPack, decodeGetPackRequest),
|
2016-10-13 18:21:47 +00:00
|
|
|
ListPacks: newServer(e.ListPacks, decodeListPacksRequest),
|
2016-09-29 04:21:39 +00:00
|
|
|
CreatePack: newServer(e.CreatePack, decodeCreatePackRequest),
|
|
|
|
ModifyPack: newServer(e.ModifyPack, decodeModifyPackRequest),
|
|
|
|
DeletePack: newServer(e.DeletePack, decodeDeletePackRequest),
|
2016-12-13 22:22:05 +00:00
|
|
|
ScheduleQueries: newServer(e.ScheduleQueries, decodeScheduleQueriesRequest),
|
|
|
|
GetScheduledQueriesInPack: newServer(e.GetScheduledQueriesInPack, decodeGetScheduledQueriesInPackRequest),
|
|
|
|
GetScheduledQuery: newServer(e.GetScheduledQuery, decodeGetScheduledQueryRequest),
|
|
|
|
ModifyScheduledQuery: newServer(e.ModifyScheduledQuery, decodeModifyScheduledQueryRequest),
|
|
|
|
DeleteScheduledQuery: newServer(e.DeleteScheduledQuery, decodeDeleteScheduledQueryRequest),
|
2016-09-29 04:21:39 +00:00
|
|
|
EnrollAgent: newServer(e.EnrollAgent, decodeEnrollAgentRequest),
|
|
|
|
GetClientConfig: newServer(e.GetClientConfig, decodeGetClientConfigRequest),
|
|
|
|
GetDistributedQueries: newServer(e.GetDistributedQueries, decodeGetDistributedQueriesRequest),
|
|
|
|
SubmitDistributedQueryResults: newServer(e.SubmitDistributedQueryResults, decodeSubmitDistributedQueryResultsRequest),
|
|
|
|
SubmitLogs: newServer(e.SubmitLogs, decodeSubmitLogsRequest),
|
2016-10-03 03:14:35 +00:00
|
|
|
GetLabel: newServer(e.GetLabel, decodeGetLabelRequest),
|
2016-10-13 18:21:47 +00:00
|
|
|
ListLabels: newServer(e.ListLabels, decodeListLabelsRequest),
|
2016-10-03 03:14:35 +00:00
|
|
|
CreateLabel: newServer(e.CreateLabel, decodeCreateLabelRequest),
|
|
|
|
DeleteLabel: newServer(e.DeleteLabel, decodeDeleteLabelRequest),
|
|
|
|
AddLabelToPack: newServer(e.AddLabelToPack, decodeAddLabelToPackRequest),
|
|
|
|
GetLabelsForPack: newServer(e.GetLabelsForPack, decodeGetLabelsForPackRequest),
|
|
|
|
DeleteLabelFromPack: newServer(e.DeleteLabelFromPack, decodeDeleteLabelFromPackRequest),
|
2016-10-06 00:10:44 +00:00
|
|
|
GetHost: newServer(e.GetHost, decodeGetHostRequest),
|
|
|
|
DeleteHost: newServer(e.DeleteHost, decodeDeleteHostRequest),
|
2016-10-13 18:21:47 +00:00
|
|
|
ListHosts: newServer(e.ListHosts, decodeListHostsRequest),
|
2016-11-02 14:59:53 +00:00
|
|
|
SearchTargets: newServer(e.SearchTargets, decodeSearchTargetsRequest),
|
2016-09-26 17:14:39 +00:00
|
|
|
}
|
2016-09-04 19:43:12 +00:00
|
|
|
}
|
2016-09-01 04:51:38 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
// MakeHandler creates an HTTP handler for the Kolide server endpoints.
|
|
|
|
func MakeHandler(ctx context.Context, svc kolide.Service, jwtKey string, logger kitlog.Logger) http.Handler {
|
|
|
|
kolideAPIOptions := []kithttp.ServerOption{
|
2016-09-04 19:43:12 +00:00
|
|
|
kithttp.ServerBefore(
|
2016-09-26 17:14:39 +00:00
|
|
|
setRequestsContexts(svc, jwtKey),
|
2016-09-04 19:43:12 +00:00
|
|
|
),
|
|
|
|
kithttp.ServerErrorLogger(logger),
|
2016-09-23 02:41:58 +00:00
|
|
|
kithttp.ServerErrorEncoder(encodeError),
|
2016-09-04 19:43:12 +00:00
|
|
|
kithttp.ServerAfter(
|
|
|
|
kithttp.SetContentType("application/json; charset=utf-8"),
|
|
|
|
),
|
|
|
|
}
|
2016-08-28 03:59:17 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
kolideEndpoints := MakeKolideServerEndpoints(svc, jwtKey)
|
|
|
|
kolideHandlers := makeKolideKitHandlers(ctx, kolideEndpoints, kolideAPIOptions)
|
|
|
|
|
2016-09-04 19:43:12 +00:00
|
|
|
r := mux.NewRouter()
|
2016-09-26 17:14:39 +00:00
|
|
|
attachKolideAPIRoutes(r, kolideHandlers)
|
2016-11-23 00:35:43 +00:00
|
|
|
r.HandleFunc("/api/v1/kolide/results/{id}",
|
|
|
|
makeStreamDistributedQueryCampaignResultsHandler(svc, jwtKey)).
|
|
|
|
Methods("GET")
|
2016-09-05 14:15:58 +00:00
|
|
|
|
2016-08-28 03:59:17 +00:00
|
|
|
return r
|
|
|
|
}
|
2016-09-26 17:14:39 +00:00
|
|
|
|
|
|
|
func attachKolideAPIRoutes(r *mux.Router, h kolideHandlers) {
|
|
|
|
r.Handle("/api/v1/kolide/login", h.Login).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/logout", h.Logout).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/forgot_password", h.ForgotPassword).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/reset_password", h.ResetPassword).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/me", h.Me).Methods("GET")
|
2016-12-14 18:11:43 +00:00
|
|
|
r.Handle("/api/v1/kolide/change_password", h.ChangePassword).Methods("POST")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/users", h.ListUsers).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/users", h.CreateUser).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/users/{id}", h.GetUser).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/users/{id}", h.ModifyUser).Methods("PATCH")
|
|
|
|
r.Handle("/api/v1/kolide/users/{id}/sessions", h.GetSessionsForUserInfo).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/users/{id}/sessions", h.DeleteSessionsForUser).Methods("DELETE")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/sessions/{id}", h.GetSessionInfo).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/sessions/{id}", h.DeleteSession).Methods("DELETE")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/config", h.GetAppConfig).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/config", h.ModifyAppConfig).Methods("PATCH")
|
2016-09-29 02:44:05 +00:00
|
|
|
r.Handle("/api/v1/kolide/invites", h.CreateInvite).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/invites", h.ListInvites).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/invites/{id}", h.DeleteInvite).Methods("DELETE")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/queries/{id}", h.GetQuery).Methods("GET")
|
2016-10-07 17:26:48 +00:00
|
|
|
r.Handle("/api/v1/kolide/queries", h.ListQueries).Methods("GET")
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/queries", h.CreateQuery).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/queries/{id}", h.ModifyQuery).Methods("PATCH")
|
|
|
|
r.Handle("/api/v1/kolide/queries/{id}", h.DeleteQuery).Methods("DELETE")
|
2016-12-09 17:12:45 +00:00
|
|
|
r.Handle("/api/v1/kolide/queries/delete", h.DeleteQueries).Methods("POST")
|
2016-11-16 21:07:50 +00:00
|
|
|
r.Handle("/api/v1/kolide/queries/run", h.CreateDistributedQueryCampaign).Methods("POST")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/packs/{id}", h.GetPack).Methods("GET")
|
2016-10-07 17:26:48 +00:00
|
|
|
r.Handle("/api/v1/kolide/packs", h.ListPacks).Methods("GET")
|
2016-09-26 17:14:39 +00:00
|
|
|
r.Handle("/api/v1/kolide/packs", h.CreatePack).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/packs/{id}", h.ModifyPack).Methods("PATCH")
|
|
|
|
r.Handle("/api/v1/kolide/packs/{id}", h.DeletePack).Methods("DELETE")
|
2016-12-13 22:22:05 +00:00
|
|
|
r.Handle("/api/v1/kolide/packs/{id}/scheduled", h.GetScheduledQueriesInPack).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/schedule", h.ScheduleQueries).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/schedule/{id}", h.GetScheduledQuery).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/schedule/{id}", h.ModifyScheduledQuery).Methods("PATCH")
|
|
|
|
r.Handle("/api/v1/kolide/schedule/{id}", h.DeleteScheduledQuery).Methods("DELETE")
|
2016-10-03 03:14:35 +00:00
|
|
|
r.Handle("/api/v1/kolide/labels/{id}", h.GetLabel).Methods("GET")
|
2016-10-07 17:26:48 +00:00
|
|
|
r.Handle("/api/v1/kolide/labels", h.ListLabels).Methods("GET")
|
2016-10-03 03:14:35 +00:00
|
|
|
r.Handle("/api/v1/kolide/labels", h.CreateLabel).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/labels/{id}", h.DeleteLabel).Methods("DELETE")
|
|
|
|
r.Handle("/api/v1/kolide/packs/{pid}/labels/{lid}", h.AddLabelToPack).Methods("POST")
|
|
|
|
r.Handle("/api/v1/kolide/packs/{pid}/labels", h.GetLabelsForPack).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/packs/{pid}/labels/{lid}", h.DeleteLabelFromPack).Methods("DELETE")
|
2016-09-29 04:21:39 +00:00
|
|
|
|
2016-10-07 17:26:48 +00:00
|
|
|
r.Handle("/api/v1/kolide/hosts", h.ListHosts).Methods("GET")
|
2016-10-06 00:10:44 +00:00
|
|
|
r.Handle("/api/v1/kolide/hosts/{id}", h.GetHost).Methods("GET")
|
|
|
|
r.Handle("/api/v1/kolide/hosts/{id}", h.DeleteHost).Methods("DELETE")
|
|
|
|
|
2016-11-02 14:59:53 +00:00
|
|
|
r.Handle("/api/v1/kolide/targets", h.SearchTargets).Methods("POST")
|
|
|
|
|
2016-09-29 04:21:39 +00:00
|
|
|
r.Handle("/api/v1/osquery/enroll", h.EnrollAgent).Methods("POST")
|
|
|
|
r.Handle("/api/v1/osquery/config", h.GetClientConfig).Methods("POST")
|
|
|
|
r.Handle("/api/v1/osquery/distributed/read", h.GetDistributedQueries).Methods("POST")
|
|
|
|
r.Handle("/api/v1/osquery/distributed/write", h.SubmitDistributedQueryResults).Methods("POST")
|
|
|
|
r.Handle("/api/v1/osquery/log", h.SubmitLogs).Methods("POST")
|
2016-09-26 17:14:39 +00:00
|
|
|
}
|
2016-11-09 17:19:07 +00:00
|
|
|
|
|
|
|
// WithSetup is an http middleware that checks if a database user exists.
|
|
|
|
// If one does, it serves the API with a setup middleware.
|
|
|
|
// If the server is already configured, the default API handler is exposed.
|
2016-12-02 18:46:31 +00:00
|
|
|
func WithSetup(svc kolide.Service, logger kitlog.Logger, next http.Handler) http.HandlerFunc {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
configRouter := http.NewServeMux()
|
|
|
|
configRouter.Handle("/api/v1/kolide/config", http.HandlerFunc(forceSetup))
|
|
|
|
configRouter.Handle("/api/v1/setup", kithttp.NewServer(
|
|
|
|
context.Background(),
|
|
|
|
makeSetupEndpoint(svc),
|
|
|
|
decodeSetupRequest,
|
|
|
|
encodeResponse,
|
|
|
|
))
|
|
|
|
if RequireSetup(svc, logger) {
|
|
|
|
configRouter.ServeHTTP(w, r)
|
|
|
|
} else {
|
|
|
|
next.ServeHTTP(w, r)
|
|
|
|
}
|
2016-11-09 17:19:07 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func forceSetup(w http.ResponseWriter, r *http.Request) {
|
|
|
|
response := map[string]bool{
|
|
|
|
"require_setup": true,
|
|
|
|
}
|
|
|
|
if err := json.NewEncoder(w).Encode(&response); err != nil {
|
|
|
|
encodeError(context.Background(), err, w)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// RequireSetup checks if the service must be configured by an administrator.
|
|
|
|
func RequireSetup(svc kolide.Service, logger kitlog.Logger) bool {
|
2016-11-10 00:35:49 +00:00
|
|
|
users, err := svc.ListUsers(context.Background(), kolide.ListOptions{Page: 0, PerPage: 1})
|
2016-11-09 17:19:07 +00:00
|
|
|
if err != nil {
|
|
|
|
logger.Log("err", err)
|
2016-12-02 18:46:31 +00:00
|
|
|
return false
|
2016-11-09 17:19:07 +00:00
|
|
|
}
|
|
|
|
return len(users) == 0
|
|
|
|
}
|