xnu-1504.9.26.tar.gz

[apple/xnu.git] / osfmk / kern / stack.c

diff --git a/osfmk/kern/stack.c b/osfmk/kern/stack.c
index f9fb6682c9680c55a074ca5a2aecdbf48158659d..a5912256932b7222172b63e616ae5dfd9f24a259 100644 (file)
--- a/osfmk/kern/stack.c
+++ b/osfmk/kern/stack.c
@@ -1,5 +1,5 @@
 /*
 /*

- * Copyright (c) 2003-2004 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2003-2007 Apple Inc. All rights reserved.

  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 


@@ -58,11 +58,11 @@ decl_simple_lock_data(static,stack_lock_data)
 
 #define STACK_CACHE_SIZE       2
 
 
 #define STACK_CACHE_SIZE       2
 

-static vm_map_t                        stack_map;

 static vm_offset_t             stack_free_list;
 
 static unsigned int            stack_free_count, stack_free_hiwat;             /* free list count */
 static vm_offset_t             stack_free_list;
 
 static unsigned int            stack_free_count, stack_free_hiwat;             /* free list count */

-static unsigned int            stack_total, stack_hiwat;                               /* current total count */
+static unsigned int            stack_hiwat;
+unsigned int                   stack_total;                            /* current total count */

 
 static unsigned int            stack_free_target;
 static int                             stack_free_delta;
 
 static unsigned int            stack_free_target;
 static int                             stack_free_delta;


@@ -71,37 +71,52 @@ static unsigned int         stack_new_count;                                                /* total new stack allocations */
 
 static vm_offset_t             stack_addr_mask;
 
 
 static vm_offset_t             stack_addr_mask;
 

+unsigned int                   kernel_stack_pages = KERNEL_STACK_SIZE / PAGE_SIZE;
+vm_offset_t                    kernel_stack_size = KERNEL_STACK_SIZE;
+vm_offset_t                    kernel_stack_mask = -KERNEL_STACK_SIZE;
+vm_offset_t                    kernel_stack_depth_max = 0;
+

 /*
  *     The next field is at the base of the stack,
  *     so the low end is left unsullied.
  */
 #define stack_next(stack)      \
 /*
  *     The next field is at the base of the stack,
  *     so the low end is left unsullied.
  */
 #define stack_next(stack)      \

-                       (*((vm_offset_t *)((stack) + KERNEL_STACK_SIZE) - 1))
+       (*((vm_offset_t *)((stack) + kernel_stack_size) - 1))
+
+static inline int
+log2(vm_offset_t size)
+{
+       int     result;
+       for (result = 0; size > 0; result++)
+               size >>= 1;
+       return result;
+}
+
+static inline vm_offset_t
+roundup_pow2(vm_offset_t size)
+{
+       return 1UL << (log2(size - 1) + 1); 
+}

 
 void
 stack_init(void)
 {
 
 void
 stack_init(void)
 {

-       vm_offset_t                     stacks, boundary;
-       vm_map_offset_t         map_addr;
-

        simple_lock_init(&stack_lock_data, 0);
        
        simple_lock_init(&stack_lock_data, 0);
        

-       if (KERNEL_STACK_SIZE < round_page(KERNEL_STACK_SIZE))
-               panic("stack_init: stack size %d not a multiple of page size %d\n",     KERNEL_STACK_SIZE, PAGE_SIZE);
-       
-       for (boundary = PAGE_SIZE; boundary <= KERNEL_STACK_SIZE; )
-               boundary <<= 1;
-
-       stack_addr_mask = boundary - 1;
-
-       if (kmem_suballoc(kernel_map, &stacks, (boundary * (2 * THREAD_MAX + 64)),
-                                                               FALSE, VM_FLAGS_ANYWHERE, &stack_map) != KERN_SUCCESS)
-               panic("stack_init: kmem_suballoc");
+       if (PE_parse_boot_argn("kernel_stack_pages",
+                              &kernel_stack_pages,
+                              sizeof (kernel_stack_pages))) {
+               kernel_stack_size = kernel_stack_pages * PAGE_SIZE;
+               printf("stack_init: kernel_stack_pages=%d kernel_stack_size=%p\n",
+                       kernel_stack_pages, (void *) kernel_stack_size);
+       }

 
 

-       map_addr = vm_map_min(stack_map);
-       if (vm_map_enter(stack_map, &map_addr, vm_map_round_page(PAGE_SIZE), 0, (VM_MAKE_TAG(VM_MEMORY_STACK) | VM_FLAGS_FIXED),
-                                               VM_OBJECT_NULL, 0, FALSE, VM_PROT_NONE, VM_PROT_NONE, VM_INHERIT_DEFAULT) != KERN_SUCCESS)
-               panic("stack_init: vm_map_enter");
+       if (kernel_stack_size < round_page(kernel_stack_size))
+               panic("stack_init: stack size %p not a multiple of page size %d\n",
+                       (void *) kernel_stack_size, PAGE_SIZE);
+       
+       stack_addr_mask = roundup_pow2(kernel_stack_size) - 1;
+       kernel_stack_mask = ~stack_addr_mask;

 }
 
 /*
 }
 
 /*


@@ -116,6 +131,7 @@ stack_alloc(
 {
        vm_offset_t             stack;
        spl_t                   s;
 {
        vm_offset_t             stack;
        spl_t                   s;

+       int                     guard_flags;

 
        assert(thread->kernel_stack == 0);
 
 
        assert(thread->kernel_stack == 0);
 


@@ -136,8 +152,27 @@ stack_alloc(
        splx(s);
                
        if (stack == 0) {
        splx(s);
                
        if (stack == 0) {

-               if (kernel_memory_allocate(stack_map, &stack, KERNEL_STACK_SIZE, stack_addr_mask, KMA_KOBJECT) != KERN_SUCCESS)
+
+               /*
+                * Request guard pages on either side of the stack.  Ask
+                * kernel_memory_allocate() for two extra pages to account
+                * for these.
+                */
+
+               guard_flags = KMA_GUARD_FIRST | KMA_GUARD_LAST;
+               if (kernel_memory_allocate(kernel_map, &stack,
+                                          kernel_stack_size + (2*PAGE_SIZE),
+                                          stack_addr_mask,
+                                          KMA_KOBJECT | guard_flags)
+                   != KERN_SUCCESS)

                        panic("stack_alloc: kernel_memory_allocate");
                        panic("stack_alloc: kernel_memory_allocate");

+
+               /*
+                * The stack address that comes back is the address of the lower
+                * guard page.  Skip past it to get the actual stack base address.
+                */
+
+               stack += PAGE_SIZE;

        }
 
        machine_stack_attach(thread, stack);
        }
 
        machine_stack_attach(thread, stack);


@@ -155,28 +190,8 @@ stack_free(
     vm_offset_t                stack = machine_stack_detach(thread);
 
        assert(stack);
     vm_offset_t                stack = machine_stack_detach(thread);
 
        assert(stack);

-       if (stack != thread->reserved_stack) {
-               struct stack_cache      *cache;
-               spl_t                           s;
-
-               s = splsched();
-               cache = &PROCESSOR_DATA(current_processor(), stack_cache);
-               if (cache->count < STACK_CACHE_SIZE) {
-                       stack_next(stack) = cache->free;
-                       cache->free = stack;
-                       cache->count++;
-               }
-               else {
-                       stack_lock();
-                       stack_next(stack) = stack_free_list;
-                       stack_free_list = stack;
-                       if (++stack_free_count > stack_free_hiwat)
-                               stack_free_hiwat = stack_free_count;
-                       stack_free_delta++;
-                       stack_unlock();
-               }
-               splx(s);
-       }
+       if (stack != thread->reserved_stack)
+               stack_free_stack(stack);

 }
 
 void
 }
 
 void


@@ -278,9 +293,24 @@ stack_collect(void)
                        stack_unlock();
                        splx(s);
 
                        stack_unlock();
                        splx(s);
 

-                       if (vm_map_remove(stack_map, vm_map_trunc_page(stack),
-                                                               vm_map_round_page(stack + KERNEL_STACK_SIZE), VM_MAP_REMOVE_KUNWIRE) != KERN_SUCCESS)
+                       /*
+                        * Get the stack base address, then decrement by one page
+                        * to account for the lower guard page.  Add two extra pages
+                        * to the size to account for the guard pages on both ends
+                        * that were originally requested when the stack was allocated
+                        * back in stack_alloc().
+                        */
+
+                       stack = (vm_offset_t)vm_map_trunc_page(stack);
+                       stack -= PAGE_SIZE;
+                       if (vm_map_remove(
+                                   kernel_map,
+                                   stack,
+                                   stack + kernel_stack_size+(2*PAGE_SIZE),
+                                   VM_MAP_REMOVE_KUNWIRE)
+                           != KERN_SUCCESS)

                                panic("stack_collect: vm_map_remove");
                                panic("stack_collect: vm_map_remove");

+                       stack = 0;

 
                        s = splsched();
                        stack_lock();
 
                        s = splsched();
                        stack_lock();


@@ -345,10 +375,10 @@ stack_fake_zone_info(int *count, vm_size_t *cur_size, vm_size_t *max_size, vm_si
        splx(s);
 
        *count      = total - free;
        splx(s);
 
        *count      = total - free;

-       *cur_size   = KERNEL_STACK_SIZE * total;
-       *max_size   = KERNEL_STACK_SIZE * hiwat;
-       *elem_size  = KERNEL_STACK_SIZE;
-       *alloc_size = KERNEL_STACK_SIZE;
+       *cur_size   = kernel_stack_size * total;
+       *max_size   = kernel_stack_size * hiwat;
+       *elem_size  = kernel_stack_size;
+       *alloc_size = kernel_stack_size;

        *collectable = 1;
        *exhaustable = 0;
 }
        *collectable = 1;
        *exhaustable = 0;
 }


@@ -383,7 +413,7 @@ processor_set_stack_usage(
        vm_size_t maxusage;
        vm_offset_t maxstack;
 
        vm_size_t maxusage;
        vm_offset_t maxstack;
 

-       register thread_t *threads;
+       register thread_t *thread_list;

        register thread_t thread;
 
        unsigned int actual;    /* this many things */
        register thread_t thread;
 
        unsigned int actual;    /* this many things */


@@ -392,19 +422,16 @@ processor_set_stack_usage(
        vm_size_t size, size_needed;
        void *addr;
 
        vm_size_t size, size_needed;
        void *addr;
 

-       if (pset == PROCESSOR_SET_NULL)
+       if (pset == PROCESSOR_SET_NULL || pset != &pset0)

                return KERN_INVALID_ARGUMENT;
 
                return KERN_INVALID_ARGUMENT;
 

-       size = 0; addr = 0;
+       size = 0;
+       addr = NULL;

 
        for (;;) {
 
        for (;;) {

-               pset_lock(pset);
-               if (!pset->active) {
-                       pset_unlock(pset);
-                       return KERN_INVALID_ARGUMENT;
-               }
+               lck_mtx_lock(&tasks_threads_lock);

 
 

-               actual = pset->thread_count;
+               actual = threads_count;

 
                /* do we have the memory we need? */
 
 
                /* do we have the memory we need? */
 


@@ -412,8 +439,7 @@ processor_set_stack_usage(
                if (size_needed <= size)
                        break;
 
                if (size_needed <= size)
                        break;
 

-               /* unlock the pset and allocate more memory */
-               pset_unlock(pset);
+               lck_mtx_unlock(&tasks_threads_lock);

 
                if (size != 0)
                        kfree(addr, size);
 
                if (size != 0)
                        kfree(addr, size);


@@ -426,18 +452,17 @@ processor_set_stack_usage(
                        return KERN_RESOURCE_SHORTAGE;
        }
 
                        return KERN_RESOURCE_SHORTAGE;
        }
 

-       /* OK, have memory and the processor_set is locked & active */
-       threads = (thread_t *) addr;
-       for (i = 0, thread = (thread_t) queue_first(&pset->threads);
-                                       !queue_end(&pset->threads, (queue_entry_t) thread);
-                                       thread = (thread_t) queue_next(&thread->pset_threads)) {
+       /* OK, have memory and list is locked */
+       thread_list = (thread_t *) addr;
+       for (i = 0, thread = (thread_t) queue_first(&threads);
+                                       !queue_end(&threads, (queue_entry_t) thread);
+                                       thread = (thread_t) queue_next(&thread->threads)) {

                thread_reference_internal(thread);
                thread_reference_internal(thread);

-               threads[i++] = thread;
+               thread_list[i++] = thread;

        }
        assert(i <= actual);
 
        }
        assert(i <= actual);
 

-       /* can unlock processor set now that we have the thread refs */
-       pset_unlock(pset);
+       lck_mtx_unlock(&tasks_threads_lock);

 
        /* calculate maxusage and free thread references */
 
 
        /* calculate maxusage and free thread references */
 


@@ -445,7 +470,7 @@ processor_set_stack_usage(
        maxusage = 0;
        maxstack = 0;
        while (i > 0) {
        maxusage = 0;
        maxstack = 0;
        while (i > 0) {

-               thread_t threadref = threads[--i];
+               thread_t threadref = thread_list[--i];

 
                if (threadref->kernel_stack != 0)
                        total++;
 
                if (threadref->kernel_stack != 0)
                        total++;


@@ -457,7 +482,7 @@ processor_set_stack_usage(
                kfree(addr, size);
 
        *totalp = total;
                kfree(addr, size);
 
        *totalp = total;

-       *residentp = *spacep = total * round_page(KERNEL_STACK_SIZE);
+       *residentp = *spacep = total * round_page(kernel_stack_size);

        *maxusagep = maxusage;
        *maxstackp = maxstack;
        return KERN_SUCCESS;
        *maxusagep = maxusage;
        *maxstackp = maxstack;
        return KERN_SUCCESS;


@@ -467,10 +492,10 @@ processor_set_stack_usage(
 
 vm_offset_t min_valid_stack_address(void)
 {
 
 vm_offset_t min_valid_stack_address(void)
 {

-       return vm_map_min(stack_map);
+       return (vm_offset_t)vm_map_min(kernel_map);

 }
 
 vm_offset_t max_valid_stack_address(void)
 {
 }
 
 vm_offset_t max_valid_stack_address(void)
 {

-       return vm_map_max(stack_map);
+       return (vm_offset_t)vm_map_max(kernel_map);

 }
 }