From 201bca587e09a3c113d1301ab6ab9741408e0c47 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mika=C3=ABl=20Cluseau?= Date: Fri, 12 Apr 2019 10:15:57 +0100 Subject: [PATCH] check status code --- cmd/dkl-local-server/upstream.go | 7 +++++++ go.sum | 1 + 2 files changed, 8 insertions(+) diff --git a/cmd/dkl-local-server/upstream.go b/cmd/dkl-local-server/upstream.go index 9391fa6..51e8b26 100644 --- a/cmd/dkl-local-server/upstream.go +++ b/cmd/dkl-local-server/upstream.go @@ -2,6 +2,7 @@ package main import ( "flag" + "fmt" "io" "log" "net/http" @@ -39,6 +40,12 @@ func (ctx *renderContext) distFetch(path ...string) (outPath string, err error) return } + if resp.StatusCode != 200 { + err = fmt.Errorf("wrong status: %s", resp.Status) + resp.Body.Close() + return + } + tempOutPath := filepath.Join(filepath.Dir(outPath), "._part_"+filepath.Base(outPath)) done := make(chan error, 1) diff --git a/go.sum b/go.sum index 40ca4de..d6cc79a 100644 --- a/go.sum +++ b/go.sum @@ -10,6 +10,7 @@ github.com/alcortesm/tgz v0.0.0-20161220082320-9c5fe88206d7/go.mod h1:6zEj6s6u/g github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 h1:kFOfPq6dUM1hTo4JG6LR5AXSUEsOjtdm0kw0FtQtMJA= github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c= github.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6/go.mod h1:grANhF5doyWs3UAsr3K4I6qtAmlQcZDesFNEHPZAzj8= +github.com/cavaliercoder/go-cpio v0.0.0-20180626203310-925f9528c45e h1:hHg27A0RSSp2Om9lubZpiMgVbvn39bsUmW9U5h0twqc= github.com/cavaliercoder/go-cpio v0.0.0-20180626203310-925f9528c45e/go.mod h1:oDpT4efm8tSYHXV5tHSdRvBet/b/QzxZ+XyyPehvm3A= github.com/cloudflare/cfssl v0.0.0-20181213083726-b94e044bb51e h1:Qux+lbuMaRzkQyTdzgtz8MgzPtzmaPQy6DXmxpdxT3U= github.com/cloudflare/cfssl v0.0.0-20181213083726-b94e044bb51e/go.mod h1:yMWuSON2oQp+43nFtAV/uvKQIFpSPerB57DCt9t8sSA=