+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include <sys/conf.h>
+#include <sys/vnode.h>
+#include <sys/vnode_internal.h>
+#include <sys/fcntl.h>
+#include <IOKit/IOPolledInterface.h>
+#include <IOKit/IOBufferMemoryDescriptor.h>
+
+IOPolledFileIOVars * gIOPolledCoreFileVars;
+
+#if IOPOLLED_COREFILE
+
+static IOReturn
+IOOpenPolledCoreFile(const char * filename)
+{
+ IOReturn err;
+ unsigned int debug;
+
+ if (gIOPolledCoreFileVars) return (kIOReturnBusy);
+ if (!IOPolledInterface::gMetaClass.getInstanceCount()) return (kIOReturnUnsupported);
+
+ debug = 0;
+ PE_parse_boot_argn("debug", &debug, sizeof (debug));
+ if (DB_DISABLE_LOCAL_CORE & debug) return (kIOReturnUnsupported);
+
+ err = IOPolledFileOpen(filename, kIOCoreDumpSize, kIOCoreDumpFreeSize,
+ NULL, 0,
+ &gIOPolledCoreFileVars, NULL, NULL, 0);
+ if (kIOReturnSuccess != err) return (err);
+
+ err = IOPolledFilePollersSetup(gIOPolledCoreFileVars, kIOPolledPreflightCoreDumpState);
+ if (kIOReturnSuccess != err)
+ {
+ IOPolledFileClose(&gIOPolledCoreFileVars, NULL, NULL, 0, 0, 0);
+ }
+
+ return (err);
+}
+
+static void
+IOClosePolledCoreFile(void)
+{
+ IOPolledFilePollersClose(gIOPolledCoreFileVars, kIOPolledPostflightCoreDumpState);
+ IOPolledFileClose(&gIOPolledCoreFileVars, NULL, NULL, 0, 0, 0);
+}
+
+static thread_call_t gIOOpenPolledCoreFileTC;
+static IONotifier * gIOPolledCoreFileNotifier;
+static IONotifier * gIOPolledCoreFileInterestNotifier;
+
+static IOReturn
+KernelCoreMediaInterest(void * target, void * refCon,
+ UInt32 messageType, IOService * provider,
+ void * messageArgument, vm_size_t argSize )
+{
+ if (kIOMessageServiceIsTerminated == messageType)
+ {
+ gIOPolledCoreFileInterestNotifier->remove();
+ gIOPolledCoreFileInterestNotifier = 0;
+ IOClosePolledCoreFile();
+ }
+
+ return (kIOReturnSuccess);
+}
+
+static void
+OpenKernelCoreMedia(thread_call_param_t p0, thread_call_param_t p1)
+{
+ IOService * newService;
+ OSString * string;
+ char filename[16];
+
+ newService = (IOService *) p1;
+ do
+ {
+ if (gIOPolledCoreFileVars) break;
+ string = OSDynamicCast(OSString, newService->getProperty(kIOBSDNameKey));
+ if (!string) break;
+ snprintf(filename, sizeof(filename), "/dev/%s", string->getCStringNoCopy());
+ if (kIOReturnSuccess != IOOpenPolledCoreFile(filename)) break;
+ gIOPolledCoreFileInterestNotifier = newService->registerInterest(
+ gIOGeneralInterest, &KernelCoreMediaInterest, NULL, 0);
+ }
+ while (false);
+
+ newService->release();
+}
+
+static bool
+NewKernelCoreMedia(void * target, void * refCon,
+ IOService * newService,
+ IONotifier * notifier)
+{
+ static volatile UInt32 onlyOneCorePartition = 0;
+ do
+ {
+ if (!OSCompareAndSwap(0, 1, &onlyOneCorePartition)) break;
+ if (gIOPolledCoreFileVars) break;
+ if (!gIOOpenPolledCoreFileTC) break;
+ newService = newService->getProvider();
+ if (!newService) break;
+ newService->retain();
+ thread_call_enter1(gIOOpenPolledCoreFileTC, newService);
+ }
+ while (false);
+
+ return (false);
+}
+
+#endif /* IOPOLLED_COREFILE */
+
+extern "C" void
+IOBSDMountChange(struct mount * mp, uint32_t op)
+{
+#if IOPOLLED_COREFILE
+
+ OSDictionary * bsdMatching;
+ OSDictionary * mediaMatching;
+ OSString * string;
+
+ if (!gIOPolledCoreFileNotifier) do
+ {
+ if (!gIOOpenPolledCoreFileTC) gIOOpenPolledCoreFileTC = thread_call_allocate(&OpenKernelCoreMedia, NULL);
+ bsdMatching = IOService::serviceMatching("IOMediaBSDClient");
+ if (!bsdMatching) break;
+ mediaMatching = IOService::serviceMatching("IOMedia");
+ string = OSString::withCStringNoCopy("5361644D-6163-11AA-AA11-00306543ECAC");
+ if (!string || !mediaMatching) break;
+ mediaMatching->setObject("Content", string);
+ string->release();
+ bsdMatching->setObject(gIOParentMatchKey, mediaMatching);
+ mediaMatching->release();
+
+ gIOPolledCoreFileNotifier = IOService::addMatchingNotification(
+ gIOFirstMatchNotification, bsdMatching,
+ &NewKernelCoreMedia, NULL, NULL, -1000);
+ }
+ while (false);
+
+#endif /* IOPOLLED_COREFILE */
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+extern "C" boolean_t
+IOTaskHasEntitlement(task_t task, const char * entitlement)
+{
+ OSObject * obj;
+ obj = IOUserClient::copyClientEntitlement(task, entitlement);
+ if (!obj) return (false);
+ obj->release();
+ return (obj != kOSBooleanFalse);
+}
+