X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=MdeModulePkg%2FUniversal%2FNetwork%2FMtftp4Dxe%2FMtftp4Impl.c;h=4c459b22b88f40a3e54bacc6c243f1498b3a1861;hb=72f01d4b4a73f55bbabee8eba386f106891541f5;hp=531bd74042a14ecf632ff94fa7435821e851c886;hpb=3dc3861a4e3d19f45db22c90ef0fbdc0493daeb0;p=mirror_edk2.git

diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
index 531bd74042..4c459b22b8 100644
--- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
+++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c
@@ -1,7 +1,7 @@
 /** @file
   Interface routine for Mtftp4.
   
-Copyright (c) 2006 - 2007, Intel Corporation<BR>
+Copyright (c) 2006 - 2009, Intel Corporation<BR>
 All rights reserved. This program and the accompanying materials
 are licensed and made available under the terms and conditions of the BSD License
 which accompanies this distribution.  The full text of the license may be found at
@@ -103,6 +103,7 @@ Mtftp4CleanOperation (
 
 **/
 EFI_STATUS
+EFIAPI
 Mtftp4GetInfoCheckPacket (
   IN EFI_MTFTP4_PROTOCOL    *This,
   IN EFI_MTFTP4_TOKEN       *Token,
@@ -110,13 +111,11 @@ Mtftp4GetInfoCheckPacket (
   IN EFI_MTFTP4_PACKET      *Packet
   )
 {
-  MTFTP4_PROTOCOL           *Instance;
   MTFTP4_GETINFO_STATE      *State;
   EFI_STATUS                Status;
   UINT16                    OpCode;
 
-  Instance = MTFTP4_PROTOCOL_FROM_THIS (This);
-  State    = &Instance->GetInfoState;
+  State   = (MTFTP4_GETINFO_STATE *) Token->Context;
   OpCode   = NTOHS (Packet->OpCode);
 
   //
@@ -983,8 +982,7 @@ EfiMtftp4GetInfo (
   )
 {
   EFI_MTFTP4_TOKEN          Token;
-  MTFTP4_PROTOCOL           *Instance;
-  MTFTP4_GETINFO_STATE      *State;
+  MTFTP4_GETINFO_STATE      State;
   EFI_STATUS                Status;
 
   if ((This == NULL) || (Filename == NULL) || (PacketLength == NULL) ||
@@ -997,11 +995,9 @@ EfiMtftp4GetInfo (
   }
 
   *PacketLength         = 0;
-  Instance              = MTFTP4_PROTOCOL_FROM_THIS (This);
-  State                 = &Instance->GetInfoState;
-  State->Packet         = Packet;
-  State->PacketLen      = PacketLength;
-  State->Status         = EFI_SUCCESS;
+  State.Packet          = Packet;
+  State.PacketLen       = PacketLength;
+  State.Status          = EFI_SUCCESS;
 
   //
   // Fill in the Token to issue an synchronous ReadFile operation
@@ -1015,6 +1011,7 @@ EfiMtftp4GetInfo (
   Token.OptionList      = OptionList;
   Token.BufferSize      = 0;
   Token.Buffer          = NULL;
+  Token.Context         = &State;
   Token.CheckPacket     = Mtftp4GetInfoCheckPacket;
   Token.TimeoutCallback = NULL;
   Token.PacketNeeded    = NULL;
@@ -1022,7 +1019,7 @@ EfiMtftp4GetInfo (
   Status                = EfiMtftp4ReadFile (This, &Token);
 
   if (EFI_ABORTED == Status) {
-    return State->Status;
+    return State.Status;
   }
 
   return Status;