]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - stand/efi/libefi/efienv.c
MFC r343911, r344238-r344241, r344247, r344254-r344255, r344260, r344268,
[FreeBSD/FreeBSD.git] / stand / efi / libefi / efienv.c
1 /*-
2  * Copyright (c) 2018 Netflix, Inc.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29
30 #include <stand.h>
31 #include <efi.h>
32 #include <efichar.h>
33 #include <efilib.h>
34
35 static EFI_GUID FreeBSDBootVarGUID = FREEBSD_BOOT_VAR_GUID;
36 static EFI_GUID GlobalBootVarGUID = EFI_GLOBAL_VARIABLE;
37
38 EFI_STATUS
39 efi_getenv(EFI_GUID *g, const char *v, void *data, size_t *len)
40 {
41         size_t ul;
42         CHAR16 *uv;
43         UINT32 attr;
44         UINTN dl;
45         EFI_STATUS rv;
46
47         uv = NULL;
48         if (utf8_to_ucs2(v, &uv, &ul) != 0)
49                 return (EFI_OUT_OF_RESOURCES);
50         dl = *len;
51         rv = RS->GetVariable(uv, g, &attr, &dl, data);
52         if (rv == EFI_SUCCESS || rv == EFI_BUFFER_TOO_SMALL)
53                 *len = dl;
54         free(uv);
55         return (rv);
56 }
57
58 EFI_STATUS
59 efi_global_getenv(const char *v, void *data, size_t *len)
60 {
61
62         return (efi_getenv(&GlobalBootVarGUID, v, data, len));
63 }
64
65 EFI_STATUS
66 efi_freebsd_getenv(const char *v, void *data, size_t *len)
67 {
68
69         return (efi_getenv(&FreeBSDBootVarGUID, v, data, len));
70 }
71
72 EFI_STATUS
73 efi_setenv_freebsd_wcs(const char *varname, CHAR16 *valstr)
74 {
75         CHAR16 *var = NULL;
76         size_t len;
77         EFI_STATUS rv;
78
79         if (utf8_to_ucs2(varname, &var, &len) != 0)
80                 return (EFI_OUT_OF_RESOURCES);
81         rv = RS->SetVariable(var, &FreeBSDBootVarGUID,
82             EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,
83             (ucs2len(valstr) + 1) * sizeof(efi_char), valstr);
84         free(var);
85         return (rv);
86 }
87