aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Library/UefiShellUfmCommandLib/dir.c64
-rw-r--r--Library/UefiShellUfmCommandLib/dir.h30
2 files changed, 94 insertions, 0 deletions
diff --git a/Library/UefiShellUfmCommandLib/dir.c b/Library/UefiShellUfmCommandLib/dir.c
new file mode 100644
index 0000000..12bc13b
--- /dev/null
+++ b/Library/UefiShellUfmCommandLib/dir.c
@@ -0,0 +1,64 @@
+#include <Library/UefiLib.h>
+#include <Library/DebugLib.h>
+#include <Library/ShellLib.h>
+#include <Library/BaseLib.h>
+#include <Library/MemoryAllocationLib.h>
+
+#include "dir.h"
+
+struct dir_list *dirl_alloc(CHAR16 *search_path, CONST UINT64 attr)
+{
+ EFI_STATUS status = EFI_SUCCESS;
+ EFI_SHELL_FILE_INFO *list_head = NULL, *node = NULL;
+ UINTN length = 0;
+ struct dir_list *dl;
+
+ status = ShellOpenFileMetaArg(search_path, EFI_FILE_MODE_READ, &list_head);
+ if(EFI_ERROR(status))
+ return NULL;
+
+ for(node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&list_head->Link);
+ !IsNull(&list_head->Link, &node->Link);
+ node = (EFI_SHELL_FILE_INFO *)GetNextNode(&list_head->Link, &node->Link))
+ {
+ ASSERT(node != NULL);
+ ASSERT(node->Info != NULL);
+ ASSERT((node->Info->Attribute & EFI_FILE_VALID_ATTR) == node->Info->Attribute);
+
+ if((node->Info->Attribute & attr) != attr ||
+ StrCmp(node->FileName, L".") == 0) {
+ EFI_SHELL_FILE_INFO *tmp;
+ tmp = (EFI_SHELL_FILE_INFO *)GetPreviousNode(&list_head->Link, &node->Link);
+
+ // I have stolen code below in the source code of InternalFreeShellFileInfoNode():
+ RemoveEntryList(&node->Link);
+ FreePool((VOID*)node->Info);
+ FreePool((VOID*)node->FileName);
+ FreePool((VOID*)node->FullName);
+ gEfiShellProtocol->CloseFile(node->Handle);
+ FreePool(node);
+ node = tmp;
+ continue;
+ }
+
+ length++;
+ }
+
+ dl = AllocatePool(sizeof(struct dir_list));
+ if(!dl) {
+ ShellCloseFileMetaArg(&list_head);
+ return NULL;
+ }
+
+ dl->list_head = list_head;
+ dl->len = length;
+ return dl;
+}
+
+VOID dirl_release(struct dir_list *dl)
+{
+ if(dl->list_head)
+ ShellCloseFileMetaArg(&dl->list_head);
+
+ FreePool(dl);
+}
diff --git a/Library/UefiShellUfmCommandLib/dir.h b/Library/UefiShellUfmCommandLib/dir.h
new file mode 100644
index 0000000..be560ad
--- /dev/null
+++ b/Library/UefiShellUfmCommandLib/dir.h
@@ -0,0 +1,30 @@
+#ifndef UFM_DIR_H
+#define UFM_DIR_H
+
+#include <Uefi.h>
+
+struct dir_list {
+ EFI_SHELL_FILE_INFO *list_head;
+ int len; // number of elements in list
+};
+
+/*
+ * Opens a directory and gets all its entries
+ *
+ * search_path: the pointer to path string
+ * attr: required file attributes
+ *
+ * return: A pointer to the allocated structure or NULL if allocation fails
+*/
+struct dir_list *dirl_alloc(CHAR16 *search_path, CONST UINT64 attr);
+
+/*
+ * Deletes the directory list, frees the structure
+ *
+ * dl: the pointer to the directory list
+ *
+ * return: VOID
+*/
+VOID dirl_release(struct dir_list *dl);
+
+#endif /* UFM_DIR_H */