From df156ec7d88974ebe7211f06256db56ed4bdd0e2 Mon Sep 17 00:00:00 2001
From: mischief <mischief@offblast.org>
Date: Tue, 25 Mar 2014 03:53:11 -0700
Subject: [PATCH] use the Host field correctly

---
 main.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index ecfc678..aada962 100644
--- a/main.go
+++ b/main.go
@@ -169,7 +169,7 @@ func (werc *Werc) genmenu(site, dir string) []MenuEntry {
 }
 
 func (werc *Werc) WercCommon(w http.ResponseWriter, r *http.Request, page *WercPage) {
-	site := r.URL.Host
+	site, _, _ := net.SplitHostPort(r.Host)
 	if site == "" {
 		site = werc.conf.MasterSite
 	}
@@ -285,7 +285,7 @@ func (werc *Werc) Werc404(w http.ResponseWriter, r *http.Request) {
 
 func (werc *Werc) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	log.Printf("%s", r.URL)
-	site := r.Host
+	site, _, _ := net.SplitHostPort(r.Host)
 	if site == "" {
 		site = werc.conf.MasterSite
 	}