fix: fixed wifi support on macOS (fixes pr #1100, fixes #1090, fixes #1076)

This commit is contained in:
Simone Margaritelli 2024-08-08 12:10:28 +02:00
commit 0dc5f66e27
2 changed files with 104 additions and 37 deletions

View file

@ -1,60 +1,55 @@
package network
/*
#cgo LDFLAGS: -framework CoreWLAN -framework Foundation
#include <stdbool.h>
#include <stdlib.h>
const char *GetSupportedFrequencies(const char *iface);
bool SetInterfaceChannel(const char *iface, int channel);
*/
import "C"
import (
"fmt"
"encoding/json"
"errors"
"net"
"regexp"
"strconv"
"github.com/bettercap/bettercap/core"
"github.com/evilsocket/islazy/str"
"unsafe"
)
const airPortPath = "/System/Library/PrivateFrameworks/Apple80211.framework/Versions/Current/Resources/airport"
var WiFiChannelParser = regexp.MustCompile(`(?m)^.*Supported Channels: (.*)$`)
// see Windows version to understand why ....
func getInterfaceName(iface net.Interface) string {
return iface.Name
}
func SetInterfaceChannel(iface string, channel int) error {
curr := GetInterfaceChannel(iface)
// the interface is already on this channel
if curr == channel {
return nil
}
cIface := C.CString(iface)
defer C.free(unsafe.Pointer(cIface))
_, err := core.Exec(airPortPath, []string{iface, fmt.Sprintf("-c%d", channel)})
if err != nil {
return err
success := C.SetInterfaceChannel(cIface, C.int(channel))
if !success {
return errors.New("failed to set interface channel")
}
SetInterfaceCurrentChannel(iface, channel)
return nil
}
func getFrequenciesFromChannels(output string) ([]int, error) {
freqs := make([]int, 0)
if output != "" {
if matches := WiFiChannelParser.FindStringSubmatch(output); len(matches) == 2 {
for _, channel := range str.Comma(matches[1]) {
re := regexp.MustCompile(`\d+`)
if channel, err := strconv.Atoi(re.FindString(channel)); err == nil {
freqs = append(freqs, Dot11Chan2Freq(channel))
}
}
}
}
return freqs, nil
}
func GetSupportedFrequencies(iface string) ([]int, error) {
out, err := core.Exec("system_profiler", []string{"SPAirPortDataType"})
cIface := C.CString(iface)
defer C.free(unsafe.Pointer(cIface))
cFrequencies := C.GetSupportedFrequencies(cIface)
if cFrequencies == nil {
return nil, errors.New("failed to get supported frequencies")
}
defer C.free(unsafe.Pointer(cFrequencies))
frequenciesStr := C.GoString(cFrequencies)
var frequencies []int
err := json.Unmarshal([]byte(frequenciesStr), &frequencies)
if err != nil {
return nil, err
}
return getFrequenciesFromChannels(out)
return frequencies, nil
}