如何使用golang获取Windows上所有驱动器的列表?
我想让我的程序遍历Windows系统上的所有驱动器,并搜索特定的文件类型。现在,我可以运行该程序,并向其传递一个驱动器号作为开始,但我希望它能自动搜索所有驱动器。目前,我需要这样做:如何使用golang获取Windows上所有驱动器的列表?,go,Go,我想让我的程序遍历Windows系统上的所有驱动器,并搜索特定的文件类型。现在,我可以运行该程序,并向其传递一个驱动器号作为开始,但我希望它能自动搜索所有驱动器。目前,我需要这样做: C:\> program.exe C: C:\> program.exe D: C:\> program.exe E: 我希望程序得到所有驱动器的列表,并在用户不必指定驱动器号的情况下遍历所有驱动器。这可以用围棋吗 与此问题类似,但使用Go代替C。最简单的方法是编写自己的函数,并尝试打开Volk
C:\> program.exe C:
C:\> program.exe D:
C:\> program.exe E:
我希望程序得到所有驱动器的列表,并在用户不必指定驱动器号的情况下遍历所有驱动器。这可以用围棋吗
与此问题类似,但使用Go代替C。最简单的方法是编写自己的函数,并尝试打开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
}
您可以根据位图选择函数并匹配字母
比如:
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
}
或者您可以直接调用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
}
}
您可以使用lib:
您可以使用
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)
}
}
为什么不
对于uuz,drive-in“ABCDEFGHIJKLMNOPQRSTUVWXYZ”{/*在驱动器上尝试一些操作并在失败时继续*/}
?使用cgo从Go使用C API。@Volker我在寻找一种更好的方法?@rightfold谢谢,但我更喜欢使用标准的Go包,而不是C.@roartechs,这是最好的选择。最后一个选项是使用包syscall。祝你好运接受这个答案,因为看起来没有更好的方法。我怀疑这可能会导致某些系统出现延迟,例如等待DVD驱动器启动。我记得当没有插入软盘时,点击A:驱动器会很慢……这不是会留下一堆打开的文件需要Close()
ed吗?调用的第一个参数应该是unitptr(bufferSize)
而不是uintpttr(不安全的指针(&bufferSize))
。
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)
}
}