Loading...
libkern/c++/OSRuntime.cpp xnu-201 xnu-4903.241.1
--- xnu/xnu-201/libkern/c++/OSRuntime.cpp
+++ xnu/xnu-4903.241.1/libkern/c++/OSRuntime.cpp
@@ -1,343 +1,587 @@
 /*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000,2008-2009 Apple Inc. All rights reserved.
  *
- * @APPLE_LICENSE_HEADER_START@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
- * The contents of this file constitute Original Code as defined in and
- * are subject to the Apple Public Source License Version 1.1 (the
- * "License").  You may not use this file except in compliance with the
- * License.  Please obtain a copy of the License at
- * http://www.apple.com/publicsource and read it before using this file.
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
  * 
- * This Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
- * License for the specific language governing rights and limitations
- * under the License.
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
  * 
- * @APPLE_LICENSE_HEADER_END@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /*
- * Copyright (c) 1997 Apple Computer, Inc.
+ * Copyright (c) 1997 Apple Inc.
  *
  */
 #include <libkern/c++/OSMetaClass.h>
+#include <libkern/c++/OSKext.h>
 #include <libkern/c++/OSLib.h>
 #include <libkern/c++/OSSymbol.h>
-#include <libkern/c++/OSBoolean.h>
+#include <IOKit/IOKitDebug.h>
 
 #include <sys/cdefs.h>
 
 __BEGIN_DECLS
 
 #include <string.h>
-
-struct mach_header;
-
 #include <mach/mach_types.h>
-#include <mach-o/mach_header.h>
+#include <libkern/kernel_mach_header.h>
+#include <libkern/prelink.h>
 #include <stdarg.h>
 
+#if PRAGMA_MARK
+#pragma mark Constants &c.
+#endif /* PRAGMA_MARK */
+OSKextLogSpec kOSRuntimeLogSpec =
+    kOSKextLogErrorLevel |
+    kOSKextLogLoadFlag |
+    kOSKextLogKextBookkeepingFlag;
+    
+#if PRAGMA_MARK
+#pragma mark Logging Bootstrap
+#endif /* PRAGMA_MARK */
+/*********************************************************************
+* kern_os Logging Bootstrap
+*
+* We can't call in to OSKext until the kernel's C++ environment is up
+* and running, so let's mask those references with a check variable.
+* We print unconditionally if C++ isn't up, but if that's the case
+* we've generally hit a serious error in kernel init!
+*********************************************************************/
+static bool gKernelCPPInitialized = false;
+
+#define OSRuntimeLog(kext, flags, format, args...)            \
+    do {                                                      \
+        if (gKernelCPPInitialized) {                          \
+            OSKextLog((kext), (flags), (format), ## args);  \
+        } else {                                              \
+            printf((format), ## args);                        \
+        }                                                     \
+    } while (0)
+
+#if PRAGMA_MARK
+#pragma mark kern_os Allocator Package
+#endif /* PRAGMA_MARK */
+/*********************************************************************
+* kern_os Allocator Package
+*********************************************************************/
+
+/*********************************************************************
+*********************************************************************/
 #if OSALLOCDEBUG
 extern int debug_iomalloc_size;
 #endif
 
-#define	MDECL(reqlen)					\
-typedef union {						\
-	struct	_mhead hdr;				\
-	char	_m[(reqlen) + sizeof (struct _mhead)];	\
-} hdr_t;						\
-hdr_t
-
-struct _mhead {
-	size_t	mlen;
-	char	dat[0];
+/*********************************************************************
+*********************************************************************/
+void *
+kern_os_malloc(size_t size)
+{
+    void *mem;
+    if (size == 0) {
+        return (0);
+    }
+
+    mem = kallocp_tag_bt((vm_size_t *)&size, VM_KERN_MEMORY_LIBKERN);
+    if (!mem) {
+        return (0);
+    }
+
+#if OSALLOCDEBUG
+    OSAddAtomic(size, &debug_iomalloc_size);
+#endif
+
+    bzero(mem, size);
+
+    return mem;
+}
+
+/*********************************************************************
+*********************************************************************/
+void
+kern_os_free(void * addr)
+{
+    size_t size;
+    size = kalloc_size(addr);
+#if OSALLOCDEBUG
+	OSAddAtomic(-size, &debug_iomalloc_size);
+#endif
+
+    kfree_addr(addr);
+}
+
+/*********************************************************************
+*********************************************************************/
+void *
+kern_os_realloc(
+    void   * addr,
+    size_t   nsize)
+{
+    void            *nmem;
+    size_t          osize;
+
+    if (!addr) {
+        return (kern_os_malloc(nsize));
+    }
+
+    osize = kalloc_size(addr);
+    if (nsize == osize) {
+        return (addr);
+    }
+
+    if (nsize == 0) {
+        kfree_addr(addr);
+        return (0);
+    }
+
+    nmem = kallocp_tag_bt((vm_size_t *)&nsize, VM_KERN_MEMORY_LIBKERN);
+    if (!nmem){
+        kfree_addr(addr);
+        return (0);
+    }
+
+#if OSALLOCDEBUG
+    OSAddAtomic((nsize - osize), &debug_iomalloc_size);
+#endif
+
+    if (nsize > osize) {
+        (void)memset((char *)nmem + osize, 0, nsize - osize);
+    }
+    (void)memcpy(nmem, addr, (nsize > osize) ? osize : nsize);
+    kfree_addr(addr);
+
+    return (nmem);
+}
+
+#if PRAGMA_MARK
+#pragma mark Libkern Init
+#endif /* PRAGMA_MARK */
+/*********************************************************************
+* Libkern Init
+*********************************************************************/
+
+#if __GNUC__ >= 3
+void __cxa_pure_virtual( void )    { panic("%s", __FUNCTION__); }
+#else
+void __pure_virtual( void )        { panic("%s", __FUNCTION__); }
+#endif
+
+extern lck_grp_t * IOLockGroup;
+extern kmod_info_t g_kernel_kmod_info;
+
+enum {
+    kOSSectionNamesDefault     = 0,
+    kOSSectionNamesBuiltinKext = 1,
+    kOSSectionNamesCount       = 2,
 };
-
-void *kern_os_malloc(
-	size_t		size)
-{
-	MDECL(size)	*mem;
-	size_t		memsize = sizeof (*mem);
-
-	if (size == 0)
-		return (0);
-
-	mem = (hdr_t *)kalloc(memsize);
-	if (!mem)
-		return (0);
-
-#if OSALLOCDEBUG
-	debug_iomalloc_size += memsize;
-#endif
-
-	mem->hdr.mlen = memsize;
-	(void) memset(mem->hdr.dat, 0, size);
-
-	return  (mem->hdr.dat);
-}
-
-void kern_os_free(
-	void		*addr)
-{
-	struct _mhead	*hdr;
-
-	if (!addr)
-		return;
-
-	hdr = (struct _mhead *) addr; hdr--;
-
-#if OSALLOCDEBUG
-	debug_iomalloc_size -= hdr->mlen;
-#endif
-
-#if 0
-	memset((vm_offset_t)hdr, 0xbb, hdr->mlen);
-#else
-	kfree((vm_offset_t)hdr, hdr->mlen);
-#endif
-}
-
-void *kern_os_realloc(
-	void		*addr,
-	size_t		nsize)
-{
-	struct _mhead	*ohdr;
-	MDECL(nsize)	*nmem;
-	size_t		nmemsize, osize;
-
-	if (!addr)
-		return (kern_os_malloc(nsize));
-
-	ohdr = (struct _mhead *) addr; ohdr--;
-	osize = ohdr->mlen - sizeof (*ohdr);
-	if (nsize == osize)
-		return (addr);
-
-	if (nsize == 0) {
-		kern_os_free(addr);
-		return (0);
-	}
-
-	nmemsize = sizeof (*nmem);
-	nmem = (hdr_t *) kalloc(nmemsize);
-	if (!nmem){
-		kern_os_free(addr);
-		return (0);
-	}
-
-#if OSALLOCDEBUG
-	debug_iomalloc_size += (nmemsize - ohdr->mlen);
-#endif
-
-	nmem->hdr.mlen = nmemsize;
-	if (nsize > osize)
-		(void) memset(&nmem->hdr.dat[osize], 0, nsize - osize);
-	(void) memcpy(nmem->hdr.dat, ohdr->dat,
-					(nsize > osize) ? osize : nsize);
-	kfree((vm_offset_t)ohdr, ohdr->mlen);
-
-	return (nmem->hdr.dat);
-}
-
-size_t kern_os_malloc_size(
-	void		*addr)
-{
-	struct _mhead	*hdr;
-
-	if (!addr)
-		return( 0);
-
-	hdr = (struct _mhead *) addr; hdr--;
-	return( hdr->mlen - sizeof (struct _mhead));
-}
-
-void __pure_virtual( void )	{ panic(__FUNCTION__); }
+enum {
+    kOSSectionNameInitializer = 0,
+    kOSSectionNameFinalizer   = 1,
+    kOSSectionNameCount       = 2
+};
+
+static const char *
+gOSStructorSectionNames[kOSSectionNamesCount][kOSSectionNameCount] = {
+    { SECT_MODINITFUNC,    SECT_MODTERMFUNC },
+    { kBuiltinInitSection, kBuiltinTermSection }
+};
+
+void OSlibkernInit(void)
+{
+    // This must be called before calling OSRuntimeInitializeCPP.
+    OSMetaClassBase::initialize();
+
+    g_kernel_kmod_info.address = (vm_address_t) &_mh_execute_header;
+    if (kOSReturnSuccess != OSRuntimeInitializeCPP(NULL)) {
+    // &g_kernel_kmod_info, gOSSectionNamesStandard, 0, 0)) {
+        panic("OSRuntime: C++ runtime failed to initialize.");
+    }
+
+    gKernelCPPInitialized = true;
+
+    return;
+}
+
+__END_DECLS
+
+#if PRAGMA_MARK
+#pragma mark C++ Runtime Load/Unload
+#endif /* PRAGMA_MARK */
+/*********************************************************************
+* kern_os C++ Runtime Load/Unload
+*********************************************************************/
+
 
 typedef void (*structor_t)(void);
 
-void OSRuntimeUnloadCPPForSegment(struct segment_command * segment) {
-
-    struct section * section;
+static bool
+OSRuntimeCallStructorsInSection(
+    OSKext                   * theKext,
+    kmod_info_t              * kmodInfo,
+    void                     * metaHandle,
+    kernel_segment_command_t * segment,
+    const char               * sectionName,
+    uintptr_t                  textStart,
+    uintptr_t                  textEnd)
+{
+    kernel_section_t * section;
+    bool result = TRUE;
 
     for (section = firstsect(segment);
-         section != 0;
-         section = nextsect(segment, section)) {
-
-        if (strcmp(section->sectname, "__destructor") == 0) {
-            structor_t * destructors = (structor_t *)section->addr;
-
-            if (destructors) {
-                int num_destructors = section->size / sizeof(structor_t);
-
-                for (int i = 0; i < num_destructors; i++) {
-                    (*destructors[i])();
+         section != NULL;
+         section = nextsect(segment, section))
+    {
+        if (strncmp(section->sectname, sectionName, sizeof(section->sectname) - 1)) continue;
+
+        structor_t * structors = (structor_t *)section->addr;
+        if (!structors) continue;
+
+        structor_t structor;
+        unsigned int num_structors = section->size / sizeof(structor_t);
+        unsigned int hit_null_structor = 0;
+        unsigned int firstIndex = 0;
+
+        if (textStart)
+        {
+            // bsearch for any in range
+            unsigned int baseIdx;
+            unsigned int lim;
+            uintptr_t value;
+            firstIndex = num_structors;
+            for (lim = num_structors, baseIdx = 0; lim; lim >>= 1)
+            {
+                value = (uintptr_t) structors[baseIdx + (lim >> 1)];
+                if (!value) panic("%s: null structor", kmodInfo->name);
+                if ((value >= textStart) && (value < textEnd))
+                {
+                    firstIndex = (baseIdx + (lim >> 1));
+                    // scan back for the first in range
+                    for (; firstIndex; firstIndex--)
+                    {
+                        value = (uintptr_t) structors[firstIndex - 1];
+                        if ((value < textStart) || (value >= textEnd)) break;
+                    }
+                    break;
                 }
-            } /* if (destructors) */
-        } /* if (strcmp...) */
+                if (textStart > value)
+                {
+                    // move right
+                    baseIdx += (lim >> 1) + 1;
+                    lim--;
+                }
+                // else move left
+            }
+            baseIdx = (baseIdx + (lim >> 1));
+        }
+        for (;
+             (firstIndex < num_structors)
+              && (!metaHandle || OSMetaClass::checkModLoad(metaHandle));
+             firstIndex++)
+        {
+            if ((structor = structors[firstIndex]))
+            {
+                if ((textStart && ((uintptr_t) structor < textStart))
+                 || (textEnd && ((uintptr_t) structor >= textEnd))) break;
+
+                (*structor)();
+            }
+            else if (!hit_null_structor)
+            {
+                hit_null_structor = 1;
+                OSRuntimeLog(theKext, kOSRuntimeLogSpec,
+                    "Null structor in kext %s segment %s!",
+                    kmodInfo->name, section->segname);
+            }
+        }
+        if (metaHandle) result = OSMetaClass::checkModLoad(metaHandle);
+        break;
     } /* for (section...) */
-
-    return;
-}
-
-void OSRuntimeUnloadCPP(kmod_info_t *ki, void *)
-{
-    if (ki && ki->address) {
-
-        struct segment_command * segment;
-        struct mach_header *header;
-
-	OSSymbol::checkForPageUnload((void *) ki->address,
-				     (void *) (ki->address + ki->size));
-
-        header = (struct mach_header *)ki->address;
-        segment = firstsegfromheader(header);
-
-        for (segment = firstsegfromheader(header);
-             segment != 0;
-             segment = nextseg(segment)) {
-
-            OSRuntimeUnloadCPPForSegment(segment);
-        }
-    }
-}
-
-kern_return_t OSRuntimeFinalizeCPP(kmod_info_t *ki, void *)
-{
-    void *metaHandle;
-
-    if (OSMetaClass::modHasInstance(ki->name)) {
-        // @@@ gvdl should have a verbose flag
-        printf("Can't unload %s due to -\n", ki->name);
-        OSMetaClass::reportModInstances(ki->name);
-        return kOSMetaClassHasInstances;
-    }
-
-    // Tell the meta class system that we are starting to unload
-    metaHandle = OSMetaClass::preModLoad(ki->name);
-    OSRuntimeUnloadCPP(ki, 0);	// Do the actual unload
-    (void) OSMetaClass::postModLoad(metaHandle);
-
-    return KMOD_RETURN_SUCCESS;
-}
-
-// Functions used by the extenTools/kmod library project
-kern_return_t OSRuntimeInitializeCPP(kmod_info_t *ki, void *)
-{
-    struct mach_header *header;
-    void *metaHandle;
-    bool load_success;
-    struct segment_command * segment;
-    struct segment_command * failure_segment;
-
-    if (!ki || !ki->address)
-        return KMOD_RETURN_FAILURE;
-    else
-        header = (struct mach_header *) ki->address;
-
-    // Tell the meta class system that we are starting the load
-    metaHandle = OSMetaClass::preModLoad(ki->name);
+    return (result);
+}
+
+/*********************************************************************
+*********************************************************************/
+kern_return_t
+OSRuntimeFinalizeCPP(
+    OSKext                   * theKext)
+{
+    kern_return_t              result = KMOD_RETURN_FAILURE;
+    void                     * metaHandle = NULL;  // do not free
+    kernel_mach_header_t     * header;
+    kernel_segment_command_t * segment;
+    kmod_info_t              * kmodInfo;
+    const char              ** sectionNames;
+    uintptr_t                  textStart;
+    uintptr_t                  textEnd;
+
+    textStart    = 0;
+    textEnd      = 0;
+    sectionNames = gOSStructorSectionNames[kOSSectionNamesDefault];
+    if (theKext) {
+        if (!theKext->isCPPInitialized()) {
+            result = KMOD_RETURN_SUCCESS;
+            goto finish;
+        }
+        kmodInfo = theKext->kmod_info;
+        if (!kmodInfo || !kmodInfo->address) {
+            result = kOSKextReturnInvalidArgument;
+            goto finish;
+        }
+        header = (kernel_mach_header_t *)kmodInfo->address;
+        if (theKext->flags.builtin) {
+            header       = (kernel_mach_header_t *)g_kernel_kmod_info.address;
+            textStart    = kmodInfo->address;
+            textEnd      = textStart + kmodInfo->size;
+            sectionNames = gOSStructorSectionNames[kOSSectionNamesBuiltinKext];
+        }
+    } else {
+        kmodInfo = &g_kernel_kmod_info;
+        header   = (kernel_mach_header_t *)kmodInfo->address;
+    }
+
+   /* OSKext checks for this condition now, but somebody might call
+    * this function directly (the symbol is exported....).
+    */
+    if (OSMetaClass::modHasInstance(kmodInfo->name)) {
+        // xxx - Don't log under errors? this is more of an info thing
+        OSRuntimeLog(theKext, kOSRuntimeLogSpec,
+            "Can't tear down kext %s C++; classes have instances:",
+            kmodInfo->name);
+        OSKext::reportOSMetaClassInstances(kmodInfo->name, kOSRuntimeLogSpec);
+        result = kOSMetaClassHasInstances;
+        goto finish;
+    }
+
+   /* Tell the meta class system that we are starting to unload.
+    * metaHandle isn't actually needed on the finalize path,
+    * so we don't check it here, even though OSMetaClass::postModLoad() will
+    * return a failure (it only does actual work on the init path anyhow).
+    */
+    metaHandle = OSMetaClass::preModLoad(kmodInfo->name);
+
+    OSSymbol::checkForPageUnload((void *)kmodInfo->address,
+        (void *)(kmodInfo->address + kmodInfo->size));
+
+    header = (kernel_mach_header_t *)kmodInfo->address;
+    segment = firstsegfromheader(header);
+
+    for (segment = firstsegfromheader(header);
+         segment != 0;
+         segment = nextsegfromheader(header, segment)) {
+
+        OSRuntimeCallStructorsInSection(theKext, kmodInfo, NULL, segment,
+                sectionNames[kOSSectionNameFinalizer], textStart, textEnd);
+    }
+
+    (void)OSMetaClass::postModLoad(metaHandle);
+
+    if (theKext) {
+        theKext->setCPPInitialized(false);
+    }
+    result = KMOD_RETURN_SUCCESS;
+finish:
+    return result;
+}
+
+/*********************************************************************
+*********************************************************************/
+kern_return_t
+OSRuntimeInitializeCPP(
+    OSKext                   * theKext)
+{
+    kern_return_t              result          = KMOD_RETURN_FAILURE;
+    kernel_mach_header_t     * header          = NULL;
+    void                     * metaHandle      = NULL;  // do not free
+    bool                       load_success    = true;
+    kernel_segment_command_t * segment         = NULL;  // do not free
+    kernel_segment_command_t * failure_segment = NULL;  // do not free
+    kmod_info_t             *  kmodInfo;
+    const char              ** sectionNames;
+    uintptr_t                  textStart;
+    uintptr_t                  textEnd;
+
+    textStart    = 0;
+    textEnd      = 0;
+    sectionNames = gOSStructorSectionNames[kOSSectionNamesDefault];
+    if (theKext) {
+        if (theKext->isCPPInitialized()) {
+            result = KMOD_RETURN_SUCCESS;
+            goto finish;
+        }
+
+        kmodInfo = theKext->kmod_info;
+        if (!kmodInfo || !kmodInfo->address) {
+            result = kOSKextReturnInvalidArgument;
+            goto finish;
+        }
+        header = (kernel_mach_header_t *)kmodInfo->address;
+
+        if (theKext->flags.builtin) {
+            header       = (kernel_mach_header_t *)g_kernel_kmod_info.address;
+            textStart    = kmodInfo->address;
+            textEnd      = textStart + kmodInfo->size;
+            sectionNames = gOSStructorSectionNames[kOSSectionNamesBuiltinKext];
+        }
+    } else {
+        kmodInfo = &g_kernel_kmod_info;
+        header   = (kernel_mach_header_t *)kmodInfo->address;
+    }
+
+   /* Tell the meta class system that we are starting the load
+    */
+    metaHandle = OSMetaClass::preModLoad(kmodInfo->name);
     assert(metaHandle);
-    if (!metaHandle)
-        return KMOD_RETURN_FAILURE;
-
-    load_success = true;
-    failure_segment = 0;
-
-   /* Scan the header for all sections named "__constructor", in any
+    if (!metaHandle) {
+        goto finish;
+    }
+
+   /* NO GOTO PAST HERE. */
+
+   /* Scan the header for all constructor sections, in any
     * segment, and invoke the constructors within those sections.
     */
     for (segment = firstsegfromheader(header);
-         segment != 0 && load_success;
-         segment = nextseg(segment)) {
-
-        struct section * section;
-
+         segment != NULL && load_success;
+         segment = nextsegfromheader(header, segment))
+    {
        /* Record the current segment in the event of a failure.
         */
         failure_segment = segment;
-
-        for (section = firstsect(segment);
-             section != 0 && load_success;
-             section = nextsect(segment, section)) {
-
-            if (strcmp(section->sectname, "__constructor") == 0) {
-                structor_t * constructors = (structor_t *)section->addr;
-
-                if (constructors) {
-                    // FIXME: can we break here under the assumption that
-                    // section names are unique within a segment?
-
-                    int num_constructors = section->size / sizeof(structor_t);
-                    int hit_null_constructor = 0;
-
-                    for (int i = 0;
-                         i < num_constructors &&
-                         OSMetaClass::checkModLoad(metaHandle);
-                         i++) {
-
-                        if (constructors[i]) {
-                            (*constructors[i])();
-                        } else if (!hit_null_constructor) {
-                            hit_null_constructor = 1;
-                            printf("Error! Null constructor in segment %s.\n",
-                                section->segname);
-                        }
-                    }
-                    load_success = OSMetaClass::checkModLoad(metaHandle);
-
-                } /* if (constructors) */
-            } /* if (strcmp...) */
-        } /* for (section...) */
+        load_success = OSRuntimeCallStructorsInSection(
+                            theKext, kmodInfo, metaHandle, segment,
+                            sectionNames[kOSSectionNameInitializer],
+                            textStart, textEnd);
     } /* for (segment...) */
 
-
-    // We failed so call all of the destructors
+   /* We failed so call all of the destructors. We must do this before
+    * calling OSMetaClass::postModLoad() as the OSMetaClass destructors
+    * will alter state (in the metaHandle) used by that function.
+    */
     if (!load_success) {
 
-       /* Scan the header for all sections named "__constructor", in any
+       /* Scan the header for all destructor sections, in any
         * segment, and invoke the constructors within those sections.
         */
         for (segment = firstsegfromheader(header);
              segment != failure_segment && segment != 0;
-             segment = nextseg(segment)) {
-
-            OSRuntimeUnloadCPPForSegment(segment);
+             segment = nextsegfromheader(header, segment)) {
+
+            OSRuntimeCallStructorsInSection(theKext, kmodInfo, NULL, segment,
+                    sectionNames[kOSSectionNameFinalizer], textStart, textEnd);
 
         } /* for (segment...) */
     }
 
-    return OSMetaClass::postModLoad(metaHandle);
-}
-
-static KMOD_LIB_DECL(__kernel__, 0);
-void OSlibkernInit(void)
-{
-    vm_address_t *headerArray = (vm_address_t *) getmachheaders();
-
-    KMOD_INFO_NAME.address = headerArray[0]; assert(!headerArray[1]);
-    if (kOSReturnSuccess != OSRuntimeInitializeCPP(&KMOD_INFO_NAME, 0))
-        panic("OSRuntime: C++ runtime failed to initialize");
-
-    OSBoolean::initialize();
-}
-
-__END_DECLS
-
-void * operator new( size_t size)
+   /* Now, regardless of success so far, do the post-init registration
+    * and cleanup. If we had to call the unloadCPP function, static
+    * destructors have removed classes from the stalled list so no
+    * metaclasses will actually be registered.
+    */
+    result = OSMetaClass::postModLoad(metaHandle);
+
+   /* If we've otherwise been fine up to now, but OSMetaClass::postModLoad()
+    * fails (typically due to a duplicate class), tear down all the C++
+    * stuff from the kext. This isn't necessary for libkern/OSMetaClass stuff,
+    * but may be necessary for other C++ code. We ignore the return value
+    * because it's only a fail when there are existing instances of libkern
+    * classes, and there had better not be any created on the C++ init path.
+    */
+    if (load_success && result != KMOD_RETURN_SUCCESS) {
+        (void)OSRuntimeFinalizeCPP(theKext); //kmodInfo, sectionNames, textStart, textEnd);
+    }
+
+    if (theKext && load_success && result == KMOD_RETURN_SUCCESS) {
+        theKext->setCPPInitialized(true);
+    }
+finish:
+    return result;
+}
+
+/*********************************************************************
+Unload a kernel segment.
+*********************************************************************/
+
+void
+OSRuntimeUnloadCPPForSegment(
+    kernel_segment_command_t * segment)
+{
+    OSRuntimeCallStructorsInSection(NULL, &g_kernel_kmod_info, NULL, segment,
+                    gOSStructorSectionNames[kOSSectionNamesDefault][kOSSectionNameFinalizer], 0, 0);
+}
+
+#if PRAGMA_MARK
+#pragma mark C++ Allocators & Deallocators
+#endif /* PRAGMA_MARK */
+/*********************************************************************
+* C++ Allocators & Deallocators
+*********************************************************************/
+void *
+operator new(size_t size)
 {
     void * result;
 
-    result = (void *) kern_os_malloc( size);
-    if( result)
-	bzero( result, size);
-    return( result);
-}
-
-void operator delete( void * addr)
-{
-    kern_os_free( addr);
-}
-
+    result = (void *) kern_os_malloc(size);
+    return result;
+}
+
+void
+operator delete(void * addr)
+#if __cplusplus >= 201103L
+								noexcept
+#endif
+{
+    kern_os_free(addr);
+    return;
+}
+
+void *
+operator new[](unsigned long sz)
+{
+    if (sz == 0) sz = 1;
+    return kern_os_malloc(sz);
+}
+
+void
+operator delete[](void * ptr)
+#if __cplusplus >= 201103L
+								noexcept
+#endif
+{
+    if (ptr) {
+        kern_os_free(ptr);
+    }
+    return;
+}
+
+/* PR-6481964 - The compiler is going to check for size overflows in calls to
+ * new[], and if there is an overflow, it will call __throw_length_error.
+ * This is an unrecoverable error by the C++ standard, so we must panic here.
+ *
+ * We have to put the function inside the std namespace because of how the
+ * compiler expects the name to be mangled.
+ */
+namespace std {
+
+void
+__throw_length_error(const char *msg __unused)
+{
+    panic("Size of array created by new[] has overflowed");
+}
+
+};