From 25114986086e5877a0b4108d8cf5e4e95f377241 Mon Sep 17 00:00:00 2001 From: Ivan Vyshnevskyi Date: Sat, 16 Dec 2017 17:41:52 +0200 Subject: [PATCH] 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 --- livereload/livereload.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/livereload/livereload.go b/livereload/livereload.go index 7bfbbaf6b..2f3cee8f0 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