mihomo/adapters/outbound/direct.go

46 lines
871 B
Go
Raw Normal View History

2019-12-08 12:17:24 +08:00
package outbound
2018-06-10 22:50:03 +08:00
import (
"context"
2018-06-10 22:50:03 +08:00
"net"
2020-02-09 17:02:48 +08:00
"github.com/Dreamacro/clash/component/dialer"
2018-06-10 22:50:03 +08:00
C "github.com/Dreamacro/clash/constant"
)
2018-12-22 23:56:42 +08:00
type Direct struct {
*Base
2018-06-10 22:50:03 +08:00
}
func (d *Direct) DialContext(ctx context.Context, metadata *C.Metadata) (C.Conn, error) {
2019-05-09 21:00:29 +08:00
address := net.JoinHostPort(metadata.Host, metadata.DstPort)
if metadata.DstIP != nil {
address = net.JoinHostPort(metadata.DstIP.String(), metadata.DstPort)
2018-12-05 21:13:29 +08:00
}
c, err := dialer.DialContext(ctx, "tcp", address)
2018-06-10 22:50:03 +08:00
if err != nil {
2018-12-22 23:56:42 +08:00
return nil, err
2018-06-10 22:50:03 +08:00
}
tcpKeepAlive(c)
return newConn(c, d), nil
2018-11-21 13:47:46 +08:00
}
2020-01-31 14:43:54 +08:00
func (d *Direct) DialUDP(metadata *C.Metadata) (C.PacketConn, error) {
2020-02-09 17:02:48 +08:00
pc, err := dialer.ListenPacket("udp", "")
2019-04-23 23:29:36 +08:00
if err != nil {
2020-01-31 14:43:54 +08:00
return nil, err
2019-04-24 10:29:29 +08:00
}
2020-01-31 14:43:54 +08:00
return newPacketConn(pc, d), nil
2019-04-23 23:29:36 +08:00
}
func NewDirect() *Direct {
2018-12-22 23:56:42 +08:00
return &Direct{
Base: &Base{
name: "DIRECT",
tp: C.Direct,
2019-04-23 23:29:36 +08:00
udp: true,
2018-12-22 23:56:42 +08:00
},
}
2018-06-10 22:50:03 +08:00
}