xnu-3789.1.32.tar.gz

[apple/xnu.git] / pexpert / gen / pe_gen.c

diff --git a/pexpert/gen/pe_gen.c b/pexpert/gen/pe_gen.c
index 43ab61a9657461724091c1a0d8c8290c4fc12dc1..60664a6a0e9a7e8c5e0a68c2fc6804d9a99056e5 100644 (file)
--- a/pexpert/gen/pe_gen.c
+++ b/pexpert/gen/pe_gen.c
@@ -31,16 +31,54 @@
 
 #include <pexpert/pexpert.h>
 #include <pexpert/protos.h>
+#include <pexpert/device_tree.h>
 #include <kern/debug.h>
 
+
 static int DEBUGFlag;
 
+static uint32_t gPEKernelConfigurationBitmask;
+
 int32_t gPESerialBaud = -1;
 
 void pe_init_debug(void)
 {
-  if (!PE_parse_boot_argn("debug", &DEBUGFlag, sizeof (DEBUGFlag)))
-    DEBUGFlag = 0;
+       boolean_t boot_arg_value;
+
+       if (!PE_parse_boot_argn("debug", &DEBUGFlag, sizeof (DEBUGFlag)))
+               DEBUGFlag = 0;
+
+       gPEKernelConfigurationBitmask = 0;
+
+       if (!PE_parse_boot_argn("assertions", &boot_arg_value, sizeof(boot_arg_value))) {
+#if MACH_ASSERT
+               boot_arg_value = TRUE;
+#else
+               boot_arg_value = FALSE;
+#endif
+       }
+       gPEKernelConfigurationBitmask |= (boot_arg_value ? kPEICanHasAssertions : 0);
+
+       if (!PE_parse_boot_argn("statistics", &boot_arg_value, sizeof(boot_arg_value))) {
+#if DEVELOPMENT || DEBUG
+               boot_arg_value = TRUE;
+#else
+               boot_arg_value = FALSE;
+#endif
+       }
+       gPEKernelConfigurationBitmask |= (boot_arg_value ? kPEICanHasStatistics : 0);
+
+#if SECURE_KERNEL
+       boot_arg_value = FALSE;
+#else
+       if (!PE_i_can_has_debugger(NULL)) {
+               boot_arg_value = FALSE;
+       } else if (!PE_parse_boot_argn("diagnostic_api", &boot_arg_value, sizeof(boot_arg_value)))  {
+               boot_arg_value = TRUE;
+       }
+#endif
+       gPEKernelConfigurationBitmask |= (boot_arg_value ? kPEICanHasDiagnosticAPI : 0);
+
 }
 
 void PE_enter_debugger(const char *cause)
@@ -49,6 +87,12 @@ void PE_enter_debugger(const char *cause)
     Debugger(cause);
 }
 
+uint32_t
+PE_i_can_has_kernel_configuration(void)
+{
+       return gPEKernelConfigurationBitmask;
+}
+
 /* extern references */
 extern void vcattach(void);
 
@@ -64,6 +108,41 @@ void PE_init_printf(boolean_t vm_initialized)
   }
 }
 
+uint32_t
+PE_get_random_seed(unsigned char *dst_random_seed, uint32_t request_size)
+{
+       DTEntry         entryP;
+       uint32_t        size = 0;
+       void            *dt_random_seed;
+
+        if ((DTLookupEntry(NULL, "/chosen", &entryP) == kSuccess)
+           && (DTGetProperty(entryP, "random-seed",
+                               (void **)&dt_random_seed, &size) == kSuccess)) {
+               unsigned char *src_random_seed;
+               unsigned int i;
+               unsigned int null_count = 0;
+
+               src_random_seed = (unsigned char *)dt_random_seed;
+
+               if (size > request_size) size = request_size;
+
+               /*
+                * Copy from the device tree into the destination buffer,
+                * count the number of null bytes and null out the device tree.
+                */
+               for (i=0 ; i< size; i++, src_random_seed++, dst_random_seed++) {
+                       *dst_random_seed = *src_random_seed;
+                       null_count += *src_random_seed == (unsigned char)0;
+                       *src_random_seed = (unsigned char)0;
+               }
+               if (null_count == size)
+                       /* All nulls is no seed - return 0 */
+                       size = 0;
+       }
+
+       return(size);
+}
+
 unsigned char appleClut8[ 256 * 3 ] = {
 // 00
        0xFF,0xFF,0xFF, 0xFF,0xFF,0xCC, 0xFF,0xFF,0x99, 0xFF,0xFF,0x66,
@@ -146,3 +225,4 @@ unsigned char appleClut8[ 256 * 3 ] = {
        0xAA,0xAA,0xAA, 0x88,0x88,0x88, 0x77,0x77,0x77, 0x55,0x55,0x55,
        0x44,0x44,0x44, 0x22,0x22,0x22, 0x11,0x11,0x11, 0x00,0x00,0x00
 };
+