mirror of
https://github.com/golang/go
synced 2024-10-04 18:31:22 -06:00
322214cf54
This CL reduces unnecessary network facility lookups introduced by recent changes below. changeset: 15798:53a4da6a4f4a net: return correct point-to-point interface address on linux changeset: 15799:a81ef8e0cc05 net: set up IPv6 scoped addressing zone for network facilities Also adds a test case for issue 4839. Benchmark results on linux/amd64, virtual machine: benchmark old ns/op new ns/op delta BenchmarkInterfaces-2 80487 80382 -0.13% BenchmarkInterfaceByIndex-2 72013 71391 -0.86% BenchmarkInterfaceByName-2 79865 80101 +0.30% BenchmarkInterfaceAddrs-2 42071 829677 +1872.09% BenchmarkInterfacesAndAddrs-2 35016 607622 +1635.27% BenchmarkInterfacesAndMulticastAddrs-2 169849 169082 -0.45% old: 15797:9c3930413c1b, new: tip Benchmark results on linux/amd64, virtual machine: benchmark old ns/op new ns/op delta BenchmarkInterfaces-2 80487 81459 +1.21% BenchmarkInterfaceByIndex-2 72013 71512 -0.70% BenchmarkInterfaceByName-2 79865 80567 +0.88% BenchmarkInterfaceAddrs-2 42071 120108 +185.49% BenchmarkInterfacesAndAddrs-2 35016 33259 -5.02% BenchmarkInterfacesAndMulticastAddrs-2 169849 82391 -51.49% old: 15797:9c3930413c1b, new: tip+CL7400055 Benchmark results on darwin/amd64: benchmark old ns/op new ns/op delta BenchmarkInterfaces-2 34402 34231 -0.50% BenchmarkInterfaceByIndex-2 13192 12956 -1.79% BenchmarkInterfaceByName-2 34791 34388 -1.16% BenchmarkInterfaceAddrs-2 36565 63906 +74.77% BenchmarkInterfacesAndAddrs-2 17497 31068 +77.56% BenchmarkInterfacesAndMulticastAddrs-2 25276 66711 +163.93% old: 15797:9c3930413c1b, new: tip Benchmark results on darwin/amd64: benchmark old ns/op new ns/op delta BenchmarkInterfaces-2 34402 31854 -7.41% BenchmarkInterfaceByIndex-2 13192 12950 -1.83% BenchmarkInterfaceByName-2 34791 31926 -8.23% BenchmarkInterfaceAddrs-2 36565 42144 +15.26% BenchmarkInterfacesAndAddrs-2 17497 17329 -0.96% BenchmarkInterfacesAndMulticastAddrs-2 25276 24870 -1.61% old: 15797:9c3930413c1b, new: tip+CL7400055 Update #4234. Fixes #4839 (again). Fixes #4866. R=golang-dev, fullung CC=golang-dev https://golang.org/cl/7400055
159 lines
4.0 KiB
Go
159 lines
4.0 KiB
Go
// Copyright 2011 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package net
|
|
|
|
import (
|
|
"os"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
func bytePtrToString(p *uint8) string {
|
|
a := (*[10000]uint8)(unsafe.Pointer(p))
|
|
i := 0
|
|
for a[i] != 0 {
|
|
i++
|
|
}
|
|
return string(a[:i])
|
|
}
|
|
|
|
func getAdapterList() (*syscall.IpAdapterInfo, error) {
|
|
b := make([]byte, 1000)
|
|
l := uint32(len(b))
|
|
a := (*syscall.IpAdapterInfo)(unsafe.Pointer(&b[0]))
|
|
// TODO(mikio): GetAdaptersInfo returns IP_ADAPTER_INFO that
|
|
// contains IPv4 address list only. We should use another API
|
|
// for fetching IPv6 stuff from the kernel.
|
|
err := syscall.GetAdaptersInfo(a, &l)
|
|
if err == syscall.ERROR_BUFFER_OVERFLOW {
|
|
b = make([]byte, l)
|
|
a = (*syscall.IpAdapterInfo)(unsafe.Pointer(&b[0]))
|
|
err = syscall.GetAdaptersInfo(a, &l)
|
|
}
|
|
if err != nil {
|
|
return nil, os.NewSyscallError("GetAdaptersInfo", err)
|
|
}
|
|
return a, nil
|
|
}
|
|
|
|
func getInterfaceList() ([]syscall.InterfaceInfo, error) {
|
|
s, err := sysSocket(syscall.AF_INET, syscall.SOCK_DGRAM, syscall.IPPROTO_UDP)
|
|
if err != nil {
|
|
return nil, os.NewSyscallError("Socket", err)
|
|
}
|
|
defer syscall.Closesocket(s)
|
|
|
|
ii := [20]syscall.InterfaceInfo{}
|
|
ret := uint32(0)
|
|
size := uint32(unsafe.Sizeof(ii))
|
|
err = syscall.WSAIoctl(s, syscall.SIO_GET_INTERFACE_LIST, nil, 0, (*byte)(unsafe.Pointer(&ii[0])), size, &ret, nil, 0)
|
|
if err != nil {
|
|
return nil, os.NewSyscallError("WSAIoctl", err)
|
|
}
|
|
c := ret / uint32(unsafe.Sizeof(ii[0]))
|
|
return ii[:c-1], nil
|
|
}
|
|
|
|
// If the ifindex is zero, interfaceTable returns mappings of all
|
|
// network interfaces. Otherwise it returns a mapping of a specific
|
|
// interface.
|
|
func interfaceTable(ifindex int) ([]Interface, error) {
|
|
ai, err := getAdapterList()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ii, err := getInterfaceList()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var ift []Interface
|
|
for ; ai != nil; ai = ai.Next {
|
|
index := ai.Index
|
|
if ifindex == 0 || ifindex == int(index) {
|
|
var flags Flags
|
|
|
|
row := syscall.MibIfRow{Index: index}
|
|
e := syscall.GetIfEntry(&row)
|
|
if e != nil {
|
|
return nil, os.NewSyscallError("GetIfEntry", e)
|
|
}
|
|
|
|
for _, ii := range ii {
|
|
ip := (*syscall.RawSockaddrInet4)(unsafe.Pointer(&ii.Address)).Addr
|
|
ipv4 := IPv4(ip[0], ip[1], ip[2], ip[3])
|
|
ipl := &ai.IpAddressList
|
|
for ipl != nil {
|
|
ips := bytePtrToString(&ipl.IpAddress.String[0])
|
|
if ipv4.Equal(parseIPv4(ips)) {
|
|
break
|
|
}
|
|
ipl = ipl.Next
|
|
}
|
|
if ipl == nil {
|
|
continue
|
|
}
|
|
if ii.Flags&syscall.IFF_UP != 0 {
|
|
flags |= FlagUp
|
|
}
|
|
if ii.Flags&syscall.IFF_LOOPBACK != 0 {
|
|
flags |= FlagLoopback
|
|
}
|
|
if ii.Flags&syscall.IFF_BROADCAST != 0 {
|
|
flags |= FlagBroadcast
|
|
}
|
|
if ii.Flags&syscall.IFF_POINTTOPOINT != 0 {
|
|
flags |= FlagPointToPoint
|
|
}
|
|
if ii.Flags&syscall.IFF_MULTICAST != 0 {
|
|
flags |= FlagMulticast
|
|
}
|
|
}
|
|
|
|
name := bytePtrToString(&ai.AdapterName[0])
|
|
|
|
ifi := Interface{
|
|
Index: int(index),
|
|
MTU: int(row.Mtu),
|
|
Name: name,
|
|
HardwareAddr: HardwareAddr(row.PhysAddr[:row.PhysAddrLen]),
|
|
Flags: flags}
|
|
ift = append(ift, ifi)
|
|
}
|
|
}
|
|
return ift, nil
|
|
}
|
|
|
|
// If the ifi is nil, interfaceAddrTable returns addresses for all
|
|
// network interfaces. Otherwise it returns addresses for a specific
|
|
// interface.
|
|
func interfaceAddrTable(ifi *Interface) ([]Addr, error) {
|
|
ai, err := getAdapterList()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var ifat []Addr
|
|
for ; ai != nil; ai = ai.Next {
|
|
index := ai.Index
|
|
if ifi == nil || ifi.Index == int(index) {
|
|
ipl := &ai.IpAddressList
|
|
for ; ipl != nil; ipl = ipl.Next {
|
|
ifa := IPAddr{IP: parseIPv4(bytePtrToString(&ipl.IpAddress.String[0]))}
|
|
ifat = append(ifat, ifa.toAddr())
|
|
}
|
|
}
|
|
}
|
|
return ifat, nil
|
|
}
|
|
|
|
// interfaceMulticastAddrTable returns addresses for a specific
|
|
// interface.
|
|
func interfaceMulticastAddrTable(ifi *Interface) ([]Addr, error) {
|
|
// TODO(mikio): Implement this like other platforms.
|
|
return nil, nil
|
|
}
|