FreeBSD/Linux Kernel Cross Reference
sys/osfmk/chud/chud_osfmk_callback.c

     /*
      * Copyright (c) 2003-2007 Apple Inc. All rights reserved.
      *
      * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
      * 
      * 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.
     * 
     * 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, QUIET ENJOYMENT OR NON-INFRINGEMENT.
     * Please see the License for the specific language governing rights and
     * limitations under the License.
     * 
     * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
     */
    
    #include <stdint.h>
    #include <mach/boolean.h>
    #include <mach/mach_types.h>
    
    #include <kern/kern_types.h>
    #include <kern/processor.h>
    #include <kern/timer_call.h>
    #include <kern/thread_call.h>
    #include <kern/kalloc.h>
    #include <kern/thread.h>
    
    #include <libkern/OSAtomic.h>
    
    #include <machine/machine_routines.h>
    #include <machine/cpu_data.h>
    
    #include <chud/chud_xnu.h>
    #include <chud/chud_xnu_private.h>
    
    #pragma mark **** timer ****
    
    __private_extern__ chud_timer_t
    chudxnu_timer_alloc(chudxnu_timer_callback_func_t func, uint32_t param0)
    {
        return (chud_timer_t)thread_call_allocate((thread_call_func_t)func, (thread_call_param_t)param0);
    }
    
    __private_extern__ kern_return_t
    chudxnu_timer_callback_enter(
            chud_timer_t timer,
            uint32_t param1,
            uint32_t time,
            uint32_t units)
    {
        uint64_t t_delay;
        clock_interval_to_deadline(time, units, &t_delay);
        thread_call_enter1_delayed((thread_call_t)timer, (thread_call_param_t)param1, t_delay);
        return KERN_SUCCESS;
    }
    
    __private_extern__ kern_return_t
    chudxnu_timer_callback_cancel(chud_timer_t timer)
    {
        thread_call_cancel((thread_call_t)timer);
        return KERN_SUCCESS;
    }
    
    __private_extern__ kern_return_t
    chudxnu_timer_free(chud_timer_t timer)
    {
        thread_call_cancel((thread_call_t)timer);
        thread_call_free((thread_call_t)timer);
        return KERN_SUCCESS;
    }
    
    static chudxnu_dtrace_callback_t 
            dtrace_callback = (chudxnu_dtrace_callback_t) NULL;
    
    kern_return_t
    chudxnu_dtrace_callback(uint64_t selector, uint64_t *args, uint32_t count)
    {
            /* it's not an error if no callback is hooked up */
            kern_return_t ret = KERN_SUCCESS;
    
            /* Make a local stack copy of the function ptr */
            chudxnu_dtrace_callback_t fn = dtrace_callback;
    
            if(fn) {
                    ret = fn(selector, args, count);
            }
    
           return ret;
   }
   
   __private_extern__ void
   chudxnu_dtrace_callback_enter(chudxnu_dtrace_callback_t fn)
   {
           chudxnu_dtrace_callback_t old_fn = dtrace_callback;
   
           /* Atomically clear the call back */
           while(!OSCompareAndSwap((UInt32)old_fn, (UInt32)fn, 
                   (volatile UInt32 *) &dtrace_callback)) {
                   old_fn = dtrace_callback;
           }
   }
   
   __private_extern__ void
   chudxnu_dtrace_callback_cancel(void)
   {
           chudxnu_dtrace_callback_enter(NULL);
   }
   

Cache object: 50b91ebc4d418761a3fe5685a2971f18