44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
|
package peerdiscovery
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"golang.org/x/net/ipv4"
|
||
|
|
"golang.org/x/net/ipv6"
|
||
|
|
)
|
||
|
|
|
||
|
|
type PacketConn4 struct {
|
||
|
|
*ipv4.PacketConn
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReadFrom wraps the ipv4 ReadFrom without a control message
|
||
|
|
func (pc4 PacketConn4) ReadFrom(buf []byte) (int, net.Addr, error) {
|
||
|
|
n, _, addr, err := pc4.PacketConn.ReadFrom(buf)
|
||
|
|
return n, addr, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// WriteTo wraps the ipv4 WriteTo without a control message
|
||
|
|
func (pc4 PacketConn4) WriteTo(buf []byte, dst net.Addr) (int, error) {
|
||
|
|
return pc4.PacketConn.WriteTo(buf, nil, dst)
|
||
|
|
}
|
||
|
|
|
||
|
|
type PacketConn6 struct {
|
||
|
|
*ipv6.PacketConn
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReadFrom wraps the ipv6 ReadFrom without a control message
|
||
|
|
func (pc6 PacketConn6) ReadFrom(buf []byte) (int, net.Addr, error) {
|
||
|
|
n, _, addr, err := pc6.PacketConn.ReadFrom(buf)
|
||
|
|
return n, addr, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// WriteTo wraps the ipv6 WriteTo without a control message
|
||
|
|
func (pc6 PacketConn6) WriteTo(buf []byte, dst net.Addr) (int, error) {
|
||
|
|
return pc6.PacketConn.WriteTo(buf, nil, dst)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetMulticastTTL wraps the hop limit of ipv6
|
||
|
|
func (pc6 PacketConn6) SetMulticastTTL(i int) error {
|
||
|
|
return pc6.SetMulticastHopLimit(i)
|
||
|
|
}
|