aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tunnel/winipcfg/utils.go
blob: c841c40c3da03d38e1056a36bebad184281737c0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
/* SPDX-License-Identifier: MIT
 *
 * Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
 */

package winipcfg

import (
	"net"
	"unsafe"

	"golang.org/x/sys/windows"
)

// SocketAddressToIP function returns IPv4 or IPv6 address from windows.SocketAddress.
// If the address is neither IPv4 not IPv6 nil is returned.
//TODO: Remove once https://go-review.googlesource.com/c/sys/+/178577 is merged.
func SocketAddressToIP(addr *windows.SocketAddress) net.IP {
	if uintptr(addr.SockaddrLength) >= unsafe.Sizeof(windows.RawSockaddrInet4{}) && addr.Sockaddr.Addr.Family == windows.AF_INET {
		return (*windows.RawSockaddrInet4)(unsafe.Pointer(addr.Sockaddr)).Addr[:]
	} else if uintptr(addr.SockaddrLength) >= unsafe.Sizeof(windows.RawSockaddrInet6{}) && addr.Sockaddr.Addr.Family == windows.AF_INET6 {
		return (*windows.RawSockaddrInet6)(unsafe.Pointer(addr.Sockaddr)).Addr[:]
	} else {
		return nil
	}
}