hc
2024-05-08 f309769f8af08599af39b6de4f675784ce76530d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
/******************************************************************************
 *
 * Copyright(c) 2007 - 2017 Realtek Corporation.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * 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.
 *
 *****************************************************************************/
#ifndef __RTW_SECURITY_H_
#define __RTW_SECURITY_H_
 
enum security_type {
   /* TYPE */
   _NO_PRIVACY_    = 0x00,
   _WEP40_        = 0x01,
   _TKIP_        = 0x02,
   _TKIP_WTMIC_    = 0x03,
   _AES_        = 0x04,
   _WEP104_    = 0x05,
   _SMS4_        = 0x06,
   _GCMP_        = 0x07,
   _SEC_TYPE_MAX_,
 
   /* EXT_SECTYPE=1 */
   _SEC_TYPE_256_    = 0x10,
   _CCMP_256_    = (_AES_ | _SEC_TYPE_256_),
   _GCMP_256_    = (_GCMP_ | _SEC_TYPE_256_),
 
#ifdef CONFIG_IEEE80211W
   /* EXT_SECTYPE=0, MGNT=1, GK=0/1, KEYID=00/01 */
   _SEC_TYPE_BIT_    = 0x20,
   _BIP_CMAC_128_    = (_SEC_TYPE_BIT_),
   _BIP_GMAC_128_    = (_SEC_TYPE_BIT_ + 1),
   _BIP_GMAC_256_    = (_SEC_TYPE_BIT_ + 2),
   /* EXT_SECTYPE=1, MGNT=1, GK=1, KEYID=00/01 */
   _BIP_CMAC_256_    = (_SEC_TYPE_BIT_ + 3),
   _BIP_MAX_,
#endif
};
 
/* 802.11W use wrong key */
#define IEEE80211W_RIGHT_KEY    0x0
#define IEEE80211W_WRONG_KEY    0x1
#define IEEE80211W_NO_KEY        0x2
 
#define CCMPH_2_PN(ch)    ((ch) & 0x000000000000ffff) \
           | (((ch) & 0xffffffff00000000) >> 16)
 
#define is_wep_enc(alg) (((alg) == _WEP40_) || ((alg) == _WEP104_))
 
const char *security_type_str(u8 value);
#ifdef CONFIG_IEEE80211W
u32 security_type_bip_to_gmcs(enum security_type type);
#endif
 
#define _WPA_IE_ID_    0xdd
#define _WPA2_IE_ID_    0x30
 
#define RTW_KEK_LEN 16
#define RTW_KCK_LEN 16
#define RTW_TKIP_MIC_LEN 8
#define RTW_REPLAY_CTR_LEN 8
 
#define INVALID_SEC_MAC_CAM_ID    0xFF
 
typedef enum {
   ENCRYP_PROTOCOL_OPENSYS,   /* open system */
   ENCRYP_PROTOCOL_WEP,       /* WEP */
   ENCRYP_PROTOCOL_WPA,       /* WPA */
   ENCRYP_PROTOCOL_WPA2,      /* WPA2 */
   ENCRYP_PROTOCOL_WAPI,      /* WAPI: Not support in this version */
   ENCRYP_PROTOCOL_MAX
} ENCRYP_PROTOCOL_E;
 
 
#ifndef Ndis802_11AuthModeWPA2
#define Ndis802_11AuthModeWPA2 (Ndis802_11AuthModeWPANone + 1)
#endif
 
#ifndef Ndis802_11AuthModeWPA2PSK
#define Ndis802_11AuthModeWPA2PSK (Ndis802_11AuthModeWPANone + 2)
#endif
 
union pn48    {
 
   u64    val;
 
#ifdef CONFIG_LITTLE_ENDIAN
 
struct {
   u8 TSC0;
   u8 TSC1;
   u8 TSC2;
   u8 TSC3;
   u8 TSC4;
   u8 TSC5;
   u8 TSC6;
   u8 TSC7;
} _byte_;
 
#elif defined(CONFIG_BIG_ENDIAN)
 
struct {
   u8 TSC7;
   u8 TSC6;
   u8 TSC5;
   u8 TSC4;
   u8 TSC3;
   u8 TSC2;
   u8 TSC1;
   u8 TSC0;
} _byte_;
 
#endif
 
};
 
union Keytype {
   u8 skey[32];
};
 
typedef struct _RT_PMKID_LIST {
   u8                        bUsed;
   u8                        Bssid[6];
   u8                        PMKID[16];
   u8                        SsidBuf[33];
   u8                        *ssid_octet;
   u16                        ssid_length;
} RT_PMKID_LIST, *PRT_PMKID_LIST;
 
 
struct security_priv {
   u32      dot11AuthAlgrthm;        /* 802.11 auth, could be open, shared, 8021x and authswitch */
   u32      dot11PrivacyAlgrthm;    /* This specify the privacy for shared auth. algorithm. */
 
   /* WEP */
   u32      dot11PrivacyKeyIndex;    /* this is only valid for legendary wep, 0~3 for key id. (tx key index) */
   union Keytype dot11DefKey[6];            /* this is only valid for def. key     */
   u32    dot11DefKeylen[6];
   u8    dot11Def_camid[6];
   u8     key_mask; /* use to restore wep key after hal_init */
 
   u32 dot118021XGrpPrivacy;    /* This specify the privacy algthm. used for Grp key */
   u32    dot118021XGrpKeyid;        /* key id used for Grp Key ( tx key index) */
   union Keytype    dot118021XGrpKey[6];    /* 802.1x Group Key, for inx0 and inx1     */
   union Keytype    dot118021XGrptxmickey[6];
   union Keytype    dot118021XGrprxmickey[6];
   union pn48        dot11Grptxpn;            /* PN48 used for Grp Key xmit. */
   union pn48        dot11Grprxpn;            /* PN48 used for Grp Key recv. */
   u8                iv_seq[4][8];
#ifdef CONFIG_IEEE80211W
   enum security_type dot11wCipher;
   u32    dot11wBIPKeyid;                        /* key id used for BIP Key ( tx key index) */
   union Keytype    dot11wBIPKey[6];        /* BIP Key, for index4 and index5 */
   union pn48        dot11wBIPtxpn;            /* PN48 used for BIP xmit. */
   union pn48        dot11wBIPrxpn;            /* PN48 used for BIP recv. */
#endif /* CONFIG_IEEE80211W */
#ifdef CONFIG_AP_MODE
   /* extend security capabilities for AP_MODE */
   unsigned int dot8021xalg;/* 0:disable, 1:psk, 2:802.1x */
   unsigned int wpa_psk;/* 0:disable, bit(0): WPA, bit(1):WPA2 */
   unsigned int wpa_group_cipher;
   unsigned int wpa2_group_cipher;
   unsigned int wpa_pairwise_cipher;
   unsigned int wpa2_pairwise_cipher;
   unsigned int akmp; /* An authentication and key management protocol */
#endif
   u8 mfp_opt;
   u8    dot118021x_bmc_cam_id;
   /*IEEE802.11-2012 Std. Table 8-101 AKM Suite Selectors*/
   u32    rsn_akm_suite_type;
 
   u8 wps_ie[MAX_WPS_IE_LEN];/* added in assoc req */
   int wps_ie_len;
 
   u8 owe_ie[MAX_OWE_IE_LEN];/* added in assoc req */
   int owe_ie_len;
 
   u8 rsnx_ie[MAX_RSNX_IE_LEN];
   int rsnx_ie_len;
 
   u8    binstallGrpkey;
#ifdef CONFIG_GTK_OL
   u8    binstallKCK_KEK;
#endif /* CONFIG_GTK_OL */
#ifdef CONFIG_IEEE80211W
   u8    binstallBIPkey;
#endif /* CONFIG_IEEE80211W */
   u8    busetkipkey;
   u8    bcheck_grpkey;
   u8    bgrpkey_handshake;
 
   u8    auth_alg;
   u8    auth_type;
   u8    extauth_status;
   /* u8    packet_cnt; */ /* unused, removed */
 
   s32    sw_encrypt;/* from registry_priv */
   s32    sw_decrypt;/* from registry_priv */
 
   s32     hw_decrypted;/* if the rx packets is hw_decrypted==_FALSE, it means the hw has not been ready. */
 
 
   /* keeps the auth_type & enc_status from upper layer ioctl(wpa_supplicant or wzc) */
   u32 ndisauthtype;    /* NDIS_802_11_AUTHENTICATION_MODE */
   u32 ndisencryptstatus;    /* NDIS_802_11_ENCRYPTION_STATUS */
 
   NDIS_802_11_WEP ndiswep;
 
   u8 assoc_info[600];
   u8 szofcapability[256]; /* for wpa2 usage */
   u8 oidassociation[512]; /* for wpa/wpa2 usage */
   u8 authenticator_ie[256];  /* store ap security information element */
   u8 supplicant_ie[256];  /* store sta security information element */
 
 
   /* for tkip countermeasure */
   systime last_mic_err_time;
   u8    btkip_countermeasure;
   u8    btkip_wait_report;
   systime btkip_countermeasure_time;
 
   /* --------------------------------------------------------------------------- */
   /* For WPA2 Pre-Authentication. */
   /* --------------------------------------------------------------------------- */
   /* u8                RegEnablePreAuth;                 */ /* Default value: Pre-Authentication enabled or not, from registry "EnablePreAuth". Added by Annie, 2005-11-01. */
   /* u8                EnablePreAuthentication;             */ /* Current Value: Pre-Authentication enabled or not. */
   RT_PMKID_LIST        PMKIDList[NUM_PMKID_CACHE];    /* Renamed from PreAuthKey[NUM_PRE_AUTH_KEY]. Annie, 2006-10-13. */
   u8                PMKIDIndex;
   /* u32                PMKIDCount;                         */ /* Added by Annie, 2006-10-13. */
   /* u8                szCapability[256];                 */ /* For WPA2-PSK using zero-config, by Annie, 2005-09-20. */
 
   u8 bWepDefaultKeyIdxSet;
 
#define DBG_SW_SEC_CNT
#ifdef DBG_SW_SEC_CNT
   u64 wep_sw_enc_cnt_bc;
   u64 wep_sw_enc_cnt_mc;
   u64 wep_sw_enc_cnt_uc;
   u64 wep_sw_dec_cnt_bc;
   u64 wep_sw_dec_cnt_mc;
   u64 wep_sw_dec_cnt_uc;
 
   u64 tkip_sw_enc_cnt_bc;
   u64 tkip_sw_enc_cnt_mc;
   u64 tkip_sw_enc_cnt_uc;
   u64 tkip_sw_dec_cnt_bc;
   u64 tkip_sw_dec_cnt_mc;
   u64 tkip_sw_dec_cnt_uc;
 
   u64 aes_sw_enc_cnt_bc;
   u64 aes_sw_enc_cnt_mc;
   u64 aes_sw_enc_cnt_uc;
   u64 aes_sw_dec_cnt_bc;
   u64 aes_sw_dec_cnt_mc;
   u64 aes_sw_dec_cnt_uc;
 
   u64 gcmp_sw_enc_cnt_bc;
   u64 gcmp_sw_enc_cnt_mc;
   u64 gcmp_sw_enc_cnt_uc;
   u64 gcmp_sw_dec_cnt_bc;
   u64 gcmp_sw_dec_cnt_mc;
   u64 gcmp_sw_dec_cnt_uc;
#endif /* DBG_SW_SEC_CNT */
};
 
#ifdef CONFIG_IEEE80211W
#define SEC_IS_BIP_KEY_INSTALLED(sec) ((sec)->binstallBIPkey)
#else
#define SEC_IS_BIP_KEY_INSTALLED(sec) _FALSE
#endif
 
#define GET_ENCRY_ALGO(psecuritypriv, psta, encry_algo, bmcst)\
   do {\
       switch (psecuritypriv->dot11AuthAlgrthm) {\
       case dot11AuthAlgrthm_Open:\
       case dot11AuthAlgrthm_Shared:\
       case dot11AuthAlgrthm_Auto:\
           encry_algo = (u8)psecuritypriv->dot11PrivacyAlgrthm;\
           break;\
       case dot11AuthAlgrthm_8021X:\
           if (bmcst)\
               encry_algo = (u8)psecuritypriv->dot118021XGrpPrivacy;\
           else\
               encry_algo = (u8) psta->dot118021XPrivacy;\
           break;\
       case dot11AuthAlgrthm_WAPI:\
           encry_algo = (u8)psecuritypriv->dot11PrivacyAlgrthm;\
           break;\
       } \
   } while (0)
 
#define _AES_IV_LEN_ 8
 
#define SET_ICE_IV_LEN(iv_len, icv_len, encrypt)\
   do {\
       switch (encrypt) {\
       case _WEP40_:\
       case _WEP104_:\
           iv_len = 4;\
           icv_len = 4;\
           break;\
       case _TKIP_:\
           iv_len = 8;\
           icv_len = 4;\
           break;\
       case _AES_:\
           iv_len = 8;\
           icv_len = 8;\
           break;\
       case _GCMP_:\
       case _GCMP_256_:\
           iv_len = 8;\
           icv_len = 16;\
           break;\
       case _CCMP_256_:\
           iv_len = 8;\
           icv_len = 16;\
           break;\
       case _SMS4_:\
           iv_len = 18;\
           icv_len = 16;\
           break;\
       default:\
           iv_len = 0;\
           icv_len = 0;\
           break;\
       } \
   } while (0)
 
 
#define GET_TKIP_PN(iv, dot11txpn)\
   do {\
       dot11txpn._byte_.TSC0 = iv[2];\
       dot11txpn._byte_.TSC1 = iv[0];\
       dot11txpn._byte_.TSC2 = iv[4];\
       dot11txpn._byte_.TSC3 = iv[5];\
       dot11txpn._byte_.TSC4 = iv[6];\
       dot11txpn._byte_.TSC5 = iv[7];\
   } while (0)
 
 
#define ROL32(A, n)    (((A) << (n)) | (((A)>>(32-(n)))  & ((1UL << (n)) - 1)))
#define ROR32(A, n)    ROL32((A), 32-(n))
 
struct mic_data {
   u32  K0, K1;         /* Key */
   u32  L, R;           /* Current state */
   u32  M;              /* Message accumulator (single word) */
   u32     nBytesInM;      /*  # bytes in M */
};
 
void rtw_secmicsetkey(struct mic_data *pmicdata, u8 *key);
void rtw_secmicappendbyte(struct mic_data *pmicdata, u8 b);
void rtw_secmicappend(struct mic_data *pmicdata, u8 *src, u32 nBytes);
void rtw_secgetmic(struct mic_data *pmicdata, u8 *dst);
 
void rtw_seccalctkipmic(
   u8 *key,
   u8 *header,
   u8 *data,
   u32 data_len,
   u8 *Miccode,
   u8   priority);
 
u32 rtw_aes_encrypt(_adapter *padapter, u8 *pxmitframe);
u32 rtw_tkip_encrypt(_adapter *padapter, u8 *pxmitframe);
void rtw_wep_encrypt(_adapter *padapter, u8  *pxmitframe);
 
u32 rtw_aes_decrypt(_adapter *padapter, u8  *precvframe);
u32 rtw_tkip_decrypt(_adapter *padapter, u8  *precvframe);
void rtw_wep_decrypt(_adapter *padapter, u8  *precvframe);
 
u32 rtw_gcmp_encrypt(_adapter *padapter, u8 *pxmitframe);
u32 rtw_gcmp_decrypt(_adapter *padapter, u8 *precvframe);
 
#ifdef CONFIG_RTW_MESH_AEK
int rtw_aes_siv_encrypt(const u8 *key, size_t key_len,
   const u8 *pw, size_t pwlen, size_t num_elem,
   const u8 *addr[], const size_t *len, u8 *out);
int rtw_aes_siv_decrypt(const u8 *key, size_t key_len,
   const u8 *iv_crypt, size_t iv_c_len, size_t num_elem,
   const u8 *addr[], const size_t *len, u8 *out);
#endif /* CONFIG_RTW_MESH_AEK */
 
#ifdef CONFIG_IEEE80211W
u8 rtw_calculate_bip_mic(enum security_type gmcs, u8 *whdr_pos, s32 len,
   const u8 *key, const u8 *data, size_t data_len, u8 *mic);
u32 rtw_bip_verify(enum security_type gmcs, u16 pkt_len,
   u8 *whdr_pos, sint flen, const u8 *key, u16 keyid, u64 *ipn);
#endif
#ifdef CONFIG_TDLS
void wpa_tdls_generate_tpk(_adapter *padapter, void *sta);
int wpa_tdls_ftie_mic(u8 *kck, u8 trans_seq,
           u8 *lnkid, u8 *rsnie, u8 *timeoutie, u8 *ftie,
           u8 *mic);
int wpa_tdls_teardown_ftie_mic(u8 *kck, u8 *lnkid, u16 reason,
           u8 dialog_token, u8 trans_seq, u8 *ftie, u8 *mic);
int tdls_verify_mic(u8 *kck, u8 trans_seq,
           u8 *lnkid, u8 *rsnie, u8 *timeoutie, u8 *ftie);
#endif /* CONFIG_TDLS */
 
void rtw_sec_restore_wep_key(_adapter *adapter);
u8 rtw_handle_tkip_countermeasure(_adapter *adapter, const char *caller);
 
#ifdef CONFIG_WOWLAN
u16 rtw_calc_crc(u8  *pdata, int length);
#endif /*CONFIG_WOWLAN*/
 
#define rtw_sec_chk_auth_alg(a, s) \
   ((a)->securitypriv.auth_alg == (s))
 
#define rtw_sec_chk_auth_type(a, s) \
   ((a)->securitypriv.auth_type == (s))
 
#endif /* __RTL871X_SECURITY_H_ */
 
u32 rtw_calc_crc32(u8 *data, size_t len);