489 lines · 15.1 KB · Go
RawDownload
// Package http provides a registration interface for http servicespackage http import (	"bytes"	"context"	"crypto/tls"	"crypto/x509"	"errors"	"fmt"	"html/template"	"log"	"net"	"net/http"	"os"	"path/filepath"	"strings"	"sync"	"time" 	"github.com/go-chi/chi/v5"	"github.com/rclone/rclone/fs"	"github.com/rclone/rclone/fs/config/flags"	"github.com/rclone/rclone/lib/atexit"	"github.com/spf13/pflag") // Help returns text describing the http server to add to the command// help.func Help(prefix string) string {	help := `### Server options Use ` + "`--{{ .Prefix }}addr`" + ` to specify which IP address and port the server shouldlisten on, eg ` + "`--{{ .Prefix }}addr 1.2.3.4:8000` or `--{{ .Prefix }}addr :8080`" + ` to listen to allIPs.  By default it only listens on localhost.  You can use port:0 to let the OS choose an available port. If you set ` + "`--{{ .Prefix }}addr`" + ` to listen on a public or LAN accessible IP addressthen using Authentication is advised - see the next section for info. You can use a unix socket by setting the url to ` + "`unix:///path/to/socket`" + `or just by using an absolute path name. Note that unix sockets bypass theauthentication - this is expected to be done with file system permissions. ` + "`--{{ .Prefix }}addr`" + ` may be repeated to listen on multiple IPs/ports/sockets. ` + "`--{{ .Prefix }}server-read-timeout` and `--{{ .Prefix }}server-write-timeout`" + ` can be used tocontrol the timeouts on the server.  Note that this is the total timefor a transfer. ` + "`--{{ .Prefix }}max-header-bytes`" + ` controls the maximum number of bytes the server willaccept in the HTTP header. ` + "`--{{ .Prefix }}baseurl`" + ` controls the URL prefix that rclone serves from.  By defaultrclone will serve from the root.  If you used ` + "`--{{ .Prefix }}baseurl \"/rclone\"`" + ` thenrclone would serve from a URL starting with "/rclone/".  This isuseful if you wish to proxy rclone serve.  Rclone automaticallyinserts leading and trailing "/" on ` + "`--{{ .Prefix }}baseurl`" + `, so ` + "`--{{ .Prefix }}baseurl \"rclone\"`" + `,` + "`--{{ .Prefix }}baseurl \"/rclone\"` and `--{{ .Prefix }}baseurl \"/rclone/\"`" + ` are all treatedidentically. #### TLS (SSL) By default this will serve over http.  If you want you can serve overhttps.  You will need to supply the ` + "`--{{ .Prefix }}cert` and `--{{ .Prefix }}key`" + ` flags.If you wish to do client side certificate validation then you will need tosupply ` + "`--{{ .Prefix }}client-ca`" + ` also. ` + "`--{{ .Prefix }}cert`" + ` should be a either a PEM encoded certificate or a concatenationof that with the CA certificate.  ` + "`--k{{ .Prefix }}ey`" + ` should be the PEM encodedprivate key and ` + "`--{{ .Prefix }}client-ca`" + ` should be the PEM encoded clientcertificate authority certificate. --{{ .Prefix }}min-tls-version is minimum TLS version that is acceptable. Valid  values are "tls1.0", "tls1.1", "tls1.2" and "tls1.3" (default  "tls1.0").`	tmpl, err := template.New("server help").Parse(help)	if err != nil {		log.Fatal("Fatal error parsing template", err)	} 	data := struct {		Prefix string	}{		Prefix: prefix,	}	buf := &bytes.Buffer{}	err = tmpl.Execute(buf, data)	if err != nil {		log.Fatal("Fatal error executing template", err)	}	return buf.String()} // Middleware function signature required by chi.Router.Use()type Middleware func(http.Handler) http.Handler // Config contains options for the http Servertype Config struct {	ListenAddr         []string      // Port to listen on	BaseURL            string        // prefix to strip from URLs	ServerReadTimeout  time.Duration // Timeout for server reading data	ServerWriteTimeout time.Duration // Timeout for server writing data	MaxHeaderBytes     int           // Maximum size of request header	TLSCert            string        // Path to TLS PEM key (concatenation of certificate and CA certificate)	TLSKey             string        // Path to TLS PEM Private key	TLSCertBody        []byte        // TLS PEM key (concatenation of certificate and CA certificate) body, ignores TLSCert	TLSKeyBody         []byte        // TLS PEM Private key body, ignores TLSKey	ClientCA           string        // Client certificate authority to verify clients with	MinTLSVersion      string        // MinTLSVersion contains the minimum TLS version that is acceptable.	AllowOrigin        string        // AllowOrigin sets the Access-Control-Allow-Origin header} // AddFlagsPrefix adds flags for the httplibfunc (cfg *Config) AddFlagsPrefix(flagSet *pflag.FlagSet, prefix string) {	flags.StringArrayVarP(flagSet, &cfg.ListenAddr, prefix+"addr", "", cfg.ListenAddr, "IPaddress:Port or :Port to bind server to")	flags.DurationVarP(flagSet, &cfg.ServerReadTimeout, prefix+"server-read-timeout", "", cfg.ServerReadTimeout, "Timeout for server reading data")	flags.DurationVarP(flagSet, &cfg.ServerWriteTimeout, prefix+"server-write-timeout", "", cfg.ServerWriteTimeout, "Timeout for server writing data")	flags.IntVarP(flagSet, &cfg.MaxHeaderBytes, prefix+"max-header-bytes", "", cfg.MaxHeaderBytes, "Maximum size of request header")	flags.StringVarP(flagSet, &cfg.TLSCert, prefix+"cert", "", cfg.TLSCert, "TLS PEM key (concatenation of certificate and CA certificate)")	flags.StringVarP(flagSet, &cfg.TLSKey, prefix+"key", "", cfg.TLSKey, "TLS PEM Private key")	flags.StringVarP(flagSet, &cfg.ClientCA, prefix+"client-ca", "", cfg.ClientCA, "Client certificate authority to verify clients with")	flags.StringVarP(flagSet, &cfg.BaseURL, prefix+"baseurl", "", cfg.BaseURL, "Prefix for URLs - leave blank for root")	flags.StringVarP(flagSet, &cfg.MinTLSVersion, prefix+"min-tls-version", "", cfg.MinTLSVersion, "Minimum TLS version that is acceptable")	flags.StringVarP(flagSet, &cfg.AllowOrigin, prefix+"allow-origin", "", cfg.AllowOrigin, "Origin which cross-domain request (CORS) can be executed from")} // AddHTTPFlagsPrefix adds flags for the httplibfunc AddHTTPFlagsPrefix(flagSet *pflag.FlagSet, prefix string, cfg *Config) {	cfg.AddFlagsPrefix(flagSet, prefix)} // DefaultCfg is the default values used for Configfunc DefaultCfg() Config {	return Config{		ListenAddr:         []string{"127.0.0.1:8080"},		ServerReadTimeout:  1 * time.Hour,		ServerWriteTimeout: 1 * time.Hour,		MaxHeaderBytes:     4096,		MinTLSVersion:      "tls1.0",	}} type instance struct {	url        string	listener   net.Listener	httpServer *http.Server} func (s instance) serve(wg *sync.WaitGroup) {	defer wg.Done()	err := s.httpServer.Serve(s.listener)	if err != http.ErrServerClosed && err != nil {		log.Printf("%s: unexpected error: %s", s.listener.Addr(), err.Error())	}} // Server contains info about the running http servertype Server struct {	wg           sync.WaitGroup	mux          chi.Router	tlsConfig    *tls.Config	instances    []instance	auth         AuthConfig	cfg          Config	template     *TemplateConfig	htmlTemplate *template.Template	usingAuth    bool // set if we are using auth middleware	atexitHandle atexit.FnHandle} // Option allows customizing the servertype Option func(*Server) // WithAuth option initializes the appropriate auth middlewarefunc WithAuth(cfg AuthConfig) Option {	return func(s *Server) {		s.auth = cfg	}} // WithConfig option applies the Config to the server, overriding defaultsfunc WithConfig(cfg Config) Option {	return func(s *Server) {		s.cfg = cfg	}} // WithTemplate option allows the parsing of a templatefunc WithTemplate(cfg TemplateConfig) Option {	return func(s *Server) {		s.template = &cfg	}} // NewServer instantiates a new http server using provided listeners and options// This function is provided if the default http server does not meet a services requirements and should not generally be used// A http server can listen using multiple listeners. For example, a listener for port 80, and a listener for port 443.// tlsListeners are ignored if opt.TLSKey is not providedfunc NewServer(ctx context.Context, options ...Option) (*Server, error) {	s := &Server{		mux: chi.NewRouter(),		cfg: DefaultCfg(),	} 	// Make sure default logger is logging where everything else is	// middleware.DefaultLogger = middleware.RequestLogger(&middleware.DefaultLogFormatter{Logger: log.Default(), NoColor: true})	// Log requests	// s.mux.Use(middleware.Logger) 	for _, opt := range options {		opt(s)	} 	// Build base router	s.mux.MethodNotAllowed(func(w http.ResponseWriter, _ *http.Request) {		http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)	})	s.mux.NotFound(func(w http.ResponseWriter, _ *http.Request) {		http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)	}) 	// Ignore passing "/" for BaseURL	s.cfg.BaseURL = strings.Trim(s.cfg.BaseURL, "/")	if s.cfg.BaseURL != "" {		s.cfg.BaseURL = "/" + s.cfg.BaseURL		s.mux.Use(MiddlewareStripPrefix(s.cfg.BaseURL))	} 	err := s.initTemplate()	if err != nil {		return nil, err	} 	err = s.initTLS()	if err != nil {		return nil, err	} 	s.mux.Use(MiddlewareCORS(s.cfg.AllowOrigin)) 	// Put this one last for dumping requests / responses	ci := fs.GetConfig(context.Background())	dump := ci.Dump	if dump != 0 {		s.mux.Use(MiddlewareDump(dump))	} 	s.initAuth() 	for _, addr := range s.cfg.ListenAddr {		var url string		var network = "tcp"		var tlsCfg *tls.Config 		if strings.HasPrefix(addr, "unix://") || filepath.IsAbs(addr) {			network = "unix"			addr = strings.TrimPrefix(addr, "unix://")			url = addr 		} else if strings.HasPrefix(addr, "tls://") || (len(s.cfg.ListenAddr) == 1 && s.tlsConfig != nil) {			tlsCfg = s.tlsConfig			addr = strings.TrimPrefix(addr, "tls://")		} 		var listener net.Listener		if tlsCfg == nil {			listener, err = net.Listen(network, addr)		} else {			listener, err = tls.Listen(network, addr, tlsCfg)		}		if err != nil {			return nil, err		} 		if network == "tcp" {			var secure string			if tlsCfg != nil {				secure = "s"			}			url = fmt.Sprintf("http%s://%s%s/", secure, listener.Addr().String(), s.cfg.BaseURL)		} 		ii := instance{			url:      url,			listener: listener,			httpServer: &http.Server{				Handler:           s.mux,				ReadTimeout:       s.cfg.ServerReadTimeout,				WriteTimeout:      s.cfg.ServerWriteTimeout,				MaxHeaderBytes:    s.cfg.MaxHeaderBytes,				ReadHeaderTimeout: 10 * time.Second, // time to send the headers				IdleTimeout:       60 * time.Second, // time to keep idle connections open				TLSConfig:         tlsCfg,				BaseContext:       NewBaseContext(ctx, url),			},		} 		s.instances = append(s.instances, ii)	} 	return s, nil} func (s *Server) initAuth() {	s.usingAuth = false 	authCertificateUserEnabled := s.tlsConfig != nil && s.tlsConfig.ClientAuth != tls.NoClientCert && s.auth.HtPasswd == "" && s.auth.BasicUser == ""	if authCertificateUserEnabled {		s.usingAuth = true		s.mux.Use(MiddlewareAuthCertificateUser())	} 	if s.auth.CustomAuthFn != nil {		s.usingAuth = true		s.mux.Use(MiddlewareAuthCustom(s.auth.CustomAuthFn, s.auth.Realm, authCertificateUserEnabled))		return	} 	if s.auth.HtPasswd != "" {		s.usingAuth = true		s.mux.Use(MiddlewareAuthHtpasswd(s.auth.HtPasswd, s.auth.Realm))		return	} 	if s.auth.BasicUser != "" {		s.usingAuth = true		s.mux.Use(MiddlewareAuthBasic(s.auth.BasicUser, s.auth.BasicPass, s.auth.Realm, s.auth.Salt))		return	}} func (s *Server) initTemplate() error {	if s.template == nil {		return nil	} 	var err error	s.htmlTemplate, err = GetTemplate(s.template.Path)	if err != nil {		err = fmt.Errorf("failed to get template: %w", err)	} 	return err} var (	// ErrInvalidMinTLSVersion - hard coded errors, allowing for easier testing	ErrInvalidMinTLSVersion = errors.New("invalid value for --min-tls-version")	// ErrTLSBodyMismatch - hard coded errors, allowing for easier testing	ErrTLSBodyMismatch = errors.New("need both TLSCertBody and TLSKeyBody to use TLS")	// ErrTLSFileMismatch - hard coded errors, allowing for easier testing	ErrTLSFileMismatch = errors.New("need both --cert and --key to use TLS")	// ErrTLSParseCA - hard coded errors, allowing for easier testing	ErrTLSParseCA = errors.New("unable to parse client certificate authority")) func (s *Server) initTLS() error {	if s.cfg.TLSCert == "" && s.cfg.TLSKey == "" && len(s.cfg.TLSCertBody) == 0 && len(s.cfg.TLSKeyBody) == 0 {		return nil	} 	if (len(s.cfg.TLSCertBody) > 0) != (len(s.cfg.TLSKeyBody) > 0) {		return ErrTLSBodyMismatch	} 	if (s.cfg.TLSCert != "") != (s.cfg.TLSKey != "") {		return ErrTLSFileMismatch	} 	var cert tls.Certificate	var err error	if len(s.cfg.TLSCertBody) > 0 {		cert, err = tls.X509KeyPair(s.cfg.TLSCertBody, s.cfg.TLSKeyBody)	} else {		cert, err = tls.LoadX509KeyPair(s.cfg.TLSCert, s.cfg.TLSKey)	}	if err != nil {		return err	} 	var minTLSVersion uint16	switch s.cfg.MinTLSVersion {	case "tls1.0":		minTLSVersion = tls.VersionTLS10	case "tls1.1":		minTLSVersion = tls.VersionTLS11	case "tls1.2":		minTLSVersion = tls.VersionTLS12	case "tls1.3":		minTLSVersion = tls.VersionTLS13	default:		return fmt.Errorf("%w: %s", ErrInvalidMinTLSVersion, s.cfg.MinTLSVersion)	} 	s.tlsConfig = &tls.Config{		MinVersion:   minTLSVersion,		Certificates: []tls.Certificate{cert},	} 	if s.cfg.ClientCA != "" {		// if !useTLS {		// 	err := errors.New("can't use --client-ca without --cert and --key")		// 	log.Fatalf(err.Error())		// }		certpool := x509.NewCertPool()		pem, err := os.ReadFile(s.cfg.ClientCA)		if err != nil {			return err		} 		if !certpool.AppendCertsFromPEM(pem) {			return ErrTLSParseCA		} 		s.tlsConfig.ClientCAs = certpool		s.tlsConfig.ClientAuth = tls.RequireAndVerifyClientCert	} 	return nil} // Serve starts the HTTP server on each listenerfunc (s *Server) Serve() {	s.wg.Add(len(s.instances))	for _, ii := range s.instances {		// TODO: decide how/when to log listening url		// log.Printf("listening on %s", ii.url)		go ii.serve(&s.wg)	}	// Install an atexit handler to shutdown gracefully	s.atexitHandle = atexit.Register(func() { _ = s.Shutdown() })} // Wait blocks while the server is serving requestsfunc (s *Server) Wait() {	s.wg.Wait()} // Router returns the server base routerfunc (s *Server) Router() chi.Router {	return s.mux} // Time to wait to Shutdown an HTTP serverconst gracefulShutdownTime = 10 * time.Second // Shutdown gracefully shuts down the serverfunc (s *Server) Shutdown() error {	// Stop the atexit handler	if s.atexitHandle != nil {		atexit.Unregister(s.atexitHandle)		s.atexitHandle = nil	}	for _, ii := range s.instances {		expiry := time.Now().Add(gracefulShutdownTime)		ctx, cancel := context.WithDeadline(context.Background(), expiry)		if err := ii.httpServer.Shutdown(ctx); err != nil {			log.Printf("error shutting down server: %s", err)		}		cancel()	}	s.wg.Wait()	return nil} // HTMLTemplate returns the parsed template, if WithTemplate option was passed.func (s *Server) HTMLTemplate() *template.Template {	return s.htmlTemplate} // URLs returns all configured URLSfunc (s *Server) URLs() []string {	var out []string	for _, ii := range s.instances {		if ii.listener.Addr().Network() == "unix" {			continue		}		out = append(out, ii.url)	}	return out} // UsingAuth returns true if authentication is requiredfunc (s *Server) UsingAuth() bool {	return s.usingAuth}