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 #include <efi.h> 29 #include <efilib.h> 30 31 struct entry { 32 EFI_HANDLE handle; 33 EFI_HANDLE alias; 34 struct devsw *dev; 35 int unit; 36 uint64_t extra; 37 }; 38 39 struct entry *entry; 40 int nentries; 41 42 int 43 efi_register_handles(struct devsw *sw, EFI_HANDLE *handles, 44 EFI_HANDLE *aliases, int count) 45 { 46 size_t sz; 47 int idx, unit; 48 49 idx = nentries; 50 nentries += count; 51 sz = nentries * sizeof(struct entry); 52 entry = (entry == NULL) ? malloc(sz) : realloc(entry, sz); 53 if (entry == NULL) 54 return (ENOMEM); 55 for (unit = 0; idx < nentries; idx++, unit++) { 56 entry[idx].handle = handles[unit]; 57 if (aliases != NULL) 58 entry[idx].alias = aliases[unit]; 59 else 60 entry[idx].alias = NULL; 61 entry[idx].dev = sw; 62 entry[idx].unit = unit; 63 } 64 return (0); 65 } 66 67 EFI_HANDLE 68 efi_find_handle(struct devsw *dev, int unit) 69 { 70 int idx; 71 72 for (idx = 0; idx < nentries; idx++) { 73 if (entry[idx].dev != dev) 74 continue; 75 if (entry[idx].unit != unit) 76 continue; 77 return (entry[idx].handle); 78 } 79 return (NULL); 80 } 81 82 int 83 efi_handle_lookup(EFI_HANDLE h, struct devsw **dev, int *unit, uint64_t *extra) 84 { 85 int idx; 86 87 for (idx = 0; idx < nentries; idx++) { 88 if (entry[idx].handle != h && entry[idx].alias != h) 89 continue; 90 if (dev != NULL) 91 *dev = entry[idx].dev; 92 if (unit != NULL) 93 *unit = entry[idx].unit; 94 if (extra != NULL) 95 *extra = entry[idx].extra; 96 return (0); 97 } 98 return (ENOENT); 99 } 100 101 int 102 efi_handle_update_dev(EFI_HANDLE h, struct devsw *dev, int unit, 103 uint64_t guid) 104 { 105 int idx; 106 107 for (idx = 0; idx < nentries; idx++) { 108 if (entry[idx].handle != h) 109 continue; 110 entry[idx].dev = dev; 111 entry[idx].unit = unit; 112 entry[idx].alias = NULL; 113 entry[idx].extra = guid; 114 return (0); 115 } 116 117 return (ENOENT); 118 } 119