mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-03-05 20:49:04 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(battery): validate returned buffer size on Windows

Browse source 
resolves #5647
This commit is contained in:
Jan De Dobbeleer 2024-09-20 08:35:15 +02:00 committed by Jan De Dobbeleer
parent 9cf42b56d5
commit 1cb66f6af9
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/runtime/battery/battery_windows.go
View file

@ -185,6 +185,10 @@ func systemGet(idx int) (*battery, error) {
return nil, errno return nil, errno
} }
if cbRequired == 0 {
return nil, errors.New("no buffer information returned")
}
// The god damn struct with ANYSIZE_ARRAY of utf16 in it is crazy. // The god damn struct with ANYSIZE_ARRAY of utf16 in it is crazy.
// So... let's emulate it with array of uint16 ;-D. // So... let's emulate it with array of uint16 ;-D.
// Keep in mind that the first two elements are actually cbSize. // Keep in mind that the first two elements are actually cbSize.
@ -250,7 +254,7 @@ func systemGet(idx int) (*battery, error) {
} }
if bqi.BatteryTag == 0 { if bqi.BatteryTag == 0 {
return nil, errors.New("BatteryTag not returned") return nil, errors.New("battery tag not returned")
} }
b := &battery{} b := &battery{}
@ -274,6 +278,7 @@ func systemGet(idx int) (*battery, error) {
b.Full = float64(bi.FullChargedCapacity) b.Full = float64(bi.FullChargedCapacity)
bws := batteryWaitStatus{BatteryTag: bqi.BatteryTag} bws := batteryWaitStatus{BatteryTag: bqi.BatteryTag}
var bs batteryStatus var bs batteryStatus
err = windows.DeviceIoControl( err = windows.DeviceIoControl(
handle, handle,