ufs_module.c revision 295550
1/*- 2 * Copyright (c) 1998 Robert Nordier 3 * All rights reserved. 4 * Copyright (c) 2001 Robert Drehmel 5 * All rights reserved. 6 * Copyright (c) 2014 Nathan Whitehorn 7 * All rights reserved. 8 * Copyright (c) 2015 Eric McCorkle 9 * All rights reverved. 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 * 32 * $FreeBSD: stable/10/sys/boot/efi/boot1/ufs_module.c 295550 2016-02-11 22:33:47Z smh $ 33 */ 34 35#include <stdarg.h> 36#include <stdbool.h> 37#include <sys/cdefs.h> 38#include <sys/param.h> 39#include <efi.h> 40 41#include "boot_module.h" 42 43static dev_info_t *devinfo; 44static dev_info_t *devices; 45 46static int 47dskread(void *buf, u_int64_t lba, int nblk) 48{ 49 int size; 50 EFI_STATUS status; 51 52 lba = lba / (devinfo->dev->Media->BlockSize / DEV_BSIZE); 53 size = nblk * DEV_BSIZE; 54 55 status = devinfo->dev->ReadBlocks(devinfo->dev, 56 devinfo->dev->Media->MediaId, lba, size, buf); 57 58 if (status != EFI_SUCCESS) { 59 DPRINTF("dskread: failed dev: %p, id: %u, lba: %lu, size: %d, " 60 "status: %lu\n", devinfo->dev, 61 devinfo->dev->Media->MediaId, lba, size, 62 EFI_ERROR_CODE(status)); 63 return (-1); 64 } 65 66 return (0); 67} 68 69#include "ufsread.c" 70 71static struct dmadat __dmadat; 72 73static int 74init_dev(dev_info_t* dev) 75{ 76 77 devinfo = dev; 78 dmadat = &__dmadat; 79 80 return fsread(0, NULL, 0); 81} 82 83static EFI_STATUS 84probe(dev_info_t* dev) 85{ 86 87 if (init_dev(dev) < 0) 88 return (EFI_UNSUPPORTED); 89 90 add_device(&devices, dev); 91 92 return (EFI_SUCCESS); 93} 94 95static EFI_STATUS 96load(const char *filepath, dev_info_t *dev, void **bufp, size_t *bufsize) 97{ 98 ufs_ino_t ino; 99 EFI_STATUS status; 100 size_t size; 101 ssize_t read; 102 void *buf; 103 104 DPRINTF("Loading '%s' from %s\n", filepath, devpath_str(dev->devpath)); 105 106 if (init_dev(dev) < 0) { 107 DPRINTF("Failed to init device\n"); 108 return (EFI_UNSUPPORTED); 109 } 110 111 if ((ino = lookup(filepath)) == 0) { 112 DPRINTF("Failed to lookup '%s' (file not found?)\n", filepath); 113 return (EFI_NOT_FOUND); 114 } 115 116 if (fsread_size(ino, NULL, 0, &size) < 0 || size <= 0) { 117 printf("Failed to read size of '%s' ino: %d\n", filepath, ino); 118 return (EFI_INVALID_PARAMETER); 119 } 120 121 if ((status = bs->AllocatePool(EfiLoaderData, size, &buf)) != 122 EFI_SUCCESS) { 123 printf("Failed to allocate read buffer %zu for '%s' (%lu)\n", 124 size, filepath, EFI_ERROR_CODE(status)); 125 return (status); 126 } 127 128 read = fsread(ino, buf, size); 129 if ((size_t)read != size) { 130 printf("Failed to read '%s' (%zd != %zu)\n", filepath, read, 131 size); 132 (void)bs->FreePool(buf); 133 return (EFI_INVALID_PARAMETER); 134 } 135 136 DPRINTF("Load complete\n"); 137 138 *bufp = buf; 139 *bufsize = size; 140 141 return (EFI_SUCCESS); 142} 143 144static void 145status() 146{ 147 int i; 148 dev_info_t *dev; 149 150 for (dev = devices, i = 0; dev != NULL; dev = dev->next, i++) 151 ; 152 153 printf("%s found ", ufs_module.name); 154 switch (i) { 155 case 0: 156 printf("no partitions\n"); 157 break; 158 case 1: 159 printf("%d partition\n", i); 160 break; 161 default: 162 printf("%d partitions\n", i); 163 } 164} 165 166static dev_info_t * 167_devices() 168{ 169 170 return (devices); 171} 172 173const boot_module_t ufs_module = 174{ 175 .name = "UFS", 176 .probe = probe, 177 .load = load, 178 .status = status, 179 .devices = _devices 180}; 181