1 /* ldbuildid.c - Build Id support routines 2 Copyright (C) 2013-2022 Free Software Foundation, Inc. 3 4 This file is part of the GNU Binutils. 5 6 This program is free software; you can redistribute it and/or modify 7 it under the terms of the GNU General Public License as published by 8 the Free Software Foundation; either version 3 of the License, or 9 (at your option) any later version. 10 11 This program is distributed in the hope that it will be useful, 12 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 GNU General Public License for more details. 15 16 You should have received a copy of the GNU General Public License 17 along with this program; if not, write to the Free Software 18 Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, 19 MA 02110-1301, USA. */ 20 21 #include "sysdep.h" 22 #include "bfd.h" 23 #include "safe-ctype.h" 24 #include "md5.h" 25 #include "sha1.h" 26 #include "ldbuildid.h" 27 #ifdef __MINGW32__ 28 #include <windows.h> 29 #include <rpcdce.h> 30 #endif 31 32 #define streq(a,b) strcmp ((a), (b)) == 0 33 34 bool 35 validate_build_id_style (const char *style) 36 { 37 if ((streq (style, "md5")) || (streq (style, "sha1")) 38 || (streq (style, "uuid")) || (startswith (style, "0x"))) 39 return true; 40 41 return false; 42 } 43 44 bfd_size_type 45 compute_build_id_size (const char *style) 46 { 47 if (streq (style, "md5") || streq (style, "uuid")) 48 return 128 / 8; 49 50 if (streq (style, "sha1")) 51 return 160 / 8; 52 53 if (startswith (style, "0x")) 54 { 55 bfd_size_type size = 0; 56 /* ID is in string form (hex). Count the bytes. */ 57 const char *id = style + 2; 58 59 do 60 { 61 if (ISXDIGIT (id[0]) && ISXDIGIT (id[1])) 62 { 63 ++size; 64 id += 2; 65 } 66 else if (*id == '-' || *id == ':') 67 ++id; 68 else 69 { 70 size = 0; 71 break; 72 } 73 } while (*id != '\0'); 74 return size; 75 } 76 77 return 0; 78 } 79 80 static unsigned char 81 read_hex (const char xdigit) 82 { 83 if (ISDIGIT (xdigit)) 84 return xdigit - '0'; 85 86 if (ISUPPER (xdigit)) 87 return xdigit - 'A' + 0xa; 88 89 if (ISLOWER (xdigit)) 90 return xdigit - 'a' + 0xa; 91 92 abort (); 93 return 0; 94 } 95 96 bool 97 generate_build_id (bfd *abfd, 98 const char *style, 99 checksum_fn checksum_contents, 100 unsigned char *id_bits, 101 int size ATTRIBUTE_UNUSED) 102 { 103 if (streq (style, "md5")) 104 { 105 struct md5_ctx ctx; 106 107 md5_init_ctx (&ctx); 108 if (!(*checksum_contents) (abfd, (sum_fn) &md5_process_bytes, &ctx)) 109 return false; 110 md5_finish_ctx (&ctx, id_bits); 111 } 112 else if (streq (style, "sha1")) 113 { 114 struct sha1_ctx ctx; 115 116 sha1_init_ctx (&ctx); 117 if (!(*checksum_contents) (abfd, (sum_fn) &sha1_process_bytes, &ctx)) 118 return false; 119 sha1_finish_ctx (&ctx, id_bits); 120 } 121 else if (streq (style, "uuid")) 122 { 123 #ifndef __MINGW32__ 124 int n; 125 int fd = open ("/dev/urandom", O_RDONLY); 126 127 if (fd < 0) 128 return false; 129 n = read (fd, id_bits, size); 130 close (fd); 131 if (n < size) 132 return false; 133 #else /* __MINGW32__ */ 134 typedef RPC_STATUS (RPC_ENTRY * UuidCreateFn) (UUID *); 135 UUID uuid; 136 UuidCreateFn uuid_create = 0; 137 HMODULE rpc_library = LoadLibrary ("rpcrt4.dll"); 138 139 if (!rpc_library) 140 return false; 141 uuid_create = (UuidCreateFn) (void (WINAPI *)(void)) GetProcAddress (rpc_library, "UuidCreate"); 142 if (!uuid_create) 143 { 144 FreeLibrary (rpc_library); 145 return false; 146 } 147 148 if (uuid_create (&uuid) != RPC_S_OK) 149 { 150 FreeLibrary (rpc_library); 151 return false; 152 } 153 FreeLibrary (rpc_library); 154 memcpy (id_bits, &uuid, 155 (size_t) size < sizeof (UUID) ? (size_t) size : sizeof (UUID)); 156 #endif /* __MINGW32__ */ 157 } 158 else if (startswith (style, "0x")) 159 { 160 /* ID is in string form (hex). Convert to bits. */ 161 const char *id = style + 2; 162 size_t n = 0; 163 164 do 165 { 166 if (ISXDIGIT (id[0]) && ISXDIGIT (id[1])) 167 { 168 id_bits[n] = read_hex (*id++) << 4; 169 id_bits[n++] |= read_hex (*id++); 170 } 171 else if (*id == '-' || *id == ':') 172 ++id; 173 else 174 abort (); /* Should have been validated earlier. */ 175 } 176 while (*id != '\0'); 177 } 178 else 179 abort (); /* Should have been validated earlier. */ 180 181 return true; 182 } 183