mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-05 20:40:16 -08:00
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package hooks
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/aquasecurity/trivy/pkg/cloud"
|
|
"github.com/aquasecurity/trivy/pkg/flag"
|
|
"github.com/aquasecurity/trivy/pkg/log"
|
|
"github.com/aquasecurity/trivy/pkg/types"
|
|
xhttp "github.com/aquasecurity/trivy/pkg/x/http"
|
|
)
|
|
|
|
const (
|
|
presignedUploadUrl = "/trivy-reports/upload-url"
|
|
)
|
|
|
|
type CloudPlatformResultsHook struct {
|
|
name string
|
|
cloudConfig *cloud.Config
|
|
client *http.Client
|
|
logger *log.Logger
|
|
}
|
|
|
|
func NewResultsHook(cloudCfg *cloud.Config) *CloudPlatformResultsHook {
|
|
return &CloudPlatformResultsHook{
|
|
name: "Trivy Cloud Results Hook",
|
|
cloudConfig: cloudCfg,
|
|
client: xhttp.Client(),
|
|
logger: log.WithPrefix(log.PrefixCloud),
|
|
}
|
|
}
|
|
|
|
func (h *CloudPlatformResultsHook) Name() string {
|
|
return h.name
|
|
}
|
|
|
|
// PreReport is not going go to be called so we return nil
|
|
func (h *CloudPlatformResultsHook) PreReport(_ context.Context, _ *types.Report, _ flag.Options) error {
|
|
return nil
|
|
}
|
|
|
|
func (h *CloudPlatformResultsHook) PostReport(ctx context.Context, report *types.Report, _ flag.Options) error {
|
|
h.logger.Debug("PostReport called with report")
|
|
jsonReport, err := json.MarshalIndent(report, "", " ")
|
|
if err != nil {
|
|
return xerrors.Errorf("failed to marshal report to JSON: %w", err)
|
|
}
|
|
|
|
return h.uploadResults(ctx, jsonReport)
|
|
}
|
|
|
|
func (h *CloudPlatformResultsHook) uploadResults(ctx context.Context, jsonReport []byte) error {
|
|
uploadUrl, err := h.getPresignedUploadUrl(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get presigned upload URL: %w", err)
|
|
}
|
|
|
|
// create a new request to upload the results
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodPut, uploadUrl, bytes.NewBuffer(jsonReport))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create upload request: %w", err)
|
|
}
|
|
|
|
req.Header.Set("Content-Type", "application/json")
|
|
resp, err := h.client.Do(req)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to upload results: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("failed to upload results: received status code %d", resp.StatusCode)
|
|
}
|
|
|
|
h.logger.Info("Report uploaded successfully to Trivy Cloud")
|
|
return nil
|
|
}
|
|
|
|
func (h *CloudPlatformResultsHook) getPresignedUploadUrl(ctx context.Context) (string, error) {
|
|
uploadUrl, err := url.JoinPath(h.cloudConfig.Api.URL, presignedUploadUrl)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to join API URL and presigned upload URL: %w", err)
|
|
}
|
|
h.logger.Debug("Requesting result upload URL", log.String("uploadUrl", uploadUrl))
|
|
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uploadUrl, http.NoBody)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create request: %w", err)
|
|
}
|
|
|
|
req.Header.Set("Authorization", "Bearer "+h.cloudConfig.Token)
|
|
resp, err := h.client.Do(req)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get upload URL: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return "", fmt.Errorf("failed to get upload URL: %w", err)
|
|
}
|
|
|
|
// read the upload URL from the response
|
|
var uploadResponse struct {
|
|
UploadURL string `json:"uploadUrl"`
|
|
}
|
|
|
|
if err := json.NewDecoder(resp.Body).Decode(&uploadResponse); err != nil {
|
|
return "", xerrors.Errorf("failed to decode upload URL response: %w", err)
|
|
}
|
|
|
|
return uploadResponse.UploadURL, nil
|
|
}
|