2 *************************************************************************
4 * 5F., No.36, Taiyuan St., Jhubei City,
8 * (c) Copyright 2002-2007, Ralink Technology, Inc.
10 * This program is free software; you can redistribute it and/or modify *
11 * it under the terms of the GNU General Public License as published by *
12 * the Free Software Foundation; either version 2 of the License, or *
13 * (at your option) any later version. *
15 * This program is distributed in the hope that it will be useful, *
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
18 * GNU General Public License for more details. *
20 * You should have received a copy of the GNU General Public License *
21 * along with this program; if not, write to the *
22 * Free Software Foundation, Inc., *
23 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
25 *************************************************************************
34 -------- ---------- ----------------------------------------------
35 John 2004-9-3 porting from RT2500
37 #include "../rt_config.h"
40 ==========================================================================
42 authenticate state machine init, including state transition and timer init
44 Sm - pointer to the auth state machine
46 The state machine looks like this
48 AUTH_REQ_IDLE AUTH_WAIT_SEQ2 AUTH_WAIT_SEQ4
49 MT2_MLME_AUTH_REQ mlme_auth_req_action invalid_state_when_auth invalid_state_when_auth
50 MT2_PEER_AUTH_EVEN drop peer_auth_even_at_seq2_action peer_auth_even_at_seq4_action
51 MT2_AUTH_TIMEOUT Drop auth_timeout_action auth_timeout_action
55 ==========================================================================
58 void AuthStateMachineInit(
61 OUT STATE_MACHINE_FUNC Trans
[])
63 StateMachineInit(Sm
, Trans
, MAX_AUTH_STATE
, MAX_AUTH_MSG
, (STATE_MACHINE_FUNC
)Drop
, AUTH_REQ_IDLE
, AUTH_MACHINE_BASE
);
66 StateMachineSetAction(Sm
, AUTH_REQ_IDLE
, MT2_MLME_AUTH_REQ
, (STATE_MACHINE_FUNC
)MlmeAuthReqAction
);
69 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ2
, MT2_MLME_AUTH_REQ
, (STATE_MACHINE_FUNC
)InvalidStateWhenAuth
);
70 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ2
, MT2_PEER_AUTH_EVEN
, (STATE_MACHINE_FUNC
)PeerAuthRspAtSeq2Action
);
71 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ2
, MT2_AUTH_TIMEOUT
, (STATE_MACHINE_FUNC
)AuthTimeoutAction
);
74 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ4
, MT2_MLME_AUTH_REQ
, (STATE_MACHINE_FUNC
)InvalidStateWhenAuth
);
75 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ4
, MT2_PEER_AUTH_EVEN
, (STATE_MACHINE_FUNC
)PeerAuthRspAtSeq4Action
);
76 StateMachineSetAction(Sm
, AUTH_WAIT_SEQ4
, MT2_AUTH_TIMEOUT
, (STATE_MACHINE_FUNC
)AuthTimeoutAction
);
78 RTMPInitTimer(pAd
, &pAd
->MlmeAux
.AuthTimer
, GET_TIMER_FUNCTION(AuthTimeout
), pAd
, FALSE
);
82 ==========================================================================
84 function to be executed at timer thread when auth timer expires
88 ==========================================================================
91 IN PVOID SystemSpecific1
,
92 IN PVOID FunctionContext
,
93 IN PVOID SystemSpecific2
,
94 IN PVOID SystemSpecific3
)
96 RTMP_ADAPTER
*pAd
= (RTMP_ADAPTER
*)FunctionContext
;
98 DBGPRINT(RT_DEBUG_TRACE
,("AUTH - AuthTimeout\n"));
100 // Do nothing if the driver is starting halt state.
101 // This might happen when timer already been fired before cancel timer with mlmehalt
102 if (RTMP_TEST_FLAG(pAd
, fRTMP_ADAPTER_HALT_IN_PROGRESS
| fRTMP_ADAPTER_NIC_NOT_EXIST
))
105 // send a de-auth to reset AP's state machine (Patch AP-Dir635)
106 if (pAd
->Mlme
.AuthMachine
.CurrState
== AUTH_WAIT_SEQ2
)
107 Cls2errAction(pAd
, pAd
->MlmeAux
.Bssid
);
110 MlmeEnqueue(pAd
, AUTH_STATE_MACHINE
, MT2_AUTH_TIMEOUT
, 0, NULL
);
111 RT28XX_MLME_HANDLER(pAd
);
116 ==========================================================================
119 IRQL = DISPATCH_LEVEL
121 ==========================================================================
123 VOID
MlmeAuthReqAction(
124 IN PRTMP_ADAPTER pAd
,
125 IN MLME_QUEUE_ELEM
*Elem
)
128 USHORT Alg
, Seq
, Status
;
130 HEADER_802_11 AuthHdr
;
131 BOOLEAN TimerCancelled
;
133 PUCHAR pOutBuffer
= NULL
;
136 // Block all authentication request durning WPA block period
137 if (pAd
->StaCfg
.bBlockAssoc
== TRUE
)
139 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Block Auth request durning WPA block period!\n"));
140 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
141 Status
= MLME_STATE_MACHINE_REJECT
;
142 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
144 else if(MlmeAuthReqSanity(pAd
, Elem
->Msg
, Elem
->MsgLen
, Addr
, &Timeout
, &Alg
))
147 RTMPCancelTimer(&pAd
->MlmeAux
.AuthTimer
, &TimerCancelled
);
148 COPY_MAC_ADDR(pAd
->MlmeAux
.Bssid
, Addr
);
149 pAd
->MlmeAux
.Alg
= Alg
;
151 Status
= MLME_SUCCESS
;
153 NStatus
= MlmeAllocateMemory(pAd
, &pOutBuffer
); //Get an unused nonpaged memory
154 if(NStatus
!= NDIS_STATUS_SUCCESS
)
156 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - MlmeAuthReqAction(Alg:%d) allocate memory failed\n", Alg
));
157 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
158 Status
= MLME_FAIL_NO_RESOURCE
;
159 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
163 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Send AUTH request seq#1 (Alg=%d)...\n", Alg
));
164 MgtMacHeaderInit(pAd
, &AuthHdr
, SUBTYPE_AUTH
, 0, Addr
, pAd
->MlmeAux
.Bssid
);
165 MakeOutgoingFrame(pOutBuffer
, &FrameLen
,
166 sizeof(HEADER_802_11
),&AuthHdr
,
171 MiniportMMRequest(pAd
, 0, pOutBuffer
, FrameLen
);
172 MlmeFreeMemory(pAd
, pOutBuffer
);
174 RTMPSetTimer(&pAd
->MlmeAux
.AuthTimer
, Timeout
);
175 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_WAIT_SEQ2
;
179 DBGPRINT_ERR(("AUTH - MlmeAuthReqAction() sanity check failed\n"));
180 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
181 Status
= MLME_INVALID_FORMAT
;
182 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
187 ==========================================================================
190 IRQL = DISPATCH_LEVEL
192 ==========================================================================
194 VOID
PeerAuthRspAtSeq2Action(
195 IN PRTMP_ADAPTER pAd
,
196 IN MLME_QUEUE_ELEM
*Elem
)
198 UCHAR Addr2
[MAC_ADDR_LEN
];
199 USHORT Seq
, Status
, RemoteStatus
, Alg
;
200 UCHAR ChlgText
[CIPHER_TEXT_LEN
];
201 UCHAR CyperChlgText
[CIPHER_TEXT_LEN
+ 8 + 8];
203 HEADER_802_11 AuthHdr
;
204 BOOLEAN TimerCancelled
;
205 PUCHAR pOutBuffer
= NULL
;
210 if (PeerAuthSanity(pAd
, Elem
->Msg
, Elem
->MsgLen
, Addr2
, &Alg
, &Seq
, &Status
, ChlgText
))
212 if (MAC_ADDR_EQUAL(pAd
->MlmeAux
.Bssid
, Addr2
) && Seq
== 2)
214 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Receive AUTH_RSP seq#2 to me (Alg=%d, Status=%d)\n", Alg
, Status
));
215 RTMPCancelTimer(&pAd
->MlmeAux
.AuthTimer
, &TimerCancelled
);
217 if (Status
== MLME_SUCCESS
)
219 // Authentication Mode "LEAP" has allow for CCX 1.X
220 if ((pAd
->MlmeAux
.Alg
== Ndis802_11AuthModeOpen
)
222 || (pAd
->StaCfg
.LeapAuthMode
== CISCO_AuthModeLEAP
)
223 #endif // LEAP_SUPPORT //
226 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
228 pAd
->Mlme
.LeapMachine
.CurrState
= LEAP_IDLE
;
229 #endif // LEAP_SUPPORT //
230 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
234 // 2. shared key, need to be challenged
236 RemoteStatus
= MLME_SUCCESS
;
238 // Get an unused nonpaged memory
239 NStatus
= MlmeAllocateMemory(pAd
, &pOutBuffer
);
240 if(NStatus
!= NDIS_STATUS_SUCCESS
)
242 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - PeerAuthRspAtSeq2Action() allocate memory fail\n"));
243 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
244 Status2
= MLME_FAIL_NO_RESOURCE
;
245 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status2
);
249 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Send AUTH request seq#3...\n"));
250 MgtMacHeaderInit(pAd
, &AuthHdr
, SUBTYPE_AUTH
, 0, Addr2
, pAd
->MlmeAux
.Bssid
);
252 // Encrypt challenge text & auth information
255 pAd
->SharedKey
[BSS0
][pAd
->StaCfg
.DefaultKeyId
].Key
,
256 pAd
->StaCfg
.DefaultKeyId
,
257 pAd
->SharedKey
[BSS0
][pAd
->StaCfg
.DefaultKeyId
].KeyLen
,
260 Alg
= cpu2le16(*(USHORT
*)&Alg
);
261 Seq
= cpu2le16(*(USHORT
*)&Seq
);
262 RemoteStatus
= cpu2le16(*(USHORT
*)&RemoteStatus
);
264 RTMPEncryptData(pAd
, (PUCHAR
) &Alg
, CyperChlgText
+ 4, 2);
265 RTMPEncryptData(pAd
, (PUCHAR
) &Seq
, CyperChlgText
+ 6, 2);
266 RTMPEncryptData(pAd
, (PUCHAR
) &RemoteStatus
, CyperChlgText
+ 8, 2);
269 RTMPEncryptData(pAd
, Element
, CyperChlgText
+ 10, 2);
270 RTMPEncryptData(pAd
, ChlgText
, CyperChlgText
+ 12, 128);
271 RTMPSetICV(pAd
, CyperChlgText
+ 140);
272 MakeOutgoingFrame(pOutBuffer
, &FrameLen
,
273 sizeof(HEADER_802_11
), &AuthHdr
,
274 CIPHER_TEXT_LEN
+ 16, CyperChlgText
,
276 MiniportMMRequest(pAd
, 0, pOutBuffer
, FrameLen
);
277 MlmeFreeMemory(pAd
, pOutBuffer
);
279 RTMPSetTimer(&pAd
->MlmeAux
.AuthTimer
, AUTH_TIMEOUT
);
280 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_WAIT_SEQ4
;
286 if (pAd
->StaCfg
.LeapAuthMode
== CISCO_AuthModeLEAP
)
288 //Invalid Authentication possible rogue AP
289 //Add this Ap to Rogue AP.
290 RogueApTableSetEntry(pAd
, &pAd
->StaCfg
.RogueApTab
, Addr2
, LEAP_REASON_INVALID_AUTH
);
292 #endif // LEAP_SUPPORT //
293 pAd
->StaCfg
.AuthFailReason
= Status
;
294 COPY_MAC_ADDR(pAd
->StaCfg
.AuthFailSta
, Addr2
);
295 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
296 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
302 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - PeerAuthSanity() sanity check fail\n"));
307 ==========================================================================
310 IRQL = DISPATCH_LEVEL
312 ==========================================================================
314 VOID
PeerAuthRspAtSeq4Action(
315 IN PRTMP_ADAPTER pAd
,
316 IN MLME_QUEUE_ELEM
*Elem
)
318 UCHAR Addr2
[MAC_ADDR_LEN
];
319 USHORT Alg
, Seq
, Status
;
320 CHAR ChlgText
[CIPHER_TEXT_LEN
];
321 BOOLEAN TimerCancelled
;
323 if(PeerAuthSanity(pAd
, Elem
->Msg
, Elem
->MsgLen
, Addr2
, &Alg
, &Seq
, &Status
, ChlgText
))
325 if(MAC_ADDR_EQUAL(pAd
->MlmeAux
.Bssid
, Addr2
) && Seq
== 4)
327 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Receive AUTH_RSP seq#4 to me\n"));
328 RTMPCancelTimer(&pAd
->MlmeAux
.AuthTimer
, &TimerCancelled
);
330 if (Status
!= MLME_SUCCESS
)
332 pAd
->StaCfg
.AuthFailReason
= Status
;
333 COPY_MAC_ADDR(pAd
->StaCfg
.AuthFailSta
, Addr2
);
336 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
337 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
342 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - PeerAuthRspAtSeq4Action() sanity check fail\n"));
347 ==========================================================================
350 IRQL = DISPATCH_LEVEL
352 ==========================================================================
354 VOID
MlmeDeauthReqAction(
355 IN PRTMP_ADAPTER pAd
,
356 IN MLME_QUEUE_ELEM
*Elem
)
358 MLME_DEAUTH_REQ_STRUCT
*pInfo
;
359 HEADER_802_11 DeauthHdr
;
360 PUCHAR pOutBuffer
= NULL
;
365 pInfo
= (MLME_DEAUTH_REQ_STRUCT
*)Elem
->Msg
;
367 NStatus
= MlmeAllocateMemory(pAd
, &pOutBuffer
); //Get an unused nonpaged memory
368 if (NStatus
!= NDIS_STATUS_SUCCESS
)
370 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - MlmeDeauthReqAction() allocate memory fail\n"));
371 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
372 Status
= MLME_FAIL_NO_RESOURCE
;
373 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_DEAUTH_CONF
, 2, &Status
);
377 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Send DE-AUTH request (Reason=%d)...\n", pInfo
->Reason
));
378 MgtMacHeaderInit(pAd
, &DeauthHdr
, SUBTYPE_DEAUTH
, 0, pInfo
->Addr
, pAd
->MlmeAux
.Bssid
);
379 MakeOutgoingFrame(pOutBuffer
, &FrameLen
,
380 sizeof(HEADER_802_11
),&DeauthHdr
,
383 MiniportMMRequest(pAd
, 0, pOutBuffer
, FrameLen
);
384 MlmeFreeMemory(pAd
, pOutBuffer
);
386 pAd
->StaCfg
.DeauthReason
= pInfo
->Reason
;
387 COPY_MAC_ADDR(pAd
->StaCfg
.DeauthSta
, pInfo
->Addr
);
388 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
389 Status
= MLME_SUCCESS
;
390 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_DEAUTH_CONF
, 2, &Status
);
392 // send wireless event - for deauthentication
393 if (pAd
->CommonCfg
.bWirelessEvent
)
394 RTMPSendWirelessEvent(pAd
, IW_DEAUTH_EVENT_FLAG
, pAd
->MacTab
.Content
[BSSID_WCID
].Addr
, BSS0
, 0);
398 ==========================================================================
401 IRQL = DISPATCH_LEVEL
403 ==========================================================================
405 VOID
AuthTimeoutAction(
406 IN PRTMP_ADAPTER pAd
,
407 IN MLME_QUEUE_ELEM
*Elem
)
410 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - AuthTimeoutAction\n"));
411 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
412 Status
= MLME_REJ_TIMEOUT
;
413 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
417 ==========================================================================
420 IRQL = DISPATCH_LEVEL
422 ==========================================================================
424 VOID
InvalidStateWhenAuth(
425 IN PRTMP_ADAPTER pAd
,
426 IN MLME_QUEUE_ELEM
*Elem
)
429 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - InvalidStateWhenAuth (state=%ld), reset AUTH state machine\n", pAd
->Mlme
.AuthMachine
.CurrState
));
430 pAd
->Mlme
.AuthMachine
.CurrState
= AUTH_REQ_IDLE
;
431 Status
= MLME_STATE_MACHINE_REJECT
;
432 MlmeEnqueue(pAd
, MLME_CNTL_STATE_MACHINE
, MT2_AUTH_CONF
, 2, &Status
);
436 ==========================================================================
440 This action should never trigger AUTH state transition, therefore we
441 separate it from AUTH state machine, and make it as a standalone service
443 IRQL = DISPATCH_LEVEL
445 ==========================================================================
448 IN PRTMP_ADAPTER pAd
,
451 HEADER_802_11 DeauthHdr
;
452 PUCHAR pOutBuffer
= NULL
;
455 USHORT Reason
= REASON_CLS2ERR
;
457 NStatus
= MlmeAllocateMemory(pAd
, &pOutBuffer
); //Get an unused nonpaged memory
458 if (NStatus
!= NDIS_STATUS_SUCCESS
)
461 DBGPRINT(RT_DEBUG_TRACE
, ("AUTH - Class 2 error, Send DEAUTH frame...\n"));
462 MgtMacHeaderInit(pAd
, &DeauthHdr
, SUBTYPE_DEAUTH
, 0, pAddr
, pAd
->MlmeAux
.Bssid
);
463 MakeOutgoingFrame(pOutBuffer
, &FrameLen
,
464 sizeof(HEADER_802_11
),&DeauthHdr
,
467 MiniportMMRequest(pAd
, 0, pOutBuffer
, FrameLen
);
468 MlmeFreeMemory(pAd
, pOutBuffer
);
470 pAd
->StaCfg
.DeauthReason
= Reason
;
471 COPY_MAC_ADDR(pAd
->StaCfg
.DeauthSta
, pAddr
);