如何使用 golang 获取 Windows 上所有驱动器的列表？答案

【问题标题】：How can I get a listing of all drives on Windows using golang?如何使用 golang 获取 Windows 上所有驱动器的列表？
【发布时间】：2023-04-06 08:09:01
【问题描述】：

我希望我的程序遍历 Windows 系统上的所有驱动器并搜索特定的文件类型。现在，我可以运行该程序并传递一个驱动器号以开始，但我希望它自动搜索所有驱动器。目前，我需要做这样的事情：

C:\> program.exe C:
C:\> program.exe D:
C:\> program.exe E:

我希望程序获取所有驱动器的列表并遍历所有驱动器，而无需用户指定驱动器号。这可以使用 Go 实现吗？

类似于这个问题Listing All Physical Drives (Windows)，但使用 Go 而不是 C。

【问题讨论】：

为什么不for _, drive in "ABCDEFGHIJKLMNOPQRSTUVWXYZ" { /* try some op on drive and continue on failure */ }？
使用 cgo 从 Go 中使用 C API。
@Volker 我在寻找更好的方法？
@rightfold 谢谢，但我更喜欢使用标准的 go 包而不是 C。
@roartechs 这真是太好了。您的最后一个选择是使用包系统调用。祝你好运！

标签： go

【解决方案1】：

你可以call函数GetLogicalDrives根据位图匹配字母。

类似：

package main

import (
    "fmt"
    "syscall"
)

func main() {

    kernel32, _ := syscall.LoadLibrary("kernel32.dll")
    getLogicalDrivesHandle, _ := syscall.GetProcAddress(kernel32, "GetLogicalDrives")

    var drives []string

    if ret, _, callErr := syscall.Syscall(uintptr(getLogicalDrivesHandle), 0, 0, 0, 0); callErr != 0 {
        // handle error
    } else {
        drives = bitsToDrives(uint32(ret))
    }

    fmt.Printf("%v", drives)

}

func bitsToDrives(bitMap uint32) (drives []string) {
    availableDrives := []string{"A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"}

    for i := range availableDrives {
        if bitMap & 1 == 1 {
            drives = append(drives, availableDrives[i])
        }
        bitMap >>= 1
    }

    return
}

【讨论】：

【解决方案2】：

您可以使用gopsutil 库：

package main

import (
    "fmt"
    "github.com/shirou/gopsutil/disk"
)

func main() {
    partitions, _ := disk.Partitions(false)
    for _, partition := range partitions {
        fmt.Println(partition.Mountpoint)
    }
}

【讨论】：

【解决方案3】：

最简单的方法是编写自己的函数，尝试打开Volker提到的“驱动器”文件夹。

import "os"

func getdrives() (r []string){
    for _, drive := range "ABCDEFGHIJKLMNOPQRSTUVWXYZ"{
        f, err := os.Open(string(drive)+":\\")
        if err == nil {
            r = append(r, string(drive))
            f.Close()
        }
    }
    return
}

【讨论】：

接受了这个答案，因为它看起来没有更好的方法。
我怀疑这可能会导致某些系统出现延迟，例如等待 DVD 驱动器启动。我记得在没有插入软盘时单击 A: 驱动器会很慢...
这不会留下一堆需要Close()ed 的打开文件吗？

【解决方案4】：

或者你可以直接调用GetLogicalDriveStrings：

package main

import (
    "fmt"
    "syscall"
    "unsafe"
)

func main() {
    kernel32, err := syscall.LoadDLL("kernel32.dll")
    getLogicalDriveStringsHandle, err := kernel32.FindProc("GetLogicalDriveStringsA")

    buffer := [1024]byte{}
    bufferSize := uint32(len(buffer))
    drives := []string{}

    hr, _, _ := getLogicalDriveStringsHandle.Call(uintptr(unsafe.Pointer(&bufferSize)), uintptr(unsafe.Pointer(&buffer)))
    if hr == 0 {
        fmt.Print("There was an error")
    } else {
        // Deal with the buffer, you need to split it
    }
}

【讨论】：

调用的第一个参数应该是unitptr(bufferSize) 而不是uintptr(unsafe.Pointer(&bufferSize))。

【解决方案5】：

您可以使用windows 包：

package main

import (
   "golang.org/x/sys/windows"
   "strings"
   "unicode/utf16"
)

func drives() ([]string, error) {
   n, e := windows.GetLogicalDriveStrings(0, nil)
   if e != nil { return nil, e }
   a := make([]uint16, n)
   windows.GetLogicalDriveStrings(n, &a[0])
   s := string(utf16.Decode(a))
   return strings.Split(strings.TrimRight(s, "\x00"), "\x00"), nil
}

func main() {
   a, e := drives()
   if e != nil {
      panic(e)
   }
   for _, s := range a {
      println(s)
   }
}

https://pkg.go.dev/golang.org/x/sys/windows#GetLogicalDriveStrings

【讨论】：