X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FUniversal%2FNetwork%2FMtftp4Dxe%2FMtftp4Impl.c;h=3ce879253751426fc25531595a08e076c33112f5;hp=214ac44745c7843c57197ebddbfcb52f658bc5d7;hb=61066b8d7475929f9f453e1ebea4f72d0c020096;hpb=36ee91ca3661d3d020a7841aacbf858d885c4728;ds=sidebyside

diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
index 214ac44745..3ce8792537 100644
--- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
+++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
@@ -260,11 +260,13 @@ Mtftp4GetInfoCheckPacket (
   IN EFI_MTFTP4_PACKET      *Packet
   )
 {
+  MTFTP4_PROTOCOL           *Instance;
   MTFTP4_GETINFO_STATE      *State;
   EFI_STATUS                Status;
   UINT16                    OpCode;
 
-  State   = (MTFTP4_GETINFO_STATE *) Token->Context;
+  Instance = MTFTP4_PROTOCOL_FROM_THIS (This);
+  State    = &Instance->GetInfoState;
   OpCode   = NTOHS (Packet->OpCode);
 
   //
@@ -335,7 +337,8 @@ EfiMtftp4GetInfo (
   )
 {
   EFI_MTFTP4_TOKEN          Token;
-  MTFTP4_GETINFO_STATE      State;
+  MTFTP4_PROTOCOL           *Instance;
+  MTFTP4_GETINFO_STATE      *State;
   EFI_STATUS                Status;
 
   if ((This == NULL) || (Filename == NULL) || (PacketLength == NULL) ||
@@ -348,9 +351,11 @@ EfiMtftp4GetInfo (
   }
 
   *PacketLength         = 0;
-  State.Packet          = Packet;
-  State.PacketLen       = PacketLength;
-  State.Status          = EFI_SUCCESS;
+  Instance              = MTFTP4_PROTOCOL_FROM_THIS (This);
+  State                 = &Instance->GetInfoState;
+  State->Packet          = Packet;
+  State->PacketLen       = PacketLength;
+  State->Status          = EFI_SUCCESS;
 
   //
   // Fill in the Token to issue an synchronous ReadFile operation
@@ -364,7 +369,6 @@ EfiMtftp4GetInfo (
   Token.OptionList      = OptionList;
   Token.BufferSize      = 0;
   Token.Buffer          = NULL;
-  Token.Context         = &State;
   Token.CheckPacket     = Mtftp4GetInfoCheckPacket;
   Token.TimeoutCallback = NULL;
   Token.PacketNeeded    = NULL;
@@ -372,7 +376,7 @@ EfiMtftp4GetInfo (
   Status                = EfiMtftp4ReadFile (This, &Token);
 
   if (EFI_ABORTED == Status) {
-    return State.Status;
+    return State->Status;
   }
 
   return Status;