sendnrw/windows_exporter
2
0
Fork 0
mirror of https://github.com/prometheus-community/windows_exporter.git synced 2026-02-20 19:56:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

mssql: expose server version info (#1741)

Browse Source 
Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>
This commit is contained in:
Jan-Otto Kröpke
2024-11-19 23:35:55 +01:00
committed by GitHub
parent 2335fba18a
commit 8c90961c10
7 changed files with 490 additions and 298 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
internal/headers/iphlpapi/iphlpapi.go
View File

@@ -3,6 +3,7 @@
package iphlpapi
import (
"encoding/binary"
"fmt"
"unsafe"
@@ -15,65 +16,99 @@ var (
)
func GetTCPConnectionStates(family uint32) (map[MIB_TCP_STATE]uint32, error) {
stateCounts := make(map[MIB_TCP_STATE]uint32)
switch family {
case windows.AF_INET:
table, err := getExtendedTcpTable[MIB_TCPROW_OWNER_PID](family, TCPTableOwnerPIDAll)
if err != nil {
return nil, err
}
for _, row := range table {
stateCounts[row.dwState]++
}
return stateCounts, nil
case windows.AF_INET6:
table, err := getExtendedTcpTable[MIB_TCP6ROW_OWNER_PID](family, TCPTableOwnerPIDAll)
if err != nil {
return nil, err
}
for _, row := range table {
stateCounts[row.dwState]++
}
return stateCounts, nil
default:
return nil, fmt.Errorf("unsupported address family %d", family)
}
}
func GetOwnerPIDOfTCPPort(family uint32, tcpPort uint16) (uint32, error) {
switch family {
case windows.AF_INET:
table, err := getExtendedTcpTable[MIB_TCPROW_OWNER_PID](family, TCPTableOwnerPIDListener)
if err != nil {
return 0, err
}
for _, row := range table {
if row.dwLocalPort.uint16() == tcpPort {
return row.dwOwningPid, nil
}
}
return 0, fmt.Errorf("no process found for port %d", tcpPort)
case windows.AF_INET6:
table, err := getExtendedTcpTable[MIB_TCP6ROW_OWNER_PID](family, TCPTableOwnerPIDListener)
if err != nil {
return 0, err
}
for _, row := range table {
if row.dwLocalPort.uint16() == tcpPort {
return row.dwOwningPid, nil
}
}
return 0, fmt.Errorf("no process found for port %d", tcpPort)
default:
return 0, fmt.Errorf("unsupported address family %d", family)
}
}
func getExtendedTcpTable[T any](ulAf uint32, tableClass uint32) ([]T, error) {
var size uint32
stateCounts := make(map[MIB_TCP_STATE]uint32)
rowSize := uint32(unsafe.Sizeof(MIB_TCPROW_OWNER_PID{}))
tableClass := TCPTableClass
ret, _, _ := procGetExtendedTcpTable.Call(
uintptr(0),
uintptr(unsafe.Pointer(&size)),
uintptr(0),
uintptr(ulAf),
uintptr(tableClass),
uintptr(0),
)
if family == windows.AF_INET6 {
rowSize = uint32(unsafe.Sizeof(MIB_TCP6ROW_OWNER_PID{}))
tableClass = TCP6TableClass
}
ret := getExtendedTcpTable(0, &size, true, family, tableClass, 0)
if ret != 0 && ret != uintptr(windows.ERROR_INSUFFICIENT_BUFFER) {
if ret != uintptr(windows.ERROR_INSUFFICIENT_BUFFER) {
return nil, fmt.Errorf("getExtendedTcpTable (size query) failed with code %d", ret)
}
buf := make([]byte, size)
ret = getExtendedTcpTable(uintptr(unsafe.Pointer(&buf[0])), &size, true, family, tableClass, 0)
ret, _, _ = procGetExtendedTcpTable.Call(
uintptr(unsafe.Pointer(&buf[0])),
uintptr(unsafe.Pointer(&size)),
uintptr(0),
uintptr(ulAf),
uintptr(tableClass),
uintptr(0),
)
if ret != 0 {
return nil, fmt.Errorf("getExtendedTcpTable (data query) failed with code %d", ret)
}
numEntries := *(*uint32)(unsafe.Pointer(&buf[0]))
for i := range numEntries {
var state MIB_TCP_STATE
if family == windows.AF_INET6 {
row := (*MIB_TCP6ROW_OWNER_PID)(unsafe.Pointer(&buf[4+i*rowSize]))
state = row.dwState
} else {
row := (*MIB_TCPROW_OWNER_PID)(unsafe.Pointer(&buf[4+i*rowSize]))
state = row.dwState
}
stateCounts[state]++
}
return stateCounts, nil
}
func getExtendedTcpTable(pTCPTable uintptr, pdwSize *uint32, bOrder bool, ulAf uint32, tableClass uint32, reserved uint32) uintptr {
ret, _, _ := procGetExtendedTcpTable.Call(
pTCPTable,
uintptr(unsafe.Pointer(pdwSize)),
uintptr(boolToInt(bOrder)),
uintptr(ulAf),
uintptr(tableClass),
uintptr(reserved),
)
return ret
}
func boolToInt(b bool) int {
if b {
return 1
}
return 0
return unsafe.Slice((*T)(unsafe.Pointer(&buf[4])), binary.LittleEndian.Uint32(buf)), nil
}