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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# Required for IP_ADAPTER_ADDRESSES_LH
{.emit:"""/*INCLUDESECTION*/
#include <winsock2.h>
""".}
import winlean
import os
import private/utils
type
IP_ADAPTER_ADDRESSES_LH {.importc, header: "<iphlpapi.h>".} = object
Next: ptr IP_ADAPTER_ADDRESSES_LH
AdapterName: cstring
FirstUnicastAddress: ptr IP_ADAPTER_UNICAST_ADDRESS_LH
Description: WideCString
FriendlyName: WideCString
IP_ADAPTER_UNICAST_ADDRESS_LH {.importc, header: "<iphlpapi.h>".} = object
Next: ptr IP_ADAPTER_UNICAST_ADDRESS_LH
Address: SOCKET_ADDRESS
OnLinkPrefixLength: uint8
SOCKET_ADDRESS {.importc, header: "<winsock2.h>"} = object
lpSockaddr: ptr SockAddr
iSockaddrLength: cint
const
ERROR_BUFFER_OVERFLOW = 0x6f
proc GetAdaptersAddresses(family: ULONG, flags: ULONG, reserved: pointer, adapterAddresses: ptr IP_ADAPTER_ADDRESSES_LH, size: ptr ULONG): ULONG {.stdcall, dynlib: "iphlpapi.dll", importc.}
proc getAdapters*(): seq[Adapter] =
result = @[]
var
bufsz: ULONG = 15*1024
rc: DWORD = ERROR_BUFFER_OVERFLOW
buffer: seq[byte]
while rc == ERROR_BUFFER_OVERFLOW:
buffer.setLen(bufsz)
rc = GetAdaptersAddresses(AF_UNSPEC, 0, nil,
cast[ptr IP_ADAPTER_ADDRESSES_LH](addr buffer[0]), addr bufsz)
if rc != 0:
raiseOSError(rc.OSErrorCode, "GetAdaptersAddresses")
let first = cast[ptr IP_ADAPTER_ADDRESSES_LH](addr buffer[0])
var curr = first
while curr != nil:
let name = $curr.AdapterName
let niceName = $curr.Description
var ips: seq[IP] = @[]
var currAddr = curr.FirstUnicastAddress
while currAddr != nil:
var
ip: IpAddress
scopeId: uint32
flowInfo: uint32
fromSockAddrPtr(currAddr.Address.lpSockaddr, ip, flowInfo, scopeId)
case ip.family
of IpAddressFamily.IPv6:
ips.add(IP(family: IpAddressFamily.IPv6,
flowInfo: flowInfo,
scopeId: scopeId,
ip: ip,
networkPrefix: currAddr.OnLinkPrefixLength.int,
niceName: $curr.FriendlyName))
of IpAddressFamily.IPv4:
ips.add(IP(family: IpAddressFamily.IPv4,
ip: ip,
networkPrefix: currAddr.OnLinkPrefixLength.int,
niceName: $curr.FriendlyName))
currAddr = currAddr.Next
result.add(Adapter(ips: ips, name: name, niceName: niceName))
curr = curr.Next
|
