livereload: Fix host comparison when ports aren't present

Compare the original hosts from 'Origin' and 'Host' headers before
attempting to do a port-less comparison.  This helps in the case when
hugo server was started with a '--port=80' so both headers do not
contain a port.

Fixes #4141

diff --git a/livereload/livereload.go b/livereload/livereload.go
index 7bfbbaf6b870738b6d419b4e7b45a88a71c1c0c8..2f3cee8f047c8e2d5717f4774998b1630087d4f0 100644 (file)
--- a/livereload/livereload.go
+++ b/livereload/livereload.go
@@ -62,6 +62,10 @@ var upgrader = &websocket.Upgrader{
                        return false
                }
 
+               if u.Host == r.Host {
+                       return true
+               }
+
                h1, _, err := net.SplitHostPort(u.Host)
                if err != nil {
                        return false