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