1 /* ldbuildid.c - Build Id support routines 2 Copyright (C) 2013-2020 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 #define strneq(a,b,n) strncmp ((a), (b), (n)) == 0 34 35 bfd_boolean 36 validate_build_id_style (const char *style) 37 { 38 if ((streq (style, "md5")) || (streq (style, "sha1")) 39 || (streq (style, "uuid")) || (strneq (style, "0x", 2))) 40 return TRUE; 41 42 return FALSE; 43 } 44 45 bfd_size_type 46 compute_build_id_size (const char *style) 47 { 48 if (streq (style, "md5") || streq (style, "uuid")) 49 return 128 / 8; 50 51 if (streq (style, "sha1")) 52 return 160 / 8; 53 54 if (strneq (style, "0x", 2)) 55 { 56 bfd_size_type size = 0; 57 /* ID is in string form (hex). Count the bytes. */ 58 const char *id = style + 2; 59 60 do 61 { 62 if (ISXDIGIT (id[0]) && ISXDIGIT (id[1])) 63 { 64 ++size; 65 id += 2; 66 } 67 else if (*id == '-' || *id == ':') 68 ++id; 69 else 70 { 71 size = 0; 72 break; 73 } 74 } while (*id != '\0'); 75 return size; 76 } 77 78 return 0; 79 } 80 81 static unsigned char 82 read_hex (const char xdigit) 83 { 84 if (ISDIGIT (xdigit)) 85 return xdigit - '0'; 86 87 if (ISUPPER (xdigit)) 88 return xdigit - 'A' + 0xa; 89 90 if (ISLOWER (xdigit)) 91 return xdigit - 'a' + 0xa; 92 93 abort (); 94 return 0; 95 } 96 97 bfd_boolean 98 generate_build_id (bfd *abfd, 99 const char *style, 100 checksum_fn checksum_contents, 101 unsigned char *id_bits, 102 int size ATTRIBUTE_UNUSED) 103 { 104 if (streq (style, "md5")) 105 { 106 struct md5_ctx ctx; 107 108 md5_init_ctx (&ctx); 109 if (!(*checksum_contents) (abfd, (sum_fn) &md5_process_bytes, &ctx)) 110 return FALSE; 111 md5_finish_ctx (&ctx, id_bits); 112 } 113 else if (streq (style, "sha1")) 114 { 115 struct sha1_ctx ctx; 116 117 sha1_init_ctx (&ctx); 118 if (!(*checksum_contents) (abfd, (sum_fn) &sha1_process_bytes, &ctx)) 119 return FALSE; 120 sha1_finish_ctx (&ctx, id_bits); 121 } 122 else if (streq (style, "uuid")) 123 { 124 #ifndef __MINGW32__ 125 int n; 126 int fd = open ("/dev/urandom", O_RDONLY); 127 128 if (fd < 0) 129 return FALSE; 130 n = read (fd, id_bits, size); 131 close (fd); 132 if (n < size) 133 return FALSE; 134 #else /* __MINGW32__ */ 135 typedef RPC_STATUS (RPC_ENTRY * UuidCreateFn) (UUID *); 136 UUID uuid; 137 UuidCreateFn uuid_create = 0; 138 HMODULE rpc_library = LoadLibrary ("rpcrt4.dll"); 139 140 if (!rpc_library) 141 return FALSE; 142 uuid_create = (UuidCreateFn) (void (WINAPI *)(void)) GetProcAddress (rpc_library, "UuidCreate"); 143 if (!uuid_create) 144 { 145 FreeLibrary (rpc_library); 146 return FALSE; 147 } 148 149 if (uuid_create (&uuid) != RPC_S_OK) 150 { 151 FreeLibrary (rpc_library); 152 return FALSE; 153 } 154 FreeLibrary (rpc_library); 155 memcpy (id_bits, &uuid, 156 (size_t) size < sizeof (UUID) ? (size_t) size : sizeof (UUID)); 157 #endif /* __MINGW32__ */ 158 } 159 else if (strneq (style, "0x", 2)) 160 { 161 /* ID is in string form (hex). Convert to bits. */ 162 const char *id = style + 2; 163 size_t n = 0; 164 165 do 166 { 167 if (ISXDIGIT (id[0]) && ISXDIGIT (id[1])) 168 { 169 id_bits[n] = read_hex (*id++) << 4; 170 id_bits[n++] |= read_hex (*id++); 171 } 172 else if (*id == '-' || *id == ':') 173 ++id; 174 else 175 abort (); /* Should have been validated earlier. */ 176 } 177 while (*id != '\0'); 178 } 179 else 180 abort (); /* Should have been validated earlier. */ 181 182 return TRUE; 183 } 184