Fix an issue in DataHubGetNextRecord.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8458 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 13 insertions, 10 deletions

diff --git a/IntelFrameworkModulePkg/Universal/DataHubDxe/DataHub.c b/IntelFrameworkModulePkg/Universal/DataHubDxe/DataHub.c
index b58de9d..a2e991a 100644
--- a/IntelFrameworkModulePkg/Universal/DataHubDxe/DataHub.c
+++ b/IntelFrameworkModulePkg/Universal/DataHubDxe/DataHub.c
@@ -243,19 +243,22 @@ DataHubGetNextRecord (
       // Use the MTC from the Filter Driver.

       //

       FilterMonotonicCount = FilterDriver->GetNextMonotonicCount;

+       

+      //

+      // The GetNextMonotonicCount field remembers the last value from the previous time.

+      // But we already processed this vaule, so we need to find the next one.

+      //

+      *Record = GetNextDataRecord (&Private->DataListHead, ClassFilter, &FilterMonotonicCount);

       if (FilterMonotonicCount != 0) {

+        *MonotonicCount = FilterMonotonicCount;

+      }

+      

+      if ((FilterDriver->GetNextMonotonicCount != 0) && (FilterMonotonicCount == 0)) {

         //

-        // The GetNextMonotonicCount field remembers the last value from the previous time.

-        // But we already processed this vaule, so we need to find the next one.

+        // If there is no new record to get exit now.

         //

-        *Record         = GetNextDataRecord (&Private->DataListHead, ClassFilter, &FilterMonotonicCount);

-        *MonotonicCount = FilterMonotonicCount;

-        if (FilterMonotonicCount == 0) {

-          //

-          // If there is no new record to get exit now.

-          //

-          return EFI_NOT_FOUND;

-        }

+        *MonotonicCount = 0;

+        return EFI_NOT_FOUND;

       }

     }

   }