NetworkPkg: Restore TPL Before Return

This patch fixes a few instances of error cases in NetworkPkg
returning after a RaiseTPL call without restoring the TPL
first.

Signed-off-by: Oliver Smith-Denny <[email protected]>

Author: os-d

NetworkPkg/Ip6Dxe/Ip6Impl.c

@@ -61,7 +61,6 @@ EfiIp6GetModeData (
     return EFI_INVALID_PARAMETER;
   }
 
-  OldTpl     = gBS->RaiseTPL (TPL_CALLBACK);
   IpInstance = IP6_INSTANCE_FROM_PROTOCOL (This);
   IpSb       = IpInstance->Service;
   IpIf       = IpInstance->Interface;
@@ -70,6 +69,8 @@ EfiIp6GetModeData (
     return EFI_INVALID_PARAMETER;
   }
 
+  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
+
   if (Ip6ModeData != NULL) {
     //
     // IsStarted is "whether the EfiIp6Configure has been called".

NetworkPkg/SnpDxe/Get_status.c

@@ -216,12 +216,12 @@ SnpUndi32GetStatus (
 
   Snp = EFI_SIMPLE_NETWORK_DEV_FROM_THIS (This);
 
-  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
-
   if (Snp == NULL) {
     return EFI_DEVICE_ERROR;
   }
 
+  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
+
   switch (Snp->Mode.State) {
     case EfiSimpleNetworkInitialized:
       break;

NetworkPkg/SnpDxe/Transmit.c

@@ -287,12 +287,12 @@ SnpUndi32Transmit (
 
   Snp = EFI_SIMPLE_NETWORK_DEV_FROM_THIS (This);
 
-  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
-
   if (Snp == NULL) {
     return EFI_DEVICE_ERROR;
   }
 
+  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
+
   switch (Snp->Mode.State) {
     case EfiSimpleNetworkInitialized:
       break;