fix X-Powered-By header

Signed-off-by: Christian Winther <jippignu@gmail.com>
jippi · Mar 14, 2018 · fc30e55 · fc30e55
1 parent 7076d25
commit fc30e55
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/internal/http_handler.go b/internal/http_handler.go
@@ -82,7 +82,6 @@ func StarServer() {
 	if err := srv.ListenAndServe(); err != nil {
 		log.Fatal(err)
 	}
-
 }
 
 // handles: /{api_version}/meta-data/iam/info
@@ -178,6 +177,7 @@ func iamSecurityCredentialsName(w http.ResponseWriter, r *http.Request) {
 
 	// send the response
 	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.Header().Set("X-Powered-By", "go-metadataproxy")
 	w.WriteHeader(http.StatusOK)
 	w.Write([]byte(*roleInfo.RoleName))
 
@@ -302,7 +302,7 @@ func passthroughHandler(w http.ResponseWriter, r *http.Request) {
 	}
 	defer resp.Body.Close()
 
-	w.Header().Add("X-Powered-By", "go-metadataproxy")
+	w.Header().Set("X-Powered-By", "go-metadataproxy")
 	w.WriteHeader(resp.StatusCode)
 	io.Copy(w, resp.Body)
 

diff --git a/internal/http_helper.go b/internal/http_helper.go
@@ -45,6 +45,7 @@ func isCompatibleAPIVersion(r *http.Request) bool {
 
 func httpError(err error, w http.ResponseWriter, r *http.Request) {
 	log.Error(err)
+	w.Header().Set("X-Powered-By", "go-metadataproxy")
 	http.NotFound(w, r)
 }