aboutsummaryrefslogtreecommitdiffstats
path: root/Library/UefiShellUfmCommandLib/fs.c
blob: fe2d5909b857dfc67ec83a14f8c149acd133672c (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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#include <Library/UefiLib.h>
#include <Library/DebugLib.h>
#include <Library/ShellLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/DevicePathLib.h>
#include <Library/ShellCommandLib.h>

#include "fs.h"

STATIC BOOLEAN search_list(CONST CHAR16 *list, CONST CHAR16 *target,
	OUT CHAR16 **var OPTIONAL, CONST BOOLEAN meta, CONST CHAR16 *delimiter)
{
	CHAR16 *temp_list, *temp_str;
	CONST CHAR16 *list_walker;
	BOOLEAN result;

	for(list_walker = list, temp_list = NULL;
		list_walker != NULL && *list_walker != CHAR_NULL; )
	{
		temp_list = StrnCatGrow(&temp_list, NULL, list_walker, 0);
		ASSERT(temp_list != NULL);

		temp_str = StrStr(temp_list, delimiter);
		if(temp_str != NULL)
			*temp_str = CHAR_NULL;

		list_walker = StrStr(list_walker, delimiter);
		while(list_walker != NULL && *list_walker == *delimiter)
			list_walker++;

		if(meta)
			result = gUnicodeCollation->MetaiMatch(gUnicodeCollation, temp_list, (CHAR16 *)target);
		else
			result = (BOOLEAN)(StrCmp(temp_list, target) == 0);

		if(result) {
			if(var != NULL)
				*var = temp_list;
			else
				FreePool(temp_list);
			return TRUE;
		}
		FreePool(temp_list);
		temp_list = NULL;
	}

	return FALSE;
}

struct fs_array *fsa_alloc(EFI_HANDLE *handles, UINTN count)
{
	UINTN i;
	struct fs_array *fsa;
	EFI_DEVICE_PATH_PROTOCOL *dev_path;
	CONST CHAR16 *map_list;

	fsa = AllocatePool(sizeof(struct fs_array));
	if(!fsa)
		return NULL;

	fsa->full_name = AllocateZeroPool(count * sizeof(CHAR16 *));
	if(!fsa->full_name) {
		fsa_release(fsa);
		return NULL;
	}

	fsa->len = count;
	for(i = 0; i < count; i++) {
		dev_path = DevicePathFromHandle(handles[i]);
		map_list = gEfiShellProtocol->GetMapFromDevicePath(&dev_path);
		search_list(map_list, L"FS*:", &fsa->full_name[i], TRUE, L";");
		ASSERT(fsa->full_name[i] != NULL);
	}

	return fsa;
}

VOID fsa_release(struct fs_array *fsa)
{
	UINTN i;
	if(fsa->full_name) {
		for(i = 0; i < fsa->len; i++) {
			if(fsa->full_name[i]) // safety measure and our good sleep
				FreePool(fsa->full_name[i]);
		}
		FreePool(fsa->full_name);
	}
	FreePool(fsa);
}