115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package connection
|
|
|
|
import (
|
|
"fmt"
|
|
"hash/fnv"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/cloudflare/cloudflared/tunnelrpc/pogs"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const (
|
|
// edgeH2muxTLSServerName is the server name to establish h2mux connection with edge
|
|
edgeH2muxTLSServerName = "cftunnel.com"
|
|
// edgeH2TLSServerName is the server name to establish http2 connection with edge
|
|
edgeH2TLSServerName = "h2.cftunnel.com"
|
|
lbProbeUserAgentPrefix = "Mozilla/5.0 (compatible; Cloudflare-Traffic-Manager/1.0; +https://www.cloudflare.com/traffic-manager/;"
|
|
)
|
|
|
|
type Config struct {
|
|
OriginClient OriginClient
|
|
GracePeriod time.Duration
|
|
ReplaceExisting bool
|
|
}
|
|
|
|
type NamedTunnelConfig struct {
|
|
Auth pogs.TunnelAuth
|
|
ID uuid.UUID
|
|
Client pogs.ClientInfo
|
|
}
|
|
|
|
type ClassicTunnelConfig struct {
|
|
Hostname string
|
|
OriginCert []byte
|
|
// feature-flag to use new edge reconnect tokens
|
|
UseReconnectToken bool
|
|
}
|
|
|
|
func (c *ClassicTunnelConfig) IsTrialZone() bool {
|
|
return c.Hostname == ""
|
|
}
|
|
|
|
type Protocol int64
|
|
|
|
const (
|
|
H2mux Protocol = iota
|
|
HTTP2
|
|
)
|
|
|
|
func SelectProtocol(s string, accountTag string, http2Percentage uint32) (Protocol, bool) {
|
|
switch s {
|
|
case "h2mux":
|
|
return H2mux, true
|
|
case "http2":
|
|
return HTTP2, true
|
|
case "auto":
|
|
if tryHTTP2(accountTag, http2Percentage) {
|
|
return HTTP2, true
|
|
}
|
|
return H2mux, true
|
|
default:
|
|
return 0, false
|
|
}
|
|
}
|
|
|
|
func tryHTTP2(accountTag string, http2Percentage uint32) bool {
|
|
h := fnv.New32a()
|
|
h.Write([]byte(accountTag))
|
|
return h.Sum32()%100 < http2Percentage
|
|
}
|
|
|
|
func (p Protocol) ServerName() string {
|
|
switch p {
|
|
case H2mux:
|
|
return edgeH2muxTLSServerName
|
|
case HTTP2:
|
|
return edgeH2TLSServerName
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func (p Protocol) String() string {
|
|
switch p {
|
|
case H2mux:
|
|
return "h2mux"
|
|
case HTTP2:
|
|
return "http2"
|
|
default:
|
|
return fmt.Sprintf("unknown protocol")
|
|
}
|
|
}
|
|
|
|
type OriginClient interface {
|
|
Proxy(w ResponseWriter, req *http.Request, isWebsocket bool) error
|
|
}
|
|
|
|
type ResponseWriter interface {
|
|
WriteRespHeaders(*http.Response) error
|
|
WriteErrorResponse(error)
|
|
io.ReadWriter
|
|
}
|
|
|
|
type ConnectedFuse interface {
|
|
Connected()
|
|
IsConnected() bool
|
|
}
|
|
|
|
func uint8ToString(input uint8) string {
|
|
return strconv.FormatUint(uint64(input), 10)
|
|
}
|