mirror of
https://github.com/fleetdm/fleet
synced 2026-05-14 20:48:35 +00:00
6014908ab6
h/t https://github.com/binref/refinery/issues/72, for #24720. No changes file as this is an unreleased bug. Also added output for version in the custom package parser tool. # 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://github.com/fleetdm/fleet/blob/main/docs/Contributing/Committing-Changes.md#changes-files) for more information. - [x] Manual QA for all new/changed functionality
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/fleetdm/fleet/v4/pkg/file"
|
|
"github.com/fleetdm/fleet/v4/pkg/fleethttp"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
)
|
|
|
|
func main() {
|
|
url := flag.String("url", "", "URL of the custom package")
|
|
path := flag.String("path", "", "File path of the custom package")
|
|
flag.Parse()
|
|
|
|
if *url == "" && *path == "" {
|
|
log.Fatal("missing -url or -path argument")
|
|
}
|
|
if *url != "" && *path != "" {
|
|
log.Fatal("cannot set both -url and -path")
|
|
}
|
|
|
|
metadata, err := processPackage(*url, *path)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Printf(
|
|
"- Name: '%s'\n- Bundle Identifier: '%s'\n- Package IDs: '%s'\n- Version: %s\n\n",
|
|
metadata.Name, metadata.BundleIdentifier, strings.Join(metadata.PackageIDs, ","), metadata.Version,
|
|
)
|
|
}
|
|
|
|
func processPackage(url, path string) (*file.InstallerMetadata, error) {
|
|
var tfr *fleet.TempFileReader
|
|
if url != "" {
|
|
client := fleethttp.NewClient()
|
|
client.Transport = fleethttp.NewSizeLimitTransport(fleet.MaxSoftwareInstallerSize)
|
|
|
|
req, err := http.NewRequest(http.MethodGet, url, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create http request: %s", err)
|
|
}
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("get request: %s", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// Allow all 2xx and 3xx status codes in this pass.
|
|
if resp.StatusCode >= 400 {
|
|
return nil, fmt.Errorf("get request failed with status: %d", resp.StatusCode)
|
|
}
|
|
|
|
tfr, err = fleet.NewTempFileReader(resp.Body, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("reading custom package: %d", resp.StatusCode)
|
|
}
|
|
defer tfr.Close()
|
|
} else { // -path
|
|
fp, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("open file: %s", err)
|
|
}
|
|
tfr = &fleet.TempFileReader{
|
|
File: fp,
|
|
}
|
|
}
|
|
|
|
metadata, err := file.ExtractInstallerMetadata(tfr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("extract installer metadata: %s", err)
|
|
}
|
|
return metadata, nil
|
|
}
|