From 57a107aa69c92f86570e3e83448346cc3a7fba5d Mon Sep 17 00:00:00 2001
From: Antoni Sawicki <tenox@google.com>
Date: Wed, 29 May 2019 23:49:39 -0700
Subject: [PATCH] log decode error instead of bailing out

---
 wrp.go | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/wrp.go b/wrp.go
index d9ece02..e7a4ed6 100644
--- a/wrp.go
+++ b/wrp.go
@@ -8,18 +8,18 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"flag"
 	"fmt"
+	_ "image"
+	"image/gif"
+	"image/png"
 	"log"
 	"net/http"
+	"net/url"
 	"strconv"
 	"time"
-	"bytes"
-	_ "image"
-	"image/png"
-	"image/gif"
-	"net/url"
 
 	"github.com/chromedp/cdproto/emulation"
 
@@ -79,15 +79,16 @@ func capture(gourl string, out http.ResponseWriter) {
 
 	log.Printf("Landed on: %s, Got %d nodes\n", loc, len(nodes))
 
-	img, err:= png.Decode(bytes.NewReader(scrcap) )
+	img, err := png.Decode(bytes.NewReader(scrcap))
 	if err != nil {
-		log.Fatal(err)
+		log.Printf("Failed to decode screenshot: %s\n", err)
+		fmt.Fprintf(out, "<BR>Unable to decode page screenshot:<BR>%s<BR>\n", err)
+		return
 	}
 	gifbuf.Reset()
 	gif.Encode(&gifbuf, img, nil)
 
 	base, _ := url.Parse(loc)
-	fmt.Fprintf(out, "<!-- Location: %s -->\n", base)
 	fmt.Fprintf(out, "<IMG SRC=\"/wrp.gif\" ALT=\"wrp\" USEMAP=\"#map\">\n<MAP NAME=\"map\">\n")
 
 	for _, n := range nodes {