/*
godlighty -- highly-customizable HTTP, HTTP/2, HTTPS server
-Copyright (C) 2021 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2021-2022 Sergey Matveev <stargrave@stargrave.org>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
import (
"bytes"
"compress/gzip"
+ "encoding/base64"
"errors"
"fmt"
- "io/ioutil"
+ "io"
"log"
"net"
"net/http"
"time"
"github.com/klauspost/compress/zstd"
+ "go.stargrave.org/godlighty/meta4"
"golang.org/x/net/webdav"
)
const (
- Index = "index.html"
- Readme = "README"
- Meta4Ext = ".meta4"
+ Index = "index.html"
+ Readme = "README"
)
var (
gzPool = sync.Pool{
- New: func() interface{} { return gzip.NewWriter(ioutil.Discard) },
+ New: func() interface{} { return gzip.NewWriter(io.Discard) },
}
zstdPool = sync.Pool{
New: func() interface{} {
w, err := zstd.NewWriter(
- ioutil.Discard,
+ io.Discard,
zstd.WithEncoderLevel(zstd.SpeedDefault),
)
if err != nil {
return
}
- if !(r.Method == "" || r.Method == http.MethodGet) {
+ if !(r.Method == "" || r.Method == http.MethodGet || r.Method == http.MethodHead) {
fmt.Printf("%s %s \"%s %+q %s\" %d %s\"%s\"\n",
r.RemoteAddr, host, r.Method, PathWithQuery(r.URL), r.Proto,
http.StatusMethodNotAllowed,
}
var readme []byte
for _, f := range append(cfg.Readmes, Readme) {
- readme, _ = ioutil.ReadFile(path.Join(pth, f))
+ readme, _ = os.ReadFile(path.Join(pth, f))
if readme != nil {
break
}
}
contentType = "text/html; charset=utf-8"
} else {
- for _, index := range append(cfg.Indexes, Index) {
+ for _, index := range append(cfg.Indices, Index) {
p := path.Join(pth, index)
if _, err := os.Stat(p); err == nil {
pth = p
}
defer fd.Close()
- if _, err = os.Stat(pth + Meta4Ext); err == nil {
- w.Header().Set("Link", "<"+path.Base(pth)+Meta4Ext+`>; rel=describedby; type="application/metalink4+xml"`)
+ if meta4fi, err := os.Stat(pth + meta4.Ext); err == nil {
+ if meta4fi.Size() > meta4.MaxSize {
+ goto SkipMeta4
+ }
+ meta4Raw, err := os.ReadFile(pth + meta4.Ext)
+ if err != nil {
+ goto SkipMeta4
+ }
+ base := path.Base(pth)
+ forHTTP, err := meta4.Parse(base, meta4Raw)
+ if err != nil {
+ goto SkipMeta4
+ }
+ w.Header().Add("Link", "<"+base+meta4.Ext+
+ `>; rel=describedby; type="application/metalink4+xml"`,
+ )
+ for _, u := range forHTTP.URLs {
+ w.Header().Add("Link", "<"+u+">; rel=duplicate")
+ }
+ if forHTTP.SHA256 != nil {
+ w.Header().Add("Digest", "SHA-256="+
+ base64.StdEncoding.EncodeToString(forHTTP.SHA256))
+ }
+ if forHTTP.SHA512 != nil {
+ w.Header().Add("Digest", "SHA-512="+
+ base64.StdEncoding.EncodeToString(forHTTP.SHA512))
+ }
}
+SkipMeta4:
if contentType == "" {
contentType = mediaType(path.Base(pth), cfg.MIMEs)