Compare commits
52 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
72d437c82e | ||
|
|
3dca29eb41 | ||
|
|
a40b5dcd0b | ||
|
|
fffe86680e | ||
|
|
d5af32e121 | ||
|
|
5141eaafaa | ||
|
|
e5c0afa013 | ||
|
|
4c7c200f92 | ||
|
|
0a709da439 | ||
|
|
1fdef9b608 | ||
|
|
2144e2c2f2 | ||
|
|
93a606b591 | ||
|
|
b00794f236 | ||
|
|
3da7fe7cee | ||
|
|
dea7600f29 | ||
|
|
7d958a4798 | ||
|
|
a5493b708a | ||
|
|
6e5c2473e7 | ||
|
|
c639233ea1 | ||
|
|
5677440876 | ||
|
|
be3d09d7f4 | ||
|
|
504da9afd8 | ||
|
|
d1cb8967b6 | ||
|
|
107b3a1785 | ||
|
|
e7a06a12bf | ||
|
|
649b20659b | ||
|
|
c9e2af98f3 | ||
|
|
d6d02e398e | ||
|
|
de0b93a4f6 | ||
|
|
ce649ecc66 | ||
|
|
688e7e2823 | ||
|
|
b38311da00 | ||
|
|
8e2ac24830 | ||
|
|
bfa3356d3a | ||
|
|
9f3564936e | ||
|
|
d8fb072826 | ||
|
|
69f0913b3d | ||
|
|
f7012b38da | ||
|
|
768ec6c17b | ||
|
|
ae7d58549d | ||
|
|
ad5d78f08f | ||
|
|
4b92c71839 | ||
|
|
19f1d6693e | ||
|
|
0e87d64ffc | ||
|
|
845f8e9bd1 | ||
|
|
cf9ab18c1f | ||
|
|
ada42ff427 | ||
|
|
fcc71b76d9 | ||
|
|
6a1ccdc644 | ||
|
|
f156be19b4 | ||
|
|
82bdffc1eb | ||
|
|
a396ec77e4 |
@@ -6,27 +6,32 @@ import (
|
|||||||
"crypto/x509/pkix"
|
"crypto/x509/pkix"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
// A Store represents a certificate store.
|
// A Store represents a TLS certificate store.
|
||||||
// It generates certificates as needed and automatically rotates expired certificates.
|
|
||||||
// The zero value for Store is an empty store ready to use.
|
// The zero value for Store is an empty store ready to use.
|
||||||
//
|
//
|
||||||
// Certificate scopes must be registered with Register before calling Get or Load.
|
// Store can be used to store server certificates.
|
||||||
// This prevents the Store from creating or loading unnecessary certificates.
|
// Servers should provide a hostname or wildcard pattern as a certificate scope.
|
||||||
|
// Servers will most likely use the methods Register, Load and Get.
|
||||||
|
//
|
||||||
|
// Store can also be used to store client certificates.
|
||||||
|
// Clients should provide a hostname as a certificate scope.
|
||||||
|
// Clients will most likely use the methods Add, Load, and Lookup.
|
||||||
//
|
//
|
||||||
// Store is safe for concurrent use by multiple goroutines.
|
// Store is safe for concurrent use by multiple goroutines.
|
||||||
type Store struct {
|
type Store struct {
|
||||||
// CreateCertificate, if not nil, is called to create a new certificate
|
// CreateCertificate, if not nil, is called by Get to create a new
|
||||||
// to replace a missing or expired certificate. If CreateCertificate
|
// certificate to replace a missing or expired certificate.
|
||||||
// is nil, a certificate with a duration of 1 year will be created.
|
|
||||||
// The provided scope is suitable for use in a certificate's DNSNames.
|
// The provided scope is suitable for use in a certificate's DNSNames.
|
||||||
CreateCertificate func(scope string) (tls.Certificate, error)
|
CreateCertificate func(scope string) (tls.Certificate, error)
|
||||||
|
|
||||||
|
scopes map[string]struct{}
|
||||||
certs map[string]tls.Certificate
|
certs map[string]tls.Certificate
|
||||||
path string
|
path string
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
@@ -35,24 +40,21 @@ type Store struct {
|
|||||||
// Register registers the provided scope with the certificate store.
|
// Register registers the provided scope with the certificate store.
|
||||||
// The scope can either be a hostname or a wildcard pattern (e.g. "*.example.com").
|
// The scope can either be a hostname or a wildcard pattern (e.g. "*.example.com").
|
||||||
// To accept all hostnames, use the special pattern "*".
|
// To accept all hostnames, use the special pattern "*".
|
||||||
|
//
|
||||||
|
// Calls to Get will only succeed for registered scopes.
|
||||||
|
// Other methods are not affected.
|
||||||
func (s *Store) Register(scope string) {
|
func (s *Store) Register(scope string) {
|
||||||
s.mu.Lock()
|
s.mu.Lock()
|
||||||
defer s.mu.Unlock()
|
defer s.mu.Unlock()
|
||||||
if s.certs == nil {
|
if s.scopes == nil {
|
||||||
s.certs = make(map[string]tls.Certificate)
|
s.scopes = make(map[string]struct{})
|
||||||
}
|
}
|
||||||
s.certs[scope] = tls.Certificate{}
|
s.scopes[scope] = struct{}{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Add adds a certificate with the given scope to the certificate store.
|
// Add registers the certificate for the given scope.
|
||||||
// If a certificate for the given scope already exists, Add will overwrite it.
|
// If a certificate already exists for scope, Add will overwrite it.
|
||||||
func (s *Store) Add(scope string, cert tls.Certificate) error {
|
func (s *Store) Add(scope string, cert tls.Certificate) error {
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
if s.certs == nil {
|
|
||||||
s.certs = make(map[string]tls.Certificate)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse certificate if not already parsed
|
// Parse certificate if not already parsed
|
||||||
if cert.Leaf == nil {
|
if cert.Leaf == nil {
|
||||||
parsed, err := x509.ParseCertificate(cert.Certificate[0])
|
parsed, err := x509.ParseCertificate(cert.Certificate[0])
|
||||||
@@ -62,6 +64,22 @@ func (s *Store) Add(scope string, cert tls.Certificate) error {
|
|||||||
cert.Leaf = parsed
|
cert.Leaf = parsed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if err := s.write(scope, cert); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
if s.certs == nil {
|
||||||
|
s.certs = make(map[string]tls.Certificate)
|
||||||
|
}
|
||||||
|
s.certs[scope] = cert
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) write(scope string, cert tls.Certificate) error {
|
||||||
|
s.mu.RLock()
|
||||||
|
defer s.mu.RUnlock()
|
||||||
if s.path != "" {
|
if s.path != "" {
|
||||||
certPath := filepath.Join(s.path, scope+".crt")
|
certPath := filepath.Join(s.path, scope+".crt")
|
||||||
keyPath := filepath.Join(s.path, scope+".key")
|
keyPath := filepath.Join(s.path, scope+".key")
|
||||||
@@ -69,35 +87,37 @@ func (s *Store) Add(scope string, cert tls.Certificate) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
s.certs[scope] = cert
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get retrieves a certificate for the given hostname.
|
// Get retrieves a certificate for the given hostname.
|
||||||
// If no matching scope has been registered, Get returns an error.
|
// If no matching scope has been registered, Get returns an error.
|
||||||
// Get generates new certificates as needed and rotates expired certificates.
|
// Get generates new certificates as needed and rotates expired certificates.
|
||||||
|
// It calls CreateCertificate to create a new certificate if it is not nil,
|
||||||
|
// otherwise it creates certificates with a duration of 100 years.
|
||||||
//
|
//
|
||||||
// Get is suitable for use in a gemini.Server's GetCertificate field.
|
// Get is suitable for use in a gemini.Server's GetCertificate field.
|
||||||
func (s *Store) Get(hostname string) (*tls.Certificate, error) {
|
func (s *Store) Get(hostname string) (*tls.Certificate, error) {
|
||||||
s.mu.RLock()
|
s.mu.RLock()
|
||||||
defer s.mu.RUnlock()
|
_, ok := s.scopes[hostname]
|
||||||
cert, ok := s.certs[hostname]
|
|
||||||
if !ok {
|
if !ok {
|
||||||
// Try wildcard
|
// Try wildcard
|
||||||
wildcard := strings.SplitN(hostname, ".", 2)
|
wildcard := strings.SplitN(hostname, ".", 2)
|
||||||
if len(wildcard) == 2 {
|
if len(wildcard) == 2 {
|
||||||
hostname = "*." + wildcard[1]
|
hostname = "*." + wildcard[1]
|
||||||
cert, ok = s.certs[hostname]
|
_, ok = s.scopes[hostname]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !ok {
|
if !ok {
|
||||||
// Try "*"
|
// Try "*"
|
||||||
cert, ok = s.certs["*"]
|
_, ok = s.scopes["*"]
|
||||||
}
|
}
|
||||||
if !ok {
|
if !ok {
|
||||||
|
s.mu.RUnlock()
|
||||||
return nil, errors.New("unrecognized scope")
|
return nil, errors.New("unrecognized scope")
|
||||||
}
|
}
|
||||||
|
cert := s.certs[hostname]
|
||||||
|
s.mu.RUnlock()
|
||||||
|
|
||||||
// If the certificate is empty or expired, generate a new one.
|
// If the certificate is empty or expired, generate a new one.
|
||||||
if cert.Leaf == nil || cert.Leaf.NotAfter.Before(time.Now()) {
|
if cert.Leaf == nil || cert.Leaf.NotAfter.Before(time.Now()) {
|
||||||
@@ -114,6 +134,14 @@ func (s *Store) Get(hostname string) (*tls.Certificate, error) {
|
|||||||
return &cert, nil
|
return &cert, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Lookup returns the certificate for the provided scope.
|
||||||
|
func (s *Store) Lookup(scope string) (tls.Certificate, bool) {
|
||||||
|
s.mu.RLock()
|
||||||
|
defer s.mu.RUnlock()
|
||||||
|
cert, ok := s.certs[scope]
|
||||||
|
return cert, ok
|
||||||
|
}
|
||||||
|
|
||||||
func (s *Store) createCertificate(scope string) (tls.Certificate, error) {
|
func (s *Store) createCertificate(scope string) (tls.Certificate, error) {
|
||||||
if s.CreateCertificate != nil {
|
if s.CreateCertificate != nil {
|
||||||
return s.CreateCertificate(scope)
|
return s.CreateCertificate(scope)
|
||||||
@@ -123,7 +151,7 @@ func (s *Store) createCertificate(scope string) (tls.Certificate, error) {
|
|||||||
Subject: pkix.Name{
|
Subject: pkix.Name{
|
||||||
CommonName: scope,
|
CommonName: scope,
|
||||||
},
|
},
|
||||||
Duration: 365 * 24 * time.Hour,
|
Duration: 100 * 365 * 24 * time.Hour,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -132,26 +160,31 @@ func (s *Store) createCertificate(scope string) (tls.Certificate, error) {
|
|||||||
// The path should lead to a directory containing certificates
|
// The path should lead to a directory containing certificates
|
||||||
// and private keys named "scope.crt" and "scope.key" respectively,
|
// and private keys named "scope.crt" and "scope.key" respectively,
|
||||||
// where "scope" is the scope of the certificate.
|
// where "scope" is the scope of the certificate.
|
||||||
// Certificates with scopes that have not been registered will be ignored.
|
|
||||||
func (s *Store) Load(path string) error {
|
func (s *Store) Load(path string) error {
|
||||||
|
if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
path = filepath.Clean(path)
|
||||||
matches, err := filepath.Glob(filepath.Join(path, "*.crt"))
|
matches, err := filepath.Glob(filepath.Join(path, "*.crt"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
for _, crtPath := range matches {
|
for _, crtPath := range matches {
|
||||||
scope := strings.TrimSuffix(filepath.Base(crtPath), ".crt")
|
|
||||||
if _, ok := s.certs[scope]; !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
keyPath := strings.TrimSuffix(crtPath, ".crt") + ".key"
|
keyPath := strings.TrimSuffix(crtPath, ".crt") + ".key"
|
||||||
cert, err := tls.LoadX509KeyPair(crtPath, keyPath)
|
cert, err := tls.LoadX509KeyPair(crtPath, keyPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
scope := strings.TrimPrefix(crtPath, path)
|
||||||
|
scope = strings.TrimPrefix(scope, "/")
|
||||||
|
scope = strings.TrimSuffix(scope, ".crt")
|
||||||
s.Add(scope, cert)
|
s.Add(scope, cert)
|
||||||
}
|
}
|
||||||
s.SetPath(path)
|
s.mu.Lock()
|
||||||
|
defer s.mu.Unlock()
|
||||||
|
s.path = path
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -170,5 +203,5 @@ func (s *Store) Entries() map[string]tls.Certificate {
|
|||||||
func (s *Store) SetPath(path string) {
|
func (s *Store) SetPath(path string) {
|
||||||
s.mu.Lock()
|
s.mu.Lock()
|
||||||
defer s.mu.Unlock()
|
defer s.mu.Unlock()
|
||||||
s.path = path
|
s.path = filepath.Clean(path)
|
||||||
}
|
}
|
||||||
|
|||||||
16
client.go
16
client.go
@@ -6,7 +6,6 @@ import (
|
|||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"net"
|
"net"
|
||||||
"net/url"
|
"net/url"
|
||||||
"time"
|
|
||||||
"unicode/utf8"
|
"unicode/utf8"
|
||||||
|
|
||||||
"golang.org/x/net/idna"
|
"golang.org/x/net/idna"
|
||||||
@@ -132,6 +131,9 @@ func (c *Client) Do(ctx context.Context, req *Request) (*Response, error) {
|
|||||||
conn.Close()
|
conn.Close()
|
||||||
return nil, ctx.Err()
|
return nil, ctx.Err()
|
||||||
case r := <-res:
|
case r := <-res:
|
||||||
|
if r.err != nil {
|
||||||
|
conn.Close()
|
||||||
|
}
|
||||||
return r.resp, r.err
|
return r.resp, r.err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -153,7 +155,7 @@ func (c *Client) do(ctx context.Context, conn net.Conn, req *Request) (*Response
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Write the request
|
// Write the request
|
||||||
if err := req.Write(w); err != nil {
|
if _, err := req.WriteTo(w); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -175,17 +177,9 @@ func (c *Client) dialContext(ctx context.Context, network, addr string) (net.Con
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Client) verifyConnection(cs tls.ConnectionState, hostname string) error {
|
func (c *Client) verifyConnection(cs tls.ConnectionState, hostname string) error {
|
||||||
cert := cs.PeerCertificates[0]
|
|
||||||
// Verify hostname
|
|
||||||
if err := verifyHostname(cert, hostname); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
// Check expiration date
|
|
||||||
if !time.Now().Before(cert.NotAfter) {
|
|
||||||
return ErrCertificateExpired
|
|
||||||
}
|
|
||||||
// See if the client trusts the certificate
|
// See if the client trusts the certificate
|
||||||
if c.TrustCertificate != nil {
|
if c.TrustCertificate != nil {
|
||||||
|
cert := cs.PeerCertificates[0]
|
||||||
return c.TrustCertificate(hostname, cert)
|
return c.TrustCertificate(hostname, cert)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
6
doc.go
6
doc.go
@@ -29,10 +29,10 @@ Servers should be configured with certificates:
|
|||||||
}
|
}
|
||||||
server.GetCertificate = certificates.Get
|
server.GetCertificate = certificates.Get
|
||||||
|
|
||||||
ServeMux is a Gemini request multiplexer.
|
Mux is a Gemini request multiplexer.
|
||||||
ServeMux can handle requests for multiple hosts and schemes.
|
Mux can handle requests for multiple hosts and schemes.
|
||||||
|
|
||||||
mux := &gemini.ServeMux{}
|
mux := &gemini.Mux{}
|
||||||
mux.HandleFunc("example.com", func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
mux.HandleFunc("example.com", func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
||||||
fmt.Fprint(w, "Welcome to example.com")
|
fmt.Fprint(w, "Welcome to example.com")
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -30,7 +30,7 @@ func main() {
|
|||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
mux := &gemini.ServeMux{}
|
mux := &gemini.Mux{}
|
||||||
mux.HandleFunc("/", profile)
|
mux.HandleFunc("/", profile)
|
||||||
mux.HandleFunc("/username", changeUsername)
|
mux.HandleFunc("/username", changeUsername)
|
||||||
|
|
||||||
|
|||||||
@@ -6,7 +6,6 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"bytes"
|
|
||||||
"context"
|
"context"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"errors"
|
"errors"
|
||||||
@@ -16,7 +15,6 @@ import (
|
|||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"time"
|
|
||||||
|
|
||||||
"git.sr.ht/~adnano/go-gemini"
|
"git.sr.ht/~adnano/go-gemini"
|
||||||
"git.sr.ht/~adnano/go-gemini/tofu"
|
"git.sr.ht/~adnano/go-gemini/tofu"
|
||||||
@@ -61,16 +59,15 @@ Otherwise, this should be safe to trust.
|
|||||||
=> `
|
=> `
|
||||||
|
|
||||||
func trustCertificate(hostname string, cert *x509.Certificate) error {
|
func trustCertificate(hostname string, cert *x509.Certificate) error {
|
||||||
host := tofu.NewHost(hostname, cert.Raw, cert.NotAfter)
|
host := tofu.NewHost(hostname, cert.Raw)
|
||||||
|
|
||||||
knownHost, ok := hosts.Lookup(hostname)
|
knownHost, ok := hosts.Lookup(hostname)
|
||||||
if ok && time.Now().Before(knownHost.Expires) {
|
if ok {
|
||||||
// Check fingerprint
|
// Check fingerprint
|
||||||
if bytes.Equal(knownHost.Fingerprint, host.Fingerprint) {
|
if knownHost.Fingerprint != host.Fingerprint {
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return errors.New("error: fingerprint does not match!")
|
return errors.New("error: fingerprint does not match!")
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
fmt.Printf(trustPrompt, hostname, host.Fingerprint)
|
fmt.Printf(trustPrompt, hostname, host.Fingerprint)
|
||||||
scanner.Scan()
|
scanner.Scan()
|
||||||
@@ -87,7 +84,7 @@ func trustCertificate(hostname string, cert *x509.Certificate) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func getInput(prompt string, sensitive bool) (input string, ok bool) {
|
func getInput(prompt string) (input string, ok bool) {
|
||||||
fmt.Printf("%s ", prompt)
|
fmt.Printf("%s ", prompt)
|
||||||
scanner.Scan()
|
scanner.Scan()
|
||||||
return scanner.Text(), true
|
return scanner.Text(), true
|
||||||
@@ -105,7 +102,7 @@ func do(req *gemini.Request, via []*gemini.Request) (*gemini.Response, error) {
|
|||||||
|
|
||||||
switch resp.Status.Class() {
|
switch resp.Status.Class() {
|
||||||
case gemini.StatusInput:
|
case gemini.StatusInput:
|
||||||
input, ok := getInput(resp.Meta, resp.Status == gemini.StatusSensitiveInput)
|
input, ok := getInput(resp.Meta)
|
||||||
if !ok {
|
if !ok {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,11 +22,11 @@ func main() {
|
|||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
mux := &gemini.ServeMux{}
|
mux := &gemini.Mux{}
|
||||||
mux.Handle("/", gemini.FileServer(os.DirFS("/var/www")))
|
mux.Handle("/", gemini.FileServer(os.DirFS("/var/www")))
|
||||||
|
|
||||||
server := &gemini.Server{
|
server := &gemini.Server{
|
||||||
Handler: logMiddleware(mux),
|
Handler: gemini.LoggingMiddleware(mux),
|
||||||
ReadTimeout: 30 * time.Second,
|
ReadTimeout: 30 * time.Second,
|
||||||
WriteTimeout: 1 * time.Minute,
|
WriteTimeout: 1 * time.Minute,
|
||||||
GetCertificate: certificates.Get,
|
GetCertificate: certificates.Get,
|
||||||
@@ -56,49 +56,3 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func logMiddleware(h gemini.Handler) gemini.Handler {
|
|
||||||
return gemini.HandlerFunc(func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
|
||||||
lw := &logResponseWriter{rw: w}
|
|
||||||
h.ServeGemini(ctx, lw, r)
|
|
||||||
host := r.TLS().ServerName
|
|
||||||
log.Printf("gemini: %s %q %d %d", host, r.URL, lw.status, lw.wrote)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
type logResponseWriter struct {
|
|
||||||
rw gemini.ResponseWriter
|
|
||||||
status gemini.Status
|
|
||||||
meta string
|
|
||||||
mediatype string
|
|
||||||
wroteHeader bool
|
|
||||||
wrote int
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *logResponseWriter) SetMediaType(mediatype string) {
|
|
||||||
w.mediatype = mediatype
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *logResponseWriter) Write(b []byte) (int, error) {
|
|
||||||
if !w.wroteHeader {
|
|
||||||
w.WriteHeader(gemini.StatusSuccess, w.mediatype)
|
|
||||||
}
|
|
||||||
n, err := w.rw.Write(b)
|
|
||||||
w.wrote += n
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *logResponseWriter) WriteHeader(status gemini.Status, meta string) {
|
|
||||||
if w.wroteHeader {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
w.status = status
|
|
||||||
w.meta = meta
|
|
||||||
w.wroteHeader = true
|
|
||||||
w.rw.WriteHeader(status, meta)
|
|
||||||
w.wrote += len(meta) + 5
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *logResponseWriter) Flush() error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -21,7 +21,7 @@ func main() {
|
|||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
mux := &gemini.ServeMux{}
|
mux := &gemini.Mux{}
|
||||||
mux.HandleFunc("/", stream)
|
mux.HandleFunc("/", stream)
|
||||||
|
|
||||||
server := &gemini.Server{
|
server := &gemini.Server{
|
||||||
|
|||||||
5
fs.go
5
fs.go
@@ -1,5 +1,3 @@
|
|||||||
// +build go1.16
|
|
||||||
|
|
||||||
package gemini
|
package gemini
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -153,7 +151,8 @@ func serveFile(w ResponseWriter, r *Request, fsys fs.FS, name string, redirect b
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Use contents of index.gmi if present
|
// Use contents of index.gmi if present
|
||||||
index, err := fsys.Open(path.Join(name, indexPage))
|
name = path.Join(name, indexPage)
|
||||||
|
index, err := fsys.Open(name)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
defer index.Close()
|
defer index.Close()
|
||||||
istat, err := index.Stat()
|
istat, err := index.Stat()
|
||||||
|
|||||||
16
gemini.go
16
gemini.go
@@ -11,16 +11,24 @@ func init() {
|
|||||||
mime.AddExtensionType(".gemini", "text/gemini")
|
mime.AddExtensionType(".gemini", "text/gemini")
|
||||||
}
|
}
|
||||||
|
|
||||||
var crlf = []byte("\r\n")
|
|
||||||
|
|
||||||
// Errors.
|
// Errors.
|
||||||
var (
|
var (
|
||||||
ErrInvalidRequest = errors.New("gemini: invalid request")
|
ErrInvalidRequest = errors.New("gemini: invalid request")
|
||||||
ErrInvalidResponse = errors.New("gemini: invalid response")
|
ErrInvalidResponse = errors.New("gemini: invalid response")
|
||||||
|
|
||||||
ErrCertificateExpired = errors.New("gemini: certificate expired")
|
|
||||||
|
|
||||||
// ErrBodyNotAllowed is returned by ResponseWriter.Write calls
|
// ErrBodyNotAllowed is returned by ResponseWriter.Write calls
|
||||||
// when the response status code does not permit a body.
|
// when the response status code does not permit a body.
|
||||||
ErrBodyNotAllowed = errors.New("gemini: response status code does not allow body")
|
ErrBodyNotAllowed = errors.New("gemini: response status code does not allow body")
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var crlf = []byte("\r\n")
|
||||||
|
|
||||||
|
func trimCRLF(b []byte) ([]byte, bool) {
|
||||||
|
// Check for CR
|
||||||
|
if len(b) < 2 || b[len(b)-2] != '\r' {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
// Trim CRLF
|
||||||
|
b = b[:len(b)-2]
|
||||||
|
return b, true
|
||||||
|
}
|
||||||
|
|||||||
2
go.mod
2
go.mod
@@ -1,5 +1,5 @@
|
|||||||
module git.sr.ht/~adnano/go-gemini
|
module git.sr.ht/~adnano/go-gemini
|
||||||
|
|
||||||
go 1.15
|
go 1.16
|
||||||
|
|
||||||
require golang.org/x/net v0.0.0-20210119194325-5f4716e94777
|
require golang.org/x/net v0.0.0-20210119194325-5f4716e94777
|
||||||
|
|||||||
11
handler.go
11
handler.go
@@ -79,18 +79,21 @@ func StripPrefix(prefix string, h Handler) Handler {
|
|||||||
//
|
//
|
||||||
// The new Handler calls h.ServeGemini to handle each request, but
|
// The new Handler calls h.ServeGemini to handle each request, but
|
||||||
// if a call runs for longer than its time limit, the handler responds with a
|
// if a call runs for longer than its time limit, the handler responds with a
|
||||||
// 40 Temporary Failure error. After such a timeout, writes by h to
|
// 40 Temporary Failure status code and the given message in its response meta.
|
||||||
// its ResponseWriter will return context.DeadlineExceeded.
|
// After such a timeout, writes by h to its ResponseWriter will return
|
||||||
func TimeoutHandler(h Handler, dt time.Duration) Handler {
|
// context.DeadlineExceeded.
|
||||||
|
func TimeoutHandler(h Handler, dt time.Duration, message string) Handler {
|
||||||
return &timeoutHandler{
|
return &timeoutHandler{
|
||||||
h: h,
|
h: h,
|
||||||
dt: dt,
|
dt: dt,
|
||||||
|
msg: message,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type timeoutHandler struct {
|
type timeoutHandler struct {
|
||||||
h Handler
|
h Handler
|
||||||
dt time.Duration
|
dt time.Duration
|
||||||
|
msg string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *timeoutHandler) ServeGemini(ctx context.Context, w ResponseWriter, r *Request) {
|
func (t *timeoutHandler) ServeGemini(ctx context.Context, w ResponseWriter, r *Request) {
|
||||||
@@ -118,7 +121,7 @@ func (t *timeoutHandler) ServeGemini(ctx context.Context, w ResponseWriter, r *R
|
|||||||
w.WriteHeader(tw.status, tw.meta)
|
w.WriteHeader(tw.status, tw.meta)
|
||||||
w.Write(buf.Bytes())
|
w.Write(buf.Bytes())
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
w.WriteHeader(StatusTemporaryFailure, "Timeout")
|
w.WriteHeader(StatusTemporaryFailure, t.msg)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
28
io.go
28
io.go
@@ -1,7 +1,6 @@
|
|||||||
package gemini
|
package gemini
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"context"
|
"context"
|
||||||
"io"
|
"io"
|
||||||
)
|
)
|
||||||
@@ -75,30 +74,3 @@ func (nopReadCloser) Read(p []byte) (int, error) {
|
|||||||
func (nopReadCloser) Close() error {
|
func (nopReadCloser) Close() error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type bufReadCloser struct {
|
|
||||||
br *bufio.Reader // used until empty
|
|
||||||
io.ReadCloser
|
|
||||||
}
|
|
||||||
|
|
||||||
func newBufReadCloser(br *bufio.Reader, rc io.ReadCloser) io.ReadCloser {
|
|
||||||
body := &bufReadCloser{ReadCloser: rc}
|
|
||||||
if br.Buffered() != 0 {
|
|
||||||
body.br = br
|
|
||||||
}
|
|
||||||
return body
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *bufReadCloser) Read(p []byte) (n int, err error) {
|
|
||||||
if b.br != nil {
|
|
||||||
if n := b.br.Buffered(); len(p) > n {
|
|
||||||
p = p[:n]
|
|
||||||
}
|
|
||||||
n, err = b.br.Read(p)
|
|
||||||
if b.br.Buffered() == 0 {
|
|
||||||
b.br = nil
|
|
||||||
}
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
return b.ReadCloser.Read(p)
|
|
||||||
}
|
|
||||||
|
|||||||
53
middleware.go
Normal file
53
middleware.go
Normal file
@@ -0,0 +1,53 @@
|
|||||||
|
package gemini
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
// LoggingMiddleware returns a handler that wraps h and logs Gemini requests
|
||||||
|
// and their responses to the log package's standard logger.
|
||||||
|
// Requests are logged with the format "gemini: {host} {URL} {status code} {bytes written}".
|
||||||
|
func LoggingMiddleware(h Handler) Handler {
|
||||||
|
return HandlerFunc(func(ctx context.Context, w ResponseWriter, r *Request) {
|
||||||
|
lw := &logResponseWriter{rw: w}
|
||||||
|
h.ServeGemini(ctx, lw, r)
|
||||||
|
host := r.ServerName()
|
||||||
|
log.Printf("gemini: %s %q %d %d", host, r.URL, lw.Status, lw.Wrote)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type logResponseWriter struct {
|
||||||
|
Status Status
|
||||||
|
Wrote int
|
||||||
|
rw ResponseWriter
|
||||||
|
mediatype string
|
||||||
|
wroteHeader bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *logResponseWriter) SetMediaType(mediatype string) {
|
||||||
|
w.mediatype = mediatype
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *logResponseWriter) Write(b []byte) (int, error) {
|
||||||
|
if !w.wroteHeader {
|
||||||
|
w.WriteHeader(StatusSuccess, w.mediatype)
|
||||||
|
}
|
||||||
|
n, err := w.rw.Write(b)
|
||||||
|
w.Wrote += n
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *logResponseWriter) WriteHeader(status Status, meta string) {
|
||||||
|
if w.wroteHeader {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.wroteHeader = true
|
||||||
|
w.Status = status
|
||||||
|
w.Wrote += len(meta) + 5
|
||||||
|
w.rw.WriteHeader(status, meta)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *logResponseWriter) Flush() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
26
mux.go
26
mux.go
@@ -10,7 +10,7 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
// ServeMux is a Gemini request multiplexer.
|
// Mux is a Gemini request multiplexer.
|
||||||
// It matches the URL of each incoming request against a list of registered
|
// It matches the URL of each incoming request against a list of registered
|
||||||
// patterns and calls the handler for the pattern that
|
// patterns and calls the handler for the pattern that
|
||||||
// most closely matches the URL.
|
// most closely matches the URL.
|
||||||
@@ -55,17 +55,17 @@ import (
|
|||||||
// scheme of "gemini".
|
// scheme of "gemini".
|
||||||
//
|
//
|
||||||
// If a subtree has been registered and a request is received naming the
|
// If a subtree has been registered and a request is received naming the
|
||||||
// subtree root without its trailing slash, ServeMux redirects that
|
// subtree root without its trailing slash, Mux redirects that
|
||||||
// request to the subtree root (adding the trailing slash). This behavior can
|
// request to the subtree root (adding the trailing slash). This behavior can
|
||||||
// be overridden with a separate registration for the path without
|
// be overridden with a separate registration for the path without
|
||||||
// the trailing slash. For example, registering "/images/" causes ServeMux
|
// the trailing slash. For example, registering "/images/" causes Mux
|
||||||
// to redirect a request for "/images" to "/images/", unless "/images" has
|
// to redirect a request for "/images" to "/images/", unless "/images" has
|
||||||
// been registered separately.
|
// been registered separately.
|
||||||
//
|
//
|
||||||
// ServeMux also takes care of sanitizing the URL request path and
|
// Mux also takes care of sanitizing the URL request path and
|
||||||
// redirecting any request containing . or .. elements or repeated slashes
|
// redirecting any request containing . or .. elements or repeated slashes
|
||||||
// to an equivalent, cleaner URL.
|
// to an equivalent, cleaner URL.
|
||||||
type ServeMux struct {
|
type Mux struct {
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
m map[muxKey]Handler
|
m map[muxKey]Handler
|
||||||
es []muxEntry // slice of entries sorted from longest to shortest
|
es []muxEntry // slice of entries sorted from longest to shortest
|
||||||
@@ -106,7 +106,7 @@ func cleanPath(p string) string {
|
|||||||
|
|
||||||
// Find a handler on a handler map given a path string.
|
// Find a handler on a handler map given a path string.
|
||||||
// Most-specific (longest) pattern wins.
|
// Most-specific (longest) pattern wins.
|
||||||
func (mux *ServeMux) match(key muxKey) Handler {
|
func (mux *Mux) match(key muxKey) Handler {
|
||||||
// Check for exact match first.
|
// Check for exact match first.
|
||||||
if r, ok := mux.m[key]; ok {
|
if r, ok := mux.m[key]; ok {
|
||||||
return r
|
return r
|
||||||
@@ -134,7 +134,7 @@ func (mux *ServeMux) match(key muxKey) Handler {
|
|||||||
// This occurs when a handler for path + "/" was already registered, but
|
// This occurs when a handler for path + "/" was already registered, but
|
||||||
// not for path itself. If the path needs appending to, it creates a new
|
// not for path itself. If the path needs appending to, it creates a new
|
||||||
// URL, setting the path to u.Path + "/" and returning true to indicate so.
|
// URL, setting the path to u.Path + "/" and returning true to indicate so.
|
||||||
func (mux *ServeMux) redirectToPathSlash(key muxKey, u *url.URL) (*url.URL, bool) {
|
func (mux *Mux) redirectToPathSlash(key muxKey, u *url.URL) (*url.URL, bool) {
|
||||||
mux.mu.RLock()
|
mux.mu.RLock()
|
||||||
shouldRedirect := mux.shouldRedirectRLocked(key)
|
shouldRedirect := mux.shouldRedirectRLocked(key)
|
||||||
mux.mu.RUnlock()
|
mux.mu.RUnlock()
|
||||||
@@ -146,8 +146,8 @@ func (mux *ServeMux) redirectToPathSlash(key muxKey, u *url.URL) (*url.URL, bool
|
|||||||
|
|
||||||
// shouldRedirectRLocked reports whether the given path and host should be redirected to
|
// shouldRedirectRLocked reports whether the given path and host should be redirected to
|
||||||
// path+"/". This should happen if a handler is registered for path+"/" but
|
// path+"/". This should happen if a handler is registered for path+"/" but
|
||||||
// not path -- see comments at ServeMux.
|
// not path -- see comments at Mux.
|
||||||
func (mux *ServeMux) shouldRedirectRLocked(key muxKey) bool {
|
func (mux *Mux) shouldRedirectRLocked(key muxKey) bool {
|
||||||
if _, exist := mux.m[key]; exist {
|
if _, exist := mux.m[key]; exist {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
@@ -177,7 +177,7 @@ func getWildcard(hostname string) (string, bool) {
|
|||||||
// the path is not in its canonical form, the handler will be an
|
// the path is not in its canonical form, the handler will be an
|
||||||
// internally-generated handler that redirects to the canonical path. If the
|
// internally-generated handler that redirects to the canonical path. If the
|
||||||
// host contains a port, it is ignored when matching handlers.
|
// host contains a port, it is ignored when matching handlers.
|
||||||
func (mux *ServeMux) Handler(r *Request) Handler {
|
func (mux *Mux) Handler(r *Request) Handler {
|
||||||
scheme := r.URL.Scheme
|
scheme := r.URL.Scheme
|
||||||
host := r.URL.Hostname()
|
host := r.URL.Hostname()
|
||||||
path := cleanPath(r.URL.Path)
|
path := cleanPath(r.URL.Path)
|
||||||
@@ -212,14 +212,14 @@ func (mux *ServeMux) Handler(r *Request) Handler {
|
|||||||
|
|
||||||
// ServeGemini dispatches the request to the handler whose
|
// ServeGemini dispatches the request to the handler whose
|
||||||
// pattern most closely matches the request URL.
|
// pattern most closely matches the request URL.
|
||||||
func (mux *ServeMux) ServeGemini(ctx context.Context, w ResponseWriter, r *Request) {
|
func (mux *Mux) ServeGemini(ctx context.Context, w ResponseWriter, r *Request) {
|
||||||
h := mux.Handler(r)
|
h := mux.Handler(r)
|
||||||
h.ServeGemini(ctx, w, r)
|
h.ServeGemini(ctx, w, r)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handle registers the handler for the given pattern.
|
// Handle registers the handler for the given pattern.
|
||||||
// If a handler already exists for pattern, Handle panics.
|
// If a handler already exists for pattern, Handle panics.
|
||||||
func (mux *ServeMux) Handle(pattern string, handler Handler) {
|
func (mux *Mux) Handle(pattern string, handler Handler) {
|
||||||
if pattern == "" {
|
if pattern == "" {
|
||||||
panic("gemini: invalid pattern")
|
panic("gemini: invalid pattern")
|
||||||
}
|
}
|
||||||
@@ -294,6 +294,6 @@ func appendSorted(es []muxEntry, e muxEntry) []muxEntry {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// HandleFunc registers the handler function for the given pattern.
|
// HandleFunc registers the handler function for the given pattern.
|
||||||
func (mux *ServeMux) HandleFunc(pattern string, handler HandlerFunc) {
|
func (mux *Mux) HandleFunc(pattern string, handler HandlerFunc) {
|
||||||
mux.Handle(pattern, handler)
|
mux.Handle(pattern, handler)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ type nopHandler struct{}
|
|||||||
|
|
||||||
func (*nopHandler) ServeGemini(context.Context, ResponseWriter, *Request) {}
|
func (*nopHandler) ServeGemini(context.Context, ResponseWriter, *Request) {}
|
||||||
|
|
||||||
func TestServeMuxMatch(t *testing.T) {
|
func TestMuxMatch(t *testing.T) {
|
||||||
type Match struct {
|
type Match struct {
|
||||||
URL string
|
URL string
|
||||||
Ok bool
|
Ok bool
|
||||||
@@ -292,7 +292,7 @@ func TestServeMuxMatch(t *testing.T) {
|
|||||||
|
|
||||||
for i, test := range tests {
|
for i, test := range tests {
|
||||||
h := &nopHandler{}
|
h := &nopHandler{}
|
||||||
var mux ServeMux
|
var mux Mux
|
||||||
mux.Handle(test.Pattern, h)
|
mux.Handle(test.Pattern, h)
|
||||||
|
|
||||||
for _, match := range tests[i].Matches {
|
for _, match := range tests[i].Matches {
|
||||||
|
|||||||
65
request.go
65
request.go
@@ -29,6 +29,7 @@ type Request struct {
|
|||||||
Certificate *tls.Certificate
|
Certificate *tls.Certificate
|
||||||
|
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
|
tls *tls.ConnectionState
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewRequest returns a new request.
|
// NewRequest returns a new request.
|
||||||
@@ -50,60 +51,78 @@ func NewRequest(rawurl string) (*Request, error) {
|
|||||||
// for specialized applications; most code should use the Server
|
// for specialized applications; most code should use the Server
|
||||||
// to read requests and handle them via the Handler interface.
|
// to read requests and handle them via the Handler interface.
|
||||||
func ReadRequest(r io.Reader) (*Request, error) {
|
func ReadRequest(r io.Reader) (*Request, error) {
|
||||||
// Read URL
|
// Limit request size
|
||||||
r = io.LimitReader(r, 1026)
|
r = io.LimitReader(r, 1026)
|
||||||
br := bufio.NewReaderSize(r, 1026)
|
br := bufio.NewReaderSize(r, 1026)
|
||||||
rawurl, err := br.ReadString('\r')
|
b, err := br.ReadBytes('\n')
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
if err == io.EOF {
|
||||||
}
|
|
||||||
// Read terminating line feed
|
|
||||||
if b, err := br.ReadByte(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
} else if b != '\n' {
|
|
||||||
return nil, ErrInvalidRequest
|
return nil, ErrInvalidRequest
|
||||||
}
|
}
|
||||||
// Trim carriage return
|
return nil, err
|
||||||
rawurl = rawurl[:len(rawurl)-1]
|
}
|
||||||
// Validate URL
|
// Read URL
|
||||||
if len(rawurl) > 1024 {
|
rawurl, ok := trimCRLF(b)
|
||||||
|
if !ok {
|
||||||
return nil, ErrInvalidRequest
|
return nil, ErrInvalidRequest
|
||||||
}
|
}
|
||||||
u, err := url.Parse(rawurl)
|
if len(rawurl) == 0 {
|
||||||
|
return nil, ErrInvalidRequest
|
||||||
|
}
|
||||||
|
u, err := url.Parse(string(rawurl))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &Request{URL: u}, nil
|
return &Request{URL: u}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write writes a Gemini request in wire format.
|
// WriteTo writes r to w in the Gemini request format.
|
||||||
// This method consults the request URL only.
|
// This method consults the request URL only.
|
||||||
func (r *Request) Write(w io.Writer) error {
|
func (r *Request) WriteTo(w io.Writer) (int64, error) {
|
||||||
bw := bufio.NewWriterSize(w, 1026)
|
bw := bufio.NewWriterSize(w, 1026)
|
||||||
url := r.URL.String()
|
url := r.URL.String()
|
||||||
if len(url) > 1024 {
|
if len(url) > 1024 {
|
||||||
return ErrInvalidRequest
|
return 0, ErrInvalidRequest
|
||||||
}
|
}
|
||||||
if _, err := bw.WriteString(url); err != nil {
|
var wrote int64
|
||||||
return err
|
n, err := bw.WriteString(url)
|
||||||
|
wrote += int64(n)
|
||||||
|
if err != nil {
|
||||||
|
return wrote, err
|
||||||
}
|
}
|
||||||
if _, err := bw.Write(crlf); err != nil {
|
n, err = bw.Write(crlf)
|
||||||
return err
|
wrote += int64(n)
|
||||||
|
if err != nil {
|
||||||
|
return wrote, err
|
||||||
}
|
}
|
||||||
return bw.Flush()
|
return wrote, bw.Flush()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Conn returns the network connection on which the request was received.
|
// Conn returns the network connection on which the request was received.
|
||||||
|
// Conn returns nil for client requests.
|
||||||
func (r *Request) Conn() net.Conn {
|
func (r *Request) Conn() net.Conn {
|
||||||
return r.conn
|
return r.conn
|
||||||
}
|
}
|
||||||
|
|
||||||
// TLS returns information about the TLS connection on which the
|
// TLS returns information about the TLS connection on which the
|
||||||
// request was received.
|
// request was received.
|
||||||
|
// TLS returns nil for client requests.
|
||||||
func (r *Request) TLS() *tls.ConnectionState {
|
func (r *Request) TLS() *tls.ConnectionState {
|
||||||
|
if r.tls == nil {
|
||||||
if tlsConn, ok := r.conn.(*tls.Conn); ok {
|
if tlsConn, ok := r.conn.(*tls.Conn); ok {
|
||||||
state := tlsConn.ConnectionState()
|
state := tlsConn.ConnectionState()
|
||||||
return &state
|
r.tls = &state
|
||||||
}
|
}
|
||||||
return nil
|
}
|
||||||
|
return r.tls
|
||||||
|
}
|
||||||
|
|
||||||
|
// ServerName returns the value of the TLS Server Name Indication extension
|
||||||
|
// sent by the client.
|
||||||
|
// ServerName returns an empty string for client requests.
|
||||||
|
func (r *Request) ServerName() string {
|
||||||
|
if tls := r.TLS(); tls != nil {
|
||||||
|
return tls.ServerName
|
||||||
|
}
|
||||||
|
return ""
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ package gemini
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"io"
|
|
||||||
"net/url"
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
@@ -36,25 +35,25 @@ func TestReadRequest(t *testing.T) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "\r\n",
|
Raw: "\r\n",
|
||||||
URL: &url.URL{},
|
Err: ErrInvalidRequest,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "gemini://example.com\n",
|
Raw: "gemini://example.com\n",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidRequest,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "gemini://example.com",
|
Raw: "gemini://example.com",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidRequest,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
// 1030 bytes
|
// 1030 bytes
|
||||||
Raw: maxURL + "xxxxxx",
|
Raw: maxURL + "xxxxxx",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidRequest,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
// 1027 bytes
|
// 1027 bytes
|
||||||
Raw: maxURL + "x" + "\r\n",
|
Raw: maxURL + "x" + "\r\n",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidRequest,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
// 1024 bytes
|
// 1024 bytes
|
||||||
@@ -119,7 +118,7 @@ func TestWriteRequest(t *testing.T) {
|
|||||||
t.Logf("%s", test.Req.URL)
|
t.Logf("%s", test.Req.URL)
|
||||||
var b strings.Builder
|
var b strings.Builder
|
||||||
bw := bufio.NewWriter(&b)
|
bw := bufio.NewWriter(&b)
|
||||||
err := test.Req.Write(bw)
|
_, err := test.Req.WriteTo(bw)
|
||||||
if err != test.Err {
|
if err != test.Err {
|
||||||
t.Errorf("expected err = %v, got %v", test.Err, err)
|
t.Errorf("expected err = %v, got %v", test.Err, err)
|
||||||
}
|
}
|
||||||
|
|||||||
91
response.go
91
response.go
@@ -3,13 +3,14 @@ package gemini
|
|||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
// The default media type for responses.
|
// The default media type for responses.
|
||||||
const defaultMediaType = "text/gemini; charset=utf-8"
|
const defaultMediaType = "text/gemini"
|
||||||
|
|
||||||
// Response represents the response from a Gemini request.
|
// Response represents the response from a Gemini request.
|
||||||
//
|
//
|
||||||
@@ -43,52 +44,56 @@ type Response struct {
|
|||||||
// ReadResponse reads a Gemini response from the provided io.ReadCloser.
|
// ReadResponse reads a Gemini response from the provided io.ReadCloser.
|
||||||
func ReadResponse(r io.ReadCloser) (*Response, error) {
|
func ReadResponse(r io.ReadCloser) (*Response, error) {
|
||||||
resp := &Response{}
|
resp := &Response{}
|
||||||
br := bufio.NewReader(r)
|
|
||||||
|
|
||||||
// Read the status
|
// Limit response header size
|
||||||
statusB := make([]byte, 2)
|
lr := io.LimitReader(r, 1029)
|
||||||
if _, err := br.Read(statusB); err != nil {
|
// Wrap the reader to remove the limit later on
|
||||||
|
wr := &struct{ io.Reader }{lr}
|
||||||
|
br := bufio.NewReader(wr)
|
||||||
|
|
||||||
|
// Read response header
|
||||||
|
b, err := br.ReadBytes('\n')
|
||||||
|
if err != nil {
|
||||||
|
if err == io.EOF {
|
||||||
|
return nil, ErrInvalidResponse
|
||||||
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
status, err := strconv.Atoi(string(statusB))
|
if len(b) < 3 {
|
||||||
|
return nil, ErrInvalidResponse
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read the status
|
||||||
|
status, err := strconv.Atoi(string(b[:2]))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, ErrInvalidResponse
|
return nil, ErrInvalidResponse
|
||||||
}
|
}
|
||||||
resp.Status = Status(status)
|
resp.Status = Status(status)
|
||||||
|
|
||||||
// Read one space
|
// Read one space
|
||||||
if b, err := br.ReadByte(); err != nil {
|
if b[2] != ' ' {
|
||||||
return nil, err
|
|
||||||
} else if b != ' ' {
|
|
||||||
return nil, ErrInvalidResponse
|
return nil, ErrInvalidResponse
|
||||||
}
|
}
|
||||||
|
|
||||||
// Read the meta
|
// Read the meta
|
||||||
meta, err := br.ReadString('\r')
|
meta, ok := trimCRLF(b[3:])
|
||||||
if err != nil {
|
if !ok {
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
// Trim carriage return
|
|
||||||
meta = meta[:len(meta)-1]
|
|
||||||
// Ensure meta is less than or equal to 1024 bytes
|
|
||||||
if len(meta) > 1024 {
|
|
||||||
return nil, ErrInvalidResponse
|
return nil, ErrInvalidResponse
|
||||||
}
|
}
|
||||||
if resp.Status.Class() == StatusSuccess && meta == "" {
|
if len(meta) == 0 || len(meta) > 1024 {
|
||||||
// Use default media type
|
|
||||||
meta = defaultMediaType
|
|
||||||
}
|
|
||||||
resp.Meta = meta
|
|
||||||
|
|
||||||
// Read terminating newline
|
|
||||||
if b, err := br.ReadByte(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
} else if b != '\n' {
|
|
||||||
return nil, ErrInvalidResponse
|
return nil, ErrInvalidResponse
|
||||||
}
|
}
|
||||||
|
resp.Meta = string(meta)
|
||||||
|
|
||||||
if resp.Status.Class() == StatusSuccess {
|
if resp.Status.Class() == StatusSuccess {
|
||||||
resp.Body = newBufReadCloser(br, r)
|
// Use unlimited reader
|
||||||
|
wr.Reader = r
|
||||||
|
|
||||||
|
type readCloser struct {
|
||||||
|
io.Reader
|
||||||
|
io.Closer
|
||||||
|
}
|
||||||
|
resp.Body = readCloser{br, r}
|
||||||
} else {
|
} else {
|
||||||
resp.Body = nopReadCloser{}
|
resp.Body = nopReadCloser{}
|
||||||
r.Close()
|
r.Close()
|
||||||
@@ -111,6 +116,29 @@ func (r *Response) TLS() *tls.ConnectionState {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WriteTo writes r to w in the Gemini response format, including the
|
||||||
|
// header and body.
|
||||||
|
//
|
||||||
|
// This method consults the Status, Meta, and Body fields of the response.
|
||||||
|
// The Response Body is closed after it is sent.
|
||||||
|
func (r *Response) WriteTo(w io.Writer) (int64, error) {
|
||||||
|
var wrote int64
|
||||||
|
n, err := fmt.Fprintf(w, "%02d %s\r\n", r.Status, r.Meta)
|
||||||
|
wrote += int64(n)
|
||||||
|
if err != nil {
|
||||||
|
return wrote, err
|
||||||
|
}
|
||||||
|
if r.Body != nil {
|
||||||
|
defer r.Body.Close()
|
||||||
|
n, err := io.Copy(w, r.Body)
|
||||||
|
wrote += n
|
||||||
|
if err != nil {
|
||||||
|
return wrote, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return wrote, nil
|
||||||
|
}
|
||||||
|
|
||||||
// A ResponseWriter interface is used by a Gemini handler to construct
|
// A ResponseWriter interface is used by a Gemini handler to construct
|
||||||
// a Gemini response.
|
// a Gemini response.
|
||||||
//
|
//
|
||||||
@@ -118,8 +146,8 @@ func (r *Response) TLS() *tls.ConnectionState {
|
|||||||
// has returned.
|
// has returned.
|
||||||
type ResponseWriter interface {
|
type ResponseWriter interface {
|
||||||
// SetMediaType sets the media type that will be sent by Write for a
|
// SetMediaType sets the media type that will be sent by Write for a
|
||||||
// successful response. If no media type is set, a default of
|
// successful response. If no media type is set, a default media type of
|
||||||
// "text/gemini; charset=utf-8" will be used.
|
// "text/gemini" will be used.
|
||||||
//
|
//
|
||||||
// Setting the media type after a call to Write or WriteHeader has
|
// Setting the media type after a call to Write or WriteHeader has
|
||||||
// no effect.
|
// no effect.
|
||||||
@@ -130,7 +158,7 @@ type ResponseWriter interface {
|
|||||||
// If WriteHeader has not yet been called, Write calls WriteHeader with
|
// If WriteHeader has not yet been called, Write calls WriteHeader with
|
||||||
// StatusSuccess and the media type set in SetMediaType before writing the data.
|
// StatusSuccess and the media type set in SetMediaType before writing the data.
|
||||||
// If no media type was set, Write uses a default media type of
|
// If no media type was set, Write uses a default media type of
|
||||||
// "text/gemini; charset=utf-8".
|
// "text/gemini".
|
||||||
Write([]byte) (int, error)
|
Write([]byte) (int, error)
|
||||||
|
|
||||||
// WriteHeader sends a Gemini response header with the provided
|
// WriteHeader sends a Gemini response header with the provided
|
||||||
@@ -151,7 +179,6 @@ type ResponseWriter interface {
|
|||||||
|
|
||||||
type responseWriter struct {
|
type responseWriter struct {
|
||||||
bw *bufio.Writer
|
bw *bufio.Writer
|
||||||
cl io.Closer
|
|
||||||
mediatype string
|
mediatype string
|
||||||
wroteHeader bool
|
wroteHeader bool
|
||||||
bodyAllowed bool
|
bodyAllowed bool
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ package gemini
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
@@ -38,6 +37,15 @@ func TestReadWriteResponse(t *testing.T) {
|
|||||||
Meta: "/redirect",
|
Meta: "/redirect",
|
||||||
SkipWrite: true, // skip write test since result won't match Raw
|
SkipWrite: true, // skip write test since result won't match Raw
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Raw: "32 " + maxURL + "\r\n",
|
||||||
|
Status: 32,
|
||||||
|
Meta: maxURL,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Raw: "33 " + maxURL + "xxxx" + "\r\n",
|
||||||
|
Err: ErrInvalidResponse,
|
||||||
|
},
|
||||||
{
|
{
|
||||||
Raw: "99 Unknown status code\r\n",
|
Raw: "99 Unknown status code\r\n",
|
||||||
Status: 99,
|
Status: 99,
|
||||||
@@ -57,15 +65,15 @@ func TestReadWriteResponse(t *testing.T) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "",
|
Raw: "",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidResponse,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "10 Search query",
|
Raw: "10 Search query",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidResponse,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "20 text/gemini\nHello, world!",
|
Raw: "20 text/gemini\nHello, world!",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidResponse,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "20 text/gemini\rHello, world!",
|
Raw: "20 text/gemini\rHello, world!",
|
||||||
@@ -73,7 +81,7 @@ func TestReadWriteResponse(t *testing.T) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "20 text/gemini\r",
|
Raw: "20 text/gemini\r",
|
||||||
Err: io.EOF,
|
Err: ErrInvalidResponse,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Raw: "abcdefghijklmnopqrstuvwxyz",
|
Raw: "abcdefghijklmnopqrstuvwxyz",
|
||||||
@@ -83,11 +91,11 @@ func TestReadWriteResponse(t *testing.T) {
|
|||||||
|
|
||||||
for _, test := range tests {
|
for _, test := range tests {
|
||||||
t.Logf("%#v", test.Raw)
|
t.Logf("%#v", test.Raw)
|
||||||
resp, err := ReadResponse(ioutil.NopCloser(strings.NewReader(test.Raw)))
|
resp, err := ReadResponse(io.NopCloser(strings.NewReader(test.Raw)))
|
||||||
if err != test.Err {
|
if err != test.Err {
|
||||||
t.Errorf("expected err = %v, got %v", test.Err, err)
|
t.Errorf("expected err = %v, got %v", test.Err, err)
|
||||||
}
|
}
|
||||||
if test.Err != nil {
|
if err != nil {
|
||||||
// No response
|
// No response
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -97,7 +105,7 @@ func TestReadWriteResponse(t *testing.T) {
|
|||||||
if resp.Meta != test.Meta {
|
if resp.Meta != test.Meta {
|
||||||
t.Errorf("expected meta = %s, got %s", test.Meta, resp.Meta)
|
t.Errorf("expected meta = %s, got %s", test.Meta, resp.Meta)
|
||||||
}
|
}
|
||||||
b, _ := ioutil.ReadAll(resp.Body)
|
b, _ := io.ReadAll(resp.Body)
|
||||||
body := string(b)
|
body := string(b)
|
||||||
if body != test.Body {
|
if body != test.Body {
|
||||||
t.Errorf("expected body = %#v, got %#v", test.Body, body)
|
t.Errorf("expected body = %#v, got %#v", test.Body, body)
|
||||||
|
|||||||
140
tofu/tofu.go
140
tofu/tofu.go
@@ -4,17 +4,16 @@ package tofu
|
|||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"bytes"
|
"bytes"
|
||||||
"crypto/sha512"
|
"crypto/sha256"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"errors"
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// KnownHosts represents a list of known hosts.
|
// KnownHosts represents a list of known hosts.
|
||||||
@@ -84,7 +83,11 @@ func (k *KnownHosts) WriteTo(w io.Writer) (int64, error) {
|
|||||||
|
|
||||||
// Load loads the known hosts entries from the provided path.
|
// Load loads the known hosts entries from the provided path.
|
||||||
func (k *KnownHosts) Load(path string) error {
|
func (k *KnownHosts) Load(path string) error {
|
||||||
f, err := os.Open(path)
|
if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_RDONLY, 0644)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -133,6 +136,9 @@ func (k *KnownHosts) Parse(r io.Reader) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
if h.Algorithm != "sha256" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
k.hosts[h.Hostname] = h
|
k.hosts[h.Hostname] = h
|
||||||
}
|
}
|
||||||
@@ -143,22 +149,17 @@ func (k *KnownHosts) Parse(r io.Reader) error {
|
|||||||
// TOFU implements basic trust on first use.
|
// TOFU implements basic trust on first use.
|
||||||
//
|
//
|
||||||
// If the host is not on file, it is added to the list.
|
// If the host is not on file, it is added to the list.
|
||||||
// If the host on file is expired, a new entry is added to the list.
|
|
||||||
// If the fingerprint does not match the one on file, an error is returned.
|
// If the fingerprint does not match the one on file, an error is returned.
|
||||||
func (k *KnownHosts) TOFU(hostname string, cert *x509.Certificate) error {
|
func (k *KnownHosts) TOFU(hostname string, cert *x509.Certificate) error {
|
||||||
host := NewHost(hostname, cert.Raw, cert.NotAfter)
|
host := NewHost(hostname, cert.Raw)
|
||||||
|
|
||||||
knownHost, ok := k.Lookup(hostname)
|
knownHost, ok := k.Lookup(hostname)
|
||||||
if !ok || time.Now().After(knownHost.Expires) {
|
if !ok {
|
||||||
k.Add(host)
|
k.Add(host)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
if host.Fingerprint != knownHost.Fingerprint {
|
||||||
// Check fingerprint
|
|
||||||
if !bytes.Equal(knownHost.Fingerprint, host.Fingerprint) {
|
|
||||||
return fmt.Errorf("fingerprint for %q does not match", hostname)
|
return fmt.Errorf("fingerprint for %q does not match", hostname)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -266,21 +267,16 @@ func (p *PersistentHosts) Entries() []Host {
|
|||||||
// TOFU implements trust on first use with a persistent set of known hosts.
|
// TOFU implements trust on first use with a persistent set of known hosts.
|
||||||
//
|
//
|
||||||
// If the host is not on file, it is added to the list.
|
// If the host is not on file, it is added to the list.
|
||||||
// If the host on file is expired, a new entry is added to the list.
|
|
||||||
// If the fingerprint does not match the one on file, an error is returned.
|
// If the fingerprint does not match the one on file, an error is returned.
|
||||||
func (p *PersistentHosts) TOFU(hostname string, cert *x509.Certificate) error {
|
func (p *PersistentHosts) TOFU(hostname string, cert *x509.Certificate) error {
|
||||||
host := NewHost(hostname, cert.Raw, cert.NotAfter)
|
host := NewHost(hostname, cert.Raw)
|
||||||
|
|
||||||
knownHost, ok := p.Lookup(hostname)
|
knownHost, ok := p.Lookup(hostname)
|
||||||
if !ok || time.Now().After(knownHost.Expires) {
|
if !ok {
|
||||||
return p.Add(host)
|
return p.Add(host)
|
||||||
}
|
}
|
||||||
|
if host.Fingerprint != knownHost.Fingerprint {
|
||||||
// Check fingerprint
|
|
||||||
if !bytes.Equal(knownHost.Fingerprint, host.Fingerprint) {
|
|
||||||
return fmt.Errorf("fingerprint for %q does not match", hostname)
|
return fmt.Errorf("fingerprint for %q does not match", hostname)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -292,21 +288,19 @@ func (p *PersistentHosts) Close() error {
|
|||||||
// Host represents a host entry with a fingerprint using a certain algorithm.
|
// Host represents a host entry with a fingerprint using a certain algorithm.
|
||||||
type Host struct {
|
type Host struct {
|
||||||
Hostname string // hostname
|
Hostname string // hostname
|
||||||
Algorithm string // fingerprint algorithm e.g. SHA-512
|
Algorithm string // fingerprint algorithm e.g. sha256
|
||||||
Fingerprint Fingerprint // fingerprint
|
Fingerprint string // fingerprint
|
||||||
Expires time.Time // unix time of the fingerprint expiration date
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewHost returns a new host with a SHA-512 fingerprint of
|
// NewHost returns a new host with a SHA256 fingerprint of
|
||||||
// the provided raw data.
|
// the provided raw data.
|
||||||
func NewHost(hostname string, raw []byte, expires time.Time) Host {
|
func NewHost(hostname string, raw []byte) Host {
|
||||||
sum := sha512.Sum512(raw)
|
sum := sha256.Sum256(raw)
|
||||||
|
|
||||||
return Host{
|
return Host{
|
||||||
Hostname: hostname,
|
Hostname: hostname,
|
||||||
Algorithm: "SHA-512",
|
Algorithm: "sha256",
|
||||||
Fingerprint: sum[:],
|
Fingerprint: base64.StdEncoding.EncodeToString(sum[:]),
|
||||||
Expires: expires,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -324,95 +318,19 @@ func (h Host) String() string {
|
|||||||
b.WriteByte(' ')
|
b.WriteByte(' ')
|
||||||
b.WriteString(h.Algorithm)
|
b.WriteString(h.Algorithm)
|
||||||
b.WriteByte(' ')
|
b.WriteByte(' ')
|
||||||
b.WriteString(h.Fingerprint.String())
|
b.WriteString(h.Fingerprint)
|
||||||
b.WriteByte(' ')
|
|
||||||
b.WriteString(strconv.FormatInt(h.Expires.Unix(), 10))
|
|
||||||
return b.String()
|
return b.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnmarshalText unmarshals the host from the provided text.
|
// UnmarshalText unmarshals the host from the provided text.
|
||||||
func (h *Host) UnmarshalText(text []byte) error {
|
func (h *Host) UnmarshalText(text []byte) error {
|
||||||
const format = "hostname algorithm hex-fingerprint expiry-unix-ts"
|
|
||||||
|
|
||||||
parts := bytes.Split(text, []byte(" "))
|
parts := bytes.Split(text, []byte(" "))
|
||||||
if len(parts) != 4 {
|
if len(parts) != 3 {
|
||||||
return fmt.Errorf("expected the format %q", format)
|
return fmt.Errorf("expected the format 'hostname algorithm fingerprint'")
|
||||||
}
|
|
||||||
|
|
||||||
if len(parts[0]) == 0 {
|
|
||||||
return errors.New("empty hostname")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
h.Hostname = string(parts[0])
|
h.Hostname = string(parts[0])
|
||||||
|
h.Algorithm = string(parts[1])
|
||||||
algorithm := string(parts[1])
|
h.Fingerprint = string(parts[2])
|
||||||
if algorithm != "SHA-512" {
|
|
||||||
return fmt.Errorf("unsupported algorithm %q", algorithm)
|
|
||||||
}
|
|
||||||
|
|
||||||
h.Algorithm = algorithm
|
|
||||||
|
|
||||||
fingerprint := make([]byte, 0, sha512.Size)
|
|
||||||
scanner := bufio.NewScanner(bytes.NewReader(parts[2]))
|
|
||||||
scanner.Split(scanFingerprint)
|
|
||||||
|
|
||||||
for scanner.Scan() {
|
|
||||||
b, err := strconv.ParseUint(scanner.Text(), 16, 8)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to parse fingerprint hash: %w", err)
|
|
||||||
}
|
|
||||||
fingerprint = append(fingerprint, byte(b))
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(fingerprint) != sha512.Size {
|
|
||||||
return fmt.Errorf("invalid fingerprint size %d, expected %d",
|
|
||||||
len(fingerprint), sha512.Size)
|
|
||||||
}
|
|
||||||
|
|
||||||
h.Fingerprint = fingerprint
|
|
||||||
|
|
||||||
unix, err := strconv.ParseInt(string(parts[3]), 10, 0)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("invalid unix timestamp: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
h.Expires = time.Unix(unix, 0)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func scanFingerprint(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
|
||||||
if atEOF && len(data) == 0 {
|
|
||||||
return 0, nil, nil
|
|
||||||
}
|
|
||||||
if i := bytes.IndexByte(data, ':'); i >= 0 {
|
|
||||||
// We have a full newline-terminated line.
|
|
||||||
return i + 1, data[0:i], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// If we're at EOF, we have a final, non-terminated hex byte
|
|
||||||
if atEOF {
|
|
||||||
return len(data), data, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Request more data.
|
|
||||||
return 0, nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Fingerprint represents a fingerprint.
|
|
||||||
type Fingerprint []byte
|
|
||||||
|
|
||||||
// String returns a string representation of the fingerprint.
|
|
||||||
func (f Fingerprint) String() string {
|
|
||||||
var sb strings.Builder
|
|
||||||
|
|
||||||
for i, b := range f {
|
|
||||||
if i > 0 {
|
|
||||||
sb.WriteByte(':')
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Fprintf(&sb, "%02X", b)
|
|
||||||
}
|
|
||||||
|
|
||||||
return sb.String()
|
|
||||||
}
|
|
||||||
|
|||||||
212
vendor.go
212
vendor.go
@@ -1,212 +0,0 @@
|
|||||||
// Hostname verification code from the crypto/x509 package.
|
|
||||||
// Modified to allow Common Names in the short term, until new certificates
|
|
||||||
// can be issued with SANs.
|
|
||||||
|
|
||||||
// Copyright 2011 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.
|
|
||||||
|
|
||||||
package gemini
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/x509"
|
|
||||||
"net"
|
|
||||||
"strings"
|
|
||||||
"unicode/utf8"
|
|
||||||
)
|
|
||||||
|
|
||||||
var oidExtensionSubjectAltName = []int{2, 5, 29, 17}
|
|
||||||
|
|
||||||
func hasSANExtension(c *x509.Certificate) bool {
|
|
||||||
for _, e := range c.Extensions {
|
|
||||||
if e.Id.Equal(oidExtensionSubjectAltName) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func validHostnamePattern(host string) bool { return validHostname(host, true) }
|
|
||||||
func validHostnameInput(host string) bool { return validHostname(host, false) }
|
|
||||||
|
|
||||||
// validHostname reports whether host is a valid hostname that can be matched or
|
|
||||||
// matched against according to RFC 6125 2.2, with some leniency to accommodate
|
|
||||||
// legacy values.
|
|
||||||
func validHostname(host string, isPattern bool) bool {
|
|
||||||
if !isPattern {
|
|
||||||
host = strings.TrimSuffix(host, ".")
|
|
||||||
}
|
|
||||||
if len(host) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, part := range strings.Split(host, ".") {
|
|
||||||
if part == "" {
|
|
||||||
// Empty label.
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if isPattern && i == 0 && part == "*" {
|
|
||||||
// Only allow full left-most wildcards, as those are the only ones
|
|
||||||
// we match, and matching literal '*' characters is probably never
|
|
||||||
// the expected behavior.
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for j, c := range part {
|
|
||||||
if 'a' <= c && c <= 'z' {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if '0' <= c && c <= '9' {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if 'A' <= c && c <= 'Z' {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if c == '-' && j != 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if c == '_' {
|
|
||||||
// Not a valid character in hostnames, but commonly
|
|
||||||
// found in deployments outside the WebPKI.
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// commonNameAsHostname reports whether the Common Name field should be
|
|
||||||
// considered the hostname that the certificate is valid for. This is a legacy
|
|
||||||
// behavior, disabled by default or if the Subject Alt Name extension is present.
|
|
||||||
//
|
|
||||||
// It applies the strict validHostname check to the Common Name field, so that
|
|
||||||
// certificates without SANs can still be validated against CAs with name
|
|
||||||
// constraints if there is no risk the CN would be matched as a hostname.
|
|
||||||
// See NameConstraintsWithoutSANs and issue 24151.
|
|
||||||
func commonNameAsHostname(c *x509.Certificate) bool {
|
|
||||||
return !hasSANExtension(c) && validHostnamePattern(c.Subject.CommonName)
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchExactly(hostA, hostB string) bool {
|
|
||||||
if hostA == "" || hostA == "." || hostB == "" || hostB == "." {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return toLowerCaseASCII(hostA) == toLowerCaseASCII(hostB)
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchHostnames(pattern, host string) bool {
|
|
||||||
pattern = toLowerCaseASCII(pattern)
|
|
||||||
host = toLowerCaseASCII(strings.TrimSuffix(host, "."))
|
|
||||||
|
|
||||||
if len(pattern) == 0 || len(host) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
patternParts := strings.Split(pattern, ".")
|
|
||||||
hostParts := strings.Split(host, ".")
|
|
||||||
|
|
||||||
if len(patternParts) != len(hostParts) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, patternPart := range patternParts {
|
|
||||||
if i == 0 && patternPart == "*" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if patternPart != hostParts[i] {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// toLowerCaseASCII returns a lower-case version of in. See RFC 6125 6.4.1. We use
|
|
||||||
// an explicitly ASCII function to avoid any sharp corners resulting from
|
|
||||||
// performing Unicode operations on DNS labels.
|
|
||||||
func toLowerCaseASCII(in string) string {
|
|
||||||
// If the string is already lower-case then there's nothing to do.
|
|
||||||
isAlreadyLowerCase := true
|
|
||||||
for _, c := range in {
|
|
||||||
if c == utf8.RuneError {
|
|
||||||
// If we get a UTF-8 error then there might be
|
|
||||||
// upper-case ASCII bytes in the invalid sequence.
|
|
||||||
isAlreadyLowerCase = false
|
|
||||||
break
|
|
||||||
}
|
|
||||||
if 'A' <= c && c <= 'Z' {
|
|
||||||
isAlreadyLowerCase = false
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if isAlreadyLowerCase {
|
|
||||||
return in
|
|
||||||
}
|
|
||||||
|
|
||||||
out := []byte(in)
|
|
||||||
for i, c := range out {
|
|
||||||
if 'A' <= c && c <= 'Z' {
|
|
||||||
out[i] += 'a' - 'A'
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return string(out)
|
|
||||||
}
|
|
||||||
|
|
||||||
// verifyHostname returns nil if c is a valid certificate for the named host.
|
|
||||||
// Otherwise it returns an error describing the mismatch.
|
|
||||||
//
|
|
||||||
// IP addresses can be optionally enclosed in square brackets and are checked
|
|
||||||
// against the IPAddresses field. Other names are checked case insensitively
|
|
||||||
// against the DNSNames field. If the names are valid hostnames, the certificate
|
|
||||||
// fields can have a wildcard as the left-most label.
|
|
||||||
//
|
|
||||||
// The legacy Common Name field is ignored unless it's a valid hostname, the
|
|
||||||
// certificate doesn't have any Subject Alternative Names, and the GODEBUG
|
|
||||||
// environment variable is set to "x509ignoreCN=0". Support for Common Name is
|
|
||||||
// deprecated will be entirely removed in the future.
|
|
||||||
func verifyHostname(c *x509.Certificate, h string) error {
|
|
||||||
// IP addresses may be written in [ ].
|
|
||||||
candidateIP := h
|
|
||||||
if len(h) >= 3 && h[0] == '[' && h[len(h)-1] == ']' {
|
|
||||||
candidateIP = h[1 : len(h)-1]
|
|
||||||
}
|
|
||||||
if ip := net.ParseIP(candidateIP); ip != nil {
|
|
||||||
// We only match IP addresses against IP SANs.
|
|
||||||
// See RFC 6125, Appendix B.2.
|
|
||||||
for _, candidate := range c.IPAddresses {
|
|
||||||
if ip.Equal(candidate) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return x509.HostnameError{c, candidateIP}
|
|
||||||
}
|
|
||||||
|
|
||||||
names := c.DNSNames
|
|
||||||
if commonNameAsHostname(c) {
|
|
||||||
names = []string{c.Subject.CommonName}
|
|
||||||
}
|
|
||||||
|
|
||||||
candidateName := toLowerCaseASCII(h) // Save allocations inside the loop.
|
|
||||||
validCandidateName := validHostnameInput(candidateName)
|
|
||||||
|
|
||||||
for _, match := range names {
|
|
||||||
// Ideally, we'd only match valid hostnames according to RFC 6125 like
|
|
||||||
// browsers (more or less) do, but in practice Go is used in a wider
|
|
||||||
// array of contexts and can't even assume DNS resolution. Instead,
|
|
||||||
// always allow perfect matches, and only apply wildcard and trailing
|
|
||||||
// dot processing to valid hostnames.
|
|
||||||
if validCandidateName && validHostnamePattern(match) {
|
|
||||||
if matchHostnames(match, candidateName) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if matchExactly(match, candidateName) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return x509.HostnameError{c, h}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user