1 files changed, 110 insertions, 0 deletions
diff --git a/wlan_mac/google_wlan_mac.c b/wlan_mac/google_wlan_mac.c
new file mode 100644
index 0000000..ee37056
--- /dev/null
+++ b/wlan_mac/google_wlan_mac.c
@@ -0,0 +1,110 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * Copyright (c) 2021, The Linux Foundation. All rights reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 and
+ * only version 2 as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ */
+
+#include <asm/setup.h>
+#include <linux/kernel.h>
+#include <linux/module.h>
+#include <linux/platform_device.h>
+#include <linux/string.h>
+#include <linux/export.h>
+#include <linux/of.h>
+#include <linux/random.h>
+#include <linux/slab.h>
+#include <linux/uaccess.h>
+#include "cnss_utils.h"
+
+#define CDB_PATH "/chosen/config"
+#define WIFI_MAC "wlan_mac1"
+
+static void set_wifi_mac(void)
+{
+	u8 mac[6] = {0};
+	unsigned int size;
+	unsigned char *mac_addr = NULL;
+	struct device_node *node;
+	unsigned int mac_found = 0;
+
+	node = of_find_node_by_path(CDB_PATH);
+	if (!node) {
+		pr_err("[WLAN] CDB Node not created under %s", CDB_PATH);
+	} else {
+		mac_addr = (unsigned char *)
+				of_get_property(node, WIFI_MAC, &size);
+	}
+
+	/* In case Missing Provisioned MAC Address, exit with error */
+	if (!mac_addr) {
+		pr_err("[WLAN] Missing Provisioned MAC address\n");
+		return;
+	}
+
+	/* Start decoding MAC Address
+	 * Note that 2 formats are supported for now
+	 * AA:BB:CC:DD:EE:FF (with separating colons) and
+	 * AABBCCDDEEFF (without separating colons) */
+	if (sscanf(mac_addr,
+		   "%02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx",
+		   &mac[0], &mac[1], &mac[2], &mac[3], &mac[4],
+		   &mac[5]) == 6) {
+		mac_found = 1;
+	} else if (sscanf(mac_addr,
+			  "%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx",
+			  &mac[0], &mac[1], &mac[2], &mac[3], &mac[4],
+			  &mac[5]) == 6) {
+		mac_found = 1;
+	}
+
+	/* Make sure Address decoding succeeds */
+	if (!mac_found) {
+		pr_err("[WLAN] Invalid format for Provisioned MAC Address\n");
+		return;
+	}
+
+	/* Make sure Provisioned MAC Address is globally Administered */
+	if (mac[0] & 2) {
+		pr_err("[WLAN] Invalid Provisioned MAC Address\n");
+		return;
+	}
+
+	/* Send provisioned MAC Address to the platform driver */
+	if (cnss_utils_set_wlan_mac_address(mac, sizeof(mac)) != 0) {
+		pr_err("[WLAN] set wlan mac address failed\n");
+		return;
+	}
+
+	/* Now derive the derived mac address
+	 * by flipping the locally administred bit */
+	mac[0] = mac[0] | 2;
+
+	if (cnss_utils_set_wlan_derived_mac_address(mac, sizeof(mac)) != 0) {
+		pr_err("[WLAN] set wlan derived  mac address failed\n");
+		return;
+	}
+
+	return;
+}
+
+static int __init wifi_mac_init(void)
+{
+	set_wifi_mac();
+
+	return 0;
+}
+
+late_initcall(wifi_mac_init);
+
+MODULE_AUTHOR("Google");
+MODULE_DESCRIPTION("Google set wlan mac");
+MODULE_LICENSE("GPL v2");
+MODULE_SOFTDEP("pre: cnss2");