mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-07-01 03:25:23 +00:00
91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
|
package quic
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"sync"
|
||
|
|
||
|
"v2ray.com/core/transport/internet/tls"
|
||
|
|
||
|
quic "github.com/lucas-clemente/quic-go"
|
||
|
|
||
|
"v2ray.com/core/common"
|
||
|
"v2ray.com/core/common/net"
|
||
|
"v2ray.com/core/transport/internet"
|
||
|
)
|
||
|
|
||
|
type clientSessions struct {
|
||
|
access sync.Mutex
|
||
|
sessions map[net.Destination]quic.Session
|
||
|
}
|
||
|
|
||
|
func (s *clientSessions) getSession(destAddr net.Addr, tlsConfig *tls.Config, sockopt *internet.SocketConfig) (quic.Session, error) {
|
||
|
s.access.Lock()
|
||
|
defer s.access.Unlock()
|
||
|
|
||
|
if s.sessions == nil {
|
||
|
s.sessions = make(map[net.Destination]quic.Session)
|
||
|
}
|
||
|
|
||
|
dest := net.DestinationFromAddr(destAddr)
|
||
|
|
||
|
if session, found := s.sessions[dest]; found {
|
||
|
return session, nil
|
||
|
}
|
||
|
|
||
|
conn, err := internet.ListenSystemPacket(context.Background(), &net.UDPAddr{
|
||
|
IP: []byte{0, 0, 0, 0},
|
||
|
Port: 0,
|
||
|
}, sockopt)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
config := &quic.Config{
|
||
|
Versions: []quic.VersionNumber{quic.VersionMilestone0_10_0},
|
||
|
ConnectionIDLength: 12,
|
||
|
KeepAlive: true,
|
||
|
}
|
||
|
|
||
|
session, err := quic.DialContext(context.Background(), conn, destAddr, "", tlsConfig.GetTLSConfig(tls.WithDestination(dest)), config)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
s.sessions[dest] = session
|
||
|
return session, nil
|
||
|
}
|
||
|
|
||
|
var client clientSessions
|
||
|
|
||
|
func Dial(ctx context.Context, dest net.Destination, streamSettings *internet.MemoryStreamConfig) (internet.Connection, error) {
|
||
|
tlsConfig := tls.ConfigFromStreamSettings(streamSettings)
|
||
|
if tlsConfig == nil {
|
||
|
return nil, newError("TLS not enabled for QUIC")
|
||
|
}
|
||
|
|
||
|
destAddr, err := net.ResolveUDPAddr("udp", dest.NetAddr())
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
session, err := client.getSession(destAddr, tlsConfig, streamSettings.SocketSettings)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
conn, err := session.OpenStreamSync()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return &interConn{
|
||
|
stream: conn,
|
||
|
local: session.LocalAddr(),
|
||
|
remote: destAddr,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
common.Must(internet.RegisterTransportDialer(protocolName, Dial))
|
||
|
}
|