mirror of
https://github.com/fleetdm/fleet
synced 2026-05-08 09:40:49 +00:00
091e11917e
> Closes https://github.com/fleetdm/fleet/issues/30678 # Checklist for submitter If some of the following don't apply, delete the relevant line. <!-- Note that API documentation changes are now addressed by the product design team. --> - [x] Input data is properly validated, `SELECT *` is avoided, SQL injection is prevented (using placeholders for values in statements) - [x] Added/updated automated tests - [x] Manual QA for all new/changed functionality - [x] For unreleased bug fixes in a release candidate, confirmed that the fix is not expected to adversely impact load test results or alerted the release DRI if additional load testing is needed.
84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package worker
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/contexts/ctxerr"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
apple_mdm "github.com/fleetdm/fleet/v4/server/mdm/apple"
|
|
kitlog "github.com/go-kit/log"
|
|
"github.com/go-kit/log/level"
|
|
)
|
|
|
|
const AppleSoftwareJobName = "apple_software"
|
|
|
|
type AppleSoftwareTask string
|
|
|
|
const VerifyVPPTask AppleSoftwareTask = "verify_vpp_installs"
|
|
|
|
type AppleSoftware struct {
|
|
Datastore fleet.Datastore
|
|
Commander *apple_mdm.MDMAppleCommander
|
|
Log kitlog.Logger
|
|
}
|
|
|
|
func (v *AppleSoftware) Name() string {
|
|
return AppleSoftwareJobName
|
|
}
|
|
|
|
type appleSoftwareArgs struct {
|
|
Task AppleSoftwareTask `json:"task"`
|
|
HostUUID string `json:"host_uuid"`
|
|
VerificationCommandUUID string `json:"verification_command_uuid"`
|
|
}
|
|
|
|
func (v *AppleSoftware) Run(ctx context.Context, argsJSON json.RawMessage) error {
|
|
var args appleSoftwareArgs
|
|
if err := json.Unmarshal(argsJSON, &args); err != nil {
|
|
return ctxerr.Wrap(ctx, err, "unmarshal args")
|
|
}
|
|
|
|
switch args.Task {
|
|
case VerifyVPPTask:
|
|
err := v.verifyVPPInstalls(ctx, args.HostUUID, args.VerificationCommandUUID)
|
|
return ctxerr.Wrap(ctx, err, "running migrate VPP token task")
|
|
|
|
default:
|
|
return ctxerr.Errorf(ctx, "unknown task: %v", args.Task)
|
|
}
|
|
}
|
|
|
|
func (v *AppleSoftware) verifyVPPInstalls(ctx context.Context, hostUUID, verificationCommandUUID string) error {
|
|
level.Debug(v.Log).Log("msg", "verifying VPP installs", "host_uuid", hostUUID, "verification_command_uuid", verificationCommandUUID)
|
|
newListCmdUUID := fleet.VerifySoftwareInstallCommandUUID()
|
|
err := v.Commander.InstalledApplicationList(ctx, []string{hostUUID}, newListCmdUUID, true)
|
|
if err != nil {
|
|
return ctxerr.Wrap(ctx, err, "sending installed application list command in verify")
|
|
}
|
|
|
|
if err := v.Datastore.ReplaceVPPInstallVerificationUUID(ctx, verificationCommandUUID, newListCmdUUID); err != nil {
|
|
return ctxerr.Wrap(ctx, err, "update install record")
|
|
}
|
|
|
|
level.Debug(v.Log).Log("msg", "new installed application list command sent", "uuid", newListCmdUUID)
|
|
|
|
return nil
|
|
}
|
|
|
|
func QueueVPPInstallVerificationJob(ctx context.Context, ds fleet.Datastore, logger kitlog.Logger, task AppleSoftwareTask, requestDelay time.Duration, hostUUID, verificationCommandUUID string) error {
|
|
args := &appleSoftwareArgs{
|
|
Task: task,
|
|
HostUUID: hostUUID,
|
|
VerificationCommandUUID: verificationCommandUUID,
|
|
}
|
|
|
|
job, err := QueueJobWithDelay(ctx, ds, AppleSoftwareJobName, args, requestDelay)
|
|
if err != nil {
|
|
return ctxerr.Wrap(ctx, err, "queueing job")
|
|
}
|
|
|
|
level.Debug(logger).Log("job_id", job.ID, "job_name", appleMDMJobName, "task", task)
|
|
return nil
|
|
}
|