mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-01-05 16:27:26 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(wifi): do not query all networks

Browse source
This commit is contained in:
Jan De Dobbeleer 2022-09-30 22:03:28 +02:00 committed by Jan De Dobbeleer
parent 4b6b128d74
commit 8a9a022baa
1 changed files with 16 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

69
src/environment/windows_win32.go
View file

@ -203,7 +203,6 @@ func (env *ShellEnvironment) getConnections() []*Connection {
var pIFTable2 *MIN_IF_TABLE2 var pIFTable2 *MIN_IF_TABLE2
_, _, _ = hGetIfTable2.Call(uintptr(unsafe.Pointer(&pIFTable2))) _, _, _ = hGetIfTable2.Call(uintptr(unsafe.Pointer(&pIFTable2)))
SSIDs, _ := env.getAllWifiSSID()
networks := make([]*Connection, 0) networks := make([]*Connection, 0)
for i := 0; i < int(pIFTable2.NumEntries); i++ { for i := 0; i < int(pIFTable2.NumEntries); i++ {
@ -220,11 +219,13 @@ func (env *ShellEnvironment) getConnections() []*Connection {
} }
var connectionType ConnectionType var connectionType ConnectionType
var ssid string
switch networkInterface.Type { switch networkInterface.Type {
case 6: case 6:
connectionType = ETHERNET connectionType = ETHERNET
case 71: case 71:
connectionType = WIFI connectionType = WIFI
ssid = env.getWiFiSSID(networkInterface.InterfaceGUID)
case 237, 234, 244: case 237, 234, 244:
connectionType = CELLULAR connectionType = CELLULAR
} }
@ -243,10 +244,7 @@ func (env *ShellEnvironment) getConnections() []*Connection {
Name: description, // we want a relatable name, alias isn't that Name: description, // we want a relatable name, alias isn't that
TransmitRate: networkInterface.TransmitLinkSpeed, TransmitRate: networkInterface.TransmitLinkSpeed,
ReceiveRate: networkInterface.ReceiveLinkSpeed, ReceiveRate: networkInterface.ReceiveLinkSpeed,
} SSID: ssid,
if SSID, OK := SSIDs[network.Name]; OK {
network.SSID = SSID
} }
networks = append(networks, network) networks = append(networks, network)
@ -322,13 +320,21 @@ type MIB_IF_ROW2 struct { //nolint: revive
OutQLen uint64 OutQLen uint64
} }
func (env *ShellEnvironment) getAllWifiSSID() (map[string]string, error) { var (
wlanapi = syscall.NewLazyDLL("wlanapi.dll")
hWlanOpenHandle = wlanapi.NewProc("WlanOpenHandle")
hWlanCloseHandle = wlanapi.NewProc("WlanCloseHandle")
hWlanQueryInterface = wlanapi.NewProc("WlanQueryInterface")
)
func (env *ShellEnvironment) getWiFiSSID(guid windows.GUID) string {
// Query wifi connection state
var pdwNegotiatedVersion uint32 var pdwNegotiatedVersion uint32
var phClientHandle uint32 var phClientHandle uint32
e, _, err := hWlanOpenHandle.Call(uintptr(uint32(2)), uintptr(unsafe.Pointer(nil)), uintptr(unsafe.Pointer(&pdwNegotiatedVersion)), uintptr(unsafe.Pointer(&phClientHandle))) e, _, err := hWlanOpenHandle.Call(uintptr(uint32(2)), uintptr(unsafe.Pointer(nil)), uintptr(unsafe.Pointer(&pdwNegotiatedVersion)), uintptr(unsafe.Pointer(&phClientHandle)))
if e != 0 { if e != 0 {
env.Log(Error, "getAllWifiSSID", err.Error()) env.Log(Error, "getAllWifiSSID", err.Error())
return nil, err return ""
} }
// defer closing handle // defer closing handle
@ -336,42 +342,11 @@ func (env *ShellEnvironment) getAllWifiSSID() (map[string]string, error) {
_, _, _ = hWlanCloseHandle.Call(uintptr(phClientHandle), uintptr(unsafe.Pointer(nil))) _, _, _ = hWlanCloseHandle.Call(uintptr(phClientHandle), uintptr(unsafe.Pointer(nil)))
}() }()
ssid := make(map[string]string)
// list interfaces
var interfaceList *WLAN_INTERFACE_INFO_LIST
e, _, err = hWlanEnumInterfaces.Call(uintptr(phClientHandle), uintptr(unsafe.Pointer(nil)), uintptr(unsafe.Pointer(&interfaceList)))
if e != 0 {
env.Log(Error, "getAllWifiSSID", err.Error())
return nil, err
}
// use first interface that is connected
numberOfInterfaces := int(interfaceList.dwNumberOfItems)
infoSize := unsafe.Sizeof(interfaceList.InterfaceInfo[0])
for i := 0; i < numberOfInterfaces; i++ {
network := (*WLAN_INTERFACE_INFO)(unsafe.Pointer(uintptr(unsafe.Pointer(&interfaceList.InterfaceInfo[0])) + uintptr(i)*infoSize))
if network.isState == 1 {
wifiInterface := strings.TrimRight(string(utf16.Decode(network.strInterfaceDescription[:])), "\x00")
ssid[wifiInterface] = env.getWiFiSSID(network, phClientHandle)
}
}
return ssid, nil
}
var (
wlanapi = syscall.NewLazyDLL("wlanapi.dll")
hWlanOpenHandle = wlanapi.NewProc("WlanOpenHandle")
hWlanCloseHandle = wlanapi.NewProc("WlanCloseHandle")
hWlanEnumInterfaces = wlanapi.NewProc("WlanEnumInterfaces")
hWlanQueryInterface = wlanapi.NewProc("WlanQueryInterface")
)
func (env *ShellEnvironment) getWiFiSSID(network *WLAN_INTERFACE_INFO, clientHandle uint32) string {
// Query wifi connection state
var dataSize uint16 var dataSize uint16
var wlanAttr *WLAN_CONNECTION_ATTRIBUTES var wlanAttr *WLAN_CONNECTION_ATTRIBUTES
e, _, _ := hWlanQueryInterface.Call(uintptr(clientHandle),
uintptr(unsafe.Pointer(&network.InterfaceGuid)), e, _, _ = hWlanQueryInterface.Call(uintptr(phClientHandle),
uintptr(unsafe.Pointer(&guid)),
uintptr(7), // wlan_intf_opcode_current_connection uintptr(7), // wlan_intf_opcode_current_connection
uintptr(unsafe.Pointer(nil)), uintptr(unsafe.Pointer(nil)),
uintptr(unsafe.Pointer(&dataSize)), uintptr(unsafe.Pointer(&dataSize)),
@ -389,18 +364,6 @@ func (env *ShellEnvironment) getWiFiSSID(network *WLAN_INTERFACE_INFO, clientHan
return string(ssid.ucSSID[0:ssid.uSSIDLength]) return string(ssid.ucSSID[0:ssid.uSSIDLength])
} }
type WLAN_INTERFACE_INFO_LIST struct { //nolint: revive
dwNumberOfItems uint32
dwIndex uint32 //nolint: unused
InterfaceInfo [256]WLAN_INTERFACE_INFO
}
type WLAN_INTERFACE_INFO struct { //nolint: revive
InterfaceGuid syscall.GUID //nolint: revive
strInterfaceDescription [256]uint16
isState uint32
}
type WLAN_CONNECTION_ATTRIBUTES struct { //nolint: revive type WLAN_CONNECTION_ATTRIBUTES struct { //nolint: revive
isState uint32 //nolint: unused isState uint32 //nolint: unused
wlanConnectionMode uint32 //nolint: unused wlanConnectionMode uint32 //nolint: unused