sys/boot/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: %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
  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         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
 144 static void
 145 status()
 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
 166 static dev_info_t *
 167 _devices()
 168 {
 169
 170         return (devices);
 171 }
 172
 173 const boot_module_t ufs_module =
 174 {
 175         .name = "UFS",
 176         .probe = probe,
 177         .load = load,
 178         .status = status,
 179         .devices = _devices
 180 };