示例#1
0
func updateBinaryFile(url string, checksum []byte) {
	fmt.Println("Downloading updated binary")
	httpResp, err := http.Get(url)
	if err != nil {
		glog.Errorf("Cannot download binary: %s", err)
		os.Exit(1)
	}
	defer httpResp.Body.Close()

	binary := httpResp.Body
	if httpResp.ContentLength > 0 {
		bar := pb.New64(httpResp.ContentLength).SetUnits(pb.U_BYTES)
		bar.Start()
		binary = bar.NewProxyReader(binary)
		defer func() {
			<-time.After(bar.RefreshRate)
			fmt.Println()
		}()
	}
	err = update.Apply(binary, update.Options{
		Hash:     crypto.SHA256,
		Checksum: checksum,
	})
	if err != nil {
		glog.Errorf("Cannot apply binary update: %s", err)
		err := update.RollbackError(err)
		if err != nil {
			glog.Errorf("Failed to rollback update: %s", err)
		}
		os.Exit(1)
	}
}
示例#2
0
func downloadChecksum(v semver.Version, downloadBinary, downloadLinkFormat string) ([]byte, error) {
	fmt.Println("Downloading updated binary checksum to validate updated binary")
	u := fmt.Sprintf(downloadLinkFormat, v, downloadBinary+".sha256")
	checksumResp, err := http.Get(u)
	if err != nil {
		return nil, err
	}
	defer checksumResp.Body.Close()

	checksum := checksumResp.Body
	if checksumResp.ContentLength > 0 {
		bar := pb.New64(checksumResp.ContentLength).SetUnits(pb.U_BYTES)
		bar.Start()
		checksum = bar.NewProxyReader(checksum)
		defer func() {
			<-time.After(2 * bar.RefreshRate)
			fmt.Println()
		}()
	}
	b, err := ioutil.ReadAll(checksum)
	if err != nil {
		return nil, err
	}
	if checksumResp.StatusCode != 200 {
		return nil, fmt.Errorf("received %d", checksumResp.StatusCode)
	}

	return hex.DecodeString(strings.TrimSpace(string(b)))
}
示例#3
0
func (m *MachineConfig) CacheMinikubeISOFromURL() error {
	fmt.Println("Downloading ISO")

	// store the miniube-iso inside the .minikube dir
	response, err := http.Get(m.MinikubeISO)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	if response.StatusCode != http.StatusOK {
		return fmt.Errorf("Received %d response from %s while trying to download minikube.iso", response.StatusCode, m.MinikubeISO)
	}

	iso := response.Body

	if response.ContentLength > 0 {
		bar := pb.New64(response.ContentLength).SetUnits(pb.U_BYTES)
		bar.Start()
		iso = bar.NewProxyReader(iso)
		defer func() {
			<-time.After(bar.RefreshRate)
			fmt.Println()
		}()
	}

	out, err := os.Create(m.GetISOCacheFilepath())
	if err != nil {
		return err
	}
	defer out.Close()
	if _, err = io.Copy(out, iso); err != nil {
		return err
	}
	return nil
}
示例#4
0
func main() {
	token := os.Getenv("GH_TOKEN")
	var tc *http.Client
	if len(token) > 0 {
		fmt.Println("Using GH_TOKEN environment variable")
		ts := oauth2.StaticTokenSource(
			&oauth2.Token{AccessToken: token},
		)
		tc = oauth2.NewClient(oauth2.NoContext, ts)
	}
	client := github.NewClient(tc)
	var (
		release *github.RepositoryRelease
		resp    *github.Response
		err     error
	)
	if len(os.Args) > 1 {
		release, resp, err = client.Repositories.GetReleaseByTag("openshift", "origin", os.Args[1])
	} else {
		release, resp, err = client.Repositories.GetLatestRelease("openshift", "origin")
	}
	if err != nil {
		fmt.Printf("Could not get latest OpenShift release: %s", err)
		os.Exit(1)
	}
	defer resp.Body.Close()

	assetID := getOpenShiftServerAssetID(release)
	if assetID == 0 {
		fmt.Println("Could not get OpenShift release URL")
		os.Exit(1)
	}
	asset, url, err := client.Repositories.DownloadReleaseAsset("openshift", "origin", assetID)
	if err != nil {
		fmt.Printf("Could not download OpenShift release asset: %s\n", err)
		os.Exit(1)
	}
	if len(url) > 0 {
		httpResp, err := http.Get(url)
		if err != nil {
			fmt.Printf("Could not download OpenShift release asset: %s\n", err)
			os.Exit(1)
		}
		defer httpResp.Body.Close()

		asset = httpResp.Body
		if httpResp.ContentLength > 0 {
			bar := pb.New64(httpResp.ContentLength).SetUnits(pb.U_BYTES)
			bar.Start()
			asset = bar.NewProxyReader(asset)
			defer func() {
				<-time.After(bar.RefreshRate)
				fmt.Println()
			}()
		}
	}

	gzf, err := gzip.NewReader(asset)
	if err != nil {
		fmt.Printf("Could not ungzip OpenShift release asset: %s\n", err)
		os.Exit(1)
	}
	defer gzf.Close()
	tr := tar.NewReader(gzf)
	for {
		hdr, err := tr.Next()
		if err == io.EOF {
			// end of tar archive
			break
		}
		if err != nil {
			fmt.Printf("Could not extract OpenShift release asset: %s\n", err)
			os.Exit(1)
		}
		if hdr.Typeflag != tar.TypeReg || filepath.Base(hdr.Name) != "kube-apiserver" {
			continue
		}
		contents, err := ioutil.ReadAll(tr)
		if err != nil {
			fmt.Printf("Could not extract OpenShift release asset: %s\n", err)
			os.Exit(1)
		}
		err = ioutil.WriteFile("out/openshift", contents, os.ModePerm)
		if err != nil {
			fmt.Printf("Could not write OpenShift binary: %s\n", err)
			os.Exit(1)
		}
	}
}