remove vendor and add iphlpapi.go to implement net windows api

Author: pytimer

Gopkg.lock

@@ -6,22 +6,17 @@ import (
 	"fmt"
 	"net"
 	"syscall"
-
-	"github.com/kbinani/win"
 )
 
-// ErrInsufficientBuffer windows api ERROR_INSUFFICIENT_BUFFER
-const ErrInsufficientBuffer = 122
-
-func decodePort(port win.DWORD) uint16 {
+func decodePort(port uint32) uint16 {
 	return syscall.Ntohs(uint16(port))
 }
 
-func parseIPv4(addr win.DWORD) string {
+func parseIPv4(addr uint32) string {
 	return fmt.Sprintf("%d.%d.%d.%d", addr&255, addr>>8&255, addr>>16&255, addr>>24&255)
 }
 
-func parseIPv6(addr [16]win.UCHAR) string {
+func parseIPv6(addr [16]byte) string {
 	var ret [16]byte
 	for i := 0; i < 16; i++ {
 		ret[i] = uint8(addr[i])
@@ -33,7 +28,7 @@ func parseIPv6(addr [16]win.UCHAR) string {
 }
 
 // TCPStatuses https://msdn.microsoft.com/en-us/library/windows/desktop/bb485761(v=vs.85).aspx
-var TCPStatuses = map[win.MIB_TCP_STATE]string{
+var TCPStatuses = map[MIB_TCP_STATE]string{
 	1:  "CLOSED",
 	2:  "LISTEN",
 	3:  "SYN_SENT",

Gopkg.toml

@@ -0,0 +1,60 @@
+// +build windows
+
+package winnetstat
+
+import (
+	"syscall"
+	"unsafe"
+
+	"golang.org/x/sys/windows"
+)
+
+var (
+	// Library
+	modiphlpapi = windows.NewLazySystemDLL("iphlpapi.dll")
+
+	// Functions
+	procGetTCPTable2        = modiphlpapi.NewProc("GetTcpTable2")
+	procGetTCP6Table2       = modiphlpapi.NewProc("GetTcp6Table2")
+	procGetExtendedTCPTable = modiphlpapi.NewProc("GetExtendedTcpTable")
+	procGetExtendedUDPTable = modiphlpapi.NewProc("GetExtendedUdpTable")
+)
+
+func getUintptrFromBool(b bool) uintptr {
+	if b {
+		return 1
+	}
+	return 0
+}
+
+func GetTcpTable2(tcpTable PMIB_TCPTABLE2, bufSize *uint32, order bool) (errcode error) {
+	r1, _, _ := syscall.Syscall(procGetTCPTable2.Addr(), 3, uintptr(unsafe.Pointer(tcpTable)), uintptr(unsafe.Pointer(bufSize)), getUintptrFromBool(order))
+	if r1 != 0 {
+		errcode = syscall.Errno(r1)
+	}
+	return
+}
+
+func GetTcp6Table2(tcpTable PMIB_TCP6TABLE2, bufSize *uint32, order bool) (errcode error) {
+	r1, _, _ := syscall.Syscall(procGetTCP6Table2.Addr(), 3, uintptr(unsafe.Pointer(tcpTable)), uintptr(unsafe.Pointer(bufSize)), getUintptrFromBool(order))
+	if r1 != 0 {
+		errcode = syscall.Errno(r1)
+	}
+	return
+}
+
+func GetExtendedUdpTable(pUdpTable uintptr, pdwSize *uint32, bOrder bool, ulAf uint32, tableClass UDP_TABLE_CLASS, reserved uint32) (errcode error) {
+	r1, _, _ := syscall.Syscall6(procGetExtendedUDPTable.Addr(), 6, pUdpTable, uintptr(unsafe.Pointer(pdwSize)), getUintptrFromBool(bOrder), uintptr(ulAf), uintptr(tableClass), uintptr(reserved))
+	if r1 != 0 {
+		errcode = syscall.Errno(r1)
+	}
+	return
+}
+
+func GetExtendedTcpTable(pTcpTable uintptr, pdwSize *uint32, bOrder bool, ulAf uint32, tableClass TCP_TABLE_CLASS, reserved uint32) (errcode error) {
+	r1, _, _ := syscall.Syscall6(procGetExtendedTCPTable.Addr(), 6, pTcpTable, uintptr(unsafe.Pointer(pdwSize)), getUintptrFromBool(bOrder), uintptr(ulAf), uintptr(tableClass), uintptr(reserved))
+	if r1 != 0 {
+		errcode = syscall.Errno(r1)
+	}
+	return
+}

common.go

@@ -119,6 +119,4 @@ func getNetStatWithKindFile(filename string) ([]NetStat, error) {
 	default:
 		return nil, fmt.Errorf("invalid kind filename: %s", filename)
 	}
-
-	return nil, nil
 }