..

1 files changed, 84 insertions, 71 deletions

diff --git a/memtestEDK/Memtest/ProtocolInformation/Logger.c b/memtestEDK/Memtest/ProtocolInformation/Logger.c
index 8f448c5..d41f7e5 100644
--- a/memtestEDK/Memtest/ProtocolInformation/Logger.c
+++ b/memtestEDK/Memtest/ProtocolInformation/Logger.c
@@ -32,6 +32,9 @@ EFI_OPEN_PROTOCOL_INFORMATION_ENTRY *OpenInfo;
 UINTN OpenInfoIndex;
 UINTN OpenInfoCount;
 
+EFI_FILE_PROTOCOL* root = NULL;
+EFI_FILE_SYSTEM_INFO* fileSystemInfo = NULL;
+
 void printFileSystemInfo(EFI_FILE_SYSTEM_INFO* fileSystemInfo) {
 
   Print(L"fileSystemInfo pointer     %p\n", fileSystemInfo);
@@ -75,6 +78,79 @@ BOOLEAN compareGuids(EFI_GUID* sfspGuid, EFI_GUID* guid) {
   return FALSE;
 }
 
+EFI_STATUS writeToFile() {
+  
+        EFI_FILE_PROTOCOL* token = NULL;      
+        efiStatus = root->Open(
+          root, 
+          &token,
+          L"memtest_log",
+          EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
+          0);
+        checkStatus(efiStatus, "Open");
+
+        UINT64 currPos;
+        efiStatus = token->GetPosition(token, &currPos);
+        checkStatus(efiStatus, "GetPosition");
+        Print(L"Current File Position: %d\n", currPos);
+
+        UINTN fileSize; // TODO get the actual size
+        CHAR16* fileBuffer = NULL;
+
+        efiStatus = token->Read(token, &fileSize, fileBuffer);
+        checkStatus(efiStatus, "Read");
+
+        Print(L"FILE SIZE: %d\n", fileSize);
+
+        fileSize += 80;
+        efiStatus = bs->AllocatePool(EfiLoaderData, (UINTN) (fileSize) , (void **)&fileBuffer);
+        checkStatus(efiStatus, "AllocatePool for file");
+
+        Print(L"FILE SIZE2: %d\n", fileSize);
+
+        efiStatus = token->Read(token, &fileSize, fileBuffer);
+        checkStatus(efiStatus, "Read");
+        printFileContent(fileSize, fileBuffer);
+
+        efiStatus = token->GetPosition(token, &currPos);
+        checkStatus(efiStatus, "GetPosition");
+        Print(L"Current File Position: %d\n", currPos);
+
+        char message[] = "\nAdding a new line";
+        UINTN length = 19;
+        UINTN sizeBytes = sizeof(message);
+        Print(L"size in bytes: %d\n", sizeBytes);
+        Print(L"Lengh: %d\n", length);
+
+        if (fileSize >=1) {
+          token->SetPosition(token, fileSize);
+        }
+
+        efiStatus = token->Write(token, &length, message);
+        checkStatus(efiStatus, "Write");
+
+        fileSize += length;
+
+        token->SetPosition(token, 0);
+
+        efiStatus = token->Read(token, &fileSize, fileBuffer);
+        printFileContent(fileSize, fileBuffer);
+
+        if (fileBuffer != NULL) {
+          bs->FreePool(fileBuffer);
+        }
+
+        if (fileSystemInfo != NULL) {
+          bs->FreePool(fileSystemInfo);
+        }
+
+        efiStatus = token->Close(token);
+        checkStatus(efiStatus, "Close");
+
+        return efiStatus;
+}
+
+
 EFI_STATUS
 EFIAPI
 UefiMain (
@@ -86,14 +162,13 @@ UefiMain (
   Console = SystemTable->ConOut;
 
   //
-  // Get all Handles
+  // Get all Handles containing the EFI_SIMPLE_FILE_SYSTEM_PROTOCOL_GUID
   //
-  efiStatus = bs->LocateHandleBuffer (
-    AllHandles,
-    NULL,
-    NULL,
-    &handleCount,
-    &handleBuffer
+  efiStatus = bs->LocateHandleBuffer(ByProtocol, 
+                                   &sfspGuid, 
+                                   NULL, 
+                                   &handleCount, 
+                                   &handleBuffer
   );
   checkStatus(efiStatus, "LocateHandleBuffer");
 
@@ -150,13 +225,11 @@ UefiMain (
         );
         checkStatus(efiStatus, "HandleProtocol");
 
-        EFI_FILE_PROTOCOL* root = NULL;
         efiStatus = fs->OpenVolume(fs, &root);
         checkStatus(efiStatus, "OpenVolume");
 
         EFI_GUID fsiGuid = EFI_FILE_SYSTEM_INFO_ID;  
         UINTN bufSize = 0;           
-        EFI_FILE_SYSTEM_INFO* fileSystemInfo = NULL;
         // First time it is expected to fail
         efiStatus = root->GetInfo(
           root,
@@ -176,73 +249,13 @@ UefiMain (
         //printFileSystemInfo(fileSystemInfo);
 
         // TODO first check if it is the right filesystem. If yes, leave the for loop
-
-        EFI_FILE_PROTOCOL* token = NULL;      
-        efiStatus = root->Open(
-          root, 
-          &token,
-          L"memtest_log",
-          EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
-          0);
-        checkStatus(efiStatus, "Open");
-
-        UINT64 currPos;
-        efiStatus = token->GetPosition(token, &currPos);
-        checkStatus(efiStatus, "GetPosition");
-        Print(L"Current File Position: %d\n", currPos);
-
-        UINTN fileSize = 500; // TODO get the actual size
-        CHAR16* fileBuffer = NULL;
-        efiStatus = bs->AllocatePool(EfiLoaderData, (UINTN) (fileSize) , (void **)&fileBuffer);
-        checkStatus(efiStatus, "AllocatePool for file");
-
-        efiStatus = token->Read(token, &fileSize, fileBuffer);
-        checkStatus(efiStatus, "Read");
-        printFileContent(fileSize, fileBuffer);
-
-        efiStatus = token->GetPosition(token, &currPos);
-        checkStatus(efiStatus, "GetPosition");
-        Print(L"Current File Position: %d\n", currPos);
-
-        char *message = "\nAdding a new line";
-        UINTN length = 19;
-        Print(L"Lengh: %d\n", length);
-
-        if (fileSize >=1) {
-          token->SetPosition(token, fileSize - 1);
-        }
-
-        efiStatus = token->Write(token, &length, message);
-        checkStatus(efiStatus, "Write");
-
-        fileSize += length;
-
-        token->SetPosition(token, 0);
-
-        efiStatus = token->Read(token, &fileSize, fileBuffer);
-        printFileContent(fileSize, fileBuffer);
-
-        if (fileBuffer != NULL) {
-          bs->FreePool(fileBuffer);
-        }
-
-        if (fileSystemInfo != NULL) {
-          bs->FreePool(fileSystemInfo);
-        }
+        break;
 
       } // end  if (compareGuids(&sfspGuid, guid))
     } // end  for (ProtocolIndex = 0; ProtocolIndex < ArrayCount; ProtocolIndex++)
 	} // end "for (handleIndex = 0; handleIndex < handleCount; handleIndex++)"
 
-
-  efiStatus = bs->LocateHandleBuffer(ByProtocol, 
-                                   &sfspGuid, 
-                                   NULL, 
-                                   &handleCount2, 
-                                   &handleBuffer2);
-
-  Print(L"Handle count2: %d\n", handleCount2);
-
+  efiStatus = writeToFile();
 
   return EFI_SUCCESS;
 }