test/serve: fix a descriptor leak, add preliminary directory support

Fix a descriptor leak in the helper, and add some minimal support for
clients that might want to scrape HTML of a directory to walk the
filesystem.

Signed-off-by: Nalin Dahyabhai <nalin@redhat.com>
This commit is contained in:
Nalin Dahyabhai 2025-03-26 15:23:00 -04:00
parent 36c3f13d85
commit cbe5823958
1 changed files with 29 additions and 4 deletions

View File

@ -3,32 +3,57 @@ package main
import (
"context"
"errors"
"fmt"
"html"
"io"
"io/fs"
"log"
"net"
"net/http"
"os"
"path"
"path/filepath"
"strconv"
"strings"
)
func sendThatFile(basepath string) func(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
filename := filepath.Join(basepath, filepath.Clean(string([]rune{filepath.Separator})+r.URL.Path))
filename := filepath.Join(basepath, filepath.Clean(string([]rune{filepath.Separator})+filepath.FromSlash(r.URL.Path)))
f, err := os.Open(filename)
if err != nil {
if errors.Is(err, os.ErrNotExist) {
http.NotFound(w, r)
return
}
http.Error(w, "whoops", http.StatusInternalServerError)
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer f.Close()
finfo, err := f.Stat()
if err != nil {
http.Error(w, "whoops", http.StatusInternalServerError)
http.Error(w, fmt.Sprintf("checking file info: %v", err), http.StatusInternalServerError)
return
}
http.ServeContent(w, r, filename, finfo.ModTime(), f)
content := io.ReadSeeker(f)
if finfo.IsDir() {
names, err := f.ReadDir(-1)
if err != nil {
http.Error(w, fmt.Sprintf("reading directory: %v", err), http.StatusInternalServerError)
}
var builder strings.Builder
builder.WriteString("<body><html>")
for _, name := range names {
suffix := ""
if name.IsDir() {
suffix = "/"
}
builder.WriteString(fmt.Sprintf("<a href=%q>%s</a><br/>\n", path.Join(r.URL.Path, name.Name())+suffix, html.EscapeString(fs.FormatDirEntry(name))))
}
builder.WriteString("</html></body>")
content = strings.NewReader(builder.String())
}
http.ServeContent(w, r, filename, finfo.ModTime(), content)
}
}