loader.efi environment related cleanups
Since we have dedicated libefi/env.c file for variable support, the following changes are done: Simple cstyle changes in env.c Moved efi variable related commands from loader/main.c to libefi/env.c Did create function to set "efi-version" environment variable in env.c. This function does serve two purposes: for first a small clean up of the loader main(), and for second, it does replace the otherwise unused efi_variable_support hack. A bit of cleanup of ficl backend functions. The TEST_MAIN has no meaning, and removed few memory leaks. The forth code is updated to use "efi-version" variable, instead of ficl environment check. Reviewed by: imp Approved by: imp (mentor) Differential Revision: https://reviews.freebsd.org/D9165
This commit is contained in:
parent
4b1be206bb
commit
d124e3965a
Notes:
svn2git
2020-12-20 02:59:44 +00:00
svn path=/head/; revision=313042
@ -65,4 +65,12 @@ EFI_STATUS main(int argc, CHAR16 *argv[]);
|
||||
void exit(EFI_STATUS status);
|
||||
void delay(int usecs);
|
||||
|
||||
/* EFI environment initialization. */
|
||||
void efi_init_environment(void);
|
||||
|
||||
/* CHAR16 utility functions. */
|
||||
int wcscmp(CHAR16 *, CHAR16 *);
|
||||
void cpy8to16(const char *, CHAR16 *, size_t);
|
||||
void cpy16to8(const CHAR16 *, char *, size_t);
|
||||
|
||||
#endif /* _LOADER_EFILIB_H */
|
||||
|
@ -3,6 +3,7 @@
|
||||
.include <src.opts.mk>
|
||||
|
||||
.if ${MK_FORTH} != "no"
|
||||
CFLAGS+= -DBOOT_FORTH
|
||||
.include "${.CURDIR}/../../Makefile.ficl"
|
||||
.endif
|
||||
|
||||
@ -10,17 +11,14 @@ LIB= efi
|
||||
INTERNALLIB=
|
||||
WARNS?= 2
|
||||
|
||||
SRCS= delay.c devpath.c efi_console.c efinet.c efipart.c errno.c \
|
||||
handles.c libefi.c
|
||||
SRCS= delay.c devpath.c efi_console.c efinet.c efipart.c env.c errno.c \
|
||||
handles.c wchar.c libefi.c
|
||||
|
||||
.if ${MACHINE_CPUARCH} == "amd64" || ${MACHINE_CPUARCH} == "i386"
|
||||
SRCS+= time.c
|
||||
.elif ${MACHINE_CPUARCH} == "aarch64" || ${MACHINE_CPUARCH} == "arm"
|
||||
SRCS+= time_event.c
|
||||
.endif
|
||||
.if ${MK_FORTH} != "no"
|
||||
SRCS+= env.c
|
||||
.endif
|
||||
|
||||
# We implement a slightly non-standard %S in that it always takes a
|
||||
# CHAR16 that's common in UEFI-land instead of a wchar_t. This only
|
||||
|
@ -26,15 +26,17 @@
|
||||
#include <sys/cdefs.h>
|
||||
__FBSDID("$FreeBSD$");
|
||||
|
||||
#include <sys/param.h>
|
||||
#include <stand.h>
|
||||
#include <string.h>
|
||||
#include <efi.h>
|
||||
#include <efilib.h>
|
||||
#include <uuid.h>
|
||||
#include <stdbool.h>
|
||||
#include "bootstrap.h"
|
||||
#ifdef BOOT_FORTH
|
||||
#include "ficl.h"
|
||||
|
||||
int efi_variable_support = 1;
|
||||
#endif
|
||||
|
||||
/*
|
||||
* Simple wrappers to the underlying UEFI functions.
|
||||
@ -42,36 +44,348 @@ int efi_variable_support = 1;
|
||||
* for details.
|
||||
*/
|
||||
EFI_STATUS
|
||||
efi_get_next_variable_name(UINTN *variable_name_size, CHAR16 *variable_name, EFI_GUID *vendor_guid)
|
||||
efi_get_next_variable_name(UINTN *variable_name_size, CHAR16 *variable_name,
|
||||
EFI_GUID *vendor_guid)
|
||||
{
|
||||
return RS->GetNextVariableName(variable_name_size, variable_name, vendor_guid);
|
||||
return (RS->GetNextVariableName(variable_name_size, variable_name,
|
||||
vendor_guid));
|
||||
}
|
||||
|
||||
EFI_STATUS
|
||||
efi_get_variable(CHAR16 *variable_name, EFI_GUID *vendor_guid, UINT32 *attributes, UINTN *data_size,
|
||||
void *data)
|
||||
efi_get_variable(CHAR16 *variable_name, EFI_GUID *vendor_guid,
|
||||
UINT32 *attributes, UINTN *data_size, void *data)
|
||||
{
|
||||
return RS->GetVariable(variable_name, vendor_guid, attributes, data_size, data);
|
||||
return (RS->GetVariable(variable_name, vendor_guid, attributes,
|
||||
data_size, data));
|
||||
}
|
||||
|
||||
EFI_STATUS
|
||||
efi_set_variable(CHAR16 *variable_name, EFI_GUID *vendor_guid, UINT32 attributes, UINTN data_size,
|
||||
void *data)
|
||||
efi_set_variable(CHAR16 *variable_name, EFI_GUID *vendor_guid,
|
||||
UINT32 attributes, UINTN data_size, void *data)
|
||||
{
|
||||
return RS->SetVariable(variable_name, vendor_guid, attributes, data_size, data);
|
||||
return (RS->SetVariable(variable_name, vendor_guid, attributes,
|
||||
data_size, data));
|
||||
}
|
||||
|
||||
void
|
||||
efi_init_environment(void)
|
||||
{
|
||||
char var[128];
|
||||
|
||||
snprintf(var, sizeof(var), "%d.%02d", ST->Hdr.Revision >> 16,
|
||||
ST->Hdr.Revision & 0xffff);
|
||||
env_setenv("efi-version", EV_VOLATILE, var, env_noset, env_nounset);
|
||||
}
|
||||
|
||||
COMMAND_SET(efishow, "efi-show", "print some or all EFI variables", command_efi_show);
|
||||
|
||||
static int
|
||||
efi_print_var(CHAR16 *varnamearg, EFI_GUID *matchguid, int lflag)
|
||||
{
|
||||
UINTN datasz, i;
|
||||
EFI_STATUS status;
|
||||
UINT32 attr;
|
||||
CHAR16 *data;
|
||||
char *str;
|
||||
uint32_t uuid_status;
|
||||
int is_ascii;
|
||||
|
||||
datasz = 0;
|
||||
status = RS->GetVariable(varnamearg, matchguid, &attr,
|
||||
&datasz, NULL);
|
||||
if (status != EFI_BUFFER_TOO_SMALL) {
|
||||
printf("Can't get the variable: error %#lx\n", status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
data = malloc(datasz);
|
||||
status = RS->GetVariable(varnamearg, matchguid, &attr,
|
||||
&datasz, data);
|
||||
if (status != EFI_SUCCESS) {
|
||||
printf("Can't get the variable: error %#lx\n", status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid_to_string((uuid_t *)matchguid, &str, &uuid_status);
|
||||
if (lflag) {
|
||||
printf("%s 0x%x %S", str, attr, varnamearg);
|
||||
} else {
|
||||
printf("%s 0x%x %S=", str, attr, varnamearg);
|
||||
is_ascii = 1;
|
||||
free(str);
|
||||
str = (char *)data;
|
||||
for (i = 0; i < datasz - 1; i++) {
|
||||
/* Quick hack to see if this ascii-ish string printable range plus tab, cr and lf */
|
||||
if ((str[i] < 32 || str[i] > 126) && str[i] != 9 && str[i] != 10 && str[i] != 13) {
|
||||
is_ascii = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (str[datasz - 1] != '\0')
|
||||
is_ascii = 0;
|
||||
if (is_ascii)
|
||||
printf("%s", str);
|
||||
else {
|
||||
for (i = 0; i < datasz / 2; i++) {
|
||||
if (isalnum(data[i]) || isspace(data[i]))
|
||||
printf("%c", data[i]);
|
||||
else
|
||||
printf("\\x%02x", data[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
free(data);
|
||||
if (pager_output("\n"))
|
||||
return (CMD_WARN);
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
static int
|
||||
command_efi_show(int argc, char *argv[])
|
||||
{
|
||||
/*
|
||||
* efi-show [-a]
|
||||
* print all the env
|
||||
* efi-show -u UUID
|
||||
* print all the env vars tagged with UUID
|
||||
* efi-show -v var
|
||||
* search all the env vars and print the ones matching var
|
||||
* eif-show -u UUID -v var
|
||||
* eif-show UUID var
|
||||
* print all the env vars that match UUID and var
|
||||
*/
|
||||
/* NB: We assume EFI_GUID is the same as uuid_t */
|
||||
int aflag = 0, gflag = 0, lflag = 0, vflag = 0;
|
||||
int ch, rv;
|
||||
unsigned i;
|
||||
EFI_STATUS status;
|
||||
EFI_GUID varguid = { 0,0,0,{0,0,0,0,0,0,0,0} };
|
||||
EFI_GUID matchguid = { 0,0,0,{0,0,0,0,0,0,0,0} };
|
||||
uint32_t uuid_status;
|
||||
CHAR16 *varname;
|
||||
CHAR16 *newnm;
|
||||
CHAR16 varnamearg[128];
|
||||
UINTN varalloc;
|
||||
UINTN varsz;
|
||||
|
||||
while ((ch = getopt(argc, argv, "ag:lv:")) != -1) {
|
||||
switch (ch) {
|
||||
case 'a':
|
||||
aflag = 1;
|
||||
break;
|
||||
case 'g':
|
||||
gflag = 1;
|
||||
uuid_from_string(optarg, (uuid_t *)&matchguid,
|
||||
&uuid_status);
|
||||
if (uuid_status != uuid_s_ok) {
|
||||
printf("uid %s could not be parsed\n", optarg);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
break;
|
||||
case 'l':
|
||||
lflag = 1;
|
||||
break;
|
||||
case 'v':
|
||||
vflag = 1;
|
||||
if (strlen(optarg) >= nitems(varnamearg)) {
|
||||
printf("Variable %s is longer than %zd characters\n",
|
||||
optarg, nitems(varnamearg));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
for (i = 0; i < strlen(optarg); i++)
|
||||
varnamearg[i] = optarg[i];
|
||||
varnamearg[i] = 0;
|
||||
break;
|
||||
default:
|
||||
printf("Invalid argument %c\n", ch);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
}
|
||||
|
||||
if (aflag && (gflag || vflag)) {
|
||||
printf("-a isn't compatible with -v or -u\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
if (aflag && optind < argc) {
|
||||
printf("-a doesn't take any args\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
if (optind == argc)
|
||||
aflag = 1;
|
||||
|
||||
argc -= optind;
|
||||
argv += optind;
|
||||
|
||||
pager_open();
|
||||
if (vflag && gflag) {
|
||||
rv = efi_print_var(varnamearg, &matchguid, lflag);
|
||||
pager_close();
|
||||
return (rv);
|
||||
}
|
||||
|
||||
if (argc == 2) {
|
||||
optarg = argv[0];
|
||||
if (strlen(optarg) >= nitems(varnamearg)) {
|
||||
printf("Variable %s is longer than %zd characters\n",
|
||||
optarg, nitems(varnamearg));
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
for (i = 0; i < strlen(optarg); i++)
|
||||
varnamearg[i] = optarg[i];
|
||||
varnamearg[i] = 0;
|
||||
optarg = argv[1];
|
||||
uuid_from_string(optarg, (uuid_t *)&matchguid,
|
||||
&uuid_status);
|
||||
if (uuid_status != uuid_s_ok) {
|
||||
printf("uid %s could not be parsed\n", optarg);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
rv = efi_print_var(varnamearg, &matchguid, lflag);
|
||||
pager_close();
|
||||
return (rv);
|
||||
}
|
||||
|
||||
if (argc > 0) {
|
||||
printf("Too many args %d\n", argc);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
/*
|
||||
* Initiate the search -- note the standard takes pain
|
||||
* to specify the initial call must be a poiner to a NULL
|
||||
* character.
|
||||
*/
|
||||
varalloc = 1024;
|
||||
varname = malloc(varalloc);
|
||||
if (varname == NULL) {
|
||||
printf("Can't allocate memory to get variables\n");
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
varname[0] = 0;
|
||||
while (1) {
|
||||
varsz = varalloc;
|
||||
status = RS->GetNextVariableName(&varsz, varname, &varguid);
|
||||
if (status == EFI_BUFFER_TOO_SMALL) {
|
||||
varalloc = varsz;
|
||||
newnm = realloc(varname, varalloc);
|
||||
if (newnm == NULL) {
|
||||
printf("Can't allocate memory to get variables\n");
|
||||
free(varname);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
varname = newnm;
|
||||
continue; /* Try again with bigger buffer */
|
||||
}
|
||||
if (status != EFI_SUCCESS)
|
||||
break;
|
||||
if (aflag) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
if (vflag) {
|
||||
if (wcscmp(varnamearg, varname) == 0) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
if (gflag) {
|
||||
if (memcmp(&varguid, &matchguid, sizeof(varguid)) == 0) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
free(varname);
|
||||
pager_close();
|
||||
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
COMMAND_SET(efiset, "efi-set", "set EFI variables", command_efi_set);
|
||||
|
||||
static int
|
||||
command_efi_set(int argc, char *argv[])
|
||||
{
|
||||
char *uuid, *var, *val;
|
||||
CHAR16 wvar[128];
|
||||
EFI_GUID guid;
|
||||
uint32_t status;
|
||||
EFI_STATUS err;
|
||||
|
||||
if (argc != 4) {
|
||||
printf("efi-set uuid var new-value\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid = argv[1];
|
||||
var = argv[2];
|
||||
val = argv[3];
|
||||
uuid_from_string(uuid, (uuid_t *)&guid, &status);
|
||||
if (status != uuid_s_ok) {
|
||||
printf("Invalid uuid %s %d\n", uuid, status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
cpy8to16(var, wvar, sizeof(wvar));
|
||||
err = RS->SetVariable(wvar, &guid,
|
||||
EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_BOOTSERVICE_ACCESS,
|
||||
strlen(val) + 1, val);
|
||||
if (EFI_ERROR(err)) {
|
||||
printf("Failed to set variable: error %lu\n", EFI_ERROR_CODE(err));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
COMMAND_SET(efiunset, "efi-unset", "delete / unset EFI variables", command_efi_unset);
|
||||
|
||||
static int
|
||||
command_efi_unset(int argc, char *argv[])
|
||||
{
|
||||
char *uuid, *var;
|
||||
CHAR16 wvar[128];
|
||||
EFI_GUID guid;
|
||||
uint32_t status;
|
||||
EFI_STATUS err;
|
||||
|
||||
if (argc != 3) {
|
||||
printf("efi-unset uuid var\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid = argv[1];
|
||||
var = argv[2];
|
||||
uuid_from_string(uuid, (uuid_t *)&guid, &status);
|
||||
if (status != uuid_s_ok) {
|
||||
printf("Invalid uuid %s\n", uuid);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
cpy8to16(var, wvar, sizeof(wvar));
|
||||
err = RS->SetVariable(wvar, &guid, 0, 0, NULL);
|
||||
if (EFI_ERROR(err)) {
|
||||
printf("Failed to unset variable: error %lu\n", EFI_ERROR_CODE(err));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
#ifdef BOOT_FORTH
|
||||
/*
|
||||
* FreeBSD's loader interaction words and extras
|
||||
* FreeBSD's loader interaction words and extras
|
||||
*
|
||||
* efi-setenv ( value n name n guid n attr -- 0 | -1)
|
||||
* efi-getenv ( guid n addr n -- addr' n' | -1 )
|
||||
* efi-unsetenv ( name n guid n'' -- )
|
||||
* efi-setenv ( value n name n guid n attr -- 0 | -1)
|
||||
* efi-getenv ( guid n addr n -- addr' n' | -1 )
|
||||
* efi-unsetenv ( name n guid n'' -- )
|
||||
*/
|
||||
|
||||
/*
|
||||
* efi-setenv
|
||||
* efi-setenv ( value n name n guid n attr -- 0 | -1)
|
||||
* efi-setenv ( value n name n guid n attr -- 0 | -1)
|
||||
*
|
||||
* Set environment variables using the SetVariable EFI runtime service.
|
||||
*
|
||||
@ -87,19 +401,18 @@ efi_set_variable(CHAR16 *variable_name, EFI_GUID *vendor_guid, UINT32 attributes
|
||||
* 4 Run time access
|
||||
* (corresponding to the same bits in the UEFI spec).
|
||||
*/
|
||||
void
|
||||
static void
|
||||
ficlEfiSetenv(FICL_VM *pVM)
|
||||
{
|
||||
#ifndef TESTMAIN
|
||||
char *value = NULL, *guid = NULL;
|
||||
CHAR16 *name = NULL;
|
||||
int i;
|
||||
#endif
|
||||
char *namep, *valuep, *guidp;
|
||||
int names, values, guids, attr;
|
||||
int status;
|
||||
EFI_STATUS status;
|
||||
uuid_t u;
|
||||
uint32_t ustatus;
|
||||
bool error = true;
|
||||
|
||||
#if FICL_ROBUST > 1
|
||||
vmCheckStack(pVM, 6, 0);
|
||||
@ -112,10 +425,9 @@ ficlEfiSetenv(FICL_VM *pVM)
|
||||
values = stackPopINT(pVM->pStack);
|
||||
valuep = (char*)stackPopPtr(pVM->pStack);
|
||||
|
||||
#ifndef TESTMAIN
|
||||
guid = (char*)ficlMalloc(guids);
|
||||
if (guid == NULL)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
goto out;
|
||||
memcpy(guid, guidp, guids);
|
||||
uuid_from_string(guid, &u, &ustatus);
|
||||
if (ustatus != uuid_s_ok) {
|
||||
@ -123,16 +435,16 @@ ficlEfiSetenv(FICL_VM *pVM)
|
||||
goto out;
|
||||
}
|
||||
|
||||
name = (CHAR16 *)ficlMalloc((names + 1) * sizeof(CHAR16));
|
||||
name = ficlMalloc((names + 1) * sizeof(CHAR16));
|
||||
if (name == NULL)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
goto out;
|
||||
for (i = 0; i < names; i++)
|
||||
name[i] = namep[i];
|
||||
name[names] = (CHAR16)0;
|
||||
name[names] = 0;
|
||||
|
||||
value = (char*)ficlMalloc(values + 1);
|
||||
value = ficlMalloc(values + 1);
|
||||
if (value == NULL)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
goto out;
|
||||
memcpy(value, valuep, values);
|
||||
|
||||
status = efi_set_variable(name, (EFI_GUID *)&u, attr, values, value);
|
||||
@ -140,21 +452,20 @@ ficlEfiSetenv(FICL_VM *pVM)
|
||||
stackPushINT(pVM->pStack, 0);
|
||||
else
|
||||
stackPushINT(pVM->pStack, -1);
|
||||
error = false;
|
||||
out:
|
||||
ficlFree(name);
|
||||
ficlFree(value);
|
||||
ficlFree(guid);
|
||||
#endif
|
||||
|
||||
return;
|
||||
if (error == true)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
}
|
||||
|
||||
void
|
||||
static void
|
||||
ficlEfiGetenv(FICL_VM *pVM)
|
||||
{
|
||||
#ifndef TESTMAIN
|
||||
char *name, *value;
|
||||
#endif
|
||||
char *namep;
|
||||
int names;
|
||||
|
||||
@ -164,7 +475,6 @@ ficlEfiGetenv(FICL_VM *pVM)
|
||||
names = stackPopINT(pVM->pStack);
|
||||
namep = (char*) stackPopPtr(pVM->pStack);
|
||||
|
||||
#ifndef TESTMAIN
|
||||
name = (char*) ficlMalloc(names+1);
|
||||
if (name == NULL)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
@ -178,18 +488,13 @@ ficlEfiGetenv(FICL_VM *pVM)
|
||||
stackPushPtr(pVM->pStack, value);
|
||||
stackPushINT(pVM->pStack, strlen(value));
|
||||
} else
|
||||
#endif
|
||||
stackPushINT(pVM->pStack, -1);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
void
|
||||
static void
|
||||
ficlEfiUnsetenv(FICL_VM *pVM)
|
||||
{
|
||||
#ifndef TESTMAIN
|
||||
char *name;
|
||||
#endif
|
||||
char *namep;
|
||||
int names;
|
||||
|
||||
@ -199,7 +504,6 @@ ficlEfiUnsetenv(FICL_VM *pVM)
|
||||
names = stackPopINT(pVM->pStack);
|
||||
namep = (char*) stackPopPtr(pVM->pStack);
|
||||
|
||||
#ifndef TESTMAIN
|
||||
name = (char*) ficlMalloc(names+1);
|
||||
if (name == NULL)
|
||||
vmThrowErr(pVM, "Error: out of memory");
|
||||
@ -208,9 +512,6 @@ ficlEfiUnsetenv(FICL_VM *pVM)
|
||||
|
||||
unsetenv(name);
|
||||
ficlFree(name);
|
||||
#endif
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
/**************************************************************************
|
||||
@ -218,17 +519,14 @@ ficlEfiUnsetenv(FICL_VM *pVM)
|
||||
**************************************************************************/
|
||||
void ficlEfiCompilePlatform(FICL_SYSTEM *pSys)
|
||||
{
|
||||
FICL_DICT *dp = pSys->dp;
|
||||
assert (dp);
|
||||
FICL_DICT *dp = pSys->dp;
|
||||
assert (dp);
|
||||
|
||||
dictAppendWord(dp, "efi-setenv", ficlEfiSetenv, FW_DEFAULT);
|
||||
dictAppendWord(dp, "efi-getenv", ficlEfiGetenv, FW_DEFAULT);
|
||||
dictAppendWord(dp, "efi-unsetenv", ficlEfiUnsetenv, FW_DEFAULT);
|
||||
|
||||
/* Would like to export the EFI version, but this will do for now */
|
||||
ficlSetEnv(pSys, "efi-boot", 1);
|
||||
|
||||
return;
|
||||
dictAppendWord(dp, "efi-setenv", ficlEfiSetenv, FW_DEFAULT);
|
||||
dictAppendWord(dp, "efi-getenv", ficlEfiGetenv, FW_DEFAULT);
|
||||
dictAppendWord(dp, "efi-unsetenv", ficlEfiUnsetenv, FW_DEFAULT);
|
||||
}
|
||||
|
||||
FICL_COMPILE_SET(ficlEfiCompilePlatform);
|
||||
|
||||
#endif /* BOOT_FORTH */
|
||||
|
73
sys/boot/efi/libefi/wchar.c
Normal file
73
sys/boot/efi/libefi/wchar.c
Normal file
@ -0,0 +1,73 @@
|
||||
/*-
|
||||
* Copyright 2016 Netflix, Inc. All Rights Reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
||||
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||||
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||||
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||||
* SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <sys/cdefs.h>
|
||||
__FBSDID("$FreeBSD$");
|
||||
|
||||
#include <efi.h>
|
||||
#include <efilib.h>
|
||||
|
||||
/*
|
||||
* CHAR16 related functions moved from loader.
|
||||
* Perhaps we should move those to libstand afterall, but they are
|
||||
* needed only by UEFI.
|
||||
*/
|
||||
|
||||
int
|
||||
wcscmp(CHAR16 *a, CHAR16 *b)
|
||||
{
|
||||
|
||||
while (*a && *b && *a == *b) {
|
||||
a++;
|
||||
b++;
|
||||
}
|
||||
return *a - *b;
|
||||
}
|
||||
|
||||
/*
|
||||
* cpy8to16 copies a traditional C string into a CHAR16 string and
|
||||
* 0 terminates it. len is the size of *dst in bytes.
|
||||
*/
|
||||
void
|
||||
cpy8to16(const char *src, CHAR16 *dst, size_t len)
|
||||
{
|
||||
len <<= 1; /* Assume CHAR16 is 2 bytes */
|
||||
while (len > 0 && *src) {
|
||||
*dst++ = *src++;
|
||||
len--;
|
||||
}
|
||||
*dst++ = (CHAR16)0;
|
||||
}
|
||||
|
||||
void
|
||||
cpy16to8(const CHAR16 *src, char *dst, size_t len)
|
||||
{
|
||||
size_t i;
|
||||
|
||||
for (i = 0; i < len && src[i]; i++)
|
||||
dst[i] = (char)src[i];
|
||||
if (i < len)
|
||||
dst[i] = '\0';
|
||||
}
|
@ -52,22 +52,6 @@ __FBSDID("$FreeBSD$");
|
||||
|
||||
extern char bootprog_info[];
|
||||
|
||||
#ifdef BOOT_FORTH
|
||||
/*
|
||||
* Normally, efi.o from libefi.a would be brought in due to a function we call
|
||||
* there that's defined there. However, none of its functions are callable from
|
||||
* here since it just adds words to the FORTH environment or implement those
|
||||
* words. So, add a reference to a symbol in efi.o to force it to be be brought
|
||||
* in so the init function there gets added to the "compile" linker set happens
|
||||
* correctly.
|
||||
*
|
||||
* This assumes there's no global analysys that notices dummy1 isn't used
|
||||
* anywhere and tries to eliminate it.
|
||||
*/
|
||||
extern int efi_variable_support;
|
||||
int *dummy1 = &efi_variable_support;
|
||||
#endif
|
||||
|
||||
struct arch_switch archsw; /* MI/MD interface boundary */
|
||||
|
||||
EFI_GUID acpi = ACPI_TABLE_GUID;
|
||||
@ -88,32 +72,6 @@ EFI_GUID inputid = SIMPLE_TEXT_INPUT_PROTOCOL;
|
||||
static void efi_zfs_probe(void);
|
||||
#endif
|
||||
|
||||
/*
|
||||
* cpy8to16 copies a traditional C string into a CHAR16 string and
|
||||
* 0 terminates it. len is the size of *dst in bytes.
|
||||
*/
|
||||
static void
|
||||
cpy8to16(const char *src, CHAR16 *dst, size_t len)
|
||||
{
|
||||
len <<= 1; /* Assume CHAR16 is 2 bytes */
|
||||
while (len > 0 && *src) {
|
||||
*dst++ = *src++;
|
||||
len--;
|
||||
}
|
||||
*dst++ = (CHAR16)0;
|
||||
}
|
||||
|
||||
static void
|
||||
cpy16to8(const CHAR16 *src, char *dst, size_t len)
|
||||
{
|
||||
size_t i;
|
||||
|
||||
for (i = 0; i < len && src[i]; i++)
|
||||
dst[i] = (char)src[i];
|
||||
if (i < len)
|
||||
dst[i] = '\0';
|
||||
}
|
||||
|
||||
static int
|
||||
has_keyboard(void)
|
||||
{
|
||||
@ -455,9 +413,7 @@ main(int argc, CHAR16 *argv[])
|
||||
}
|
||||
}
|
||||
|
||||
snprintf(var, sizeof(var), "%d.%02d", ST->Hdr.Revision >> 16,
|
||||
ST->Hdr.Revision & 0xffff);
|
||||
env_setenv("efi-version", EV_VOLATILE, var, env_noset, env_nounset);
|
||||
efi_init_environment();
|
||||
setenv("LINES", "24", 1); /* optional */
|
||||
|
||||
for (k = 0; k < ST->NumberOfTableEntries; k++) {
|
||||
@ -476,19 +432,6 @@ main(int argc, CHAR16 *argv[])
|
||||
return (EFI_SUCCESS); /* keep compiler happy */
|
||||
}
|
||||
|
||||
/* XXX move to lib stand ? */
|
||||
static int
|
||||
wcscmp(CHAR16 *a, CHAR16 *b)
|
||||
{
|
||||
|
||||
while (*a && *b && *a == *b) {
|
||||
a++;
|
||||
b++;
|
||||
}
|
||||
return *a - *b;
|
||||
}
|
||||
|
||||
|
||||
COMMAND_SET(reboot, "reboot", "reboot the system", command_reboot);
|
||||
|
||||
static int
|
||||
@ -770,305 +713,6 @@ command_reloadbe(int argc, char *argv[])
|
||||
}
|
||||
#endif
|
||||
|
||||
COMMAND_SET(efishow, "efi-show", "print some or all EFI variables", command_efi_show);
|
||||
|
||||
static int
|
||||
efi_print_var(CHAR16 *varnamearg, EFI_GUID *matchguid, int lflag)
|
||||
{
|
||||
UINTN datasz, i;
|
||||
EFI_STATUS status;
|
||||
UINT32 attr;
|
||||
CHAR16 *data;
|
||||
char *str;
|
||||
uint32_t uuid_status;
|
||||
int is_ascii;
|
||||
|
||||
datasz = 0;
|
||||
status = RS->GetVariable(varnamearg, matchguid, &attr,
|
||||
&datasz, NULL);
|
||||
if (status != EFI_BUFFER_TOO_SMALL) {
|
||||
printf("Can't get the variable: error %#lx\n", status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
data = malloc(datasz);
|
||||
status = RS->GetVariable(varnamearg, matchguid, &attr,
|
||||
&datasz, data);
|
||||
if (status != EFI_SUCCESS) {
|
||||
printf("Can't get the variable: error %#lx\n", status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid_to_string((uuid_t *)matchguid, &str, &uuid_status);
|
||||
if (lflag) {
|
||||
printf("%s 0x%x %S", str, attr, varnamearg);
|
||||
} else {
|
||||
printf("%s 0x%x %S=", str, attr, varnamearg);
|
||||
is_ascii = 1;
|
||||
free(str);
|
||||
str = (char *)data;
|
||||
for (i = 0; i < datasz - 1; i++) {
|
||||
/* Quick hack to see if this ascii-ish string printable range plus tab, cr and lf */
|
||||
if ((str[i] < 32 || str[i] > 126) && str[i] != 9 && str[i] != 10 && str[i] != 13) {
|
||||
is_ascii = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (str[datasz - 1] != '\0')
|
||||
is_ascii = 0;
|
||||
if (is_ascii)
|
||||
printf("%s", str);
|
||||
else {
|
||||
for (i = 0; i < datasz / 2; i++) {
|
||||
if (isalnum(data[i]) || isspace(data[i]))
|
||||
printf("%c", data[i]);
|
||||
else
|
||||
printf("\\x%02x", data[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
free(data);
|
||||
if (pager_output("\n"))
|
||||
return (CMD_WARN);
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
static int
|
||||
command_efi_show(int argc, char *argv[])
|
||||
{
|
||||
/*
|
||||
* efi-show [-a]
|
||||
* print all the env
|
||||
* efi-show -u UUID
|
||||
* print all the env vars tagged with UUID
|
||||
* efi-show -v var
|
||||
* search all the env vars and print the ones matching var
|
||||
* eif-show -u UUID -v var
|
||||
* eif-show UUID var
|
||||
* print all the env vars that match UUID and var
|
||||
*/
|
||||
/* NB: We assume EFI_GUID is the same as uuid_t */
|
||||
int aflag = 0, gflag = 0, lflag = 0, vflag = 0;
|
||||
int ch, rv;
|
||||
unsigned i;
|
||||
EFI_STATUS status;
|
||||
EFI_GUID varguid = { 0,0,0,{0,0,0,0,0,0,0,0} };
|
||||
EFI_GUID matchguid = { 0,0,0,{0,0,0,0,0,0,0,0} };
|
||||
uint32_t uuid_status;
|
||||
CHAR16 *varname;
|
||||
CHAR16 *newnm;
|
||||
CHAR16 varnamearg[128];
|
||||
UINTN varalloc;
|
||||
UINTN varsz;
|
||||
|
||||
while ((ch = getopt(argc, argv, "ag:lv:")) != -1) {
|
||||
switch (ch) {
|
||||
case 'a':
|
||||
aflag = 1;
|
||||
break;
|
||||
case 'g':
|
||||
gflag = 1;
|
||||
uuid_from_string(optarg, (uuid_t *)&matchguid,
|
||||
&uuid_status);
|
||||
if (uuid_status != uuid_s_ok) {
|
||||
printf("uid %s could not be parsed\n", optarg);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
break;
|
||||
case 'l':
|
||||
lflag = 1;
|
||||
break;
|
||||
case 'v':
|
||||
vflag = 1;
|
||||
if (strlen(optarg) >= nitems(varnamearg)) {
|
||||
printf("Variable %s is longer than %zd characters\n",
|
||||
optarg, nitems(varnamearg));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
for (i = 0; i < strlen(optarg); i++)
|
||||
varnamearg[i] = optarg[i];
|
||||
varnamearg[i] = 0;
|
||||
break;
|
||||
default:
|
||||
printf("Invalid argument %c\n", ch);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
}
|
||||
|
||||
if (aflag && (gflag || vflag)) {
|
||||
printf("-a isn't compatible with -v or -u\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
if (aflag && optind < argc) {
|
||||
printf("-a doesn't take any args");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
if (optind == argc)
|
||||
aflag = 1;
|
||||
|
||||
argc -= optind;
|
||||
argv += optind;
|
||||
|
||||
pager_open();
|
||||
if (vflag && gflag) {
|
||||
rv = efi_print_var(varnamearg, &matchguid, lflag);
|
||||
pager_close();
|
||||
return (rv);
|
||||
}
|
||||
|
||||
if (argc == 2) {
|
||||
optarg = argv[0];
|
||||
if (strlen(optarg) >= nitems(varnamearg)) {
|
||||
printf("Variable %s is longer than %zd characters\n",
|
||||
optarg, nitems(varnamearg));
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
for (i = 0; i < strlen(optarg); i++)
|
||||
varnamearg[i] = optarg[i];
|
||||
varnamearg[i] = 0;
|
||||
optarg = argv[1];
|
||||
uuid_from_string(optarg, (uuid_t *)&matchguid,
|
||||
&uuid_status);
|
||||
if (uuid_status != uuid_s_ok) {
|
||||
printf("uid %s could not be parsed\n", optarg);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
rv = efi_print_var(varnamearg, &matchguid, lflag);
|
||||
pager_close();
|
||||
return (rv);
|
||||
}
|
||||
|
||||
if (argc > 0) {
|
||||
printf("Too many args %d\n", argc);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
|
||||
/*
|
||||
* Initiate the search -- note the standard takes pain
|
||||
* to specify the initial call must be a poiner to a NULL
|
||||
* character.
|
||||
*/
|
||||
varalloc = 1024;
|
||||
varname = malloc(varalloc);
|
||||
if (varname == NULL) {
|
||||
printf("Can't allocate memory to get variables\n");
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
varname[0] = 0;
|
||||
while (1) {
|
||||
varsz = varalloc;
|
||||
status = RS->GetNextVariableName(&varsz, varname, &varguid);
|
||||
if (status == EFI_BUFFER_TOO_SMALL) {
|
||||
varalloc = varsz;
|
||||
newnm = malloc(varalloc);
|
||||
if (newnm == NULL) {
|
||||
printf("Can't allocate memory to get variables\n");
|
||||
free(varname);
|
||||
pager_close();
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
memcpy(newnm, varname, varsz);
|
||||
free(varname);
|
||||
varname = newnm;
|
||||
continue; /* Try again with bigger buffer */
|
||||
}
|
||||
if (status != EFI_SUCCESS)
|
||||
break;
|
||||
if (aflag) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
if (vflag) {
|
||||
if (wcscmp(varnamearg, varname) == 0) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
if (gflag) {
|
||||
if (memcmp(&varguid, &matchguid, sizeof(varguid)) == 0) {
|
||||
if (efi_print_var(varname, &varguid, lflag) != CMD_OK)
|
||||
break;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
free(varname);
|
||||
pager_close();
|
||||
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
COMMAND_SET(efiset, "efi-set", "set EFI variables", command_efi_set);
|
||||
|
||||
static int
|
||||
command_efi_set(int argc, char *argv[])
|
||||
{
|
||||
char *uuid, *var, *val;
|
||||
CHAR16 wvar[128];
|
||||
EFI_GUID guid;
|
||||
uint32_t status;
|
||||
EFI_STATUS err;
|
||||
|
||||
if (argc != 4) {
|
||||
printf("efi-set uuid var new-value\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid = argv[1];
|
||||
var = argv[2];
|
||||
val = argv[3];
|
||||
uuid_from_string(uuid, (uuid_t *)&guid, &status);
|
||||
if (status != uuid_s_ok) {
|
||||
printf("Invalid uuid %s %d\n", uuid, status);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
cpy8to16(var, wvar, sizeof(wvar));
|
||||
err = RS->SetVariable(wvar, &guid,
|
||||
EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_BOOTSERVICE_ACCESS,
|
||||
strlen(val) + 1, val);
|
||||
if (EFI_ERROR(err)) {
|
||||
printf("Failed to set variable: error %lu\n", EFI_ERROR_CODE(err));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
COMMAND_SET(efiunset, "efi-unset", "delete / unset EFI variables", command_efi_unset);
|
||||
|
||||
static int
|
||||
command_efi_unset(int argc, char *argv[])
|
||||
{
|
||||
char *uuid, *var;
|
||||
CHAR16 wvar[128];
|
||||
EFI_GUID guid;
|
||||
uint32_t status;
|
||||
EFI_STATUS err;
|
||||
|
||||
if (argc != 3) {
|
||||
printf("efi-unset uuid var\n");
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
uuid = argv[1];
|
||||
var = argv[2];
|
||||
uuid_from_string(uuid, (uuid_t *)&guid, &status);
|
||||
if (status != uuid_s_ok) {
|
||||
printf("Invalid uuid %s\n", uuid);
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
cpy8to16(var, wvar, sizeof(wvar));
|
||||
err = RS->SetVariable(wvar, &guid, 0, 0, NULL);
|
||||
if (EFI_ERROR(err)) {
|
||||
printf("Failed to unset variable: error %lu\n", EFI_ERROR_CODE(err));
|
||||
return (CMD_ERROR);
|
||||
}
|
||||
return (CMD_OK);
|
||||
}
|
||||
|
||||
#ifdef LOADER_FDT_SUPPORT
|
||||
extern int command_fdt_internal(int argc, char *argv[]);
|
||||
|
||||
|
@ -46,9 +46,9 @@ include /boot/support.4th
|
||||
include /boot/color.4th
|
||||
include /boot/delay.4th
|
||||
include /boot/check-password.4th
|
||||
s" efi-boot" environment? [if] [if]
|
||||
s" efi-version" getenv? [if]
|
||||
include /boot/efi.4th
|
||||
[then] [then]
|
||||
[then]
|
||||
|
||||
only forth definitions
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user