IScsiProto.c 86 KB

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