ld/ldbuildid.c

   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
  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_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 }