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 spin_unlock(&lock); 48 } 49 50 void phys_alloc_set_minimum_alignment(phys_addr_t align) 51 { 52 assert(align && !(align & (align - 1))); 53 spin_lock(&lock); 54 align_min = align; 55 spin_unlock(&lock); 56 } 57 58 static phys_addr_t phys_alloc_aligned_safe(phys_addr_t size, 59 phys_addr_t align, bool safe) 60 { 61 phys_addr_t addr, size_orig = size; 62 u64 top_safe = top; 63 64 if (safe && sizeof(long) == 4) 65 top_safe = MIN(top, 1ULL << 32); 66 67 align = MAX(align, align_min); 68 69 spin_lock(&lock); 70 71 addr = ALIGN(base, align); 72 size += addr - base; 73 74 if ((top_safe - base) < size) { 75 printf("%s: requested=0x%llx (align=0x%llx), " 76 "need=0x%llx, but free=0x%llx. " 77 "top=0x%llx, top_safe=0x%llx\n", __func__, 78 size_orig, align, size, top_safe - base, 79 top, top_safe); 80 spin_unlock(&lock); 81 return INVALID_PHYS_ADDR; 82 } 83 84 base += size; 85 86 if (nr_regions < PHYS_ALLOC_NR_REGIONS) { 87 regions[nr_regions].base = addr; 88 regions[nr_regions].size = size_orig; 89 ++nr_regions; 90 } else { 91 printf("%s: WARNING: no free log entries, " 92 "can't log allocation...\n", __func__); 93 } 94 95 spin_unlock(&lock); 96 97 return addr; 98 } 99 100 static phys_addr_t phys_zalloc_aligned_safe(phys_addr_t size, 101 phys_addr_t align, bool safe) 102 { 103 phys_addr_t addr = phys_alloc_aligned_safe(size, align, safe); 104 if (addr == INVALID_PHYS_ADDR) 105 return addr; 106 107 memset(phys_to_virt(addr), 0, size); 108 return addr; 109 } 110 111 phys_addr_t phys_alloc_aligned(phys_addr_t size, phys_addr_t align) 112 { 113 return phys_alloc_aligned_safe(size, align, false); 114 } 115 116 phys_addr_t phys_zalloc_aligned(phys_addr_t size, phys_addr_t align) 117 { 118 return phys_zalloc_aligned_safe(size, align, false); 119 } 120 121 phys_addr_t phys_alloc(phys_addr_t size) 122 { 123 return phys_alloc_aligned(size, align_min); 124 } 125 126 phys_addr_t phys_zalloc(phys_addr_t size) 127 { 128 return phys_zalloc_aligned(size, align_min); 129 } 130 131 static void *early_malloc(size_t size) 132 { 133 phys_addr_t addr = phys_alloc_aligned_safe(size, align_min, true); 134 if (addr == INVALID_PHYS_ADDR) 135 return NULL; 136 137 return phys_to_virt(addr); 138 } 139 140 static void *early_calloc(size_t nmemb, size_t size) 141 { 142 phys_addr_t addr = phys_zalloc_aligned_safe(nmemb * size, 143 align_min, true); 144 if (addr == INVALID_PHYS_ADDR) 145 return NULL; 146 147 return phys_to_virt(addr); 148 } 149 150 static void early_free(void *ptr __unused) 151 { 152 } 153 154 static void *early_memalign(size_t alignment, size_t size) 155 { 156 phys_addr_t addr; 157 158 assert(alignment && !(alignment & (alignment - 1))); 159 160 addr = phys_alloc_aligned_safe(size, alignment, true); 161 if (addr == INVALID_PHYS_ADDR) 162 return NULL; 163 164 return phys_to_virt(addr); 165 } 166 167 static struct alloc_ops early_alloc_ops = { 168 .malloc = early_malloc, 169 .calloc = early_calloc, 170 .free = early_free, 171 .memalign = early_memalign, 172 }; 173 174 struct alloc_ops *alloc_ops = &early_alloc_ops; 175