]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/dev/dtrace/profile_prvd.c
xnu-7195.101.1.tar.gz
[apple/xnu.git] / bsd / dev / dtrace / profile_prvd.c
index 60ebf9bd7db873c98768b34397e766c7214733c4..450a69f4b0c5aacfab619d97e69b2dacb3aae9af 100644 (file)
  * Use is subject to license terms.
  */
 
  * Use is subject to license terms.
  */
 
-/* #pragma ident       "@(#)profile.c  1.7     07/01/10 SMI" */
-
-#if !defined(__APPLE__)
-#include <sys/errno.h>
-#include <sys/stat.h>
-#include <sys/modctl.h>
-#include <sys/conf.h>
-#include <sys/systm.h>
-#include <sys/ddi.h>
-#include <sys/sunddi.h>
-#include <sys/cpuvar.h>
-#include <sys/kmem.h>
-#include <sys/strsubr.h>
-#include <sys/dtrace.h>
-#include <sys/cyclic.h>
-#include <sys/atomic.h>
-#else
-#ifdef KERNEL
-#ifndef _KERNEL
-#define _KERNEL /* Solaris vs. Darwin */
-#endif
-#endif
-
 #include <kern/cpu_data.h>
 #include <kern/thread.h>
 #include <kern/assert.h>
 #include <kern/cpu_data.h>
 #include <kern/thread.h>
 #include <kern/assert.h>
@@ -69,6 +46,8 @@
 
 #if defined(__x86_64__)
 extern x86_saved_state_t *find_kern_regs(thread_t);
 
 #if defined(__x86_64__)
 extern x86_saved_state_t *find_kern_regs(thread_t);
+#elif defined (__arm__) || defined(__arm64__)
+extern struct arm_saved_state *find_kern_regs(thread_t);
 #else
 #error Unknown architecture
 #endif
 #else
 #error Unknown architecture
 #endif
@@ -77,9 +56,7 @@ extern x86_saved_state_t *find_kern_regs(thread_t);
 #define ASSERT(x) do {} while(0)
 
 extern void profile_init(void);
 #define ASSERT(x) do {} while(0)
 
 extern void profile_init(void);
-#endif /* __APPLE__ */
 
 
-static dev_info_t *profile_devi;
 static dtrace_provider_id_t profile_id;
 
 /*
 static dtrace_provider_id_t profile_id;
 
 /*
@@ -99,7 +76,7 @@ static dtrace_provider_id_t profile_id;
  * On SPARC, the picture is further complicated because the compiler
  * optimizes away tail-calls -- so the following frames are optimized away:
  *
  * On SPARC, the picture is further complicated because the compiler
  * optimizes away tail-calls -- so the following frames are optimized away:
  *
- *     profile_fire
+ *      profile_fire
  *     cyclic_expire
  *
  * This gives three frames.  However, on DEBUG kernels, the cyclic_expire
  *     cyclic_expire
  *
  * This gives three frames.  However, on DEBUG kernels, the cyclic_expire
@@ -114,45 +91,47 @@ static dtrace_provider_id_t profile_id;
 
 #if defined(__x86_64__)
 #define PROF_ARTIFICIAL_FRAMES  9
 
 #if defined(__x86_64__)
 #define PROF_ARTIFICIAL_FRAMES  9
+#elif defined(__arm__) || defined(__arm64__)
+#define PROF_ARTIFICIAL_FRAMES 8
 #else
 #error Unknown architecture
 #endif
 
 #else
 #error Unknown architecture
 #endif
 
-#define        PROF_NAMELEN            15
+#define PROF_NAMELEN            15
 
 
-#define        PROF_PROFILE            0
-#define        PROF_TICK               1
-#define        PROF_PREFIX_PROFILE     "profile-"
-#define        PROF_PREFIX_TICK        "tick-"
+#define PROF_PROFILE            0
+#define PROF_TICK               1
+#define PROF_PREFIX_PROFILE     "profile-"
+#define PROF_PREFIX_TICK        "tick-"
 
 typedef struct profile_probe {
 
 typedef struct profile_probe {
-       char            prof_name[PROF_NAMELEN];
-       dtrace_id_t     prof_id;
-       int             prof_kind;
-       hrtime_t        prof_interval;
-       cyclic_id_t     prof_cyclic;
+       char            prof_name[PROF_NAMELEN];
+       dtrace_id_t     prof_id;
+       int             prof_kind;
+       hrtime_t        prof_interval;
+       cyclic_id_t     prof_cyclic;
 } profile_probe_t;
 
 typedef struct profile_probe_percpu {
 } profile_probe_t;
 
 typedef struct profile_probe_percpu {
-       hrtime_t        profc_expected;
-       hrtime_t        profc_interval;
-       profile_probe_t *profc_probe;
+       hrtime_t        profc_expected;
+       hrtime_t        profc_interval;
+       profile_probe_t *profc_probe;
 } profile_probe_percpu_t;
 
 } profile_probe_percpu_t;
 
-hrtime_t       profile_interval_min = NANOSEC / 5000;          /* 5000 hz */
-int            profile_aframes = 0;                            /* override */
+hrtime_t        profile_interval_min = NANOSEC / 5000;          /* 5000 hz */
+int             profile_aframes = 0;                            /* override */
 
 static int profile_rates[] = {
 
 static int profile_rates[] = {
-    97, 199, 499, 997, 1999,
-    4001, 4999, 0, 0, 0,
-    0, 0, 0, 0, 0,
-    0, 0, 0, 0, 0
+       97, 199, 499, 997, 1999,
+       4001, 4999, 0, 0, 0,
+       0, 0, 0, 0, 0,
+       0, 0, 0, 0, 0
 };
 
 static int profile_ticks[] = {
 };
 
 static int profile_ticks[] = {
-    1, 10, 100, 500, 1000,
-    5000, 0, 0, 0, 0,
-    0, 0, 0, 0, 0
+       1, 10, 100, 500, 1000,
+       5000, 0, 0, 0, 0,
+       0, 0, 0, 0, 0
 };
 
 /*
 };
 
 /*
@@ -162,9 +141,9 @@ static int profile_ticks[] = {
  * this gets its value from PROFILE_MAX_DEFAULT or profile-max-probes if it's
  * present in the profile.conf file.
  */
  * this gets its value from PROFILE_MAX_DEFAULT or profile-max-probes if it's
  * present in the profile.conf file.
  */
-#define        PROFILE_MAX_DEFAULT     1000    /* default max. number of probes */
-static uint32_t profile_max;           /* maximum number of profile probes */
-static uint32_t profile_total; /* current number of profile probes */
+#define PROFILE_MAX_DEFAULT     1000    /* default max. number of probes */
+static uint32_t profile_max;            /* maximum number of profile probes */
+static uint32_t profile_total;  /* current number of profile probes */
 
 static void
 profile_fire(void *arg)
 
 static void
 profile_fire(void *arg)
@@ -176,17 +155,12 @@ profile_fire(void *arg)
        late = dtrace_gethrtime() - pcpu->profc_expected;
        pcpu->profc_expected += pcpu->profc_interval;
 
        late = dtrace_gethrtime() - pcpu->profc_expected;
        pcpu->profc_expected += pcpu->profc_interval;
 
-#if !defined(__APPLE__)
-       dtrace_probe(prof->prof_id, CPU->cpu_profile_pc,
-           CPU->cpu_profile_upc, late, 0, 0);
-#else
 #if defined(__x86_64__)
        x86_saved_state_t *kern_regs = find_kern_regs(current_thread());
 
        if (NULL != kern_regs) {
                /* Kernel was interrupted. */
 #if defined(__x86_64__)
        x86_saved_state_t *kern_regs = find_kern_regs(current_thread());
 
        if (NULL != kern_regs) {
                /* Kernel was interrupted. */
-               dtrace_probe(prof->prof_id, saved_state64(kern_regs)->isf.rip,  0x0, 0, 0, 0);
-
+               dtrace_probe(prof->prof_id, saved_state64(kern_regs)->isf.rip, 0x0, late, 0, 0);
        } else {
                pal_register_cache_state(current_thread(), VALID);
                /* Possibly a user interrupt */
        } else {
                pal_register_cache_state(current_thread(), VALID);
                /* Possibly a user interrupt */
@@ -195,21 +169,64 @@ profile_fire(void *arg)
                if (NULL == tagged_regs) {
                        /* Too bad, so sad, no useful interrupt state. */
                        dtrace_probe(prof->prof_id, 0xcafebabe,
                if (NULL == tagged_regs) {
                        /* Too bad, so sad, no useful interrupt state. */
                        dtrace_probe(prof->prof_id, 0xcafebabe,
-                       0x0, 0, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                           0x0, late, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
                } else if (is_saved_state64(tagged_regs)) {
                        x86_saved_state64_t *regs = saved_state64(tagged_regs);
 
                } else if (is_saved_state64(tagged_regs)) {
                        x86_saved_state64_t *regs = saved_state64(tagged_regs);
 
-                       dtrace_probe(prof->prof_id, 0x0, regs->isf.rip, 0, 0, 0);
+                       dtrace_probe(prof->prof_id, 0x0, regs->isf.rip, late, 0, 0);
                } else {
                        x86_saved_state32_t *regs = saved_state32(tagged_regs);
 
                } else {
                        x86_saved_state32_t *regs = saved_state32(tagged_regs);
 
-                       dtrace_probe(prof->prof_id, 0x0, regs->eip, 0, 0, 0);
+                       dtrace_probe(prof->prof_id, 0x0, regs->eip, late, 0, 0);
+               }
+       }
+#elif defined(__arm__)
+       {
+               arm_saved_state_t *arm_kern_regs = (arm_saved_state_t *) find_kern_regs(current_thread());
+
+               // We should only come in here from interrupt context, so we should always have valid kernel regs
+               assert(NULL != arm_kern_regs);
+
+               if (arm_kern_regs->cpsr & 0xF) {
+                       /* Kernel was interrupted. */
+                       dtrace_probe(prof->prof_id, arm_kern_regs->pc, 0x0, late, 0, 0);
+               } else {
+                       /* Possibly a user interrupt */
+                       arm_saved_state_t   *arm_user_regs = (arm_saved_state_t *)find_user_regs(current_thread());
+
+                       if (NULL == arm_user_regs) {
+                               /* Too bad, so sad, no useful interrupt state. */
+                               dtrace_probe(prof->prof_id, 0xcafebabe, 0x0, late, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                       } else {
+                               dtrace_probe(prof->prof_id, 0x0, arm_user_regs->pc, late, 0, 0);
+                       }
+               }
+       }
+#elif defined(__arm64__)
+       {
+               arm_saved_state_t *arm_kern_regs = (arm_saved_state_t *) find_kern_regs(current_thread());
+
+               // We should only come in here from interrupt context, so we should always have valid kernel regs
+               assert(NULL != arm_kern_regs);
+
+               if (saved_state64(arm_kern_regs)->cpsr & 0xF) {
+                       /* Kernel was interrupted. */
+                       dtrace_probe(prof->prof_id, saved_state64(arm_kern_regs)->pc, 0x0, late, 0, 0);
+               } else {
+                       /* Possibly a user interrupt */
+                       arm_saved_state_t   *arm_user_regs = (arm_saved_state_t *)find_user_regs(current_thread());
+
+                       if (NULL == arm_user_regs) {
+                               /* Too bad, so sad, no useful interrupt state. */
+                               dtrace_probe(prof->prof_id, 0xcafebabe, 0x0, late, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                       } else {
+                               dtrace_probe(prof->prof_id, 0x0, get_saved_state_pc(arm_user_regs), late, 0, 0);
+                       }
                }
        }
 #else
 #error Unknown architecture
 #endif
                }
        }
 #else
 #error Unknown architecture
 #endif
-#endif /* __APPLE__ */
 }
 
 static void
 }
 
 static void
@@ -217,16 +234,12 @@ profile_tick(void *arg)
 {
        profile_probe_t *prof = arg;
 
 {
        profile_probe_t *prof = arg;
 
-#if !defined(__APPLE__)
-       dtrace_probe(prof->prof_id, CPU->cpu_profile_pc,
-           CPU->cpu_profile_upc, 0, 0, 0);
-#else
 #if defined(__x86_64__)
        x86_saved_state_t *kern_regs = find_kern_regs(current_thread());
 
        if (NULL != kern_regs) {
                /* Kernel was interrupted. */
 #if defined(__x86_64__)
        x86_saved_state_t *kern_regs = find_kern_regs(current_thread());
 
        if (NULL != kern_regs) {
                /* Kernel was interrupted. */
-               dtrace_probe(prof->prof_id, saved_state64(kern_regs)->isf.rip,  0x0, 0, 0, 0);
+               dtrace_probe(prof->prof_id, saved_state64(kern_regs)->isf.rip, 0x0, 0, 0, 0);
        } else {
                pal_register_cache_state(current_thread(), VALID);
                /* Possibly a user interrupt */
        } else {
                pal_register_cache_state(current_thread(), VALID);
                /* Possibly a user interrupt */
@@ -235,7 +248,7 @@ profile_tick(void *arg)
                if (NULL == tagged_regs) {
                        /* Too bad, so sad, no useful interrupt state. */
                        dtrace_probe(prof->prof_id, 0xcafebabe,
                if (NULL == tagged_regs) {
                        /* Too bad, so sad, no useful interrupt state. */
                        dtrace_probe(prof->prof_id, 0xcafebabe,
-                       0x0, 0, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                           0x0, 0, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
                } else if (is_saved_state64(tagged_regs)) {
                        x86_saved_state64_t *regs = saved_state64(tagged_regs);
 
                } else if (is_saved_state64(tagged_regs)) {
                        x86_saved_state64_t *regs = saved_state64(tagged_regs);
 
@@ -246,10 +259,48 @@ profile_tick(void *arg)
                        dtrace_probe(prof->prof_id, 0x0, regs->eip, 0, 0, 0);
                }
        }
                        dtrace_probe(prof->prof_id, 0x0, regs->eip, 0, 0, 0);
                }
        }
+#elif defined(__arm__)
+       {
+               arm_saved_state_t *arm_kern_regs = (arm_saved_state_t *) find_kern_regs(current_thread());
+
+               if (NULL != arm_kern_regs) {
+                       /* Kernel was interrupted. */
+                       dtrace_probe(prof->prof_id, arm_kern_regs->pc, 0x0, 0, 0, 0);
+               } else {
+                       /* Possibly a user interrupt */
+                       arm_saved_state_t   *arm_user_regs = (arm_saved_state_t *)find_user_regs(current_thread());
+
+                       if (NULL == arm_user_regs) {
+                               /* Too bad, so sad, no useful interrupt state. */
+                               dtrace_probe(prof->prof_id, 0xcafebabe, 0x0, 0, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                       } else {
+                               dtrace_probe(prof->prof_id, 0x0, arm_user_regs->pc, 0, 0, 0);
+                       }
+               }
+       }
+#elif defined(__arm64__)
+       {
+               arm_saved_state_t *arm_kern_regs = (arm_saved_state_t *) find_kern_regs(current_thread());
+
+               if (NULL != arm_kern_regs) {
+                       /* Kernel was interrupted. */
+                       dtrace_probe(prof->prof_id, saved_state64(arm_kern_regs)->pc, 0x0, 0, 0, 0);
+               } else {
+                       /* Possibly a user interrupt */
+                       arm_saved_state_t   *arm_user_regs = (arm_saved_state_t *)find_user_regs(current_thread());
+
+                       if (NULL == arm_user_regs) {
+                               /* Too bad, so sad, no useful interrupt state. */
+                               dtrace_probe(prof->prof_id, 0xcafebabe, 0x0, 0, 0, 0); /* XXX_BOGUS also see profile_usermode() below. */
+                       } else {
+                               dtrace_probe(prof->prof_id, 0x0, get_saved_state_pc(arm_user_regs), 0, 0, 0);
+                       }
+               }
+       }
+
 #else
 #error Unknown architecture
 #endif
 #else
 #error Unknown architecture
 #endif
-#endif /* __APPLE__ */
 }
 
 static void
 }
 
 static void
@@ -257,26 +308,26 @@ profile_create(hrtime_t interval, const char *name, int kind)
 {
        profile_probe_t *prof;
 
 {
        profile_probe_t *prof;
 
-       if (interval < profile_interval_min)
+       if (interval < profile_interval_min) {
                return;
                return;
+       }
 
 
-       if (dtrace_probe_lookup(profile_id, NULL, NULL, name) != 0)
+       if (dtrace_probe_lookup(profile_id, NULL, NULL, name) != 0) {
                return;
                return;
+       }
 
 
-       atomic_add_32(&profile_total, 1);
+       os_atomic_inc(&profile_total, relaxed);
        if (profile_total > profile_max) {
        if (profile_total > profile_max) {
-               atomic_add_32(&profile_total, -1);
+               os_atomic_dec(&profile_total, relaxed);
                return;
        }
 
                return;
        }
 
-#if !defined(__APPLE__)
-       prof = kmem_zalloc(sizeof (profile_probe_t), KM_SLEEP);
-#else
-       if (PROF_TICK == kind)
-               prof = kmem_zalloc(sizeof (profile_probe_t), KM_SLEEP);
-       else
-               prof = kmem_zalloc(sizeof (profile_probe_t) + NCPU*sizeof(profile_probe_percpu_t), KM_SLEEP);
-#endif /* __APPLE__ */
+       if (PROF_TICK == kind) {
+               prof = kmem_zalloc(sizeof(profile_probe_t), KM_SLEEP);
+       } else {
+               prof = kmem_zalloc(sizeof(profile_probe_t) + NCPU * sizeof(profile_probe_percpu_t), KM_SLEEP);
+       }
+
        (void) strlcpy(prof->prof_name, name, sizeof(prof->prof_name));
        prof->prof_interval = interval;
        prof->prof_cyclic = CYCLIC_NONE;
        (void) strlcpy(prof->prof_name, name, sizeof(prof->prof_name));
        prof->prof_interval = interval;
        prof->prof_cyclic = CYCLIC_NONE;
@@ -295,38 +346,6 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
        hrtime_t val = 0, mult = 1, len;
        const char *name, *suffix = NULL;
 
        hrtime_t val = 0, mult = 1, len;
        const char *name, *suffix = NULL;
 
-#if !defined(__APPLE__)
-       const struct {
-               char *prefix;
-               int kind;
-       } types[] = {
-               { PROF_PREFIX_PROFILE, PROF_PROFILE },
-               { PROF_PREFIX_TICK, PROF_TICK },
-               { NULL, NULL }
-       };
-
-       const struct {
-               char *name;
-               hrtime_t mult;
-       } suffixes[] = {
-               { "ns",         NANOSEC / NANOSEC },
-               { "nsec",       NANOSEC / NANOSEC },
-               { "us",         NANOSEC / MICROSEC },
-               { "usec",       NANOSEC / MICROSEC },
-               { "ms",         NANOSEC / MILLISEC },
-               { "msec",       NANOSEC / MILLISEC },
-               { "s",          NANOSEC / SEC },
-               { "sec",        NANOSEC / SEC },
-               { "m",          NANOSEC * (hrtime_t)60 },
-               { "min",        NANOSEC * (hrtime_t)60 },
-               { "h",          NANOSEC * (hrtime_t)(60 * 60) },
-               { "hour",       NANOSEC * (hrtime_t)(60 * 60) },
-               { "d",          NANOSEC * (hrtime_t)(24 * 60 * 60) },
-               { "day",        NANOSEC * (hrtime_t)(24 * 60 * 60) },
-               { "hz",         0 },
-               { NULL }
-       };
-#else
        const struct {
                const char *prefix;
                int kind;
        const struct {
                const char *prefix;
                int kind;
@@ -340,25 +359,23 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
                const char *name;
                hrtime_t mult;
        } suffixes[] = {
                const char *name;
                hrtime_t mult;
        } suffixes[] = {
-               { "ns",         NANOSEC / NANOSEC },
-               { "nsec",       NANOSEC / NANOSEC },
-               { "us",         NANOSEC / MICROSEC },
-               { "usec",       NANOSEC / MICROSEC },
-               { "ms",         NANOSEC / MILLISEC },
-               { "msec",       NANOSEC / MILLISEC },
-               { "s",          NANOSEC / SEC },
-               { "sec",        NANOSEC / SEC },
-               { "m",          NANOSEC * (hrtime_t)60 },
-               { "min",        NANOSEC * (hrtime_t)60 },
-               { "h",          NANOSEC * (hrtime_t)(60 * 60) },
-               { "hour",       NANOSEC * (hrtime_t)(60 * 60) },
-               { "d",          NANOSEC * (hrtime_t)(24 * 60 * 60) },
-               { "day",        NANOSEC * (hrtime_t)(24 * 60 * 60) },
-               { "hz",         0 },
+               { "ns", NANOSEC / NANOSEC },
+               { "nsec", NANOSEC / NANOSEC },
+               { "us", NANOSEC / MICROSEC },
+               { "usec", NANOSEC / MICROSEC },
+               { "ms", NANOSEC / MILLISEC },
+               { "msec", NANOSEC / MILLISEC },
+               { "s", NANOSEC / SEC },
+               { "sec", NANOSEC / SEC },
+               { "m", NANOSEC * (hrtime_t)60 },
+               { "min", NANOSEC * (hrtime_t)60 },
+               { "h", NANOSEC * (hrtime_t)(60 * 60) },
+               { "hour", NANOSEC * (hrtime_t)(60 * 60) },
+               { "d", NANOSEC * (hrtime_t)(24 * 60 * 60) },
+               { "day", NANOSEC * (hrtime_t)(24 * 60 * 60) },
+               { "hz", 0 },
                { NULL, 0 }
                { NULL, 0 }
-       };              
-#endif /* __APPLE__ */
-
+       };
 
        if (desc == NULL) {
                char n[PROF_NAMELEN];
 
        if (desc == NULL) {
                char n[PROF_NAMELEN];
@@ -366,26 +383,20 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
                /*
                 * If no description was provided, provide all of our probes.
                 */
                /*
                 * If no description was provided, provide all of our probes.
                 */
-#if !defined(__APPLE__)
-               for (i = 0; i < sizeof (profile_rates) / sizeof (int); i++) {
-#else
-               for (i = 0; i < (int)(sizeof (profile_rates) / sizeof (int)); i++) {
-#endif /* __APPLE__ */
-                       if ((rate = profile_rates[i]) == 0)
+               for (i = 0; i < (int)(sizeof(profile_rates) / sizeof(int)); i++) {
+                       if ((rate = profile_rates[i]) == 0) {
                                continue;
                                continue;
+                       }
 
                        (void) snprintf(n, PROF_NAMELEN, "%s%d",
                            PROF_PREFIX_PROFILE, rate);
                        profile_create(NANOSEC / rate, n, PROF_PROFILE);
                }
 
 
                        (void) snprintf(n, PROF_NAMELEN, "%s%d",
                            PROF_PREFIX_PROFILE, rate);
                        profile_create(NANOSEC / rate, n, PROF_PROFILE);
                }
 
-#if !defined(__APPLE__)
-               for (i = 0; i < sizeof (profile_ticks) / sizeof (int); i++) {
-#else
-               for (i = 0; i < (int)(sizeof (profile_ticks) / sizeof (int)); i++) {
-#endif /* __APPLE__ */
-                       if ((rate = profile_ticks[i]) == 0)
+               for (i = 0; i < (int)(sizeof(profile_ticks) / sizeof(int)); i++) {
+                       if ((rate = profile_ticks[i]) == 0) {
                                continue;
                                continue;
+                       }
 
                        (void) snprintf(n, PROF_NAMELEN, "%s%d",
                            PROF_PREFIX_TICK, rate);
 
                        (void) snprintf(n, PROF_NAMELEN, "%s%d",
                            PROF_PREFIX_TICK, rate);
@@ -400,13 +411,15 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
        for (i = 0; types[i].prefix != NULL; i++) {
                len = strlen(types[i].prefix);
 
        for (i = 0; types[i].prefix != NULL; i++) {
                len = strlen(types[i].prefix);
 
-               if (strncmp(name, types[i].prefix, len) != 0)
+               if (strncmp(name, types[i].prefix, len) != 0) {
                        continue;
                        continue;
+               }
                break;
        }
 
                break;
        }
 
-       if (types[i].prefix == NULL)
+       if (types[i].prefix == NULL) {
                return;
                return;
+       }
 
        kind = types[i].kind;
        j = strlen(name) - len;
 
        kind = types[i].kind;
        j = strlen(name) - len;
@@ -415,8 +428,9 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
         * We need to start before any time suffix.
         */
        for (j = strlen(name); j >= len; j--) {
         * We need to start before any time suffix.
         */
        for (j = strlen(name); j >= len; j--) {
-               if (name[j] >= '0' && name[j] <= '9')
+               if (name[j] >= '0' && name[j] <= '9') {
                        break;
                        break;
+               }
                suffix = &name[j];
        }
 
                suffix = &name[j];
        }
 
@@ -426,35 +440,32 @@ profile_provide(void *arg, const dtrace_probedesc_t *desc)
         * Now determine the numerical value present in the probe name.
         */
        for (; j >= len; j--) {
         * Now determine the numerical value present in the probe name.
         */
        for (; j >= len; j--) {
-               if (name[j] < '0' || name[j] > '9')
+               if (name[j] < '0' || name[j] > '9') {
                        return;
                        return;
+               }
 
                val += (name[j] - '0') * mult;
                mult *= (hrtime_t)10;
        }
 
 
                val += (name[j] - '0') * mult;
                mult *= (hrtime_t)10;
        }
 
-       if (val == 0)
+       if (val == 0) {
                return;
                return;
+       }
 
        /*
         * Look-up the suffix to determine the multiplier.
         */
        for (i = 0, mult = 0; suffixes[i].name != NULL; i++) {
 
        /*
         * Look-up the suffix to determine the multiplier.
         */
        for (i = 0, mult = 0; suffixes[i].name != NULL; i++) {
-#if !defined(__APPLE__)
-               if (strcasecmp(suffixes[i].name, suffix) == 0) {
-                       mult = suffixes[i].mult;
-                       break;
-               }
-#else
+               /* APPLE NOTE: Darwin employs size bounded string operations */
                if (strncasecmp(suffixes[i].name, suffix, strlen(suffixes[i].name) + 1) == 0) {
                        mult = suffixes[i].mult;
                        break;
                }
                if (strncasecmp(suffixes[i].name, suffix, strlen(suffixes[i].name) + 1) == 0) {
                        mult = suffixes[i].mult;
                        break;
                }
-#endif /* __APPLE__ */
        }
 
        }
 
-       if (suffixes[i].name == NULL && *suffix != '\0')
+       if (suffixes[i].name == NULL && *suffix != '\0') {
                return;
                return;
+       }
 
        if (mult == 0) {
                /*
 
        if (mult == 0) {
                /*
@@ -476,17 +487,15 @@ profile_destroy(void *arg, dtrace_id_t id, void *parg)
        profile_probe_t *prof = parg;
 
        ASSERT(prof->prof_cyclic == CYCLIC_NONE);
        profile_probe_t *prof = parg;
 
        ASSERT(prof->prof_cyclic == CYCLIC_NONE);
-#if !defined(__APPLE__)
-       kmem_free(prof, sizeof (profile_probe_t));
-#else
-       if (prof->prof_kind == PROF_TICK)
-               kmem_free(prof, sizeof (profile_probe_t));
-       else
-               kmem_free(prof, sizeof (profile_probe_t) + NCPU*sizeof(profile_probe_percpu_t));
-#endif /* __APPLE__ */
+
+       if (prof->prof_kind == PROF_TICK) {
+               kmem_free(prof, sizeof(profile_probe_t));
+       } else {
+               kmem_free(prof, sizeof(profile_probe_t) + NCPU * sizeof(profile_probe_percpu_t));
+       }
 
        ASSERT(profile_total >= 1);
 
        ASSERT(profile_total >= 1);
-       atomic_add_32(&profile_total, -1);
+       os_atomic_dec(&profile_total, relaxed);
 }
 
 /*ARGSUSED*/
 }
 
 /*ARGSUSED*/
@@ -497,11 +506,7 @@ profile_online(void *arg, dtrace_cpu_t *cpu, cyc_handler_t *hdlr, cyc_time_t *wh
        profile_probe_t *prof = arg;
        profile_probe_percpu_t *pcpu;
 
        profile_probe_t *prof = arg;
        profile_probe_percpu_t *pcpu;
 
-#if !defined(__APPLE__)
-       pcpu = kmem_zalloc(sizeof (profile_probe_percpu_t), KM_SLEEP);
-#else
        pcpu = ((profile_probe_percpu_t *)(&(prof[1]))) + cpu_number();
        pcpu = ((profile_probe_percpu_t *)(&(prof[1]))) + cpu_number();
-#endif /* __APPLE__ */
        pcpu->profc_probe = prof;
 
        hdlr->cyh_func = profile_fire;
        pcpu->profc_probe = prof;
 
        hdlr->cyh_func = profile_fire;
@@ -509,11 +514,7 @@ profile_online(void *arg, dtrace_cpu_t *cpu, cyc_handler_t *hdlr, cyc_time_t *wh
        hdlr->cyh_level = CY_HIGH_LEVEL;
 
        when->cyt_interval = prof->prof_interval;
        hdlr->cyh_level = CY_HIGH_LEVEL;
 
        when->cyt_interval = prof->prof_interval;
-#if !defined(__APPLE__)
        when->cyt_when = dtrace_gethrtime() + when->cyt_interval;
        when->cyt_when = dtrace_gethrtime() + when->cyt_interval;
-#else
-       when->cyt_when = 0;
-#endif /* __APPLE__ */
 
        pcpu->profc_expected = when->cyt_when;
        pcpu->profc_interval = when->cyt_interval;
 
        pcpu->profc_expected = when->cyt_when;
        pcpu->profc_interval = when->cyt_interval;
@@ -526,11 +527,7 @@ profile_offline(void *arg, dtrace_cpu_t *cpu, void *oarg)
        profile_probe_percpu_t *pcpu = oarg;
 
        ASSERT(pcpu->profc_probe == arg);
        profile_probe_percpu_t *pcpu = oarg;
 
        ASSERT(pcpu->profc_probe == arg);
-#if !defined(__APPLE__)
-       kmem_free(pcpu, sizeof (profile_probe_percpu_t));
-#else
 #pragma unused(pcpu,arg,cpu) /* __APPLE__ */
 #pragma unused(pcpu,arg,cpu) /* __APPLE__ */
-#endif /* __APPLE__ */
 }
 
 /*ARGSUSED*/
 }
 
 /*ARGSUSED*/
@@ -564,20 +561,13 @@ profile_enable(void *arg, dtrace_id_t id, void *parg)
                omni.cyo_arg = prof;
        }
 
                omni.cyo_arg = prof;
        }
 
-#if !defined(__APPLE__)
-       if (prof->prof_kind == PROF_TICK) {
-               prof->prof_cyclic = cyclic_add(&hdlr, &when);
-       } else {
-               prof->prof_cyclic = cyclic_add_omni(&omni);
-       }
-#else
        if (prof->prof_kind == PROF_TICK) {
                prof->prof_cyclic = cyclic_timer_add(&hdlr, &when);
        } else {
                prof->prof_cyclic = (cyclic_id_t)cyclic_add_omni(&omni); /* cast puns cyclic_id_list_t with cyclic_id_t */
        }
        if (prof->prof_kind == PROF_TICK) {
                prof->prof_cyclic = cyclic_timer_add(&hdlr, &when);
        } else {
                prof->prof_cyclic = (cyclic_id_t)cyclic_add_omni(&omni); /* cast puns cyclic_id_list_t with cyclic_id_t */
        }
-#endif /* __APPLE__ */
-       return(0);
+
+       return 0;
 }
 
 /*ARGSUSED*/
 }
 
 /*ARGSUSED*/
@@ -589,98 +579,105 @@ profile_disable(void *arg, dtrace_id_t id, void *parg)
        ASSERT(prof->prof_cyclic != CYCLIC_NONE);
        ASSERT(MUTEX_HELD(&cpu_lock));
 
        ASSERT(prof->prof_cyclic != CYCLIC_NONE);
        ASSERT(MUTEX_HELD(&cpu_lock));
 
-#if !defined(__APPLE__)
-       cyclic_remove(prof->prof_cyclic);
-#else
 #pragma unused(arg,id)
        if (prof->prof_kind == PROF_TICK) {
                cyclic_timer_remove(prof->prof_cyclic);
        } else {
                cyclic_remove_omni((cyclic_id_list_t)prof->prof_cyclic); /* cast puns cyclic_id_list_t with cyclic_id_t */
        }
 #pragma unused(arg,id)
        if (prof->prof_kind == PROF_TICK) {
                cyclic_timer_remove(prof->prof_cyclic);
        } else {
                cyclic_remove_omni((cyclic_id_list_t)prof->prof_cyclic); /* cast puns cyclic_id_list_t with cyclic_id_t */
        }
-#endif /* __APPLE__ */
        prof->prof_cyclic = CYCLIC_NONE;
 }
 
        prof->prof_cyclic = CYCLIC_NONE;
 }
 
-#if !defined(__APPLE__)
-/*ARGSUSED*/
-static int
-profile_usermode(void *arg, dtrace_id_t id, void *parg)
+static uint64_t
+profile_getarg(void *arg, dtrace_id_t id, void *parg, int argno, int aframes)
 {
 {
-       return (CPU->cpu_profile_pc == 0);
+#pragma unused(arg, id, parg, argno, aframes)
+       /*
+        * All the required arguments for the profile probe are passed directly
+        * to dtrace_probe, and we do not go through dtrace_getarg which doesn't
+        * know how to hop to the kernel stack from the interrupt stack like
+        * dtrace_getpcstack
+        */
+       return 0;
 }
 }
-#else
+
+static void
+profile_getargdesc(void *arg, dtrace_id_t id, void *parg, dtrace_argdesc_t *desc)
+{
+#pragma unused(arg, id)
+       profile_probe_t *prof = parg;
+       const char *argdesc = NULL;
+       switch (desc->dtargd_ndx) {
+       case 0:
+               argdesc = "void*";
+               break;
+       case 1:
+               argdesc = "user_addr_t";
+               break;
+       case 2:
+               if (prof->prof_kind == PROF_PROFILE) {
+                       argdesc = "hrtime_t";
+               }
+               break;
+       }
+       if (argdesc) {
+               strlcpy(desc->dtargd_native, argdesc, DTRACE_ARGTYPELEN);
+       } else {
+               desc->dtargd_ndx = DTRACE_ARGNONE;
+       }
+}
+
+/*
+ * APPLE NOTE:  profile_usermode call not supported.
+ */
 static int
 profile_usermode(void *arg, dtrace_id_t id, void *parg)
 {
 #pragma unused(arg,id,parg)
        return 1; /* XXX_BOGUS */
 }
 static int
 profile_usermode(void *arg, dtrace_id_t id, void *parg)
 {
 #pragma unused(arg,id,parg)
        return 1; /* XXX_BOGUS */
 }
-#endif /* __APPLE__ */
 
 static dtrace_pattr_t profile_attr = {
 
 static dtrace_pattr_t profile_attr = {
-{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
-{ DTRACE_STABILITY_UNSTABLE, DTRACE_STABILITY_UNSTABLE, DTRACE_CLASS_UNKNOWN },
-{ DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
-{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
-{ DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
+       { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
+       { DTRACE_STABILITY_UNSTABLE, DTRACE_STABILITY_UNSTABLE, DTRACE_CLASS_UNKNOWN },
+       { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
+       { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
+       { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_COMMON },
 };
 
 static dtrace_pops_t profile_pops = {
 };
 
 static dtrace_pops_t profile_pops = {
-       profile_provide,
-       NULL,
-       profile_enable,
-       profile_disable,
-       NULL,
-       NULL,
-       NULL,
-       NULL,
-       profile_usermode,
-       profile_destroy
+       .dtps_provide =         profile_provide,
+       .dtps_provide_module =  NULL,
+       .dtps_enable =          profile_enable,
+       .dtps_disable =         profile_disable,
+       .dtps_suspend =         NULL,
+       .dtps_resume =          NULL,
+       .dtps_getargdesc =      profile_getargdesc,
+       .dtps_getargval =       profile_getarg,
+       .dtps_usermode =        profile_usermode,
+       .dtps_destroy =         profile_destroy
 };
 
 static int
 };
 
 static int
-profile_attach(dev_info_t *devi, ddi_attach_cmd_t cmd)
+profile_attach(dev_info_t *devi)
 {
 {
-       switch (cmd) {
-       case DDI_ATTACH:
-               break;
-       case DDI_RESUME:
-               return (DDI_SUCCESS);
-       default:
-               return (DDI_FAILURE);
-       }
-
-#if !defined(__APPLE__)
-       if (ddi_create_minor_node(devi, "profile", S_IFCHR, 0,
-           DDI_PSEUDO, NULL) == DDI_FAILURE ||
-           dtrace_register("profile", &profile_attr,
-           DTRACE_PRIV_KERNEL | DTRACE_PRIV_USER, NULL,
-           &profile_pops, NULL, &profile_id) != 0) {
-               ddi_remove_minor_node(devi, NULL);
-               return (DDI_FAILURE);
-       }
-       
-       profile_max = ddi_getprop(DDI_DEV_T_ANY, devi, DDI_PROP_DONTPASS,
-           "profile-max-probes", PROFILE_MAX_DEFAULT);
-#else
        if (ddi_create_minor_node(devi, "profile", S_IFCHR, 0,
            DDI_PSEUDO, 0) == DDI_FAILURE ||
            dtrace_register("profile", &profile_attr,
            DTRACE_PRIV_KERNEL | DTRACE_PRIV_USER, NULL,
            &profile_pops, NULL, &profile_id) != 0) {
                ddi_remove_minor_node(devi, NULL);
        if (ddi_create_minor_node(devi, "profile", S_IFCHR, 0,
            DDI_PSEUDO, 0) == DDI_FAILURE ||
            dtrace_register("profile", &profile_attr,
            DTRACE_PRIV_KERNEL | DTRACE_PRIV_USER, NULL,
            &profile_pops, NULL, &profile_id) != 0) {
                ddi_remove_minor_node(devi, NULL);
-               return (DDI_FAILURE);
+               return DDI_FAILURE;
        }
 
        profile_max = PROFILE_MAX_DEFAULT;
        }
 
        profile_max = PROFILE_MAX_DEFAULT;
-#endif /* __APPLE__ */
 
 
-       ddi_report_dev(devi);
-       profile_devi = devi;
-       return (DDI_SUCCESS);
+       return DDI_SUCCESS;
 }
 
 }
 
+/*
+ * APPLE NOTE:  profile_detach not implemented
+ */
 #if !defined(__APPLE__)
 static int
 profile_detach(dev_info_t *devi, ddi_detach_cmd_t cmd)
 #if !defined(__APPLE__)
 static int
 profile_detach(dev_info_t *devi, ddi_detach_cmd_t cmd)
@@ -689,114 +686,24 @@ profile_detach(dev_info_t *devi, ddi_detach_cmd_t cmd)
        case DDI_DETACH:
                break;
        case DDI_SUSPEND:
        case DDI_DETACH:
                break;
        case DDI_SUSPEND:
-               return (DDI_SUCCESS);
+               return DDI_SUCCESS;
        default:
        default:
-               return (DDI_FAILURE);
+               return DDI_FAILURE;
        }
 
        }
 
-       if (dtrace_unregister(profile_id) != 0)
-               return (DDI_FAILURE);
-
-       ddi_remove_minor_node(devi, NULL);
-       return (DDI_SUCCESS);
-}
-
-/*ARGSUSED*/
-static int
-profile_info(dev_info_t *dip, ddi_info_cmd_t infocmd, void *arg, void **result)
-{
-       int error;
-
-       switch (infocmd) {
-       case DDI_INFO_DEVT2DEVINFO:
-               *result = (void *)profile_devi;
-               error = DDI_SUCCESS;
-               break;
-       case DDI_INFO_DEVT2INSTANCE:
-               *result = (void *)0;
-               error = DDI_SUCCESS;
-               break;
-       default:
-               error = DDI_FAILURE;
+       if (dtrace_unregister(profile_id) != 0) {
+               return DDI_FAILURE;
        }
        }
-       return (error);
-}
-
-/*ARGSUSED*/
-static int
-profile_open(dev_t *devp, int flag, int otyp, cred_t *cred_p)
-{
-       return (0);
-}
-
-static struct cb_ops profile_cb_ops = {
-       profile_open,           /* open */
-       nodev,                  /* close */
-       nulldev,                /* strategy */
-       nulldev,                /* print */
-       nodev,                  /* dump */
-       nodev,                  /* read */
-       nodev,                  /* write */
-       nodev,                  /* ioctl */
-       nodev,                  /* devmap */
-       nodev,                  /* mmap */
-       nodev,                  /* segmap */
-       nochpoll,               /* poll */
-       ddi_prop_op,            /* cb_prop_op */
-       0,                      /* streamtab  */
-       D_NEW | D_MP            /* Driver compatibility flag */
-};
-
-static struct dev_ops profile_ops = {
-       DEVO_REV,               /* devo_rev, */
-       0,                      /* refcnt  */
-       profile_info,           /* get_dev_info */
-       nulldev,                /* identify */
-       nulldev,                /* probe */
-       profile_attach,         /* attach */
-       profile_detach,         /* detach */
-       nodev,                  /* reset */
-       &profile_cb_ops,        /* driver operations */
-       NULL,                   /* bus operations */
-       nodev                   /* dev power */
-};
-
-/*
- * Module linkage information for the kernel.
- */
-static struct modldrv modldrv = {
-       &mod_driverops,         /* module type (this is a pseudo driver) */
-       "Profile Interrupt Tracing",    /* name of module */
-       &profile_ops,           /* driver ops */
-};
-
-static struct modlinkage modlinkage = {
-       MODREV_1,
-       (void *)&modldrv,
-       NULL
-};
-
-int
-_init(void)
-{
-       return (mod_install(&modlinkage));
-}
 
 
-int
-_info(struct modinfo *modinfop)
-{
-       return (mod_info(&modlinkage, modinfop));
+       ddi_remove_minor_node(devi, NULL);
+       return DDI_SUCCESS;
 }
 }
+#endif /* __APPLE__ */
 
 
-int
-_fini(void)
-{
-       return (mod_remove(&modlinkage));
-}
-#else
 d_open_t _profile_open;
 
 d_open_t _profile_open;
 
-int _profile_open(dev_t dev, int flags, int devtype, struct proc *p)
+int
+_profile_open(dev_t dev, int flags, int devtype, struct proc *p)
 {
 #pragma unused(dev,flags,devtype,p)
        return 0;
 {
 #pragma unused(dev,flags,devtype,p)
        return 0;
@@ -804,47 +711,32 @@ int _profile_open(dev_t dev, int flags, int devtype, struct proc *p)
 
 #define PROFILE_MAJOR  -24 /* let the kernel pick the device number */
 
 
 #define PROFILE_MAJOR  -24 /* let the kernel pick the device number */
 
-/*
- * A struct describing which functions will get invoked for certain
- * actions.
- */
-static struct cdevsw profile_cdevsw =
+static const struct cdevsw profile_cdevsw =
 {
 {
-       _profile_open,          /* open */
-       eno_opcl,                       /* close */
-       eno_rdwrt,                      /* read */
-       eno_rdwrt,                      /* write */
-       eno_ioctl,                      /* ioctl */
-       (stop_fcn_t *)nulldev, /* stop */
-       (reset_fcn_t *)nulldev, /* reset */
-       NULL,                           /* tty's */
-       eno_select,                     /* select */
-       eno_mmap,                       /* mmap */
-       eno_strat,                      /* strategy */
-       eno_getc,                       /* getc */
-       eno_putc,                       /* putc */
-       0                                       /* type */
+       .d_open = _profile_open,
+       .d_close = eno_opcl,
+       .d_read = eno_rdwrt,
+       .d_write = eno_rdwrt,
+       .d_ioctl = eno_ioctl,
+       .d_stop = (stop_fcn_t *)nulldev,
+       .d_reset = (reset_fcn_t *)nulldev,
+       .d_select = eno_select,
+       .d_mmap = eno_mmap,
+       .d_strategy = eno_strat,
+       .d_reserved_1 = eno_getc,
+       .d_reserved_2 = eno_putc,
 };
 
 };
 
-static int gProfileInited = 0;
-
-void profile_init( void )
+void
+profile_init( void )
 {
 {
-       if (0 == gProfileInited)
-       {
-               int majdevno = cdevsw_add(PROFILE_MAJOR, &profile_cdevsw);
-               
-               if (majdevno < 0) {
-                       printf("profile_init: failed to allocate a major number!\n");
-                       gProfileInited = 0;
-                       return;
-               }
+       int majdevno = cdevsw_add(PROFILE_MAJOR, &profile_cdevsw);
 
 
-               profile_attach( (dev_info_t     *)(uintptr_t)majdevno, DDI_ATTACH );
+       if (majdevno < 0) {
+               printf("profile_init: failed to allocate a major number!\n");
+               return;
+       }
 
 
-               gProfileInited = 1;
-       } else
-               panic("profile_init: called twice!\n");
+       profile_attach((dev_info_t*)(uintptr_t)majdevno);
 }
 #undef PROFILE_MAJOR
 }
 #undef PROFILE_MAJOR
-#endif /* __APPLE__ */