1/* 2 * Copyright (c) 2010 Apple Inc. All rights reserved. 3 * 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. The rights granted to you under the License 10 * may not be used to create, or enable the creation or redistribution of, 11 * unlawful or unlicensed copies of an Apple operating system, or to 12 * circumvent, violate, or enable the circumvention or violation of, any 13 * terms of an Apple operating system software license agreement. 14 * 15 * Please obtain a copy of the License at 16 * http://www.opensource.apple.com/apsl/ and read it before using this file. 17 * 18 * The Original Code and all software distributed under the License are 19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23 * Please see the License for the specific language governing rights and 24 * limitations under the License. 25 * 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27 */ 28 29#include <sys/cdefs.h> 30#include <stdbool.h> 31 32#include <IOKit/assert.h> 33#include <IOKit/system.h> 34#include <IOKit/IOLib.h> 35#include <IOKit/IOMemoryDescriptor.h> 36#include <IOKit/IOKitDebug.h> 37 38__BEGIN_DECLS 39 40#include <pexpert/pexpert.h> 41 42static volatile UInt32 alreadyFetched = 0; 43static IOMemoryDescriptor * newData; 44 45IOMemoryDescriptor * 46IOGetBootKeyStoreData(void); 47void 48IOSetKeyStoreData(IOMemoryDescriptor * data); 49 50__END_DECLS 51 52#if 1 53#define DEBG(fmt, args...) { kprintf(fmt, ## args); } 54#else 55#define DEBG(fmt, args...) {} 56#endif 57 58void 59IOSetKeyStoreData(IOMemoryDescriptor * data) 60{ 61 newData = data; 62 alreadyFetched = 0; 63} 64 65IOMemoryDescriptor * 66IOGetBootKeyStoreData(void) 67{ 68 IOMemoryDescriptor *memoryDescriptor; 69 boot_args *args = (boot_args *)PE_state.bootArgs; 70 IOOptionBits options; 71 IOAddressRange ranges; 72 73 if (!OSCompareAndSwap(0, 1, &alreadyFetched)) 74 return (NULL); 75 76 if (newData) 77 { 78 IOMemoryDescriptor * data = newData; 79 newData = NULL; 80 return (data); 81 } 82 83 DEBG("%s: data at address %u size %u\n", __func__, 84 args->keyStoreDataStart, 85 args->keyStoreDataSize); 86 87 if (args->keyStoreDataStart == 0) 88 return (NULL); 89 90 ranges.address = args->keyStoreDataStart; 91 ranges.length = args->keyStoreDataSize; 92 93 options = kIODirectionInOut | kIOMemoryTypePhysical64 | kIOMemoryMapperNone; 94 95 memoryDescriptor = IOMemoryDescriptor::withOptions(&ranges, 96 1, 97 0, 98 NULL, 99 options); 100 101 DEBG("%s: memory descriptor %p\n", __func__, memoryDescriptor); 102 103 return memoryDescriptor; 104} 105