fix: macos

This commit is contained in:
Luke B. Silver 2024-05-24 22:04:46 +00:00
commit 71822229a0
2 changed files with 82 additions and 38 deletions

49
network/corewlan_bridge.m Normal file
View file

@ -0,0 +1,49 @@
#import <Foundation/Foundation.h>
#import <CoreWLAN/CoreWLAN.h>
const char *GetSupportedFrequencies(const char *iface) {
@autoreleasepool {
NSString *interfaceName = [NSString stringWithUTF8String:iface];
CWInterface *interface = [CWInterface interfaceWithName:interfaceName];
if (!interface) {
return NULL;
}
NSSet *supportedChannels = [interface supportedWLANChannels];
NSMutableArray *frequencies = [NSMutableArray arrayWithCapacity:[supportedChannels count]];
for (CWChannel *channel in supportedChannels) {
[frequencies addObject:@(channel.frequency)];
}
NSError *error = nil;
NSData *jsonData = [NSJSONSerialization dataWithJSONObject:frequencies options:0 error:&error];
if (!jsonData) {
NSLog(@"Failed to serialize frequencies: %@", error);
return NULL;
}
NSString *jsonString = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
return strdup([jsonString UTF8String]);
}
}
bool SetInterfaceChannel(const char *iface, int channel) {
@autoreleasepool {
NSString *interfaceName = [NSString stringWithUTF8String:iface];
CWInterface *interface = [CWInterface interfaceWithName:interfaceName];
if (!interface) {
return false;
}
NSError *error = nil;
CWChannel *newChannel = [[CWChannel alloc] initWithChannelNumber:channel channelWidth:kCWChannelWidthUnknown];
[interface setWLANChannel:newChannel error:&error];
if (error) {
NSLog(@"Failed to set channel: %@", error);
return false;
}
return true;
}
}

View file

@ -1,60 +1,55 @@
package network package network
/*
#cgo LDFLAGS: -framework CoreWLAN
#include <stdbool.h>
#include <stdlib.h>
const char *GetSupportedFrequencies(const char *iface);
bool SetInterfaceChannel(const char *iface, int channel);
*/
import "C"
import ( import (
"fmt" "encoding/json"
"errors"
"net" "net"
"regexp" "unsafe"
"strconv"
"github.com/bettercap/bettercap/core"
"github.com/evilsocket/islazy/str"
) )
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 { func getInterfaceName(iface net.Interface) string {
return iface.Name return iface.Name
} }
func SetInterfaceChannel(iface string, channel int) error { func SetInterfaceChannel(iface string, channel int) error {
curr := GetInterfaceChannel(iface) cIface := C.CString(iface)
// the interface is already on this channel defer C.free(unsafe.Pointer(cIface))
if curr == channel {
return nil
}
_, err := core.Exec(airPortPath, []string{iface, fmt.Sprintf("-c%d", channel)}) success := C.SetInterfaceChannel(cIface, C.int(channel))
if err != nil { if !success {
return err return errors.New("failed to set interface channel")
} }
SetInterfaceCurrentChannel(iface, channel) SetInterfaceCurrentChannel(iface, channel)
return nil 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) { 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 { if err != nil {
return nil, err return nil, err
} }
return getFrequenciesFromChannels(out)
} return frequencies, nil
}