Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
nu11secur1ty
GitHub Repository: nu11secur1ty/Kali-Linux
Path: blob/master/ALFA-W1F1/RTL8814AU/os_dep/linux/mlme_linux.c
1307 views
1
/******************************************************************************
2
*
3
* Copyright(c) 2007 - 2017 Realtek Corporation.
4
*
5
* This program is free software; you can redistribute it and/or modify it
6
* under the terms of version 2 of the GNU General Public License as
7
* published by the Free Software Foundation.
8
*
9
* This program is distributed in the hope that it will be useful, but WITHOUT
10
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
12
* more details.
13
*
14
*****************************************************************************/
15
16
17
#define _MLME_OSDEP_C_
18
19
#include <drv_types.h>
20
21
22
#ifdef RTK_DMP_PLATFORM
23
void Linkup_workitem_callback(struct work_struct *work)
24
{
25
struct mlme_priv *pmlmepriv = container_of(work, struct mlme_priv, Linkup_workitem);
26
_adapter *padapter = container_of(pmlmepriv, _adapter, mlmepriv);
27
28
29
30
#if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 12))
31
kobject_uevent(&padapter->pnetdev->dev.kobj, KOBJ_LINKUP);
32
#else
33
kobject_hotplug(&padapter->pnetdev->class_dev.kobj, KOBJ_LINKUP);
34
#endif
35
36
}
37
38
void Linkdown_workitem_callback(struct work_struct *work)
39
{
40
struct mlme_priv *pmlmepriv = container_of(work, struct mlme_priv, Linkdown_workitem);
41
_adapter *padapter = container_of(pmlmepriv, _adapter, mlmepriv);
42
43
44
45
#if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 12))
46
kobject_uevent(&padapter->pnetdev->dev.kobj, KOBJ_LINKDOWN);
47
#else
48
kobject_hotplug(&padapter->pnetdev->class_dev.kobj, KOBJ_LINKDOWN);
49
#endif
50
51
}
52
#endif
53
54
extern void rtw_indicate_wx_assoc_event(_adapter *padapter);
55
extern void rtw_indicate_wx_disassoc_event(_adapter *padapter);
56
57
void rtw_os_indicate_connect(_adapter *adapter)
58
{
59
struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
60
61
#ifdef CONFIG_IOCTL_CFG80211
62
if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) == _TRUE) ||
63
(check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE))
64
rtw_cfg80211_ibss_indicate_connect(adapter);
65
else
66
rtw_cfg80211_indicate_connect(adapter);
67
#endif /* CONFIG_IOCTL_CFG80211 */
68
69
rtw_indicate_wx_assoc_event(adapter);
70
71
#ifdef CONFIG_RTW_MESH
72
#if CONFIG_RTW_MESH_CTO_MGATE_CARRIER
73
if (!rtw_mesh_cto_mgate_required(adapter))
74
#endif
75
#endif
76
rtw_netif_carrier_on(adapter->pnetdev);
77
78
if (adapter->pid[2] != 0)
79
rtw_signal_process(adapter->pid[2], SIGALRM);
80
81
#ifdef RTK_DMP_PLATFORM
82
_set_workitem(&adapter->mlmepriv.Linkup_workitem);
83
#endif
84
85
86
}
87
88
extern void indicate_wx_scan_complete_event(_adapter *padapter);
89
void rtw_os_indicate_scan_done(_adapter *padapter, bool aborted)
90
{
91
#ifdef CONFIG_IOCTL_CFG80211
92
rtw_cfg80211_indicate_scan_done(padapter, aborted);
93
#endif
94
indicate_wx_scan_complete_event(padapter);
95
}
96
97
static RT_PMKID_LIST backupPMKIDList[NUM_PMKID_CACHE];
98
void rtw_reset_securitypriv(_adapter *adapter)
99
{
100
u8 backupPMKIDIndex = 0;
101
u8 backupTKIPCountermeasure = 0x00;
102
u32 backupTKIPcountermeasure_time = 0;
103
/* add for CONFIG_IEEE80211W, none 11w also can use */
104
_irqL irqL;
105
106
_enter_critical_bh(&adapter->security_key_mutex, &irqL);
107
108
if (adapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) { /* 802.1x */
109
/* Added by Albert 2009/02/18 */
110
/* We have to backup the PMK information for WiFi PMK Caching test item. */
111
/* */
112
/* Backup the btkip_countermeasure information. */
113
/* When the countermeasure is trigger, the driver have to disconnect with AP for 60 seconds. */
114
115
_rtw_memset(&backupPMKIDList[0], 0x00, sizeof(RT_PMKID_LIST) * NUM_PMKID_CACHE);
116
117
_rtw_memcpy(&backupPMKIDList[0], &adapter->securitypriv.PMKIDList[0], sizeof(RT_PMKID_LIST) * NUM_PMKID_CACHE);
118
backupPMKIDIndex = adapter->securitypriv.PMKIDIndex;
119
backupTKIPCountermeasure = adapter->securitypriv.btkip_countermeasure;
120
backupTKIPcountermeasure_time = adapter->securitypriv.btkip_countermeasure_time;
121
_rtw_memset((unsigned char *)&adapter->securitypriv, 0, sizeof(struct security_priv));
122
123
/* Added by Albert 2009/02/18 */
124
/* Restore the PMK information to securitypriv structure for the following connection. */
125
_rtw_memcpy(&adapter->securitypriv.PMKIDList[0], &backupPMKIDList[0], sizeof(RT_PMKID_LIST) * NUM_PMKID_CACHE);
126
adapter->securitypriv.PMKIDIndex = backupPMKIDIndex;
127
adapter->securitypriv.btkip_countermeasure = backupTKIPCountermeasure;
128
adapter->securitypriv.btkip_countermeasure_time = backupTKIPcountermeasure_time;
129
130
adapter->securitypriv.ndisauthtype = Ndis802_11AuthModeOpen;
131
adapter->securitypriv.ndisencryptstatus = Ndis802_11WEPDisabled;
132
133
adapter->securitypriv.extauth_status = WLAN_STATUS_UNSPECIFIED_FAILURE;
134
135
} else { /* reset values in securitypriv */
136
/* if(adapter->mlmepriv.fw_state & WIFI_STATION_STATE) */
137
/* { */
138
struct security_priv *psec_priv = &adapter->securitypriv;
139
140
psec_priv->dot11AuthAlgrthm = dot11AuthAlgrthm_Open; /* open system */
141
psec_priv->dot11PrivacyAlgrthm = _NO_PRIVACY_;
142
psec_priv->dot11PrivacyKeyIndex = 0;
143
144
psec_priv->dot118021XGrpPrivacy = _NO_PRIVACY_;
145
psec_priv->dot118021XGrpKeyid = 1;
146
147
psec_priv->ndisauthtype = Ndis802_11AuthModeOpen;
148
psec_priv->ndisencryptstatus = Ndis802_11WEPDisabled;
149
/* } */
150
151
psec_priv->extauth_status = WLAN_STATUS_UNSPECIFIED_FAILURE;
152
}
153
/* add for CONFIG_IEEE80211W, none 11w also can use */
154
_exit_critical_bh(&adapter->security_key_mutex, &irqL);
155
156
RTW_INFO(FUNC_ADPT_FMT" - End to Disconnect\n", FUNC_ADPT_ARG(adapter));
157
}
158
159
void rtw_os_indicate_disconnect(_adapter *adapter, u16 reason, u8 locally_generated)
160
{
161
/* RT_PMKID_LIST backupPMKIDList[NUM_PMKID_CACHE]; */
162
163
164
rtw_netif_carrier_off(adapter->pnetdev); /* Do it first for tx broadcast pkt after disconnection issue! */
165
166
#ifdef CONFIG_IOCTL_CFG80211
167
rtw_cfg80211_indicate_disconnect(adapter, reason, locally_generated);
168
#endif /* CONFIG_IOCTL_CFG80211 */
169
170
rtw_indicate_wx_disassoc_event(adapter);
171
172
#ifdef RTK_DMP_PLATFORM
173
_set_workitem(&adapter->mlmepriv.Linkdown_workitem);
174
#endif
175
/* modify for CONFIG_IEEE80211W, none 11w also can use the same command */
176
rtw_reset_securitypriv_cmd(adapter);
177
178
179
}
180
181
void rtw_report_sec_ie(_adapter *adapter, u8 authmode, u8 *sec_ie)
182
{
183
uint len;
184
u8 *buff, *p, i;
185
union iwreq_data wrqu;
186
187
188
189
buff = NULL;
190
if (authmode == _WPA_IE_ID_) {
191
192
buff = rtw_zmalloc(IW_CUSTOM_MAX);
193
if (NULL == buff) {
194
RTW_INFO(FUNC_ADPT_FMT ": alloc memory FAIL!!\n",
195
FUNC_ADPT_ARG(adapter));
196
return;
197
}
198
p = buff;
199
200
p += sprintf(p, "ASSOCINFO(ReqIEs=");
201
202
len = sec_ie[1] + 2;
203
len = (len < IW_CUSTOM_MAX) ? len : IW_CUSTOM_MAX;
204
205
for (i = 0; i < len; i++)
206
p += sprintf(p, "%02x", sec_ie[i]);
207
208
p += sprintf(p, ")");
209
210
_rtw_memset(&wrqu, 0, sizeof(wrqu));
211
212
wrqu.data.length = p - buff;
213
214
wrqu.data.length = (wrqu.data.length < IW_CUSTOM_MAX) ? wrqu.data.length : IW_CUSTOM_MAX;
215
216
#ifndef CONFIG_IOCTL_CFG80211
217
wireless_send_event(adapter->pnetdev, IWEVCUSTOM, &wrqu, buff);
218
#endif
219
220
rtw_mfree(buff, IW_CUSTOM_MAX);
221
}
222
223
224
}
225
226
#ifdef CONFIG_AP_MODE
227
228
void rtw_indicate_sta_assoc_event(_adapter *padapter, struct sta_info *psta)
229
{
230
union iwreq_data wrqu;
231
struct sta_priv *pstapriv = &padapter->stapriv;
232
233
if (psta == NULL)
234
return;
235
236
if (psta->cmn.aid > pstapriv->max_aid)
237
return;
238
239
if (pstapriv->sta_aid[psta->cmn.aid - 1] != psta)
240
return;
241
242
243
wrqu.addr.sa_family = ARPHRD_ETHER;
244
245
_rtw_memcpy(wrqu.addr.sa_data, psta->cmn.mac_addr, ETH_ALEN);
246
247
RTW_INFO("+rtw_indicate_sta_assoc_event\n");
248
249
#ifndef CONFIG_IOCTL_CFG80211
250
wireless_send_event(padapter->pnetdev, IWEVREGISTERED, &wrqu, NULL);
251
#endif
252
253
}
254
255
void rtw_indicate_sta_disassoc_event(_adapter *padapter, struct sta_info *psta)
256
{
257
union iwreq_data wrqu;
258
struct sta_priv *pstapriv = &padapter->stapriv;
259
260
if (psta == NULL)
261
return;
262
263
if (psta->cmn.aid > pstapriv->max_aid)
264
return;
265
266
if (pstapriv->sta_aid[psta->cmn.aid - 1] != psta)
267
return;
268
269
270
wrqu.addr.sa_family = ARPHRD_ETHER;
271
272
_rtw_memcpy(wrqu.addr.sa_data, psta->cmn.mac_addr, ETH_ALEN);
273
274
RTW_INFO("+rtw_indicate_sta_disassoc_event\n");
275
276
#ifndef CONFIG_IOCTL_CFG80211
277
wireless_send_event(padapter->pnetdev, IWEVEXPIRED, &wrqu, NULL);
278
#endif
279
280
}
281
282
283
#ifdef CONFIG_HOSTAPD_MLME
284
285
static int mgnt_xmit_entry(struct sk_buff *skb, struct net_device *pnetdev)
286
{
287
struct hostapd_priv *phostapdpriv = rtw_netdev_priv(pnetdev);
288
_adapter *padapter = (_adapter *)phostapdpriv->padapter;
289
290
/* RTW_INFO("%s\n", __FUNCTION__); */
291
292
return rtw_hal_hostap_mgnt_xmit_entry(padapter, skb);
293
}
294
295
static int mgnt_netdev_open(struct net_device *pnetdev)
296
{
297
struct hostapd_priv *phostapdpriv = rtw_netdev_priv(pnetdev);
298
299
RTW_INFO("mgnt_netdev_open: MAC Address:" MAC_FMT "\n", MAC_ARG(pnetdev->dev_addr));
300
301
302
init_usb_anchor(&phostapdpriv->anchored);
303
304
rtw_netif_wake_queue(pnetdev);
305
306
rtw_netif_carrier_on(pnetdev);
307
308
/* rtw_write16(phostapdpriv->padapter, 0x0116, 0x0100); */ /* only excluding beacon */
309
310
return 0;
311
}
312
static int mgnt_netdev_close(struct net_device *pnetdev)
313
{
314
struct hostapd_priv *phostapdpriv = rtw_netdev_priv(pnetdev);
315
316
RTW_INFO("%s\n", __FUNCTION__);
317
318
usb_kill_anchored_urbs(&phostapdpriv->anchored);
319
320
rtw_netif_carrier_off(pnetdev);
321
322
rtw_netif_stop_queue(pnetdev);
323
324
/* rtw_write16(phostapdpriv->padapter, 0x0116, 0x3f3f); */
325
326
return 0;
327
}
328
329
#if (LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 29))
330
static const struct net_device_ops rtl871x_mgnt_netdev_ops = {
331
.ndo_open = mgnt_netdev_open,
332
.ndo_stop = mgnt_netdev_close,
333
.ndo_start_xmit = mgnt_xmit_entry,
334
#if 0
335
.ndo_set_mac_address = r871x_net_set_mac_address,
336
.ndo_get_stats = r871x_net_get_stats,
337
.ndo_do_ioctl = r871x_mp_ioctl,
338
#endif
339
};
340
#endif
341
342
int hostapd_mode_init(_adapter *padapter)
343
{
344
unsigned char mac[ETH_ALEN];
345
struct hostapd_priv *phostapdpriv;
346
struct net_device *pnetdev;
347
348
pnetdev = rtw_alloc_etherdev(sizeof(struct hostapd_priv));
349
if (!pnetdev)
350
return -ENOMEM;
351
352
/* SET_MODULE_OWNER(pnetdev); */
353
ether_setup(pnetdev);
354
355
/* pnetdev->type = ARPHRD_IEEE80211; */
356
357
phostapdpriv = rtw_netdev_priv(pnetdev);
358
phostapdpriv->pmgnt_netdev = pnetdev;
359
phostapdpriv->padapter = padapter;
360
padapter->phostapdpriv = phostapdpriv;
361
362
/* pnetdev->init = NULL; */
363
364
#if (LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 29))
365
366
RTW_INFO("register rtl871x_mgnt_netdev_ops to netdev_ops\n");
367
368
pnetdev->netdev_ops = &rtl871x_mgnt_netdev_ops;
369
370
#else
371
372
pnetdev->open = mgnt_netdev_open;
373
374
pnetdev->stop = mgnt_netdev_close;
375
376
pnetdev->hard_start_xmit = mgnt_xmit_entry;
377
378
/* pnetdev->set_mac_address = r871x_net_set_mac_address; */
379
380
/* pnetdev->get_stats = r871x_net_get_stats; */
381
382
/* pnetdev->do_ioctl = r871x_mp_ioctl; */
383
384
#endif
385
386
pnetdev->watchdog_timeo = HZ; /* 1 second timeout */
387
388
/* pnetdev->wireless_handlers = NULL; */
389
390
391
392
393
if (dev_alloc_name(pnetdev, "mgnt.wlan%d") < 0)
394
RTW_INFO("hostapd_mode_init(): dev_alloc_name, fail!\n");
395
396
397
/* SET_NETDEV_DEV(pnetdev, pintfpriv->udev); */
398
399
400
mac[0] = 0x00;
401
mac[1] = 0xe0;
402
mac[2] = 0x4c;
403
mac[3] = 0x87;
404
mac[4] = 0x11;
405
mac[5] = 0x12;
406
407
_rtw_memcpy(pnetdev->dev_addr, mac, ETH_ALEN);
408
409
410
rtw_netif_carrier_off(pnetdev);
411
412
413
/* Tell the network stack we exist */
414
if (register_netdev(pnetdev) != 0) {
415
RTW_INFO("hostapd_mode_init(): register_netdev fail!\n");
416
417
if (pnetdev)
418
rtw_free_netdev(pnetdev);
419
}
420
421
return 0;
422
423
}
424
425
void hostapd_mode_unload(_adapter *padapter)
426
{
427
struct hostapd_priv *phostapdpriv = padapter->phostapdpriv;
428
struct net_device *pnetdev = phostapdpriv->pmgnt_netdev;
429
430
unregister_netdev(pnetdev);
431
rtw_free_netdev(pnetdev);
432
433
}
434
435
#endif
436
#endif
437
438