AdGuardDNS/forward/type.go

package forward

import "net"

type transportType int

const (
	typeUdp transportType = iota
	typeTcp
	typeTls
	typeTotalCount // keep this last
)

func stringToTransportType(s string) transportType {
	switch s {
	case "udp":
		return typeUdp
	case "tcp":
		return typeTcp
	case "tcp-tls":
		return typeTls
	}

	return typeUdp
}

func (t *Transport) transportTypeFromConn(pc *persistConn) transportType {
	if _, ok := pc.c.Conn.(*net.UDPConn); ok {
		return typeUdp
	}

	if t.tlsConfig == nil {
		return typeTcp
	}

	return typeTls
}