blob: 12bc13b6fed5bf8e91234fae21c14bec09db4a5d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
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);
}
|
