osfmk/i386/hibernate_i386.c

   1 /*
   2  * Copyright (c) 2004 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28
  29 #include <kern/machine.h>
  30 #include <kern/misc_protos.h>
  31 #include <kern/thread.h>
  32 #include <kern/processor.h>
  33 #include <kern/kalloc.h>
  34 #include <mach/machine.h>
  35 #include <mach/processor_info.h>
  36 #include <mach/mach_types.h>
  37 #include <i386/pmap.h>
  38 #include <kern/cpu_data.h>
  39 #include <IOKit/IOPlatformExpert.h>
  40
  41 #include <pexpert/i386/efi.h>
  42
  43 #include <IOKit/IOHibernatePrivate.h>
  44 #include <vm/vm_page.h>
  45 #include <i386/i386_lowmem.h>
  46
  47 #define MAX_BANKS       32
  48
  49 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
  50
  51 hibernate_page_list_t *
  52 hibernate_page_list_allocate(void)
  53 {
  54     ppnum_t                 base, num;
  55     vm_size_t               size;
  56     uint32_t                bank, num_banks;
  57     uint32_t                pages, page_count;
  58     hibernate_page_list_t * list;
  59     hibernate_bitmap_t *    bitmap;
  60
  61     EfiMemoryRange *        mptr;
  62     uint32_t                mcount, msize, i;
  63     hibernate_bitmap_t      dram_ranges[MAX_BANKS];
  64     boot_args *             args = (boot_args *) PE_state.bootArgs;
  65
  66     mptr = (EfiMemoryRange *)ml_static_ptovirt(args->MemoryMap);
  67     if (args->MemoryMapDescriptorSize == 0)
  68         panic("Invalid memory map descriptor size");
  69     msize = args->MemoryMapDescriptorSize;
  70     mcount = args->MemoryMapSize / msize;
  71
  72     num_banks = 0;
  73     for (i = 0; i < mcount; i++, mptr = (EfiMemoryRange *)(((vm_offset_t)mptr) + msize))
  74     {
  75         base = (ppnum_t) (mptr->PhysicalStart >> I386_PGSHIFT);
  76         num = (ppnum_t) mptr->NumberOfPages;
  77         if (!num)
  78             continue;
  79
  80         switch (mptr->Type)
  81         {
  82             // any kind of dram
  83             case kEfiLoaderCode:
  84             case kEfiLoaderData:
  85             case kEfiBootServicesCode:
  86             case kEfiBootServicesData:
  87             case kEfiConventionalMemory:
  88             case kEfiACPIReclaimMemory:
  89             case kEfiACPIMemoryNVS:
  90             case kEfiPalCode:
  91
  92                 for (bank = 0; bank < num_banks; bank++)
  93                 {
  94                     if (dram_ranges[bank].first_page <= base)
  95                         continue;
  96                     if ((base + num) == dram_ranges[bank].first_page)
  97                     {
  98                         dram_ranges[bank].first_page = base;
  99                         num = 0;
 100                     }
 101                     break;
 102                 }
 103                 if (!num) break;
 104
 105                 if (bank && (base == (1 + dram_ranges[bank - 1].last_page)))
 106                     bank--;
 107                 else
 108                 {
 109                     num_banks++;
 110                     if (num_banks >= MAX_BANKS) break;
 111                     bcopy(&dram_ranges[bank],
 112                           &dram_ranges[bank + 1],
 113                           (num_banks - bank - 1) * sizeof(hibernate_bitmap_t));
 114                     dram_ranges[bank].first_page = base;
 115                 }
 116                 dram_ranges[bank].last_page = base + num - 1;
 117                 break;
 118
 119             // runtime services will be restarted, so no save
 120             case kEfiRuntimeServicesCode:
 121             case kEfiRuntimeServicesData:
 122             // non dram
 123             case kEfiReservedMemoryType:
 124             case kEfiUnusableMemory:
 125             case kEfiMemoryMappedIO:
 126             case kEfiMemoryMappedIOPortSpace:
 127             default:
 128                 break;
 129         }
 130     }
 131
 132     if (num_banks >= MAX_BANKS)
 133         return (NULL);
 134
 135     // size the hibernation bitmap
 136
 137     size = sizeof(hibernate_page_list_t);
 138     page_count = 0;
 139     for (bank = 0; bank < num_banks; bank++) {
 140         pages = dram_ranges[bank].last_page + 1 - dram_ranges[bank].first_page;
 141         page_count += pages;
 142         size += sizeof(hibernate_bitmap_t) + ((pages + 31) >> 5) * sizeof(uint32_t);
 143     }
 144
 145     list = (hibernate_page_list_t *)kalloc(size);
 146     if (!list)
 147         return (list);
 148
 149     list->list_size  = (uint32_t)size;
 150     list->page_count = page_count;
 151     list->bank_count = num_banks;
 152
 153     // convert to hibernation bitmap.
 154
 155     bitmap = &list->bank_bitmap[0];
 156     for (bank = 0; bank < num_banks; bank++)
 157     {
 158         bitmap->first_page = dram_ranges[bank].first_page;
 159         bitmap->last_page  = dram_ranges[bank].last_page;
 160         bitmap->bitmapwords = (bitmap->last_page + 1
 161                                - bitmap->first_page + 31) >> 5;
 162         kprintf("hib bank[%d]: 0x%x000 end 0x%xfff\n", bank,
 163                 bitmap->first_page,
 164                 bitmap->last_page);
 165         bitmap = (hibernate_bitmap_t *) &bitmap->bitmap[bitmap->bitmapwords];
 166     }
 167
 168     return (list);
 169 }
 170
 171 // mark pages not to be saved, but available for scratch usage during restore
 172
 173 void
 174 hibernate_page_list_setall_machine( __unused hibernate_page_list_t * page_list,
 175                                     __unused hibernate_page_list_t * page_list_wired,
 176                                     __unused uint32_t * pagesOut)
 177 {
 178 }
 179
 180 // mark pages not to be saved and not for scratch usage during restore
 181 void
 182 hibernate_page_list_set_volatile( hibernate_page_list_t * page_list,
 183                                   hibernate_page_list_t * page_list_wired,
 184                                   uint32_t * pagesOut)
 185 {
 186     boot_args * args = (boot_args *) PE_state.bootArgs;
 187
 188 #if !defined(x86_64)
 189     hibernate_set_page_state(page_list, page_list_wired,
 190                 I386_HIB_PAGETABLE, I386_HIB_PAGETABLE_COUNT,
 191                 kIOHibernatePageStateFree);
 192     *pagesOut -= I386_HIB_PAGETABLE_COUNT;
 193 #endif
 194
 195     if (args->efiRuntimeServicesPageStart)
 196     {
 197         hibernate_set_page_state(page_list, page_list_wired,
 198                     args->efiRuntimeServicesPageStart, args->efiRuntimeServicesPageCount,
 199                     kIOHibernatePageStateFree);
 200         *pagesOut -= args->efiRuntimeServicesPageCount;
 201     }
 202 }
 203
 204 kern_return_t
 205 hibernate_processor_setup(IOHibernateImageHeader * header)
 206 {
 207     boot_args * args = (boot_args *) PE_state.bootArgs;
 208
 209     cpu_datap(0)->cpu_hibernate = 1;
 210     header->processorFlags = 0;
 211
 212     header->runtimePages     = args->efiRuntimeServicesPageStart;
 213     header->runtimePageCount = args->efiRuntimeServicesPageCount;
 214     if (args->Version == kBootArgsVersion1 && args->Revision >= kBootArgsRevision1_5) {
 215         header->runtimeVirtualPages = args->efiRuntimeServicesVirtualPageStart;
 216     } else {
 217         header->runtimeVirtualPages = 0;
 218     }
 219
 220     return (KERN_SUCCESS);
 221 }
 222
 223 void
 224 hibernate_vm_lock(void)
 225 {
 226     if (current_cpu_datap()->cpu_hibernate)
 227     {
 228         vm_page_lock_queues();
 229         lck_mtx_lock(&vm_page_queue_free_lock);
 230
 231         if (vm_page_local_q) {
 232                 uint32_t  i;
 233
 234                 for (i = 0; i < vm_page_local_q_count; i++) {
 235                         struct vpl      *lq;
 236
 237                         lq = &vm_page_local_q[i].vpl_un.vpl;
 238
 239                         VPL_LOCK(&lq->vpl_lock);
 240                 }
 241         }
 242     }
 243 }
 244
 245 void
 246 hibernate_vm_unlock(void)
 247 {
 248     if (current_cpu_datap()->cpu_hibernate)
 249     {
 250         if (vm_page_local_q) {
 251                 uint32_t  i;
 252
 253                 for (i = 0; i < vm_page_local_q_count; i++) {
 254                         struct vpl      *lq;
 255
 256                         lq = &vm_page_local_q[i].vpl_un.vpl;
 257
 258                         VPL_UNLOCK(&lq->vpl_lock);
 259                 }
 260         }
 261         lck_mtx_unlock(&vm_page_queue_free_lock);
 262         vm_page_unlock_queues();
 263     }
 264 }