Loading...
bsd/dev/random/randomdev.c xnu-1699.32.7 xnu-1228
--- xnu/xnu-1699.32.7/bsd/dev/random/randomdev.c
+++ xnu/xnu-1228/bsd/dev/random/randomdev.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 1999-2009 Apple, Inc. All rights reserved.
+ * Copyright (c) 1999-2006 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
@@ -26,17 +26,6 @@
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 
-/*
-	WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING!
-	
-	THIS FILE IS NEEDED TO PASS FIPS ACCEPTANCE FOR THE RANDOM NUMBER GENERATOR.
-	IF YOU ALTER IT IN ANY WAY, WE WILL NEED TO GO THOUGH FIPS ACCEPTANCE AGAIN,
-	AN OPERATION THAT IS VERY EXPENSIVE AND TIME CONSUMING.  IN OTHER WORDS,
-	DON'T MESS WITH THIS FILE.
-
-	WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING!
-*/
-
 #include <sys/param.h>
 #include <sys/systm.h>
 #include <sys/proc.h>
@@ -54,14 +43,10 @@
 
 #include <dev/random/randomdev.h>
 #include <dev/random/YarrowCoreLib/include/yarrow.h>
-
-#include <libkern/OSByteOrder.h>
-#include <libkern/OSAtomic.h>
+#include <libkern/crypto/sha1.h>
 
 #include <mach/mach_time.h>
 #include <machine/machine_routines.h>
-
-#include "fips_sha1.h"
 
 #define RANDOM_MAJOR  -1 /* let the kernel pick the device number */
 
@@ -89,41 +74,32 @@
 	0					/* type */
 };
 
-
-/*
-	WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING!
-
-	ANY CODE PROTECTED UNDER "#ifdef __arm__" IS SERIOUSLY SUPPOSED TO BE THERE!
-	IF YOU REMOVE ARM CODE, RANDOM WILL NOT MEAN ANYTHING FOR iPHONES ALL OVER.
-	PLEASE DON'T TOUCH __arm__ CODE IN THIS FILE!
-
-	WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING! WARNING!
-*/
-
-
 /* Used to detect whether we've already been initialized */
-static UInt8 gRandomInstalled = 0;
+static int gRandomInstalled = 0;
 static PrngRef gPrngRef;
 static int gRandomError = 1;
 static lck_grp_t *gYarrowGrp;
 static lck_attr_t *gYarrowAttr;
 static lck_grp_attr_t *gYarrowGrpAttr;
 static lck_mtx_t *gYarrowMutex = 0;
-static UInt8 gYarrowInitializationLock = 0;
+
+void CheckReseed(void);
 
 #define RESEED_TICKS 50 /* how long a reseed operation can take */
 
 
-typedef u_int8_t BlockWord;
-enum {kBSize = 20};
+enum {kBSizeInBits = 160}; // MUST be a multiple of 32!!!
+enum {kBSizeInBytes = kBSizeInBits / 8};
+typedef u_int32_t BlockWord;
+enum {kWordSizeInBits = 32};
+enum {kBSize = 5};
 typedef BlockWord Block[kBSize];
-enum {kBlockSize = sizeof(Block)};
 
 /* define prototypes to keep the compiler happy... */
 
 void add_blocks(Block a, Block b, BlockWord carry);
 void fips_initialize(void);
-void random_block(Block b, int addOptional);
+void random_block(Block b);
 u_int32_t CalculateCRC(u_int8_t* buffer, size_t length);
 
 /*
@@ -136,26 +112,26 @@
 void
 add_blocks(Block a, Block b, BlockWord carry)
 {
-	int i = kBlockSize - 1;
-	while (i >= 0)
+	int i = kBSize;
+	while (--i >= 0)
 	{
-		u_int32_t c = (u_int32_t)carry +
-					  (u_int32_t)a[i] +
-					  (u_int32_t)b[i];
-		a[i] = c & 0xff;
-		carry = c >> 8;
-		i -= 1;
+		u_int64_t c = (u_int64_t)carry +
+					  (u_int64_t)a[i] +
+					  (u_int64_t)b[i];
+		a[i] = c & ((1LL << kWordSizeInBits) - 1);
+		carry = c >> kWordSizeInBits;
 	}
 }
 
 
 
-static char zeros[(512 - kBSize * 8) / 8];
-static Block g_xkey;
-static Block g_random_data;
-static int g_bytes_used;
-static unsigned char g_SelfTestInitialized = 0;
-static u_int32_t gLastBlockChecksum;
+struct sha1_ctxt g_sha1_ctx;
+char zeros[(512 - kBSizeInBits) / 8];
+Block g_xkey;
+Block g_random_data;
+int g_bytes_used;
+unsigned char g_SelfTestInitialized = 0;
+u_int32_t gLastBlockChecksum;
 
 static const u_int32_t g_crc_table[] =
 {
@@ -218,52 +194,29 @@
  * get a random block of data per fips 186-2
  */
 void
-random_block(Block b, int addOptional)
-{
-	SHA1_CTX sha1_ctx;
-	
+random_block(Block b)
+{
 	int repeatCount = 0;
 	do
 	{
 		// do one iteration
-		
-		if (addOptional)
-		{
-			// create an xSeed to add.
-			Block xSeed;
-			prngOutput (gPrngRef, (BYTE*) &xSeed, sizeof (xSeed));
-			
-			// add the seed to the previous value of g_xkey
-			add_blocks (g_xkey, xSeed, 0);
-		}
-		
-		// initialize the value of H
-		FIPS_SHA1Init(&sha1_ctx);
-		
-		// to stay compatible with the FIPS specification, we need to flip the bytes in
-		// g_xkey to little endian byte order.  In our case, this makes exactly no difference
-		// (random is random), but we need to do it anyway to keep FIPS happy
-		
+		Block xSeed;
+		prngOutput (gPrngRef, (BYTE*) &xSeed, sizeof (xSeed));
+		
+		// add the seed to the previous value of g_xkey
+		add_blocks (g_xkey, xSeed, 0);
+
 		// compute "G"
-		FIPS_SHA1Update(&sha1_ctx, g_xkey, kBlockSize);
+		SHA1Update (&g_sha1_ctx, (const u_int8_t *) &g_xkey, sizeof (g_xkey));
 		
 		// add zeros to fill the internal SHA-1 buffer
-		FIPS_SHA1Update (&sha1_ctx, (const u_int8_t *)zeros, sizeof (zeros));
-		
-		// we have to do a byte order correction here because the sha1 math is being done internally
-		// as u_int32_t, not a stream of bytes.  Since we maintain our data as a byte stream, we need
-		// to convert
-		
-		u_int32_t* finger = (u_int32_t*) b;
-		
-		unsigned j;
-		for (j = 0; j < kBlockSize / sizeof (u_int32_t); ++j)
-		{
-			*finger++ = OSSwapHostToBigInt32(sha1_ctx.h.b32[j]);
-		}		
+		SHA1Update (&g_sha1_ctx, (const u_int8_t *)zeros, sizeof (zeros));
+		
+		// write the resulting block
+		memmove(b, g_sha1_ctx.h.b8, sizeof (Block));
 		
 		// calculate the CRC-32 of the block
-		u_int32_t new_crc = CalculateCRC(sha1_ctx.h.b8, sizeof (Block));
+		u_int32_t new_crc = CalculateCRC(g_sha1_ctx.h.b8, sizeof (Block));
 		
 		// make sure we don't repeat
 		int cmp = new_crc == gLastBlockChecksum;
@@ -309,27 +262,6 @@
 {
     prng_error_status perr;
 
-	/* Multiple threads can enter this as a result of an earlier
-	 * check of gYarrowMutex.  We make sure that only one of them
-	 * can enter at a time.  If one of them enters and discovers
-	 * that gYarrowMutex is no longer NULL, we know that another
-	 * thread has initialized the Yarrow state and we can exit.
-	 */
-	
-	/* The first thread that enters this function will find
-	 * gYarrowInitializationLock set to 0.  It will atomically
-	 * set the value to 1 and, seeing that it was zero, drop
-	 * out of the loop.  Other threads will see that the value is
-	 * 1 and continue to loop until we are initialized.
-     */
-
-	while (OSTestAndSet(0, &gYarrowInitializationLock)); /* serialize access to this function */
-	
-	if (gYarrowMutex) {
-		/*  we've already been initialized, clear and get out */
-		goto function_exit;
-	}
-
     /* create a Yarrow object */
     perr = prngInitialize(&gPrngRef);
     if (perr != 0) {
@@ -340,12 +272,11 @@
 	/* clear the error flag, reads and write should then work */
     gRandomError = 0;
 
+   {
     struct timeval tt;
     char buffer [16];
 
     /* get a little non-deterministic data as an initial seed. */
-	/* On OSX, securityd will add much more entropy as soon as it */
-	/* comes up.  On iOS, entropy is added with each system interrupt. */
     microtime(&tt);
 
     /*
@@ -359,7 +290,7 @@
     if (perr != 0) {
         /* an error, complain */
         printf ("Couldn't seed Yarrow.\n");
-        goto function_exit;
+        return;
     }
     
     /* turn the data around */
@@ -367,6 +298,7 @@
     
     /* and scramble it some more */
     perr = prngForceReseed(gPrngRef, RESEED_TICKS);
+    }
     
     /* make a mutex to control access */
     gYarrowGrpAttr = lck_grp_attr_alloc_init();
@@ -375,33 +307,21 @@
     gYarrowMutex   = lck_mtx_alloc_init(gYarrowGrp, gYarrowAttr);
 	
 	fips_initialize ();
-
-function_exit:
-	/* allow other threads to figure out whether or not we have been initialized. */
-	gYarrowInitializationLock = 0;
-}
-
-const Block kKnownAnswer = {0x92, 0xb4, 0x04, 0xe5, 0x56, 0x58, 0x8c, 0xed, 0x6c, 0x1a, 0xcd, 0x4e, 0xbf, 0x05, 0x3f, 0x68, 0x09, 0xf7, 0x3a, 0x93};
+}
 
 void
 fips_initialize(void)
 {
-	/* So that we can do the self test, set the seed to zero */
-	memset(&g_xkey, 0, sizeof(g_xkey));
+	/* Read the initial value of g_xkey from yarrow */
+	prngOutput (gPrngRef, (BYTE*) &g_xkey, sizeof (g_xkey));
+	
+	/* initialize our SHA1 generator */
+	SHA1Init (&g_sha1_ctx);
 	
 	/* other initializations */
 	memset (zeros, 0, sizeof (zeros));
 	g_bytes_used = 0;
-	random_block(g_random_data, FALSE);
-	
-	// check here to see if we got the initial data we were expecting
-	if (memcmp(kKnownAnswer, g_random_data, kBlockSize) != 0)
-	{
-		panic("FIPS random self test failed");
-	}
-	
-	// now do the random block again to make sure that userland doesn't get predicatable data
-	random_block(g_random_data, TRUE);
+	random_block(g_random_data);
 }
 
 /*
@@ -413,11 +333,16 @@
 {
 	int ret;
 
-	if (OSTestAndSet(0, &gRandomInstalled)) {
-		/* do this atomically so that it works correctly with
-		 multiple threads */
+	if (gRandomInstalled)
 		return;
-	}
+
+	/* install us in the file system */
+	gRandomInstalled = 1;
+
+#ifndef ARM_BOARD_CONFIG_S5L8900XFPGA_1136JFS
+	/* setup yarrow and the mutex */
+	PreliminarySetup();
+#endif
 
 	ret = cdevsw_add(RANDOM_MAJOR, &random_cdevsw);
 	if (ret < 0) {
@@ -435,9 +360,6 @@
 	 */
 	devfs_make_node(makedev (ret, 1), DEVFS_CHAR,
 		UID_ROOT, GID_WHEEL, 0666, "urandom", 0);
-
-	/* setup yarrow and the mutex if needed*/
-	PreliminarySetup();
 }
 
 int
@@ -517,6 +439,7 @@
 
     while (uio_resid(uio) > 0 && retCode == 0) {
         /* get the user's data */
+        // LP64todo - fix this!  uio_resid may be 64-bit value
         int bytesToInput = min(uio_resid(uio), sizeof (rdBuffer));
         retCode = uiomove(rdBuffer, bytesToInput, uio);
         if (retCode != 0)
@@ -558,18 +481,18 @@
    /* lock down the mutex */
     lck_mtx_lock(gYarrowMutex);
 
-
+    CheckReseed();
 	int bytes_remaining = uio_resid(uio);
     while (bytes_remaining > 0 && retCode == 0) {
         /* get the user's data */
 		int bytes_to_read = 0;
 		
-		int bytes_available = kBlockSize - g_bytes_used;
+		int bytes_available = kBSizeInBytes - g_bytes_used;
         if (bytes_available == 0)
 		{
-			random_block(g_random_data, TRUE);
+			random_block(g_random_data);
 			g_bytes_used = 0;
-			bytes_available = kBlockSize;
+			bytes_available = kBSizeInBytes;
 		}
 		
 		bytes_to_read = min (bytes_remaining, bytes_available);
@@ -595,20 +518,24 @@
 read_random(void* buffer, u_int numbytes)
 {
     if (gYarrowMutex == 0) { /* are we initialized? */
+#ifndef ARM_BOARD_CONFIG_S5L8900XFPGA_1136JFS
         PreliminarySetup ();
+#endif
     }
     
     lck_mtx_lock(gYarrowMutex);
+    CheckReseed();
+
 	int bytes_read = 0;
 
 	int bytes_remaining = numbytes;
     while (bytes_remaining > 0) {
-        int bytes_to_read = min(bytes_remaining, kBlockSize - g_bytes_used);
+        int bytes_to_read = min(bytes_remaining, kBSizeInBytes - g_bytes_used);
         if (bytes_to_read == 0)
 		{
-			random_block(g_random_data, TRUE);
+			random_block(g_random_data);
 			g_bytes_used = 0;
-			bytes_to_read = min(bytes_remaining, kBlockSize);
+			bytes_to_read = min(bytes_remaining, kBSizeInBytes);
 		}
 		
 		memmove ((u_int8_t*) buffer + bytes_read, ((u_int8_t*)g_random_data)+ g_bytes_used, bytes_to_read);
@@ -621,13 +548,17 @@
 }
 
 /*
- * Return an u_int32_t pseudo-random number.
- */
-u_int32_t
+ * Return an unsigned long pseudo-random number.
+ */
+u_long
 RandomULong(void)
 {
-	u_int32_t buf;
+	u_long buf;
 	read_random(&buf, sizeof (buf));
 	return (buf);
 }
 
+void
+CheckReseed(void)
+{
+}