906 lines
27 KiB
Go
906 lines
27 KiB
Go
package origin
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/pkg/errors"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/cloudflare/cloudflared/buffer"
|
|
"github.com/cloudflare/cloudflared/cmd/cloudflared/buildinfo"
|
|
"github.com/cloudflare/cloudflared/cmd/cloudflared/ui"
|
|
"github.com/cloudflare/cloudflared/connection"
|
|
"github.com/cloudflare/cloudflared/h2mux"
|
|
"github.com/cloudflare/cloudflared/logger"
|
|
"github.com/cloudflare/cloudflared/signal"
|
|
"github.com/cloudflare/cloudflared/tunnelrpc"
|
|
"github.com/cloudflare/cloudflared/tunnelrpc/pogs"
|
|
tunnelpogs "github.com/cloudflare/cloudflared/tunnelrpc/pogs"
|
|
"github.com/cloudflare/cloudflared/validation"
|
|
"github.com/cloudflare/cloudflared/websocket"
|
|
)
|
|
|
|
const (
|
|
dialTimeout = 15 * time.Second
|
|
openStreamTimeout = 30 * time.Second
|
|
muxerTimeout = 5 * time.Second
|
|
lbProbeUserAgentPrefix = "Mozilla/5.0 (compatible; Cloudflare-Traffic-Manager/1.0; +https://www.cloudflare.com/traffic-manager/;"
|
|
TagHeaderNamePrefix = "Cf-Warp-Tag-"
|
|
DuplicateConnectionError = "EDUPCONN"
|
|
FeatureSerializedHeaders = "serialized_headers"
|
|
FeatureQuickReconnects = "quick_reconnects"
|
|
)
|
|
|
|
type registerRPCName string
|
|
|
|
const (
|
|
register registerRPCName = "register"
|
|
reconnect registerRPCName = "reconnect"
|
|
)
|
|
|
|
type TunnelConfig struct {
|
|
BuildInfo *buildinfo.BuildInfo
|
|
ClientID string
|
|
ClientTlsConfig *tls.Config
|
|
CloseConnOnce *sync.Once // Used to close connectedSignal no more than once
|
|
CompressionQuality uint64
|
|
EdgeAddrs []string
|
|
GracePeriod time.Duration
|
|
HAConnections int
|
|
HTTPTransport http.RoundTripper
|
|
HeartbeatInterval time.Duration
|
|
Hostname string
|
|
HTTPHostHeader string
|
|
IncidentLookup IncidentLookup
|
|
IsAutoupdated bool
|
|
IsFreeTunnel bool
|
|
LBPool string
|
|
Logger logger.Service
|
|
TransportLogger logger.Service
|
|
MaxHeartbeats uint64
|
|
Metrics *TunnelMetrics
|
|
MetricsUpdateFreq time.Duration
|
|
NoChunkedEncoding bool
|
|
OriginCert []byte
|
|
ReportedVersion string
|
|
Retries uint
|
|
RunFromTerminal bool
|
|
Tags []tunnelpogs.Tag
|
|
TlsConfig *tls.Config
|
|
WSGI bool
|
|
// OriginUrl may not be used if a user specifies a unix socket.
|
|
OriginUrl string
|
|
|
|
// feature-flag to use new edge reconnect tokens
|
|
UseReconnectToken bool
|
|
|
|
NamedTunnel *NamedTunnelConfig
|
|
ReplaceExisting bool
|
|
TunnelEventChan chan<- ui.TunnelEvent
|
|
}
|
|
|
|
type dupConnRegisterTunnelError struct{}
|
|
|
|
var errDuplicationConnection = &dupConnRegisterTunnelError{}
|
|
|
|
func (e dupConnRegisterTunnelError) Error() string {
|
|
return "already connected to this server, trying another address"
|
|
}
|
|
|
|
type muxerShutdownError struct{}
|
|
|
|
func (e muxerShutdownError) Error() string {
|
|
return "muxer shutdown"
|
|
}
|
|
|
|
// RegisterTunnel error from server
|
|
type serverRegisterTunnelError struct {
|
|
cause error
|
|
permanent bool
|
|
}
|
|
|
|
func (e serverRegisterTunnelError) Error() string {
|
|
return e.cause.Error()
|
|
}
|
|
|
|
// RegisterTunnel error from client
|
|
type clientRegisterTunnelError struct {
|
|
cause error
|
|
}
|
|
|
|
func newClientRegisterTunnelError(cause error, counter *prometheus.CounterVec, name registerRPCName) clientRegisterTunnelError {
|
|
counter.WithLabelValues(cause.Error(), string(name)).Inc()
|
|
return clientRegisterTunnelError{cause: cause}
|
|
}
|
|
|
|
func (e clientRegisterTunnelError) Error() string {
|
|
return e.cause.Error()
|
|
}
|
|
|
|
func (c *TunnelConfig) muxerConfig(handler h2mux.MuxedStreamHandler) h2mux.MuxerConfig {
|
|
return h2mux.MuxerConfig{
|
|
Timeout: muxerTimeout,
|
|
Handler: handler,
|
|
IsClient: true,
|
|
HeartbeatInterval: c.HeartbeatInterval,
|
|
MaxHeartbeats: c.MaxHeartbeats,
|
|
Logger: c.TransportLogger,
|
|
CompressionQuality: h2mux.CompressionSetting(c.CompressionQuality),
|
|
}
|
|
}
|
|
|
|
func (c *TunnelConfig) RegistrationOptions(connectionID uint8, OriginLocalIP string, uuid uuid.UUID) *tunnelpogs.RegistrationOptions {
|
|
policy := tunnelrpc.ExistingTunnelPolicy_balance
|
|
if c.HAConnections <= 1 && c.LBPool == "" {
|
|
policy = tunnelrpc.ExistingTunnelPolicy_disconnect
|
|
}
|
|
return &tunnelpogs.RegistrationOptions{
|
|
ClientID: c.ClientID,
|
|
Version: c.ReportedVersion,
|
|
OS: fmt.Sprintf("%s_%s", c.BuildInfo.GoOS, c.BuildInfo.GoArch),
|
|
ExistingTunnelPolicy: policy,
|
|
PoolName: c.LBPool,
|
|
Tags: c.Tags,
|
|
ConnectionID: connectionID,
|
|
OriginLocalIP: OriginLocalIP,
|
|
IsAutoupdated: c.IsAutoupdated,
|
|
RunFromTerminal: c.RunFromTerminal,
|
|
CompressionQuality: c.CompressionQuality,
|
|
UUID: uuid.String(),
|
|
Features: c.SupportedFeatures(),
|
|
}
|
|
}
|
|
|
|
func (c *TunnelConfig) ConnectionOptions(originLocalAddr string, numPreviousAttempts uint8) *tunnelpogs.ConnectionOptions {
|
|
// attempt to parse out origin IP, but don't fail since it's informational field
|
|
host, _, _ := net.SplitHostPort(originLocalAddr)
|
|
originIP := net.ParseIP(host)
|
|
|
|
return &tunnelpogs.ConnectionOptions{
|
|
Client: c.NamedTunnel.Client,
|
|
OriginLocalIP: originIP,
|
|
ReplaceExisting: c.ReplaceExisting,
|
|
CompressionQuality: uint8(c.CompressionQuality),
|
|
NumPreviousAttempts: numPreviousAttempts,
|
|
}
|
|
}
|
|
|
|
func (c *TunnelConfig) SupportedFeatures() []string {
|
|
features := []string{FeatureSerializedHeaders}
|
|
if c.NamedTunnel == nil {
|
|
features = append(features, FeatureQuickReconnects)
|
|
}
|
|
return features
|
|
}
|
|
|
|
type NamedTunnelConfig struct {
|
|
Auth pogs.TunnelAuth
|
|
ID uuid.UUID
|
|
Client pogs.ClientInfo
|
|
}
|
|
|
|
func StartTunnelDaemon(ctx context.Context, config *TunnelConfig, connectedSignal *signal.Signal, cloudflaredID uuid.UUID, reconnectCh chan ReconnectSignal) error {
|
|
s, err := NewSupervisor(config, cloudflaredID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.Run(ctx, connectedSignal, reconnectCh)
|
|
}
|
|
|
|
func ServeTunnelLoop(ctx context.Context,
|
|
credentialManager *reconnectCredentialManager,
|
|
config *TunnelConfig,
|
|
addr *net.TCPAddr,
|
|
connectionIndex uint8,
|
|
connectedSignal *signal.Signal,
|
|
cloudflaredUUID uuid.UUID,
|
|
bufferPool *buffer.Pool,
|
|
reconnectCh chan ReconnectSignal,
|
|
) error {
|
|
config.Metrics.incrementHaConnections()
|
|
defer config.Metrics.decrementHaConnections()
|
|
backoff := BackoffHandler{MaxRetries: config.Retries}
|
|
connectedFuse := h2mux.NewBooleanFuse()
|
|
go func() {
|
|
if connectedFuse.Await() {
|
|
connectedSignal.Notify()
|
|
}
|
|
}()
|
|
// Ensure the above goroutine will terminate if we return without connecting
|
|
defer connectedFuse.Fuse(false)
|
|
for {
|
|
err, recoverable := ServeTunnel(
|
|
ctx,
|
|
credentialManager,
|
|
config,
|
|
config.Logger,
|
|
addr, connectionIndex,
|
|
connectedFuse,
|
|
&backoff,
|
|
cloudflaredUUID,
|
|
bufferPool,
|
|
reconnectCh,
|
|
)
|
|
if recoverable {
|
|
if duration, ok := backoff.GetBackoffDuration(ctx); ok {
|
|
if config.TunnelEventChan != nil {
|
|
config.TunnelEventChan <- ui.TunnelEvent{Index: connectionIndex, EventType: ui.Reconnecting}
|
|
}
|
|
|
|
config.Logger.Infof("Retrying connection %d in %s seconds", connectionIndex, duration)
|
|
backoff.Backoff(ctx)
|
|
continue
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
|
|
func ServeTunnel(
|
|
ctx context.Context,
|
|
credentialManager *reconnectCredentialManager,
|
|
config *TunnelConfig,
|
|
logger logger.Service,
|
|
addr *net.TCPAddr,
|
|
connectionIndex uint8,
|
|
connectedFuse *h2mux.BooleanFuse,
|
|
backoff *BackoffHandler,
|
|
cloudflaredUUID uuid.UUID,
|
|
bufferPool *buffer.Pool,
|
|
reconnectCh chan ReconnectSignal,
|
|
) (err error, recoverable bool) {
|
|
// Treat panics as recoverable errors
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
var ok bool
|
|
err, ok = r.(error)
|
|
if !ok {
|
|
err = fmt.Errorf("ServeTunnel: %v", r)
|
|
}
|
|
recoverable = true
|
|
}
|
|
}()
|
|
|
|
// If launch-ui flag is set, send disconnect msg
|
|
if config.TunnelEventChan != nil {
|
|
defer func() {
|
|
config.TunnelEventChan <- ui.TunnelEvent{Index: connectionIndex, EventType: ui.Disconnected}
|
|
}()
|
|
}
|
|
|
|
connectionTag := uint8ToString(connectionIndex)
|
|
|
|
// Returns error from parsing the origin URL or handshake errors
|
|
handler, originLocalAddr, err := NewTunnelHandler(ctx, config, addr, connectionIndex, bufferPool)
|
|
if err != nil {
|
|
switch err.(type) {
|
|
case connection.DialError:
|
|
logger.Errorf("Connection %d unable to dial edge: %s", connectionIndex, err)
|
|
case h2mux.MuxerHandshakeError:
|
|
logger.Errorf("Connection %d handshake with edge server failed: %s", connectionIndex, err)
|
|
default:
|
|
logger.Errorf("Connection %d failed: %s", connectionIndex, err)
|
|
return err, false
|
|
}
|
|
return err, true
|
|
}
|
|
|
|
errGroup, serveCtx := errgroup.WithContext(ctx)
|
|
|
|
errGroup.Go(func() (err error) {
|
|
defer func() {
|
|
if err == nil {
|
|
connectedFuse.Fuse(true)
|
|
backoff.SetGracePeriod()
|
|
}
|
|
}()
|
|
|
|
if config.NamedTunnel != nil {
|
|
return RegisterConnection(ctx, handler.muxer, config, connectionIndex, originLocalAddr, uint8(backoff.retries))
|
|
}
|
|
|
|
if config.UseReconnectToken && connectedFuse.Value() {
|
|
err := ReconnectTunnel(serveCtx, handler.muxer, config, logger, connectionIndex, originLocalAddr, cloudflaredUUID, credentialManager)
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
// log errors and proceed to RegisterTunnel
|
|
logger.Errorf("Couldn't reconnect connection %d. Reregistering it instead. Error was: %v", connectionIndex, err)
|
|
}
|
|
return RegisterTunnel(serveCtx, credentialManager, handler.muxer, config, logger, connectionIndex, originLocalAddr, cloudflaredUUID)
|
|
})
|
|
|
|
errGroup.Go(func() error {
|
|
updateMetricsTickC := time.Tick(config.MetricsUpdateFreq)
|
|
for {
|
|
select {
|
|
case <-serveCtx.Done():
|
|
// UnregisterTunnel blocks until the RPC call returns
|
|
if connectedFuse.Value() {
|
|
if config.NamedTunnel != nil {
|
|
_ = UnregisterConnection(ctx, handler.muxer, config)
|
|
} else {
|
|
_ = UnregisterTunnel(handler.muxer, config.GracePeriod, config.TransportLogger)
|
|
}
|
|
}
|
|
handler.muxer.Shutdown()
|
|
return nil
|
|
case <-updateMetricsTickC:
|
|
handler.UpdateMetrics(connectionTag)
|
|
}
|
|
}
|
|
})
|
|
|
|
errGroup.Go(func() error {
|
|
for {
|
|
select {
|
|
case reconnect := <-reconnectCh:
|
|
return &reconnect
|
|
case <-serveCtx.Done():
|
|
return nil
|
|
}
|
|
}
|
|
})
|
|
|
|
errGroup.Go(func() error {
|
|
// All routines should stop when muxer finish serving. When muxer is shutdown
|
|
// gracefully, it doesn't return an error, so we need to return errMuxerShutdown
|
|
// here to notify other routines to stop
|
|
err := handler.muxer.Serve(serveCtx)
|
|
if err == nil {
|
|
return muxerShutdownError{}
|
|
}
|
|
return err
|
|
})
|
|
|
|
err = errGroup.Wait()
|
|
if err != nil {
|
|
switch err := err.(type) {
|
|
case *dupConnRegisterTunnelError:
|
|
// don't retry this connection anymore, let supervisor pick new a address
|
|
return err, false
|
|
case *serverRegisterTunnelError:
|
|
logger.Errorf("Register tunnel error from server side: %s", err.cause)
|
|
// Don't send registration error return from server to Sentry. They are
|
|
// logged on server side
|
|
if incidents := config.IncidentLookup.ActiveIncidents(); len(incidents) > 0 {
|
|
logger.Error(activeIncidentsMsg(incidents))
|
|
}
|
|
return err.cause, !err.permanent
|
|
case *clientRegisterTunnelError:
|
|
logger.Errorf("Register tunnel error on client side: %s", err.cause)
|
|
return err, true
|
|
case *muxerShutdownError:
|
|
logger.Info("Muxer shutdown")
|
|
return err, true
|
|
case *ReconnectSignal:
|
|
logger.Infof("Restarting connection %d due to reconnect signal in %d seconds", connectionIndex, err.Delay)
|
|
err.DelayBeforeReconnect()
|
|
return err, true
|
|
default:
|
|
if err == context.Canceled {
|
|
logger.Debugf("Serve tunnel error: %s", err)
|
|
return err, false
|
|
}
|
|
logger.Errorf("Serve tunnel error: %s", err)
|
|
return err, true
|
|
}
|
|
}
|
|
return nil, true
|
|
}
|
|
|
|
func RegisterConnection(
|
|
ctx context.Context,
|
|
muxer *h2mux.Muxer,
|
|
config *TunnelConfig,
|
|
connectionIndex uint8,
|
|
originLocalAddr string,
|
|
numPreviousAttempts uint8,
|
|
) error {
|
|
const registerConnection = "registerConnection"
|
|
|
|
config.TransportLogger.Debug("initiating RPC stream for RegisterConnection")
|
|
rpc, err := connection.NewRPCClient(ctx, muxer, config.TransportLogger, openStreamTimeout)
|
|
if err != nil {
|
|
// RPC stream open error
|
|
return newClientRegisterTunnelError(err, config.Metrics.rpcFail, registerConnection)
|
|
}
|
|
defer rpc.Close()
|
|
|
|
conn, err := rpc.RegisterConnection(
|
|
ctx,
|
|
config.NamedTunnel.Auth,
|
|
config.NamedTunnel.ID,
|
|
connectionIndex,
|
|
config.ConnectionOptions(originLocalAddr, numPreviousAttempts),
|
|
)
|
|
if err != nil {
|
|
if err.Error() == DuplicateConnectionError {
|
|
config.Metrics.regFail.WithLabelValues("dup_edge_conn", registerConnection).Inc()
|
|
return errDuplicationConnection
|
|
}
|
|
config.Metrics.regFail.WithLabelValues("server_error", registerConnection).Inc()
|
|
return serverRegistrationErrorFromRPC(err)
|
|
}
|
|
|
|
config.Metrics.regSuccess.WithLabelValues(registerConnection).Inc()
|
|
config.Logger.Infof("Connection %d registered with %s using ID %s", connectionIndex, conn.Location, conn.UUID)
|
|
|
|
// If launch-ui flag is set, send connect msg
|
|
if config.TunnelEventChan != nil {
|
|
config.TunnelEventChan <- ui.TunnelEvent{Index: connectionIndex, EventType: ui.Connected, Location: conn.Location}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func serverRegistrationErrorFromRPC(err error) *serverRegisterTunnelError {
|
|
if retryable, ok := err.(*tunnelpogs.RetryableError); ok {
|
|
return &serverRegisterTunnelError{
|
|
cause: retryable.Unwrap(),
|
|
permanent: false,
|
|
}
|
|
}
|
|
return &serverRegisterTunnelError{
|
|
cause: err,
|
|
permanent: true,
|
|
}
|
|
}
|
|
|
|
func UnregisterConnection(
|
|
ctx context.Context,
|
|
muxer *h2mux.Muxer,
|
|
config *TunnelConfig,
|
|
) error {
|
|
config.TransportLogger.Debug("initiating RPC stream for UnregisterConnection")
|
|
rpc, err := connection.NewRPCClient(ctx, muxer, config.TransportLogger, openStreamTimeout)
|
|
if err != nil {
|
|
// RPC stream open error
|
|
return newClientRegisterTunnelError(err, config.Metrics.rpcFail, register)
|
|
}
|
|
defer rpc.Close()
|
|
|
|
return rpc.UnregisterConnection(ctx)
|
|
}
|
|
|
|
func RegisterTunnel(
|
|
ctx context.Context,
|
|
credentialManager *reconnectCredentialManager,
|
|
muxer *h2mux.Muxer,
|
|
config *TunnelConfig,
|
|
logger logger.Service,
|
|
connectionID uint8,
|
|
originLocalIP string,
|
|
uuid uuid.UUID,
|
|
) error {
|
|
config.TransportLogger.Debug("initiating RPC stream to register")
|
|
if config.TunnelEventChan != nil {
|
|
config.TunnelEventChan <- ui.TunnelEvent{EventType: ui.RegisteringTunnel}
|
|
}
|
|
tunnelServer, err := connection.NewRPCClient(ctx, muxer, config.TransportLogger, openStreamTimeout)
|
|
if err != nil {
|
|
// RPC stream open error
|
|
return newClientRegisterTunnelError(err, config.Metrics.rpcFail, register)
|
|
}
|
|
defer tunnelServer.Close()
|
|
// Request server info without blocking tunnel registration; must use capnp library directly.
|
|
serverInfoPromise := tunnelrpc.TunnelServer{Client: tunnelServer.Client}.GetServerInfo(ctx, func(tunnelrpc.TunnelServer_getServerInfo_Params) error {
|
|
return nil
|
|
})
|
|
LogServerInfo(serverInfoPromise.Result(), connectionID, config.Metrics, logger, config.TunnelEventChan)
|
|
registration := tunnelServer.RegisterTunnel(
|
|
ctx,
|
|
config.OriginCert,
|
|
config.Hostname,
|
|
config.RegistrationOptions(connectionID, originLocalIP, uuid),
|
|
)
|
|
if registrationErr := registration.DeserializeError(); registrationErr != nil {
|
|
// RegisterTunnel RPC failure
|
|
return processRegisterTunnelError(registrationErr, config.Metrics, register)
|
|
}
|
|
|
|
// Send free tunnel URL to UI
|
|
if config.TunnelEventChan != nil {
|
|
config.TunnelEventChan <- ui.TunnelEvent{EventType: ui.SetUrl, Url: registration.Url}
|
|
}
|
|
credentialManager.SetEventDigest(connectionID, registration.EventDigest)
|
|
return processRegistrationSuccess(config, logger, connectionID, registration, register, credentialManager)
|
|
}
|
|
|
|
func processRegistrationSuccess(
|
|
config *TunnelConfig,
|
|
logger logger.Service,
|
|
connectionID uint8,
|
|
registration *tunnelpogs.TunnelRegistration,
|
|
name registerRPCName,
|
|
credentialManager *reconnectCredentialManager,
|
|
) error {
|
|
for _, logLine := range registration.LogLines {
|
|
logger.Info(logLine)
|
|
}
|
|
|
|
if registration.TunnelID != "" {
|
|
config.Metrics.tunnelsHA.AddTunnelID(connectionID, registration.TunnelID)
|
|
logger.Infof("Each HA connection's tunnel IDs: %v", config.Metrics.tunnelsHA.String())
|
|
}
|
|
|
|
// Print out the user's trial zone URL in a nice box (if they requested and got one)
|
|
if isTrialTunnel := config.Hostname == ""; isTrialTunnel {
|
|
if url, err := url.Parse(registration.Url); err == nil {
|
|
for _, line := range asciiBox(trialZoneMsg(url.String()), 2) {
|
|
logger.Info(line)
|
|
}
|
|
} else {
|
|
logger.Error("Failed to connect tunnel, please try again.")
|
|
return fmt.Errorf("empty URL in response from Cloudflare edge")
|
|
}
|
|
}
|
|
|
|
credentialManager.SetConnDigest(connectionID, registration.ConnDigest)
|
|
config.Metrics.userHostnamesCounts.WithLabelValues(registration.Url).Inc()
|
|
|
|
logger.Infof("Route propagating, it may take up to 1 minute for your new route to become functional")
|
|
config.Metrics.regSuccess.WithLabelValues(string(name)).Inc()
|
|
return nil
|
|
}
|
|
|
|
func processRegisterTunnelError(err tunnelpogs.TunnelRegistrationError, metrics *TunnelMetrics, name registerRPCName) error {
|
|
if err.Error() == DuplicateConnectionError {
|
|
metrics.regFail.WithLabelValues("dup_edge_conn", string(name)).Inc()
|
|
return errDuplicationConnection
|
|
}
|
|
metrics.regFail.WithLabelValues("server_error", string(name)).Inc()
|
|
return serverRegisterTunnelError{
|
|
cause: err,
|
|
permanent: err.IsPermanent(),
|
|
}
|
|
}
|
|
|
|
func UnregisterTunnel(muxer *h2mux.Muxer, gracePeriod time.Duration, logger logger.Service) error {
|
|
logger.Debug("initiating RPC stream to unregister")
|
|
ctx := context.Background()
|
|
tunnelServer, err := connection.NewRPCClient(ctx, muxer, logger, openStreamTimeout)
|
|
if err != nil {
|
|
// RPC stream open error
|
|
return err
|
|
}
|
|
defer tunnelServer.Close()
|
|
|
|
// gracePeriod is encoded in int64 using capnproto
|
|
return tunnelServer.UnregisterTunnel(ctx, gracePeriod.Nanoseconds())
|
|
}
|
|
|
|
func LogServerInfo(
|
|
promise tunnelrpc.ServerInfo_Promise,
|
|
connectionID uint8,
|
|
metrics *TunnelMetrics,
|
|
logger logger.Service,
|
|
tunnelEventChan chan<- ui.TunnelEvent,
|
|
) {
|
|
serverInfoMessage, err := promise.Struct()
|
|
if err != nil {
|
|
logger.Errorf("Failed to retrieve server information: %s", err)
|
|
return
|
|
}
|
|
serverInfo, err := tunnelpogs.UnmarshalServerInfo(serverInfoMessage)
|
|
if err != nil {
|
|
logger.Errorf("Failed to retrieve server information: %s", err)
|
|
return
|
|
}
|
|
// If launch-ui flag is set, send connect msg
|
|
if tunnelEventChan != nil {
|
|
tunnelEventChan <- ui.TunnelEvent{Index: connectionID, EventType: ui.Connected, Location: serverInfo.LocationName}
|
|
}
|
|
logger.Infof("Connected to %s", serverInfo.LocationName)
|
|
metrics.registerServerLocation(uint8ToString(connectionID), serverInfo.LocationName)
|
|
}
|
|
|
|
type TunnelHandler struct {
|
|
originUrl string
|
|
httpHostHeader string
|
|
muxer *h2mux.Muxer
|
|
httpClient http.RoundTripper
|
|
tlsConfig *tls.Config
|
|
tags []tunnelpogs.Tag
|
|
metrics *TunnelMetrics
|
|
// connectionID is only used by metrics, and prometheus requires labels to be string
|
|
connectionID string
|
|
logger logger.Service
|
|
noChunkedEncoding bool
|
|
|
|
bufferPool *buffer.Pool
|
|
}
|
|
|
|
// NewTunnelHandler returns a TunnelHandler, origin LAN IP and error
|
|
func NewTunnelHandler(ctx context.Context,
|
|
config *TunnelConfig,
|
|
addr *net.TCPAddr,
|
|
connectionID uint8,
|
|
bufferPool *buffer.Pool,
|
|
) (*TunnelHandler, string, error) {
|
|
originURL, err := validation.ValidateUrl(config.OriginUrl)
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("unable to parse origin URL %#v", originURL)
|
|
}
|
|
h := &TunnelHandler{
|
|
originUrl: originURL,
|
|
httpHostHeader: config.HTTPHostHeader,
|
|
httpClient: config.HTTPTransport,
|
|
tlsConfig: config.ClientTlsConfig,
|
|
tags: config.Tags,
|
|
metrics: config.Metrics,
|
|
connectionID: uint8ToString(connectionID),
|
|
logger: config.Logger,
|
|
noChunkedEncoding: config.NoChunkedEncoding,
|
|
bufferPool: bufferPool,
|
|
}
|
|
if h.httpClient == nil {
|
|
h.httpClient = http.DefaultTransport
|
|
}
|
|
|
|
edgeConn, err := connection.DialEdge(ctx, dialTimeout, config.TlsConfig, addr)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
// Establish a muxed connection with the edge
|
|
// Client mux handshake with agent server
|
|
h.muxer, err = h2mux.Handshake(edgeConn, edgeConn, config.muxerConfig(h), h.metrics.activeStreams)
|
|
if err != nil {
|
|
return nil, "", errors.Wrap(err, "h2mux handshake with edge error")
|
|
}
|
|
return h, edgeConn.LocalAddr().String(), nil
|
|
}
|
|
|
|
func (h *TunnelHandler) AppendTagHeaders(r *http.Request) {
|
|
for _, tag := range h.tags {
|
|
r.Header.Add(TagHeaderNamePrefix+tag.Name, tag.Value)
|
|
}
|
|
}
|
|
|
|
func (h *TunnelHandler) ServeStream(stream *h2mux.MuxedStream) error {
|
|
h.metrics.incrementRequests(h.connectionID)
|
|
defer h.metrics.decrementConcurrentRequests(h.connectionID)
|
|
|
|
req, reqErr := h.createRequest(stream)
|
|
if reqErr != nil {
|
|
h.writeErrorResponse(stream, reqErr)
|
|
return reqErr
|
|
}
|
|
|
|
cfRay := findCfRayHeader(req)
|
|
lbProbe := isLBProbeRequest(req)
|
|
h.logRequest(req, cfRay, lbProbe)
|
|
|
|
var resp *http.Response
|
|
var respErr error
|
|
if websocket.IsWebSocketUpgrade(req) {
|
|
resp, respErr = h.serveWebsocket(stream, req)
|
|
} else {
|
|
resp, respErr = h.serveHTTP(stream, req)
|
|
}
|
|
if respErr != nil {
|
|
h.writeErrorResponse(stream, respErr)
|
|
return respErr
|
|
}
|
|
h.logResponseOk(resp, cfRay, lbProbe)
|
|
return nil
|
|
}
|
|
|
|
func (h *TunnelHandler) createRequest(stream *h2mux.MuxedStream) (*http.Request, error) {
|
|
req, err := http.NewRequest("GET", h.originUrl, h2mux.MuxedStreamReader{MuxedStream: stream})
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Unexpected error from http.NewRequest")
|
|
}
|
|
err = h2mux.H2RequestHeadersToH1Request(stream.Headers, req)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "invalid request received")
|
|
}
|
|
h.AppendTagHeaders(req)
|
|
return req, nil
|
|
}
|
|
|
|
func (h *TunnelHandler) serveWebsocket(stream *h2mux.MuxedStream, req *http.Request) (*http.Response, error) {
|
|
if h.httpHostHeader != "" {
|
|
req.Header.Set("Host", h.httpHostHeader)
|
|
req.Host = h.httpHostHeader
|
|
}
|
|
|
|
conn, response, err := websocket.ClientConnect(req, h.tlsConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer conn.Close()
|
|
err = stream.WriteHeaders(h2mux.H1ResponseToH2ResponseHeaders(response))
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Error writing response header")
|
|
}
|
|
// Copy to/from stream to the undelying connection. Use the underlying
|
|
// connection because cloudflared doesn't operate on the message themselves
|
|
websocket.Stream(conn.UnderlyingConn(), stream)
|
|
|
|
return response, nil
|
|
}
|
|
|
|
func (h *TunnelHandler) serveHTTP(stream *h2mux.MuxedStream, req *http.Request) (*http.Response, error) {
|
|
// Support for WSGI Servers by switching transfer encoding from chunked to gzip/deflate
|
|
if h.noChunkedEncoding {
|
|
req.TransferEncoding = []string{"gzip", "deflate"}
|
|
cLength, err := strconv.Atoi(req.Header.Get("Content-Length"))
|
|
if err == nil {
|
|
req.ContentLength = int64(cLength)
|
|
}
|
|
}
|
|
|
|
// Request origin to keep connection alive to improve performance
|
|
req.Header.Set("Connection", "keep-alive")
|
|
|
|
if h.httpHostHeader != "" {
|
|
req.Header.Set("Host", h.httpHostHeader)
|
|
req.Host = h.httpHostHeader
|
|
}
|
|
|
|
response, err := h.httpClient.RoundTrip(req)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Error proxying request to origin")
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
headers := h2mux.H1ResponseToH2ResponseHeaders(response)
|
|
headers = append(headers, h2mux.CreateResponseMetaHeader(h2mux.ResponseMetaHeaderField, h2mux.ResponseSourceOrigin))
|
|
err = stream.WriteHeaders(headers)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Error writing response header")
|
|
}
|
|
if h.isEventStream(response) {
|
|
h.writeEventStream(stream, response.Body)
|
|
} else {
|
|
// Use CopyBuffer, because Copy only allocates a 32KiB buffer, and cross-stream
|
|
// compression generates dictionary on first write
|
|
buf := h.bufferPool.Get()
|
|
defer h.bufferPool.Put(buf)
|
|
io.CopyBuffer(stream, response.Body, buf)
|
|
}
|
|
return response, nil
|
|
}
|
|
|
|
func (h *TunnelHandler) writeEventStream(stream *h2mux.MuxedStream, responseBody io.ReadCloser) {
|
|
reader := bufio.NewReader(responseBody)
|
|
for {
|
|
line, err := reader.ReadBytes('\n')
|
|
if err != nil {
|
|
break
|
|
}
|
|
stream.Write(line)
|
|
}
|
|
}
|
|
|
|
func (h *TunnelHandler) isEventStream(response *http.Response) bool {
|
|
if response.Header.Get("content-type") == "text/event-stream" {
|
|
h.logger.Debug("Detected Server-Side Events from Origin")
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (h *TunnelHandler) writeErrorResponse(stream *h2mux.MuxedStream, err error) {
|
|
h.logger.Errorf("HTTP request error: %s", err)
|
|
stream.WriteHeaders([]h2mux.Header{
|
|
{Name: ":status", Value: "502"},
|
|
h2mux.CreateResponseMetaHeader(h2mux.ResponseMetaHeaderField, h2mux.ResponseSourceCloudflared),
|
|
})
|
|
stream.Write([]byte("502 Bad Gateway"))
|
|
h.metrics.incrementResponses(h.connectionID, "502")
|
|
}
|
|
|
|
func (h *TunnelHandler) logRequest(req *http.Request, cfRay string, lbProbe bool) {
|
|
logger := h.logger
|
|
if cfRay != "" {
|
|
logger.Debugf("CF-RAY: %s %s %s %s", cfRay, req.Method, req.URL, req.Proto)
|
|
} else if lbProbe {
|
|
logger.Debugf("CF-RAY: %s Load Balancer health check %s %s %s", cfRay, req.Method, req.URL, req.Proto)
|
|
} else {
|
|
logger.Infof("CF-RAY: %s All requests should have a CF-RAY header. Please open a support ticket with Cloudflare. %s %s %s ", cfRay, req.Method, req.URL, req.Proto)
|
|
}
|
|
logger.Debugf("CF-RAY: %s Request Headers %+v", cfRay, req.Header)
|
|
|
|
if contentLen := req.ContentLength; contentLen == -1 {
|
|
logger.Debugf("CF-RAY: %s Request Content length unknown", cfRay)
|
|
} else {
|
|
logger.Debugf("CF-RAY: %s Request content length %d", cfRay, contentLen)
|
|
}
|
|
}
|
|
|
|
func (h *TunnelHandler) logResponseOk(r *http.Response, cfRay string, lbProbe bool) {
|
|
h.metrics.incrementResponses(h.connectionID, "200")
|
|
logger := h.logger
|
|
if cfRay != "" {
|
|
logger.Debugf("CF-RAY: %s %s", cfRay, r.Status)
|
|
} else if lbProbe {
|
|
logger.Debugf("Response to Load Balancer health check %s", r.Status)
|
|
} else {
|
|
logger.Infof("%s", r.Status)
|
|
}
|
|
logger.Debugf("CF-RAY: %s Response Headers %+v", cfRay, r.Header)
|
|
|
|
if contentLen := r.ContentLength; contentLen == -1 {
|
|
logger.Debugf("CF-RAY: %s Response content length unknown", cfRay)
|
|
} else {
|
|
logger.Debugf("CF-RAY: %s Response content length %d", cfRay, contentLen)
|
|
}
|
|
}
|
|
|
|
func (h *TunnelHandler) UpdateMetrics(connectionID string) {
|
|
h.metrics.updateMuxerMetrics(connectionID, h.muxer.Metrics())
|
|
}
|
|
|
|
func uint8ToString(input uint8) string {
|
|
return strconv.FormatUint(uint64(input), 10)
|
|
}
|
|
|
|
// Print out the given lines in a nice ASCII box.
|
|
func asciiBox(lines []string, padding int) (box []string) {
|
|
maxLen := maxLen(lines)
|
|
spacer := strings.Repeat(" ", padding)
|
|
|
|
border := "+" + strings.Repeat("-", maxLen+(padding*2)) + "+"
|
|
|
|
box = append(box, border)
|
|
for _, line := range lines {
|
|
box = append(box, "|"+spacer+line+strings.Repeat(" ", maxLen-len(line))+spacer+"|")
|
|
}
|
|
box = append(box, border)
|
|
return
|
|
}
|
|
|
|
func maxLen(lines []string) int {
|
|
max := 0
|
|
for _, line := range lines {
|
|
if len(line) > max {
|
|
max = len(line)
|
|
}
|
|
}
|
|
return max
|
|
}
|
|
|
|
func trialZoneMsg(url string) []string {
|
|
return []string{
|
|
"Your free tunnel has started! Visit it:",
|
|
" " + url,
|
|
}
|
|
}
|
|
|
|
func activeIncidentsMsg(incidents []Incident) string {
|
|
preamble := "There is an active Cloudflare incident that may be related:"
|
|
if len(incidents) > 1 {
|
|
preamble = "There are active Cloudflare incidents that may be related:"
|
|
}
|
|
incidentStrings := []string{}
|
|
for _, incident := range incidents {
|
|
incidentString := fmt.Sprintf("%s (%s)", incident.Name, incident.URL())
|
|
incidentStrings = append(incidentStrings, incidentString)
|
|
}
|
|
return preamble + " " + strings.Join(incidentStrings, "; ")
|
|
|
|
}
|
|
|
|
func findCfRayHeader(h1 *http.Request) string {
|
|
return h1.Header.Get("Cf-Ray")
|
|
}
|
|
|
|
func isLBProbeRequest(req *http.Request) bool {
|
|
return strings.HasPrefix(req.UserAgent(), lbProbeUserAgentPrefix)
|
|
}
|