110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CheckDetails is an arbitrary json structure, we would only care about submission_url
|
||
|
|
type CheckDetails struct {
|
||
|
|
SubmissionURL string `json:"submission_url"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check definition
|
||
|
|
type Check struct {
|
||
|
|
Cid string `json:"_cid"`
|
||
|
|
Active bool `json:"_active"`
|
||
|
|
BrokerCid string `json:"_broker"`
|
||
|
|
CheckBundleCid string `json:"_check_bundle"`
|
||
|
|
CheckUUID string `json:"_check_uuid"`
|
||
|
|
Details CheckDetails `json:"_details"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// FetchCheckByID fetch a check configuration by id
|
||
|
|
func (a *API) FetchCheckByID(id IDType) (*Check, error) {
|
||
|
|
cid := CIDType(fmt.Sprintf("/check/%d", int(id)))
|
||
|
|
return a.FetchCheckByCID(cid)
|
||
|
|
}
|
||
|
|
|
||
|
|
// FetchCheckByCID fetch a check configuration by cid
|
||
|
|
func (a *API) FetchCheckByCID(cid CIDType) (*Check, error) {
|
||
|
|
result, err := a.Get(string(cid))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
check := new(Check)
|
||
|
|
json.Unmarshal(result, check)
|
||
|
|
|
||
|
|
return check, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// FetchCheckBySubmissionURL fetch a check configuration by submission_url
|
||
|
|
func (a *API) FetchCheckBySubmissionURL(submissionURL URLType) (*Check, error) {
|
||
|
|
|
||
|
|
u, err := url.Parse(string(submissionURL))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// valid trap url: scheme://host[:port]/module/httptrap/UUID/secret
|
||
|
|
|
||
|
|
// does it smell like a valid trap url path
|
||
|
|
if u.Path[:17] != "/module/httptrap/" {
|
||
|
|
return nil, fmt.Errorf("[ERROR] Invalid submission URL '%s', unrecognized path", submissionURL)
|
||
|
|
}
|
||
|
|
|
||
|
|
// extract uuid/secret
|
||
|
|
pathParts := strings.Split(u.Path[17:len(u.Path)], "/")
|
||
|
|
if len(pathParts) != 2 {
|
||
|
|
return nil, fmt.Errorf("[ERROR] Invalid submission URL '%s', UUID not where expected", submissionURL)
|
||
|
|
}
|
||
|
|
|
||
|
|
uuid := pathParts[0]
|
||
|
|
|
||
|
|
query := SearchQueryType(fmt.Sprintf("f__check_uuid=%s", uuid))
|
||
|
|
|
||
|
|
checks, err := a.CheckSearch(query)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(checks) == 0 {
|
||
|
|
return nil, fmt.Errorf("[ERROR] No checks found with UUID %s", uuid)
|
||
|
|
}
|
||
|
|
|
||
|
|
numActive := 0
|
||
|
|
checkID := -1
|
||
|
|
|
||
|
|
for idx, check := range checks {
|
||
|
|
if check.Active {
|
||
|
|
numActive++
|
||
|
|
checkID = idx
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if numActive > 1 {
|
||
|
|
return nil, fmt.Errorf("[ERROR] Multiple checks with same UUID %s", uuid)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &checks[checkID], nil
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// CheckSearch returns a list of checks matching a query/filter
|
||
|
|
func (a *API) CheckSearch(query SearchQueryType) ([]Check, error) {
|
||
|
|
queryURL := fmt.Sprintf("/check?%s", string(query))
|
||
|
|
|
||
|
|
result, err := a.Get(queryURL)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
var checks []Check
|
||
|
|
json.Unmarshal(result, &checks)
|
||
|
|
|
||
|
|
return checks, nil
|
||
|
|
}
|