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