sendnrw/windows_exporter
2
0
Fork 0
mirror of https://github.com/prometheus-community/windows_exporter.git synced 2026-03-05 01:56:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev_cs_collector' of https://github.com/benridley/windows_exporter into dev_cs_collector

Browse Source
This commit is contained in:
Ben Ridley
2021-03-29 10:14:26 -07:00
parent 33c6b2c6a5 248b7214e3
commit 863b7d8ab4
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
headers/netapi32/netapi32.go
View File

@@ -74,17 +74,19 @@ func netApiBufferFree(buffer *wKSTAInfo102) {
// NetWkstaGetInfo returns information about the configuration of a workstation. // NetWkstaGetInfo returns information about the configuration of a workstation.
// https://docs.microsoft.com/en-us/windows/win32/api/lmwksta/nf-lmwksta-netwkstagetinfo // https://docs.microsoft.com/en-us/windows/win32/api/lmwksta/nf-lmwksta-netwkstagetinfo
func netWkstaGetInfo() (*wKSTAInfo102, uint32, error) { func netWkstaGetInfo() (wKSTAInfo102, uint32, error) {
var lpwi *wKSTAInfo102 var lpwi *wKSTAInfo102
pLevel := uintptr(102) pLevel := uintptr(102)
r1, _, _ := procNetWkstaGetInfo.Call(0, pLevel, uintptr(unsafe.Pointer(&lpwi))) r1, _, _ := procNetWkstaGetInfo.Call(0, pLevel, uintptr(unsafe.Pointer(&lpwi)))
defer netApiBufferFree(lpwi)
if ret := *(*uint32)(unsafe.Pointer(&r1)); ret != 0 { if ret := *(*uint32)(unsafe.Pointer(&r1)); ret != 0 {
return nil, ret, errors.New(NetApiStatus[ret]) return wKSTAInfo102{}, ret, errors.New(NetApiStatus[ret])
} }
return lpwi, 0, nil deref := *lpwi
return deref, 0, nil
} }
// GetWorkstationInfo is an idiomatic wrapper for netWkstaGetInfo // GetWorkstationInfo is an idiomatic wrapper for netWkstaGetInfo
@@ -102,7 +104,5 @@ func GetWorkstationInfo() (WorkstationInfo, error) {
LanRoot: windows.UTF16PtrToString(info.wki102_lanroot), LanRoot: windows.UTF16PtrToString(info.wki102_lanroot),
LoggedOnUsers: info.wki102_logged_on_users, LoggedOnUsers: info.wki102_logged_on_users,
} }
// Free the memory allocated by netapi
netApiBufferFree(info)
return workstationInfo, nil return workstationInfo, nil
} }