removed accidental commit
[grub2/phcoder.git] / commands / xnu_uuid.c
blob1302881fa5a08c0acb380e6ab36287a4954b177a
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.
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.
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.
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/>.
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>
35 struct tohash
37 grub_uint8_t prefix[16];
38 grub_uint64_t serial;
39 } __attribute__ ((packed));
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};
47 #define rol(x,n) ( ((x) << (n)) | ((x) >> (32-(n))) )
48 #define ror(x,n) ( ((x) >> (n)) | ((x) << (32-(n))) )
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;
57 static void
58 md5_init( void *context )
60 MD5_CONTEXT *ctx = context;
62 ctx->A = 0x67452301;
63 ctx->B = 0xefcdab89;
64 ctx->C = 0x98badcfe;
65 ctx->D = 0x10325476;
67 ctx->nblocks = 0;
68 ctx->count = 0;
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))
81 /****************
82 * transform n*64 grub_uint8_ts
84 static void
85 transform ( MD5_CONTEXT *ctx, const unsigned char *data )
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;
94 #ifdef WORDS_BIGENDIAN
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 )
100 p2[3] = *p1++;
101 p2[2] = *p1++;
102 p2[1] = *p1++;
103 p2[0] = *p1++;
106 #else
107 memcpy( correct_words, data, 64 );
108 #endif
110 #define OP(a, b, c, d, s, T) \
111 do \
113 a += FF (b, c, d) + (*cwp++) + T; \
114 a = rol(a, s); \
115 a += b; \
117 while (0)
119 /* Before we start, one word about the strange constants.
120 They are defined in RFC 1321 as
122 T[i] = (int) (4294967296.0 * fabs (sin (i))), i=1..64
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);
143 #undef OP
144 #define OP(f, a, b, c, d, k, s, T) \
145 do \
147 a += f (b, c, d) + correct_words[k] + T; \
148 a = rol(a, s); \
149 a += b; \
151 while (0)
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);
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);
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);
207 /* Put checksum in context given as argument. */
208 ctx->A += A;
209 ctx->B += B;
210 ctx->C += C;
211 ctx->D += D;
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.
218 static void
219 md5_write( void *context, const void *inbuf_arg , grub_size_t inlen)
221 const unsigned char *inbuf = inbuf_arg;
222 MD5_CONTEXT *hd = context;
224 if( hd->count == 64 ) /* flush the buffer */
226 transform( hd, hd->buf );
227 // _gcry_burn_stack (80+6*sizeof(void*));
228 hd->count = 0;
229 hd->nblocks++;
231 if( !inbuf )
232 return;
234 if( hd->count )
236 for( ; inlen && hd->count < 64; inlen-- )
237 hd->buf[hd->count++] = *inbuf++;
238 md5_write( hd, NULL, 0 );
239 if( !inlen )
240 return;
242 // _gcry_burn_stack (80+6*sizeof(void*));
244 while( inlen >= 64 )
246 transform( hd, inbuf );
247 hd->count = 0;
248 hd->nblocks++;
249 inlen -= 64;
250 inbuf += 64;
252 for( ; inlen && hd->count < 64; inlen-- )
253 hd->buf[hd->count++] = *inbuf++;
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.
264 static void
265 md5_final( void *context)
267 MD5_CONTEXT *hd = context;
268 grub_uint32_t t, msb, lsb;
269 grub_uint8_t *p;
271 md5_write(hd, NULL, 0); /* flush */;
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;
287 if( hd->count < 56 ) /* enough room */
289 hd->buf[hd->count++] = 0x80; /* pad */
290 while( hd->count < 56 )
291 hd->buf[hd->count++] = 0; /* pad */
293 else /* need one extra block */
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 */
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*));
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
329 * GRUB2 Crypto Interface
330 * Written by Michael Gorven
332 static grub_err_t
333 md5 (const char *in, grub_size_t insize, char *out)
335 MD5_CONTEXT hd;
337 md5_init (&hd);
338 md5_write (&hd, in, insize);
339 md5_final (&hd);
340 grub_memcpy (out, hd.buf, 16);
342 return GRUB_ERR_NONE;
345 static grub_err_t
346 grub_cmd_xnu_uuid (grub_command_t cmd __attribute__ ((unused)),
347 int argc, char **args)
349 struct tohash hashme;
350 grub_uint8_t xnu_uuid[16];
351 char uuid_string[sizeof ("xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx")];
352 char *ptr;
354 if (argc < 1)
355 return grub_error (GRUB_ERR_BAD_ARGUMENT, "UUID required");
357 hashme.serial = grub_cpu_to_be64 (grub_strtoull (args[0], 0, 16));
358 grub_memcpy (hashme.prefix, hash_prefix, sizeof (hashme.prefix));
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);
382 return GRUB_ERR_NONE;
385 static grub_command_t cmd;
388 GRUB_MOD_INIT (xnu_uuid)
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.");
396 GRUB_MOD_FINI (xnu_uuid)
398 grub_unregister_command (cmd);