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

add error handling

Browse Source 
Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>
This commit is contained in:
Jan-Otto Kröpke
2023-07-10 02:12:22 +02:00
parent ad1ab35399
commit 9e368d49e7
1 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
wmi/perflib/perflib.go
View File

@@ -305,10 +305,16 @@ func QueryPerformanceData(query string) ([]*PerfObject, error) {
objOffset := int64(header.HeaderLength) objOffset := int64(header.HeaderLength)
for i := 0; i < numObjects; i++ { for i := 0; i < numObjects; i++ {
r.Seek(objOffset, io.SeekStart) _, err := r.Seek(objOffset, io.SeekStart)
if err != nil {
return nil, err
}
obj := new(perfObjectType) obj := new(perfObjectType)
obj.BinaryReadFrom(r) err = obj.BinaryReadFrom(r)
if err != nil {
return nil, err
}
numCounterDefs := int(obj.NumCounters) numCounterDefs := int(obj.NumCounters)
numInstances := int(obj.NumInstances) numInstances := int(obj.NumInstances)
@@ -336,7 +342,10 @@ func QueryPerformanceData(query string) ([]*PerfObject, error) {
for i := 0; i < numCounterDefs; i++ { for i := 0; i < numCounterDefs; i++ {
def := new(perfCounterDefinition) def := new(perfCounterDefinition)
def.BinaryReadFrom(r) err := def.BinaryReadFrom(r)
if err != nil {
return nil, err
}
counterDefs[i] = &PerfCounterDef{ counterDefs[i] = &PerfCounterDef{
Name: def.LookupName(), Name: def.LookupName(),
@@ -356,9 +365,15 @@ func QueryPerformanceData(query string) ([]*PerfObject, error) {
if obj.NumInstances <= 0 { if obj.NumInstances <= 0 {
blockOffset := objOffset + int64(obj.DefinitionLength) blockOffset := objOffset + int64(obj.DefinitionLength)
r.Seek(blockOffset, io.SeekStart) _, err := r.Seek(blockOffset, io.SeekStart)
if err != nil {
return nil, err
}
_, counters := parseCounterBlock(buffer, r, blockOffset, counterDefs) _, counters, err := parseCounterBlock(buffer, r, blockOffset, counterDefs)
if err != nil {
return nil, err
}
instances[0] = &PerfInstance{ instances[0] = &PerfInstance{
Name: "", Name: "",
@@ -370,14 +385,23 @@ func QueryPerformanceData(query string) ([]*PerfObject, error) {
instOffset := objOffset + int64(obj.DefinitionLength) instOffset := objOffset + int64(obj.DefinitionLength)
for i := 0; i < numInstances; i++ { for i := 0; i < numInstances; i++ {
r.Seek(instOffset, io.SeekStart) _, err := r.Seek(instOffset, io.SeekStart)
if err != nil {
return nil, err
}
inst := new(perfInstanceDefinition) inst := new(perfInstanceDefinition)
inst.BinaryReadFrom(r) err = inst.BinaryReadFrom(r)
if err != nil {
return nil, err
}
name, _ := readUTF16StringAtPos(r, instOffset+int64(inst.NameOffset), inst.NameLength) name, _ := readUTF16StringAtPos(r, instOffset+int64(inst.NameOffset), inst.NameLength)
pos := instOffset + int64(inst.ByteLength) pos := instOffset + int64(inst.ByteLength)
offset, counters := parseCounterBlock(buffer, r, pos, counterDefs) offset, counters, err := parseCounterBlock(buffer, r, pos, counterDefs)
if err != nil {
return nil, err
}
instances[i] = &PerfInstance{ instances[i] = &PerfInstance{
Name: name, Name: name,
@@ -396,10 +420,16 @@ func QueryPerformanceData(query string) ([]*PerfObject, error) {
return objects, nil return objects, nil
} }
func parseCounterBlock(b []byte, r io.ReadSeeker, pos int64, defs []*PerfCounterDef) (int64, []*PerfCounter) { func parseCounterBlock(b []byte, r io.ReadSeeker, pos int64, defs []*PerfCounterDef) (int64, []*PerfCounter, error) {
r.Seek(pos, io.SeekStart) _, err := r.Seek(pos, io.SeekStart)
if err != nil {
return 0, nil, err
}
block := new(perfCounterBlock) block := new(perfCounterBlock)
block.BinaryReadFrom(r) err = block.BinaryReadFrom(r)
if err != nil {
return 0, nil, err
}
counters := make([]*PerfCounter, len(defs)) counters := make([]*PerfCounter, len(defs))
@@ -419,7 +449,7 @@ func parseCounterBlock(b []byte, r io.ReadSeeker, pos int64, defs []*PerfCounter
} }
} }
return int64(block.ByteLength), counters return int64(block.ByteLength), counters, nil
} }
func convertCounterValue(counterDef *perfCounterDefinition, buffer []byte, valueOffset int64) (value int64) { func convertCounterValue(counterDef *perfCounterDefinition, buffer []byte, valueOffset int64) (value int64) {