]> git.proxmox.com Git - mirror_edk2.git/blame_incremental - MdeModulePkg/Universal/Network/IScsiDxe/IScsiProto.c
Update code to support VS2013 tool chain.
[mirror_edk2.git] / MdeModulePkg / Universal / Network / IScsiDxe / IScsiProto.c
... / ...
CommitLineData
1/** @file\r
2 The implementation of iSCSI protocol based on RFC3720.\r
3\r
4Copyright (c) 2004 - 2014, Intel Corporation. All rights reserved.<BR>\r
5This program and the accompanying materials\r
6are licensed and made available under the terms and conditions of the BSD License\r
7which accompanies this distribution. The full text of the license may be found at\r
8http://opensource.org/licenses/bsd-license.php\r
9\r
10THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12\r
13**/\r
14\r
15#include "IScsiImpl.h"\r
16\r
17UINT32 mDataSegPad = 0;\r
18\r
19/**\r
20 Attach the iSCSI connection to the iSCSI session. \r
21\r
22 @param[in, out] Session The iSCSI session.\r
23 @param[in, out] Conn The iSCSI connection.\r
24**/\r
25VOID\r
26IScsiAttatchConnection (\r
27 IN OUT ISCSI_SESSION *Session,\r
28 IN OUT ISCSI_CONNECTION *Conn\r
29 )\r
30{\r
31 InsertTailList (&Session->Conns, &Conn->Link);\r
32 Conn->Session = Session;\r
33 Session->NumConns++;\r
34}\r
35\r
36/**\r
37 Detach the iSCSI connection from the session it belongs to. \r
38\r
39 @param[in, out] Conn The iSCSI connection.\r
40**/\r
41VOID\r
42IScsiDetatchConnection (\r
43 IN OUT ISCSI_CONNECTION *Conn\r
44 )\r
45{\r
46 RemoveEntryList (&Conn->Link);\r
47 Conn->Session->NumConns--;\r
48 Conn->Session = NULL;\r
49}\r
50\r
51/**\r
52 Check the sequence number according to RFC3720. \r
53\r
54 @param[in, out] ExpSN The currently expected sequence number.\r
55 @param[in] NewSN The sequence number to check.\r
56\r
57 @retval EFI_SUCCESS The check passed and the ExpSN is increased.\r
58 @retval EFI_NOT_READY Response was sent due to a retransmission request.\r
59 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol error happened.\r
60**/\r
61EFI_STATUS\r
62IScsiCheckSN (\r
63 IN OUT UINT32 *ExpSN,\r
64 IN UINT32 NewSN\r
65 )\r
66{\r
67 if (!ISCSI_SEQ_EQ (NewSN, *ExpSN)) {\r
68 if (ISCSI_SEQ_LT (NewSN, *ExpSN)) {\r
69 //\r
70 // Duplicate\r
71 //\r
72 return EFI_NOT_READY;\r
73 } else {\r
74 return EFI_PROTOCOL_ERROR;\r
75 }\r
76 } else {\r
77 //\r
78 // Advance the ExpSN\r
79 //\r
80 (*ExpSN)++;\r
81 return EFI_SUCCESS;\r
82 }\r
83}\r
84\r
85/**\r
86 Update the sequence numbers for the iSCSI command.\r
87\r
88 @param[in, out] Session The iSCSI session.\r
89 @param[in] MaxCmdSN Maximum CmdSN from the target.\r
90 @param[in] ExpCmdSN Next expected CmdSN from the target.\r
91**/\r
92VOID\r
93IScsiUpdateCmdSN (\r
94 IN OUT ISCSI_SESSION *Session,\r
95 IN UINT32 MaxCmdSN,\r
96 IN UINT32 ExpCmdSN\r
97 )\r
98{\r
99 if (ISCSI_SEQ_LT (MaxCmdSN, ExpCmdSN - 1)) {\r
100 return ;\r
101 }\r
102\r
103 if (ISCSI_SEQ_GT (MaxCmdSN, Session->MaxCmdSN)) {\r
104 Session->MaxCmdSN = MaxCmdSN;\r
105 }\r
106\r
107 if (ISCSI_SEQ_GT (ExpCmdSN, Session->ExpCmdSN)) {\r
108 Session->ExpCmdSN = ExpCmdSN;\r
109 }\r
110}\r
111\r
112/**\r
113 This function does the iSCSI connection login.\r
114\r
115 @param[in, out] Conn The iSCSI connection to login.\r
116\r
117 @retval EFI_SUCCESS The iSCSI connection is logged into the iSCSI target.\r
118 @retval EFI_TIMEOUT Timeout happened during the login procedure.\r
119 @retval Others Other errors as indicated. \r
120**/\r
121EFI_STATUS\r
122IScsiConnLogin (\r
123 IN OUT ISCSI_CONNECTION *Conn\r
124 )\r
125{\r
126 EFI_STATUS Status;\r
127\r
128 //\r
129 // Start the timer, wait 16 seconds to establish the TCP connection.\r
130 //\r
131 Status = gBS->SetTimer (Conn->TimeoutEvent, TimerRelative, 16 * TICKS_PER_SECOND);\r
132 if (EFI_ERROR (Status)) {\r
133 return Status;\r
134 }\r
135 //\r
136 // try to establish the tcp connection\r
137 //\r
138 Status = Tcp4IoConnect (&Conn->Tcp4Io, Conn->TimeoutEvent);\r
139 if (EFI_ERROR (Status)) {\r
140 return Status;\r
141 }\r
142\r
143 gBS->SetTimer (Conn->TimeoutEvent, TimerCancel, 0);\r
144 Conn->State = CONN_STATE_IN_LOGIN;\r
145\r
146 //\r
147 // connection is established, start the iSCSI Login\r
148 //\r
149 do {\r
150 Status = IScsiSendLoginReq (Conn);\r
151 if (EFI_ERROR (Status)) {\r
152 break;\r
153 }\r
154\r
155 Status = IScsiReceiveLoginRsp (Conn);\r
156 if (EFI_ERROR (Status)) {\r
157 break;\r
158 }\r
159 } while (Conn->CurrentStage != ISCSI_FULL_FEATURE_PHASE);\r
160\r
161 return Status;\r
162}\r
163\r
164/**\r
165 Reset the iSCSI connection.\r
166\r
167 @param[in, out] Conn The iSCSI connection to reset.\r
168**/\r
169VOID\r
170IScsiConnReset (\r
171 IN OUT ISCSI_CONNECTION *Conn\r
172 )\r
173{\r
174 Tcp4IoReset (&Conn->Tcp4Io);\r
175}\r
176\r
177/**\r
178 Create a TCP connection for the iSCSI session.\r
179\r
180 @param[in] Private The iSCSI driver data.\r
181 @param[in] Session Maximum CmdSN from the target.\r
182\r
183 @return The newly created iSCSI connection.\r
184**/\r
185ISCSI_CONNECTION *\r
186IScsiCreateConnection (\r
187 IN ISCSI_DRIVER_DATA *Private,\r
188 IN ISCSI_SESSION *Session\r
189 )\r
190{\r
191 ISCSI_CONNECTION *Conn;\r
192 TCP4_IO_CONFIG_DATA Tcp4IoConfig;\r
193 EFI_STATUS Status;\r
194\r
195 Conn = AllocatePool (sizeof (ISCSI_CONNECTION));\r
196 if (Conn == NULL) {\r
197 return NULL;\r
198 }\r
199\r
200 Conn->Signature = ISCSI_CONNECTION_SIGNATURE;\r
201 Conn->State = CONN_STATE_FREE;\r
202 Conn->CurrentStage = ISCSI_SECURITY_NEGOTIATION;\r
203 Conn->NextStage = ISCSI_LOGIN_OPERATIONAL_NEGOTIATION;\r
204 Conn->CHAPStep = ISCSI_CHAP_INITIAL;\r
205 Conn->ExpStatSN = 0;\r
206 Conn->PartialReqSent = FALSE;\r
207 Conn->PartialRspRcvd = FALSE;\r
208 Conn->Cid = Session->NextCid++;\r
209\r
210 Status = gBS->CreateEvent (\r
211 EVT_TIMER,\r
212 TPL_CALLBACK,\r
213 NULL,\r
214 NULL,\r
215 &Conn->TimeoutEvent\r
216 );\r
217 if (EFI_ERROR (Status)) {\r
218 FreePool (Conn);\r
219 return NULL;\r
220 }\r
221\r
222 NetbufQueInit (&Conn->RspQue);\r
223\r
224 //\r
225 // set the default connection-only parameters\r
226 //\r
227 Conn->MaxRecvDataSegmentLength = DEFAULT_MAX_RECV_DATA_SEG_LEN;\r
228 Conn->HeaderDigest = IScsiDigestNone;\r
229 Conn->DataDigest = IScsiDigestNone;\r
230\r
231 CopyMem (&Tcp4IoConfig.LocalIp, &Session->ConfigData.NvData.LocalIp, sizeof (EFI_IPv4_ADDRESS));\r
232 CopyMem (&Tcp4IoConfig.SubnetMask, &Session->ConfigData.NvData.SubnetMask, sizeof (EFI_IPv4_ADDRESS));\r
233 CopyMem (&Tcp4IoConfig.Gateway, &Session->ConfigData.NvData.Gateway, sizeof (EFI_IPv4_ADDRESS));\r
234 CopyMem (&Tcp4IoConfig.RemoteIp, &Session->ConfigData.NvData.TargetIp, sizeof (EFI_IPv4_ADDRESS));\r
235\r
236 Tcp4IoConfig.RemotePort = Session->ConfigData.NvData.TargetPort;\r
237\r
238 //\r
239 // Create the tcp4 IO for this connection\r
240 //\r
241 Status = Tcp4IoCreateSocket (\r
242 Private->Image,\r
243 Private->Controller,\r
244 &Tcp4IoConfig,\r
245 &Conn->Tcp4Io\r
246 );\r
247 if (EFI_ERROR (Status)) {\r
248 gBS->CloseEvent (Conn->TimeoutEvent);\r
249 FreePool (Conn);\r
250 Conn = NULL;\r
251 }\r
252\r
253 return Conn;\r
254}\r
255\r
256/**\r
257 Destroy an iSCSI connection.\r
258\r
259 @param[in] Conn The connection to destroy.\r
260**/\r
261VOID\r
262IScsiDestroyConnection (\r
263 IN ISCSI_CONNECTION *Conn\r
264 )\r
265{\r
266 Tcp4IoDestroySocket (&Conn->Tcp4Io);\r
267 NetbufQueFlush (&Conn->RspQue);\r
268 gBS->CloseEvent (Conn->TimeoutEvent);\r
269 FreePool (Conn);\r
270}\r
271\r
272/**\r
273 Login the iSCSI session.\r
274\r
275 @param[in] Private The iSCSI driver data.\r
276\r
277 @retval EFI_SUCCESS The iSCSI session login procedure finished.\r
278 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
279 @retval EFI_NO_MEDIA There was a media error.\r
280 @retval Others Other errors as indicated.\r
281\r
282**/\r
283EFI_STATUS\r
284IScsiSessionLogin (\r
285 IN ISCSI_DRIVER_DATA *Private\r
286 )\r
287{\r
288 EFI_STATUS Status;\r
289 ISCSI_SESSION *Session;\r
290 ISCSI_CONNECTION *Conn;\r
291 EFI_TCP4_PROTOCOL *Tcp4;\r
292 BOOLEAN MediaPresent;\r
293\r
294 Session = &Private->Session;\r
295\r
296 //\r
297 // Check media status before session login\r
298 //\r
299 MediaPresent = TRUE;\r
300 NetLibDetectMedia (Private->Controller, &MediaPresent);\r
301 if (!MediaPresent) {\r
302 return EFI_NO_MEDIA;\r
303 }\r
304\r
305 //\r
306 // Set session identifier\r
307 //\r
308 CopyMem (Session->Isid, Session->ConfigData.NvData.IsId, 6);\r
309\r
310 //\r
311 // Create a connection for the session.\r
312 //\r
313 Conn = IScsiCreateConnection (Private, Session);\r
314 if (Conn == NULL) {\r
315 return EFI_OUT_OF_RESOURCES;\r
316 }\r
317\r
318 IScsiAttatchConnection (Session, Conn);\r
319\r
320 //\r
321 // Login througth the newly created connection.\r
322 //\r
323 Status = IScsiConnLogin (Conn);\r
324 if (EFI_ERROR (Status)) {\r
325 IScsiConnReset (Conn);\r
326 IScsiDetatchConnection (Conn);\r
327 IScsiDestroyConnection (Conn);\r
328 } else {\r
329 Session->State = SESSION_STATE_LOGGED_IN;\r
330\r
331 gBS->OpenProtocol (\r
332 Conn->Tcp4Io.Handle,\r
333 &gEfiTcp4ProtocolGuid,\r
334 (VOID **)&Tcp4,\r
335 Private->Image,\r
336 Private->ExtScsiPassThruHandle,\r
337 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER\r
338 );\r
339 }\r
340\r
341 return Status;\r
342}\r
343\r
344/**\r
345 Build and send the iSCSI login request to the iSCSI target according to\r
346 the current login stage.\r
347\r
348 @param[in] Conn The connection in the iSCSI login phase.\r
349\r
350 @retval EFI_SUCCESS The iSCSI login request PDU is built and sent on this\r
351 connection.\r
352 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
353 @retval EFI_DEVICE_ERROR Some kind of device error happened.\r
354**/\r
355EFI_STATUS\r
356IScsiSendLoginReq (\r
357 IN ISCSI_CONNECTION *Conn\r
358 )\r
359{\r
360 NET_BUF *Pdu;\r
361 EFI_STATUS Status;\r
362\r
363 //\r
364 // build the Login Request PDU\r
365 //\r
366 Pdu = IScsiPrepareLoginReq (Conn);\r
367 if (Pdu == NULL) {\r
368 return EFI_DEVICE_ERROR;\r
369 }\r
370 //\r
371 // Send it to the iSCSI target.\r
372 //\r
373 Status = Tcp4IoTransmit (&Conn->Tcp4Io, Pdu);\r
374\r
375 NetbufFree (Pdu);\r
376\r
377 return Status;\r
378}\r
379\r
380/**\r
381 Receive and process the iSCSI login response.\r
382\r
383 @param[in] Conn The connection in the iSCSI login phase.\r
384 \r
385 @retval EFI_SUCCESS The iSCSI login response PDU is received and processed.\r
386 @retval Others Other errors as indicated.\r
387**/\r
388EFI_STATUS\r
389IScsiReceiveLoginRsp (\r
390 IN ISCSI_CONNECTION *Conn\r
391 )\r
392{\r
393 EFI_STATUS Status;\r
394 NET_BUF *Pdu;\r
395\r
396 Pdu = NULL;\r
397\r
398 //\r
399 // Receive the iSCSI login response.\r
400 //\r
401 Status = IScsiReceivePdu (Conn, &Pdu, NULL, FALSE, FALSE, NULL);\r
402 if (EFI_ERROR (Status)) {\r
403 return Status;\r
404 }\r
405 //\r
406 // A Login Response is received, process it.\r
407 //\r
408 ASSERT (Pdu != NULL);\r
409 Status = IScsiProcessLoginRsp (Conn, Pdu);\r
410\r
411 NetbufFree (Pdu);\r
412\r
413 return Status;\r
414}\r
415\r
416/**\r
417 Add an iSCSI key-value pair as a string into the data segment of the Login Request PDU.\r
418 The DataSegmentLength and the actual size of the net buffer containing this PDU will be\r
419 updated.\r
420\r
421 @param[in, out] Pdu The iSCSI PDU whose data segment the key-value pair will\r
422 be added to.\r
423 @param[in] Key The key name string.\r
424 @param[in] Value The value string.\r
425\r
426 @retval EFI_SUCCESS The key-valu pair is added to the PDU's datasegment and\r
427 the correspondence length fields are updated.\r
428 @retval EFI_OUT_OF_RESOURCES There is not enough space in the PDU to add the key-value\r
429 pair.\r
430**/\r
431EFI_STATUS\r
432IScsiAddKeyValuePair (\r
433 IN OUT NET_BUF *Pdu,\r
434 IN CHAR8 *Key,\r
435 IN CHAR8 *Value\r
436 )\r
437{\r
438 UINT32 DataSegLen;\r
439 UINT32 KeyLen;\r
440 UINT32 ValueLen;\r
441 UINT32 TotalLen;\r
442 ISCSI_LOGIN_REQUEST *LoginReq;\r
443 CHAR8 *Data;\r
444\r
445 LoginReq = (ISCSI_LOGIN_REQUEST *) NetbufGetByte (Pdu, 0, NULL);\r
446 if (LoginReq == NULL) {\r
447 return EFI_PROTOCOL_ERROR;\r
448 }\r
449 DataSegLen = NTOH24 (LoginReq->DataSegmentLength);\r
450\r
451 KeyLen = (UINT32) AsciiStrLen (Key);\r
452 ValueLen = (UINT32) AsciiStrLen (Value);\r
453\r
454 //\r
455 // 1 byte for the key value separator '=' and 1 byte for the null\r
456 // delimiter after the value.\r
457 //\r
458 TotalLen = KeyLen + 1 + ValueLen + 1;\r
459\r
460 //\r
461 // Allocate the space for the key-value pair.\r
462 //\r
463 Data = (CHAR8 *)NetbufAllocSpace (Pdu, TotalLen, NET_BUF_TAIL);\r
464 if (Data == NULL) {\r
465 return EFI_OUT_OF_RESOURCES;\r
466 }\r
467 //\r
468 // Add the key.\r
469 //\r
470 CopyMem (Data, Key, KeyLen);\r
471 Data += KeyLen;\r
472\r
473 *Data = '=';\r
474 Data++;\r
475\r
476 //\r
477 // Add the value.\r
478 //\r
479 CopyMem (Data, Value, ValueLen);\r
480 Data += ValueLen;\r
481\r
482 *Data = '\0';\r
483\r
484 //\r
485 // update the DataSegmentLength\r
486 //\r
487 ISCSI_SET_DATASEG_LEN (LoginReq, DataSegLen + TotalLen);\r
488\r
489 return EFI_SUCCESS;\r
490}\r
491\r
492/**\r
493 Prepare the iSCSI login request to be sent according to the current login status.\r
494\r
495 @param[in, out] Conn The connection in the iSCSI login phase.\r
496\r
497 @return The pointer to the net buffer containing the iSCSI login request built.\r
498 @retval Others Other errors as indicated.\r
499**/\r
500NET_BUF *\r
501IScsiPrepareLoginReq (\r
502 IN OUT ISCSI_CONNECTION *Conn\r
503 )\r
504{\r
505 ISCSI_SESSION *Session;\r
506 NET_BUF *Nbuf;\r
507 ISCSI_LOGIN_REQUEST *LoginReq;\r
508 EFI_STATUS Status;\r
509\r
510 Session = Conn->Session;\r
511\r
512 Nbuf = NetbufAlloc (sizeof (ISCSI_LOGIN_REQUEST) + DEFAULT_MAX_RECV_DATA_SEG_LEN);\r
513 if (Nbuf == NULL) {\r
514 return NULL;\r
515 }\r
516\r
517 LoginReq = (ISCSI_LOGIN_REQUEST *) NetbufAllocSpace (Nbuf, sizeof (ISCSI_LOGIN_REQUEST), NET_BUF_TAIL);\r
518 ASSERT (LoginReq != NULL);\r
519 ZeroMem (LoginReq, sizeof (ISCSI_LOGIN_REQUEST));\r
520\r
521 //\r
522 // Init the login request pdu\r
523 //\r
524 ISCSI_SET_OPCODE (LoginReq, ISCSI_OPCODE_LOGIN_REQ, ISCSI_REQ_IMMEDIATE);\r
525 ISCSI_SET_STAGES (LoginReq, Conn->CurrentStage, Conn->NextStage);\r
526 LoginReq->VersionMax = ISCSI_VERSION_MAX;\r
527 LoginReq->VersionMin = ISCSI_VERSION_MIN;\r
528 LoginReq->Tsih = HTONS (Session->Tsih);\r
529 LoginReq->InitiatorTaskTag = HTONL (Session->InitiatorTaskTag);\r
530 LoginReq->Cid = HTONS (Conn->Cid);\r
531 LoginReq->CmdSN = HTONL (Session->CmdSN);\r
532\r
533 //\r
534 // For the first Login Request on a coonection this is ExpStatSN for the\r
535 // old connection and this field is only valid if the Login Request restarts\r
536 // a connection.\r
537 // For subsequent Login Requests it is used to acknowledge the Login Responses\r
538 // with their increasing StatSN values.\r
539 //\r
540 LoginReq->ExpStatSN = HTONL (Conn->ExpStatSN);\r
541 CopyMem (LoginReq->Isid, Session->Isid, sizeof (LoginReq->Isid));\r
542\r
543 if (Conn->PartialRspRcvd) {\r
544 //\r
545 // A partial response, initiator must send an empty Login Request.\r
546 //\r
547 return Nbuf;\r
548 }\r
549\r
550 switch (Conn->CurrentStage) {\r
551 case ISCSI_SECURITY_NEGOTIATION:\r
552 Status = IScsiCHAPToSendReq (Conn, Nbuf);\r
553 break;\r
554\r
555 case ISCSI_LOGIN_OPERATIONAL_NEGOTIATION:\r
556 Status = IScsiFillOpParams (Conn, Nbuf);\r
557 ISCSI_SET_FLAG (LoginReq, ISCSI_LOGIN_REQ_PDU_FLAG_TRANSIT);\r
558 break;\r
559\r
560 default:\r
561 //\r
562 // something error happens...\r
563 //\r
564 Status = EFI_DEVICE_ERROR;\r
565 break;\r
566 }\r
567\r
568 if (EFI_ERROR (Status)) {\r
569 NetbufFree (Nbuf);\r
570 Nbuf = NULL;\r
571 } else {\r
572 //\r
573 // Pad the data segment if needed.\r
574 //\r
575 IScsiPadSegment (Nbuf, ISCSI_GET_DATASEG_LEN (LoginReq));\r
576 //\r
577 // Check whether we will issue the stage transition signal?\r
578 //\r
579 Conn->TransitInitiated = (BOOLEAN) ISCSI_FLAG_ON (LoginReq, ISCSI_LOGIN_REQ_PDU_FLAG_TRANSIT);\r
580 }\r
581\r
582 return Nbuf;\r
583}\r
584\r
585/**\r
586 Process the iSCSI Login Response.\r
587\r
588 @param[in, out] Conn The connection on which the iSCSI login response is received.\r
589 @param[in, out] Pdu The iSCSI login response PDU.\r
590\r
591 @retval EFI_SUCCESS The iSCSI login response PDU is processed and all check are passed.\r
592 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol error happened.\r
593 @retval EFI_MEDIA_CHANGED Target is redirected.\r
594 @retval Others Other errors as indicated.\r
595**/\r
596EFI_STATUS\r
597IScsiProcessLoginRsp (\r
598 IN OUT ISCSI_CONNECTION *Conn,\r
599 IN OUT NET_BUF *Pdu\r
600 )\r
601{\r
602 EFI_STATUS Status;\r
603 ISCSI_SESSION *Session;\r
604 ISCSI_LOGIN_RESPONSE *LoginRsp;\r
605 BOOLEAN Transit;\r
606 BOOLEAN Continue;\r
607 UINT8 CurrentStage;\r
608 UINT8 NextStage;\r
609 UINT8 *DataSeg;\r
610 UINT32 DataSegLen;\r
611\r
612 Session = Conn->Session;\r
613\r
614 LoginRsp = (ISCSI_LOGIN_RESPONSE *) NetbufGetByte (Pdu, 0, NULL);\r
615 if (LoginRsp == NULL) {\r
616 return EFI_PROTOCOL_ERROR;\r
617 }\r
618 if (!ISCSI_CHECK_OPCODE (LoginRsp, ISCSI_OPCODE_LOGIN_RSP)) {\r
619 //\r
620 // It's not a Login Response\r
621 //\r
622 return EFI_PROTOCOL_ERROR;\r
623 }\r
624 //\r
625 // Get the data segment if any.\r
626 //\r
627 DataSegLen = ISCSI_GET_DATASEG_LEN (LoginRsp);\r
628 if (DataSegLen != 0) {\r
629 DataSeg = NetbufGetByte (Pdu, sizeof (ISCSI_LOGIN_RESPONSE), NULL);\r
630 } else {\r
631 DataSeg = NULL;\r
632 }\r
633 //\r
634 // Check the status class in the login response PDU.\r
635 //\r
636 switch (LoginRsp->StatusClass) {\r
637 case ISCSI_LOGIN_STATUS_SUCCESS:\r
638 //\r
639 // Just break here, the response and the data segment will be processed later.\r
640 //\r
641 break;\r
642\r
643 case ISCSI_LOGIN_STATUS_REDIRECTION:\r
644 //\r
645 // The target may be moved to a different address\r
646 //\r
647 if (DataSeg == NULL) {\r
648 return EFI_PROTOCOL_ERROR;\r
649 }\r
650 //\r
651 // Process the TargetAddress key-value strings in the data segment to update the\r
652 // target address info.\r
653 //\r
654 Status = IScsiUpdateTargetAddress (Session, (CHAR8 *)DataSeg, DataSegLen);\r
655 if (EFI_ERROR (Status)) {\r
656 return Status;\r
657 }\r
658 //\r
659 // Session will be restarted on this error status because the Target is\r
660 // redirected by this Login Response.\r
661 //\r
662 return EFI_MEDIA_CHANGED;\r
663\r
664 default:\r
665 //\r
666 // Initiator Error, Target Error, or any other undefined error code.\r
667 //\r
668 return EFI_PROTOCOL_ERROR;\r
669 }\r
670 //\r
671 // The status is sucess, extract the wanted fields from the header segment.\r
672 //\r
673 Transit = (BOOLEAN) ISCSI_FLAG_ON (LoginRsp, ISCSI_LOGIN_RSP_PDU_FLAG_TRANSIT);\r
674 Continue = (BOOLEAN) ISCSI_FLAG_ON (LoginRsp, ISCSI_LOGIN_RSP_PDU_FLAG_CONTINUE);\r
675\r
676 CurrentStage = (UINT8) ISCSI_GET_CURRENT_STAGE (LoginRsp);\r
677 NextStage = (UINT8) ISCSI_GET_NEXT_STAGE (LoginRsp);\r
678\r
679 LoginRsp->InitiatorTaskTag = NTOHL (LoginRsp->InitiatorTaskTag);\r
680\r
681 if ((Transit && Continue) ||\r
682 (CurrentStage != Conn->CurrentStage) ||\r
683 (!Conn->TransitInitiated && Transit) ||\r
684 (Transit && (NextStage != Conn->NextStage)) ||\r
685 (CompareMem (Session->Isid, LoginRsp->Isid, sizeof (LoginRsp->Isid)) != 0) ||\r
686 (LoginRsp->InitiatorTaskTag != Session->InitiatorTaskTag)\r
687 ) {\r
688 //\r
689 // A Login Response with the C bit set to 1 MUST have the T bit set to 0;\r
690 // The CSG in the Login Response MUST be the same with the I-end of this connection;\r
691 // The T bit can't be 1 if the last Login Response sent by the initiator doesn't\r
692 // initiate the transistion;\r
693 // The NSG MUST be the same with the I-end of this connection if Transit is required.\r
694 // The ISID in the Login Response MUST be the same with this session.\r
695 //\r
696 return EFI_PROTOCOL_ERROR;\r
697 }\r
698\r
699 LoginRsp->StatSN = NTOHL (LoginRsp->StatSN);\r
700 LoginRsp->ExpCmdSN = NTOHL (LoginRsp->ExpCmdSN);\r
701 LoginRsp->MaxCmdSN = NTOHL (LoginRsp->MaxCmdSN);\r
702\r
703 if ((Conn->CurrentStage == ISCSI_SECURITY_NEGOTIATION) && (Conn->CHAPStep == ISCSI_CHAP_INITIAL)) {\r
704 //\r
705 // If the Login Request is a leading Login Request, the target MUST use\r
706 // the value presented in CmdSN as the target value for ExpCmdSN.\r
707 //\r
708 if ((Session->State == SESSION_STATE_FREE) && (Session->CmdSN != LoginRsp->ExpCmdSN)) {\r
709 return EFI_PROTOCOL_ERROR;\r
710 }\r
711\r
712 //\r
713 // It's the initial Login Response, initialize the local ExpStatSN, MaxCmdSN\r
714 // and ExpCmdSN.\r
715 //\r
716 Conn->ExpStatSN = LoginRsp->StatSN + 1;\r
717 Session->MaxCmdSN = LoginRsp->MaxCmdSN;\r
718 Session->ExpCmdSN = LoginRsp->ExpCmdSN;\r
719 } else {\r
720 //\r
721 // Check the StatSN of this PDU\r
722 //\r
723 Status = IScsiCheckSN (&Conn->ExpStatSN, LoginRsp->StatSN);\r
724 if (!EFI_ERROR (Status)) {\r
725 //\r
726 // Update the MaxCmdSN and ExpCmdSN\r
727 //\r
728 IScsiUpdateCmdSN (Session, LoginRsp->MaxCmdSN, LoginRsp->ExpCmdSN);\r
729 } else {\r
730 return Status;\r
731 }\r
732 }\r
733 //\r
734 // Trim off the header segment.\r
735 //\r
736 NetbufTrim (Pdu, sizeof (ISCSI_LOGIN_RESPONSE), NET_BUF_HEAD);\r
737\r
738 //\r
739 // Queue this login response first in case it's a partial response so that\r
740 // later when the full response list is received we can combine these scattered\r
741 // responses' data segment and then process it.\r
742 //\r
743 NET_GET_REF (Pdu);\r
744 NetbufQueAppend (&Conn->RspQue, Pdu);\r
745\r
746 Conn->PartialRspRcvd = Continue;\r
747 if (Continue) {\r
748 //\r
749 // It's a partial response, have to wait for another or more Request/Response\r
750 // conversations to get the full response.\r
751 //\r
752 return EFI_SUCCESS;\r
753 }\r
754\r
755 switch (CurrentStage) {\r
756 case ISCSI_SECURITY_NEGOTIATION:\r
757 //\r
758 // In security negotiation stage, let CHAP module handle it.\r
759 //\r
760 Status = IScsiCHAPOnRspReceived (Conn);\r
761 break;\r
762\r
763 case ISCSI_LOGIN_OPERATIONAL_NEGOTIATION:\r
764 //\r
765 // Response received with negotiation resonse on iSCSI parameters, check them.\r
766 //\r
767 Status = IScsiCheckOpParams (Conn);\r
768 break;\r
769\r
770 default:\r
771 //\r
772 // Should never get here.\r
773 //\r
774 Status = EFI_PROTOCOL_ERROR;\r
775 break;\r
776 }\r
777\r
778 if (Transit && (Status == EFI_SUCCESS)) {\r
779 //\r
780 // Do the state transition.\r
781 //\r
782 Conn->CurrentStage = Conn->NextStage;\r
783\r
784 if (Conn->CurrentStage == ISCSI_LOGIN_OPERATIONAL_NEGOTIATION) {\r
785 Conn->NextStage = ISCSI_FULL_FEATURE_PHASE;\r
786 } else {\r
787 //\r
788 // CurrentStage is iSCSI Full Feature, it's the Login-Final Response,\r
789 // get the TSIH from the Login Response.\r
790 //\r
791 Session->Tsih = NTOHS (LoginRsp->Tsih);\r
792 }\r
793 }\r
794 //\r
795 // Flush the response(s) received.\r
796 //\r
797 NetbufQueFlush (&Conn->RspQue);\r
798\r
799 return Status;\r
800}\r
801\r
802/**\r
803 Updated the target information according the data received in the iSCSI\r
804 login response with an target redirection status.\r
805\r
806 @param[in, out] Session The iSCSI session.\r
807 @param[in] Data The data segment which should contain the\r
808 TargetAddress key-value list.\r
809 @param[in] Len Length of the data.\r
810 \r
811 @retval EFI_SUCCESS The target address is updated.\r
812 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
813 @retval EFI_NOT_FOUND The TargetAddress key is not found.\r
814 @retval Others Other errors as indicated.\r
815**/\r
816EFI_STATUS\r
817IScsiUpdateTargetAddress (\r
818 IN OUT ISCSI_SESSION *Session,\r
819 IN CHAR8 *Data,\r
820 IN UINT32 Len\r
821 )\r
822{\r
823 LIST_ENTRY *KeyValueList;\r
824 CHAR8 *TargetAddress;\r
825 CHAR8 *IpStr;\r
826 EFI_STATUS Status;\r
827 UINTN Number;\r
828\r
829 KeyValueList = IScsiBuildKeyValueList (Data, Len);\r
830 if (KeyValueList == NULL) {\r
831 return EFI_OUT_OF_RESOURCES;\r
832 }\r
833\r
834 Status = EFI_NOT_FOUND;\r
835\r
836 while (TRUE) {\r
837 TargetAddress = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_TARGET_ADDRESS);\r
838 if (TargetAddress == NULL) {\r
839 break;\r
840 }\r
841\r
842 if (!NET_IS_DIGIT (TargetAddress[0])) {\r
843 //\r
844 // The domainname of the target may be presented in three formats: a DNS host name,\r
845 // a dotted-decimal IPv4 address, or a bracketed IPv6 address. Only accept dotted\r
846 // IPv4 address.\r
847 //\r
848 continue;\r
849 }\r
850\r
851 IpStr = TargetAddress;\r
852\r
853 while ((*TargetAddress != 0) && (*TargetAddress != ':') && (*TargetAddress != ',')) {\r
854 //\r
855 // NULL, ':' or ',' ends the IPv4 string.\r
856 //\r
857 TargetAddress++;\r
858 }\r
859\r
860 if (*TargetAddress == ',') {\r
861 //\r
862 // Comma and the portal group tag MUST be ommitted if the TargetAddress is sent\r
863 // as the result of a redirection.\r
864 //\r
865 continue;\r
866 } else if (*TargetAddress == ':') {\r
867 *TargetAddress = '\0';\r
868\r
869 TargetAddress++;\r
870\r
871 Number = AsciiStrDecimalToUintn (TargetAddress);\r
872 if (Number > 0xFFFF) {\r
873 continue;\r
874 } else {\r
875 Session->ConfigData.NvData.TargetPort = (UINT16) Number;\r
876 }\r
877 } else {\r
878 //\r
879 // The string only contains the IPv4 address. Use the well known port.\r
880 //\r
881 Session->ConfigData.NvData.TargetPort = ISCSI_WELL_KNOWN_PORT;\r
882 }\r
883 //\r
884 // Update the target IP address.\r
885 //\r
886 Status = IScsiAsciiStrToIp (IpStr, &Session->ConfigData.NvData.TargetIp);\r
887 if (EFI_ERROR (Status)) {\r
888 continue;\r
889 } else {\r
890 break;\r
891 }\r
892 }\r
893\r
894 IScsiFreeKeyValueList (KeyValueList);\r
895\r
896 return Status;\r
897}\r
898\r
899/**\r
900 The callback function to free the net buffer list.\r
901\r
902 @param[in] Arg The opaque parameter.\r
903**/\r
904VOID\r
905EFIAPI\r
906IScsiFreeNbufList (\r
907 VOID *Arg\r
908 )\r
909{\r
910 ASSERT (Arg != NULL);\r
911\r
912 NetbufFreeList ((LIST_ENTRY *) Arg);\r
913 FreePool (Arg);\r
914}\r
915\r
916/**\r
917 The callback function called in NetBufFree, it does nothing.\r
918\r
919 @param[in] Arg The opaque parameter.\r
920**/\r
921VOID\r
922EFIAPI\r
923IScsiNbufExtFree (\r
924 VOID *Arg\r
925 )\r
926{\r
927}\r
928\r
929/**\r
930 Receive an iSCSI response PDU. An iSCSI response PDU contains an iSCSI PDU header and\r
931 an optional data segment. The two parts will be put into two blocks of buffers in the\r
932 net buffer. The digest check will be conducted in this function if needed and the digests\r
933 will be trimmed from the PDU buffer.\r
934\r
935 @param[in] Conn The iSCSI connection to receive data from.\r
936 @param[out] Pdu The received iSCSI pdu.\r
937 @param[in] Context The context used to describe information on the caller provided\r
938 buffer to receive data segment of the iSCSI pdu, it's optional.\r
939 @param[in] HeaderDigest Whether there will be header digest received.\r
940 @param[in] DataDigest Whether there will be data digest.\r
941 @param[in] TimeoutEvent The timeout event, it's optional.\r
942\r
943 @retval EFI_SUCCESS An iSCSI pdu is received.\r
944 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
945 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol error happened.\r
946 @retval Others Other errors as indicated.\r
947**/\r
948EFI_STATUS\r
949IScsiReceivePdu (\r
950 IN ISCSI_CONNECTION *Conn,\r
951 OUT NET_BUF **Pdu,\r
952 IN ISCSI_IN_BUFFER_CONTEXT *Context, OPTIONAL\r
953 IN BOOLEAN HeaderDigest,\r
954 IN BOOLEAN DataDigest,\r
955 IN EFI_EVENT TimeoutEvent OPTIONAL\r
956 )\r
957{\r
958 LIST_ENTRY *NbufList;\r
959 UINT32 Len;\r
960 NET_BUF *PduHdr;\r
961 UINT8 *Header;\r
962 EFI_STATUS Status;\r
963 UINT32 PadLen;\r
964 UINT32 InDataOffset;\r
965 NET_FRAGMENT Fragment[2];\r
966 UINT32 FragmentCount;\r
967 NET_BUF *DataSeg;\r
968 UINT32 PadAndCRC32[2];\r
969\r
970 NbufList = AllocatePool (sizeof (LIST_ENTRY));\r
971 if (NbufList == NULL) {\r
972 return EFI_OUT_OF_RESOURCES;\r
973 }\r
974\r
975 InitializeListHead (NbufList);\r
976\r
977 //\r
978 // The header digest will be received together with the PDU header if exists.\r
979 //\r
980 Len = sizeof (ISCSI_BASIC_HEADER) + (HeaderDigest ? sizeof (UINT32) : 0);\r
981 PduHdr = NetbufAlloc (Len);\r
982 if (PduHdr == NULL) {\r
983 FreePool (NbufList);\r
984 return EFI_OUT_OF_RESOURCES;\r
985 }\r
986\r
987 Header = NetbufAllocSpace (PduHdr, Len, NET_BUF_TAIL);\r
988 ASSERT (Header != NULL);\r
989 InsertTailList (NbufList, &PduHdr->List);\r
990\r
991 //\r
992 // First step, receive the BHS of the PDU.\r
993 //\r
994 Status = Tcp4IoReceive (&Conn->Tcp4Io, PduHdr, FALSE, TimeoutEvent);\r
995 if (EFI_ERROR (Status)) {\r
996 goto ON_EXIT;\r
997 }\r
998\r
999 if (HeaderDigest) {\r
1000 //\r
1001 // TODO: check the header-digest.\r
1002 //\r
1003 //\r
1004 // Trim off the digest.\r
1005 //\r
1006 NetbufTrim (PduHdr, sizeof (UINT32), NET_BUF_TAIL);\r
1007 }\r
1008\r
1009 Len = ISCSI_GET_DATASEG_LEN (Header);\r
1010 if (Len == 0) {\r
1011 //\r
1012 // No data segment.Form the pdu from a list of pdu segments.\r
1013 //\r
1014 *Pdu = NetbufFromBufList (NbufList, 0, 0, IScsiFreeNbufList, NbufList);\r
1015 if (*Pdu == NULL) {\r
1016 Status = EFI_OUT_OF_RESOURCES;\r
1017 goto ON_EXIT;\r
1018 }\r
1019 return Status;\r
1020 }\r
1021 //\r
1022 // Get the length of the padding bytes of the data segment.\r
1023 //\r
1024 PadLen = ISCSI_GET_PAD_LEN (Len);\r
1025\r
1026 switch (ISCSI_GET_OPCODE (Header)) {\r
1027 case ISCSI_OPCODE_SCSI_DATA_IN:\r
1028 //\r
1029 // Try to use the buffer described by Context if the PDU is an\r
1030 // iSCSI SCSI data in pdu so as to reduce memory copy overhead.\r
1031 //\r
1032 InDataOffset = ISCSI_GET_BUFFER_OFFSET (Header);\r
1033 if ((Context == NULL) || ((InDataOffset + Len) > Context->InDataLen)) {\r
1034 Status = EFI_PROTOCOL_ERROR;\r
1035 goto ON_EXIT;\r
1036 }\r
1037\r
1038 Fragment[0].Len = Len;\r
1039 Fragment[0].Bulk = Context->InData + InDataOffset;\r
1040\r
1041 if (DataDigest || (PadLen != 0)) {\r
1042 //\r
1043 // The data segment is padded, use two fragments to receive it.\r
1044 // The first to receive the useful data. The second to receive the padding.\r
1045 //\r
1046 Fragment[1].Len = PadLen + (DataDigest ? sizeof (UINT32) : 0);\r
1047 Fragment[1].Bulk = (UINT8 *)PadAndCRC32 + (4 - PadLen);\r
1048 FragmentCount = 2;\r
1049 } else {\r
1050 FragmentCount = 1;\r
1051 }\r
1052\r
1053 DataSeg = NetbufFromExt (&Fragment[0], FragmentCount, 0, 0, IScsiNbufExtFree, NULL);\r
1054 if (DataSeg == NULL) {\r
1055 Status = EFI_OUT_OF_RESOURCES;\r
1056 goto ON_EXIT;\r
1057 }\r
1058\r
1059 break;\r
1060\r
1061 case ISCSI_OPCODE_SCSI_RSP:\r
1062 case ISCSI_OPCODE_NOP_IN:\r
1063 case ISCSI_OPCODE_LOGIN_RSP:\r
1064 case ISCSI_OPCODE_TEXT_RSP:\r
1065 case ISCSI_OPCODE_ASYNC_MSG:\r
1066 case ISCSI_OPCODE_REJECT:\r
1067 case ISCSI_OPCODE_VENDOR_T0:\r
1068 case ISCSI_OPCODE_VENDOR_T1:\r
1069 case ISCSI_OPCODE_VENDOR_T2:\r
1070 //\r
1071 // Allocate buffer to receive the data segment.\r
1072 //\r
1073 Len += PadLen + (DataDigest ? sizeof (UINT32) : 0);\r
1074 DataSeg = NetbufAlloc (Len);\r
1075 if (DataSeg == NULL) {\r
1076 Status = EFI_OUT_OF_RESOURCES;\r
1077 goto ON_EXIT;\r
1078 }\r
1079\r
1080 NetbufAllocSpace (DataSeg, Len, NET_BUF_TAIL);\r
1081 break;\r
1082\r
1083 default:\r
1084 Status = EFI_PROTOCOL_ERROR;\r
1085 goto ON_EXIT;\r
1086 }\r
1087\r
1088 InsertTailList (NbufList, &DataSeg->List);\r
1089\r
1090 //\r
1091 // Receive the data segment with the data digest if any.\r
1092 //\r
1093 Status = Tcp4IoReceive (&Conn->Tcp4Io, DataSeg, FALSE, TimeoutEvent);\r
1094 if (EFI_ERROR (Status)) {\r
1095 goto ON_EXIT;\r
1096 }\r
1097\r
1098 if (DataDigest) {\r
1099 //\r
1100 // TODO: Check the data digest.\r
1101 //\r
1102 NetbufTrim (DataSeg, sizeof (UINT32), NET_BUF_TAIL);\r
1103 }\r
1104\r
1105 if (PadLen != 0) {\r
1106 //\r
1107 // Trim off the padding bytes in the data segment.\r
1108 //\r
1109 NetbufTrim (DataSeg, PadLen, NET_BUF_TAIL);\r
1110 }\r
1111\r
1112 //\r
1113 // Form the pdu from a list of pdu segments.\r
1114 //\r
1115 *Pdu = NetbufFromBufList (NbufList, 0, 0, IScsiFreeNbufList, NbufList);\r
1116 if (*Pdu == NULL) {\r
1117 Status = EFI_OUT_OF_RESOURCES;\r
1118 }\r
1119\r
1120ON_EXIT:\r
1121\r
1122 if (EFI_ERROR (Status)) {\r
1123 //\r
1124 // Free the Nbufs in this NbufList and the NbufList itself.\r
1125 //\r
1126 IScsiFreeNbufList (NbufList);\r
1127 }\r
1128\r
1129 return Status;\r
1130}\r
1131\r
1132/**\r
1133 Check and get the result of the prameter negotiation.\r
1134\r
1135 @param[in, out] Conn The connection in iSCSI login.\r
1136\r
1137 @retval EFI_SUCCESS The parmeter check is passed and negotiation is finished.\r
1138 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol error happened.\r
1139 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
1140**/\r
1141EFI_STATUS\r
1142IScsiCheckOpParams (\r
1143 IN OUT ISCSI_CONNECTION *Conn\r
1144 )\r
1145{\r
1146 EFI_STATUS Status;\r
1147 LIST_ENTRY *KeyValueList;\r
1148 CHAR8 *Data;\r
1149 UINT32 Len;\r
1150 ISCSI_SESSION *Session;\r
1151 CHAR8 *Value;\r
1152 UINTN NumericValue;\r
1153\r
1154 ASSERT (Conn->RspQue.BufNum != 0);\r
1155\r
1156 Session = Conn->Session;\r
1157\r
1158 Len = Conn->RspQue.BufSize;\r
1159 Data = AllocatePool (Len);\r
1160 if (Data == NULL) {\r
1161 return EFI_OUT_OF_RESOURCES;\r
1162 }\r
1163\r
1164 NetbufQueCopy (&Conn->RspQue, 0, Len, (UINT8 *) Data);\r
1165\r
1166 Status = EFI_PROTOCOL_ERROR;\r
1167\r
1168 //\r
1169 // Extract the Key-Value pairs into a list.\r
1170 //\r
1171 KeyValueList = IScsiBuildKeyValueList (Data, Len);\r
1172 if (KeyValueList == NULL) {\r
1173 FreePool (Data);\r
1174 return Status;\r
1175 }\r
1176 //\r
1177 // HeaderDigest\r
1178 //\r
1179 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_HEADER_DIGEST);\r
1180 if (Value == NULL) {\r
1181 goto ON_ERROR;\r
1182 }\r
1183\r
1184 if (AsciiStrCmp (Value, "CRC32") == 0) {\r
1185 if (Conn->HeaderDigest != IScsiDigestCRC32) {\r
1186 goto ON_ERROR;\r
1187 }\r
1188 } else if (AsciiStrCmp (Value, ISCSI_KEY_VALUE_NONE) == 0) {\r
1189 Conn->HeaderDigest = IScsiDigestNone;\r
1190 } else {\r
1191 goto ON_ERROR;\r
1192 }\r
1193 //\r
1194 // DataDigest\r
1195 //\r
1196 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_DATA_DIGEST);\r
1197 if (Value == NULL) {\r
1198 goto ON_ERROR;\r
1199 }\r
1200\r
1201 if (AsciiStrCmp (Value, "CRC32") == 0) {\r
1202 if (Conn->DataDigest != IScsiDigestCRC32) {\r
1203 goto ON_ERROR;\r
1204 }\r
1205 } else if (AsciiStrCmp (Value, ISCSI_KEY_VALUE_NONE) == 0) {\r
1206 Conn->DataDigest = IScsiDigestNone;\r
1207 } else {\r
1208 goto ON_ERROR;\r
1209 }\r
1210 //\r
1211 // ErrorRecoveryLevel, result fuction is Minimum.\r
1212 //\r
1213 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_ERROR_RECOVERY_LEVEL);\r
1214 if (Value == NULL) {\r
1215 goto ON_ERROR;\r
1216 }\r
1217\r
1218 NumericValue = AsciiStrDecimalToUintn (Value);\r
1219 if (NumericValue > 2) {\r
1220 goto ON_ERROR;\r
1221 }\r
1222\r
1223 Session->ErrorRecoveryLevel = (UINT8) MIN (Session->ErrorRecoveryLevel, NumericValue);\r
1224\r
1225 //\r
1226 // InitialR2T, result function is OR.\r
1227 //\r
1228 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_INITIAL_R2T);\r
1229 if (Value == NULL) {\r
1230 goto ON_ERROR;\r
1231 }\r
1232\r
1233 Session->InitialR2T = (BOOLEAN) (Session->InitialR2T || (AsciiStrCmp (Value, "Yes") == 0));\r
1234\r
1235 //\r
1236 // ImmediateData, result function is AND.\r
1237 //\r
1238 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_IMMEDIATE_DATA);\r
1239 if (Value == NULL) {\r
1240 goto ON_ERROR;\r
1241 }\r
1242\r
1243 Session->ImmediateData = (BOOLEAN) (Session->ImmediateData && (AsciiStrCmp (Value, "Yes") == 0));\r
1244\r
1245 //\r
1246 // MaxRecvDataSegmentLength is declarative.\r
1247 //\r
1248 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_MAX_RECV_DATA_SEGMENT_LENGTH);\r
1249 if (Value != NULL) {\r
1250 Conn->MaxRecvDataSegmentLength = (UINT32) AsciiStrDecimalToUintn (Value);\r
1251 }\r
1252 //\r
1253 // MaxBurstLength, result funtion is Mininum.\r
1254 //\r
1255 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_MAX_BURST_LENGTH);\r
1256 if (Value == NULL) {\r
1257 goto ON_ERROR;\r
1258 }\r
1259\r
1260 NumericValue = AsciiStrDecimalToUintn (Value);\r
1261 Session->MaxBurstLength = (UINT32) MIN (Session->MaxBurstLength, NumericValue);\r
1262\r
1263 //\r
1264 // FirstBurstLength, result function is Minimum. Irrelevant when InitialR2T=Yes and\r
1265 // ImmediateData=No.\r
1266 // This Key/Value is negotiation type.\r
1267 //\r
1268 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_FIRST_BURST_LENGTH);\r
1269 if (Value == NULL) {\r
1270 goto ON_ERROR;\r
1271 }\r
1272\r
1273 NumericValue = AsciiStrDecimalToUintn (Value);\r
1274 Session->FirstBurstLength = (UINT32) MIN (Session->FirstBurstLength, NumericValue);\r
1275\r
1276 //\r
1277 // MaxConnections, result function is Minimum.\r
1278 //\r
1279 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_MAX_CONNECTIONS);\r
1280 if (Value == NULL) {\r
1281 goto ON_ERROR;\r
1282 }\r
1283\r
1284 NumericValue = AsciiStrDecimalToUintn (Value);\r
1285 if ((NumericValue == 0) || (NumericValue > 65535)) {\r
1286 goto ON_ERROR;\r
1287 }\r
1288\r
1289 Session->MaxConnections = (UINT32) MIN (Session->MaxConnections, NumericValue);\r
1290\r
1291 //\r
1292 // DataPDUInOrder, result function is OR.\r
1293 //\r
1294 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_DATA_PDU_IN_ORDER);\r
1295 if (Value == NULL) {\r
1296 goto ON_ERROR;\r
1297 }\r
1298\r
1299 Session->DataPDUInOrder = (BOOLEAN) (Session->DataPDUInOrder || (AsciiStrCmp (Value, "Yes") == 0));\r
1300\r
1301 //\r
1302 // DataSequenceInorder, result function is OR.\r
1303 //\r
1304 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_DATA_SEQUENCE_IN_ORDER);\r
1305 if (Value == NULL) {\r
1306 goto ON_ERROR;\r
1307 }\r
1308\r
1309 Session->DataSequenceInOrder = (BOOLEAN) (Session->DataSequenceInOrder || (AsciiStrCmp (Value, "Yes") == 0));\r
1310\r
1311 //\r
1312 // DefaultTime2Wait, result function is Maximum.\r
1313 //\r
1314 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_DEFAULT_TIME2WAIT);\r
1315 if (Value == NULL) {\r
1316 goto ON_ERROR;\r
1317 }\r
1318\r
1319 NumericValue = AsciiStrDecimalToUintn (Value);\r
1320 if (NumericValue == 0) {\r
1321 Session->DefaultTime2Wait = 0;\r
1322 } else if (NumericValue > 3600) {\r
1323 goto ON_ERROR;\r
1324 } else {\r
1325 Session->DefaultTime2Wait = (UINT32) MAX (Session->DefaultTime2Wait, NumericValue);\r
1326 }\r
1327 //\r
1328 // DefaultTime2Retain, result function is Minimum.\r
1329 //\r
1330 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_DEFAULT_TIME2RETAIN);\r
1331 if (Value == NULL) {\r
1332 goto ON_ERROR;\r
1333 }\r
1334\r
1335 NumericValue = AsciiStrDecimalToUintn (Value);\r
1336 if (NumericValue == 0) {\r
1337 Session->DefaultTime2Retain = 0;\r
1338 } else if (NumericValue > 3600) {\r
1339 goto ON_ERROR;\r
1340 } else {\r
1341 Session->DefaultTime2Retain = (UINT32) MIN (Session->DefaultTime2Retain, NumericValue);\r
1342 }\r
1343 //\r
1344 // MaxOutstandingR2T, result function is Minimum.\r
1345 //\r
1346 Value = IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_MAX_OUTSTANDING_R2T);\r
1347 if (Value == NULL) {\r
1348 goto ON_ERROR;\r
1349 }\r
1350\r
1351 NumericValue = AsciiStrDecimalToUintn (Value);\r
1352 if ((NumericValue == 0) || (NumericValue > 65535)) {\r
1353 goto ON_ERROR;\r
1354 }\r
1355\r
1356 Session->MaxOutstandingR2T = (UINT16) MIN (Session->MaxOutstandingR2T, NumericValue);\r
1357\r
1358 //\r
1359 // Remove declarative key-value paris if any.\r
1360 //\r
1361 IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_SESSION_TYPE);\r
1362 IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_TARGET_ALIAS);\r
1363 IScsiGetValueByKeyFromList (KeyValueList, ISCSI_KEY_TARGET_PORTAL_GROUP_TAG);\r
1364\r
1365 if (IsListEmpty (KeyValueList)) {\r
1366 //\r
1367 // Succeed if no more keys in the list.\r
1368 //\r
1369 Status = EFI_SUCCESS;\r
1370 }\r
1371\r
1372ON_ERROR:\r
1373\r
1374 IScsiFreeKeyValueList (KeyValueList);\r
1375\r
1376 FreePool (Data);\r
1377\r
1378 return Status;\r
1379}\r
1380\r
1381/**\r
1382 Fill the oprational prameters.\r
1383\r
1384 @param[in] Conn The connection in iSCSI login.\r
1385 @param[in, out] Pdu The iSCSI login request PDU to fill the parameters.\r
1386\r
1387 @retval EFI_SUCCESS The parmeters are filled into the iSCSI login request PDU.\r
1388**/\r
1389EFI_STATUS\r
1390IScsiFillOpParams (\r
1391 IN ISCSI_CONNECTION *Conn,\r
1392 IN OUT NET_BUF *Pdu\r
1393 )\r
1394{\r
1395 ISCSI_SESSION *Session;\r
1396 CHAR8 Value[256];\r
1397\r
1398 Session = Conn->Session;\r
1399\r
1400 AsciiSPrint (Value, sizeof (Value), "%a", (Conn->HeaderDigest == IScsiDigestCRC32) ? "None,CRC32" : "None");\r
1401 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_HEADER_DIGEST, Value);\r
1402\r
1403 AsciiSPrint (Value, sizeof (Value), "%a", (Conn->DataDigest == IScsiDigestCRC32) ? "None,CRC32" : "None");\r
1404 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_DATA_DIGEST, Value);\r
1405\r
1406 AsciiSPrint (Value, sizeof (Value), "%d", Session->ErrorRecoveryLevel);\r
1407 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_ERROR_RECOVERY_LEVEL, Value);\r
1408\r
1409 AsciiSPrint (Value, sizeof (Value), "%a", Session->InitialR2T ? "Yes" : "No");\r
1410 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_INITIAL_R2T, Value);\r
1411\r
1412 AsciiSPrint (Value, sizeof (Value), "%a", Session->ImmediateData ? "Yes" : "No");\r
1413 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_IMMEDIATE_DATA, Value);\r
1414\r
1415 AsciiSPrint (Value, sizeof (Value), "%d", MAX_RECV_DATA_SEG_LEN_IN_FFP);\r
1416 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_MAX_RECV_DATA_SEGMENT_LENGTH, Value);\r
1417\r
1418 AsciiSPrint (Value, sizeof (Value), "%d", Session->MaxBurstLength);\r
1419 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_MAX_BURST_LENGTH, Value);\r
1420\r
1421 AsciiSPrint (Value, sizeof (Value), "%d", Session->FirstBurstLength);\r
1422 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_FIRST_BURST_LENGTH, Value);\r
1423\r
1424 AsciiSPrint (Value, sizeof (Value), "%d", Session->MaxConnections);\r
1425 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_MAX_CONNECTIONS, Value);\r
1426\r
1427 AsciiSPrint (Value, sizeof (Value), "%a", Session->DataPDUInOrder ? "Yes" : "No");\r
1428 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_DATA_PDU_IN_ORDER, Value);\r
1429\r
1430 AsciiSPrint (Value, sizeof (Value), "%a", Session->DataSequenceInOrder ? "Yes" : "No");\r
1431 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_DATA_SEQUENCE_IN_ORDER, Value);\r
1432\r
1433 AsciiSPrint (Value, sizeof (Value), "%d", Session->DefaultTime2Wait);\r
1434 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_DEFAULT_TIME2WAIT, Value);\r
1435\r
1436 AsciiSPrint (Value, sizeof (Value), "%d", Session->DefaultTime2Retain);\r
1437 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_DEFAULT_TIME2RETAIN, Value);\r
1438\r
1439 AsciiSPrint (Value, sizeof (Value), "%d", Session->MaxOutstandingR2T);\r
1440 IScsiAddKeyValuePair (Pdu, ISCSI_KEY_MAX_OUTSTANDING_R2T, Value);\r
1441\r
1442 return EFI_SUCCESS;\r
1443}\r
1444\r
1445/**\r
1446 Pad the iSCSI AHS or data segment to an integer number of 4 byte words.\r
1447\r
1448 @param[in, out] Pdu The iSCSI pdu which contains segments to pad.\r
1449 @param[in] Len The length of the last semgnet in the PDU.\r
1450\r
1451 @retval EFI_SUCCESS The segment is padded or no need to pad it.\r
1452 @retval EFI_OUT_OF_RESOURCES There is not enough remaining free space to add the\r
1453 padding bytes.\r
1454**/\r
1455EFI_STATUS\r
1456IScsiPadSegment (\r
1457 IN OUT NET_BUF *Pdu,\r
1458 IN UINT32 Len\r
1459 )\r
1460{\r
1461 UINT32 PadLen;\r
1462 UINT8 *Data;\r
1463\r
1464 PadLen = ISCSI_GET_PAD_LEN (Len);\r
1465\r
1466 if (PadLen != 0) {\r
1467 Data = NetbufAllocSpace (Pdu, PadLen, NET_BUF_TAIL);\r
1468 if (Data == NULL) {\r
1469 return EFI_OUT_OF_RESOURCES;\r
1470 }\r
1471\r
1472 ZeroMem (Data, PadLen);\r
1473 }\r
1474\r
1475 return EFI_SUCCESS;\r
1476}\r
1477\r
1478/**\r
1479 Build a key-value list from the data segment.\r
1480\r
1481 @param[in] Data The data segment containing the key-value pairs.\r
1482 @param[in] Len Length of the data segment.\r
1483\r
1484 @return The key-value list.\r
1485 @retval NULL Other errors as indicated.\r
1486**/\r
1487LIST_ENTRY *\r
1488IScsiBuildKeyValueList (\r
1489 IN CHAR8 *Data,\r
1490 IN UINT32 Len\r
1491 )\r
1492{\r
1493 LIST_ENTRY *ListHead;\r
1494 ISCSI_KEY_VALUE_PAIR *KeyValuePair;\r
1495\r
1496 ListHead = AllocatePool (sizeof (LIST_ENTRY));\r
1497 if (ListHead == NULL) {\r
1498 return NULL;\r
1499 }\r
1500\r
1501 InitializeListHead (ListHead);\r
1502\r
1503 while (Len > 0) {\r
1504 KeyValuePair = AllocatePool (sizeof (ISCSI_KEY_VALUE_PAIR));\r
1505 if (KeyValuePair == NULL) {\r
1506 goto ON_ERROR;\r
1507 }\r
1508\r
1509 InitializeListHead (&KeyValuePair->List);\r
1510\r
1511 KeyValuePair->Key = Data;\r
1512\r
1513 while ((Len > 0) && (*Data != '=')) {\r
1514 Len--;\r
1515 Data++;\r
1516 }\r
1517\r
1518 if (*Data == '=') {\r
1519 *Data = '\0';\r
1520\r
1521 Data++;\r
1522 Len--;\r
1523 } else {\r
1524 FreePool (KeyValuePair);\r
1525 goto ON_ERROR;\r
1526 }\r
1527\r
1528 KeyValuePair->Value = Data;\r
1529\r
1530 InsertTailList (ListHead, &KeyValuePair->List);;\r
1531\r
1532 Data += AsciiStrLen (KeyValuePair->Value) + 1;\r
1533 Len -= (UINT32) AsciiStrLen (KeyValuePair->Value) + 1;\r
1534 }\r
1535\r
1536 return ListHead;\r
1537\r
1538ON_ERROR:\r
1539\r
1540 IScsiFreeKeyValueList (ListHead);\r
1541\r
1542 return NULL;\r
1543}\r
1544\r
1545/**\r
1546 Get the value string by the key name from the key-value list. If found,\r
1547 the key-value entry will be removed from the list.\r
1548\r
1549 @param[in, out] KeyValueList The key-value list.\r
1550 @param[in] Key The key name to find.\r
1551\r
1552 @return The value string.\r
1553**/\r
1554CHAR8 *\r
1555IScsiGetValueByKeyFromList (\r
1556 IN OUT LIST_ENTRY *KeyValueList,\r
1557 IN CHAR8 *Key\r
1558 )\r
1559{\r
1560 LIST_ENTRY *Entry;\r
1561 ISCSI_KEY_VALUE_PAIR *KeyValuePair;\r
1562 CHAR8 *Value;\r
1563\r
1564 Value = NULL;\r
1565\r
1566 NET_LIST_FOR_EACH (Entry, KeyValueList) {\r
1567 KeyValuePair = NET_LIST_USER_STRUCT (Entry, ISCSI_KEY_VALUE_PAIR, List);\r
1568\r
1569 if (AsciiStrCmp (KeyValuePair->Key, Key) == 0) {\r
1570 Value = KeyValuePair->Value;\r
1571\r
1572 RemoveEntryList (&KeyValuePair->List);\r
1573 FreePool (KeyValuePair);\r
1574 break;\r
1575 }\r
1576 }\r
1577\r
1578 return Value;\r
1579}\r
1580\r
1581/**\r
1582 Free the key-value list.\r
1583\r
1584 @param[in] KeyValueList The key-value list.\r
1585**/\r
1586VOID\r
1587IScsiFreeKeyValueList (\r
1588 IN LIST_ENTRY *KeyValueList\r
1589 )\r
1590{\r
1591 LIST_ENTRY *Entry;\r
1592 ISCSI_KEY_VALUE_PAIR *KeyValuePair;\r
1593\r
1594 while (!IsListEmpty (KeyValueList)) {\r
1595 Entry = NetListRemoveHead (KeyValueList);\r
1596 KeyValuePair = NET_LIST_USER_STRUCT (Entry, ISCSI_KEY_VALUE_PAIR, List);\r
1597\r
1598 FreePool (KeyValuePair);\r
1599 }\r
1600\r
1601 FreePool (KeyValueList);\r
1602}\r
1603\r
1604/**\r
1605 Normalize the iSCSI name according to RFC.\r
1606\r
1607 @param[in, out] Name The iSCSI name.\r
1608 @param[in] Len length of the iSCSI name.\r
1609\r
1610 @retval EFI_SUCCESS The iSCSI name is valid and normalized.\r
1611 @retval EFI_PROTOCOL_ERROR The iSCSI name is mal-formatted or not in the IQN format.\r
1612**/\r
1613EFI_STATUS\r
1614IScsiNormalizeName (\r
1615 IN OUT CHAR8 *Name,\r
1616 IN UINTN Len\r
1617 )\r
1618{\r
1619 UINTN Index;\r
1620\r
1621 for (Index = 0; Index < Len; Index++) {\r
1622 if (NET_IS_UPPER_CASE_CHAR (Name[Index])) {\r
1623 //\r
1624 // Convert the upper-case characters to lower-case ones\r
1625 //\r
1626 Name[Index] = (CHAR8) (Name[Index] - 'A' + 'a');\r
1627 }\r
1628\r
1629 if (!NET_IS_LOWER_CASE_CHAR (Name[Index]) &&\r
1630 !NET_IS_DIGIT (Name[Index]) &&\r
1631 (Name[Index] != '-') &&\r
1632 (Name[Index] != '.') &&\r
1633 (Name[Index] != ':')\r
1634 ) {\r
1635 //\r
1636 // ASCII dash, dot, colon lower-case characters and digit characters\r
1637 // are allowed.\r
1638 //\r
1639 return EFI_PROTOCOL_ERROR;\r
1640 }\r
1641 }\r
1642\r
1643 if ((Len < 4) || (CompareMem (Name, "iqn.", 4) != 0)) {\r
1644 //\r
1645 // Only IQN format is accepted now.\r
1646 //\r
1647 return EFI_PROTOCOL_ERROR;\r
1648 }\r
1649\r
1650 return EFI_SUCCESS;\r
1651}\r
1652\r
1653/**\r
1654 Create an iSCSI task control block.\r
1655\r
1656 @param[in] Conn The connection on which the task control block will be created.\r
1657 @param[out] Tcb The newly created task control block.\r
1658\r
1659 @retval EFI_SUCCESS The task control block is created.\r
1660 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
1661 @retval EFI_NOT_READY The target can not accept new commands.\r
1662**/\r
1663EFI_STATUS\r
1664IScsiNewTcb (\r
1665 IN ISCSI_CONNECTION *Conn,\r
1666 OUT ISCSI_TCB **Tcb\r
1667 )\r
1668{\r
1669 ISCSI_SESSION *Session;\r
1670 ISCSI_TCB *NewTcb;\r
1671\r
1672 ASSERT (Tcb != NULL);\r
1673\r
1674 Session = Conn->Session;\r
1675\r
1676 if (ISCSI_SEQ_GT (Session->CmdSN, Session->MaxCmdSN)) {\r
1677 return EFI_NOT_READY;\r
1678 }\r
1679\r
1680 NewTcb = AllocateZeroPool (sizeof (ISCSI_TCB));\r
1681 if (NewTcb == NULL) {\r
1682 return EFI_OUT_OF_RESOURCES;\r
1683 }\r
1684\r
1685 InitializeListHead (&NewTcb->Link);\r
1686\r
1687 NewTcb->SoFarInOrder = TRUE;\r
1688 NewTcb->InitiatorTaskTag = Session->InitiatorTaskTag;\r
1689 NewTcb->CmdSN = Session->CmdSN;\r
1690 NewTcb->Conn = Conn;\r
1691\r
1692 InsertTailList (&Session->TcbList, &NewTcb->Link);\r
1693\r
1694 //\r
1695 // Advance the initiator task tag.\r
1696 //\r
1697 Session->InitiatorTaskTag++;\r
1698 Session->CmdSN++;\r
1699\r
1700 *Tcb = NewTcb;\r
1701\r
1702 return EFI_SUCCESS;\r
1703}\r
1704\r
1705/**\r
1706 Delete the tcb from the connection and destroy it.\r
1707\r
1708 @param[in] Tcb The tcb to delete.\r
1709**/\r
1710VOID\r
1711IScsiDelTcb (\r
1712 IN ISCSI_TCB *Tcb\r
1713 )\r
1714{\r
1715 RemoveEntryList (&Tcb->Link);\r
1716\r
1717 FreePool (Tcb);\r
1718}\r
1719\r
1720/**\r
1721 Find the task control block by the initator task tag.\r
1722\r
1723 @param[in] TcbList The tcb list.\r
1724 @param[in] InitiatorTaskTag The initiator task tag.\r
1725\r
1726 @return The task control block found.\r
1727**/\r
1728ISCSI_TCB *\r
1729IScsiFindTcbByITT (\r
1730 IN LIST_ENTRY *TcbList,\r
1731 IN UINT32 InitiatorTaskTag\r
1732 )\r
1733{\r
1734 ISCSI_TCB *Tcb;\r
1735 LIST_ENTRY *Entry;\r
1736\r
1737 Tcb = NULL;\r
1738\r
1739 NET_LIST_FOR_EACH (Entry, TcbList) {\r
1740 Tcb = NET_LIST_USER_STRUCT (Entry, ISCSI_TCB, Link);\r
1741\r
1742 if (Tcb->InitiatorTaskTag == InitiatorTaskTag) {\r
1743 break;\r
1744 }\r
1745\r
1746 Tcb = NULL;\r
1747 }\r
1748\r
1749 return Tcb;\r
1750}\r
1751\r
1752/**\r
1753 Create a data segment, pad it and calculate the CRC if needed.\r
1754\r
1755 @param[in] Data The data to fill into the data segment.\r
1756 @param[in] Len Length of the data.\r
1757 @param[in] DataDigest Whether to calculate CRC for this data segment.\r
1758\r
1759 @return The net buffer wrapping the data segment.\r
1760**/\r
1761NET_BUF *\r
1762IScsiNewDataSegment (\r
1763 IN UINT8 *Data,\r
1764 IN UINT32 Len,\r
1765 IN BOOLEAN DataDigest\r
1766 )\r
1767{\r
1768 NET_FRAGMENT Fragment[2];\r
1769 UINT32 FragmentCount;\r
1770 UINT32 PadLen;\r
1771 NET_BUF *DataSeg;\r
1772\r
1773 Fragment[0].Len = Len;\r
1774 Fragment[0].Bulk = Data;\r
1775\r
1776 PadLen = ISCSI_GET_PAD_LEN (Len);\r
1777 if (PadLen != 0) {\r
1778 Fragment[1].Len = PadLen;\r
1779 Fragment[1].Bulk = (UINT8 *) &mDataSegPad;\r
1780\r
1781 FragmentCount = 2;\r
1782 } else {\r
1783 FragmentCount = 1;\r
1784 }\r
1785\r
1786 DataSeg = NetbufFromExt (&Fragment[0], FragmentCount, 0, 0, IScsiNbufExtFree, NULL);\r
1787\r
1788 return DataSeg;\r
1789}\r
1790\r
1791/**\r
1792 Create a iSCSI SCSI command PDU to encapsulate the command issued\r
1793 by SCSI through the EXT SCSI PASS THRU Protocol.\r
1794\r
1795 @param[in] Packet The EXT SCSI PASS THRU request packet containing the SCSI command.\r
1796 @param[in] Lun The LUN.\r
1797 @param[in] Tcb The tcb assocated with this SCSI command.\r
1798\r
1799 @return The created iSCSI SCSI command PDU.\r
1800 @retval NULL Other errors as indicated.\r
1801**/\r
1802NET_BUF *\r
1803IScsiNewScsiCmdPdu (\r
1804 IN EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet,\r
1805 IN UINT64 Lun,\r
1806 IN ISCSI_TCB *Tcb\r
1807 )\r
1808{\r
1809 LIST_ENTRY *NbufList;\r
1810 NET_BUF *Pdu;\r
1811 NET_BUF *PduHeader;\r
1812 NET_BUF *DataSeg;\r
1813 SCSI_COMMAND *ScsiCmd;\r
1814 UINT8 AHSLength;\r
1815 UINT32 Length;\r
1816 ISCSI_ADDITIONAL_HEADER *Header;\r
1817 ISCSI_BI_EXP_READ_DATA_LEN_AHS *BiExpReadDataLenAHS;\r
1818 ISCSI_SESSION *Session;\r
1819 UINT32 ImmediateDataLen;\r
1820\r
1821 AHSLength = 0;\r
1822\r
1823 if (Packet->DataDirection == DataBi) {\r
1824 //\r
1825 // Bi directional Read/Write command, the bidirectional expected\r
1826 // read data length AHS is required.\r
1827 //\r
1828 AHSLength += sizeof (ISCSI_BI_EXP_READ_DATA_LEN_AHS);\r
1829 }\r
1830\r
1831 if (Packet->CdbLength > 16) {\r
1832 //\r
1833 // The CDB exceeds 16 bytes, an extended CDB AHS is required.\r
1834 //\r
1835 AHSLength = (UINT8) (AHSLength + (ISCSI_ROUNDUP (Packet->CdbLength - 16) + sizeof (ISCSI_ADDITIONAL_HEADER)));\r
1836 }\r
1837\r
1838 Length = sizeof (SCSI_COMMAND) + AHSLength;\r
1839 PduHeader = NetbufAlloc (Length);\r
1840 if (PduHeader == NULL) {\r
1841 return NULL;\r
1842 }\r
1843\r
1844 ScsiCmd = (SCSI_COMMAND *) NetbufAllocSpace (PduHeader, Length, NET_BUF_TAIL);\r
1845 if (ScsiCmd == NULL) {\r
1846 NetbufFree (PduHeader);\r
1847 return NULL;\r
1848 } \r
1849 Header = (ISCSI_ADDITIONAL_HEADER *) (ScsiCmd + 1);\r
1850\r
1851 ZeroMem (ScsiCmd, Length);\r
1852\r
1853 ISCSI_SET_OPCODE (ScsiCmd, ISCSI_OPCODE_SCSI_CMD, 0);\r
1854 ISCSI_SET_FLAG (ScsiCmd, ISCSI_TASK_ATTR_SIMPLE);\r
1855\r
1856 //\r
1857 // Set the READ/WRITE flags according to the IO type of this request.\r
1858 //\r
1859 switch (Packet->DataDirection) {\r
1860 case DataIn:\r
1861 ISCSI_SET_FLAG (ScsiCmd, SCSI_CMD_PDU_FLAG_READ);\r
1862 ScsiCmd->ExpDataXferLength = NTOHL (Packet->InTransferLength);\r
1863 break;\r
1864\r
1865 case DataOut:\r
1866 ISCSI_SET_FLAG (ScsiCmd, SCSI_CMD_PDU_FLAG_WRITE);\r
1867 ScsiCmd->ExpDataXferLength = NTOHL (Packet->OutTransferLength);\r
1868 break;\r
1869\r
1870 case DataBi:\r
1871 ISCSI_SET_FLAG (ScsiCmd, SCSI_CMD_PDU_FLAG_READ | SCSI_CMD_PDU_FLAG_WRITE);\r
1872 ScsiCmd->ExpDataXferLength = NTOHL (Packet->OutTransferLength);\r
1873\r
1874 //\r
1875 // Fill the bidirectional expected read data length AHS.\r
1876 //\r
1877 BiExpReadDataLenAHS = (ISCSI_BI_EXP_READ_DATA_LEN_AHS *) Header;\r
1878 Header = (ISCSI_ADDITIONAL_HEADER *) (BiExpReadDataLenAHS + 1);\r
1879\r
1880 BiExpReadDataLenAHS->Length = NTOHS (5);\r
1881 BiExpReadDataLenAHS->Type = ISCSI_AHS_TYPE_BI_EXP_READ_DATA_LEN;\r
1882 BiExpReadDataLenAHS->ExpReadDataLength = NTOHL (Packet->InTransferLength);\r
1883\r
1884 break;\r
1885 }\r
1886\r
1887 ScsiCmd->TotalAHSLength = AHSLength;\r
1888 CopyMem (ScsiCmd->Lun, &Lun, sizeof (ScsiCmd->Lun));\r
1889 ScsiCmd->InitiatorTaskTag = NTOHL (Tcb->InitiatorTaskTag);\r
1890 ScsiCmd->CmdSN = NTOHL (Tcb->CmdSN);\r
1891 ScsiCmd->ExpStatSN = NTOHL (Tcb->Conn->ExpStatSN);\r
1892\r
1893 CopyMem (ScsiCmd->Cdb, Packet->Cdb, sizeof (ScsiCmd->Cdb));\r
1894\r
1895 if (Packet->CdbLength > 16) {\r
1896 Header->Length = NTOHS ((UINT16) (Packet->CdbLength - 15));\r
1897 Header->Type = ISCSI_AHS_TYPE_EXT_CDB;\r
1898\r
1899 CopyMem (Header + 1, (UINT8 *) Packet->Cdb + 16, Packet->CdbLength - 16);\r
1900 }\r
1901\r
1902 Pdu = PduHeader;\r
1903 Session = Tcb->Conn->Session;\r
1904 ImmediateDataLen = 0;\r
1905\r
1906 if (Session->ImmediateData && (Packet->OutTransferLength != 0)) {\r
1907 //\r
1908 // Send immediate data in this SCSI Command PDU. The length of the immeidate\r
1909 // data is the minimum of FirstBurstLength, the data length to be xfered and\r
1910 // the MaxRecvdataSegmentLength on this connection.\r
1911 //\r
1912 ImmediateDataLen = MIN (Session->FirstBurstLength, Packet->OutTransferLength);\r
1913 ImmediateDataLen = MIN (ImmediateDataLen, Tcb->Conn->MaxRecvDataSegmentLength);\r
1914\r
1915 //\r
1916 // Update the data segment length in the PDU header.\r
1917 //\r
1918 ISCSI_SET_DATASEG_LEN (ScsiCmd, ImmediateDataLen);\r
1919\r
1920 //\r
1921 // Create the data segment.\r
1922 //\r
1923 DataSeg = IScsiNewDataSegment ((UINT8 *) Packet->OutDataBuffer, ImmediateDataLen, FALSE);\r
1924 if (DataSeg == NULL) {\r
1925 NetbufFree (PduHeader);\r
1926 Pdu = NULL;\r
1927 goto ON_EXIT;\r
1928 }\r
1929\r
1930 NbufList = AllocatePool (sizeof (LIST_ENTRY));\r
1931 if (NbufList == NULL) {\r
1932 NetbufFree (PduHeader);\r
1933 NetbufFree (DataSeg);\r
1934\r
1935 Pdu = NULL;\r
1936 goto ON_EXIT;\r
1937 }\r
1938\r
1939 InitializeListHead (NbufList);\r
1940 InsertTailList (NbufList, &PduHeader->List);\r
1941 InsertTailList (NbufList, &DataSeg->List);\r
1942\r
1943 Pdu = NetbufFromBufList (NbufList, 0, 0, IScsiFreeNbufList, NbufList);\r
1944 if (Pdu == NULL) {\r
1945 IScsiFreeNbufList (NbufList);\r
1946 }\r
1947 }\r
1948\r
1949 if (Session->InitialR2T ||\r
1950 (ImmediateDataLen == Session->FirstBurstLength) ||\r
1951 (ImmediateDataLen == Packet->OutTransferLength)\r
1952 ) {\r
1953 //\r
1954 // Unsolicited data out sequence is not allowed,\r
1955 // or FirstBurstLength data is already sent out by immediate data\r
1956 // or all the OUT data accompany this SCSI packet is sent as\r
1957 // immediate data, the final flag should be set on this SCSI Command\r
1958 // PDU.\r
1959 //\r
1960 ISCSI_SET_FLAG (ScsiCmd, ISCSI_BHS_FLAG_FINAL);\r
1961 }\r
1962\r
1963ON_EXIT:\r
1964\r
1965 return Pdu;\r
1966}\r
1967\r
1968/**\r
1969 Create a new iSCSI SCSI Data Out PDU.\r
1970\r
1971 @param[in] Data The data to put into the Data Out PDU.\r
1972 @param[in] Len Length of the data.\r
1973 @param[in] DataSN The DataSN of the Data Out PDU.\r
1974 @param[in] Tcb The task control block of this Data Out PDU.\r
1975 @param[in] Lun The LUN.\r
1976\r
1977 @return The net buffer wrapping the Data Out PDU.\r
1978 @retval NULL Other errors as indicated.\r
1979**/\r
1980NET_BUF *\r
1981IScsiNewDataOutPdu (\r
1982 IN UINT8 *Data,\r
1983 IN UINT32 Len,\r
1984 IN UINT32 DataSN,\r
1985 IN ISCSI_TCB *Tcb,\r
1986 IN UINT64 Lun\r
1987 )\r
1988{\r
1989 LIST_ENTRY *NbufList;\r
1990 NET_BUF *PduHdr;\r
1991 NET_BUF *DataSeg;\r
1992 NET_BUF *Pdu;\r
1993 ISCSI_SCSI_DATA_OUT *DataOutHdr;\r
1994 ISCSI_XFER_CONTEXT *XferContext;\r
1995\r
1996 NbufList = AllocatePool (sizeof (LIST_ENTRY));\r
1997 if (NbufList == NULL) {\r
1998 return NULL;\r
1999 }\r
2000\r
2001 InitializeListHead (NbufList);\r
2002\r
2003 //\r
2004 // Allocate memory for the BHS.\r
2005 //\r
2006 PduHdr = NetbufAlloc (sizeof (ISCSI_SCSI_DATA_OUT));\r
2007 if (PduHdr == NULL) {\r
2008 FreePool (NbufList);\r
2009 return NULL;\r
2010 }\r
2011 //\r
2012 // Insert the BHS into the buffer list.\r
2013 //\r
2014 InsertTailList (NbufList, &PduHdr->List);\r
2015\r
2016 DataOutHdr = (ISCSI_SCSI_DATA_OUT *) NetbufAllocSpace (PduHdr, sizeof (ISCSI_SCSI_DATA_OUT), NET_BUF_TAIL);\r
2017 ASSERT (DataOutHdr != NULL);\r
2018 XferContext = &Tcb->XferContext;\r
2019\r
2020 ZeroMem (DataOutHdr, sizeof (ISCSI_SCSI_DATA_OUT));\r
2021\r
2022 //\r
2023 // Set the flags and fields of the Data Out PDU BHS.\r
2024 //\r
2025 ISCSI_SET_OPCODE (DataOutHdr, ISCSI_OPCODE_SCSI_DATA_OUT, 0);\r
2026 ISCSI_SET_DATASEG_LEN (DataOutHdr, Len);\r
2027\r
2028 DataOutHdr->InitiatorTaskTag = HTONL (Tcb->InitiatorTaskTag);\r
2029 DataOutHdr->TargetTransferTag = HTONL (XferContext->TargetTransferTag);\r
2030 DataOutHdr->ExpStatSN = HTONL (Tcb->Conn->ExpStatSN);\r
2031 DataOutHdr->DataSN = HTONL (DataSN);\r
2032 DataOutHdr->BufferOffset = HTONL (XferContext->Offset);\r
2033\r
2034 if (XferContext->TargetTransferTag != ISCSI_RESERVED_TAG) {\r
2035 CopyMem (&DataOutHdr->Lun, &Lun, sizeof (DataOutHdr->Lun));\r
2036 }\r
2037 //\r
2038 // Build the data segment for this Data Out PDU.\r
2039 //\r
2040 DataSeg = IScsiNewDataSegment (Data, Len, FALSE);\r
2041 if (DataSeg == NULL) {\r
2042 IScsiFreeNbufList (NbufList);\r
2043 return NULL;\r
2044 }\r
2045 //\r
2046 // Put the data segment into the buffer list and combine it with the BHS\r
2047 // into a full Data Out PDU.\r
2048 //\r
2049 InsertTailList (NbufList, &DataSeg->List);\r
2050 Pdu = NetbufFromBufList (NbufList, 0, 0, IScsiFreeNbufList, NbufList);\r
2051 if (Pdu == NULL) {\r
2052 IScsiFreeNbufList (NbufList);\r
2053 }\r
2054\r
2055 return Pdu;\r
2056}\r
2057\r
2058/**\r
2059 Generate a consecutive sequence of iSCSI SCSI Data Out PDUs.\r
2060\r
2061 @param[in] Data The data which will be carried by the sequence of iSCSI SCSI Data Out PDUs.\r
2062 @param[in] Tcb The task control block of the data to send out.\r
2063 @param[in] Lun The LUN the data will be sent to.\r
2064\r
2065 @return A list of net buffers with each of them wraps an iSCSI SCSI Data Out PDU.\r
2066 @retval NULL Other errors as indicated.\r
2067**/\r
2068LIST_ENTRY *\r
2069IScsiGenerateDataOutPduSequence (\r
2070 IN UINT8 *Data,\r
2071 IN ISCSI_TCB *Tcb,\r
2072 IN UINT64 Lun\r
2073 )\r
2074{\r
2075 LIST_ENTRY *PduList;\r
2076 UINT32 DataSN;\r
2077 UINT32 DataLen;\r
2078 NET_BUF *DataOutPdu;\r
2079 ISCSI_CONNECTION *Conn;\r
2080 ISCSI_XFER_CONTEXT *XferContext;\r
2081 UINT8 *DataOutPacket;\r
2082\r
2083 PduList = AllocatePool (sizeof (LIST_ENTRY));\r
2084 if (PduList == NULL) {\r
2085 return NULL;\r
2086 }\r
2087\r
2088 InitializeListHead (PduList);\r
2089\r
2090 DataSN = 0;\r
2091 Conn = Tcb->Conn;\r
2092 DataOutPdu = NULL;\r
2093 XferContext = &Tcb->XferContext;\r
2094\r
2095 while (XferContext->DesiredLength > 0) {\r
2096 //\r
2097 // Determine the length of data this Data Out PDU can carry.\r
2098 //\r
2099 DataLen = MIN (XferContext->DesiredLength, Conn->MaxRecvDataSegmentLength);\r
2100\r
2101 //\r
2102 // Create a Data Out PDU.\r
2103 //\r
2104 DataOutPdu = IScsiNewDataOutPdu (Data, DataLen, DataSN, Tcb, Lun);\r
2105 if (DataOutPdu == NULL) {\r
2106 IScsiFreeNbufList (PduList);\r
2107 PduList = NULL;\r
2108\r
2109 goto ON_EXIT;\r
2110 }\r
2111\r
2112 InsertTailList (PduList, &DataOutPdu->List);\r
2113\r
2114 //\r
2115 // Update the context and DataSN.\r
2116 //\r
2117 XferContext->Offset += DataLen;\r
2118 XferContext->DesiredLength -= DataLen;\r
2119 DataSN++;\r
2120 Data += DataLen;\r
2121 }\r
2122 //\r
2123 // Set the F bit for the last data out PDU in this sequence.\r
2124 //\r
2125 DataOutPacket = NetbufGetByte (DataOutPdu, 0, NULL);\r
2126 if (DataOutPacket == NULL) {\r
2127 IScsiFreeNbufList (PduList);\r
2128 PduList = NULL;\r
2129 goto ON_EXIT;\r
2130 }\r
2131\r
2132 ISCSI_SET_FLAG (DataOutPacket, ISCSI_BHS_FLAG_FINAL);\r
2133\r
2134ON_EXIT:\r
2135\r
2136 return PduList;\r
2137}\r
2138\r
2139/**\r
2140 Send the Data in a sequence of Data Out PDUs one by one.\r
2141\r
2142 @param[in] Data The data to carry by Data Out PDUs.\r
2143 @param[in] Lun The LUN the data will be sent to.\r
2144 @param[in] Tcb The task control block.\r
2145\r
2146 @retval EFI_SUCCES The data is sent out to the LUN.\r
2147 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
2148 @retval Others Other errors as indicated.\r
2149**/\r
2150EFI_STATUS\r
2151IScsiSendDataOutPduSequence (\r
2152 IN UINT8 *Data,\r
2153 IN UINT64 Lun,\r
2154 IN ISCSI_TCB *Tcb\r
2155 )\r
2156{\r
2157 LIST_ENTRY *DataOutPduList;\r
2158 LIST_ENTRY *Entry;\r
2159 NET_BUF *Pdu;\r
2160 EFI_STATUS Status;\r
2161\r
2162 //\r
2163 // Generate the Data Out PDU sequence.\r
2164 //\r
2165 DataOutPduList = IScsiGenerateDataOutPduSequence (Data, Tcb, Lun);\r
2166 if (DataOutPduList == NULL) {\r
2167 return EFI_OUT_OF_RESOURCES;\r
2168 }\r
2169\r
2170 Status = EFI_SUCCESS;\r
2171\r
2172 //\r
2173 // Send the Data Out PDU's one by one.\r
2174 //\r
2175 NET_LIST_FOR_EACH (Entry, DataOutPduList) {\r
2176 Pdu = NET_LIST_USER_STRUCT (Entry, NET_BUF, List);\r
2177\r
2178 Status = Tcp4IoTransmit (&Tcb->Conn->Tcp4Io, Pdu);\r
2179 if (EFI_ERROR (Status)) {\r
2180 break;\r
2181 }\r
2182 }\r
2183\r
2184 IScsiFreeNbufList (DataOutPduList);\r
2185\r
2186 return Status;\r
2187}\r
2188\r
2189/**\r
2190 Process the received iSCSI SCSI Data In PDU.\r
2191\r
2192 @param[in] Pdu The Data In PDU received.\r
2193 @param[in] Tcb The task control block.\r
2194 @param[in, out] Packet The EXT SCSI PASS THRU request packet.\r
2195\r
2196 @retval EFI_SUCCES The check on the Data IN PDU is passed and some update\r
2197 actions are taken.\r
2198 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol errror happened.\r
2199 @retval EFI_BAD_BUFFER_SIZEE The buffer was not the proper size for the request.\r
2200 @retval Others Other errors as indicated.\r
2201**/\r
2202EFI_STATUS\r
2203IScsiOnDataInRcvd (\r
2204 IN NET_BUF *Pdu,\r
2205 IN ISCSI_TCB *Tcb,\r
2206 IN OUT EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet\r
2207 )\r
2208{\r
2209 ISCSI_SCSI_DATA_IN *DataInHdr;\r
2210 EFI_STATUS Status;\r
2211\r
2212 DataInHdr = (ISCSI_SCSI_DATA_IN *) NetbufGetByte (Pdu, 0, NULL);\r
2213 if (DataInHdr == NULL) {\r
2214 return EFI_PROTOCOL_ERROR;\r
2215 }\r
2216\r
2217 DataInHdr->InitiatorTaskTag = NTOHL (DataInHdr->InitiatorTaskTag);\r
2218 DataInHdr->ExpCmdSN = NTOHL (DataInHdr->ExpCmdSN);\r
2219 DataInHdr->MaxCmdSN = NTOHL (DataInHdr->MaxCmdSN);\r
2220 DataInHdr->DataSN = NTOHL (DataInHdr->DataSN);\r
2221\r
2222 //\r
2223 // Check the DataSN.\r
2224 //\r
2225 Status = IScsiCheckSN (&Tcb->ExpDataSN, DataInHdr->DataSN);\r
2226 if (EFI_ERROR (Status)) {\r
2227 return Status;\r
2228 }\r
2229\r
2230 if (DataInHdr->InitiatorTaskTag != Tcb->InitiatorTaskTag) {\r
2231 return EFI_PROTOCOL_ERROR;\r
2232 }\r
2233 //\r
2234 // Update the command related sequence numbers.\r
2235 //\r
2236 IScsiUpdateCmdSN (Tcb->Conn->Session, DataInHdr->MaxCmdSN, DataInHdr->ExpCmdSN);\r
2237\r
2238 if (ISCSI_FLAG_ON (DataInHdr, SCSI_DATA_IN_PDU_FLAG_STATUS_VALID)) {\r
2239 if (!ISCSI_FLAG_ON (DataInHdr, ISCSI_BHS_FLAG_FINAL)) {\r
2240 //\r
2241 // The S bit is on but the F bit is off.\r
2242 //\r
2243 return EFI_PROTOCOL_ERROR;\r
2244 }\r
2245\r
2246 Tcb->StatusXferd = TRUE;\r
2247\r
2248 if (ISCSI_FLAG_ON (DataInHdr, SCSI_DATA_IN_PDU_FLAG_OVERFLOW | SCSI_DATA_IN_PDU_FLAG_UNDERFLOW)) {\r
2249 //\r
2250 // Underflow and Overflow are mutual flags.\r
2251 //\r
2252 return EFI_PROTOCOL_ERROR;\r
2253 }\r
2254 //\r
2255 // S bit is on, the StatSN is valid.\r
2256 //\r
2257 Status = IScsiCheckSN (&Tcb->Conn->ExpStatSN, NTOHL (DataInHdr->StatSN));\r
2258 if (EFI_ERROR (Status)) {\r
2259 return Status;\r
2260 }\r
2261\r
2262 Packet->HostAdapterStatus = 0;\r
2263 Packet->TargetStatus = DataInHdr->Status;\r
2264\r
2265 if (ISCSI_FLAG_ON (DataInHdr, SCSI_RSP_PDU_FLAG_OVERFLOW)) {\r
2266 Packet->InTransferLength += NTOHL (DataInHdr->ResidualCount);\r
2267 Status = EFI_BAD_BUFFER_SIZE;\r
2268 }\r
2269\r
2270 if (ISCSI_FLAG_ON (DataInHdr, SCSI_RSP_PDU_FLAG_UNDERFLOW)) {\r
2271 Packet->InTransferLength -= NTOHL (DataInHdr->ResidualCount);\r
2272 }\r
2273 }\r
2274\r
2275 return Status;\r
2276}\r
2277\r
2278/**\r
2279 Process the received iSCSI R2T PDU.\r
2280\r
2281 @param[in] Pdu The R2T PDU received.\r
2282 @param[in] Tcb The task control block.\r
2283 @param[in] Lun The Lun.\r
2284 @param[in, out] Packet The EXT SCSI PASS THRU request packet.\r
2285\r
2286 @retval EFI_SUCCES The R2T PDU is valid and the solicited data is sent out.\r
2287 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol errror happened.\r
2288 @retval Others Other errors as indicated.\r
2289**/\r
2290EFI_STATUS\r
2291IScsiOnR2TRcvd (\r
2292 IN NET_BUF *Pdu,\r
2293 IN ISCSI_TCB *Tcb,\r
2294 IN UINT64 Lun,\r
2295 IN OUT EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet\r
2296 )\r
2297{\r
2298 ISCSI_READY_TO_TRANSFER *R2THdr;\r
2299 EFI_STATUS Status;\r
2300 ISCSI_XFER_CONTEXT *XferContext;\r
2301 UINT8 *Data;\r
2302\r
2303 R2THdr = (ISCSI_READY_TO_TRANSFER *) NetbufGetByte (Pdu, 0, NULL);\r
2304 if (R2THdr == NULL) {\r
2305 return EFI_PROTOCOL_ERROR;\r
2306 }\r
2307\r
2308 R2THdr->InitiatorTaskTag = NTOHL (R2THdr->InitiatorTaskTag);\r
2309 R2THdr->TargetTransferTag = NTOHL (R2THdr->TargetTransferTag);\r
2310 R2THdr->StatSN = NTOHL (R2THdr->StatSN);\r
2311 R2THdr->R2TSeqNum = NTOHL (R2THdr->R2TSeqNum);\r
2312 R2THdr->BufferOffset = NTOHL (R2THdr->BufferOffset);\r
2313 R2THdr->DesiredDataTransferLength = NTOHL (R2THdr->DesiredDataTransferLength);\r
2314\r
2315 if ((R2THdr->InitiatorTaskTag != Tcb->InitiatorTaskTag) || !ISCSI_SEQ_EQ (R2THdr->StatSN, Tcb->Conn->ExpStatSN)) {\r
2316 return EFI_PROTOCOL_ERROR;;\r
2317 }\r
2318 //\r
2319 // Check the sequence number.\r
2320 //\r
2321 Status = IScsiCheckSN (&Tcb->ExpDataSN, R2THdr->R2TSeqNum);\r
2322 if (EFI_ERROR (Status)) {\r
2323 return Status;\r
2324 }\r
2325\r
2326 XferContext = &Tcb->XferContext;\r
2327 XferContext->TargetTransferTag = R2THdr->TargetTransferTag;\r
2328 XferContext->Offset = R2THdr->BufferOffset;\r
2329 XferContext->DesiredLength = R2THdr->DesiredDataTransferLength;\r
2330\r
2331 if (((XferContext->Offset + XferContext->DesiredLength) > Packet->OutTransferLength) ||\r
2332 (XferContext->DesiredLength > Tcb->Conn->Session->MaxBurstLength)\r
2333 ) {\r
2334 return EFI_PROTOCOL_ERROR;\r
2335 }\r
2336 //\r
2337 // Send the data solicited by this R2T.\r
2338 //\r
2339 Data = (UINT8 *) Packet->OutDataBuffer + XferContext->Offset;\r
2340 Status = IScsiSendDataOutPduSequence (Data, Lun, Tcb);\r
2341\r
2342 return Status;\r
2343}\r
2344\r
2345/**\r
2346 Process the received iSCSI SCSI Response PDU.\r
2347\r
2348 @param[in] Pdu The Response PDU received.\r
2349 @param[in] Tcb The task control block.\r
2350 @param[in, out] Packet The EXT SCSI PASS THRU request packet.\r
2351\r
2352 @retval EFI_SUCCES The Response PDU is processed.\r
2353 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol errror happened.\r
2354 @retval EFI_BAD_BUFFER_SIZEE The buffer was not the proper size for the request.\r
2355 @retval Others Other errors as indicated.\r
2356**/\r
2357EFI_STATUS\r
2358IScsiOnScsiRspRcvd (\r
2359 IN NET_BUF *Pdu,\r
2360 IN ISCSI_TCB *Tcb,\r
2361 IN OUT EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet\r
2362 )\r
2363{\r
2364 SCSI_RESPONSE *ScsiRspHdr;\r
2365 ISCSI_SENSE_DATA *SenseData;\r
2366 EFI_STATUS Status;\r
2367 UINT32 DataSegLen;\r
2368\r
2369 ScsiRspHdr = (SCSI_RESPONSE *) NetbufGetByte (Pdu, 0, NULL);\r
2370 if (ScsiRspHdr == NULL) {\r
2371 return EFI_PROTOCOL_ERROR;\r
2372 }\r
2373\r
2374 ScsiRspHdr->InitiatorTaskTag = NTOHL (ScsiRspHdr->InitiatorTaskTag);\r
2375 if (ScsiRspHdr->InitiatorTaskTag != Tcb->InitiatorTaskTag) {\r
2376 return EFI_PROTOCOL_ERROR;\r
2377 }\r
2378\r
2379 ScsiRspHdr->StatSN = NTOHL (ScsiRspHdr->StatSN);\r
2380\r
2381 Status = IScsiCheckSN (&Tcb->Conn->ExpStatSN, ScsiRspHdr->StatSN);\r
2382 if (EFI_ERROR (Status)) {\r
2383 return Status;\r
2384 }\r
2385\r
2386 ScsiRspHdr->MaxCmdSN = NTOHL (ScsiRspHdr->MaxCmdSN);\r
2387 ScsiRspHdr->ExpCmdSN = NTOHL (ScsiRspHdr->ExpCmdSN);\r
2388 IScsiUpdateCmdSN (Tcb->Conn->Session, ScsiRspHdr->MaxCmdSN, ScsiRspHdr->ExpCmdSN);\r
2389\r
2390 Tcb->StatusXferd = TRUE;\r
2391\r
2392 Packet->HostAdapterStatus = ScsiRspHdr->Response;\r
2393 if (Packet->HostAdapterStatus != ISCSI_SERVICE_RSP_COMMAND_COMPLETE_AT_TARGET) {\r
2394 return EFI_SUCCESS;\r
2395 }\r
2396\r
2397 Packet->TargetStatus = ScsiRspHdr->Status;\r
2398\r
2399 if (ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_BI_READ_OVERFLOW | SCSI_RSP_PDU_FLAG_BI_READ_UNDERFLOW) ||\r
2400 ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_OVERFLOW | SCSI_RSP_PDU_FLAG_UNDERFLOW)\r
2401 ) {\r
2402 return EFI_PROTOCOL_ERROR;\r
2403 }\r
2404\r
2405 if (ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_BI_READ_OVERFLOW)) {\r
2406 Packet->InTransferLength += NTOHL (ScsiRspHdr->BiReadResidualCount);\r
2407 Status = EFI_BAD_BUFFER_SIZE;\r
2408 }\r
2409\r
2410 if (ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_BI_READ_UNDERFLOW)) {\r
2411 Packet->InTransferLength -= NTOHL (ScsiRspHdr->BiReadResidualCount);\r
2412 }\r
2413\r
2414 if (ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_OVERFLOW)) {\r
2415 if (Packet->DataDirection == DataIn) {\r
2416 Packet->InTransferLength += NTOHL (ScsiRspHdr->ResidualCount);\r
2417 } else {\r
2418 Packet->OutTransferLength += NTOHL (ScsiRspHdr->ResidualCount);\r
2419 }\r
2420\r
2421 Status = EFI_BAD_BUFFER_SIZE;\r
2422 }\r
2423\r
2424 if (ISCSI_FLAG_ON (ScsiRspHdr, SCSI_RSP_PDU_FLAG_UNDERFLOW)) {\r
2425 if (Packet->DataDirection == DataIn) {\r
2426 Packet->InTransferLength -= NTOHL (ScsiRspHdr->ResidualCount);\r
2427 } else {\r
2428 Packet->OutTransferLength -= NTOHL (ScsiRspHdr->ResidualCount);\r
2429 }\r
2430 }\r
2431\r
2432 DataSegLen = ISCSI_GET_DATASEG_LEN (ScsiRspHdr);\r
2433 if (DataSegLen != 0) {\r
2434 SenseData = (ISCSI_SENSE_DATA *) NetbufGetByte (Pdu, sizeof (SCSI_RESPONSE), NULL);\r
2435 if (SenseData == NULL) {\r
2436 return EFI_PROTOCOL_ERROR;\r
2437 }\r
2438\r
2439 SenseData->Length = NTOHS (SenseData->Length);\r
2440\r
2441 Packet->SenseDataLength = (UINT8) MIN (SenseData->Length, Packet->SenseDataLength);\r
2442 if (Packet->SenseDataLength != 0) {\r
2443 CopyMem (Packet->SenseData, &SenseData->Data[0], Packet->SenseDataLength);\r
2444 }\r
2445 } else {\r
2446 Packet->SenseDataLength = 0;\r
2447 }\r
2448\r
2449 return Status;\r
2450}\r
2451\r
2452/**\r
2453 Process the received NOP In PDU.\r
2454\r
2455 @param[in] Pdu The NOP In PDU received.\r
2456 @param[in] Tcb The task control block.\r
2457\r
2458 @retval EFI_SUCCES The NOP In PDU is processed and the related sequence\r
2459 numbers are updated.\r
2460 @retval EFI_PROTOCOL_ERROR Some kind of iSCSI protocol errror happened.\r
2461**/\r
2462EFI_STATUS\r
2463IScsiOnNopInRcvd (\r
2464 IN NET_BUF *Pdu,\r
2465 IN ISCSI_TCB *Tcb\r
2466 )\r
2467{\r
2468 ISCSI_NOP_IN *NopInHdr;\r
2469 EFI_STATUS Status;\r
2470\r
2471 NopInHdr = (ISCSI_NOP_IN *) NetbufGetByte (Pdu, 0, NULL);\r
2472 if (NopInHdr == NULL) {\r
2473 return EFI_PROTOCOL_ERROR;\r
2474 }\r
2475\r
2476 NopInHdr->StatSN = NTOHL (NopInHdr->StatSN);\r
2477 NopInHdr->ExpCmdSN = NTOHL (NopInHdr->ExpCmdSN);\r
2478 NopInHdr->MaxCmdSN = NTOHL (NopInHdr->MaxCmdSN);\r
2479\r
2480 if (NopInHdr->InitiatorTaskTag == ISCSI_RESERVED_TAG) {\r
2481 if (NopInHdr->StatSN != Tcb->Conn->ExpStatSN) {\r
2482 return EFI_PROTOCOL_ERROR;\r
2483 }\r
2484 } else {\r
2485 Status = IScsiCheckSN (&Tcb->Conn->ExpStatSN, NopInHdr->StatSN);\r
2486 if (EFI_ERROR (Status)) {\r
2487 return Status;\r
2488 }\r
2489 }\r
2490\r
2491 IScsiUpdateCmdSN (Tcb->Conn->Session, NopInHdr->MaxCmdSN, NopInHdr->ExpCmdSN);\r
2492\r
2493 return EFI_SUCCESS;\r
2494}\r
2495\r
2496/**\r
2497 Execute the SCSI command issued through the EXT SCSI PASS THRU protocol.\r
2498\r
2499 @param[in] PassThru The EXT SCSI PASS THRU protocol.\r
2500 @param[in] Target The target ID.\r
2501 @param[in] Lun The LUN.\r
2502 @param[in, out] Packet The request packet containing IO request, SCSI command\r
2503 buffer and buffers to read/write.\r
2504 \r
2505 @retval EFI_SUCCES The SCSI command is executed and the result is updated to \r
2506 the Packet.\r
2507 @retval EFI_DEVICE_ERROR Session state was not as required.\r
2508 @retval EFI_OUT_OF_RESOURCES Failed to allocate memory.\r
2509 @retval EFI_NOT_READY The target can not accept new commands.\r
2510 @retval Others Other errors as indicated.\r
2511**/\r
2512EFI_STATUS\r
2513IScsiExecuteScsiCommand (\r
2514 IN EFI_EXT_SCSI_PASS_THRU_PROTOCOL *PassThru,\r
2515 IN UINT8 *Target,\r
2516 IN UINT64 Lun,\r
2517 IN OUT EFI_EXT_SCSI_PASS_THRU_SCSI_REQUEST_PACKET *Packet\r
2518 )\r
2519{\r
2520 EFI_STATUS Status;\r
2521 ISCSI_DRIVER_DATA *Private;\r
2522 ISCSI_SESSION *Session;\r
2523 EFI_EVENT TimeoutEvent;\r
2524 ISCSI_CONNECTION *Conn;\r
2525 ISCSI_TCB *Tcb;\r
2526 NET_BUF *Pdu;\r
2527 ISCSI_XFER_CONTEXT *XferContext;\r
2528 UINT8 *Data;\r
2529 ISCSI_IN_BUFFER_CONTEXT InBufferContext;\r
2530 UINT64 Timeout;\r
2531 UINT8 *PduHdr;\r
2532\r
2533 Private = ISCSI_DRIVER_DATA_FROM_EXT_SCSI_PASS_THRU (PassThru);\r
2534 Session = &Private->Session;\r
2535 Status = EFI_SUCCESS;\r
2536 Tcb = NULL;\r
2537 TimeoutEvent = NULL;\r
2538 Timeout = 0;\r
2539\r
2540 if (Session->State != SESSION_STATE_LOGGED_IN) {\r
2541 Status = EFI_DEVICE_ERROR;\r
2542 goto ON_EXIT;\r
2543 }\r
2544\r
2545 Conn = NET_LIST_USER_STRUCT_S (\r
2546 Session->Conns.ForwardLink,\r
2547 ISCSI_CONNECTION,\r
2548 Link,\r
2549 ISCSI_CONNECTION_SIGNATURE\r
2550 );\r
2551\r
2552 if (Packet->Timeout != 0) {\r
2553 Timeout = MultU64x32 (Packet->Timeout, 2);\r
2554 }\r
2555\r
2556 Status = IScsiNewTcb (Conn, &Tcb);\r
2557 if (EFI_ERROR (Status)) {\r
2558 goto ON_EXIT;\r
2559 }\r
2560 //\r
2561 // Encapsulate the SCSI request packet into an iSCSI SCSI Command PDU.\r
2562 //\r
2563 Pdu = IScsiNewScsiCmdPdu (Packet, Lun, Tcb);\r
2564 if (Pdu == NULL) {\r
2565 Status = EFI_OUT_OF_RESOURCES;\r
2566 goto ON_EXIT;\r
2567 }\r
2568\r
2569 XferContext = &Tcb->XferContext;\r
2570 PduHdr = NetbufGetByte (Pdu, 0, NULL);\r
2571 if (PduHdr == NULL) {\r
2572 Status = EFI_PROTOCOL_ERROR;\r
2573 NetbufFree (Pdu);\r
2574 goto ON_EXIT;\r
2575 }\r
2576 XferContext->Offset = ISCSI_GET_DATASEG_LEN (PduHdr);\r
2577\r
2578 //\r
2579 // Transmit the SCSI Command PDU.\r
2580 //\r
2581 Status = Tcp4IoTransmit (&Conn->Tcp4Io, Pdu);\r
2582\r
2583 NetbufFree (Pdu);\r
2584\r
2585 if (EFI_ERROR (Status)) {\r
2586 goto ON_EXIT;\r
2587 }\r
2588\r
2589 if (!Session->InitialR2T &&\r
2590 (XferContext->Offset < Session->FirstBurstLength) &&\r
2591 (XferContext->Offset < Packet->OutTransferLength)\r
2592 ) {\r
2593 //\r
2594 // Unsolicited Data-Out sequence is allowed, there is remaining SCSI\r
2595 // OUT data and the limit of FirstBurstLength is not reached.\r
2596 //\r
2597 XferContext->TargetTransferTag = ISCSI_RESERVED_TAG;\r
2598 XferContext->DesiredLength = MIN (\r
2599 Session->FirstBurstLength,\r
2600 Packet->OutTransferLength - XferContext->Offset\r
2601 );\r
2602\r
2603 Data = (UINT8 *) Packet->OutDataBuffer + XferContext->Offset;\r
2604 Status = IScsiSendDataOutPduSequence (Data, Lun, Tcb);\r
2605 if (EFI_ERROR (Status)) {\r
2606 goto ON_EXIT;\r
2607 }\r
2608 }\r
2609\r
2610 InBufferContext.InData = (UINT8 *) Packet->InDataBuffer;\r
2611 InBufferContext.InDataLen = Packet->InTransferLength;\r
2612\r
2613 while (!Tcb->StatusXferd) {\r
2614 //\r
2615 // Start the timeout timer.\r
2616 //\r
2617 if (Timeout != 0) {\r
2618 Status = gBS->SetTimer (Conn->TimeoutEvent, TimerRelative, Timeout);\r
2619 if (EFI_ERROR (Status)) {\r
2620 goto ON_EXIT;\r
2621 }\r
2622 TimeoutEvent = Conn->TimeoutEvent; \r
2623 }\r
2624 //\r
2625 // try to receive PDU from target.\r
2626 //\r
2627 Status = IScsiReceivePdu (Conn, &Pdu, &InBufferContext, FALSE, FALSE, TimeoutEvent);\r
2628 if (EFI_ERROR (Status)) {\r
2629 goto ON_EXIT;\r
2630 }\r
2631\r
2632 PduHdr = NetbufGetByte (Pdu, 0, NULL);\r
2633 if (PduHdr == NULL) {\r
2634 Status = EFI_PROTOCOL_ERROR;\r
2635 NetbufFree (Pdu);\r
2636 goto ON_EXIT;\r
2637 }\r
2638 switch (ISCSI_GET_OPCODE (PduHdr)) {\r
2639 case ISCSI_OPCODE_SCSI_DATA_IN:\r
2640 Status = IScsiOnDataInRcvd (Pdu, Tcb, Packet);\r
2641 break;\r
2642\r
2643 case ISCSI_OPCODE_R2T:\r
2644 Status = IScsiOnR2TRcvd (Pdu, Tcb, Lun, Packet);\r
2645 break;\r
2646\r
2647 case ISCSI_OPCODE_SCSI_RSP:\r
2648 Status = IScsiOnScsiRspRcvd (Pdu, Tcb, Packet);\r
2649 break;\r
2650\r
2651 case ISCSI_OPCODE_NOP_IN:\r
2652 Status = IScsiOnNopInRcvd (Pdu, Tcb);\r
2653 break;\r
2654\r
2655 case ISCSI_OPCODE_VENDOR_T0:\r
2656 case ISCSI_OPCODE_VENDOR_T1:\r
2657 case ISCSI_OPCODE_VENDOR_T2:\r
2658 //\r
2659 // These messages are vendor specific, skip them.\r
2660 //\r
2661 break;\r
2662\r
2663 default:\r
2664 Status = EFI_PROTOCOL_ERROR;\r
2665 break;\r
2666 }\r
2667\r
2668 NetbufFree (Pdu);\r
2669\r
2670 if (EFI_ERROR (Status)) {\r
2671 break;\r
2672 }\r
2673 }\r
2674\r
2675ON_EXIT:\r
2676\r
2677 if (TimeoutEvent != NULL) {\r
2678 gBS->SetTimer (TimeoutEvent, TimerCancel, 0);\r
2679 }\r
2680\r
2681 if (Tcb != NULL) {\r
2682 IScsiDelTcb (Tcb);\r
2683 }\r
2684\r
2685 return Status;\r
2686}\r
2687\r
2688/**\r
2689 Reinstate the session on some error.\r
2690\r
2691 @param[in, out] Private The iSCSI driver data.\r
2692\r
2693 @retval EFI_SUCCES The session is reinstated from some error.\r
2694 @retval Other Reinstatement failed.\r
2695**/\r
2696EFI_STATUS\r
2697IScsiSessionReinstatement (\r
2698 IN OUT ISCSI_DRIVER_DATA *Private\r
2699 )\r
2700{\r
2701 ISCSI_SESSION *Session;\r
2702 EFI_STATUS Status;\r
2703\r
2704 Session = &Private->Session;\r
2705 ASSERT (Session->State != SESSION_STATE_FREE);\r
2706\r
2707 //\r
2708 // Abort the session and re-init it.\r
2709 //\r
2710 IScsiSessionAbort (Session);\r
2711 IScsiSessionInit (Session, TRUE);\r
2712\r
2713 //\r
2714 // Login again.\r
2715 //\r
2716 Status = IScsiSessionLogin (Private);\r
2717\r
2718 return Status;\r
2719}\r
2720\r
2721/**\r
2722 Initialize some session parameters before login.\r
2723\r
2724 @param[in, out] Session The iSCSI session.\r
2725 @param[in] Recovery Whether the request is from a fresh new start or recovery.\r
2726**/\r
2727VOID\r
2728IScsiSessionInit (\r
2729 IN OUT ISCSI_SESSION *Session,\r
2730 IN BOOLEAN Recovery\r
2731 )\r
2732{\r
2733 if (!Recovery) {\r
2734 Session->Signature = ISCSI_SESSION_SIGNATURE;\r
2735 Session->State = SESSION_STATE_FREE;\r
2736\r
2737 InitializeListHead (&Session->Conns);\r
2738 InitializeListHead (&Session->TcbList);\r
2739 }\r
2740\r
2741 Session->Tsih = 0;\r
2742\r
2743 Session->CmdSN = 1;\r
2744 Session->InitiatorTaskTag = 1;\r
2745 Session->NextCid = 1;\r
2746\r
2747 Session->TargetPortalGroupTag = 0;\r
2748 Session->MaxConnections = ISCSI_MAX_CONNS_PER_SESSION;\r
2749 Session->InitialR2T = FALSE;\r
2750 Session->ImmediateData = TRUE;\r
2751 Session->MaxBurstLength = 262144;\r
2752 Session->FirstBurstLength = MAX_RECV_DATA_SEG_LEN_IN_FFP;\r
2753 Session->DefaultTime2Wait = 2;\r
2754 Session->DefaultTime2Retain = 20;\r
2755 Session->MaxOutstandingR2T = DEFAULT_MAX_OUTSTANDING_R2T;\r
2756 Session->DataPDUInOrder = TRUE;\r
2757 Session->DataSequenceInOrder = TRUE;\r
2758 Session->ErrorRecoveryLevel = 0;\r
2759}\r
2760\r
2761/**\r
2762 Abort the iSCSI session, that is, reset all the connection and free the\r
2763 resources.\r
2764\r
2765 @param[in, out] Session The iSCSI session.\r
2766\r
2767 @retval EFI_SUCCES The session is aborted.\r
2768**/\r
2769EFI_STATUS\r
2770IScsiSessionAbort (\r
2771 IN OUT ISCSI_SESSION *Session\r
2772 )\r
2773{\r
2774 ISCSI_DRIVER_DATA *Private;\r
2775 ISCSI_CONNECTION *Conn;\r
2776\r
2777 if (Session->State != SESSION_STATE_LOGGED_IN) {\r
2778 return EFI_SUCCESS;\r
2779 }\r
2780\r
2781 ASSERT (!IsListEmpty (&Session->Conns));\r
2782\r
2783 Private = ISCSI_DRIVER_DATA_FROM_SESSION (Session);\r
2784\r
2785 while (!IsListEmpty (&Session->Conns)) {\r
2786 Conn = NET_LIST_USER_STRUCT_S (\r
2787 Session->Conns.ForwardLink,\r
2788 ISCSI_CONNECTION,\r
2789 Link,\r
2790 ISCSI_CONNECTION_SIGNATURE\r
2791 );\r
2792\r
2793 gBS->CloseProtocol (\r
2794 Conn->Tcp4Io.Handle,\r
2795 &gEfiTcp4ProtocolGuid,\r
2796 Private->Image,\r
2797 Private->ExtScsiPassThruHandle\r
2798 );\r
2799\r
2800 IScsiConnReset (Conn);\r
2801\r
2802 IScsiDetatchConnection (Conn);\r
2803 IScsiDestroyConnection (Conn);\r
2804 }\r
2805\r
2806 Session->State = SESSION_STATE_FAILED;\r
2807\r
2808 return EFI_SUCCESS;\r
2809}\r