stand/efi/boot1/ufs_module.c

   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 };