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$ 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 43 static dev_info_t *devinfo; 44 static dev_info_t *devices; 45 46 static int 47 dskread(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: %ju, size: %d, " 60 "status: %lu\n", devinfo->dev, 61 devinfo->dev->Media->MediaId, (uintmax_t)lba, size, 62 EFI_ERROR_CODE(status)); 63 return (-1); 64 } 65 66 return (0); 67 } 68 69 #include "ufsread.c" 70 71 static struct dmadat __dmadat; 72 73 static int 74 init_dev(dev_info_t* dev) 75 { 76 77 devinfo = dev; 78 dmadat = &__dmadat; 79 80 return fsread(0, NULL, 0); 81 } 82 83 static EFI_STATUS 84 probe(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 95 static EFI_STATUS 96 load(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 #ifdef EFI_DEBUG 105 { 106 CHAR16 *text = efi_devpath_name(dev->devpath); 107 DPRINTF("Loading '%s' from %S\n", filepath, text); 108 efi_free_devpath_name(text); 109 } 110 #endif 111 if (init_dev(dev) < 0) { 112 DPRINTF("Failed to init device\n"); 113 return (EFI_UNSUPPORTED); 114 } 115 116 if ((ino = lookup(filepath)) == 0) { 117 DPRINTF("Failed to lookup '%s' (file not found?)\n", filepath); 118 return (EFI_NOT_FOUND); 119 } 120 121 if (fsread_size(ino, NULL, 0, &size) < 0 || size <= 0) { 122 printf("Failed to read size of '%s' ino: %d\n", filepath, ino); 123 return (EFI_INVALID_PARAMETER); 124 } 125 126 if ((status = BS->AllocatePool(EfiLoaderData, size, &buf)) != 127 EFI_SUCCESS) { 128 printf("Failed to allocate read buffer %zu for '%s' (%lu)\n", 129 size, filepath, EFI_ERROR_CODE(status)); 130 return (status); 131 } 132 133 read = fsread(ino, buf, size); 134 if ((size_t)read != size) { 135 printf("Failed to read '%s' (%zd != %zu)\n", filepath, read, 136 size); 137 (void)BS->FreePool(buf); 138 return (EFI_INVALID_PARAMETER); 139 } 140 141 DPRINTF("Load complete\n"); 142 143 *bufp = buf; 144 *bufsize = size; 145 146 return (EFI_SUCCESS); 147 } 148 149 static void 150 status(void) 151 { 152 int i; 153 dev_info_t *dev; 154 155 for (dev = devices, i = 0; dev != NULL; dev = dev->next, i++) 156 ; 157 158 printf("%s found ", ufs_module.name); 159 switch (i) { 160 case 0: 161 printf("no partitions\n"); 162 break; 163 case 1: 164 printf("%d partition\n", i); 165 break; 166 default: 167 printf("%d partitions\n", i); 168 } 169 } 170 171 static dev_info_t * 172 _devices(void) 173 { 174 175 return (devices); 176 } 177 178 const boot_module_t ufs_module = 179 { 180 .name = "UFS", 181 .probe = probe, 182 .load = load, 183 .status = status, 184 .devices = _devices 185 }; 186