aboutsummaryrefslogtreecommitdiffstats
path: root/device
diff options
context:
space:
mode:
authorJason A. Donenfeld <Jason@zx2c4.com>2019-10-21 13:29:57 +0200
committerJason A. Donenfeld <Jason@zx2c4.com>2019-10-21 13:29:57 +0200
commitffffbbcc8a3344a45c45884f11a1aee1407615ab (patch)
tree0c73dacb74ebf195b49841751f775ca33f6314c5 /device
parentdevice: remove dead error reporting code (diff)
downloadwireguard-go-ffffbbcc8a3344a45c45884f11a1aee1407615ab.tar.xz
wireguard-go-ffffbbcc8a3344a45c45884f11a1aee1407615ab.zip
device: allow blackholing sockets
Diffstat (limited to 'device')
-rw-r--r--device/boundif_windows.go6
-rw-r--r--device/conn_default.go12
2 files changed, 14 insertions, 4 deletions
diff --git a/device/boundif_windows.go b/device/boundif_windows.go
index 7879a43..6908415 100644
--- a/device/boundif_windows.go
+++ b/device/boundif_windows.go
@@ -18,7 +18,7 @@ const (
sockoptIPV6_UNICAST_IF = 31
)
-func (device *Device) BindSocketToInterface4(interfaceIndex uint32) error {
+func (device *Device) BindSocketToInterface4(interfaceIndex uint32, blackhole bool) error {
/* MSDN says for IPv4 this needs to be in net byte order, so that it's like an IP address with leading zeros. */
bytes := make([]byte, 4)
binary.BigEndian.PutUint32(bytes, interfaceIndex)
@@ -41,10 +41,11 @@ func (device *Device) BindSocketToInterface4(interfaceIndex uint32) error {
if err != nil {
return err
}
+ device.net.bind.(*nativeBind).blackhole4 = blackhole
return nil
}
-func (device *Device) BindSocketToInterface6(interfaceIndex uint32) error {
+func (device *Device) BindSocketToInterface6(interfaceIndex uint32, blackhole bool) error {
sysconn, err := device.net.bind.(*nativeBind).ipv6.SyscallConn()
if err != nil {
return err
@@ -58,5 +59,6 @@ func (device *Device) BindSocketToInterface6(interfaceIndex uint32) error {
if err != nil {
return err
}
+ device.net.bind.(*nativeBind).blackhole6 = blackhole
return nil
}
diff --git a/device/conn_default.go b/device/conn_default.go
index 820bb96..661f57d 100644
--- a/device/conn_default.go
+++ b/device/conn_default.go
@@ -21,8 +21,10 @@ import (
*/
type nativeBind struct {
- ipv4 *net.UDPConn
- ipv6 *net.UDPConn
+ ipv4 *net.UDPConn
+ ipv6 *net.UDPConn
+ blackhole4 bool
+ blackhole6 bool
}
type NativeEndpoint net.UDPAddr
@@ -159,11 +161,17 @@ func (bind *nativeBind) Send(buff []byte, endpoint Endpoint) error {
if bind.ipv4 == nil {
return syscall.EAFNOSUPPORT
}
+ if bind.blackhole4 {
+ return nil
+ }
_, err = bind.ipv4.WriteToUDP(buff, (*net.UDPAddr)(nend))
} else {
if bind.ipv6 == nil {
return syscall.EAFNOSUPPORT
}
+ if bind.blackhole6 {
+ return nil
+ }
_, err = bind.ipv6.WriteToUDP(buff, (*net.UDPAddr)(nend))
}
return err