drv_wifi.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630
  1. /*
  2. * Copyright (c) 2019 Winner Microelectronics Co., Ltd.
  3. *
  4. * SPDX-License-Identifier: Apache-2.0
  5. *
  6. * Change Logs:
  7. * Date Author Notes
  8. * 2018-09-15 flyingcys 1st version
  9. */
  10. #include <rtthread.h>
  11. #include <wlan_dev.h>
  12. #include "wm_type_def.h"
  13. #include "wm_wifi.h"
  14. #include "drv_wifi.h"
  15. #define DBG_ENABLE
  16. #define DBG_LEVEL DBG_INFO
  17. #define DBG_SECTION_NAME "WIFI"
  18. #define DBG_COLOR
  19. #include <rtdbg.h>
  20. #include "wm_ram_config.h"
  21. #include "wm_efuse.h"
  22. #include "wm_params.h"
  23. #include "wm_debug.h"
  24. #include "tls_ieee80211.h"
  25. //#define ETH_RX_DUMP
  26. //#define ETH_TX_DUMP
  27. //#define MINI_DUMP
  28. #define MAX_ADDR_LEN (6)
  29. struct drv_wifi
  30. {
  31. struct rt_wlan_device *wlan;
  32. rt_uint8_t dev_addr[MAX_ADDR_LEN];
  33. };
  34. static const struct rt_wlan_dev_ops ops;
  35. static struct drv_wifi wifi_sta;
  36. static struct drv_wifi wifi_ap;
  37. extern int hed_rf_current_channel;
  38. extern u8 *wpa_supplicant_get_mac(void);
  39. extern u8 *hostapd_get_mac(void);
  40. extern uint8_t* tls_wifi_buffer_acquire(int total_len);
  41. extern void *tls_wl_init(u8 *tx_gain, u8* mac_addr, u8 *hwver);
  42. extern int wpa_supplicant_init(u8 *mac_addr);
  43. extern void wpa_supplicant_set_mac(u8 *mac);
  44. extern void tls_wifi_buffer_release(rt_bool_t is_apsta, rt_uint8_t* buffer);
  45. #if defined(ETH_RX_DUMP) || defined(ETH_TX_DUMP)
  46. static void packet_dump(const char *msg, const void *ptr, rt_uint32_t len)
  47. {
  48. rt_uint32_t j;
  49. rt_uint8_t *p = (rt_uint8_t *)ptr;
  50. rt_kprintf("%s %d byte\n", msg, len);
  51. #ifdef MINI_DUMP
  52. return;
  53. #endif
  54. for (j = 0; j < len; j++)
  55. {
  56. if ((j % 8) == 0)
  57. {
  58. rt_kprintf(" ");
  59. }
  60. if ((j % 16) == 0)
  61. {
  62. rt_kprintf("\r\n");
  63. }
  64. rt_kprintf("%02x ", *p ++);
  65. }
  66. rt_kprintf("\n\n");
  67. }
  68. #endif /* dump */
  69. err_t tls_netif_set_addr(void *ipaddr, void *netmask, void *gw)
  70. {
  71. LOG_D("%s %d\r\n", __FUNCTION__, __LINE__);
  72. return 0;
  73. }
  74. static int wm_ethernetif_input(const rt_uint8_t *bssid, rt_uint8_t *buf, rt_uint32_t buf_len)
  75. {
  76. rt_err_t err = -RT_ERROR;
  77. if (0 == compare_ether_addr(bssid, wifi_ap.dev_addr))
  78. err = rt_wlan_dev_report_data(wifi_ap.wlan, (void *)buf, buf_len);
  79. else
  80. err = rt_wlan_dev_report_data(wifi_sta.wlan, (void *)buf, buf_len);
  81. return err ? -1 : 0;
  82. }
  83. static void wm_wlan_client_event(u8 *mac, enum tls_wifi_client_event_type event)
  84. {
  85. struct rt_wlan_buff buff;
  86. struct rt_wlan_info sta;
  87. rt_memcpy(sta.bssid, mac, MAX_ADDR_LEN);
  88. buff.data = &sta;
  89. buff.len = sizeof(sta);
  90. if (WM_WIFI_CLIENT_EVENT_ONLINE == event)
  91. {
  92. rt_wlan_dev_indicate_event_handle(wifi_ap.wlan, RT_WLAN_DEV_EVT_AP_ASSOCIATED, &buff);
  93. }
  94. else
  95. {
  96. rt_wlan_dev_indicate_event_handle(wifi_ap.wlan, RT_WLAN_DEV_EVT_AP_DISASSOCIATED, &buff);
  97. }
  98. }
  99. static void wm_wlan_status_changed(rt_uint8_t status)
  100. {
  101. LOG_D("status:%d", status);
  102. switch (status)
  103. {
  104. case WIFI_JOIN_SUCCESS:
  105. LOG_D("NETIF_WIFI_JOIN_SUCCESS");
  106. rt_wlan_dev_indicate_event_handle(wifi_sta.wlan, RT_WLAN_DEV_EVT_CONNECT, RT_NULL);
  107. break;
  108. case WIFI_JOIN_FAILED:
  109. LOG_D("NETIF_WIFI_JOIN_FAILED");
  110. rt_wlan_dev_indicate_event_handle(wifi_sta.wlan, RT_WLAN_DEV_EVT_CONNECT_FAIL, RT_NULL);
  111. break;
  112. case WIFI_DISCONNECTED:
  113. LOG_D("NETIF_WIFI_DISCONNECTED");
  114. rt_wlan_dev_indicate_event_handle(wifi_sta.wlan, RT_WLAN_DEV_EVT_DISCONNECT, RT_NULL);
  115. break;
  116. case WIFI_SOFTAP_SUCCESS:
  117. LOG_D("WIFI_SOFTAP_SUCCESS");
  118. rt_wlan_dev_indicate_event_handle(wifi_ap.wlan, RT_WLAN_DEV_EVT_AP_START, RT_NULL);
  119. break;
  120. case WIFI_SOFTAP_CLOSED:
  121. LOG_D("WIFI_SOFTAP_CLOSED");
  122. rt_wlan_dev_indicate_event_handle(wifi_ap.wlan, RT_WLAN_DEV_EVT_AP_STOP, RT_NULL);
  123. break;
  124. default:
  125. break;
  126. }
  127. }
  128. static rt_err_t wm_wlan_init(void)
  129. {
  130. extern rt_uint8_t tx_gain_group[];
  131. rt_uint8_t mac_addr[6] = {0x00, 0x25, 0x08, 0x09, 0x01, 0x0F};
  132. /*PARAM GAIN,MAC default*/
  133. tls_ft_param_init();
  134. tls_param_load_factory_default();
  135. tls_param_init(); /*add param to init sysparam_lock sem*/
  136. tls_get_tx_gain(&tx_gain_group[0]);
  137. TLS_DBGPRT_INFO("tx gain ");
  138. TLS_DBGPRT_DUMP((char *)(&tx_gain_group[0]), 12);
  139. if (tls_wifi_mem_cfg(WIFI_MEM_START_ADDR, 7, 7)) /*wifi tx&rx mem customized interface*/
  140. {
  141. LOG_E("wl mem initial failured\n");
  142. }
  143. tls_get_mac_addr(&mac_addr[0]);
  144. TLS_DBGPRT_INFO("mac addr ");
  145. TLS_DBGPRT_DUMP((char *)(&mac_addr[0]), 6);
  146. if (tls_wl_init(NULL, &mac_addr[0], NULL) == NULL)
  147. {
  148. LOG_I("wl driver initial failured\n");
  149. }
  150. if (wpa_supplicant_init(mac_addr))
  151. {
  152. LOG_I("supplicant initial failured\n");
  153. }
  154. rt_memcpy(wifi_sta.dev_addr, wpa_supplicant_get_mac(), MAX_ADDR_LEN);
  155. LOG_D("sta_mac:%02x-%02x-%02x-%02x-%02x-%02x\r\n", wifi_sta.dev_addr[0], wifi_sta.dev_addr[1], wifi_sta.dev_addr[2], wifi_sta.dev_addr[3], wifi_sta.dev_addr[4], wifi_sta.dev_addr[5]);
  156. rt_memcpy(wifi_ap.dev_addr, hostapd_get_mac(), MAX_ADDR_LEN);
  157. LOG_D("ap_mac:%02x-%02x-%02x-%02x-%02x-%02x\r\n", wifi_ap.dev_addr[0], wifi_ap.dev_addr[1], wifi_ap.dev_addr[2], wifi_ap.dev_addr[3], wifi_ap.dev_addr[4], wifi_ap.dev_addr[5]);
  158. return RT_EOK;
  159. }
  160. static void wm_wlan_promisc_dataframe_callback(u8 *data, u32 data_len)
  161. {
  162. rt_wlan_dev_promisc_handler(wifi_sta.wlan, data, data_len);
  163. }
  164. static rt_err_t drv_wlan_init(struct rt_wlan_device *wlan)
  165. {
  166. static int inited = 0;
  167. if (inited)
  168. return RT_EOK;
  169. wm_wlan_init();
  170. tls_ethernet_data_rx_callback(wm_ethernetif_input);
  171. tls_wifi_status_change_cb_register(wm_wlan_status_changed);
  172. tls_wifi_softap_client_event_register(wm_wlan_client_event);
  173. inited = 1;
  174. return RT_EOK;
  175. }
  176. static rt_err_t drv_wlan_mode(struct rt_wlan_device *wlan, rt_wlan_mode_t mode)
  177. {
  178. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  179. u8 wmode = IEEE80211_MODE_INFRA;
  180. if (mode == RT_WLAN_AP)
  181. wmode = IEEE80211_MODE_AP;
  182. tls_param_set(TLS_PARAM_ID_WPROTOCOL, (void *)&wmode, TRUE);
  183. return RT_EOK;
  184. }
  185. static void wm_wlan_scan_callback(void)
  186. {
  187. int buflen = 2000;
  188. char *buf = RT_NULL;
  189. int err;
  190. struct rt_wlan_info wlan_info;
  191. struct rt_wlan_buff buff;
  192. buf = rt_malloc(buflen);
  193. if (RT_NULL == buf)
  194. {
  195. LOG_E("rt_malloc failed...\r\n");
  196. return;
  197. }
  198. err = tls_wifi_get_scan_rslt((u8 *)buf, buflen);
  199. if (err != WM_SUCCESS)
  200. {
  201. rt_free(buf);
  202. return;
  203. }
  204. struct tls_scan_bss_t *scan_res = (struct tls_scan_bss_t *)buf;
  205. struct tls_bss_info_t *bss_info = (struct tls_bss_info_t *)scan_res->bss;
  206. if (scan_res->count <= 0)
  207. {
  208. rt_free(buf);
  209. return;
  210. }
  211. int i;
  212. for (i = 0; i < scan_res->count; i ++)
  213. {
  214. rt_memset(&wlan_info, 0, sizeof(wlan_info));
  215. rt_memcpy(&wlan_info.bssid[0], bss_info->bssid, 6);
  216. rt_memcpy(wlan_info.ssid.val, bss_info->ssid, bss_info->ssid_len);
  217. wlan_info.ssid.len = bss_info->ssid_len;
  218. if (bss_info->ssid_len)
  219. wlan_info.hidden = 0;
  220. else
  221. wlan_info.hidden = 1;
  222. wlan_info.channel = (rt_int16_t)bss_info->channel;
  223. wlan_info.rssi = -(char)(0x100 - bss_info->rssi);
  224. wlan_info.datarate = bss_info->max_data_rate * 1000000;
  225. wlan_info.band = RT_802_11_BAND_2_4GHZ;
  226. wlan_info.security = SECURITY_OPEN;
  227. if (WM_WIFI_AUTH_MODE_WEP_AUTO & bss_info->privacy)
  228. wlan_info.security |= WEP_ENABLED;
  229. if (WM_WIFI_AUTH_MODE_WPA_PSK_TKIP & bss_info->privacy)
  230. wlan_info.security |= WPA_SECURITY | TKIP_ENABLED;
  231. if (WM_WIFI_AUTH_MODE_WPA_PSK_CCMP & bss_info->privacy)
  232. wlan_info.security |= WPA_SECURITY | AES_ENABLED;
  233. if (WM_WIFI_AUTH_MODE_WPA2_PSK_CCMP & bss_info->privacy)
  234. wlan_info.security |= WPA2_SECURITY | AES_ENABLED;
  235. if (WM_WIFI_AUTH_MODE_WPA2_PSK_TKIP & bss_info->privacy)
  236. wlan_info.security |= WPA2_SECURITY | TKIP_ENABLED;
  237. if (bss_info->wps_support)
  238. wlan_info.security |= WPS_ENABLED;
  239. if (WM_WIFI_AUTH_MODE_UNKNOWN == bss_info->privacy)
  240. wlan_info.security = SECURITY_UNKNOWN;
  241. /* rtt incompleted... */
  242. if (wlan_info.security & SECURITY_WPA2_MIXED_PSK)
  243. wlan_info.security = SECURITY_WPA2_MIXED_PSK;
  244. else if (wlan_info.security & SECURITY_WPA2_TKIP_PSK)
  245. wlan_info.security = SECURITY_WPA2_TKIP_PSK;
  246. else if (wlan_info.security & SECURITY_WPA2_AES_PSK)
  247. wlan_info.security = SECURITY_WPA2_AES_PSK;
  248. else if (wlan_info.security & SECURITY_WPA_AES_PSK)
  249. wlan_info.security = SECURITY_WPA_AES_PSK;
  250. else if (wlan_info.security & SECURITY_WPA_TKIP_PSK)
  251. wlan_info.security = SECURITY_WPA_TKIP_PSK;
  252. else if (wlan_info.security & SECURITY_WEP_PSK)
  253. wlan_info.security = SECURITY_WEP_PSK;
  254. else if ((SECURITY_UNKNOWN == wlan_info.security) && bss_info->wps_support)
  255. wlan_info.security = SECURITY_WPS_SECURE;
  256. LOG_D("%s-%x", wlan_info.ssid.val, wlan_info.security);
  257. bss_info ++;
  258. buff.data = &wlan_info;
  259. buff.len = sizeof(wlan_info);
  260. rt_wlan_dev_indicate_event_handle(wifi_sta.wlan, RT_WLAN_DEV_EVT_SCAN_REPORT, &buff);
  261. }
  262. rt_free(buf);
  263. rt_wlan_dev_indicate_event_handle(wifi_sta.wlan, RT_WLAN_DEV_EVT_SCAN_DONE, RT_NULL);
  264. }
  265. static rt_err_t drv_wlan_scan(struct rt_wlan_device *wlan, struct rt_scan_info *scan_info)
  266. {
  267. int ret;
  268. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  269. /* register scan complt callback*/
  270. tls_wifi_scan_result_cb_register(wm_wlan_scan_callback);
  271. /* trigger the scan */
  272. ret = tls_wifi_scan();
  273. if ((ret == WM_WIFI_SCANNING_BUSY) || (ret == WM_FAILED))
  274. return -RT_ERROR;
  275. return RT_EOK;
  276. }
  277. static rt_err_t drv_wlan_join(struct rt_wlan_device *wlan, struct rt_sta_info *sta_info)
  278. {
  279. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  280. tls_wifi_disconnect();
  281. tls_wifi_connect((u8 *)sta_info->ssid.val, sta_info->ssid.len, (u8 *)sta_info->key.val, sta_info->key.len);
  282. return RT_EOK;
  283. }
  284. static rt_err_t drv_wlan_softap(struct rt_wlan_device *wlan, struct rt_ap_info *ap_info)
  285. {
  286. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  287. struct tls_softap_info_t apinfo;
  288. struct tls_ip_info_t ipinfo;
  289. rt_memset(&apinfo, 0, sizeof(apinfo));
  290. rt_memcpy(apinfo.ssid, ap_info->ssid.val, ap_info->ssid.len);
  291. apinfo.channel = ap_info->channel;
  292. switch (ap_info->security) /* only support wpa2-psk and open */
  293. {
  294. case SECURITY_WEP_PSK:
  295. apinfo.encrypt = IEEE80211_ENCRYT_WEP40;
  296. break;
  297. case SECURITY_WPA_TKIP_PSK:
  298. apinfo.encrypt = IEEE80211_ENCRYT_TKIP_WPA;
  299. break;
  300. case SECURITY_WPA_AES_PSK:
  301. apinfo.encrypt = IEEE80211_ENCRYT_CCMP_WPA;
  302. break;
  303. case SECURITY_WPA2_TKIP_PSK:
  304. apinfo.encrypt = IEEE80211_ENCRYT_TKIP_WPA2;
  305. break;
  306. case SECURITY_WPA2_AES_PSK:
  307. case SECURITY_WPA2_MIXED_PSK:
  308. apinfo.encrypt = IEEE80211_ENCRYT_CCMP_WPA2;
  309. break;
  310. default:
  311. apinfo.encrypt = IEEE80211_ENCRYT_NONE;
  312. break;
  313. }
  314. apinfo.keyinfo.format = 1; /* ascii */
  315. apinfo.keyinfo.index = 1; /* index */
  316. apinfo.keyinfo.key_len = ap_info->key.len;
  317. rt_memcpy(apinfo.keyinfo.key, ap_info->key.val, ap_info->key.len);
  318. LOG_D("ch=%d, hd=%d, key=%s, sec=%x, ssid=%s", ap_info->channel, ap_info->hidden, ap_info->key.val, ap_info->security, ap_info->ssid.val);
  319. rt_memset(&ipinfo, 0, sizeof(ipinfo));
  320. ipinfo.ip_addr[0] = 192;
  321. ipinfo.ip_addr[1] = 168;
  322. ipinfo.ip_addr[2] = 48;
  323. ipinfo.ip_addr[3] = 1;
  324. ipinfo.netmask[0] = 255;
  325. ipinfo.netmask[1] = 255;
  326. ipinfo.netmask[2] = 255;
  327. ipinfo.netmask[3] = 0;
  328. u8 ssid_set = ap_info->hidden ? 0 : 1;
  329. tls_param_set(TLS_PARAM_ID_BRDSSID, (void *)&ssid_set, FALSE);
  330. rt_memcpy(ipinfo.dnsname, "local.w60x", sizeof("local.w60x"));
  331. int ret = tls_wifi_softap_create(&apinfo, &ipinfo);
  332. return (ret == WM_SUCCESS) ? RT_EOK : RT_ERROR;
  333. }
  334. static rt_err_t drv_wlan_disconnect(struct rt_wlan_device *wlan)
  335. {
  336. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  337. tls_wifi_disconnect();
  338. return RT_EOK;
  339. }
  340. static rt_err_t drv_wlan_ap_stop(struct rt_wlan_device *wlan)
  341. {
  342. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  343. tls_wifi_softap_destroy();
  344. return RT_EOK;
  345. }
  346. static rt_err_t drv_wlan_ap_deauth(struct rt_wlan_device *wlan, rt_uint8_t mac[])
  347. {
  348. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  349. struct tls_wifi_hdr_mac_t machdr;
  350. struct tls_wifi_tx_rate_t tx;
  351. rt_memset(&machdr, 0, sizeof(machdr));
  352. rt_memcpy(machdr.da_addr, mac, MAX_ADDR_LEN);
  353. rt_memcpy(machdr.sa_addr, hostapd_get_mac(), MAX_ADDR_LEN);
  354. rt_memcpy(machdr.bssid, hostapd_get_mac(), MAX_ADDR_LEN);
  355. rt_memset(&tx, 0, sizeof(tx));
  356. tx.tx_rate = WM_WIFI_TX_RATEIDX_1M;
  357. tx.tx_gain = tls_wifi_get_tx_gain_max(WM_WIFI_TX_RATEIDX_1M);
  358. unsigned short reason = WLAN_REASON_UNSPECIFIED;/* htons */
  359. int ret = tls_wifi_send_mgmt(WM_WIFI_MGMT_TYPE_DEAUTH, &machdr, (u8 *)&reason, sizeof(reason), &tx);
  360. return (0 == ret) ? RT_EOK : RT_ERROR;
  361. }
  362. static rt_err_t drv_wlan_scan_stop(struct rt_wlan_device *wlan)
  363. {
  364. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  365. return RT_EOK;
  366. }
  367. static int drv_wlan_get_rssi(struct rt_wlan_device *wlan)
  368. {
  369. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  370. struct tls_curr_bss_t bss;
  371. rt_memset(&bss, 0, sizeof(bss));
  372. tls_wifi_get_current_bss(&bss);
  373. return -bss.rssi - 1;
  374. }
  375. static rt_err_t drv_wlan_set_powersave(struct rt_wlan_device *wlan, int level)
  376. {
  377. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  378. tls_wifi_set_psflag(level ? TRUE : FALSE, FALSE);
  379. return RT_EOK;
  380. }
  381. static int drv_wlan_get_powersave(struct rt_wlan_device *wlan)
  382. {
  383. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  384. return tls_wifi_get_psflag();
  385. }
  386. static rt_err_t drv_wlan_cfg_promisc(struct rt_wlan_device *wlan, rt_bool_t start)
  387. {
  388. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  389. if (RT_TRUE == start)
  390. {
  391. tls_wifi_set_listen_mode(1);
  392. tls_wifi_data_recv_cb_register(wm_wlan_promisc_dataframe_callback);
  393. }
  394. else
  395. {
  396. tls_wifi_set_listen_mode(0);
  397. tls_wifi_data_recv_cb_register(RT_NULL);
  398. }
  399. return RT_EOK;
  400. }
  401. static rt_err_t drv_wlan_cfg_filter(struct rt_wlan_device *wlan, struct rt_wlan_filter *filter)
  402. {
  403. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  404. return RT_EINVAL;/* not support */
  405. }
  406. static rt_err_t drv_wlan_set_channel(struct rt_wlan_device *wlan, int channel)
  407. {
  408. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  409. tls_wifi_change_chanel(channel - 1);
  410. return RT_EOK;
  411. }
  412. static int drv_wlan_get_channel(struct rt_wlan_device *wlan)
  413. {
  414. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  415. return hed_rf_current_channel;
  416. }
  417. static rt_err_t drv_wlan_set_country(struct rt_wlan_device *wlan, rt_country_code_t country_code)
  418. {
  419. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  420. u8 region = (u8)country_code;
  421. tls_param_set(TLS_PARAM_ID_COUNTRY_REGION, (void *)&region, TRUE);
  422. return RT_EOK;
  423. }
  424. static rt_country_code_t drv_wlan_get_country(struct rt_wlan_device *wlan)
  425. {
  426. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  427. u8 region = RT_COUNTRY_CHINA;
  428. tls_param_get(TLS_PARAM_ID_COUNTRY_REGION, (void *)&region, FALSE);
  429. return (rt_country_code_t)region; //RT_EOK;
  430. }
  431. static rt_err_t drv_wlan_set_mac(struct rt_wlan_device *wlan, rt_uint8_t mac[])
  432. {
  433. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  434. if (wlan->user_data == &wifi_sta) /* ap don't support */
  435. {
  436. wpa_supplicant_set_mac(mac);
  437. tls_set_mac_addr(mac);
  438. }
  439. return RT_EOK;
  440. }
  441. static rt_err_t drv_wlan_get_mac(struct rt_wlan_device *wlan, rt_uint8_t mac[])
  442. {
  443. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  444. if (wlan->user_data == &wifi_sta)
  445. rt_memcpy(mac, wpa_supplicant_get_mac(), MAX_ADDR_LEN);
  446. else
  447. rt_memcpy(mac, hostapd_get_mac(), MAX_ADDR_LEN);
  448. return RT_EOK;
  449. }
  450. static int drv_wlan_recv(struct rt_wlan_device *wlan, void *buff, int len)
  451. {
  452. return RT_EOK;
  453. }
  454. static int drv_wlan_send(struct rt_wlan_device *wlan, void *buff, int len)
  455. {
  456. rt_uint8_t *dst = tls_wifi_buffer_acquire(len);
  457. if (dst == NULL)
  458. return -RT_ENOMEM;
  459. #if ETH_PAD_SIZE
  460. pbuf_header(p, -ETH_PAD_SIZE); /* Drop the padding word */
  461. #endif
  462. #if defined(ETH_TX_DUMP)
  463. packet_dump("Tx", buff, len);
  464. #endif
  465. rt_memcpy(dst, buff, len);
  466. #if TLS_CONFIG_AP
  467. if (netif != tls_get_netif())
  468. tls_wifi_buffer_release(true, dst);
  469. else
  470. #endif
  471. tls_wifi_buffer_release(false, dst);
  472. #if ETH_PAD_SIZE
  473. pbuf_header(p, ETH_PAD_SIZE); /* Reclaim the padding word */
  474. #endif
  475. return RT_EOK;
  476. }
  477. static const struct rt_wlan_dev_ops ops =
  478. {
  479. .wlan_init = drv_wlan_init,
  480. .wlan_mode = drv_wlan_mode,
  481. .wlan_scan = drv_wlan_scan,
  482. .wlan_join = drv_wlan_join,
  483. .wlan_softap = drv_wlan_softap,
  484. .wlan_disconnect = drv_wlan_disconnect,
  485. .wlan_ap_stop = drv_wlan_ap_stop,
  486. .wlan_ap_deauth = drv_wlan_ap_deauth,
  487. .wlan_scan_stop = drv_wlan_scan_stop,
  488. .wlan_get_rssi = drv_wlan_get_rssi,
  489. .wlan_set_powersave = drv_wlan_set_powersave,
  490. .wlan_get_powersave = drv_wlan_get_powersave,
  491. .wlan_cfg_promisc = drv_wlan_cfg_promisc,
  492. .wlan_cfg_filter = drv_wlan_cfg_filter,
  493. .wlan_set_channel = drv_wlan_set_channel,
  494. .wlan_get_channel = drv_wlan_get_channel,
  495. .wlan_set_country = drv_wlan_set_country,
  496. .wlan_get_country = drv_wlan_get_country,
  497. .wlan_set_mac = drv_wlan_set_mac,
  498. .wlan_get_mac = drv_wlan_get_mac,
  499. .wlan_recv = drv_wlan_recv,
  500. .wlan_send = drv_wlan_send,
  501. };
  502. int wm_hw_wifi_init(void)
  503. {
  504. static struct rt_wlan_device wlan;
  505. static struct rt_wlan_device wlan2;
  506. LOG_D("F:%s L:%d", __FUNCTION__, __LINE__);
  507. rt_memset(&wifi_sta, 0, sizeof(wifi_sta));
  508. rt_err_t ret = rt_wlan_dev_register(&wlan, RT_WLAN_DEVICE_STA_NAME, &ops, 0, &wifi_sta);
  509. wifi_sta.wlan = &wlan;
  510. rt_memset(&wifi_ap, 0, sizeof(wifi_ap));
  511. ret |= rt_wlan_dev_register(&wlan2, RT_WLAN_DEVICE_AP_NAME, &ops, 0, &wifi_ap);
  512. wifi_ap.wlan = &wlan2;
  513. return ret; //RT_EOK;
  514. }
  515. INIT_DEVICE_EXPORT(wm_hw_wifi_init);