mirror of
https://github.com/fleetdm/fleet
synced 2026-05-22 08:28:52 +00:00
27b8a1364f
> Related issue; #19526 # 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] Changes file added for user-visible changes in `changes/`, `orbit/changes/` or `ee/fleetd-chrome/changes`. See [Changes files](https://fleetdm.com/docs/contributing/committing-changes#changes-files) for more information. - [x] Added/updated tests - [x] Manual QA for all new/changed functionality
102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package s3
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"path"
|
|
|
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
"github.com/fleetdm/fleet/v4/server/config"
|
|
"github.com/fleetdm/fleet/v4/server/contexts/ctxerr"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
)
|
|
|
|
const (
|
|
desktopPath = "desktop"
|
|
executable = "fleet-osquery"
|
|
)
|
|
|
|
type installerNotFoundError struct{}
|
|
|
|
var _ fleet.NotFoundError = (*installerNotFoundError)(nil)
|
|
|
|
func (p installerNotFoundError) Error() string {
|
|
return "installer not found"
|
|
}
|
|
|
|
func (p installerNotFoundError) IsNotFound() bool {
|
|
return true
|
|
}
|
|
|
|
// InstallerStore contains methods to retrieve installers from S3
|
|
type InstallerStore struct {
|
|
*s3store
|
|
}
|
|
|
|
// NewInstallerStore creates a new instance with the given S3 config
|
|
func NewInstallerStore(config config.S3Config) (*InstallerStore, error) {
|
|
s3store, err := newS3store(config.CarvesToInternalCfg())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &InstallerStore{s3store}, nil
|
|
}
|
|
|
|
// Get retrieves the requested installer from S3
|
|
func (i *InstallerStore) Get(ctx context.Context, installer fleet.Installer) (io.ReadCloser, int64, error) {
|
|
key := i.keyForInstaller(installer)
|
|
req, err := i.s3client.GetObject(&s3.GetObjectInput{Bucket: &i.bucket, Key: &key})
|
|
if err != nil {
|
|
if aerr, ok := err.(awserr.Error); ok {
|
|
switch aerr.Code() {
|
|
case s3.ErrCodeNoSuchKey, s3.ErrCodeNoSuchBucket, "NotFound":
|
|
return nil, int64(0), installerNotFoundError{}
|
|
}
|
|
}
|
|
|
|
return nil, int64(0), ctxerr.Wrap(ctx, err, "retrieving installer from store")
|
|
}
|
|
|
|
return req.Body, *req.ContentLength, nil
|
|
}
|
|
|
|
// Put uploads an installer to S3
|
|
func (i *InstallerStore) Put(ctx context.Context, installer fleet.Installer) (string, error) {
|
|
key := i.keyForInstaller(installer)
|
|
_, err := i.s3client.PutObject(&s3.PutObjectInput{
|
|
Bucket: &i.bucket,
|
|
Body: installer.Content,
|
|
Key: &key,
|
|
})
|
|
return key, err
|
|
}
|
|
|
|
// Exists checks if an installer exists in the S3 bucket
|
|
func (i *InstallerStore) Exists(ctx context.Context, installer fleet.Installer) (bool, error) {
|
|
key := i.keyForInstaller(installer)
|
|
_, err := i.s3client.HeadObject(&s3.HeadObjectInput{Bucket: &i.bucket, Key: &key})
|
|
if err != nil {
|
|
if aerr, ok := err.(awserr.Error); ok {
|
|
switch aerr.Code() {
|
|
case s3.ErrCodeNoSuchKey, s3.ErrCodeNoSuchBucket, "NotFound":
|
|
return false, nil
|
|
}
|
|
}
|
|
|
|
return false, ctxerr.Wrap(ctx, err, "checking existence on file store")
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// keyForInstaller builds an S3 key to search for the installer
|
|
func (i *InstallerStore) keyForInstaller(installer fleet.Installer) string {
|
|
file := fmt.Sprintf("%s.%s", executable, installer.Kind)
|
|
dir := ""
|
|
if installer.Desktop {
|
|
dir = desktopPath
|
|
}
|
|
return path.Join(i.prefix, installer.EnrollSecret, dir, file)
|
|
}
|