1 /* 2 * Copyright (C) 2014, Red Hat Inc, Andrew Jones <drjones@redhat.com> 3 * 4 * This work is licensed under the terms of the GNU LGPL, version 2. 5 */ 6 #include "alloc.h" 7 #include "asm/spinlock.h" 8 #include "asm/io.h" 9 10 #define MIN(a, b) ((a) < (b) ? (a) : (b)) 11 #define MAX(a, b) ((a) > (b) ? (a) : (b)) 12 13 #define PHYS_ALLOC_NR_REGIONS 256 14 15 struct phys_alloc_region { 16 phys_addr_t base; 17 phys_addr_t size; 18 }; 19 20 static struct phys_alloc_region regions[PHYS_ALLOC_NR_REGIONS]; 21 static int nr_regions; 22 23 static struct spinlock lock; 24 static phys_addr_t base, top, align_min; 25 26 void phys_alloc_show(void) 27 { 28 int i; 29 30 spin_lock(&lock); 31 printf("phys_alloc minimum alignment: 0x%llx\n", align_min); 32 for (i = 0; i < nr_regions; ++i) 33 printf("%016llx-%016llx [%s]\n", 34 regions[i].base, 35 regions[i].base + regions[i].size - 1, 36 "USED"); 37 printf("%016llx-%016llx [%s]\n", base, top - 1, "FREE"); 38 spin_unlock(&lock); 39 } 40 41 void phys_alloc_init(phys_addr_t base_addr, phys_addr_t size) 42 { 43 spin_lock(&lock); 44 base = base_addr; 45 top = base + size; 46 align_min = DEFAULT_MINIMUM_ALIGNMENT; 47 nr_regions = 0; 48 spin_unlock(&lock); 49 } 50 51 void phys_alloc_set_minimum_alignment(phys_addr_t align) 52 { 53 assert(align && !(align & (align - 1))); 54 spin_lock(&lock); 55 align_min = align; 56 spin_unlock(&lock); 57 } 58 59 static phys_addr_t phys_alloc_aligned_safe(phys_addr_t size, 60 phys_addr_t align, bool safe) 61 { 62 static bool warned = false; 63 phys_addr_t addr, size_orig = size; 64 u64 top_safe; 65 66 spin_lock(&lock); 67 68 top_safe = top; 69 70 if (safe && sizeof(long) == 4) 71 top_safe = MIN(top, 1ULL << 32); 72 73 align = MAX(align, align_min); 74 75 addr = ALIGN(base, align); 76 size += addr - base; 77 78 if ((top_safe - base) < size) { 79 printf("phys_alloc: requested=0x%llx (align=0x%llx), " 80 "need=0x%llx, but free=0x%llx. " 81 "top=0x%llx, top_safe=0x%llx\n", 82 size_orig, align, size, top_safe - base, 83 top, top_safe); 84 spin_unlock(&lock); 85 return INVALID_PHYS_ADDR; 86 } 87 88 base += size; 89 90 if (nr_regions < PHYS_ALLOC_NR_REGIONS) { 91 regions[nr_regions].base = addr; 92 regions[nr_regions].size = size_orig; 93 ++nr_regions; 94 } else if (!warned) { 95 printf("WARNING: phys_alloc: No free log entries, " 96 "can no longer log allocations...\n"); 97 warned = true; 98 } 99 100 spin_unlock(&lock); 101 102 return addr; 103 } 104 105 static phys_addr_t phys_zalloc_aligned_safe(phys_addr_t size, 106 phys_addr_t align, bool safe) 107 { 108 phys_addr_t addr = phys_alloc_aligned_safe(size, align, safe); 109 if (addr == INVALID_PHYS_ADDR) 110 return addr; 111 112 memset(phys_to_virt(addr), 0, size); 113 return addr; 114 } 115 116 phys_addr_t phys_alloc_aligned(phys_addr_t size, phys_addr_t align) 117 { 118 return phys_alloc_aligned_safe(size, align, false); 119 } 120 121 phys_addr_t phys_zalloc_aligned(phys_addr_t size, phys_addr_t align) 122 { 123 return phys_zalloc_aligned_safe(size, align, false); 124 } 125 126 phys_addr_t phys_alloc(phys_addr_t size) 127 { 128 return phys_alloc_aligned(size, align_min); 129 } 130 131 phys_addr_t phys_zalloc(phys_addr_t size) 132 { 133 return phys_zalloc_aligned(size, align_min); 134 } 135 136 static void *early_malloc(size_t size) 137 { 138 phys_addr_t addr = phys_alloc_aligned_safe(size, align_min, true); 139 if (addr == INVALID_PHYS_ADDR) 140 return NULL; 141 142 return phys_to_virt(addr); 143 } 144 145 static void *early_calloc(size_t nmemb, size_t size) 146 { 147 phys_addr_t addr = phys_zalloc_aligned_safe(nmemb * size, 148 align_min, true); 149 if (addr == INVALID_PHYS_ADDR) 150 return NULL; 151 152 return phys_to_virt(addr); 153 } 154 155 static void early_free(void *ptr __unused) 156 { 157 } 158 159 static void *early_memalign(size_t alignment, size_t size) 160 { 161 phys_addr_t addr; 162 163 assert(alignment && !(alignment & (alignment - 1))); 164 165 addr = phys_alloc_aligned_safe(size, alignment, true); 166 if (addr == INVALID_PHYS_ADDR) 167 return NULL; 168 169 return phys_to_virt(addr); 170 } 171 172 static struct alloc_ops early_alloc_ops = { 173 .malloc = early_malloc, 174 .calloc = early_calloc, 175 .free = early_free, 176 .memalign = early_memalign, 177 }; 178 179 struct alloc_ops *alloc_ops = &early_alloc_ops; 180