Home | History | Annotate | Line # | Download | only in libefi
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