feature (HTTP2): support HTTP2 as a server

2026-01-03 06:17:35 +01:00 · 2019-05-12 02:33:10 +10:00 · 2019-05-12 02:33:10 +10:00 · 01d36f1601
commit 01d36f1601
parent bceeadecae
1 changed files with 121 additions and 0 deletions
--- a/server/plugin/plg_starter_http2/index.go
+++ b/server/plugin/plg_starter_http2/index.go
@ -0,0 +1,121 @@
+package plg_start_http2
+
+/*
+ * In golang, HTTP2 server are written in the same way as HTTPS server, the only difference beeing
+ * describe in the documentation: https://golang.org/src/net/http/doc.go#L81
+ * In our implementation, we use the 'TLSNextProto' trick
+ */
+
+import (
+	"crypto/tls"
+	"fmt"
+	"github.com/gorilla/mux"
+	. "github.com/mickael-kerjean/filestash/server/common"
+	"github.com/mickael-kerjean/filestash/server/common/ssl"
+	"golang.org/x/crypto/acme/autocert"
+	"net/http"
+	"path/filepath"
+	"time"
+)
+
+func init() {
+	domain := Config.Get("general.host").String()
+
+	Hooks.Register.Starter(func (r *mux.Router) {
+		Log.Info("[https] starting...%s", domain)
+		srv := &http.Server{
+			Addr: fmt.Sprintf(":https"),
+			Handler: r,
+			TLSConfig: &tls.Config{
+				MinVersion: tls.VersionTLS12,
+				CipherSuites: []uint16{
+					tls.TLS_ECDHE_ECDSA_WITH_AES_256_GCM_SHA384,
+					tls.TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384,
+					tls.TLS_ECDHE_ECDSA_WITH_CHACHA20_POLY1305,
+					tls.TLS_ECDHE_RSA_WITH_CHACHA20_POLY1305,
+					tls.TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256,
+					tls.TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256,
+				},
+				PreferServerCipherSuites: true,
+				CurvePreferences: []tls.CurveID{
+					tls.CurveP256,
+					tls.X25519,
+				},
+			},
+		}
+
+		switch domain {
+		case "":
+			TLSCert, roots, err := ssl.GenerateSelfSigned()
+			if err != nil {
+				return
+			}
+			srv.TLSConfig.Certificates = []tls.Certificate{ TLSCert }
+			HTTPClient.Transport.(*TransformedTransport).Orig.(*http.Transport).TLSClientConfig = &tls.Config{
+				RootCAs: roots,
+			}
+			HTTP.Transport.(*TransformedTransport).Orig.(*http.Transport).TLSClientConfig = &tls.Config{
+				RootCAs: roots,
+			}
+		default:
+			mngr := autocert.Manager{
+				Prompt:     autocert.AcceptTOS,
+				HostPolicy: autocert.HostWhitelist(domain),
+				Cache:      autocert.DirCache(filepath.Join(GetCurrentDir(), CERT_PATH)),
+			}
+			srv.TLSConfig.GetCertificate = mngr.GetCertificate
+		}
+
+		go func() {
+			if err := srv.ListenAndServeTLS("", ""); err != nil {
+				Log.Error("[https]: listen_serve %v", err)
+				return
+			}
+		}()
+		go func() {
+			srv := http.Server{
+				Addr: fmt.Sprintf(":http"),
+				ReadTimeout:  5 * time.Second,
+				WriteTimeout: 5 * time.Second,
+				Handler: http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+					w.Header().Set("Connection", "close")
+					http.Redirect(
+						w,
+						req,
+						"https://" + req.Host + req.URL.String(),
+						http.StatusMovedPermanently,
+					)
+				}),
+			}
+			if err := srv.ListenAndServe(); err != nil {
+				Log.Error("[https]: http_redirect %v", err)
+				return
+			}
+		}()
+		go ensureAppHasBooted("https://127.0.0.1/about", fmt.Sprintf("[https] started"))
+	})
+}
+
+
+func ensureAppHasBooted(address string, message string) {
+	i := 0
+	for {
+		if i > 10 {
+			Log.Warning("[https] no boot")
+			break
+		}
+		time.Sleep(250 * time.Millisecond)
+		res, err := HTTPClient.Get(address)
+		if err != nil {
+			i += 1
+			continue
+		}
+		res.Body.Close()
+		if res.StatusCode != http.StatusOK {
+			i += 1
+			continue
+		}
+		Log.Info(message)
+		break
+	}
+}