snow/wireguard-go
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

wintun: also check for numbered suffix and friendly name

Browse source
This commit is contained in:
Jason A. Donenfeld 2019-08-28 08:08:07 -06:00
parent 7305b4ce93
commit 3bf3322b2c
1 changed files with 22 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
tun/wintun/wintun_windows.go
View file

@ -84,6 +84,14 @@ func makeWintun(deviceInfoSet setupapi.DevInfo, deviceInfoData *setupapi.DevInfo
}, nil }, nil
} }
func removeNumberedSuffix(ifname string) string {
removed := strings.TrimRight(ifname, "0123456789")
if removed != ifname && len(removed) > 1 && removed[len(removed)-1] == ' ' {
return removed[:len(removed)-1]
}
return ifname
}
// GetInterface finds a Wintun interface by its name. This function returns // GetInterface finds a Wintun interface by its name. This function returns
// the interface if found, or windows.ERROR_OBJECT_NOT_FOUND otherwise. If // the interface if found, or windows.ERROR_OBJECT_NOT_FOUND otherwise. If
// the interface is found but not a Wintun-class, this function returns // the interface is found but not a Wintun-class, this function returns
@ -123,14 +131,9 @@ func GetInterface(ifname string) (*Wintun, error) {
continue continue
} }
ifname2 = strings.ToLower(ifname2) ifname2 = strings.ToLower(ifname2)
ifname3 := strings.TrimRight(ifname2, "0123456789") ifname3 := removeNumberedSuffix(ifname2)
if ifname3 != ifname2 && len(ifname3) > 1 && ifname3[len(ifname3)-1] == ' ' {
ifname3 = ifname3[:len(ifname3)-1]
} else {
ifname3 = ""
}
if ifname == ifname2 || (len(ifname3) > 0 && ifname == ifname3) { if ifname == ifname2 || ifname == ifname3 {
err = devInfoList.BuildDriverInfoList(deviceData, setupapi.SPDIT_COMPATDRIVER) err = devInfoList.BuildDriverInfoList(deviceData, setupapi.SPDIT_COMPATDRIVER)
if err != nil { if err != nil {
return nil, fmt.Errorf("SetupDiBuildDriverInfoList failed: %v", err) return nil, fmt.Errorf("SetupDiBuildDriverInfoList failed: %v", err)
@ -472,14 +475,24 @@ func DeleteMatchingInterfaces(matches func(wintun *Wintun) bool) (deviceInstance
if !isWintun { if !isWintun {
continue continue
} }
deviceDescVal, err := devInfoList.DeviceRegistryProperty(deviceData, setupapi.SPDRP_DEVICEDESC) deviceDescVal, err := devInfoList.DeviceRegistryProperty(deviceData, setupapi.SPDRP_DEVICEDESC)
if err != nil { if err != nil {
errors = append(errors, fmt.Errorf("DeviceRegistryPropertyString(SPDRP_DEVICEDESC) failed: %v", err)) errors = append(errors, fmt.Errorf("DeviceRegistryPropertyString(SPDRP_DEVICEDESC) failed: %v", err))
continue continue
} }
if deviceDesc, ok := deviceDescVal.(string); !ok || deviceDesc != deviceTypeName { deviceDesc, _ := deviceDescVal.(string)
friendlyNameVal, err := devInfoList.DeviceRegistryProperty(deviceData, setupapi.SPDRP_FRIENDLYNAME)
if err != nil {
errors = append(errors, fmt.Errorf("DeviceRegistryPropertyString(SPDRP_FRIENDLYNAME) failed: %v", err))
continue continue
} }
friendlyName, _ := friendlyNameVal.(string)
if friendlyName != deviceTypeName && deviceDesc != deviceTypeName &&
removeNumberedSuffix(friendlyName) != deviceTypeName && removeNumberedSuffix(deviceDesc) != deviceTypeName {
continue
}
wintun, err := makeWintun(devInfoList, deviceData) wintun, err := makeWintun(devInfoList, deviceData)
if err != nil { if err != nil {
errors = append(errors, fmt.Errorf("makeWintun failed: %v", err)) errors = append(errors, fmt.Errorf("makeWintun failed: %v", err))
@ -564,7 +577,7 @@ setloop:
if err == windows.ERROR_DUP_NAME { if err == windows.ERROR_DUP_NAME {
duplicateGuid, err2 := iphlpapi.InterfaceGUIDFromAlias(availableIfname) duplicateGuid, err2 := iphlpapi.InterfaceGUIDFromAlias(availableIfname)
if err2 == nil { if err2 == nil {
for j := 0; j < 1000; j++ { for j := 0; j < maxSuffix; j++ {
proposal := fmt.Sprintf("%s %d", ifname, j+1) proposal := fmt.Sprintf("%s %d", ifname, j+1)
if proposal == availableIfname { if proposal == availableIfname {
continue continue