multipath.go

package chained

import (
	"context"
	"fmt"
	"net"

	"github.com/getlantern/common/config"
	"github.com/getlantern/errors"
	"github.com/getlantern/flashlight/v7/bandit"
	"github.com/getlantern/flashlight/v7/common"
	"github.com/getlantern/flashlight/v7/ops"
	"github.com/getlantern/multipath"
)

type mpDialerAdapter struct {
	impl  proxyImpl
	label string
}

func (d *mpDialerAdapter) DialContext(ctx context.Context) (net.Conn, error) {
	var op *ops.Op
	if value := ctx.Value("op"); value != nil {
		if existing, ok := value.(*ops.Op); ok {
			op = existing.Begin("dial_subflow")
		}
	}
	if op == nil {
		op = ops.Begin("dial_subflow")
	}
	defer op.End()
	return d.impl.dialServer(op, ctx)
}

func (d *mpDialerAdapter) Label() string {
	return d.label
}

type multipathImpl struct {
	nopCloser
	dialer multipath.Dialer
}

func (impl *multipathImpl) dialServer(op *ops.Op, ctx context.Context) (net.Conn, error) {
	return impl.dialer.DialContext(context.WithValue(ctx, "op", op))
}

func (impl *multipathImpl) FormatStats() []string {
	return impl.dialer.(multipath.Stats).FormatStats()
}

func CreateMPDialer(configDir, endpoint string, ss map[string]*config.ProxyConfig, uc common.UserConfig) (bandit.Dialer, error) {
	if len(ss) < 1 {
		return nil, errors.New("no dialers")
	}
	var p *proxy
	var err error
	var dialers []multipath.Dialer
	for name, s := range ss {
		if p == nil {
			// Note: we pass the first server info to newProxy for the attributes shared by all paths
			p, err = newProxy(endpoint, endpoint+":0", "multipath", "multipath", s, uc)
			if err != nil {
				return nil, err
			}
		}
		addr, transport, _, err := extractParams(s)
		if err != nil {
			return nil, err
		}
		impl, err := createImpl(configDir, name, addr, transport, s, uc, p.reportDialCore)
		if err != nil {
			log.Errorf("failed to add %v to %v, continuing: %v", s.Addr, name, err)
			continue
		}
		label := fmt.Sprintf("%-38s at %21s", fmt.Sprintf("%s(%s)", name, transport), addr)
		dialers = append(dialers, &mpDialerAdapter{impl, label})
	}
	if len(dialers) == 0 {
		return nil, errors.New("no subflow dialer")
	}
	p.impl = &multipathImpl{dialer: multipath.NewDialer(endpoint, dialers)}
	return p, nil
}

func groupByMultipathEndpoint(proxies map[string]*config.ProxyConfig) map[string]map[string]*config.ProxyConfig {
	groups := make(map[string]map[string]*config.ProxyConfig)
	for name, s := range proxies {
		group, exists := groups[s.MultipathEndpoint]
		if !exists {
			group = make(map[string]*config.ProxyConfig)
			groups[s.MultipathEndpoint] = group
		}
		group[name] = s
	}
	return groups
}