Return error when a file download fails.

VirusTotal · Nov 15, 2018 · 704ab0a · 704ab0a
1 parent 5af35c3
commit 704ab0a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/download.go b/cmd/download.go
@@ -15,6 +15,7 @@ package cmd
 
 import (
 	"fmt"
+	"net/http"
 	"os"
 	"path"
 	"regexp"
@@ -57,6 +58,9 @@ Loop:
 		case <-t.C:
 			callback(resp)
 		case <-resp.Done:
+			if resp.HTTPResponse.StatusCode != http.StatusOK {
+				return fmt.Errorf("error downloading file: %d", resp.HTTPResponse.StatusCode)
+			}
 			callback(resp)
 			break Loop
 		}