1 /* ldbuildid.c - Build Id support routines
2 Copyright (C) 2013-2024 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
validate_build_id_style(const char * style)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
compute_build_id_size(const char * style)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
read_hex(const char xdigit)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
generate_build_id(bfd * abfd,const char * style,checksum_fn checksum_contents,unsigned char * id_bits,int size ATTRIBUTE_UNUSED)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_choose_process_bytes (),
118 &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 (startswith (style, "0x"))
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