123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250 |
- /** @file
- Miscellaneous routines for HttpDxe driver.
- Copyright (c) 2015 - 2021, Intel Corporation. All rights reserved.<BR>
- (C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>
- SPDX-License-Identifier: BSD-2-Clause-Patent
- **/
- #include "HttpDriver.h"
- /**
- The common notify function used in HTTP driver.
- @param[in] Event The event signaled.
- @param[in] Context The context.
- **/
- VOID
- EFIAPI
- HttpCommonNotify (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- if ((Event == NULL) || (Context == NULL)) {
- return;
- }
- *((BOOLEAN *)Context) = TRUE;
- }
- /**
- The notify function associated with Tx4Token for Tcp4->Transmit() or Tx6Token for Tcp6->Transmit().
- @param[in] Context The context.
- **/
- VOID
- EFIAPI
- HttpTcpTransmitNotifyDpc (
- IN VOID *Context
- )
- {
- HTTP_TOKEN_WRAP *Wrap;
- HTTP_PROTOCOL *HttpInstance;
- if (Context == NULL) {
- return;
- }
- Wrap = (HTTP_TOKEN_WRAP *)Context;
- HttpInstance = Wrap->HttpInstance;
- if (!HttpInstance->LocalAddressIsIPv6) {
- Wrap->HttpToken->Status = Wrap->TcpWrap.Tx4Token.CompletionToken.Status;
- gBS->SignalEvent (Wrap->HttpToken->Event);
- //
- // Free resources.
- //
- if (Wrap->TcpWrap.Tx4Token.Packet.TxData->FragmentTable[0].FragmentBuffer != NULL) {
- FreePool (Wrap->TcpWrap.Tx4Token.Packet.TxData->FragmentTable[0].FragmentBuffer);
- }
- if (Wrap->TcpWrap.Tx4Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (Wrap->TcpWrap.Tx4Token.CompletionToken.Event);
- }
- } else {
- Wrap->HttpToken->Status = Wrap->TcpWrap.Tx6Token.CompletionToken.Status;
- gBS->SignalEvent (Wrap->HttpToken->Event);
- //
- // Free resources.
- //
- if (Wrap->TcpWrap.Tx6Token.Packet.TxData->FragmentTable[0].FragmentBuffer != NULL) {
- FreePool (Wrap->TcpWrap.Tx6Token.Packet.TxData->FragmentTable[0].FragmentBuffer);
- }
- if (Wrap->TcpWrap.Tx6Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (Wrap->TcpWrap.Tx6Token.CompletionToken.Event);
- }
- }
- Wrap->TcpWrap.IsTxDone = TRUE;
- //
- // Check pending TxTokens and sent out.
- //
- NetMapIterate (&Wrap->HttpInstance->TxTokens, HttpTcpTransmit, NULL);
- }
- /**
- Request HttpTcpTransmitNotifyDpc as a DPC at TPL_CALLBACK.
- @param Event The receive event delivered to TCP for transmit.
- @param Context Context for the callback.
- **/
- VOID
- EFIAPI
- HttpTcpTransmitNotify (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- //
- // Request HttpTcpTransmitNotifyDpc as a DPC at TPL_CALLBACK
- //
- QueueDpc (TPL_CALLBACK, HttpTcpTransmitNotifyDpc, Context);
- }
- /**
- The notify function associated with Rx4Token for Tcp4->Receive () or Rx6Token for Tcp6->Receive().
- @param[in] Context The context.
- **/
- VOID
- EFIAPI
- HttpTcpReceiveNotifyDpc (
- IN VOID *Context
- )
- {
- HTTP_TOKEN_WRAP *Wrap;
- NET_MAP_ITEM *Item;
- UINTN Length;
- EFI_STATUS Status;
- HTTP_PROTOCOL *HttpInstance;
- BOOLEAN UsingIpv6;
- if (Context == NULL) {
- return;
- }
- Wrap = (HTTP_TOKEN_WRAP *)Context;
- HttpInstance = Wrap->HttpInstance;
- UsingIpv6 = HttpInstance->LocalAddressIsIPv6;
- if (UsingIpv6) {
- gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
- Wrap->TcpWrap.Rx6Token.CompletionToken.Event = NULL;
- if (EFI_ERROR (Wrap->TcpWrap.Rx6Token.CompletionToken.Status)) {
- DEBUG ((DEBUG_ERROR, "HttpTcpReceiveNotifyDpc: %r!\n", Wrap->TcpWrap.Rx6Token.CompletionToken.Status));
- Wrap->HttpToken->Status = Wrap->TcpWrap.Rx6Token.CompletionToken.Status;
- gBS->SignalEvent (Wrap->HttpToken->Event);
- Item = NetMapFindKey (&HttpInstance->RxTokens, Wrap->HttpToken);
- if (Item != NULL) {
- NetMapRemoveItem (&HttpInstance->RxTokens, Item, NULL);
- }
- FreePool (Wrap);
- Wrap = NULL;
- return;
- }
- } else {
- gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
- Wrap->TcpWrap.Rx4Token.CompletionToken.Event = NULL;
- if (EFI_ERROR (Wrap->TcpWrap.Rx4Token.CompletionToken.Status)) {
- DEBUG ((DEBUG_ERROR, "HttpTcpReceiveNotifyDpc: %r!\n", Wrap->TcpWrap.Rx4Token.CompletionToken.Status));
- Wrap->HttpToken->Status = Wrap->TcpWrap.Rx4Token.CompletionToken.Status;
- gBS->SignalEvent (Wrap->HttpToken->Event);
- Item = NetMapFindKey (&HttpInstance->RxTokens, Wrap->HttpToken);
- if (Item != NULL) {
- NetMapRemoveItem (&HttpInstance->RxTokens, Item, NULL);
- }
- FreePool (Wrap);
- Wrap = NULL;
- return;
- }
- }
- //
- // Check whether we receive a complete HTTP message.
- //
- ASSERT (HttpInstance->MsgParser != NULL);
- if (UsingIpv6) {
- Length = (UINTN)Wrap->TcpWrap.Rx6Data.FragmentTable[0].FragmentLength;
- } else {
- Length = (UINTN)Wrap->TcpWrap.Rx4Data.FragmentTable[0].FragmentLength;
- }
- //
- // Record the CallbackData data.
- //
- HttpInstance->CallbackData.Wrap = (VOID *)Wrap;
- HttpInstance->CallbackData.ParseData = Wrap->HttpToken->Message->Body;
- HttpInstance->CallbackData.ParseDataLength = Length;
- //
- // Parse Body with CallbackData data.
- //
- Status = HttpParseMessageBody (
- HttpInstance->MsgParser,
- Length,
- Wrap->HttpToken->Message->Body
- );
- if (EFI_ERROR (Status)) {
- return;
- }
- if (HttpIsMessageComplete (HttpInstance->MsgParser)) {
- //
- // Free the MsgParse since we already have a full HTTP message.
- //
- HttpFreeMsgParser (HttpInstance->MsgParser);
- HttpInstance->MsgParser = NULL;
- }
- Wrap->HttpToken->Message->BodyLength = Length;
- ASSERT (HttpInstance->CacheBody == NULL);
- //
- // We receive part of header of next HTTP msg.
- //
- if (HttpInstance->NextMsg != NULL) {
- Wrap->HttpToken->Message->BodyLength = HttpInstance->NextMsg -
- (CHAR8 *)Wrap->HttpToken->Message->Body;
- HttpInstance->CacheLen = Length - Wrap->HttpToken->Message->BodyLength;
- if (HttpInstance->CacheLen != 0) {
- HttpInstance->CacheBody = AllocateZeroPool (HttpInstance->CacheLen);
- if (HttpInstance->CacheBody == NULL) {
- return;
- }
- CopyMem (HttpInstance->CacheBody, HttpInstance->NextMsg, HttpInstance->CacheLen);
- HttpInstance->NextMsg = HttpInstance->CacheBody;
- HttpInstance->CacheOffset = 0;
- }
- }
- Item = NetMapFindKey (&Wrap->HttpInstance->RxTokens, Wrap->HttpToken);
- if (Item != NULL) {
- NetMapRemoveItem (&Wrap->HttpInstance->RxTokens, Item, NULL);
- }
- Wrap->TcpWrap.IsRxDone = TRUE;
- if (UsingIpv6) {
- Wrap->HttpToken->Status = Wrap->TcpWrap.Rx6Token.CompletionToken.Status;
- } else {
- Wrap->HttpToken->Status = Wrap->TcpWrap.Rx4Token.CompletionToken.Status;
- }
- gBS->SignalEvent (Wrap->HttpToken->Event);
- //
- // Check pending RxTokens and receive the HTTP message.
- //
- NetMapIterate (&Wrap->HttpInstance->RxTokens, HttpTcpReceive, NULL);
- FreePool (Wrap);
- Wrap = NULL;
- }
- /**
- Request HttpTcpReceiveNotifyDpc as a DPC at TPL_CALLBACK.
- @param Event The receive event delivered to TCP for receive.
- @param Context Context for the callback.
- **/
- VOID
- EFIAPI
- HttpTcpReceiveNotify (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- //
- // Request HttpTcpTransmitNotifyDpc as a DPC at TPL_CALLBACK
- //
- QueueDpc (TPL_CALLBACK, HttpTcpReceiveNotifyDpc, Context);
- }
- /**
- Create events for the TCP connection token and TCP close token.
- @param[in] HttpInstance Pointer to HTTP_PROTOCOL structure.
- @retval EFI_SUCCESS The events are created successfully.
- @retval others Other error as indicated.
- **/
- EFI_STATUS
- HttpCreateTcpConnCloseEvent (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- if (!HttpInstance->LocalAddressIsIPv6) {
- //
- // Create events for various asynchronous operations.
- //
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsTcp4ConnDone,
- &HttpInstance->Tcp4ConnToken.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- goto ERROR;
- }
- //
- // Initialize Tcp4CloseToken
- //
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsTcp4CloseDone,
- &HttpInstance->Tcp4CloseToken.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- goto ERROR;
- }
- } else {
- //
- // Create events for various asynchronous operations.
- //
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsTcp6ConnDone,
- &HttpInstance->Tcp6ConnToken.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- goto ERROR;
- }
- //
- // Initialize Tcp6CloseToken
- //
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsTcp6CloseDone,
- &HttpInstance->Tcp6CloseToken.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- goto ERROR;
- }
- }
- return EFI_SUCCESS;
- ERROR:
- //
- // Error handling
- //
- HttpCloseTcpConnCloseEvent (HttpInstance);
- return Status;
- }
- /**
- Close events in the TCP connection token and TCP close token.
- @param[in] HttpInstance Pointer to HTTP_PROTOCOL structure.
- **/
- VOID
- HttpCloseTcpConnCloseEvent (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- ASSERT (HttpInstance != NULL);
- if (HttpInstance->LocalAddressIsIPv6) {
- if (NULL != HttpInstance->Tcp6ConnToken.CompletionToken.Event) {
- gBS->CloseEvent (HttpInstance->Tcp6ConnToken.CompletionToken.Event);
- HttpInstance->Tcp6ConnToken.CompletionToken.Event = NULL;
- }
- if (NULL != HttpInstance->Tcp6CloseToken.CompletionToken.Event) {
- gBS->CloseEvent (HttpInstance->Tcp6CloseToken.CompletionToken.Event);
- HttpInstance->Tcp6CloseToken.CompletionToken.Event = NULL;
- }
- } else {
- if (NULL != HttpInstance->Tcp4ConnToken.CompletionToken.Event) {
- gBS->CloseEvent (HttpInstance->Tcp4ConnToken.CompletionToken.Event);
- HttpInstance->Tcp4ConnToken.CompletionToken.Event = NULL;
- }
- if (NULL != HttpInstance->Tcp4CloseToken.CompletionToken.Event) {
- gBS->CloseEvent (HttpInstance->Tcp4CloseToken.CompletionToken.Event);
- HttpInstance->Tcp4CloseToken.CompletionToken.Event = NULL;
- }
- }
- }
- /**
- Create event for the TCP transmit token.
- @param[in] Wrap Point to HTTP token's wrap data.
- @retval EFI_SUCCESS The events is created successfully.
- @retval others Other error as indicated.
- **/
- EFI_STATUS
- HttpCreateTcpTxEvent (
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- EFI_STATUS Status;
- HTTP_PROTOCOL *HttpInstance;
- HTTP_TCP_TOKEN_WRAP *TcpWrap;
- HttpInstance = Wrap->HttpInstance;
- TcpWrap = &Wrap->TcpWrap;
- if (!HttpInstance->LocalAddressIsIPv6) {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpTcpTransmitNotify,
- Wrap,
- &TcpWrap->Tx4Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- TcpWrap->Tx4Data.Push = TRUE;
- TcpWrap->Tx4Data.Urgent = FALSE;
- TcpWrap->Tx4Data.FragmentCount = 1;
- TcpWrap->Tx4Token.Packet.TxData = &Wrap->TcpWrap.Tx4Data;
- TcpWrap->Tx4Token.CompletionToken.Status = EFI_NOT_READY;
- } else {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpTcpTransmitNotify,
- Wrap,
- &TcpWrap->Tx6Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- TcpWrap->Tx6Data.Push = TRUE;
- TcpWrap->Tx6Data.Urgent = FALSE;
- TcpWrap->Tx6Data.FragmentCount = 1;
- TcpWrap->Tx6Token.Packet.TxData = &Wrap->TcpWrap.Tx6Data;
- TcpWrap->Tx6Token.CompletionToken.Status = EFI_NOT_READY;
- }
- return EFI_SUCCESS;
- }
- /**
- Create event for the TCP receive token which is used to receive HTTP header.
- @param[in] HttpInstance Pointer to HTTP_PROTOCOL structure.
- @retval EFI_SUCCESS The events is created successfully.
- @retval others Other error as indicated.
- **/
- EFI_STATUS
- HttpCreateTcpRxEventForHeader (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- if (!HttpInstance->LocalAddressIsIPv6) {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsRxDone,
- &HttpInstance->Rx4Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- HttpInstance->Rx4Data.FragmentCount = 1;
- HttpInstance->Rx4Token.Packet.RxData = &HttpInstance->Rx4Data;
- HttpInstance->Rx4Token.CompletionToken.Status = EFI_NOT_READY;
- } else {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpCommonNotify,
- &HttpInstance->IsRxDone,
- &HttpInstance->Rx6Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- HttpInstance->Rx6Data.FragmentCount = 1;
- HttpInstance->Rx6Token.Packet.RxData = &HttpInstance->Rx6Data;
- HttpInstance->Rx6Token.CompletionToken.Status = EFI_NOT_READY;
- }
- return EFI_SUCCESS;
- }
- /**
- Create event for the TCP receive token which is used to receive HTTP body.
- @param[in] Wrap Point to HTTP token's wrap data.
- @retval EFI_SUCCESS The events is created successfully.
- @retval others Other error as indicated.
- **/
- EFI_STATUS
- HttpCreateTcpRxEvent (
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- EFI_STATUS Status;
- HTTP_PROTOCOL *HttpInstance;
- HTTP_TCP_TOKEN_WRAP *TcpWrap;
- HttpInstance = Wrap->HttpInstance;
- TcpWrap = &Wrap->TcpWrap;
- if (!HttpInstance->LocalAddressIsIPv6) {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpTcpReceiveNotify,
- Wrap,
- &TcpWrap->Rx4Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- TcpWrap->Rx4Data.FragmentCount = 1;
- TcpWrap->Rx4Token.Packet.RxData = &Wrap->TcpWrap.Rx4Data;
- TcpWrap->Rx4Token.CompletionToken.Status = EFI_NOT_READY;
- } else {
- Status = gBS->CreateEvent (
- EVT_NOTIFY_SIGNAL,
- TPL_NOTIFY,
- HttpTcpReceiveNotify,
- Wrap,
- &TcpWrap->Rx6Token.CompletionToken.Event
- );
- if (EFI_ERROR (Status)) {
- return Status;
- }
- TcpWrap->Rx6Data.FragmentCount = 1;
- TcpWrap->Rx6Token.Packet.RxData = &Wrap->TcpWrap.Rx6Data;
- TcpWrap->Rx6Token.CompletionToken.Status = EFI_NOT_READY;
- }
- return EFI_SUCCESS;
- }
- /**
- Close Events for Tcp Receive Tokens for HTTP body and HTTP header.
- @param[in] Wrap Pointer to HTTP token's wrap data.
- **/
- VOID
- HttpCloseTcpRxEvent (
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- HTTP_PROTOCOL *HttpInstance;
- ASSERT (Wrap != NULL);
- HttpInstance = Wrap->HttpInstance;
- if (HttpInstance->LocalAddressIsIPv6) {
- if (Wrap->TcpWrap.Rx6Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
- }
- if (HttpInstance->Rx6Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (HttpInstance->Rx6Token.CompletionToken.Event);
- HttpInstance->Rx6Token.CompletionToken.Event = NULL;
- }
- } else {
- if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
- }
- if (HttpInstance->Rx4Token.CompletionToken.Event != NULL) {
- gBS->CloseEvent (HttpInstance->Rx4Token.CompletionToken.Event);
- HttpInstance->Rx4Token.CompletionToken.Event = NULL;
- }
- }
- }
- /**
- Initialize the HTTP_PROTOCOL structure to the unconfigured state.
- @param[in, out] HttpInstance Pointer to HTTP_PROTOCOL structure.
- @param[in] IpVersion Indicate us TCP4 protocol or TCP6 protocol.
- @retval EFI_SUCCESS HTTP_PROTOCOL structure is initialized successfully.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpInitProtocol (
- IN OUT HTTP_PROTOCOL *HttpInstance,
- IN BOOLEAN IpVersion
- )
- {
- EFI_STATUS Status;
- VOID *Interface;
- BOOLEAN UsingIpv6;
- ASSERT (HttpInstance != NULL);
- UsingIpv6 = IpVersion;
- if (!UsingIpv6) {
- //
- // Create TCP4 child.
- //
- Status = NetLibCreateServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip4DriverBindingHandle,
- &gEfiTcp4ServiceBindingProtocolGuid,
- &HttpInstance->Tcp4ChildHandle
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- (VOID **)&Interface,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Service->ControllerHandle,
- EFI_OPEN_PROTOCOL_BY_DRIVER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- (VOID **)&HttpInstance->Tcp4,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle,
- EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Service->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- (VOID **)&Interface,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle,
- EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- } else {
- //
- // Create TCP6 Child.
- //
- Status = NetLibCreateServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip6DriverBindingHandle,
- &gEfiTcp6ServiceBindingProtocolGuid,
- &HttpInstance->Tcp6ChildHandle
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- (VOID **)&Interface,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Service->ControllerHandle,
- EFI_OPEN_PROTOCOL_BY_DRIVER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- (VOID **)&HttpInstance->Tcp6,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle,
- EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- Status = gBS->OpenProtocol (
- HttpInstance->Service->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- (VOID **)&Interface,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle,
- EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- }
- HttpInstance->Url = AllocateZeroPool (HTTP_URL_BUFFER_LEN);
- if (HttpInstance->Url == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- goto ON_ERROR;
- }
- return EFI_SUCCESS;
- ON_ERROR:
- if (HttpInstance->Tcp4ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Service->ControllerHandle
- );
- gBS->CloseProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle
- );
- NetLibDestroyServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip4DriverBindingHandle,
- &gEfiTcp4ServiceBindingProtocolGuid,
- HttpInstance->Tcp4ChildHandle
- );
- }
- if (HttpInstance->Service->Tcp4ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Service->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle
- );
- }
- if (HttpInstance->Tcp6ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Service->ControllerHandle
- );
- gBS->CloseProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle
- );
- NetLibDestroyServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip6DriverBindingHandle,
- &gEfiTcp6ServiceBindingProtocolGuid,
- HttpInstance->Tcp6ChildHandle
- );
- }
- if (HttpInstance->Service->Tcp6ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Service->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle
- );
- }
- return EFI_UNSUPPORTED;
- }
- /**
- Clean up the HTTP child, release all the resources used by it.
- @param[in] HttpInstance The HTTP child to clean up.
- **/
- VOID
- HttpCleanProtocol (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- HttpCloseConnection (HttpInstance);
- HttpCloseTcpConnCloseEvent (HttpInstance);
- if (HttpInstance->TimeoutEvent != NULL) {
- gBS->CloseEvent (HttpInstance->TimeoutEvent);
- HttpInstance->TimeoutEvent = NULL;
- }
- if (HttpInstance->CacheBody != NULL) {
- FreePool (HttpInstance->CacheBody);
- HttpInstance->CacheBody = NULL;
- HttpInstance->NextMsg = NULL;
- }
- if (HttpInstance->RemoteHost != NULL) {
- FreePool (HttpInstance->RemoteHost);
- HttpInstance->RemoteHost = NULL;
- }
- if (HttpInstance->MsgParser != NULL) {
- HttpFreeMsgParser (HttpInstance->MsgParser);
- HttpInstance->MsgParser = NULL;
- }
- if (HttpInstance->Url != NULL) {
- FreePool (HttpInstance->Url);
- HttpInstance->Url = NULL;
- }
- NetMapClean (&HttpInstance->TxTokens);
- NetMapClean (&HttpInstance->RxTokens);
- if ((HttpInstance->TlsSb != NULL) && (HttpInstance->TlsChildHandle != NULL)) {
- //
- // Destroy the TLS instance.
- //
- HttpInstance->TlsSb->DestroyChild (HttpInstance->TlsSb, HttpInstance->TlsChildHandle);
- HttpInstance->TlsChildHandle = NULL;
- }
- if (HttpInstance->Tcp4ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Service->ControllerHandle
- );
- gBS->CloseProtocol (
- HttpInstance->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle
- );
- NetLibDestroyServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip4DriverBindingHandle,
- &gEfiTcp4ServiceBindingProtocolGuid,
- HttpInstance->Tcp4ChildHandle
- );
- }
- if (HttpInstance->Service->Tcp4ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Service->Tcp4ChildHandle,
- &gEfiTcp4ProtocolGuid,
- HttpInstance->Service->Ip4DriverBindingHandle,
- HttpInstance->Handle
- );
- }
- if (HttpInstance->Tcp6ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Service->ControllerHandle
- );
- gBS->CloseProtocol (
- HttpInstance->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle
- );
- NetLibDestroyServiceChild (
- HttpInstance->Service->ControllerHandle,
- HttpInstance->Service->Ip6DriverBindingHandle,
- &gEfiTcp6ServiceBindingProtocolGuid,
- HttpInstance->Tcp6ChildHandle
- );
- }
- if (HttpInstance->Service->Tcp6ChildHandle != NULL) {
- gBS->CloseProtocol (
- HttpInstance->Service->Tcp6ChildHandle,
- &gEfiTcp6ProtocolGuid,
- HttpInstance->Service->Ip6DriverBindingHandle,
- HttpInstance->Handle
- );
- }
- TlsCloseTxRxEvent (HttpInstance);
- }
- /**
- Establish TCP connection with HTTP server.
- @param[in] HttpInstance The HTTP instance private data.
- @retval EFI_SUCCESS The TCP connection is established.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpCreateConnection (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- //
- // Connect to Http server
- //
- if (!HttpInstance->LocalAddressIsIPv6) {
- HttpInstance->IsTcp4ConnDone = FALSE;
- HttpInstance->Tcp4ConnToken.CompletionToken.Status = EFI_NOT_READY;
- Status = HttpInstance->Tcp4->Connect (HttpInstance->Tcp4, &HttpInstance->Tcp4ConnToken);
- HttpNotify (HttpEventConnectTcp, Status);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpCreateConnection: Tcp4->Connect() = %r\n", Status));
- return Status;
- }
- while (!HttpInstance->IsTcp4ConnDone) {
- HttpInstance->Tcp4->Poll (HttpInstance->Tcp4);
- }
- Status = HttpInstance->Tcp4ConnToken.CompletionToken.Status;
- } else {
- HttpInstance->IsTcp6ConnDone = FALSE;
- HttpInstance->Tcp6ConnToken.CompletionToken.Status = EFI_NOT_READY;
- Status = HttpInstance->Tcp6->Connect (HttpInstance->Tcp6, &HttpInstance->Tcp6ConnToken);
- HttpNotify (HttpEventConnectTcp, Status);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpCreateConnection: Tcp6->Connect() = %r\n", Status));
- return Status;
- }
- while (!HttpInstance->IsTcp6ConnDone) {
- HttpInstance->Tcp6->Poll (HttpInstance->Tcp6);
- }
- Status = HttpInstance->Tcp6ConnToken.CompletionToken.Status;
- }
- if (!EFI_ERROR (Status)) {
- HttpInstance->State = HTTP_STATE_TCP_CONNECTED;
- }
- return Status;
- }
- /**
- Close existing TCP connection.
- @param[in] HttpInstance The HTTP instance private data.
- @retval EFI_SUCCESS The TCP connection is closed.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpCloseConnection (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- if (HttpInstance->State == HTTP_STATE_TCP_CONNECTED) {
- if (HttpInstance->LocalAddressIsIPv6) {
- HttpInstance->Tcp6CloseToken.AbortOnClose = TRUE;
- HttpInstance->IsTcp6CloseDone = FALSE;
- Status = HttpInstance->Tcp6->Close (HttpInstance->Tcp6, &HttpInstance->Tcp6CloseToken);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- while (!HttpInstance->IsTcp6CloseDone) {
- HttpInstance->Tcp6->Poll (HttpInstance->Tcp6);
- }
- } else {
- HttpInstance->Tcp4CloseToken.AbortOnClose = TRUE;
- HttpInstance->IsTcp4CloseDone = FALSE;
- Status = HttpInstance->Tcp4->Close (HttpInstance->Tcp4, &HttpInstance->Tcp4CloseToken);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- while (!HttpInstance->IsTcp4CloseDone) {
- HttpInstance->Tcp4->Poll (HttpInstance->Tcp4);
- }
- }
- }
- HttpInstance->State = HTTP_STATE_TCP_CLOSED;
- return EFI_SUCCESS;
- }
- /**
- Configure TCP4 protocol child.
- @param[in] HttpInstance The HTTP instance private data.
- @param[in] Wrap The HTTP token's wrap data.
- @retval EFI_SUCCESS The TCP4 protocol child is configured.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpConfigureTcp4 (
- IN HTTP_PROTOCOL *HttpInstance,
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- EFI_STATUS Status;
- EFI_TCP4_CONFIG_DATA *Tcp4CfgData;
- EFI_TCP4_ACCESS_POINT *Tcp4AP;
- EFI_TCP4_OPTION *Tcp4Option;
- ASSERT (HttpInstance != NULL);
- Tcp4CfgData = &HttpInstance->Tcp4CfgData;
- ZeroMem (Tcp4CfgData, sizeof (EFI_TCP4_CONFIG_DATA));
- Tcp4CfgData->TypeOfService = HTTP_TOS_DEAULT;
- Tcp4CfgData->TimeToLive = HTTP_TTL_DEAULT;
- Tcp4CfgData->ControlOption = &HttpInstance->Tcp4Option;
- Tcp4AP = &Tcp4CfgData->AccessPoint;
- Tcp4AP->UseDefaultAddress = HttpInstance->IPv4Node.UseDefaultAddress;
- if (!Tcp4AP->UseDefaultAddress) {
- IP4_COPY_ADDRESS (&Tcp4AP->StationAddress, &HttpInstance->IPv4Node.LocalAddress);
- IP4_COPY_ADDRESS (&Tcp4AP->SubnetMask, &HttpInstance->IPv4Node.LocalSubnet);
- }
- Tcp4AP->StationPort = HttpInstance->IPv4Node.LocalPort;
- Tcp4AP->RemotePort = HttpInstance->RemotePort;
- Tcp4AP->ActiveFlag = TRUE;
- IP4_COPY_ADDRESS (&Tcp4AP->RemoteAddress, &HttpInstance->RemoteAddr);
- Tcp4Option = Tcp4CfgData->ControlOption;
- Tcp4Option->ReceiveBufferSize = HTTP_BUFFER_SIZE_DEAULT;
- Tcp4Option->SendBufferSize = HTTP_BUFFER_SIZE_DEAULT;
- Tcp4Option->MaxSynBackLog = HTTP_MAX_SYN_BACK_LOG;
- Tcp4Option->ConnectionTimeout = HTTP_CONNECTION_TIMEOUT;
- Tcp4Option->DataRetries = HTTP_DATA_RETRIES;
- Tcp4Option->FinTimeout = HTTP_FIN_TIMEOUT;
- Tcp4Option->KeepAliveProbes = HTTP_KEEP_ALIVE_PROBES;
- Tcp4Option->KeepAliveTime = HTTP_KEEP_ALIVE_TIME;
- Tcp4Option->KeepAliveInterval = HTTP_KEEP_ALIVE_INTERVAL;
- Tcp4Option->EnableNagle = TRUE;
- Tcp4CfgData->ControlOption = Tcp4Option;
- if ((HttpInstance->State == HTTP_STATE_TCP_CONNECTED) ||
- (HttpInstance->State == HTTP_STATE_TCP_CLOSED))
- {
- Status = HttpInstance->Tcp4->Configure (HttpInstance->Tcp4, NULL);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpConfigureTcp4(NULL) - %r\n", Status));
- return Status;
- }
- HttpInstance->State = HTTP_STATE_TCP_UNCONFIGED;
- }
- Status = HttpInstance->Tcp4->Configure (HttpInstance->Tcp4, Tcp4CfgData);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpConfigureTcp4 - %r\n", Status));
- return Status;
- }
- Status = HttpCreateTcpConnCloseEvent (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- Status = HttpCreateTcpTxEvent (Wrap);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- HttpInstance->State = HTTP_STATE_TCP_CONFIGED;
- return EFI_SUCCESS;
- }
- /**
- Configure TCP6 protocol child.
- @param[in] HttpInstance The HTTP instance private data.
- @param[in] Wrap The HTTP token's wrap data.
- @retval EFI_SUCCESS The TCP6 protocol child is configured.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpConfigureTcp6 (
- IN HTTP_PROTOCOL *HttpInstance,
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- EFI_STATUS Status;
- EFI_TCP6_CONFIG_DATA *Tcp6CfgData;
- EFI_TCP6_ACCESS_POINT *Tcp6Ap;
- EFI_TCP6_OPTION *Tcp6Option;
- ASSERT (HttpInstance != NULL);
- Tcp6CfgData = &HttpInstance->Tcp6CfgData;
- ZeroMem (Tcp6CfgData, sizeof (EFI_TCP6_CONFIG_DATA));
- Tcp6CfgData->TrafficClass = 0;
- Tcp6CfgData->HopLimit = 255;
- Tcp6CfgData->ControlOption = &HttpInstance->Tcp6Option;
- Tcp6Ap = &Tcp6CfgData->AccessPoint;
- Tcp6Ap->ActiveFlag = TRUE;
- Tcp6Ap->StationPort = HttpInstance->Ipv6Node.LocalPort;
- Tcp6Ap->RemotePort = HttpInstance->RemotePort;
- IP6_COPY_ADDRESS (&Tcp6Ap->StationAddress, &HttpInstance->Ipv6Node.LocalAddress);
- IP6_COPY_ADDRESS (&Tcp6Ap->RemoteAddress, &HttpInstance->RemoteIpv6Addr);
- Tcp6Option = Tcp6CfgData->ControlOption;
- Tcp6Option->ReceiveBufferSize = HTTP_BUFFER_SIZE_DEAULT;
- Tcp6Option->SendBufferSize = HTTP_BUFFER_SIZE_DEAULT;
- Tcp6Option->MaxSynBackLog = HTTP_MAX_SYN_BACK_LOG;
- Tcp6Option->ConnectionTimeout = HTTP_CONNECTION_TIMEOUT;
- Tcp6Option->DataRetries = HTTP_DATA_RETRIES;
- Tcp6Option->FinTimeout = HTTP_FIN_TIMEOUT;
- Tcp6Option->KeepAliveProbes = HTTP_KEEP_ALIVE_PROBES;
- Tcp6Option->KeepAliveTime = HTTP_KEEP_ALIVE_TIME;
- Tcp6Option->KeepAliveInterval = HTTP_KEEP_ALIVE_INTERVAL;
- Tcp6Option->EnableNagle = TRUE;
- if ((HttpInstance->State == HTTP_STATE_TCP_CONNECTED) ||
- (HttpInstance->State == HTTP_STATE_TCP_CLOSED))
- {
- Status = HttpInstance->Tcp6->Configure (HttpInstance->Tcp6, NULL);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpConfigureTcp6(NULL) - %r\n", Status));
- return Status;
- }
- HttpInstance->State = HTTP_STATE_TCP_UNCONFIGED;
- }
- Status = HttpInstance->Tcp6->Configure (HttpInstance->Tcp6, Tcp6CfgData);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "HttpConfigureTcp6 - %r\n", Status));
- return Status;
- }
- Status = HttpCreateTcpConnCloseEvent (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- Status = HttpCreateTcpTxEvent (Wrap);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- HttpInstance->State = HTTP_STATE_TCP_CONFIGED;
- return EFI_SUCCESS;
- }
- /**
- Check existing TCP connection, if in error state, recover TCP4 connection. Then,
- connect one TLS session if required.
- @param[in] HttpInstance The HTTP instance private data.
- @retval EFI_SUCCESS The TCP connection is established.
- @retval EFI_NOT_READY TCP4 protocol child is not created or configured.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpConnectTcp4 (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- EFI_TCP4_CONNECTION_STATE Tcp4State;
- if ((HttpInstance->State < HTTP_STATE_TCP_CONFIGED) || (HttpInstance->Tcp4 == NULL)) {
- return EFI_NOT_READY;
- }
- Status = HttpInstance->Tcp4->GetModeData (
- HttpInstance->Tcp4,
- &Tcp4State,
- NULL,
- NULL,
- NULL,
- NULL
- );
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp4 GetModeData fail - %x\n", Status));
- return Status;
- }
- if (Tcp4State == Tcp4StateEstablished) {
- return EFI_SUCCESS;
- } else if (Tcp4State > Tcp4StateEstablished ) {
- HttpCloseConnection (HttpInstance);
- }
- Status = HttpCreateConnection (HttpInstance);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp4 Connection fail - %x\n", Status));
- return Status;
- }
- //
- // Tls session connection.
- //
- if (HttpInstance->UseHttps) {
- if (HttpInstance->TimeoutEvent == NULL) {
- //
- // Create TimeoutEvent for TLS connection.
- //
- Status = gBS->CreateEvent (
- EVT_TIMER,
- TPL_CALLBACK,
- NULL,
- NULL,
- &HttpInstance->TimeoutEvent
- );
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- }
- //
- // Start the timer, and wait Timeout seconds for connection.
- //
- Status = gBS->SetTimer (HttpInstance->TimeoutEvent, TimerRelative, HTTP_CONNECTION_TIMEOUT * TICKS_PER_SECOND);
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- Status = TlsConnectSession (HttpInstance, HttpInstance->TimeoutEvent);
- HttpNotify (HttpEventTlsConnectSession, Status);
- gBS->SetTimer (HttpInstance->TimeoutEvent, TimerCancel, 0);
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- }
- return Status;
- }
- /**
- Check existing TCP connection, if in error state, recover TCP6 connection. Then,
- connect one TLS session if required.
- @param[in] HttpInstance The HTTP instance private data.
- @retval EFI_SUCCESS The TCP connection is established.
- @retval EFI_NOT_READY TCP6 protocol child is not created or configured.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpConnectTcp6 (
- IN HTTP_PROTOCOL *HttpInstance
- )
- {
- EFI_STATUS Status;
- EFI_TCP6_CONNECTION_STATE Tcp6State;
- if ((HttpInstance->State < HTTP_STATE_TCP_CONFIGED) || (HttpInstance->Tcp6 == NULL)) {
- return EFI_NOT_READY;
- }
- Status = HttpInstance->Tcp6->GetModeData (
- HttpInstance->Tcp6,
- &Tcp6State,
- NULL,
- NULL,
- NULL,
- NULL
- );
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp6 GetModeData fail - %x\n", Status));
- return Status;
- }
- if (Tcp6State == Tcp6StateEstablished) {
- return EFI_SUCCESS;
- } else if (Tcp6State > Tcp6StateEstablished ) {
- HttpCloseConnection (HttpInstance);
- }
- Status = HttpCreateConnection (HttpInstance);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp6 Connection fail - %x\n", Status));
- return Status;
- }
- //
- // Tls session connection.
- //
- if (HttpInstance->UseHttps) {
- if (HttpInstance->TimeoutEvent == NULL) {
- //
- // Create TimeoutEvent for TLS connection.
- //
- Status = gBS->CreateEvent (
- EVT_TIMER,
- TPL_CALLBACK,
- NULL,
- NULL,
- &HttpInstance->TimeoutEvent
- );
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- }
- //
- // Start the timer, and wait Timeout seconds for connection.
- //
- Status = gBS->SetTimer (HttpInstance->TimeoutEvent, TimerRelative, HTTP_CONNECTION_TIMEOUT * TICKS_PER_SECOND);
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- Status = TlsConnectSession (HttpInstance, HttpInstance->TimeoutEvent);
- HttpNotify (HttpEventTlsConnectSession, Status);
- gBS->SetTimer (HttpInstance->TimeoutEvent, TimerCancel, 0);
- if (EFI_ERROR (Status)) {
- TlsCloseTxRxEvent (HttpInstance);
- return Status;
- }
- }
- return Status;
- }
- /**
- Initialize Http session.
- @param[in] HttpInstance The HTTP instance private data.
- @param[in] Wrap The HTTP token's wrap data.
- @param[in] Configure The Flag indicates whether need to initialize session.
- @param[in] TlsConfigure The Flag indicates whether it's the new Tls session.
- @retval EFI_SUCCESS The initialization of session is done.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpInitSession (
- IN HTTP_PROTOCOL *HttpInstance,
- IN HTTP_TOKEN_WRAP *Wrap,
- IN BOOLEAN Configure,
- IN BOOLEAN TlsConfigure
- )
- {
- EFI_STATUS Status;
- ASSERT (HttpInstance != NULL);
- //
- // Configure Tls session.
- //
- if (TlsConfigure) {
- Status = TlsConfigureSession (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- }
- if (!HttpInstance->LocalAddressIsIPv6) {
- //
- // Configure TCP instance.
- //
- if (Configure) {
- Status = HttpConfigureTcp4 (HttpInstance, Wrap);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- }
- //
- // Connect TCP.
- //
- Status = HttpConnectTcp4 (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- } else {
- //
- // Configure TCP instance.
- //
- if (Configure) {
- Status = HttpConfigureTcp6 (HttpInstance, Wrap);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- }
- //
- // Connect TCP.
- //
- Status = HttpConnectTcp6 (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- }
- return EFI_SUCCESS;
- }
- /**
- Send the HTTP or HTTPS message through TCP4 or TCP6.
- @param[in] HttpInstance The HTTP instance private data.
- @param[in] Wrap The HTTP token's wrap data.
- @param[in] TxString Buffer containing the HTTP message string.
- @param[in] TxStringLen Length of the HTTP message string in bytes.
- @retval EFI_SUCCESS The HTTP message is queued into TCP transmit queue.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpTransmitTcp (
- IN HTTP_PROTOCOL *HttpInstance,
- IN HTTP_TOKEN_WRAP *Wrap,
- IN UINT8 *TxString,
- IN UINTN TxStringLen
- )
- {
- EFI_STATUS Status;
- EFI_TCP4_IO_TOKEN *Tx4Token;
- EFI_TCP4_PROTOCOL *Tcp4;
- EFI_TCP6_IO_TOKEN *Tx6Token;
- EFI_TCP6_PROTOCOL *Tcp6;
- UINT8 *TlsRecord;
- UINT16 PayloadSize;
- NET_FRAGMENT TempFragment;
- NET_FRAGMENT Fragment;
- UINTN RecordCount;
- UINTN RemainingLen;
- Status = EFI_SUCCESS;
- TlsRecord = NULL;
- PayloadSize = 0;
- TempFragment.Len = 0;
- TempFragment.Bulk = NULL;
- Fragment.Len = 0;
- Fragment.Bulk = NULL;
- RecordCount = 0;
- RemainingLen = 0;
- //
- // Need to encrypt data.
- //
- if (HttpInstance->UseHttps) {
- //
- // Allocate enough buffer for each TLS plaintext records.
- //
- TlsRecord = AllocateZeroPool (TLS_RECORD_HEADER_LENGTH + TLS_PLAINTEXT_RECORD_MAX_PAYLOAD_LENGTH);
- if (TlsRecord == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- return Status;
- }
- //
- // Allocate enough buffer for all TLS ciphertext records.
- //
- RecordCount = TxStringLen / TLS_PLAINTEXT_RECORD_MAX_PAYLOAD_LENGTH + 1;
- Fragment.Bulk = AllocateZeroPool (RecordCount * (TLS_RECORD_HEADER_LENGTH + TLS_CIPHERTEXT_RECORD_MAX_PAYLOAD_LENGTH));
- if (Fragment.Bulk == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- goto ON_ERROR;
- }
- //
- // Encrypt each TLS plaintext records.
- //
- RemainingLen = TxStringLen;
- while (RemainingLen != 0) {
- PayloadSize = (UINT16)MIN (TLS_PLAINTEXT_RECORD_MAX_PAYLOAD_LENGTH, RemainingLen);
- ((TLS_RECORD_HEADER *)TlsRecord)->ContentType = TlsContentTypeApplicationData;
- ((TLS_RECORD_HEADER *)TlsRecord)->Version.Major = HttpInstance->TlsConfigData.Version.Major;
- ((TLS_RECORD_HEADER *)TlsRecord)->Version.Minor = HttpInstance->TlsConfigData.Version.Minor;
- ((TLS_RECORD_HEADER *)TlsRecord)->Length = PayloadSize;
- CopyMem (TlsRecord + TLS_RECORD_HEADER_LENGTH, TxString + (TxStringLen - RemainingLen), PayloadSize);
- Status = TlsProcessMessage (
- HttpInstance,
- TlsRecord,
- TLS_RECORD_HEADER_LENGTH + PayloadSize,
- EfiTlsEncrypt,
- &TempFragment
- );
- if (EFI_ERROR (Status)) {
- goto ON_ERROR;
- }
- //
- // Record the processed/encrypted Packet.
- //
- CopyMem (Fragment.Bulk + Fragment.Len, TempFragment.Bulk, TempFragment.Len);
- Fragment.Len += TempFragment.Len;
- FreePool (TempFragment.Bulk);
- TempFragment.Len = 0;
- TempFragment.Bulk = NULL;
- RemainingLen -= (UINTN)PayloadSize;
- ZeroMem (TlsRecord, TLS_RECORD_HEADER_LENGTH + TLS_PLAINTEXT_RECORD_MAX_PAYLOAD_LENGTH);
- }
- FreePool (TlsRecord);
- TlsRecord = NULL;
- }
- if (!HttpInstance->LocalAddressIsIPv6) {
- Tcp4 = HttpInstance->Tcp4;
- Tx4Token = &Wrap->TcpWrap.Tx4Token;
- if (HttpInstance->UseHttps) {
- Tx4Token->Packet.TxData->DataLength = Fragment.Len;
- Tx4Token->Packet.TxData->FragmentTable[0].FragmentLength = Fragment.Len;
- Tx4Token->Packet.TxData->FragmentTable[0].FragmentBuffer = (VOID *)Fragment.Bulk;
- } else {
- Tx4Token->Packet.TxData->DataLength = (UINT32)TxStringLen;
- Tx4Token->Packet.TxData->FragmentTable[0].FragmentLength = (UINT32)TxStringLen;
- Tx4Token->Packet.TxData->FragmentTable[0].FragmentBuffer = (VOID *)TxString;
- }
- Tx4Token->CompletionToken.Status = EFI_NOT_READY;
- Wrap->TcpWrap.IsTxDone = FALSE;
- Status = Tcp4->Transmit (Tcp4, Tx4Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Transmit failed: %r\n", Status));
- goto ON_ERROR;
- }
- } else {
- Tcp6 = HttpInstance->Tcp6;
- Tx6Token = &Wrap->TcpWrap.Tx6Token;
- if (HttpInstance->UseHttps) {
- Tx6Token->Packet.TxData->DataLength = Fragment.Len;
- Tx6Token->Packet.TxData->FragmentTable[0].FragmentLength = Fragment.Len;
- Tx6Token->Packet.TxData->FragmentTable[0].FragmentBuffer = (VOID *)Fragment.Bulk;
- } else {
- Tx6Token->Packet.TxData->DataLength = (UINT32)TxStringLen;
- Tx6Token->Packet.TxData->FragmentTable[0].FragmentLength = (UINT32)TxStringLen;
- Tx6Token->Packet.TxData->FragmentTable[0].FragmentBuffer = (VOID *)TxString;
- }
- Tx6Token->CompletionToken.Status = EFI_NOT_READY;
- Wrap->TcpWrap.IsTxDone = FALSE;
- Status = Tcp6->Transmit (Tcp6, Tx6Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Transmit failed: %r\n", Status));
- goto ON_ERROR;
- }
- }
- return Status;
- ON_ERROR:
- if (HttpInstance->UseHttps) {
- if (TlsRecord != NULL) {
- FreePool (TlsRecord);
- TlsRecord = NULL;
- }
- if (Fragment.Bulk != NULL) {
- FreePool (Fragment.Bulk);
- Fragment.Bulk = NULL;
- }
- }
- return Status;
- }
- /**
- Check whether the user's token or event has already
- been enqueue on HTTP Tx or Rx Token list.
- @param[in] Map The container of either user's transmit or receive
- token.
- @param[in] Item Current item to check against.
- @param[in] Context The Token to check against.
- @retval EFI_ACCESS_DENIED The token or event has already been enqueued in IP
- @retval EFI_SUCCESS The current item isn't the same token/event as the
- context.
- **/
- EFI_STATUS
- EFIAPI
- HttpTokenExist (
- IN NET_MAP *Map,
- IN NET_MAP_ITEM *Item,
- IN VOID *Context
- )
- {
- EFI_HTTP_TOKEN *Token;
- EFI_HTTP_TOKEN *TokenInItem;
- Token = (EFI_HTTP_TOKEN *)Context;
- TokenInItem = (EFI_HTTP_TOKEN *)Item->Key;
- if ((Token == TokenInItem) || (Token->Event == TokenInItem->Event)) {
- return EFI_ACCESS_DENIED;
- }
- return EFI_SUCCESS;
- }
- /**
- Check whether the HTTP message associated with Tx4Token or Tx6Token is already sent out.
- @param[in] Map The container of Tx4Token or Tx6Token.
- @param[in] Item Current item to check against.
- @param[in] Context The Token to check against.
- @retval EFI_NOT_READY The HTTP message is still queued in the list.
- @retval EFI_SUCCESS The HTTP message has been sent out.
- **/
- EFI_STATUS
- EFIAPI
- HttpTcpNotReady (
- IN NET_MAP *Map,
- IN NET_MAP_ITEM *Item,
- IN VOID *Context
- )
- {
- HTTP_TOKEN_WRAP *ValueInItem;
- ValueInItem = (HTTP_TOKEN_WRAP *)Item->Value;
- if (!ValueInItem->TcpWrap.IsTxDone) {
- return EFI_NOT_READY;
- }
- return EFI_SUCCESS;
- }
- /**
- Transmit the HTTP or HTTPS message by processing the associated HTTP token.
- @param[in] Map The container of Tx4Token or Tx6Token.
- @param[in] Item Current item to check against.
- @param[in] Context The Token to check against.
- @retval EFI_OUT_OF_RESOURCES Failed to allocate resources.
- @retval EFI_SUCCESS The HTTP message is queued into TCP transmit
- queue.
- **/
- EFI_STATUS
- EFIAPI
- HttpTcpTransmit (
- IN NET_MAP *Map,
- IN NET_MAP_ITEM *Item,
- IN VOID *Context
- )
- {
- HTTP_TOKEN_WRAP *ValueInItem;
- EFI_STATUS Status;
- CHAR8 *RequestMsg;
- CHAR8 *Url;
- UINTN UrlSize;
- UINTN RequestMsgSize;
- RequestMsg = NULL;
- ValueInItem = (HTTP_TOKEN_WRAP *)Item->Value;
- if (ValueInItem->TcpWrap.IsTxDone) {
- return EFI_SUCCESS;
- }
- //
- // Parse the URI of the remote host.
- //
- UrlSize = StrLen (ValueInItem->HttpToken->Message->Data.Request->Url) + 1;
- Url = AllocatePool (UrlSize);
- if (Url == NULL) {
- return EFI_OUT_OF_RESOURCES;
- }
- UnicodeStrToAsciiStrS (ValueInItem->HttpToken->Message->Data.Request->Url, Url, UrlSize);
- //
- // Create request message.
- //
- Status = HttpGenRequestMessage (
- ValueInItem->HttpToken->Message,
- Url,
- &RequestMsg,
- &RequestMsgSize
- );
- FreePool (Url);
- if (EFI_ERROR (Status) || (NULL == RequestMsg)) {
- return Status;
- }
- ASSERT (RequestMsg != NULL);
- //
- // Transmit the request message.
- //
- Status = HttpTransmitTcp (
- ValueInItem->HttpInstance,
- ValueInItem,
- (UINT8 *)RequestMsg,
- RequestMsgSize
- );
- FreePool (RequestMsg);
- return Status;
- }
- /**
- Receive the HTTP response by processing the associated HTTP token.
- @param[in] Map The container of Rx4Token or Rx6Token.
- @param[in] Item Current item to check against.
- @param[in] Context The Token to check against.
- @retval EFI_SUCCESS The HTTP response is queued into TCP receive
- queue.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- EFIAPI
- HttpTcpReceive (
- IN NET_MAP *Map,
- IN NET_MAP_ITEM *Item,
- IN VOID *Context
- )
- {
- //
- // Process the queued HTTP response.
- //
- return HttpResponseWorker ((HTTP_TOKEN_WRAP *)Item->Value);
- }
- /**
- Receive the HTTP header by processing the associated HTTP token.
- @param[in] HttpInstance The HTTP instance private data.
- @param[in, out] SizeofHeaders The HTTP header length.
- @param[in, out] BufferSize The size of buffer to cache the header message.
- @param[in] Timeout The time to wait for receiving the header packet.
- @retval EFI_SUCCESS The HTTP header is received.
- @retval Others Other errors as indicated.
- **/
- EFI_STATUS
- HttpTcpReceiveHeader (
- IN HTTP_PROTOCOL *HttpInstance,
- IN OUT UINTN *SizeofHeaders,
- IN OUT UINTN *BufferSize,
- IN EFI_EVENT Timeout
- )
- {
- EFI_STATUS Status;
- EFI_TCP4_IO_TOKEN *Rx4Token;
- EFI_TCP4_PROTOCOL *Tcp4;
- EFI_TCP6_IO_TOKEN *Rx6Token;
- EFI_TCP6_PROTOCOL *Tcp6;
- CHAR8 **EndofHeader;
- CHAR8 **HttpHeaders;
- CHAR8 *Buffer;
- NET_FRAGMENT Fragment;
- ASSERT (HttpInstance != NULL);
- EndofHeader = HttpInstance->EndofHeader;
- HttpHeaders = HttpInstance->HttpHeaders;
- Tcp4 = HttpInstance->Tcp4;
- Tcp6 = HttpInstance->Tcp6;
- Buffer = NULL;
- Rx4Token = NULL;
- Rx6Token = NULL;
- Fragment.Len = 0;
- Fragment.Bulk = NULL;
- if (HttpInstance->LocalAddressIsIPv6) {
- ASSERT (Tcp6 != NULL);
- } else {
- ASSERT (Tcp4 != NULL);
- }
- if (!HttpInstance->UseHttps) {
- Status = HttpCreateTcpRxEventForHeader (HttpInstance);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- }
- if (!HttpInstance->LocalAddressIsIPv6) {
- if (!HttpInstance->UseHttps) {
- Rx4Token = &HttpInstance->Rx4Token;
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = AllocateZeroPool (DEF_BUF_LEN);
- if (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- return Status;
- }
- }
- //
- // Receive the HTTP headers only when EFI_HTTP_RESPONSE_DATA is not NULL.
- //
- while (*EndofHeader == NULL) {
- if (!HttpInstance->UseHttps) {
- HttpInstance->IsRxDone = FALSE;
- Rx4Token->Packet.RxData->DataLength = DEF_BUF_LEN;
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentLength = DEF_BUF_LEN;
- Status = Tcp4->Receive (Tcp4, Rx4Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp4 receive failed: %r\n", Status));
- return Status;
- }
- while (!HttpInstance->IsRxDone && ((Timeout == NULL) || EFI_ERROR (gBS->CheckEvent (Timeout)))) {
- Tcp4->Poll (Tcp4);
- }
- if (!HttpInstance->IsRxDone) {
- //
- // Cancel the Token before close its Event.
- //
- Tcp4->Cancel (HttpInstance->Tcp4, &Rx4Token->CompletionToken);
- gBS->CloseEvent (Rx4Token->CompletionToken.Event);
- Rx4Token->CompletionToken.Status = EFI_TIMEOUT;
- }
- Status = Rx4Token->CompletionToken.Status;
- if (EFI_ERROR (Status)) {
- return Status;
- }
- Fragment.Len = Rx4Token->Packet.RxData->FragmentTable[0].FragmentLength;
- Fragment.Bulk = (UINT8 *)Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer;
- } else {
- if (Fragment.Bulk != NULL) {
- FreePool (Fragment.Bulk);
- Fragment.Bulk = NULL;
- }
- Status = HttpsReceive (HttpInstance, &Fragment, Timeout);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp4 receive failed: %r\n", Status));
- return Status;
- }
- }
- //
- // Append the response string along with a Null-terminator.
- //
- *BufferSize = *SizeofHeaders + Fragment.Len;
- Buffer = AllocatePool (*BufferSize + 1);
- if (Buffer == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- return Status;
- }
- if (*HttpHeaders != NULL) {
- CopyMem (Buffer, *HttpHeaders, *SizeofHeaders);
- FreePool (*HttpHeaders);
- }
- CopyMem (
- Buffer + *SizeofHeaders,
- Fragment.Bulk,
- Fragment.Len
- );
- *(Buffer + *BufferSize) = '\0';
- *HttpHeaders = Buffer;
- *SizeofHeaders = *BufferSize;
- //
- // Check whether we received end of HTTP headers.
- //
- *EndofHeader = AsciiStrStr (*HttpHeaders, HTTP_END_OF_HDR_STR);
- }
- //
- // Free the buffer.
- //
- if ((Rx4Token != NULL) && (Rx4Token->Packet.RxData != NULL) && (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL)) {
- FreePool (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
- Fragment.Bulk = NULL;
- }
- if (Fragment.Bulk != NULL) {
- FreePool (Fragment.Bulk);
- Fragment.Bulk = NULL;
- }
- } else {
- if (!HttpInstance->UseHttps) {
- Rx6Token = &HttpInstance->Rx6Token;
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = AllocateZeroPool (DEF_BUF_LEN);
- if (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- return Status;
- }
- }
- //
- // Receive the HTTP headers only when EFI_HTTP_RESPONSE_DATA is not NULL.
- //
- while (*EndofHeader == NULL) {
- if (!HttpInstance->UseHttps) {
- HttpInstance->IsRxDone = FALSE;
- Rx6Token->Packet.RxData->DataLength = DEF_BUF_LEN;
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentLength = DEF_BUF_LEN;
- Status = Tcp6->Receive (Tcp6, Rx6Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp6 receive failed: %r\n", Status));
- return Status;
- }
- while (!HttpInstance->IsRxDone && ((Timeout == NULL) || EFI_ERROR (gBS->CheckEvent (Timeout)))) {
- Tcp6->Poll (Tcp6);
- }
- if (!HttpInstance->IsRxDone) {
- //
- // Cancel the Token before close its Event.
- //
- Tcp6->Cancel (HttpInstance->Tcp6, &Rx6Token->CompletionToken);
- gBS->CloseEvent (Rx6Token->CompletionToken.Event);
- Rx6Token->CompletionToken.Status = EFI_TIMEOUT;
- }
- Status = Rx6Token->CompletionToken.Status;
- if (EFI_ERROR (Status)) {
- return Status;
- }
- Fragment.Len = Rx6Token->Packet.RxData->FragmentTable[0].FragmentLength;
- Fragment.Bulk = (UINT8 *)Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer;
- } else {
- if (Fragment.Bulk != NULL) {
- FreePool (Fragment.Bulk);
- Fragment.Bulk = NULL;
- }
- Status = HttpsReceive (HttpInstance, &Fragment, Timeout);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp6 receive failed: %r\n", Status));
- return Status;
- }
- }
- //
- // Append the response string along with a Null-terminator.
- //
- *BufferSize = *SizeofHeaders + Fragment.Len;
- Buffer = AllocatePool (*BufferSize + 1);
- if (Buffer == NULL) {
- Status = EFI_OUT_OF_RESOURCES;
- return Status;
- }
- if (*HttpHeaders != NULL) {
- CopyMem (Buffer, *HttpHeaders, *SizeofHeaders);
- FreePool (*HttpHeaders);
- }
- CopyMem (
- Buffer + *SizeofHeaders,
- Fragment.Bulk,
- Fragment.Len
- );
- *(Buffer + *BufferSize) = '\0';
- *HttpHeaders = Buffer;
- *SizeofHeaders = *BufferSize;
- //
- // Check whether we received end of HTTP headers.
- //
- *EndofHeader = AsciiStrStr (*HttpHeaders, HTTP_END_OF_HDR_STR);
- }
- //
- // Free the buffer.
- //
- if ((Rx6Token != NULL) && (Rx6Token->Packet.RxData != NULL) && (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL)) {
- FreePool (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
- Fragment.Bulk = NULL;
- }
- if (Fragment.Bulk != NULL) {
- FreePool (Fragment.Bulk);
- Fragment.Bulk = NULL;
- }
- }
- //
- // Skip the CRLF after the HTTP headers.
- //
- *EndofHeader = *EndofHeader + AsciiStrLen (HTTP_END_OF_HDR_STR);
- *SizeofHeaders = *EndofHeader - *HttpHeaders;
- return EFI_SUCCESS;
- }
- /**
- Receive the HTTP body by processing the associated HTTP token.
- @param[in] Wrap The HTTP token's wrap data.
- @param[in] HttpMsg The HTTP message data.
- @retval EFI_SUCCESS The HTTP body is received.
- @retval Others Other error as indicated.
- **/
- EFI_STATUS
- HttpTcpReceiveBody (
- IN HTTP_TOKEN_WRAP *Wrap,
- IN EFI_HTTP_MESSAGE *HttpMsg
- )
- {
- EFI_STATUS Status;
- HTTP_PROTOCOL *HttpInstance;
- EFI_TCP6_PROTOCOL *Tcp6;
- EFI_TCP6_IO_TOKEN *Rx6Token;
- EFI_TCP4_PROTOCOL *Tcp4;
- EFI_TCP4_IO_TOKEN *Rx4Token;
- HttpInstance = Wrap->HttpInstance;
- Tcp4 = HttpInstance->Tcp4;
- Tcp6 = HttpInstance->Tcp6;
- Rx4Token = NULL;
- Rx6Token = NULL;
- if (HttpInstance->LocalAddressIsIPv6) {
- ASSERT (Tcp6 != NULL);
- } else {
- ASSERT (Tcp4 != NULL);
- }
- if (HttpInstance->LocalAddressIsIPv6) {
- Rx6Token = &Wrap->TcpWrap.Rx6Token;
- Rx6Token->Packet.RxData->DataLength = (UINT32)MIN (MAX_UINT32, HttpMsg->BodyLength);
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentLength = (UINT32)MIN (MAX_UINT32, HttpMsg->BodyLength);
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = (VOID *)HttpMsg->Body;
- Rx6Token->CompletionToken.Status = EFI_NOT_READY;
- Status = Tcp6->Receive (Tcp6, Rx6Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp6 receive failed: %r\n", Status));
- return Status;
- }
- } else {
- Rx4Token = &Wrap->TcpWrap.Rx4Token;
- Rx4Token->Packet.RxData->DataLength = (UINT32)MIN (MAX_UINT32, HttpMsg->BodyLength);
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentLength = (UINT32)MIN (MAX_UINT32, HttpMsg->BodyLength);
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = (VOID *)HttpMsg->Body;
- Rx4Token->CompletionToken.Status = EFI_NOT_READY;
- Status = Tcp4->Receive (Tcp4, Rx4Token);
- if (EFI_ERROR (Status)) {
- DEBUG ((DEBUG_ERROR, "Tcp4 receive failed: %r\n", Status));
- return Status;
- }
- }
- return EFI_SUCCESS;
- }
- /**
- Clean up Tcp Tokens while the Tcp transmission error occurs.
- @param[in] Wrap Pointer to HTTP token's wrap data.
- **/
- VOID
- HttpTcpTokenCleanup (
- IN HTTP_TOKEN_WRAP *Wrap
- )
- {
- HTTP_PROTOCOL *HttpInstance;
- EFI_TCP4_IO_TOKEN *Rx4Token;
- EFI_TCP6_IO_TOKEN *Rx6Token;
- ASSERT (Wrap != NULL);
- HttpInstance = Wrap->HttpInstance;
- Rx4Token = NULL;
- Rx6Token = NULL;
- if (HttpInstance->LocalAddressIsIPv6) {
- Rx6Token = &Wrap->TcpWrap.Rx6Token;
- if (Rx6Token->CompletionToken.Event != NULL) {
- gBS->CloseEvent (Rx6Token->CompletionToken.Event);
- Rx6Token->CompletionToken.Event = NULL;
- }
- FreePool (Wrap);
- Rx6Token = &HttpInstance->Rx6Token;
- if (Rx6Token->CompletionToken.Event != NULL) {
- gBS->CloseEvent (Rx6Token->CompletionToken.Event);
- Rx6Token->CompletionToken.Event = NULL;
- }
- if (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
- FreePool (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
- Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
- }
- } else {
- Rx4Token = &Wrap->TcpWrap.Rx4Token;
- if (Rx4Token->CompletionToken.Event != NULL) {
- gBS->CloseEvent (Rx4Token->CompletionToken.Event);
- Rx4Token->CompletionToken.Event = NULL;
- }
- FreePool (Wrap);
- Rx4Token = &HttpInstance->Rx4Token;
- if (Rx4Token->CompletionToken.Event != NULL) {
- gBS->CloseEvent (Rx4Token->CompletionToken.Event);
- Rx4Token->CompletionToken.Event = NULL;
- }
- if (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
- FreePool (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
- Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
- }
- }
- }
- /**
- Send Events via EDKII_HTTP_CALLBACK_PROTOCOL.
- @param[in] Event The event that occurs in the current state.
- @param[in] EventStatus The Status of Event, EFI_SUCCESS or other errors.
- **/
- VOID
- HttpNotify (
- IN EDKII_HTTP_CALLBACK_EVENT Event,
- IN EFI_STATUS EventStatus
- )
- {
- EFI_STATUS Status;
- EFI_HANDLE *Handles;
- UINTN Index;
- UINTN HandleCount;
- EFI_HANDLE Handle;
- EDKII_HTTP_CALLBACK_PROTOCOL *HttpCallback;
- DEBUG ((DEBUG_INFO, "HttpNotify: Event - %d, EventStatus - %r\n", Event, EventStatus));
- Handles = NULL;
- HandleCount = 0;
- Status = gBS->LocateHandleBuffer (
- ByProtocol,
- &gEdkiiHttpCallbackProtocolGuid,
- NULL,
- &HandleCount,
- &Handles
- );
- if (Status == EFI_SUCCESS) {
- for (Index = 0; Index < HandleCount; Index++) {
- Handle = Handles[Index];
- Status = gBS->HandleProtocol (
- Handle,
- &gEdkiiHttpCallbackProtocolGuid,
- (VOID **)&HttpCallback
- );
- if (Status == EFI_SUCCESS) {
- DEBUG ((DEBUG_INFO, "HttpNotify: Notifying %p\n", HttpCallback));
- HttpCallback->Callback (
- HttpCallback,
- Event,
- EventStatus
- );
- }
- }
- FreePool (Handles);
- }
- }
|