1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
9 UINT64
ticks_read(void) {
11 __asm__
volatile ("rdtsc" : "=a" (a
), "=d" (d
));
14 #elif defined(__i386__)
15 UINT64
ticks_read(void) {
17 __asm__
volatile ("rdtsc" : "=A" (val
));
20 #elif defined(__aarch64__)
21 UINT64
ticks_read(void) {
23 __asm__
volatile ("mrs %0, cntpct_el0" : "=r" (val
));
27 UINT64
ticks_read(void) {
33 #if defined(__aarch64__)
34 UINT64
ticks_freq(void) {
36 __asm__
volatile ("mrs %0, cntfrq_el0": "=r" (freq
));
40 /* count TSC ticks during a millisecond delay */
41 UINT64
ticks_freq(void) {
42 UINT64 ticks_start
, ticks_end
;
44 ticks_start
= ticks_read();
46 ticks_end
= ticks_read();
48 return (ticks_end
- ticks_start
) * 1000UL;
52 UINT64
time_usec(void) {
66 return 1000UL * 1000UL * ticks
/ freq
;
69 EFI_STATUS
parse_boolean(const CHAR8
*v
, BOOLEAN
*b
) {
73 return EFI_INVALID_PARAMETER
;
75 if (strcmpa(v
, (CHAR8
*)"1") == 0 ||
76 strcmpa(v
, (CHAR8
*)"yes") == 0 ||
77 strcmpa(v
, (CHAR8
*)"y") == 0 ||
78 strcmpa(v
, (CHAR8
*)"true") == 0 ||
79 strcmpa(v
, (CHAR8
*)"t") == 0 ||
80 strcmpa(v
, (CHAR8
*)"on") == 0) {
85 if (strcmpa(v
, (CHAR8
*)"0") == 0 ||
86 strcmpa(v
, (CHAR8
*)"no") == 0 ||
87 strcmpa(v
, (CHAR8
*)"n") == 0 ||
88 strcmpa(v
, (CHAR8
*)"false") == 0 ||
89 strcmpa(v
, (CHAR8
*)"f") == 0 ||
90 strcmpa(v
, (CHAR8
*)"off") == 0) {
95 return EFI_INVALID_PARAMETER
;
98 EFI_STATUS
efivar_set_raw(const EFI_GUID
*vendor
, const CHAR16
*name
, const void *buf
, UINTN size
, UINT32 flags
) {
101 assert(buf
|| size
== 0);
103 flags
|= EFI_VARIABLE_BOOTSERVICE_ACCESS
| EFI_VARIABLE_RUNTIME_ACCESS
;
104 return RT
->SetVariable((CHAR16
*) name
, (EFI_GUID
*) vendor
, flags
, size
, (void *) buf
);
107 EFI_STATUS
efivar_set(const EFI_GUID
*vendor
, const CHAR16
*name
, const CHAR16
*value
, UINT32 flags
) {
111 return efivar_set_raw(vendor
, name
, value
, value
? StrSize(value
) : 0, flags
);
114 EFI_STATUS
efivar_set_uint_string(const EFI_GUID
*vendor
, const CHAR16
*name
, UINTN i
, UINT32 flags
) {
120 SPrint(str
, ELEMENTSOF(str
), L
"%u", i
);
121 return efivar_set(vendor
, name
, str
, flags
);
124 EFI_STATUS
efivar_set_uint32_le(const EFI_GUID
*vendor
, const CHAR16
*name
, UINT32 value
, UINT32 flags
) {
130 buf
[0] = (UINT8
)(value
>> 0U & 0xFF);
131 buf
[1] = (UINT8
)(value
>> 8U & 0xFF);
132 buf
[2] = (UINT8
)(value
>> 16U & 0xFF);
133 buf
[3] = (UINT8
)(value
>> 24U & 0xFF);
135 return efivar_set_raw(vendor
, name
, buf
, sizeof(buf
), flags
);
138 EFI_STATUS
efivar_set_uint64_le(const EFI_GUID
*vendor
, const CHAR16
*name
, UINT64 value
, UINT32 flags
) {
144 buf
[0] = (UINT8
)(value
>> 0U & 0xFF);
145 buf
[1] = (UINT8
)(value
>> 8U & 0xFF);
146 buf
[2] = (UINT8
)(value
>> 16U & 0xFF);
147 buf
[3] = (UINT8
)(value
>> 24U & 0xFF);
148 buf
[4] = (UINT8
)(value
>> 32U & 0xFF);
149 buf
[5] = (UINT8
)(value
>> 40U & 0xFF);
150 buf
[6] = (UINT8
)(value
>> 48U & 0xFF);
151 buf
[7] = (UINT8
)(value
>> 56U & 0xFF);
153 return efivar_set_raw(vendor
, name
, buf
, sizeof(buf
), flags
);
156 EFI_STATUS
efivar_get(const EFI_GUID
*vendor
, const CHAR16
*name
, CHAR16
**value
) {
157 _cleanup_freepool_ CHAR16
*buf
= NULL
;
165 err
= efivar_get_raw(vendor
, name
, (CHAR8
**)&buf
, &size
);
169 /* Make sure there are no incomplete characters in the buffer */
170 if ((size
% sizeof(CHAR16
)) != 0)
171 return EFI_INVALID_PARAMETER
;
176 /* Return buffer directly if it happens to be NUL terminated already */
177 if (size
>= sizeof(CHAR16
) && buf
[size
/sizeof(CHAR16
)] == 0) {
178 *value
= TAKE_PTR(buf
);
182 /* Make sure a terminating NUL is available at the end */
183 val
= xallocate_pool(size
+ sizeof(CHAR16
));
185 CopyMem(val
, buf
, size
);
186 val
[size
/ sizeof(CHAR16
)] = 0; /* NUL terminate */
192 EFI_STATUS
efivar_get_uint_string(const EFI_GUID
*vendor
, const CHAR16
*name
, UINTN
*i
) {
193 _cleanup_freepool_ CHAR16
*val
= NULL
;
200 err
= efivar_get(vendor
, name
, &val
);
207 EFI_STATUS
efivar_get_uint32_le(const EFI_GUID
*vendor
, const CHAR16
*name
, UINT32
*ret
) {
208 _cleanup_freepool_ CHAR8
*buf
= NULL
;
215 err
= efivar_get_raw(vendor
, name
, &buf
, &size
);
216 if (!EFI_ERROR(err
) && ret
) {
217 if (size
!= sizeof(UINT32
))
218 return EFI_BUFFER_TOO_SMALL
;
220 *ret
= (UINT32
) buf
[0] << 0U | (UINT32
) buf
[1] << 8U | (UINT32
) buf
[2] << 16U |
221 (UINT32
) buf
[3] << 24U;
227 EFI_STATUS
efivar_get_uint64_le(const EFI_GUID
*vendor
, const CHAR16
*name
, UINT64
*ret
) {
228 _cleanup_freepool_ CHAR8
*buf
= NULL
;
235 err
= efivar_get_raw(vendor
, name
, &buf
, &size
);
236 if (!EFI_ERROR(err
) && ret
) {
237 if (size
!= sizeof(UINT64
))
238 return EFI_BUFFER_TOO_SMALL
;
240 *ret
= (UINT64
) buf
[0] << 0U | (UINT64
) buf
[1] << 8U | (UINT64
) buf
[2] << 16U |
241 (UINT64
) buf
[3] << 24U | (UINT64
) buf
[4] << 32U | (UINT64
) buf
[5] << 40U |
242 (UINT64
) buf
[6] << 48U | (UINT64
) buf
[7] << 56U;
248 EFI_STATUS
efivar_get_raw(const EFI_GUID
*vendor
, const CHAR16
*name
, CHAR8
**buffer
, UINTN
*size
) {
249 _cleanup_freepool_ CHAR8
*buf
= NULL
;
256 l
= sizeof(CHAR16
*) * EFI_MAXIMUM_VARIABLE_SIZE
;
257 buf
= xallocate_pool(l
);
259 err
= RT
->GetVariable((CHAR16
*) name
, (EFI_GUID
*) vendor
, NULL
, &l
, buf
);
260 if (!EFI_ERROR(err
)) {
263 *buffer
= TAKE_PTR(buf
);
272 EFI_STATUS
efivar_get_boolean_u8(const EFI_GUID
*vendor
, const CHAR16
*name
, BOOLEAN
*ret
) {
273 _cleanup_freepool_ CHAR8
*b
= NULL
;
281 err
= efivar_get_raw(vendor
, name
, &b
, &size
);
288 void efivar_set_time_usec(const EFI_GUID
*vendor
, const CHAR16
*name
, UINT64 usec
) {
299 SPrint(str
, ELEMENTSOF(str
), L
"%ld", usec
);
300 efivar_set(vendor
, name
, str
, 0);
303 static INTN
utf8_to_16(const CHAR8
*stra
, CHAR16
*c
) {
310 if (!(stra
[0] & 0x80))
312 else if ((stra
[0] & 0xe0) == 0xc0)
314 else if ((stra
[0] & 0xf0) == 0xe0)
316 else if ((stra
[0] & 0xf8) == 0xf0)
318 else if ((stra
[0] & 0xfc) == 0xf8)
320 else if ((stra
[0] & 0xfe) == 0xfc)
330 unichar
= stra
[0] & 0x1f;
333 unichar
= stra
[0] & 0x0f;
336 unichar
= stra
[0] & 0x07;
339 unichar
= stra
[0] & 0x03;
342 unichar
= stra
[0] & 0x01;
346 for (UINTN i
= 1; i
< len
; i
++) {
347 if ((stra
[i
] & 0xc0) != 0x80)
350 unichar
|= stra
[i
] & 0x3f;
357 CHAR16
*xstra_to_str(const CHAR8
*stra
) {
366 str
= xnew(CHAR16
, len
+ 1);
373 utf8len
= utf8_to_16(stra
+ i
, str
+ strlen
);
375 /* invalid utf8 sequence, skip the garbage */
387 CHAR16
*xstra_to_path(const CHAR8
*stra
) {
396 str
= xnew(CHAR16
, len
+ 2);
404 utf8len
= utf8_to_16(stra
+ i
, str
+ strlen
);
406 /* invalid utf8 sequence, skip the garbage */
411 if (str
[strlen
] == '/')
413 if (str
[strlen
] == '\\' && str
[strlen
-1] == '\\') {
414 /* skip double slashes */
426 CHAR8
*strchra(const CHAR8
*s
, CHAR8 c
) {
438 EFI_STATUS
file_read(EFI_FILE_HANDLE dir
, const CHAR16
*name
, UINTN off
, UINTN size
, CHAR8
**ret
, UINTN
*ret_size
) {
439 _cleanup_(FileHandleClosep
) EFI_FILE_HANDLE handle
= NULL
;
440 _cleanup_freepool_ CHAR8
*buf
= NULL
;
446 err
= dir
->Open(dir
, &handle
, (CHAR16
*) name
, EFI_FILE_MODE_READ
, 0ULL);
451 _cleanup_freepool_ EFI_FILE_INFO
*info
= NULL
;
453 err
= get_file_info_harder(handle
, &info
, NULL
);
457 size
= info
->FileSize
+1;
461 err
= handle
->SetPosition(handle
, off
);
466 buf
= xallocate_pool(size
+ 1);
467 err
= handle
->Read(handle
, &size
, buf
);
473 *ret
= TAKE_PTR(buf
);
480 void log_error_stall(const CHAR16
*fmt
, ...) {
485 INT32 attr
= ST
->ConOut
->Mode
->Attribute
;
486 ST
->ConOut
->SetAttribute(ST
->ConOut
, EFI_LIGHTRED
|EFI_BACKGROUND_BLACK
);
488 if (ST
->ConOut
->Mode
->CursorColumn
> 0)
497 ST
->ConOut
->SetAttribute(ST
->ConOut
, attr
);
499 /* Give the user a chance to see the message. */
500 BS
->Stall(3 * 1000 * 1000);
503 EFI_STATUS
log_oom(void) {
504 log_error_stall(L
"Out of memory.");
505 return EFI_OUT_OF_RESOURCES
;
508 void *memmem_safe(const void *haystack
, UINTN haystack_len
, const void *needle
, UINTN needle_len
) {
509 assert(haystack
|| haystack_len
== 0);
510 assert(needle
|| needle_len
== 0);
513 return (void*)haystack
;
515 for (const CHAR8
*h
= haystack
, *n
= needle
; haystack_len
>= needle_len
; h
++, haystack_len
--)
516 if (*h
== *n
&& CompareMem(h
+ 1, n
+ 1, needle_len
- 1) == 0)
522 void print_at(UINTN x
, UINTN y
, UINTN attr
, const CHAR16
*str
) {
524 ST
->ConOut
->SetCursorPosition(ST
->ConOut
, x
, y
);
525 ST
->ConOut
->SetAttribute(ST
->ConOut
, attr
);
526 ST
->ConOut
->OutputString(ST
->ConOut
, (CHAR16
*)str
);
529 void clear_screen(UINTN attr
) {
530 ST
->ConOut
->SetAttribute(ST
->ConOut
, attr
);
531 ST
->ConOut
->ClearScreen(ST
->ConOut
);
534 void sort_pointer_array(
537 compare_pointer_func_t compare
) {
539 assert(array
|| n_members
== 0);
545 for (UINTN i
= 1; i
< n_members
; i
++) {
546 BOOLEAN more
= FALSE
;
548 for (UINTN k
= 0; k
< n_members
- i
; k
++) {
551 if (compare(array
[k
], array
[k
+1]) <= 0)
555 array
[k
] = array
[k
+1];
564 EFI_STATUS
get_file_info_harder(
565 EFI_FILE_HANDLE handle
,
569 UINTN size
= OFFSETOF(EFI_FILE_INFO
, FileName
) + 256;
570 _cleanup_freepool_ EFI_FILE_INFO
*fi
= NULL
;
576 /* A lot like LibFileInfo() but with useful error propagation */
578 fi
= xallocate_pool(size
);
579 err
= handle
->GetInfo(handle
, &GenericFileInfo
, &size
, fi
);
580 if (err
== EFI_BUFFER_TOO_SMALL
) {
582 fi
= xallocate_pool(size
); /* GetInfo tells us the required size, let's use that now */
583 err
= handle
->GetInfo(handle
, &GenericFileInfo
, &size
, fi
);
597 EFI_STATUS
readdir_harder(
598 EFI_FILE_HANDLE handle
,
599 EFI_FILE_INFO
**buffer
,
600 UINTN
*buffer_size
) {
609 /* buffer/buffer_size are both in and output parameters. Should be zero-initialized initially, and
610 * the specified buffer needs to be freed by caller, after final use. */
613 sz
= OFFSETOF(EFI_FILE_INFO
, FileName
) /* + 256 */;
614 *buffer
= xallocate_pool(sz
);
619 err
= handle
->Read(handle
, &sz
, *buffer
);
620 if (err
== EFI_BUFFER_TOO_SMALL
) {
622 *buffer
= xallocate_pool(sz
);
624 err
= handle
->Read(handle
, &sz
, *buffer
);
630 /* End of directory */
639 UINTN
strnlena(const CHAR8
*p
, UINTN maxlen
) {
645 for (c
= 0; c
< maxlen
; c
++)
652 CHAR8
*xstrndup8(const CHAR8
*p
, UINTN sz
) {
655 /* Following efilib's naming scheme this function would be called strndupa(), but we already have a
656 * function named like this in userspace, and it does something different there, hence to minimize
657 * confusion, let's pick a different name here */
659 assert(p
|| sz
== 0);
661 sz
= strnlena(p
, sz
);
663 n
= xallocate_pool(sz
+ 1);
672 BOOLEAN
is_ascii(const CHAR16
*f
) {
683 CHAR16
**strv_free(CHAR16
**v
) {
687 for (CHAR16
**i
= v
; *i
; i
++)
694 EFI_STATUS
open_directory(
695 EFI_FILE_HANDLE root
,
697 EFI_FILE_HANDLE
*ret
) {
699 _cleanup_(FileHandleClosep
) EFI_FILE_HANDLE dir
= NULL
;
700 _cleanup_freepool_ EFI_FILE_INFO
*file_info
= NULL
;
705 /* Opens a file, and then verifies it is actually a directory */
707 err
= root
->Open(root
, &dir
, (CHAR16
*) path
, EFI_FILE_MODE_READ
, 0ULL);
711 err
= get_file_info_harder(dir
, &file_info
, NULL
);
714 if (!FLAGS_SET(file_info
->Attribute
, EFI_FILE_DIRECTORY
))
715 return EFI_LOAD_ERROR
;
717 *ret
= TAKE_PTR(dir
);
721 UINT64
get_os_indications_supported(void) {
725 /* Returns the supported OS indications. If we can't acquire it, returns a zeroed out mask, i.e. no
726 * supported features. */
728 err
= efivar_get_uint64_le(EFI_GLOBAL_GUID
, L
"OsIndicationsSupported", &osind
);