commands/xnu_uuid.c

   1 /* xnu_uuid.c - transform 64-bit serial number
   2    to 128-bit uuid suitable for xnu. */
   3 /*
   4  *  GRUB  --  GRand Unified Bootloader
   5  *  Copyright (C) 1995,1996,1998,1999,2001,2002,
   6  *                2003, 2009  Free Software Foundation, Inc.
   7  *
   8  *  GRUB is free software: you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation, either version 3 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  GRUB is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 #include <grub/types.h>
  23 #include <grub/misc.h>
  24 #include <grub/mm.h>
  25 #include <grub/err.h>
  26 #include <grub/dl.h>
  27 #include <grub/device.h>
  28 #include <grub/disk.h>
  29 #include <grub/fs.h>
  30 #include <grub/file.h>
  31 #include <grub/misc.h>
  32 #include <grub/env.h>
  33 #include <grub/command.h>
  34
  35 struct tohash
  36 {
  37   grub_uint8_t prefix[16];
  38   grub_uint64_t serial;
  39 } __attribute__ ((packed));
  40
  41 /* This prefix is used by xnu and boot-132 to hash
  42    together with volume serial. */
  43 static grub_uint8_t hash_prefix[16]
  44   = {0xB3, 0xE2, 0x0F, 0x39, 0xF2, 0x92, 0x11, 0xD6,
  45      0x97, 0xA4, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC};
  46
  47 #define rol(x,n) ( ((x) << (n)) | ((x) >> (32-(n))) )
  48 #define ror(x,n) ( ((x) >> (n)) | ((x) << (32-(n))) )
  49
  50 typedef struct {
  51   grub_uint32_t A,B,C,D;          /* chaining variables */
  52   grub_uint32_t  nblocks;
  53   grub_uint8_t buf[64];
  54   int  count;
  55 } MD5_CONTEXT;
  56
  57 static void
  58 md5_init( void *context )
  59 {
  60   MD5_CONTEXT *ctx = context;
  61
  62   ctx->A = 0x67452301;
  63   ctx->B = 0xefcdab89;
  64   ctx->C = 0x98badcfe;
  65   ctx->D = 0x10325476;
  66
  67   ctx->nblocks = 0;
  68   ctx->count = 0;
  69 }
  70
  71 /* These are the four functions used in the four steps of the MD5 algorithm
  72    and defined in the RFC 1321.  The first function is a little bit optimized
  73    (as found in Colin Plumbs public domain implementation).  */
  74 /* #define FF(b, c, d) ((b & c) | (~b & d)) */
  75 #define FF(b, c, d) (d ^ (b & (c ^ d)))
  76 #define FG(b, c, d) FF (d, b, c)
  77 #define FH(b, c, d) (b ^ c ^ d)
  78 #define FI(b, c, d) (c ^ (b | ~d))
  79
  80
  81 /****************
  82  * transform n*64 grub_uint8_ts
  83  */
  84 static void
  85 transform ( MD5_CONTEXT *ctx, const unsigned char *data )
  86 {
  87   grub_uint32_t correct_words[16];
  88   register grub_uint32_t A = ctx->A;
  89   register grub_uint32_t B = ctx->B;
  90   register grub_uint32_t C = ctx->C;
  91   register grub_uint32_t D = ctx->D;
  92   grub_uint32_t *cwp = correct_words;
  93
  94 #ifdef WORDS_BIGENDIAN
  95   {
  96     int i;
  97     grub_uint8_t *p2, *p1;
  98     for(i=0, p1=data, p2=(grub_uint8_t*)correct_words; i < 16; i++, p2 += 4 )
  99     {
 100       p2[3] = *p1++;
 101       p2[2] = *p1++;
 102       p2[1] = *p1++;
 103       p2[0] = *p1++;
 104     }
 105   }
 106 #else
 107   memcpy( correct_words, data, 64 );
 108 #endif
 109
 110 #define OP(a, b, c, d, s, T) \
 111   do                                       \
 112   {                                        \
 113     a += FF (b, c, d) + (*cwp++) + T;    \
 114     a = rol(a, s);                         \
 115     a += b;                                \
 116   }                                        \
 117   while (0)
 118
 119   /* Before we start, one word about the strange constants.
 120      They are defined in RFC 1321 as
 121
 122      T[i] = (int) (4294967296.0 * fabs (sin (i))), i=1..64
 123      */
 124
 125   /* Round 1.  */
 126   OP (A, B, C, D,  7, 0xd76aa478);
 127   OP (D, A, B, C, 12, 0xe8c7b756);
 128   OP (C, D, A, B, 17, 0x242070db);
 129   OP (B, C, D, A, 22, 0xc1bdceee);
 130   OP (A, B, C, D,  7, 0xf57c0faf);
 131   OP (D, A, B, C, 12, 0x4787c62a);
 132   OP (C, D, A, B, 17, 0xa8304613);
 133   OP (B, C, D, A, 22, 0xfd469501);
 134   OP (A, B, C, D,  7, 0x698098d8);
 135   OP (D, A, B, C, 12, 0x8b44f7af);
 136   OP (C, D, A, B, 17, 0xffff5bb1);
 137   OP (B, C, D, A, 22, 0x895cd7be);
 138   OP (A, B, C, D,  7, 0x6b901122);
 139   OP (D, A, B, C, 12, 0xfd987193);
 140   OP (C, D, A, B, 17, 0xa679438e);
 141   OP (B, C, D, A, 22, 0x49b40821);
 142
 143 #undef OP
 144 #define OP(f, a, b, c, d, k, s, T)  \
 145   do                                                                  \
 146   {                                                           \
 147     a += f (b, c, d) + correct_words[k] + T;                  \
 148     a = rol(a, s);                                                    \
 149     a += b;                                                   \
 150   }                                                           \
 151   while (0)
 152
 153   /* Round 2.  */
 154   OP (FG, A, B, C, D,  1,  5, 0xf61e2562);
 155   OP (FG, D, A, B, C,  6,  9, 0xc040b340);
 156   OP (FG, C, D, A, B, 11, 14, 0x265e5a51);
 157   OP (FG, B, C, D, A,  0, 20, 0xe9b6c7aa);
 158   OP (FG, A, B, C, D,  5,  5, 0xd62f105d);
 159   OP (FG, D, A, B, C, 10,  9, 0x02441453);
 160   OP (FG, C, D, A, B, 15, 14, 0xd8a1e681);
 161   OP (FG, B, C, D, A,  4, 20, 0xe7d3fbc8);
 162   OP (FG, A, B, C, D,  9,  5, 0x21e1cde6);
 163   OP (FG, D, A, B, C, 14,  9, 0xc33707d6);
 164   OP (FG, C, D, A, B,  3, 14, 0xf4d50d87);
 165   OP (FG, B, C, D, A,  8, 20, 0x455a14ed);
 166   OP (FG, A, B, C, D, 13,  5, 0xa9e3e905);
 167   OP (FG, D, A, B, C,  2,  9, 0xfcefa3f8);
 168   OP (FG, C, D, A, B,  7, 14, 0x676f02d9);
 169   OP (FG, B, C, D, A, 12, 20, 0x8d2a4c8a);
 170
 171   /* Round 3.  */
 172   OP (FH, A, B, C, D,  5,  4, 0xfffa3942);
 173   OP (FH, D, A, B, C,  8, 11, 0x8771f681);
 174   OP (FH, C, D, A, B, 11, 16, 0x6d9d6122);
 175   OP (FH, B, C, D, A, 14, 23, 0xfde5380c);
 176   OP (FH, A, B, C, D,  1,  4, 0xa4beea44);
 177   OP (FH, D, A, B, C,  4, 11, 0x4bdecfa9);
 178   OP (FH, C, D, A, B,  7, 16, 0xf6bb4b60);
 179   OP (FH, B, C, D, A, 10, 23, 0xbebfbc70);
 180   OP (FH, A, B, C, D, 13,  4, 0x289b7ec6);
 181   OP (FH, D, A, B, C,  0, 11, 0xeaa127fa);
 182   OP (FH, C, D, A, B,  3, 16, 0xd4ef3085);
 183   OP (FH, B, C, D, A,  6, 23, 0x04881d05);
 184   OP (FH, A, B, C, D,  9,  4, 0xd9d4d039);
 185   OP (FH, D, A, B, C, 12, 11, 0xe6db99e5);
 186   OP (FH, C, D, A, B, 15, 16, 0x1fa27cf8);
 187   OP (FH, B, C, D, A,  2, 23, 0xc4ac5665);
 188
 189   /* Round 4.  */
 190   OP (FI, A, B, C, D,  0,  6, 0xf4292244);
 191   OP (FI, D, A, B, C,  7, 10, 0x432aff97);
 192   OP (FI, C, D, A, B, 14, 15, 0xab9423a7);
 193   OP (FI, B, C, D, A,  5, 21, 0xfc93a039);
 194   OP (FI, A, B, C, D, 12,  6, 0x655b59c3);
 195   OP (FI, D, A, B, C,  3, 10, 0x8f0ccc92);
 196   OP (FI, C, D, A, B, 10, 15, 0xffeff47d);
 197   OP (FI, B, C, D, A,  1, 21, 0x85845dd1);
 198   OP (FI, A, B, C, D,  8,  6, 0x6fa87e4f);
 199   OP (FI, D, A, B, C, 15, 10, 0xfe2ce6e0);
 200   OP (FI, C, D, A, B,  6, 15, 0xa3014314);
 201   OP (FI, B, C, D, A, 13, 21, 0x4e0811a1);
 202   OP (FI, A, B, C, D,  4,  6, 0xf7537e82);
 203   OP (FI, D, A, B, C, 11, 10, 0xbd3af235);
 204   OP (FI, C, D, A, B,  2, 15, 0x2ad7d2bb);
 205   OP (FI, B, C, D, A,  9, 21, 0xeb86d391);
 206
 207   /* Put checksum in context given as argument.  */
 208   ctx->A += A;
 209   ctx->B += B;
 210   ctx->C += C;
 211   ctx->D += D;
 212 }
 213
 214 /* The routine updates the message-digest context to
 215  * account for the presence of each of the characters inBuf[0..inLen-1]
 216  * in the message whose digest is being computed.
 217  */
 218 static void
 219 md5_write( void *context, const void *inbuf_arg , grub_size_t inlen)
 220 {
 221   const unsigned char *inbuf = inbuf_arg;
 222   MD5_CONTEXT *hd = context;
 223
 224   if( hd->count == 64 )  /* flush the buffer */
 225   {
 226     transform( hd, hd->buf );
 227     //      _gcry_burn_stack (80+6*sizeof(void*));
 228     hd->count = 0;
 229     hd->nblocks++;
 230   }
 231   if( !inbuf )
 232     return;
 233
 234   if( hd->count )
 235   {
 236     for( ; inlen && hd->count < 64; inlen-- )
 237       hd->buf[hd->count++] = *inbuf++;
 238     md5_write( hd, NULL, 0 );
 239     if( !inlen )
 240       return;
 241   }
 242   //  _gcry_burn_stack (80+6*sizeof(void*));
 243
 244   while( inlen >= 64 )
 245   {
 246     transform( hd, inbuf );
 247     hd->count = 0;
 248     hd->nblocks++;
 249     inlen -= 64;
 250     inbuf += 64;
 251   }
 252   for( ; inlen && hd->count < 64; inlen-- )
 253     hd->buf[hd->count++] = *inbuf++;
 254
 255 }
 256
 257
 258
 259 /* The routine final terminates the message-digest computation and
 260  * ends with the desired message digest in mdContext->digest[0...15].
 261  * The handle is prepared for a new MD5 cycle.
 262  * Returns 16 grub_uint8_ts representing the digest.
 263  */
 264 static void
 265 md5_final( void *context)
 266 {
 267   MD5_CONTEXT *hd = context;
 268   grub_uint32_t t, msb, lsb;
 269   grub_uint8_t *p;
 270
 271   md5_write(hd, NULL, 0); /* flush */;
 272
 273   t = hd->nblocks;
 274   /* multiply by 64 to make a grub_uint8_t count */
 275   lsb = t << 6;
 276   msb = t >> 26;
 277   /* add the count */
 278   t = lsb;
 279   if( (lsb += hd->count) < t )
 280     msb++;
 281   /* multiply by 8 to make a bit count */
 282   t = lsb;
 283   lsb <<= 3;
 284   msb <<= 3;
 285   msb |= t >> 29;
 286
 287   if( hd->count < 56 )  /* enough room */
 288   {
 289     hd->buf[hd->count++] = 0x80; /* pad */
 290     while( hd->count < 56 )
 291       hd->buf[hd->count++] = 0;  /* pad */
 292   }
 293   else  /* need one extra block */
 294   {
 295     hd->buf[hd->count++] = 0x80; /* pad character */
 296     while( hd->count < 64 )
 297       hd->buf[hd->count++] = 0;
 298     md5_write(hd, NULL, 0);  /* flush */;
 299     grub_memset(hd->buf, 0, 56 ); /* fill next block with zeroes */
 300   }
 301   /* append the 64 bit count */
 302   hd->buf[56] = lsb        ;
 303   hd->buf[57] = lsb >>  8;
 304   hd->buf[58] = lsb >> 16;
 305   hd->buf[59] = lsb >> 24;
 306   hd->buf[60] = msb        ;
 307   hd->buf[61] = msb >>  8;
 308   hd->buf[62] = msb >> 16;
 309   hd->buf[63] = msb >> 24;
 310   transform( hd, hd->buf );
 311   //  _gcry_burn_stack (80+6*sizeof(void*));
 312
 313   p = hd->buf;
 314 #ifdef WORDS_BIGENDIAN
 315 #define X(a) do { *p++ = hd->a      ; *p++ = hd->a >> 8;      \
 316   *p++ = hd->a >> 16; *p++ = hd->a >> 24; } while(0)
 317 #else /* little endian */
 318 #define X(a) do { *(grub_uint32_t*)p = (*hd).a ; p += 4; } while(0)
 319 #endif
 320   X(A);
 321   X(B);
 322   X(C);
 323   X(D);
 324 #undef X
 325
 326 }
 327
 328 /**
 329  * GRUB2 Crypto Interface
 330  * Written by Michael Gorven
 331  */
 332 static grub_err_t
 333 md5 (const char *in, grub_size_t insize, char *out)
 334 {
 335   MD5_CONTEXT hd;
 336
 337   md5_init (&hd);
 338   md5_write (&hd, in, insize);
 339   md5_final (&hd);
 340   grub_memcpy (out, hd.buf, 16);
 341
 342   return GRUB_ERR_NONE;
 343 }
 344
 345 static grub_err_t
 346 grub_cmd_xnu_uuid (grub_command_t cmd __attribute__ ((unused)),
 347                    int argc, char **args)
 348 {
 349   struct tohash hashme;
 350   grub_uint8_t xnu_uuid[16];
 351   char uuid_string[sizeof ("xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx")];
 352   char *ptr;
 353
 354   if (argc < 1)
 355     return grub_error (GRUB_ERR_BAD_ARGUMENT, "UUID required");
 356
 357   hashme.serial = grub_cpu_to_be64 (grub_strtoull (args[0], 0, 16));
 358   grub_memcpy (hashme.prefix, hash_prefix, sizeof (hashme.prefix));
 359
 360   md5 ((char *) &hashme, sizeof (hashme), (char *) xnu_uuid);
 361   xnu_uuid[6] = (xnu_uuid[6] & 0xf) | 0x30;
 362   xnu_uuid[8] = (xnu_uuid[8] & 0x3f) | 0x80;
 363   grub_sprintf (uuid_string,
 364                 "%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x",
 365                 (unsigned int) xnu_uuid[0], (unsigned int) xnu_uuid[1],
 366                 (unsigned int) xnu_uuid[2], (unsigned int) xnu_uuid[3],
 367                 (unsigned int) xnu_uuid[4], (unsigned int) xnu_uuid[5],
 368                 (unsigned int) ((xnu_uuid[6] & 0xf) | 0x30),
 369                 (unsigned int) xnu_uuid[7],
 370                 (unsigned int) ((xnu_uuid[8] & 0x3f) | 0x80),
 371                 (unsigned int) xnu_uuid[9],
 372                 (unsigned int) xnu_uuid[10], (unsigned int) xnu_uuid[11],
 373                 (unsigned int) xnu_uuid[12], (unsigned int) xnu_uuid[13],
 374                 (unsigned int) xnu_uuid[14], (unsigned int) xnu_uuid[15]);
 375   for (ptr = uuid_string; *ptr; ptr++)
 376     *ptr = grub_toupper (*ptr);
 377   if (argc == 1)
 378     grub_printf ("%s", uuid_string);
 379   if (argc > 1)
 380     grub_env_set (args[1], uuid_string);
 381
 382   return GRUB_ERR_NONE;
 383 }
 384
 385 static grub_command_t cmd;
 386
 387
 388 GRUB_MOD_INIT (xnu_uuid)
 389 {
 390   cmd = grub_register_command ("xnu_uuid", grub_cmd_xnu_uuid,
 391                                "xnu_uuid GRUBUUID [VARNAME]",
 392                                "Transform 64-bit UUID to format "
 393                                "suitable for xnu.");
 394 }
 395
 396 GRUB_MOD_FINI (xnu_uuid)
 397 {
 398   grub_unregister_command (cmd);
 399 }