ld/ldbuildid.c

   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 }