wip: migrate to mono-repo. SPN has already been moved to spn/
This commit is contained in:
99
service/process/api.go
Normal file
99
service/process/api.go
Normal file
@@ -0,0 +1,99 @@
|
||||
package process
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
"github.com/safing/portbase/api"
|
||||
"github.com/safing/portmaster/service/profile"
|
||||
)
|
||||
|
||||
func registerAPIEndpoints() error {
|
||||
if err := api.RegisterEndpoint(api.Endpoint{
|
||||
Name: "Get Process Tag Metadata",
|
||||
Description: "Get information about process tags.",
|
||||
Path: "process/tags",
|
||||
Read: api.PermitUser,
|
||||
BelongsTo: module,
|
||||
StructFunc: handleProcessTagMetadata,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := api.RegisterEndpoint(api.Endpoint{
|
||||
Name: "Get Processes by Profile",
|
||||
Description: "Get all recently active processes using the given profile",
|
||||
Path: "process/list/by-profile/{source:[a-z]+}/{id:[A-z0-9-]+}",
|
||||
Read: api.PermitUser,
|
||||
BelongsTo: module,
|
||||
StructFunc: handleGetProcessesByProfile,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := api.RegisterEndpoint(api.Endpoint{
|
||||
Name: "Get Process Group Leader By PID",
|
||||
Description: "Load a process group leader by a child PID",
|
||||
Path: "process/group-leader/{pid:[0-9]+}",
|
||||
Read: api.PermitUser,
|
||||
BelongsTo: module,
|
||||
StructFunc: handleGetProcessGroupLeader,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func handleProcessTagMetadata(ar *api.Request) (i interface{}, err error) {
|
||||
tagRegistryLock.Lock()
|
||||
defer tagRegistryLock.Unlock()
|
||||
|
||||
// Create response struct.
|
||||
resp := struct {
|
||||
Tags []TagDescription
|
||||
}{
|
||||
Tags: make([]TagDescription, 0, len(tagRegistry)*2),
|
||||
}
|
||||
|
||||
// Get all tag descriptions.
|
||||
for _, th := range tagRegistry {
|
||||
resp.Tags = append(resp.Tags, th.TagDescriptions()...)
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
func handleGetProcessesByProfile(ar *api.Request) (any, error) {
|
||||
source := ar.URLVars["source"]
|
||||
id := ar.URLVars["id"]
|
||||
if id == "" || source == "" {
|
||||
return nil, api.ErrorWithStatus(fmt.Errorf("missing profile source/id"), http.StatusBadRequest)
|
||||
}
|
||||
|
||||
result := GetProcessesWithProfile(ar.Context(), profile.ProfileSource(source), id, true)
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func handleGetProcessGroupLeader(ar *api.Request) (any, error) {
|
||||
pid, err := strconv.ParseInt(ar.URLVars["pid"], 10, 0)
|
||||
if err != nil {
|
||||
return nil, api.ErrorWithStatus(err, http.StatusBadRequest)
|
||||
}
|
||||
|
||||
process, err := GetOrFindProcess(ar.Context(), int(pid))
|
||||
if err != nil {
|
||||
return nil, api.ErrorWithStatus(err, http.StatusInternalServerError)
|
||||
}
|
||||
err = process.FindProcessGroupLeader(ar.Context())
|
||||
switch {
|
||||
case process.Leader() != nil:
|
||||
return process.Leader(), nil
|
||||
case err != nil:
|
||||
return nil, api.ErrorWithStatus(err, http.StatusInternalServerError)
|
||||
default:
|
||||
return nil, api.ErrorWithStatus(errors.New("leader not found"), http.StatusNotFound)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user