net: rfkill-wlan: prior to use wifi addr store in vendor storage
[firefly-linux-kernel-4.4.55.git] / net / rfkill / rfkill-wlan.c
1 /*
2  * Copyright (C) 2012 ROCKCHIP, Inc.
3  *
4  * This software is licensed under the terms of the GNU General Public
5  * License version 2, as published by the Free Software Foundation, and
6  * may be copied, distributed, and modified under those terms.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11  * GNU General Public License for more details.
12  *
13  */
14 /* Rock-chips rfkill driver for wifi
15 */
16
17 #include <linux/kernel.h>
18 #include <linux/platform_device.h>
19 #include <linux/module.h>
20 #include <linux/rfkill.h>
21 #include <linux/init.h>
22 #include <linux/slab.h>
23 #include <linux/regulator/consumer.h>
24 #include <linux/delay.h>
25 #include <linux/rfkill-wlan.h>
26 #include <linux/rfkill-bt.h>
27 #include <linux/wakelock.h>
28 #include <linux/interrupt.h>
29 #include <asm/irq.h>
30 #include <linux/suspend.h>
31 #include <linux/proc_fs.h>
32 #include <linux/uaccess.h>
33 #include <linux/gpio.h>
34 #include <linux/rockchip/iomap.h>
35 #include <dt-bindings/gpio/gpio.h>
36 #include <linux/skbuff.h>
37 #include <linux/fb.h>
38 #include <linux/rockchip/grf.h>
39 #include <linux/rockchip/common.h>
40 #include <linux/regmap.h>
41 #include <linux/mfd/syscon.h>
42 #include <linux/mmc/host.h>
43 #ifdef CONFIG_OF
44 #include <linux/of.h>
45 #include <linux/of_device.h>
46 #include <linux/of_gpio.h>
47 #endif
48 #include <linux/soc/rockchip/rk_vendor_storage.h>
49
50 #if 0
51 #define DBG(x...)   printk(KERN_INFO "[WLAN_RFKILL]: "x)
52 #else
53 #define DBG(x...)
54 #endif
55
56 #define LOG(x...)   printk(KERN_INFO "[WLAN_RFKILL]: "x)
57
58 extern struct mmc_host *primary_sdio_host;
59 extern void mmc_pwrseq_power_off(struct mmc_host *host);
60
61 struct rfkill_wlan_data {
62         struct rksdmmc_gpio_wifi_moudle *pdata;
63     struct wake_lock            wlan_irq_wl;
64 };
65
66 static struct rfkill_wlan_data *g_rfkill = NULL;
67 static int power_set_time = 0;
68
69 static const char wlan_name[] = 
70 #if defined (CONFIG_BCM4330)
71     #if defined (CONFIG_BT_MODULE_NH660)
72         "nh660"
73     #else
74         "bcm4330"
75     #endif
76 #elif defined (CONFIG_RK903)
77     #if defined(CONFIG_RKWIFI_26M)
78         "rk903_26M"
79     #else
80         "rk903"
81     #endif
82 #elif defined(CONFIG_BCM4329)
83         "bcm4329"
84 #elif defined(CONFIG_MV8787)
85         "mv8787"
86 #elif defined(CONFIG_AP6210)
87     #if defined(CONFIG_RKWIFI_26M)
88         "ap6210"
89     #else
90         "ap6210_24M"
91     #endif
92 #elif defined(CONFIG_AP6330)
93                 "ap6330"
94 #elif defined(CONFIG_AP6476)
95                 "ap6476"
96 #elif defined(CONFIG_AP6493)
97                 "ap6493"
98 #else
99         "wlan_default"
100 #endif
101 ;
102
103 static char wifi_chip_type_string[64];
104 int get_wifi_chip_type(void)
105 {
106     int type;
107     if (strcmp(wifi_chip_type_string, "ap6210") == 0) {
108         type = WIFI_AP6210;
109     } else if (strcmp(wifi_chip_type_string, "ap6212") == 0) {
110         type = WIFI_AP6212;
111     } else if (strcmp(wifi_chip_type_string, "rk901") == 0) {
112         type = WIFI_RK901;    
113     } else if (strcmp(wifi_chip_type_string, "rk903") == 0) {
114         type = WIFI_RK903;  
115     } else if (strcmp(wifi_chip_type_string, "ap6181") == 0) {
116         type = WIFI_AP6181;
117     } else if (strcmp(wifi_chip_type_string, "ap6234") == 0) {
118         type = WIFI_AP6234;                            
119     } else if (strcmp(wifi_chip_type_string, "ap6330") == 0) {
120         type = WIFI_AP6330;
121     } else if (strcmp(wifi_chip_type_string, "ap6335") == 0) {
122         type = WIFI_AP6335;
123     } else if (strcmp(wifi_chip_type_string, "ap6354") == 0) {
124         type = WIFI_AP6354;
125     } else if (strcmp(wifi_chip_type_string, "ap6441") == 0) {
126         type = WIFI_AP6441;
127     } else if (strcmp(wifi_chip_type_string, "ap6476") == 0) {
128         type = WIFI_AP6476;    
129     } else if (strcmp(wifi_chip_type_string, "ap6493") == 0) {
130         type = WIFI_AP6493;                    
131     } else if (strcmp(wifi_chip_type_string, "rtl8188eu") == 0) {
132         type = WIFI_RTL8188EU;
133     } else if (strcmp(wifi_chip_type_string, "rtl8192du") == 0) {
134         type = WIFI_RTL8192DU;
135     } else if (strcmp(wifi_chip_type_string, "rtl8723as") == 0) {
136         type = WIFI_RTL8723AS;        
137     } else if (strcmp(wifi_chip_type_string, "rtl8723bs_vq0") == 0) {
138         type = WIFI_RTL8723BS_VQ0;        
139     } else if (strcmp(wifi_chip_type_string, "rtl8723bs") == 0) {
140         type = WIFI_RTL8723BS;
141     } else if (strcmp(wifi_chip_type_string, "rtl8723au") == 0) {
142         type = WIFI_RTL8723AU;        
143     } else if (strcmp(wifi_chip_type_string, "rtl8723bu") == 0) {
144         type = WIFI_RTL8723BU;
145     } else if (strcmp(wifi_chip_type_string, "rtl8189es") == 0) {
146         type = WIFI_RTL8189ES;
147     } else if (strcmp(wifi_chip_type_string, "rtl8812au") == 0) {
148         type = WIFI_RTL8812AU;                        
149     } else if (strcmp(wifi_chip_type_string, "esp8089") == 0) {
150         type = WIFI_ESP8089;
151     } else {
152         type = WIFI_AP6210;
153     }
154     return type;
155 }
156 EXPORT_SYMBOL(get_wifi_chip_type);
157
158 /***********************************************************
159  * 
160  * Broadcom Wifi Static Memory
161  * 
162  **********************************************************/
163 #ifdef CONFIG_RKWIFI
164 #define BCM_STATIC_MEMORY_SUPPORT 0
165 #else
166 #define BCM_STATIC_MEMORY_SUPPORT 0
167 #endif
168 //===========================
169 #if BCM_STATIC_MEMORY_SUPPORT
170 #define PREALLOC_WLAN_SEC_NUM           4
171 #define PREALLOC_WLAN_BUF_NUM           160
172 #define PREALLOC_WLAN_SECTION_HEADER    0
173 #define WLAN_SKB_BUF_NUM        16
174
175 #define WLAN_SECTION_SIZE_0     (12 * 1024)
176 #define WLAN_SECTION_SIZE_1     (12 * 1024)
177 #define WLAN_SECTION_SIZE_2     (32 * 1024)
178 #define WLAN_SECTION_SIZE_3     (136* 1024)
179 #define WLAN_SECTION_SIZE_4     (4  * 1024)
180 #define WLAN_SECTION_SIZE_5     (64 * 1024)
181 #define WLAN_SECTION_SIZE_6     (4  * 1024)
182 #define WLAN_SECTION_SIZE_7     (4  * 1024)
183
184 static struct sk_buff *wlan_static_skb[WLAN_SKB_BUF_NUM+1];
185
186 struct wifi_mem_prealloc {
187     void *mem_ptr;
188     unsigned long size;
189 };
190
191 static struct wifi_mem_prealloc wifi_mem_array[8] = {
192     {NULL, (WLAN_SECTION_SIZE_0)},
193     {NULL, (WLAN_SECTION_SIZE_1)},
194     {NULL, (WLAN_SECTION_SIZE_2)},
195     {NULL, (WLAN_SECTION_SIZE_3)},
196     {NULL, (WLAN_SECTION_SIZE_4)},
197     {NULL, (WLAN_SECTION_SIZE_5)},
198     {NULL, (WLAN_SECTION_SIZE_6)},
199     {NULL, (WLAN_SECTION_SIZE_7)}
200 };
201
202 static int rockchip_init_wifi_mem(void)
203 {
204     int i;
205     int j;
206
207     for (i = 0 ; i < WLAN_SKB_BUF_NUM ; i++) {
208         wlan_static_skb[i] = dev_alloc_skb(
209                ((i < (WLAN_SKB_BUF_NUM / 2)) ? (PAGE_SIZE*1) : (PAGE_SIZE*2)));
210
211         if (!wlan_static_skb[i])
212             goto err_skb_alloc;
213     }
214
215     wlan_static_skb[i] = dev_alloc_skb((PAGE_SIZE*4));
216     if (!wlan_static_skb[i])
217         goto err_skb_alloc;
218
219     for (i = 0 ; i <= 7; i++) {
220         wifi_mem_array[i].mem_ptr =
221                kmalloc(wifi_mem_array[i].size, GFP_KERNEL);
222
223         if (!wifi_mem_array[i].mem_ptr)
224             goto err_mem_alloc;
225     }
226     return 0;
227
228 err_mem_alloc:
229     pr_err("Failed to mem_alloc for WLAN\n");
230     for (j = 0 ; j < i ; j++)
231         kfree(wifi_mem_array[j].mem_ptr);
232     i = WLAN_SKB_BUF_NUM;
233 err_skb_alloc:
234     pr_err("Failed to skb_alloc for WLAN\n");
235     for (j = 0 ; j < i ; j++)
236         dev_kfree_skb(wlan_static_skb[j]);
237     dev_kfree_skb(wlan_static_skb[j]);
238
239     return -ENOMEM;
240 }
241
242 void *rockchip_mem_prealloc(int section, unsigned long size)
243 {
244     //printk("rockchip_mem_prealloc: section = %d, size = %d\n", section, size);
245     if (section == PREALLOC_WLAN_SEC_NUM)
246         return wlan_static_skb;
247
248     if ((section < 0) || (section > 7))
249         return NULL;
250
251     if (wifi_mem_array[section].size < size)
252         return NULL;
253
254     return wifi_mem_array[section].mem_ptr;
255 }
256 #else
257 void *rockchip_mem_prealloc(int section, unsigned long size) { return NULL;}
258 #endif
259 EXPORT_SYMBOL(rockchip_mem_prealloc);
260
261 /**************************************************************************
262  *
263  * get wifi power state Func
264  *
265  *************************************************************************/
266 static int wifi_power_state = 0;
267 int rfkill_get_wifi_power_state(int *power, int *vref_ctrl_enable)
268 {
269     struct rfkill_wlan_data *mrfkill = g_rfkill;
270
271     if (mrfkill == NULL) {
272         LOG("%s: rfkill-wlan driver has not Successful initialized\n", __func__);
273         return -1;
274     }
275
276     *power = wifi_power_state;
277
278     return 0;
279 }
280
281 /**************************************************************************
282  *
283  * Wifi Power Control Func
284  * 0 -> power off
285  * 1 -> power on
286  *
287  *************************************************************************/
288 int rockchip_wifi_power(int on)
289 {
290         struct rfkill_wlan_data *mrfkill = g_rfkill;
291     struct rksdmmc_gpio *poweron, *reset;
292     struct regulator *ldo = NULL;
293     int power = 0;
294     bool toggle = false;
295
296     LOG("%s: %d\n", __func__, on);
297
298         if (!on && primary_sdio_host)
299                 mmc_pwrseq_power_off(primary_sdio_host);
300
301     if (mrfkill == NULL) {
302         LOG("%s: rfkill-wlan driver has not Successful initialized\n", __func__);
303         return -1;
304     }
305
306     if (mrfkill->pdata->wifi_power_remain && power_set_time) {
307         LOG("%s: wifi power is setted to be remain on. skip anything to the power control", __func__);
308         return 0;
309     } else power_set_time++;
310
311     if (!rfkill_get_bt_power_state(&power, &toggle)) {
312         if (toggle == true && power == 1) {
313             LOG("%s: wifi shouldn't control the power, it was enabled by BT!\n", __func__);
314             return 0;
315         }
316     }
317
318     if (mrfkill->pdata->mregulator.power_ctrl_by_pmu) {
319         int ret = -1;
320         char *ldostr;
321         int level = mrfkill->pdata->mregulator.enable;
322
323         ldostr = mrfkill->pdata->mregulator.pmu_regulator;
324         if (ldostr == NULL) {
325             LOG("%s: wifi power set to be controled by pmic, but which one?\n", __func__);
326             return -1;
327         }
328         ldo = regulator_get(NULL, ldostr);
329         if (ldo == NULL || IS_ERR(ldo)) {
330             LOG("\n\n\n%s get ldo error,please mod this\n\n\n", __func__);
331             return -1;
332         } else {
333                         if (on == level) {
334                                 regulator_set_voltage(ldo, 3000000, 3000000);
335                             LOG("%s: %s enabled\n", __func__, ldostr);
336                                 ret = regulator_enable(ldo);
337                 wifi_power_state = 1;
338                             LOG("wifi turn on power.\n");
339             } else {
340                                 LOG("%s: %s disabled\n", __func__, ldostr);
341                 while (regulator_is_enabled(ldo) > 0) {
342                                     ret = regulator_disable(ldo);
343                 }
344                 wifi_power_state = 0;
345                             LOG("wifi shut off power.\n");
346                         }
347                         regulator_put(ldo);
348                         msleep(100);
349                 }
350     } else {
351                 poweron = &mrfkill->pdata->power_n;
352                 reset = &mrfkill->pdata->reset_n;
353
354                 if (on){
355                         if (gpio_is_valid(poweron->io)) {
356                                 gpio_set_value(poweron->io, poweron->enable);
357                                 msleep(100);
358                         }
359
360                         if (gpio_is_valid(reset->io)) {
361                                 gpio_set_value(reset->io, reset->enable);
362                                 msleep(100);
363                         }
364
365             wifi_power_state = 1;
366                         LOG("wifi turn on power. %d\n", poweron->io);
367                 }else{
368                         if (gpio_is_valid(poweron->io)) {
369                                 gpio_set_value(poweron->io, !(poweron->enable));
370                                 msleep(100);
371                         }
372
373                         if (gpio_is_valid(reset->io)) {
374                                 gpio_set_value(reset->io, !(reset->enable));
375                         }
376
377             wifi_power_state = 0;
378                         LOG("wifi shut off power.\n");
379                 }
380     }
381
382     return 0;
383 }
384 EXPORT_SYMBOL(rockchip_wifi_power);
385
386 /**************************************************************************
387  *
388  * Wifi Sdio Detect Func
389  *
390  *************************************************************************/
391 extern int mmc_host_rescan(struct mmc_host *host, int val, int irq_type);
392 int rockchip_wifi_set_carddetect(int val)
393 {
394         return mmc_host_rescan(NULL, val, 1);
395 }
396 EXPORT_SYMBOL(rockchip_wifi_set_carddetect);
397
398 /**************************************************************************
399  *
400  * Wifi Get Interrupt irq Func
401  *
402  *************************************************************************/
403 int rockchip_wifi_get_oob_irq(void)
404 {
405     struct rfkill_wlan_data *mrfkill = g_rfkill;
406     struct rksdmmc_gpio *wifi_int_irq;
407
408     LOG("%s: Enter\n", __func__);
409
410     if (mrfkill == NULL) {
411         LOG("%s: rfkill-wlan driver has not Successful initialized\n", __func__);
412         return -1;
413     }
414     
415     wifi_int_irq = &mrfkill->pdata->wifi_int_b;
416     if (gpio_is_valid(wifi_int_irq->io)) {
417         return gpio_to_irq(wifi_int_irq->io);
418         //return wifi_int_irq->io;
419     } else {
420         LOG("%s: wifi OOB pin isn't defined.\n", __func__);
421     }
422     
423     return -1;
424 }
425 EXPORT_SYMBOL(rockchip_wifi_get_oob_irq);
426
427 /**************************************************************************
428  *
429  * Wifi Reset Func
430  *
431  *************************************************************************/
432 int rockchip_wifi_reset(int on)
433 {
434     return 0;
435 }
436 EXPORT_SYMBOL(rockchip_wifi_reset);
437
438 /**************************************************************************
439  *
440  * Wifi MAC custom Func
441  *
442  *************************************************************************/
443 #include <linux/etherdevice.h>
444 #include <linux/errno.h>
445 u8 wifi_custom_mac_addr[6] = {0,0,0,0,0,0};
446
447 //#define RANDOM_ADDRESS_SAVE
448 static int get_wifi_addr_vendor(unsigned char *addr)
449 {
450         int ret;
451
452         ret = rk_vendor_read(WIFI_MAC_ID, addr, 6);
453         if (ret != 6 || is_zero_ether_addr(addr)) {
454                 LOG("%s: rk_vendor_read wifi mac address failed (%d)\n",
455                     __func__, ret);
456 #ifdef RANDOM_ADDRESS_SAVE
457                 random_ether_addr(addr);
458                 LOG("%s: generate random wifi mac address: "
459                     "%02x:%02x:%02x:%02x:%02x:%02x\n",
460                     __func__, addr[0], addr[1], addr[2],
461                     addr[3], addr[4], addr[5]);
462                 ret = rk_vendor_write(WIFI_MAC_ID, addr, 6);
463                 if (ret != 0) {
464                         LOG("%s: rk_vendor_write"
465                                 " wifi mac address failed (%d)\n",
466                                 __func__, ret);
467                         memset(addr, 0, 6);
468                         return -1;
469                 }
470 #else
471                 return -1;
472 #endif
473         } else {
474                 LOG("%s: rk_vendor_read wifi mac address: "
475                     "%02x:%02x:%02x:%02x:%02x:%02x\n",
476                     __func__, addr[0], addr[1], addr[2],
477                     addr[3], addr[4], addr[5]);
478         }
479         return 0;
480 }
481
482 int rockchip_wifi_mac_addr(unsigned char *buf)
483 {
484         char mac_buf[20] = {0};
485
486         LOG("%s: enter.\n", __func__);
487
488         // from vendor storage
489         if (is_zero_ether_addr(wifi_custom_mac_addr)) {
490                 if (get_wifi_addr_vendor(wifi_custom_mac_addr) != 0)
491                         return -1;
492         }
493
494         sprintf(mac_buf, "%02x:%02x:%02x:%02x:%02x:%02x",
495                 wifi_custom_mac_addr[0], wifi_custom_mac_addr[1],
496                 wifi_custom_mac_addr[2], wifi_custom_mac_addr[3],
497                 wifi_custom_mac_addr[4], wifi_custom_mac_addr[5]);
498         LOG("falsh wifi_custom_mac_addr=[%s]\n", mac_buf);
499
500         if (is_valid_ether_addr(wifi_custom_mac_addr)) {
501                 if (!strncmp(wifi_chip_type_string, "rtl", 3))
502                         wifi_custom_mac_addr[0] &= ~0x2; // for p2p
503         } else {
504                 LOG("This mac address is not valid, ignored...\n");
505                 return -1;
506         }
507
508         memcpy(buf, wifi_custom_mac_addr, 6);
509
510         return 0;
511 }
512 EXPORT_SYMBOL(rockchip_wifi_mac_addr);
513
514 /**************************************************************************
515  *
516  * wifi get country code func
517  *
518  *************************************************************************/
519 struct cntry_locales_custom {
520     char iso_abbrev[4];  /* ISO 3166-1 country abbreviation */
521     char custom_locale[4];   /* Custom firmware locale */
522     int custom_locale_rev;        /* Custom local revisin default -1 */
523 };
524
525 static struct cntry_locales_custom country_cloc;
526
527 void *rockchip_wifi_country_code(char *ccode)
528 {
529     struct cntry_locales_custom *mcloc;
530
531     LOG("%s: set country code [%s]\n", __func__, ccode);
532     mcloc = &country_cloc;
533     memcpy(mcloc->custom_locale, ccode, 4);
534     mcloc->custom_locale_rev = 0;
535
536     return mcloc;
537 }
538 EXPORT_SYMBOL(rockchip_wifi_country_code);
539 /**************************************************************************/
540
541 static int rfkill_rk_setup_gpio(struct rksdmmc_gpio *gpio, const char* prefix, const char* name)
542 {
543     if (gpio_is_valid(gpio->io)) {
544         int ret=0;
545         sprintf(gpio->name, "%s_%s", prefix, name);
546         ret = gpio_request(gpio->io, gpio->name);
547         if (ret) {
548             LOG("Failed to get %s gpio.\n", gpio->name);
549             return -1;
550         }
551     }
552
553     return 0;
554 }
555
556 #ifdef CONFIG_OF
557 static int wlan_platdata_parse_dt(struct device *dev,
558                   struct rksdmmc_gpio_wifi_moudle *data)
559 {
560     struct device_node *node = dev->of_node;
561     const char *strings;
562     u32 value;
563     int gpio,ret;
564     enum of_gpio_flags flags;
565
566     if (!node)
567         return -ENODEV;
568
569     memset(data, 0, sizeof(*data));
570
571 #ifdef CONFIG_MFD_SYSCON
572     data->grf = syscon_regmap_lookup_by_phandle(node, "rockchip,grf");
573     if (IS_ERR(data->grf)) {
574             LOG("can't find rockchip,grf property\n");
575             //return -1;
576     }
577 #endif
578
579     ret = of_property_read_string(node, "wifi_chip_type", &strings);
580     if (ret) {
581         LOG("%s: Can not read wifi_chip_type, set default to rkwifi.\n", __func__);
582         strcpy(wifi_chip_type_string, "rkwifi");
583     } else {
584         strcpy(wifi_chip_type_string, strings);
585     }
586     LOG("%s: wifi_chip_type = %s\n", __func__, wifi_chip_type_string);
587
588     if (of_find_property(node, "keep_wifi_power_on", NULL)) {
589         data->wifi_power_remain = true;
590         LOG("%s: wifi power will enabled while kernel starting and keep on.\n", __func__);
591     } else {
592         data->wifi_power_remain = false;
593         LOG("%s: enable wifi power control.\n", __func__);
594     }
595
596     if (of_find_property(node, "power_ctrl_by_pmu", NULL)) {
597         data->mregulator.power_ctrl_by_pmu = true;
598         ret = of_property_read_string(node, "power_pmu_regulator", &strings);
599         if (ret) {
600             LOG("%s: Can not read property: power_pmu_regulator.\n", __func__);
601             data->mregulator.power_ctrl_by_pmu = false;
602         } else {
603             LOG("%s: wifi power controled by pmu(%s).\n", __func__, strings);
604             sprintf(data->mregulator.pmu_regulator, "%s", strings);
605         }
606         ret = of_property_read_u32(node, "power_pmu_enable_level", &value);
607         if (ret) {
608             LOG("%s: Can not read property: power_pmu_enable_level.\n", __func__);
609             data->mregulator.power_ctrl_by_pmu = false;
610         } else {
611             LOG("%s: wifi power controled by pmu(level = %s).\n", __func__, (value == 1)?"HIGH":"LOW");
612             data->mregulator.enable = value;
613         }
614         } else {
615                 data->mregulator.power_ctrl_by_pmu = false;
616                 LOG("%s: wifi power controled by gpio.\n", __func__);
617         gpio = of_get_named_gpio_flags(node, "WIFI,poweren_gpio", 0, &flags);
618         if (gpio_is_valid(gpio)){
619                         data->power_n.io = gpio;
620                         data->power_n.enable = (flags == GPIO_ACTIVE_HIGH)? 1:0;
621                         LOG("%s: get property: WIFI,poweren_gpio = %d, flags = %d.\n", __func__, gpio, flags);
622         } else data->power_n.io = -1;
623         gpio = of_get_named_gpio_flags(node, "WIFI,reset_gpio", 0, &flags);
624         if (gpio_is_valid(gpio)){
625                         data->reset_n.io = gpio;
626                         data->reset_n.enable = (flags == GPIO_ACTIVE_HIGH)? 1:0;
627                         LOG("%s: get property: WIFI,reset_gpio = %d, flags = %d.\n", __func__, gpio, flags);
628         } else data->reset_n.io = -1;
629         gpio = of_get_named_gpio_flags(node, "WIFI,host_wake_irq", 0, &flags);
630         if (gpio_is_valid(gpio)){
631                         data->wifi_int_b.io = gpio;
632                         data->wifi_int_b.enable = flags;
633                         LOG("%s: get property: WIFI,host_wake_irq = %d, flags = %d.\n", __func__, gpio, flags);
634         } else data->wifi_int_b.io = -1;
635         }
636
637     return 0;
638 }
639 #endif //CONFIG_OF
640
641 #if defined(CONFIG_HAS_EARLYSUSPEND)
642 #include <linux/earlysuspend.h>
643
644 static void wlan_early_suspend(struct early_suspend *h)
645 {
646     LOG("%s :enter\n", __func__);
647
648     return;
649 }
650
651 static void wlan_late_resume(struct early_suspend *h)
652 {
653     LOG("%s :enter\n", __func__);
654
655     return;
656 }
657
658 struct early_suspend wlan_early_suspend {
659     .level = EARLY_SUSPEND_LEVEL_BLANK_SCREEN - 20;
660     .suspend = wlan_early_suspend;
661     .resume = wlan_late_resume; 
662 }
663 #endif
664
665 static void rfkill_wlan_early_suspend(void)
666 {
667     //LOG("%s :enter\n", __func__);
668
669     return;
670 }
671
672 static void rfkill_wlan_later_resume(void)
673 {
674     //LOG("%s :enter\n", __func__);
675
676     return;
677 }
678
679 static int rfkill_wlan_fb_event_notify(struct notifier_block *self,
680                        unsigned long action, void *data)
681 {
682
683     struct fb_event *event = data;
684     int blank_mode = *((int *)event->data);
685
686     switch (blank_mode) {
687     case FB_BLANK_UNBLANK:
688         rfkill_wlan_later_resume();
689         break;
690     case FB_BLANK_NORMAL:
691         rfkill_wlan_early_suspend();
692         break;
693     default:
694         rfkill_wlan_early_suspend();
695         break;
696     }
697
698     return 0;
699 }
700
701 static struct notifier_block rfkill_wlan_fb_notifier = {
702     .notifier_call = rfkill_wlan_fb_event_notify,
703 };
704
705
706 static int rfkill_wlan_probe(struct platform_device *pdev)
707 {
708         struct rfkill_wlan_data *rfkill;
709         struct rksdmmc_gpio_wifi_moudle *pdata = pdev->dev.platform_data;
710         int ret = -1;
711
712     LOG("Enter %s\n", __func__);
713
714         if (!pdata) {
715 #ifdef CONFIG_OF
716         pdata = kzalloc(sizeof(struct rksdmmc_gpio_wifi_moudle), GFP_KERNEL);
717         if (!pdata)
718             return -ENOMEM;
719
720         ret = wlan_platdata_parse_dt(&pdev->dev, pdata);
721         if (ret < 0) {
722 #endif
723                     LOG("%s: No platform data specified\n", __func__);
724             return ret;
725 #ifdef CONFIG_OF
726         }
727 #endif
728         }
729
730         rfkill = kzalloc(sizeof(*rfkill), GFP_KERNEL);
731         if (!rfkill)
732         goto rfkill_alloc_fail;
733
734         rfkill->pdata = pdata;
735     g_rfkill = rfkill;
736
737     LOG("%s: init gpio\n", __func__);
738
739     if (!pdata->mregulator.power_ctrl_by_pmu) {
740         ret = rfkill_rk_setup_gpio(&pdata->power_n, wlan_name, "wlan_poweren");
741         if (ret) goto fail_alloc;
742
743         ret = rfkill_rk_setup_gpio(&pdata->reset_n, wlan_name, "wlan_reset");
744         if (ret) goto fail_alloc;
745     }
746
747     wake_lock_init(&(rfkill->wlan_irq_wl), WAKE_LOCK_SUSPEND, "rfkill_wlan_wake");
748
749     // Turn off wifi power as default
750     if (gpio_is_valid(pdata->power_n.io))
751     {
752         gpio_direction_output(pdata->power_n.io, !pdata->power_n.enable);
753     }
754
755     if (pdata->wifi_power_remain)
756     {
757         rockchip_wifi_power(1);
758     }
759
760 #if BCM_STATIC_MEMORY_SUPPORT
761     rockchip_init_wifi_mem();
762 #endif
763
764 #if defined(CONFIG_HAS_EARLYSUSPEND)
765     register_early_suspend(wlan_early_suspend);
766 #endif
767     
768     fb_register_client(&rfkill_wlan_fb_notifier);
769
770     LOG("Exit %s\n", __func__);
771
772         return 0;
773
774 fail_alloc:
775         kfree(rfkill);
776 rfkill_alloc_fail:
777     kfree(pdata);
778
779     g_rfkill = NULL;
780
781         return ret;
782 }
783
784 static int rfkill_wlan_remove(struct platform_device *pdev)
785 {
786         struct rfkill_wlan_data *rfkill = platform_get_drvdata(pdev);
787
788     LOG("Enter %s\n", __func__);
789
790     wake_lock_destroy(&rfkill->wlan_irq_wl);
791
792     fb_unregister_client(&rfkill_wlan_fb_notifier);
793     
794     if (gpio_is_valid(rfkill->pdata->power_n.io))
795         gpio_free(rfkill->pdata->power_n.io);
796     
797     if (gpio_is_valid(rfkill->pdata->reset_n.io))
798         gpio_free(rfkill->pdata->reset_n.io);
799     
800 //    if (gpio_is_valid(rfkill->pdata->vddio.io))
801 //        gpio_free(rfkill->pdata->vddio.io);
802 //
803 //    if (gpio_is_valid(rfkill->pdata->bgf_int_b.io))
804 //        gpio_free(rfkill->pdata->bgf_int_b.io);
805 //    
806 //    if (gpio_is_valid(rfkill->pdata->gps_sync.io))
807 //        gpio_free(rfkill->pdata->gps_sync.io);
808 //    
809 //    if (gpio_is_valid(rfkill->pdata->ANTSEL2.io))
810 //        gpio_free(rfkill->pdata->ANTSEL2.io);
811 //
812 //    if (gpio_is_valid(rfkill->pdata->ANTSEL3.io))
813 //        gpio_free(rfkill->pdata->ANTSEL3.io);
814 //    
815 //    if (gpio_is_valid(rfkill->pdata->GPS_LAN.io))
816 //        gpio_free(rfkill->pdata->GPS_LAN.io);
817
818     kfree(rfkill);
819     g_rfkill = NULL;
820
821         return 0;
822 }
823
824 static int rfkill_wlan_suspend(struct platform_device *pdev, pm_message_t state)
825 {
826     LOG("Enter %s\n", __func__);
827     return 0;
828 }
829
830 static int rfkill_wlan_resume(struct platform_device *pdev)
831 {
832     LOG("Enter %s\n", __func__);
833     return 0;
834 }
835
836 #ifdef CONFIG_OF
837 static struct of_device_id wlan_platdata_of_match[] = {
838     { .compatible = "wlan-platdata" },
839     { }
840 };
841 MODULE_DEVICE_TABLE(of, wlan_platdata_of_match);
842 #endif //CONFIG_OF
843
844 static struct platform_driver rfkill_wlan_driver = {
845         .probe = rfkill_wlan_probe,
846         .remove = rfkill_wlan_remove,
847     .suspend = rfkill_wlan_suspend,
848     .resume = rfkill_wlan_resume,
849         .driver = {
850                 .name = "wlan-platdata",
851                 .owner = THIS_MODULE,
852         .of_match_table = of_match_ptr(wlan_platdata_of_match),
853         },
854 };
855
856 static int __init rfkill_wlan_init(void)
857 {
858     LOG("Enter %s\n", __func__);
859         return platform_driver_register(&rfkill_wlan_driver);
860 }
861
862 static void __exit rfkill_wlan_exit(void)
863 {
864     LOG("Enter %s\n", __func__);
865         platform_driver_unregister(&rfkill_wlan_driver);
866 }
867
868 module_init(rfkill_wlan_init);
869 module_exit(rfkill_wlan_exit);
870
871 MODULE_DESCRIPTION("rock-chips rfkill for wifi v0.1");
872 MODULE_AUTHOR("gwl@rock-chips.com");
873 MODULE_LICENSE("GPL");