Add WLAN Integration

omnibus/config/templates/datadog-agent-mac-app/Info.plist.erb

@@ -32,6 +32,8 @@
     <string>Copyright (c) 2009-<%= year %>, Datadog Inc.</string>
     <key>NSPrincipalClass</key>
     <string>NSApplication</string>
+    <key>NSLocationUsageDescription</key>
+    <string>Datadog needs access to location services to retrieve Wi-Fi network information.</string>
     <key>NSLocalNetworkUsageDescription</key>
     <string>Datadog needs access to local network services to function"</string>
 </dict>

pkg/collector/corechecks/net/wlan/wlan.go

@@ -0,0 +1,121 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2016-present Datadog, Inc.
+
+//nolint:revive // TODO(PLINT) Fix revive linter
+package wlan
+
+import (
+	"strings"
+	"time"
+
+	"github.com/DataDog/datadog-agent/pkg/collector/check"
+	core "github.com/DataDog/datadog-agent/pkg/collector/corechecks"
+	"github.com/DataDog/datadog-agent/pkg/util/log"
+	"github.com/DataDog/datadog-agent/pkg/util/option"
+)
+
+const (
+	CheckName                    = "wlan"
+	defaultMinCollectionInterval = 15
+)
+
+var getWiFiInfo = GetWiFiInfo
+
+// WiFiInfo contains information about the WiFi connection as defined in wlan_darwin.h
+type WiFiInfo struct {
+	Rssi            int
+	Ssid            string
+	Bssid           string
+	Channel         int
+	Noise           int
+	TransmitRate    float64
+	HardwareAddress string
+	// See: https://developer.apple.com/documentation/corewlan/cwphymode?language=objc
+	ActivePHYMode int
+}
+
+// WLANCheck monitors the status of the WLAN interface
+type WLANCheck struct {
+	core.CheckBase
+	lastChannelID int
+	lastBSSID     string
+	lastSSID      string
+	isWarmedUp    bool
+}
+
+func (c *WLANCheck) String() string {
+	return "wlan"
+}
+
+// Run runs the check
+func (c *WLANCheck) Run() error {
+	sender, err := c.GetSender()
+	if err != nil {
+		return err
+	}
+	wifiInfo, err := getWiFiInfo()
+	if err != nil {
+		log.Error(err)
+		return err
+	}
+
+	if wifiInfo.ActivePHYMode == 0 {
+		log.Warn("No active Wi-Fi interface detected: ActivePHYMode is none.")
+		return nil
+	}
+
+	ssid := wifiInfo.Ssid
+	if ssid == "" {
+		ssid = "unknown"
+	}
+	bssid := wifiInfo.Bssid
+	if bssid == "" {
+		bssid = "unknown"
+	}
+	hardwareAddress := strings.ToLower(strings.Replace(wifiInfo.HardwareAddress, " ", "_", -1))
+
+	tags := []string{}
+	tags = append(tags, "ssid:"+ssid)
+	tags = append(tags, "bssid:"+bssid)
+	tags = append(tags, "mac_address:"+hardwareAddress)
+
+	sender.Gauge("wlan.rssi", float64(wifiInfo.Rssi), "", tags)
+	sender.Gauge("wlan.noise", float64(wifiInfo.Noise), "", tags)
+	sender.Gauge("wlan.transmit_rate", float64(wifiInfo.TransmitRate), "", tags)
+
+	// channel swap events
+	if c.isWarmedUp && c.lastChannelID != wifiInfo.Channel {
+		sender.Count("wlan.channel_swap_events", 1.0, "", tags)
+	} else {
+		sender.Count("wlan.channel_swap_events", 0.0, "", tags)
+	}
+
+	// roaming events / ssid swap events
+	if c.isWarmedUp && c.lastBSSID != "" && c.lastSSID != "" && c.lastBSSID == wifiInfo.Bssid && c.lastSSID != wifiInfo.Ssid {
+		sender.Count("wlan.roaming_events", 1.0, "", tags)
+	} else {
+		sender.Count("wlan.roaming_events", 0.0, "", tags)
+	}
+
+	// update last values
+	c.lastChannelID = wifiInfo.Channel
+	c.lastBSSID = wifiInfo.Bssid
+	c.lastSSID = wifiInfo.Ssid
+	c.isWarmedUp = true
+
+	sender.Commit()
+	return nil
+}
+
+// Factory creates a new check factory
+func Factory() option.Option[func() check.Check] {
+	return option.New(newCheck)
+}
+
+func newCheck() check.Check {
+	return &WLANCheck{
+		CheckBase: core.NewCheckBaseWithInterval(CheckName, time.Duration(defaultMinCollectionInterval)*time.Second),
+	}
+}

pkg/collector/corechecks/net/wlan/wlan_darwin.go

@@ -0,0 +1,36 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2016-present Datadog, Inc.
+
+//go:build darwin
+
+//nolint:revive // TODO(PLINT) Fix revive linter
+package wlan
+
+/*
+#cgo CFLAGS: -I .
+#cgo LDFLAGS: -framework CoreWLAN -framework CoreLocation -framework Foundation
+#include "wlan_darwin.h"
+*/
+import "C"
+import (
+	"github.com/DataDog/datadog-agent/pkg/util/log"
+)
+
+func GetWiFiInfo() (WiFiInfo, error) {
+	C.InitLocationServices()
+	log.Info("Initialized Location Manager")
+
+	info := C.GetWiFiInformation()
+	return WiFiInfo{
+		Rssi:            int(info.rssi),
+		Ssid:            C.GoString(info.ssid),
+		Bssid:           C.GoString(info.bssid),
+		Channel:         int(info.channel),
+		Noise:           int(info.noise),
+		TransmitRate:    float64(info.transmitRate),
+		HardwareAddress: C.GoString(info.hardwareAddress),
+		ActivePHYMode:   int(info.activePHYMode),
+	}, nil
+}

pkg/collector/corechecks/net/wlan/wlan_darwin.h

@@ -0,0 +1,18 @@
+#ifndef WLAN_DARWIN_H
+#define WLAN_DARWIN_H
+
+typedef struct {
+    int rssi;
+    const char *ssid;
+    const char *bssid;
+    int channel;
+    int noise;
+    double transmitRate;
+    const char *hardwareAddress;
+    int activePHYMode;
+} WiFiInfo;
+
+WiFiInfo GetWiFiInformation();
+void InitLocationServices();
+
+#endif

pkg/collector/corechecks/net/wlan/wlan_darwin.m

@@ -0,0 +1,99 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2016-present Datadog, Inc.
+
+#include <Foundation/Foundation.h>
+#include <CoreWLAN/CoreWLAN.h>
+#include <CoreLocation/CoreLocation.h>
+#include "wlan_darwin.h"
+
+
+@interface LocationManager : NSObject <CLLocationManagerDelegate>
+@property (nonatomic, strong) CLLocationManager *locationManager;
+@end
+
+@implementation LocationManager
+
+- (instancetype)init {
+    self = [super init];
+    if (self) {
+        [self setupLocationServices];
+    }
+    return self;
+}
+
+- (void)setupLocationServices {
+    self.locationManager = [[CLLocationManager alloc] init];
+    self.locationManager.delegate = self;
+
+    if (@available(macOS 11.0, *)) {
+        if (self.locationManager.authorizationStatus == kCLAuthorizationStatusNotDetermined) {
+            [self.locationManager requestWhenInUseAuthorization];
+        }
+    } else {
+        NSLog(@"Location services not available on this OS version");
+    }
+}
+
+#pragma mark - CLLocationManagerDelegate
+
+- (void)locationManagerDidChangeAuthorization:(CLLocationManager *)manager {
+    if (@available(macOS 11.0, *)) {
+        CLAuthorizationStatus status = manager.authorizationStatus;
+
+        switch (status) {
+            case kCLAuthorizationStatusAuthorizedAlways:
+                NSLog(@"Location services authorized!");
+                [self.locationManager startUpdatingLocation];
+                break;
+
+            case kCLAuthorizationStatusDenied:
+                NSLog(@"Location services denied by user");
+                break;
+
+            case kCLAuthorizationStatusRestricted:
+                NSLog(@"Location services restricted");
+                break;
+            default:
+                NSLog(@"Location services status undetermined: %d", status);
+                break;
+        }
+    } else {
+        NSLog(@"Location services not available on this OS version");
+    }
+}
+
+- (void)locationManager:(CLLocationManager *)manager didUpdateLocations:(NSArray<CLLocation *> *)locations {
+    CLLocation *location = [locations lastObject];
+    NSLog(@"Location updated: %@", location);
+}
+
+- (void)locationManager:(CLLocationManager *)manager didFailWithError:(NSError *)error {
+    NSLog(@"Location update failed with error: %@", error.localizedDescription);
+}
+
+@end
+
+// Wrapper function to start location updates
+void InitLocationServices() {
+    LocationManager *locationManager = [[LocationManager alloc] init];
+}
+
+// GetWiFiInformation return wifi data
+WiFiInfo GetWiFiInformation() {
+    CWInterface *wifiInterface = [[CWWiFiClient sharedWiFiClient] interface];
+
+    WiFiInfo info;
+
+    info.rssi = (int)wifiInterface.rssiValue;
+    info.ssid = [[wifiInterface ssid] UTF8String];
+    info.bssid = [[wifiInterface bssid] UTF8String];
+    info.channel = (int)wifiInterface.wlanChannel.channelNumber;
+    info.noise = (int)wifiInterface.noiseMeasurement;
+    info.transmitRate = wifiInterface.transmitRate;
+    info.hardwareAddress = [[wifiInterface hardwareAddress] UTF8String];
+    info.activePHYMode = (int)wifiInterface.activePHYMode;
+
+    return info;
+}

pkg/collector/corechecks/net/wlan/wlan_nodarwin.go

@@ -0,0 +1,15 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2016-present Datadog, Inc.
+
+//go:build !darwin
+
+//nolint:revive // TODO(PLINT) Fix revive linter
+package wlan
+
+import "fmt"
+
+func GetWiFiInfo() (WiFiInfo, error) {
+	return WiFiInfo{}, fmt.Errorf("wifi info only supported on macOS")
+}