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
33 #include <stdarg.h>
34 #include <stdbool.h>
35 #include <sys/cdefs.h>
36 #include <sys/param.h>
37 #include <efi.h>
38
39 #include "boot_module.h"
40
41 static dev_info_t *devinfo;
42 static dev_info_t *devices;
43
44 static int
dskread(void * buf,u_int64_t lba,int nblk)45 dskread(void *buf, u_int64_t lba, int nblk)
46 {
47 int size;
48 EFI_STATUS status;
49
50 lba = lba / (devinfo->dev->Media->BlockSize / DEV_BSIZE);
51 size = nblk * DEV_BSIZE;
52
53 status = devinfo->dev->ReadBlocks(devinfo->dev,
54 devinfo->dev->Media->MediaId, lba, size, buf);
55
56 if (status != EFI_SUCCESS) {
57 DPRINTF("dskread: failed dev: %p, id: %u, lba: %ju, size: %d, "
58 "status: %lu\n", devinfo->dev,
59 devinfo->dev->Media->MediaId, (uintmax_t)lba, size,
60 EFI_ERROR_CODE(status));
61 return (-1);
62 }
63
64 return (0);
65 }
66
67 #include "ufsread.c"
68
69 static struct dmadat __dmadat;
70
71 static int
init_dev(dev_info_t * dev)72 init_dev(dev_info_t* dev)
73 {
74
75 devinfo = dev;
76 dmadat = &__dmadat;
77
78 return fsread(0, NULL, 0);
79 }
80
81 static EFI_STATUS
probe(dev_info_t * dev)82 probe(dev_info_t* dev)
83 {
84
85 if (init_dev(dev) < 0)
86 return (EFI_UNSUPPORTED);
87
88 add_device(&devices, dev);
89
90 return (EFI_SUCCESS);
91 }
92
93 static EFI_STATUS
load(const char * filepath,dev_info_t * dev,void ** bufp,size_t * bufsize)94 load(const char *filepath, dev_info_t *dev, void **bufp, size_t *bufsize)
95 {
96 ufs_ino_t ino;
97 EFI_STATUS status;
98 size_t size;
99 ssize_t read;
100 void *buf;
101
102 DPRINTF("Loading '%s' from %s\n", filepath, devpath_str(dev->devpath));
103
104 if (init_dev(dev) < 0) {
105 DPRINTF("Failed to init device\n");
106 return (EFI_UNSUPPORTED);
107 }
108
109 if ((ino = lookup(filepath)) == 0) {
110 DPRINTF("Failed to lookup '%s' (file not found?)\n", filepath);
111 return (EFI_NOT_FOUND);
112 }
113
114 if (fsread_size(ino, NULL, 0, &size) < 0 || size <= 0) {
115 printf("Failed to read size of '%s' ino: %d\n", filepath, ino);
116 return (EFI_INVALID_PARAMETER);
117 }
118
119 if ((status = bs->AllocatePool(EfiLoaderData, size, &buf)) !=
120 EFI_SUCCESS) {
121 printf("Failed to allocate read buffer %zu for '%s' (%lu)\n",
122 size, filepath, EFI_ERROR_CODE(status));
123 return (status);
124 }
125
126 read = fsread(ino, buf, size);
127 if ((size_t)read != size) {
128 printf("Failed to read '%s' (%zd != %zu)\n", filepath, read,
129 size);
130 (void)bs->FreePool(buf);
131 return (EFI_INVALID_PARAMETER);
132 }
133
134 DPRINTF("Load complete\n");
135
136 *bufp = buf;
137 *bufsize = size;
138
139 return (EFI_SUCCESS);
140 }
141
142 static void
status(void)143 status(void)
144 {
145 int i;
146 dev_info_t *dev;
147
148 for (dev = devices, i = 0; dev != NULL; dev = dev->next, i++)
149 ;
150
151 printf("%s found ", ufs_module.name);
152 switch (i) {
153 case 0:
154 printf("no partitions\n");
155 break;
156 case 1:
157 printf("%d partition\n", i);
158 break;
159 default:
160 printf("%d partitions\n", i);
161 }
162 }
163
164 static dev_info_t *
_devices(void)165 _devices(void)
166 {
167
168 return (devices);
169 }
170
171 const boot_module_t ufs_module =
172 {
173 .name = "UFS",
174 .probe = probe,
175 .load = load,
176 .status = status,
177 .devices = _devices
178 };
179