Merge pull request #1347 from atoulme/windows_errors

Return all Windows partitions
pull/1362/head
shirou 3 years ago committed by GitHub
commit 769daafb9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -80,6 +80,9 @@ func UsageWithContext(ctx context.Context, path string) (*UsageStat, error) {
} }
func PartitionsWithContext(ctx context.Context, all bool) ([]PartitionStat, error) { func PartitionsWithContext(ctx context.Context, all bool) ([]PartitionStat, error) {
warnings := common.Warnings{
Verbose: true,
}
var ret []PartitionStat var ret []PartitionStat
lpBuffer := make([]byte, 254) lpBuffer := make([]byte, 254)
diskret, _, err := procGetLogicalDriveStringsW.Call( diskret, _, err := procGetLogicalDriveStringsW.Call(
@ -94,7 +97,9 @@ func PartitionsWithContext(ctx context.Context, all bool) ([]PartitionStat, erro
typepath, _ := windows.UTF16PtrFromString(path) typepath, _ := windows.UTF16PtrFromString(path)
typeret, _, _ := procGetDriveType.Call(uintptr(unsafe.Pointer(typepath))) typeret, _, _ := procGetDriveType.Call(uintptr(unsafe.Pointer(typepath)))
if typeret == 0 { if typeret == 0 {
return ret, windows.GetLastError() err := windows.GetLastError()
warnings.Add(err)
continue
} }
// 2: DRIVE_REMOVABLE 3: DRIVE_FIXED 4: DRIVE_REMOTE 5: DRIVE_CDROM // 2: DRIVE_REMOVABLE 3: DRIVE_FIXED 4: DRIVE_REMOTE 5: DRIVE_CDROM
@ -118,7 +123,8 @@ func PartitionsWithContext(ctx context.Context, all bool) ([]PartitionStat, erro
if typeret == 5 || typeret == 2 { if typeret == 5 || typeret == 2 {
continue // device is not ready will happen if there is no disk in the drive continue // device is not ready will happen if there is no disk in the drive
} }
return ret, err warnings.Add(err)
continue
} }
opts := []string{"rw"} opts := []string{"rw"}
if lpFileSystemFlags&fileReadOnlyVolume != 0 { if lpFileSystemFlags&fileReadOnlyVolume != 0 {
@ -138,7 +144,7 @@ func PartitionsWithContext(ctx context.Context, all bool) ([]PartitionStat, erro
} }
} }
} }
return ret, nil return ret, warnings.Reference()
} }
func IOCountersWithContext(ctx context.Context, names ...string) (map[string]IOCountersStat, error) { func IOCountersWithContext(ctx context.Context, names ...string) (map[string]IOCountersStat, error) {

@ -395,7 +395,7 @@ func SensorsTemperaturesWithContext(ctx context.Context) ([]TemperatureStat, err
} }
} }
var warns Warnings var warns common.Warnings
if len(files) == 0 { // handle distributions without hwmon, like raspbian #391, parse legacy thermal_zone files if len(files) == 0 { // handle distributions without hwmon, like raspbian #391, parse legacy thermal_zone files
files, err = filepath.Glob(common.HostSys("/class/thermal/thermal_zone*/")) files, err = filepath.Glob(common.HostSys("/class/thermal/thermal_zone*/"))

@ -1,11 +1,10 @@
package host package common
import ( import "fmt"
"fmt"
)
type Warnings struct { type Warnings struct {
List []error List []error
Verbose bool
} }
func (w *Warnings) Add(err error) { func (w *Warnings) Add(err error) {
@ -20,5 +19,12 @@ func (w *Warnings) Reference() error {
} }
func (w *Warnings) Error() string { func (w *Warnings) Error() string {
if w.Verbose {
str := ""
for i, e := range w.List {
str += fmt.Sprintf("\tError %d: %s\n", i, e.Error())
}
return str
}
return fmt.Sprintf("Number of warnings: %v", len(w.List)) return fmt.Sprintf("Number of warnings: %v", len(w.List))
} }
Loading…
Cancel
Save