mirror of
https://github.com/prometheus-community/windows_exporter.git
synced 2026-10-06 12:49:05 +02:00
44fd6ccc85
Co-authored-by: EisenbergD <dominik.eisenberg@beiersdorf.com> Co-authored-by: Jan-Otto Kröpke <mail@jkroepke.de> Signed-off-by: EisenbergD <dominik.eisenberg@beiersdorf.com> Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>
469 lines
12 KiB
Go
469 lines
12 KiB
Go
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// Copyright The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//go:build windows
|
|
|
|
package mi_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prometheus-community/windows_exporter/internal/mi"
|
|
"github.com/prometheus-community/windows_exporter/internal/utils/testutils"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
type win32Process struct {
|
|
Name string `mi:"Name"`
|
|
}
|
|
|
|
// win32DiskDrive is used to exercise UINT16A (uint16[]) unmarshalling.
|
|
// Win32_DiskDrive.Capabilities is a reliably-populated uint16[] on any host.
|
|
type win32DiskDrive struct {
|
|
Capabilities []uint16 `mi:"Capabilities"`
|
|
}
|
|
|
|
// win32DiskDriveWrongType maps the uint16[] Capabilities property onto an
|
|
// incompatible Go type to exercise the UINT16A type guard.
|
|
type win32DiskDriveWrongType struct {
|
|
Capabilities []string `mi:"Capabilities"`
|
|
}
|
|
|
|
func Test_MI_Application_Initialize(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Application_TestConnection(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
err = session.TestConnection()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Query(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "select Name from win32_process where handle = 0")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, instance)
|
|
|
|
count, err := instance.GetElementCount()
|
|
require.NoError(t, err)
|
|
require.NotZero(t, count)
|
|
|
|
element, err := instance.GetElement("Name")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, element)
|
|
|
|
value, err := element.GetValue()
|
|
require.NoError(t, err)
|
|
require.Equal(t, "System Idle Process", value)
|
|
require.NotEmpty(t, value)
|
|
|
|
require.False(t, moreResults)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_QueryUnmarshal(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
var processes []win32Process
|
|
|
|
queryProcess, err := mi.NewQuery("select Name from win32_process where handle = 0")
|
|
require.NoError(t, err)
|
|
|
|
err = session.QueryUnmarshal(&processes, mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, queryProcess)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []win32Process{{Name: "System Idle Process"}}, processes)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_EmptyQuery(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "SELECT Name, Status FROM win32_PrintJob")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.NoError(t, err)
|
|
require.Empty(t, instance)
|
|
require.False(t, moreResults)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Query_Unmarshal(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "SELECT Name FROM Win32_Process WHERE Handle = 0 OR Handle = 4")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
var processes []win32Process
|
|
|
|
err = operation.Unmarshal(&processes)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []win32Process{{Name: "System Idle Process"}, {Name: "System"}}, processes)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_FD_Leak(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
session, err := application.NewSession(nil)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
currentFileHandle, err := testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
queryPrinter, err := mi.NewQuery("SELECT Name FROM Win32_Process")
|
|
require.NoError(t, err)
|
|
|
|
for range 300 {
|
|
var processes []win32Process
|
|
|
|
err := session.Query(&processes, mi.NamespaceRootCIMv2, queryPrinter, -1)
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
}
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
}
|
|
|
|
func Test_MI_QueryTimeout(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operationOptions, err := application.NewOperationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operationOptions)
|
|
|
|
err = operationOptions.SetTimeout(1 * time.Millisecond)
|
|
require.NoError(t, err)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, operationOptions, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "select Name from win32_process where handle = 0")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.ErrorIs(t, err, mi.MI_RESULT_INVALID_OPERATION_TIMEOUT)
|
|
require.False(t, moreResults)
|
|
require.Empty(t, instance)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Query_Uint16Array(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
session, err := application.NewSession(nil)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(
|
|
mi.OperationFlagsStandardRTTI,
|
|
nil,
|
|
mi.NamespaceRootCIMv2,
|
|
mi.QueryDialectWQL,
|
|
"SELECT Capabilities FROM Win32_DiskDrive",
|
|
)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
found := false
|
|
|
|
for {
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.NoError(t, err)
|
|
|
|
if instance == nil {
|
|
break
|
|
}
|
|
|
|
element, err := instance.GetElement("Capabilities")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, element)
|
|
|
|
value, err := element.GetValue()
|
|
require.NoError(t, err)
|
|
|
|
capabilities, ok := value.([]uint16)
|
|
require.True(t, ok, "Capabilities should unmarshal to []uint16")
|
|
|
|
if len(capabilities) > 0 {
|
|
found = true
|
|
}
|
|
|
|
if !moreResults {
|
|
break
|
|
}
|
|
}
|
|
|
|
require.True(t, found, "expected at least one disk with a non-empty uint16[] Capabilities")
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_QueryUnmarshal_Uint16Array(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
session, err := application.NewSession(nil)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
query, err := mi.NewQuery("SELECT Capabilities FROM Win32_DiskDrive")
|
|
require.NoError(t, err)
|
|
|
|
var disks []win32DiskDrive
|
|
|
|
err = session.Query(&disks, mi.NamespaceRootCIMv2, query, -1)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, disks)
|
|
|
|
found := false
|
|
|
|
for _, disk := range disks {
|
|
if len(disk.Capabilities) > 0 {
|
|
found = true
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
require.True(t, found, "expected at least one disk with a non-empty uint16[] Capabilities")
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_QueryUnmarshal_Uint16Array_WrongType(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
session, err := application.NewSession(nil)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
query, err := mi.NewQuery("SELECT Capabilities FROM Win32_DiskDrive")
|
|
require.NoError(t, err)
|
|
|
|
var disks []win32DiskDriveWrongType
|
|
|
|
// Unmarshalling a uint16[] into a []string field must error, not panic.
|
|
err = session.Query(&disks, mi.NamespaceRootCIMv2, query, -1)
|
|
require.Error(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|