drv_wifi.c 18 KB

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