From: Ivan Vyshnevskyi <ivan@vyshnevskyi.com>
Date: Sat, 16 Dec 2017 15:41:52 +0000 (+0200)
Subject: livereload: Fix host comparison when ports aren't present
X-Git-Tag: v0.32~18
X-Git-Url: http://git.maquefel.me/?a=commitdiff_plain;h=25114986086e5877a0b4108d8cf5e4e95f377241;p=brevno-suite%2Fhugo

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 7bfbbaf6..2f3cee8f 100644
--- 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