Respond with 404 if requesting server's config file

config/config.go

@@ -44,6 +44,9 @@ type Config struct {
 
 	// MaxCPU is the maximum number of cores for the whole process to use
 	MaxCPU int
+
+	// The path to the configuration file from which this was loaded
+	ConfigFile string
 }
 
 // Address returns the host:port of c as a string.
@@ -75,7 +78,16 @@ func Load(filename string) ([]Config, error) {
 		return nil, err
 	}
 
-	return p.parse()
+	cfgs, err := p.parse()
+	if err != nil {
+		return []Config{}, err
+	}
+
+	for i := 0; i < len(cfgs); i++ {
+		cfgs[i].ConfigFile = filename
+	}
+
+	return cfgs, nil
 }
 
 // IsNotFound returns whether or not the error is

server/fileserver.go

@@ -10,20 +10,22 @@ import (
 	"github.com/mholt/caddy/middleware/browse"
 )
 
-// This FileServer is adapted from the one in net/http
-// by the Go authors. Some modifications have been made.
+// This FileServer is adapted from the one in net/http by
+// the Go authors. Significant modifications have been made.
+//
 //
 // License:
 //
 // Copyright 2009 The Go Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style
 // license that can be found in the LICENSE file.
-func FileServer(root http.FileSystem) middleware.Handler {
-	return &fileHandler{root}
+func FileServer(root http.FileSystem, hide []string) middleware.Handler {
+	return &fileHandler{root: root, hide: hide}
 }
 
 type fileHandler struct {
 	root http.FileSystem
+	hide []string // list of files to treat as "Not Found"
 }
 
 func (f *fileHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
@@ -32,12 +34,12 @@ func (f *fileHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, er
 		upath = "/" + upath
 		r.URL.Path = upath
 	}
-	return serveFile(w, r, f.root, path.Clean(upath))
+	return f.serveFile(w, r, path.Clean(upath))
 }
 
 // name is '/'-separated, not filepath.Separator.
-func serveFile(w http.ResponseWriter, r *http.Request, fs http.FileSystem, name string) (int, error) {
-	f, err := fs.Open(name)
+func (fh *fileHandler) serveFile(w http.ResponseWriter, r *http.Request, name string) (int, error) {
+	f, err := fh.root.Open(name)
 	if err != nil {
 		if os.IsPermission(err) {
 			return http.StatusForbidden, err
@@ -58,7 +60,7 @@ func serveFile(w http.ResponseWriter, r *http.Request, fs http.FileSystem, name
 	if d.IsDir() {
 		for _, indexPage := range browse.IndexPages {
 			index := strings.TrimSuffix(name, "/") + "/" + indexPage
-			ff, err := fs.Open(index)
+			ff, err := fh.root.Open(index)
 			if err == nil {
 				defer ff.Close()
 				dd, err := ff.Stat()
@@ -78,6 +80,14 @@ func serveFile(w http.ResponseWriter, r *http.Request, fs http.FileSystem, name
 		return http.StatusNotFound, nil
 	}
 
+	// If the file is supposed to be hidden, return a 404
+	// (TODO: If the slice gets large, a set may be faster)
+	for _, hiddenPath := range fh.hide {
+		if d.Name() == hiddenPath {
+			return http.StatusNotFound, nil
+		}
+	}
+
 	// Note: Errors generated by ServeContent are written immediately
 	// to the response. This usually only happens if seeking fails (rare).
 	http.ServeContent(w, r, d.Name(), d.ModTime(), f)

server/server.go

@@ -134,7 +134,7 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 // on its config. This method should be called last before
 // ListenAndServe begins.
 func (s *Server) buildStack() error {
-	s.fileServer = FileServer(http.Dir(s.config.Root))
+	s.fileServer = FileServer(http.Dir(s.config.Root), []string{s.config.ConfigFile})
 
 	// TODO: We only compile middleware for the "/" scope.
 	// Partial support for multiple location contexts already