/*
 * Copyright (C) 2012 ROCKCHIP, Inc.
 *
 * This software is licensed under the terms of the GNU General Public
 * License version 2, as published by the Free Software Foundation, and
 * may be copied, distributed, and modified under those terms.
 *
 * 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.
 *
 */
/* Rock-chips rfkill driver for wifi
*/

#include <linux/kernel.h>
#include <linux/platform_device.h>
#include <linux/module.h>
#include <linux/rfkill.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/regulator/consumer.h>
#include <linux/delay.h>
#include <linux/rfkill-wlan.h>
#include <linux/rfkill-bt.h>
#include <linux/wakelock.h>
#include <linux/interrupt.h>
#include <asm/irq.h>
#include <linux/suspend.h>
#include <linux/proc_fs.h>
#include <linux/uaccess.h>
#include <linux/gpio.h>
#include <dt-bindings/gpio/gpio.h>
#include <linux/skbuff.h>
#include <linux/fb.h>
#include <linux/rockchip/grf.h>
#include <linux/regmap.h>
#include <linux/mfd/syscon.h>
#include <linux/mmc/host.h>
#ifdef CONFIG_OF
#include <linux/of.h>
#include <linux/of_device.h>
#include <linux/of_gpio.h>
#endif
#include <linux/soc/rockchip/rk_vendor_storage.h>
#include <linux/device.h>

#include "../../drivers/mmc/core/pwrseq.h"

#if 0
#define DBG(x...) pr_info("[WLAN_RFKILL]: " x)
#else
#define DBG(x...)
#endif

#define LOG(x...) pr_info("[WLAN_RFKILL]: " x)

struct rfkill_wlan_data {
	struct rksdmmc_gpio_wifi_moudle *pdata;
	struct wake_lock wlan_irq_wl;
};

static struct rfkill_wlan_data *g_rfkill = NULL;
static int power_set_time = 0;
static int wifi_bt_vbat_state;
static int wifi_power_state;

static const char wlan_name[] = "rkwifi";

static char wifi_chip_type_string[64];
/***********************************************************
 * 
 * Broadcom Wifi Static Memory
 * 
 **********************************************************/
#ifdef CONFIG_RKWIFI
#define BCM_STATIC_MEMORY_SUPPORT 0
#else
#define BCM_STATIC_MEMORY_SUPPORT 0
#endif
//===========================
#if BCM_STATIC_MEMORY_SUPPORT
#define PREALLOC_WLAN_SEC_NUM 4
#define PREALLOC_WLAN_BUF_NUM 160
#define PREALLOC_WLAN_SECTION_HEADER 0
#define WLAN_SKB_BUF_NUM 16

#define WLAN_SECTION_SIZE_0 (12 * 1024)
#define WLAN_SECTION_SIZE_1 (12 * 1024)
#define WLAN_SECTION_SIZE_2 (32 * 1024)
#define WLAN_SECTION_SIZE_3 (136 * 1024)
#define WLAN_SECTION_SIZE_4 (4 * 1024)
#define WLAN_SECTION_SIZE_5 (64 * 1024)
#define WLAN_SECTION_SIZE_6 (4 * 1024)
#define WLAN_SECTION_SIZE_7 (4 * 1024)

static struct sk_buff *wlan_static_skb[WLAN_SKB_BUF_NUM + 1];

struct wifi_mem_prealloc {
	void *mem_ptr;
	unsigned long size;
};

static struct wifi_mem_prealloc wifi_mem_array[8] = {
	{ NULL, (WLAN_SECTION_SIZE_0) }, { NULL, (WLAN_SECTION_SIZE_1) },
	{ NULL, (WLAN_SECTION_SIZE_2) }, { NULL, (WLAN_SECTION_SIZE_3) },
	{ NULL, (WLAN_SECTION_SIZE_4) }, { NULL, (WLAN_SECTION_SIZE_5) },
	{ NULL, (WLAN_SECTION_SIZE_6) }, { NULL, (WLAN_SECTION_SIZE_7) }
};

static int rockchip_init_wifi_mem(void)
{
	int i;
	int j;

	for (i = 0; i < WLAN_SKB_BUF_NUM; i++) {
		wlan_static_skb[i] =
			dev_alloc_skb(((i < (WLAN_SKB_BUF_NUM / 2)) ?
				(PAGE_SIZE * 1) :
				(PAGE_SIZE * 2)));

		if (!wlan_static_skb[i])
			goto err_skb_alloc;
	}

	wlan_static_skb[i] = dev_alloc_skb((PAGE_SIZE * 4));
	if (!wlan_static_skb[i])
		goto err_skb_alloc;

	for (i = 0; i <= 7; i++) {
		wifi_mem_array[i].mem_ptr =
			kmalloc(wifi_mem_array[i].size, GFP_KERNEL);

		if (!wifi_mem_array[i].mem_ptr)
			goto err_mem_alloc;
	}
	return 0;

err_mem_alloc:
	pr_err("Failed to mem_alloc for WLAN\n");
	for (j = 0; j < i; j++)
		kfree(wifi_mem_array[j].mem_ptr);
	i = WLAN_SKB_BUF_NUM;
err_skb_alloc:
	pr_err("Failed to skb_alloc for WLAN\n");
	for (j = 0; j < i; j++)
		dev_kfree_skb(wlan_static_skb[j]);
	dev_kfree_skb(wlan_static_skb[j]);

	return -ENOMEM;
}

void *rockchip_mem_prealloc(int section, unsigned long size)
{
	if (section == PREALLOC_WLAN_SEC_NUM)
		return wlan_static_skb;

	if (section < 0 || section > 7)
		return NULL;

	if (wifi_mem_array[section].size < size)
		return NULL;

	return wifi_mem_array[section].mem_ptr;
}
#else
void *rockchip_mem_prealloc(int section, unsigned long size)
{
	return NULL;
}
#endif
EXPORT_SYMBOL(rockchip_mem_prealloc);

int rfkill_set_wifi_bt_power(int on)
{
	struct rfkill_wlan_data *mrfkill = g_rfkill;
	struct rksdmmc_gpio *vbat;

	LOG("%s: %d\n", __func__, on);

	if (!mrfkill) {
		LOG("%s: rfkill-wlan driver has not Successful initialized\n",
		    __func__);
		return -1;
	}

	vbat = &mrfkill->pdata->vbat_n;
	if (on) {
		if (gpio_is_valid(vbat->io))
			gpio_direction_output(vbat->io, vbat->enable);
	} else {
		if (gpio_is_valid(vbat->io))
			gpio_direction_output(vbat->io, !(vbat->enable));
	}
	wifi_bt_vbat_state = on;
	return 0;
}

/**************************************************************************
 *
 * get wifi power state Func
 *
 *************************************************************************/
int rfkill_get_wifi_power_state(int *power)
{
	struct rfkill_wlan_data *mrfkill = g_rfkill;

	if (!mrfkill) {
		LOG("%s: rfkill-wlan driver has not Successful initialized\n",
		    __func__);
		return -1;
	}

	*power = wifi_power_state;

	return 0;
}
EXPORT_SYMBOL(rfkill_get_wifi_power_state);

/**************************************************************************
 *
 * Wifi Power Control Func
 * 0 -> power off
 * 1 -> power on
 *
 *************************************************************************/
int rockchip_wifi_power(int on)
{
	struct rfkill_wlan_data *mrfkill = g_rfkill;
	struct rksdmmc_gpio *poweron, *reset;
	struct regulator *ldo = NULL;
	int bt_power = 0;
	bool toggle = false;

	LOG("%s: %d\n", __func__, on);

	if (!mrfkill) {
		LOG("%s: rfkill-wlan driver has not Successful initialized\n",
		    __func__);
		return -1;
	}

	if (mrfkill->pdata->wifi_power_remain && power_set_time) {
		LOG("%s: wifi power is setted to be remain on.", __func__);
		return 0;
	}
	power_set_time++;

	if (!rfkill_get_bt_power_state(&bt_power, &toggle)) {
		LOG("%s: toggle = %s\n", __func__, toggle ? "true" : "false");
	}

	if (mrfkill->pdata->mregulator.power_ctrl_by_pmu) {
		int ret = -1;
		char *ldostr;
		int level = mrfkill->pdata->mregulator.enable;

		ldostr = mrfkill->pdata->mregulator.pmu_regulator;
		if (!ldostr)
			return -1;
		ldo = regulator_get(NULL, ldostr);
		if (!ldo || IS_ERR(ldo)) {
			LOG("\n\n\n%s get ldo error,please mod this\n\n\n",
			    __func__);
			return -1;
		}
		if (on == level) {
			regulator_set_voltage(ldo, 3000000, 3000000);
			LOG("%s: %s enabled\n", __func__, ldostr);
			ret = regulator_enable(ldo);
			if (ret)
				LOG("ldo enable failed\n");
			wifi_power_state = 1;
			LOG("wifi turn on power.\n");
		} else {
			LOG("%s: %s disabled\n", __func__, ldostr);
			while (regulator_is_enabled(ldo) > 0) {
				ret = regulator_disable(ldo);
				if (ret)
					LOG("ldo disable failed\n");
			}
			wifi_power_state = 0;
			LOG("wifi shut off power.\n");
		}
		regulator_put(ldo);
		msleep(100);
	} else {
		poweron = &mrfkill->pdata->power_n;
		reset = &mrfkill->pdata->reset_n;

		if (on) {
			if (toggle) {
				rfkill_set_wifi_bt_power(1);
				msleep(100);
			}

			if (gpio_is_valid(poweron->io)) {
				gpio_direction_output(poweron->io, poweron->enable);
				msleep(100);
			}

			if (gpio_is_valid(reset->io)) {
				gpio_direction_output(reset->io, reset->enable);
				msleep(100);
			}

			wifi_power_state = 1;
			LOG("wifi turn on power [GPIO%d-%d]\n", poweron->io, poweron->enable);
		} else {
			if (gpio_is_valid(poweron->io)) {
				printk("wifi power off\n");
				gpio_direction_output(poweron->io, !(poweron->enable));
				msleep(100);
			}

			if (gpio_is_valid(reset->io)) {
				gpio_direction_output(reset->io, !(reset->enable));
			}

			wifi_power_state = 0;
			if (toggle) {
				if (!bt_power) {
					LOG("%s: wifi will set vbat to low\n", __func__);
					rfkill_set_wifi_bt_power(0);
				} else {
					LOG("%s: wifi shouldn't control the vbat\n", __func__);
				}
			}
			LOG("wifi shut off power [GPIO%d-%d]\n", poweron->io, !poweron->enable);
		}
	}

	return 0;
}
EXPORT_SYMBOL(rockchip_wifi_power);

/**************************************************************************
 *
 * Wifi Sdio Detect Func
 *
 *************************************************************************/
int rockchip_wifi_set_carddetect(int val)
{
	return 0;
}
EXPORT_SYMBOL(rockchip_wifi_set_carddetect);

/**************************************************************************
 *
 * Wifi Get Interrupt irq Func
 *
 *************************************************************************/
int rockchip_wifi_get_oob_irq(void)
{
	struct rfkill_wlan_data *mrfkill = g_rfkill;
	struct rksdmmc_gpio *wifi_int_irq;

	LOG("%s: Enter\n", __func__);

	if (!mrfkill) {
		LOG("%s: rfkill-wlan driver has not Successful initialized\n",
		    __func__);
		return -1;
	}

	wifi_int_irq = &mrfkill->pdata->wifi_int_b;
	if (gpio_is_valid(wifi_int_irq->io)) {
		return gpio_to_irq(wifi_int_irq->io);
		//return wifi_int_irq->io;
	} else {
		LOG("%s: wifi OOB pin isn't defined.\n", __func__);
	}

	return -1;
}
EXPORT_SYMBOL(rockchip_wifi_get_oob_irq);

int rockchip_wifi_get_oob_irq_flag(void)
{
	struct rfkill_wlan_data *mrfkill = g_rfkill;
	struct rksdmmc_gpio *wifi_int_irq;
	int gpio_flags = -1;

	if (mrfkill) {
		wifi_int_irq = &mrfkill->pdata->wifi_int_b;
		if (gpio_is_valid(wifi_int_irq->io))
			gpio_flags = wifi_int_irq->enable;
	}

	return gpio_flags;
}
EXPORT_SYMBOL(rockchip_wifi_get_oob_irq_flag);

/**************************************************************************
 *
 * Wifi Reset Func
 *
 *************************************************************************/
int rockchip_wifi_reset(int on)
{
	return 0;
}
EXPORT_SYMBOL(rockchip_wifi_reset);

/**************************************************************************
 *
 * Wifi MAC custom Func
 *
 *************************************************************************/
#include <linux/etherdevice.h>
#include <linux/errno.h>
u8 wifi_custom_mac_addr[6] = { 0, 0, 0, 0, 0, 0 };

//#define RANDOM_ADDRESS_SAVE
static int get_wifi_addr_vendor(unsigned char *addr)
{
	int ret;
	int count = 5;

	while (count-- > 0) {
		if (is_rk_vendor_ready())
			break;
		/* sleep 500ms wait rk vendor driver ready */
		msleep(500);
	}
	ret = rk_vendor_read(WIFI_MAC_ID, addr, 6);
	if (ret != 6 || is_zero_ether_addr(addr)) {
		LOG("%s: rk_vendor_read wifi mac address failed (%d)\n",
		    __func__, ret);
#ifdef CONFIG_WIFI_GENERATE_RANDOM_MAC_ADDR
		random_ether_addr(addr);
		LOG("%s: generate random wifi mac address: "
		    "%02x:%02x:%02x:%02x:%02x:%02x\n",
		    __func__, addr[0], addr[1], addr[2], addr[3], addr[4],
		    addr[5]);
		ret = rk_vendor_write(WIFI_MAC_ID, addr, 6);
		if (ret != 0) {
			LOG("%s: rk_vendor_write failed %d\n",
			    __func__, ret);
			memset(addr, 0, 6);
			return -1;
		}
#else
		return -1;
#endif
	} else {
		LOG("%s: rk_vendor_read wifi mac address: "
		    "%02x:%02x:%02x:%02x:%02x:%02x\n",
		    __func__, addr[0], addr[1], addr[2], addr[3], addr[4],
		    addr[5]);
	}
	return 0;
}

int rockchip_wifi_mac_addr(unsigned char *buf)
{
	char mac_buf[20] = { 0 };

	LOG("%s: enter.\n", __func__);

	// from vendor storage
	if (is_zero_ether_addr(wifi_custom_mac_addr)) {
		if (get_wifi_addr_vendor(wifi_custom_mac_addr) != 0)
			return -1;
	}

	sprintf(mac_buf, "%02x:%02x:%02x:%02x:%02x:%02x",
		wifi_custom_mac_addr[0], wifi_custom_mac_addr[1],
		wifi_custom_mac_addr[2], wifi_custom_mac_addr[3],
		wifi_custom_mac_addr[4], wifi_custom_mac_addr[5]);
	LOG("falsh wifi_custom_mac_addr=[%s]\n", mac_buf);

	if (is_valid_ether_addr(wifi_custom_mac_addr)) {
		if (!strncmp(wifi_chip_type_string, "rtl", 3))
			wifi_custom_mac_addr[0] &= ~0x2; // for p2p
	} else {
		LOG("This mac address is not valid, ignored...\n");
		return -1;
	}

	memcpy(buf, wifi_custom_mac_addr, 6);

	return 0;
}
EXPORT_SYMBOL(rockchip_wifi_mac_addr);

/**************************************************************************
 *
 * wifi get country code func
 *
 *************************************************************************/
struct cntry_locales_custom {
	char iso_abbrev[4]; /* ISO 3166-1 country abbreviation */
	char custom_locale[4]; /* Custom firmware locale */
	int custom_locale_rev; /* Custom local revisin default -1 */
};

static struct cntry_locales_custom country_cloc;

void *rockchip_wifi_country_code(char *ccode)
{
	struct cntry_locales_custom *mcloc;

	LOG("%s: set country code [%s]\n", __func__, ccode);
	mcloc = &country_cloc;
	memcpy(mcloc->custom_locale, ccode, 4);
	mcloc->custom_locale_rev = 0;

	return mcloc;
}
EXPORT_SYMBOL(rockchip_wifi_country_code);
/**************************************************************************/

static int rfkill_rk_setup_gpio(struct rksdmmc_gpio *gpio, const char *prefix,
				const char *name)
{
	if (gpio_is_valid(gpio->io)) {
		int ret = 0;

		sprintf(gpio->name, "%s_%s", prefix, name);
		ret = gpio_request(gpio->io, gpio->name);
		if (ret) {
			LOG("Failed to get %s gpio.\n", gpio->name);
			return -1;
		}
	}

	return 0;
}

#ifdef CONFIG_OF
static int wlan_platdata_parse_dt(struct device *dev,
				  struct rksdmmc_gpio_wifi_moudle *data)
{
	struct device_node *node = dev->of_node;
	const char *strings;
	u32 value;
	int gpio, ret;
	enum of_gpio_flags flags;
	u32 ext_clk_value = 0;

	if (!node)
		return -ENODEV;

	memset(data, 0, sizeof(*data));

#ifdef CONFIG_MFD_SYSCON
	data->grf = syscon_regmap_lookup_by_phandle(node, "rockchip,grf");
	if (IS_ERR(data->grf)) {
		LOG("can't find rockchip,grf property\n");
		//return -1;
	}
#endif

	ret = of_property_read_string(node, "wifi_chip_type", &strings);
	if (ret) {
		LOG("%s: Can not read wifi_chip_type, set default to rkwifi.\n",
		    __func__);
		strcpy(wifi_chip_type_string, "rkwifi");
	} else {
		if (strings && strlen(strings) < 64)
			strcpy(wifi_chip_type_string, strings);
	}
	LOG("%s: wifi_chip_type = %s\n", __func__, wifi_chip_type_string);

	if (of_find_property(node, "keep_wifi_power_on", NULL)) {
		data->wifi_power_remain = true;
		LOG("%s: wifi power remain\n", __func__);
	} else {
		data->wifi_power_remain = false;
		LOG("%s: enable wifi power control.\n", __func__);
	}

	if (of_find_property(node, "power_ctrl_by_pmu", NULL)) {
		data->mregulator.power_ctrl_by_pmu = true;
		ret = of_property_read_string(node, "power_pmu_regulator",
					      &strings);
		if (ret) {
			LOG("%s: Can not read property: power_pmu_regulator.\n",
			    __func__);
			data->mregulator.power_ctrl_by_pmu = false;
		} else {
			LOG("%s: wifi power controlled by pmu(%s).\n", __func__,
			    strings);
			sprintf(data->mregulator.pmu_regulator, "%s", strings);
		}
		ret = of_property_read_u32(node, "power_pmu_enable_level",
					   &value);
		if (ret) {
			LOG("%s: Can not read: power_pmu_enable_level.\n",
			    __func__);
			data->mregulator.power_ctrl_by_pmu = false;
		} else {
			LOG("%s: wifi power controlled by pmu(level = %s).\n",
			    __func__, (value == 1) ? "HIGH" : "LOW");
			data->mregulator.enable = value;
		}
	} else {
		data->mregulator.power_ctrl_by_pmu = false;
		LOG("%s: wifi power controled by gpio.\n", __func__);
		gpio = of_get_named_gpio_flags(node, "WIFI,poweren_gpio", 0,
					       &flags);
		if (gpio_is_valid(gpio)) {
			data->power_n.io = gpio;
			data->power_n.enable =
				(flags == GPIO_ACTIVE_HIGH) ? 1 : 0;
			LOG("%s: WIFI,poweren_gpio = %d flags = %d.\n",
			    __func__, gpio, flags);
		} else {
			data->power_n.io = -1;
		}
		gpio = of_get_named_gpio_flags(node, "WIFI,vbat_gpio", 0,
					       &flags);
		if (gpio_is_valid(gpio)) {
			data->vbat_n.io = gpio;
			data->vbat_n.enable =
				(flags == GPIO_ACTIVE_HIGH) ? 1 : 0;
			LOG("%s: WIFI,vbat_gpio = %d, flags = %d.\n",
			    __func__, gpio, flags);
		} else {
			data->vbat_n.io = -1;
		}
		gpio = of_get_named_gpio_flags(node, "WIFI,reset_gpio", 0,
					       &flags);
		if (gpio_is_valid(gpio)) {
			data->reset_n.io = gpio;
			data->reset_n.enable =
				(flags == GPIO_ACTIVE_HIGH) ? 1 : 0;
			LOG("%s: WIFI,reset_gpio = %d, flags = %d.\n",
			    __func__, gpio, flags);
		} else {
			data->reset_n.io = -1;
		}
		gpio = of_get_named_gpio_flags(node, "WIFI,host_wake_irq", 0,
					       &flags);
		if (gpio_is_valid(gpio)) {
			data->wifi_int_b.io = gpio;
			data->wifi_int_b.enable = !flags;
			LOG("%s: WIFI,host_wake_irq = %d, flags = %d.\n",
			    __func__, gpio, flags);
		} else {
			data->wifi_int_b.io = -1;
		}
	}

	data->ext_clk = devm_clk_get(dev, "clk_wifi");
	if (IS_ERR(data->ext_clk)) {
		LOG("%s: The ref_wifi_clk not found !\n", __func__);
	} else {
		of_property_read_u32(node, "ref-clock-frequency",
				     &ext_clk_value);
		if (ext_clk_value > 0) {
			ret = clk_set_rate(data->ext_clk, ext_clk_value);
			if (ret)
				LOG("%s: set ref clk error!\n", __func__);
		}

		ret = clk_prepare_enable(data->ext_clk);
		if (ret)
			LOG("%s: enable ref clk error!\n", __func__);

		/* WIFI clock (REF_CLKOUT) output enable.
		 * 1'b0: drive disable
		 * 1'b1: output enable
		 */
		if (of_machine_is_compatible("rockchip,rk3308"))
			regmap_write(data->grf, 0x0314, 0x00020002);
	}

	return 0;
}
#endif //CONFIG_OF

#if defined(CONFIG_HAS_EARLYSUSPEND)
#include <linux/earlysuspend.h>

static void wlan_early_suspend(struct early_suspend *h)
{
	LOG("%s :enter\n", __func__);

	return;
}

static void wlan_late_resume(struct early_suspend *h)
{
	LOG("%s :enter\n", __func__);

	return;
}

struct early_suspend wlan_early_suspend {
	.level = EARLY_SUSPEND_LEVEL_BLANK_SCREEN - 20;
	.suspend = wlan_early_suspend;
	.resume = wlan_late_resume;
}
#endif

static void
rfkill_wlan_early_suspend(void)
{
	//LOG("%s :enter\n", __func__);

	return;
}

static void rfkill_wlan_later_resume(void)
{
	//LOG("%s :enter\n", __func__);

	return;
}

static int rfkill_wlan_fb_event_notify(struct notifier_block *self,
				       unsigned long action, void *data)
{
	struct fb_event *event = data;
	int blank_mode = *((int *)event->data);

	switch (blank_mode) {
	case FB_BLANK_UNBLANK:
		rfkill_wlan_later_resume();
		break;
	case FB_BLANK_NORMAL:
		rfkill_wlan_early_suspend();
		break;
	default:
		rfkill_wlan_early_suspend();
		break;
	}

	return 0;
}

static struct notifier_block rfkill_wlan_fb_notifier = {
	.notifier_call = rfkill_wlan_fb_event_notify,
};

static ssize_t wifi_power_show(struct class *cls, struct class_attribute *attr, char *_buf)
{
	return sprintf(_buf, "%d\n", wifi_power_state);
}

static ssize_t wifi_power_store(struct class *cls, struct class_attribute *attr, const char *_buf, size_t _count)
{
	long poweren = 0;

	if (kstrtol(_buf, 10, &poweren) < 0)
		return -EINVAL;

	LOG("%s: poweren = %ld\n", __func__, poweren);

	if (poweren > 0)
		rockchip_wifi_power(1);
	else
		rockchip_wifi_power(0);

	return _count;
}

static CLASS_ATTR_RW(wifi_power);

static ssize_t wifi_bt_vbat_show(struct class *cls, struct class_attribute *attr, char *_buf)
{
	return sprintf(_buf, "%d\n", wifi_bt_vbat_state);
}

static ssize_t wifi_bt_vbat_store(struct class *cls, struct class_attribute *attr, const char *_buf, size_t _count)
{
	long vbat = 0;

	if (kstrtol(_buf, 10, &vbat) < 0)
		return -EINVAL;

	LOG("%s: vbat = %ld\n", __func__, vbat);

	if (vbat > 0)
		rfkill_set_wifi_bt_power(1);
	else
		rfkill_set_wifi_bt_power(0);

	return _count;
}

static CLASS_ATTR_RW(wifi_bt_vbat);

static ssize_t wifi_set_carddetect_store(struct class *cls, struct class_attribute *attr, const char *_buf, size_t _count)
{
	long val = 0;

	if (kstrtol(_buf, 10, &val) < 0)
		return -EINVAL;

	LOG("%s: val = %ld\n", __func__, val);

	if (val > 0)
		rockchip_wifi_set_carddetect(1);
	else
		rockchip_wifi_set_carddetect(0);

	return _count;
}

static CLASS_ATTR_WO(wifi_set_carddetect);

static struct attribute *rkwifi_power_attrs[] = {
	&class_attr_wifi_power.attr,
	&class_attr_wifi_bt_vbat.attr,
	&class_attr_wifi_set_carddetect.attr,
	NULL,
};
ATTRIBUTE_GROUPS(rkwifi_power);

/** Device model classes */
static struct class rkwifi_power = {
	.name        = "rkwifi",
	.class_groups = rkwifi_power_groups,
};

static int rfkill_wlan_probe(struct platform_device *pdev)
{
	struct rfkill_wlan_data *rfkill;
	struct rksdmmc_gpio_wifi_moudle *pdata = pdev->dev.platform_data;
	int ret = -1;

	LOG("Enter %s\n", __func__);

	class_register(&rkwifi_power);

	if (!pdata) {
#ifdef CONFIG_OF
		pdata = kzalloc(sizeof(*pdata), GFP_KERNEL);
		if (!pdata)
			return -ENOMEM;

		ret = wlan_platdata_parse_dt(&pdev->dev, pdata);
		if (ret < 0) {
#endif
			LOG("%s: No platform data specified\n", __func__);
			return ret;
#ifdef CONFIG_OF
		}
#endif
	}

	rfkill = kzalloc(sizeof(*rfkill), GFP_KERNEL);
	if (!rfkill)
		goto rfkill_alloc_fail;

	rfkill->pdata = pdata;
	g_rfkill = rfkill;

	LOG("%s: init gpio\n", __func__);

	if (!pdata->mregulator.power_ctrl_by_pmu) {
		ret = rfkill_rk_setup_gpio(&pdata->vbat_n, wlan_name,
					   "wlan_vbat");
		if (ret)
			goto fail_alloc;

		ret = rfkill_rk_setup_gpio(&pdata->reset_n, wlan_name,
					   "wlan_reset");
		if (ret)
			goto fail_alloc;
	}

	wake_lock_init(&rfkill->wlan_irq_wl, WAKE_LOCK_SUSPEND,
		       "rfkill_wlan_wake");

	rfkill_set_wifi_bt_power(1);

#ifdef CONFIG_SDIO_KEEPALIVE
	if (gpio_is_valid(pdata->power_n.io) &&
		gpio_direction_output(pdata->power_n.io, pdata->power_n.enable);
#endif


	if (pdata->wifi_power_remain)
		rockchip_wifi_power(1);

#if BCM_STATIC_MEMORY_SUPPORT
	rockchip_init_wifi_mem();
#endif

#if defined(CONFIG_HAS_EARLYSUSPEND)
	register_early_suspend(wlan_early_suspend);
#endif

	fb_register_client(&rfkill_wlan_fb_notifier);

	LOG("Exit %s\n", __func__);

	return 0;

fail_alloc:
	kfree(rfkill);
rfkill_alloc_fail:
	kfree(pdata);

	g_rfkill = NULL;

	return ret;
}

static int rfkill_wlan_remove(struct platform_device *pdev)
{
	struct rfkill_wlan_data *rfkill = platform_get_drvdata(pdev);

	LOG("Enter %s\n", __func__);

	wake_lock_destroy(&rfkill->wlan_irq_wl);

	fb_unregister_client(&rfkill_wlan_fb_notifier);

	if (gpio_is_valid(rfkill->pdata->power_n.io))
		gpio_free(rfkill->pdata->power_n.io);

	if (gpio_is_valid(rfkill->pdata->reset_n.io))
		gpio_free(rfkill->pdata->reset_n.io);

	kfree(rfkill);
	g_rfkill = NULL;

	return 0;
}

static void rfkill_wlan_shutdown(struct platform_device *pdev)
{
	LOG("Enter %s\n", __func__);

	rockchip_wifi_power(0);
	rfkill_set_wifi_bt_power(0);
}

static int rfkill_wlan_suspend(struct platform_device *pdev, pm_message_t state)
{
	LOG("Enter %s\n", __func__);
	return 0;
}

static int rfkill_wlan_resume(struct platform_device *pdev)
{
	LOG("Enter %s\n", __func__);
	return 0;
}

#ifdef CONFIG_OF
static struct of_device_id wlan_platdata_of_match[] = {
	{ .compatible = "wlan-platdata" },
	{}
};
MODULE_DEVICE_TABLE(of, wlan_platdata_of_match);
#endif //CONFIG_OF

static struct platform_driver rfkill_wlan_driver = {
	.probe = rfkill_wlan_probe,
	.remove = rfkill_wlan_remove,
	.shutdown = rfkill_wlan_shutdown,
    .suspend = rfkill_wlan_suspend,
    .resume = rfkill_wlan_resume,
	.driver = {
		.name = "wlan-platdata",
		.owner = THIS_MODULE,
        .of_match_table = of_match_ptr(wlan_platdata_of_match),
	},
};

int __init rfkill_wlan_init(void)
{
	LOG("Enter %s\n", __func__);
	return platform_driver_register(&rfkill_wlan_driver);
}

void __exit rfkill_wlan_exit(void)
{
	LOG("Enter %s\n", __func__);
	platform_driver_unregister(&rfkill_wlan_driver);
}

MODULE_DESCRIPTION("rock-chips rfkill for wifi v0.1");
MODULE_AUTHOR("gwl@rock-chips.com");
MODULE_LICENSE("GPL");