From 7bfcd667a18377c796f5231b88058830df494c04 Mon Sep 17 00:00:00 2001 From: WeebDataHoarder <57538841+WeebDataHoarder@users.noreply.github.com> Date: Tue, 18 Jan 2022 19:32:19 +0100 Subject: [PATCH] Fix server header map --- httputils/server.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/httputils/server.go b/httputils/server.go index cb27cfe..f461b76 100644 --- a/httputils/server.go +++ b/httputils/server.go @@ -112,8 +112,10 @@ func (server *Server) GetExtraHeaders() map[string]string { server.extraHeadersMutex.RLock() r := make(map[string]string) - for key, value := range server.extraHeaders { - r[key] = value + if server.extraHeaders != nil { + for key, value := range server.extraHeaders { + r[key] = value + } } server.extraHeadersMutex.RUnlock() return r @@ -121,6 +123,9 @@ func (server *Server) GetExtraHeaders() map[string]string { func (server *Server) AddExtraHeader(key, value string) { server.extraHeadersMutex.Lock() + if server.extraHeaders == nil { + server.extraHeaders = make(map[string]string) + } server.extraHeaders[key] = value server.extraHeadersMutex.Unlock() }