1 /*- 2 * Copyright (c) 2006 Marcel Moolenaar 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 #include <efi.h> 31 #include <efilib.h> 32 33 struct entry { 34 EFI_HANDLE handle; 35 EFI_HANDLE alias; 36 struct devsw *dev; 37 int unit; 38 uint64_t extra; 39 }; 40 41 struct entry *entry; 42 int nentries; 43 44 int 45 efi_register_handles(struct devsw *sw, EFI_HANDLE *handles, 46 EFI_HANDLE *aliases, int count) 47 { 48 size_t sz; 49 int idx, unit; 50 51 idx = nentries; 52 nentries += count; 53 sz = nentries * sizeof(struct entry); 54 entry = (entry == NULL) ? malloc(sz) : realloc(entry, sz); 55 if (entry == NULL) 56 return (ENOMEM); 57 for (unit = 0; idx < nentries; idx++, unit++) { 58 entry[idx].handle = handles[unit]; 59 if (aliases != NULL) 60 entry[idx].alias = aliases[unit]; 61 else 62 entry[idx].alias = NULL; 63 entry[idx].dev = sw; 64 entry[idx].unit = unit; 65 } 66 return (0); 67 } 68 69 EFI_HANDLE 70 efi_find_handle(struct devsw *dev, int unit) 71 { 72 int idx; 73 74 for (idx = 0; idx < nentries; idx++) { 75 if (entry[idx].dev != dev) 76 continue; 77 if (entry[idx].unit != unit) 78 continue; 79 return (entry[idx].handle); 80 } 81 return (NULL); 82 } 83 84 int 85 efi_handle_lookup(EFI_HANDLE h, struct devsw **dev, int *unit, uint64_t *extra) 86 { 87 int idx; 88 89 for (idx = 0; idx < nentries; idx++) { 90 if (entry[idx].handle != h && entry[idx].alias != h) 91 continue; 92 if (dev != NULL) 93 *dev = entry[idx].dev; 94 if (unit != NULL) 95 *unit = entry[idx].unit; 96 if (extra != NULL) 97 *extra = entry[idx].extra; 98 return (0); 99 } 100 return (ENOENT); 101 } 102 103 int 104 efi_handle_update_dev(EFI_HANDLE h, struct devsw *dev, int unit, 105 uint64_t guid) 106 { 107 int idx; 108 109 for (idx = 0; idx < nentries; idx++) { 110 if (entry[idx].handle != h) 111 continue; 112 entry[idx].dev = dev; 113 entry[idx].unit = unit; 114 entry[idx].alias = NULL; 115 entry[idx].extra = guid; 116 return (0); 117 } 118 119 return (ENOENT); 120 } 121