mirror of https://github.com/bjdgyc/anylink.git
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package dtls
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/pion/dtls/v2/pkg/protocol"
|
|
"github.com/pion/dtls/v2/pkg/protocol/recordlayer"
|
|
"github.com/pion/udp"
|
|
)
|
|
|
|
// Listen creates a DTLS listener
|
|
func Listen(network string, laddr *net.UDPAddr, config *Config) (net.Listener, error) {
|
|
if err := validateConfig(config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
lc := udp.ListenConfig{
|
|
AcceptFilter: func(packet []byte) bool {
|
|
pkts, err := recordlayer.UnpackDatagram(packet)
|
|
if err != nil || len(pkts) < 1 {
|
|
return false
|
|
}
|
|
h := &recordlayer.Header{}
|
|
if err := h.Unmarshal(pkts[0]); err != nil {
|
|
return false
|
|
}
|
|
return h.ContentType == protocol.ContentTypeHandshake
|
|
},
|
|
}
|
|
parent, err := lc.Listen(network, laddr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &listener{
|
|
config: config,
|
|
parent: parent,
|
|
}, nil
|
|
}
|
|
|
|
// NewListener creates a DTLS listener which accepts connections from an inner Listener.
|
|
func NewListener(inner net.Listener, config *Config) (net.Listener, error) {
|
|
if err := validateConfig(config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &listener{
|
|
config: config,
|
|
parent: inner,
|
|
}, nil
|
|
}
|
|
|
|
// listener represents a DTLS listener
|
|
type listener struct {
|
|
config *Config
|
|
parent net.Listener
|
|
}
|
|
|
|
// Accept waits for and returns the next connection to the listener.
|
|
// You have to either close or read on all connection that are created.
|
|
// Connection handshake will timeout using ConnectContextMaker in the Config.
|
|
// If you want to specify the timeout duration, set ConnectContextMaker.
|
|
func (l *listener) Accept() (net.Conn, error) {
|
|
c, err := l.parent.Accept()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return Server(c, l.config)
|
|
}
|
|
|
|
// Close closes the listener.
|
|
// Any blocked Accept operations will be unblocked and return errors.
|
|
// Already Accepted connections are not closed.
|
|
func (l *listener) Close() error {
|
|
return l.parent.Close()
|
|
}
|
|
|
|
// Addr returns the listener's network address.
|
|
func (l *listener) Addr() net.Addr {
|
|
return l.parent.Addr()
|
|
}
|