Initial commit (code only without large binaries)

EdgeNode/internal/http3/server.go

@@ -0,0 +1,86 @@
+// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package http3
+
+import (
+	"context"
+	"errors"
+	"github.com/quic-go/quic-go"
+	http3quic "github.com/quic-go/quic-go/http3"
+	"net"
+	"net/http"
+)
+
+type Server struct {
+	Addr        string
+	Handler     http.Handler
+	ConnState   func(conn net.Conn, state http.ConnState)
+	ConnContext ContextFunc
+
+	rawServer *http3quic.Server
+	isClosed  bool
+
+	listener Listener
+}
+
+func (this *Server) Serve(listener Listener) error {
+	if listener == nil {
+		return errors.New("listener must not be nil")
+	}
+
+	this.listener = listener
+
+	this.rawServer = &http3quic.Server{
+		Addr:    this.Addr,
+		Handler: this.Handler,
+	}
+
+	for {
+		conn, err := listener.Accept(context.Background(), this.ConnContext)
+		if err != nil {
+			if this.isClosed {
+				return nil
+			}
+			continue
+		}
+		go func() {
+			// 通知ConnState
+			if this.ConnState != nil {
+				netConn, isNetConn := conn.(net.Conn)
+				if isNetConn {
+					this.ConnState(netConn, http.StateNew)
+					this.ConnState(netConn, http.StateActive)
+				}
+			}
+
+			_ = this.rawServer.ServeQUICConn(conn)
+
+			// 关闭连接
+			_ = conn.CloseWithError(quic.ApplicationErrorCode(quic.NoError), "")
+
+			// 通知ConnState
+			if this.ConnState != nil {
+				netConn, isNetConn := conn.(net.Conn)
+				if isNetConn {
+					this.ConnState(netConn, http.StateClosed)
+				}
+			}
+		}()
+	}
+}
+
+func (this *Server) SetQuicHeaders(headers http.Header) {
+	if this.isClosed {
+		return
+	}
+	headers.Set("Alt-Svc", `h3="`+this.Addr+`"; ma=2592000,h3-29="`+this.Addr+`"; ma=2592000`)
+}
+
+func (this *Server) Close() error {
+	this.isClosed = true
+
+	_ = this.listener.Close()
+
+	return this.rawServer.Close()
+}