efi.c revision 1.2.10.3 1 1.2.10.3 yamt /* $NetBSD: efi.c,v 1.2.10.3 2006/12/30 20:46:20 yamt Exp $ */
2 1.2.10.2 yamt
3 1.2.10.2 yamt /*-
4 1.2.10.2 yamt * Copyright (c) 2000 Doug Rabson
5 1.2.10.2 yamt * All rights reserved.
6 1.2.10.2 yamt *
7 1.2.10.2 yamt * Redistribution and use in source and binary forms, with or without
8 1.2.10.2 yamt * modification, are permitted provided that the following conditions
9 1.2.10.2 yamt * are met:
10 1.2.10.2 yamt * 1. Redistributions of source code must retain the above copyright
11 1.2.10.2 yamt * notice, this list of conditions and the following disclaimer.
12 1.2.10.2 yamt * 2. Redistributions in binary form must reproduce the above copyright
13 1.2.10.2 yamt * notice, this list of conditions and the following disclaimer in the
14 1.2.10.2 yamt * documentation and/or other materials provided with the distribution.
15 1.2.10.2 yamt *
16 1.2.10.2 yamt * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 1.2.10.2 yamt * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 1.2.10.2 yamt * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 1.2.10.2 yamt * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 1.2.10.2 yamt * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 1.2.10.2 yamt * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 1.2.10.2 yamt * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 1.2.10.2 yamt * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 1.2.10.2 yamt * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 1.2.10.2 yamt * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 1.2.10.2 yamt * SUCH DAMAGE.
27 1.2.10.2 yamt */
28 1.2.10.2 yamt
29 1.2.10.2 yamt #include <sys/cdefs.h>
30 1.2.10.3 yamt /* __FBSDID("$FreeBSD: src/sys/boot/efi/libefi/libefi.c,v 1.6 2003/04/03 21:36:29 obrien Exp $"); */
31 1.2.10.2 yamt
32 1.2.10.2 yamt #include <efi.h>
33 1.2.10.2 yamt #include <efilib.h>
34 1.2.10.2 yamt #include <lib/libsa/stand.h>
35 1.2.10.2 yamt
36 1.2.10.2 yamt EFI_HANDLE IH;
37 1.2.10.2 yamt EFI_SYSTEM_TABLE *ST;
38 1.2.10.2 yamt EFI_BOOT_SERVICES *BS;
39 1.2.10.2 yamt EFI_RUNTIME_SERVICES *RS;
40 1.2.10.2 yamt
41 1.2.10.2 yamt static EFI_PHYSICAL_ADDRESS heap;
42 1.2.10.2 yamt static UINTN heapsize;
43 1.2.10.2 yamt
44 1.2.10.2 yamt static CHAR16 *
45 1.2.10.2 yamt arg_skipsep(CHAR16 *argp)
46 1.2.10.2 yamt {
47 1.2.10.2 yamt
48 1.2.10.2 yamt while (*argp == ' ' || *argp == '\t')
49 1.2.10.2 yamt argp++;
50 1.2.10.2 yamt return (argp);
51 1.2.10.2 yamt }
52 1.2.10.2 yamt
53 1.2.10.2 yamt static CHAR16 *
54 1.2.10.2 yamt arg_skipword(CHAR16 *argp)
55 1.2.10.2 yamt {
56 1.2.10.2 yamt
57 1.2.10.2 yamt while (*argp && *argp != ' ' && *argp != '\t')
58 1.2.10.2 yamt argp++;
59 1.2.10.2 yamt return (argp);
60 1.2.10.2 yamt }
61 1.2.10.2 yamt
62 1.2.10.2 yamt void *
63 1.2.10.2 yamt efi_get_table(EFI_GUID *tbl)
64 1.2.10.2 yamt {
65 1.2.10.2 yamt EFI_GUID *id;
66 1.2.10.2 yamt int i;
67 1.2.10.2 yamt
68 1.2.10.2 yamt for (i = 0; i < ST->NumberOfTableEntries; i++) {
69 1.2.10.2 yamt id = &ST->ConfigurationTable[i].VendorGuid;
70 1.2.10.2 yamt if (!memcmp(id, tbl, sizeof(EFI_GUID)))
71 1.2.10.2 yamt return (ST->ConfigurationTable[i].VendorTable);
72 1.2.10.2 yamt }
73 1.2.10.2 yamt return (NULL);
74 1.2.10.2 yamt }
75 1.2.10.2 yamt
76 1.2.10.2 yamt void efi_exit(EFI_STATUS exit_code)
77 1.2.10.2 yamt {
78 1.2.10.2 yamt
79 1.2.10.2 yamt BS->FreePages(heap, EFI_SIZE_TO_PAGES(heapsize));
80 1.2.10.2 yamt BS->Exit(IH, exit_code, 0, NULL);
81 1.2.10.2 yamt }
82 1.2.10.2 yamt
83 1.2.10.2 yamt void
84 1.2.10.2 yamt efi_main(EFI_HANDLE image_handle, EFI_SYSTEM_TABLE *system_table)
85 1.2.10.2 yamt {
86 1.2.10.2 yamt static EFI_GUID image_protocol = LOADED_IMAGE_PROTOCOL;
87 1.2.10.2 yamt EFI_LOADED_IMAGE *img;
88 1.2.10.2 yamt CHAR16 *argp, *args, **argv;
89 1.2.10.2 yamt EFI_STATUS status;
90 1.2.10.2 yamt int argc, addprog;
91 1.2.10.2 yamt
92 1.2.10.2 yamt IH = image_handle;
93 1.2.10.2 yamt ST = system_table;
94 1.2.10.2 yamt BS = ST->BootServices;
95 1.2.10.2 yamt RS = ST->RuntimeServices;
96 1.2.10.2 yamt
97 1.2.10.2 yamt heapsize = 512*1024;
98 1.2.10.2 yamt status = BS->AllocatePages(AllocateAnyPages, EfiLoaderData,
99 1.2.10.2 yamt EFI_SIZE_TO_PAGES(heapsize), &heap);
100 1.2.10.2 yamt if (status != EFI_SUCCESS)
101 1.2.10.2 yamt BS->Exit(IH, status, 0, NULL);
102 1.2.10.2 yamt
103 1.2.10.2 yamt setheap((void *)heap, (void *)(heap + heapsize));
104 1.2.10.2 yamt
105 1.2.10.2 yamt /* Use efi_exit() from here on... */
106 1.2.10.2 yamt
107 1.2.10.2 yamt status = BS->HandleProtocol(IH, &image_protocol, (VOID**)&img);
108 1.2.10.2 yamt if (status != EFI_SUCCESS)
109 1.2.10.2 yamt efi_exit(status);
110 1.2.10.2 yamt
111 1.2.10.2 yamt /*
112 1.2.10.2 yamt * Pre-process the (optional) load options. If the option string
113 1.2.10.2 yamt * is given as an ASCII string, we use a poor man's ASCII to
114 1.2.10.2 yamt * Unicode-16 translation. The size of the option string as given
115 1.2.10.2 yamt * to us includes the terminating null character. We assume the
116 1.2.10.2 yamt * string is an ASCII string if strlen() plus the terminating
117 1.2.10.2 yamt * '\0' is less than LoadOptionsSize. Even if all Unicode-16
118 1.2.10.2 yamt * characters have the upper 8 bits non-zero, the terminating
119 1.2.10.2 yamt * null character will cause a one-off.
120 1.2.10.2 yamt * If the string is already in Unicode-16, we make a copy so that
121 1.2.10.2 yamt * we know we can always modify the string.
122 1.2.10.2 yamt */
123 1.2.10.2 yamt if (img->LoadOptionsSize > 0 && img->LoadOptions != NULL) {
124 1.2.10.2 yamt if (img->LoadOptionsSize == strlen(img->LoadOptions) + 1) {
125 1.2.10.2 yamt args = alloc(img->LoadOptionsSize << 1);
126 1.2.10.2 yamt for (argc = 0; argc < img->LoadOptionsSize; argc++)
127 1.2.10.2 yamt args[argc] = ((char*)img->LoadOptions)[argc];
128 1.2.10.2 yamt } else {
129 1.2.10.2 yamt args = alloc(img->LoadOptionsSize);
130 1.2.10.2 yamt memcpy(args, img->LoadOptions, img->LoadOptionsSize);
131 1.2.10.2 yamt }
132 1.2.10.2 yamt } else
133 1.2.10.2 yamt args = NULL;
134 1.2.10.2 yamt
135 1.2.10.2 yamt /*
136 1.2.10.2 yamt * Use a quick and dirty algorithm to build the argv vector. We
137 1.2.10.2 yamt * first count the number of words. Then, after allocating the
138 1.2.10.2 yamt * vector, we split the string up. We don't deal with quotes or
139 1.2.10.2 yamt * other more advanced shell features.
140 1.2.10.2 yamt * The EFI shell will pas the name of the image as the first
141 1.2.10.2 yamt * word in the argument list. This does not happen if we're
142 1.2.10.2 yamt * loaded by the boot manager. This is not so easy to figure
143 1.2.10.2 yamt * out though. The ParentHandle is not always NULL, because
144 1.2.10.2 yamt * there can be a function (=image) that will perform the task
145 1.2.10.2 yamt * for the boot manager.
146 1.2.10.2 yamt */
147 1.2.10.2 yamt /* Part 1: Figure out if we need to add our program name. */
148 1.2.10.2 yamt addprog = (args == NULL || img->ParentHandle == NULL ||
149 1.2.10.2 yamt img->FilePath == NULL) ? 1 : 0;
150 1.2.10.2 yamt if (!addprog) {
151 1.2.10.2 yamt addprog =
152 1.2.10.2 yamt (DevicePathType(img->FilePath) != MEDIA_DEVICE_PATH ||
153 1.2.10.2 yamt DevicePathSubType(img->FilePath) != MEDIA_FILEPATH_DP ||
154 1.2.10.2 yamt DevicePathNodeLength(img->FilePath) <=
155 1.2.10.2 yamt sizeof(FILEPATH_DEVICE_PATH)) ? 1 : 0;
156 1.2.10.2 yamt if (!addprog) {
157 1.2.10.2 yamt /* XXX todo. */
158 1.2.10.2 yamt }
159 1.2.10.2 yamt }
160 1.2.10.2 yamt /* Part 2: count words. */
161 1.2.10.2 yamt argc = (addprog) ? 1 : 0;
162 1.2.10.2 yamt argp = args;
163 1.2.10.2 yamt while (argp != NULL && *argp != 0) {
164 1.2.10.2 yamt argp = arg_skipsep(argp);
165 1.2.10.2 yamt if (*argp == 0)
166 1.2.10.2 yamt break;
167 1.2.10.2 yamt argc++;
168 1.2.10.2 yamt argp = arg_skipword(argp);
169 1.2.10.2 yamt }
170 1.2.10.2 yamt /* Part 3: build vector. */
171 1.2.10.2 yamt argv = alloc((argc + 1) * sizeof(CHAR16*));
172 1.2.10.2 yamt argc = 0;
173 1.2.10.2 yamt if (addprog)
174 1.2.10.2 yamt argv[argc++] = L"loader.efi";
175 1.2.10.2 yamt argp = args;
176 1.2.10.2 yamt while (argp != NULL && *argp != 0) {
177 1.2.10.2 yamt argp = arg_skipsep(argp);
178 1.2.10.2 yamt if (*argp == 0)
179 1.2.10.2 yamt break;
180 1.2.10.2 yamt argv[argc++] = argp;
181 1.2.10.2 yamt argp = arg_skipword(argp);
182 1.2.10.2 yamt /* Terminate the words. */
183 1.2.10.2 yamt if (*argp != 0)
184 1.2.10.2 yamt *argp++ = 0;
185 1.2.10.2 yamt }
186 1.2.10.2 yamt argv[argc] = NULL;
187 1.2.10.2 yamt
188 1.2.10.2 yamt status = main(argc, argv);
189 1.2.10.2 yamt efi_exit(status);
190 1.2.10.2 yamt }
191