2
* GRUB -- GRand Unified Bootloader
3
* Copyright (C) 2009, 2010 Free Software Foundation, Inc.
5
* GRUB is free software: you can redistribute it and/or modify
6
* it under the terms of the GNU General Public License as published by
7
* the Free Software Foundation, either version 3 of the License, or
8
* (at your option) any later version.
10
* GRUB is distributed in the hope that it will be useful,
11
* but WITHOUT ANY WARRANTY; without even the implied warranty of
12
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
* GNU General Public License for more details.
15
* You should have received a copy of the GNU General Public License
16
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
19
#include <grub/relocator.h>
20
#include <grub/relocator_private.h>
21
#include <grub/mm_private.h>
22
#include <grub/misc.h>
23
#include <grub/cache.h>
24
#include <grub/memory.h>
28
struct grub_relocator_chunk *chunks;
29
grub_phys_addr_t postchunks;
30
grub_phys_addr_t highestaddr;
31
grub_phys_addr_t highestnonpostaddr;
32
grub_size_t relocators_size;
35
struct grub_relocator_subchunk
37
enum {CHUNK_TYPE_IN_REGION, CHUNK_TYPE_REGION_START,
38
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
39
CHUNK_TYPE_FIRMWARE, CHUNK_TYPE_LEFTOVER
43
grub_phys_addr_t start;
46
struct grub_relocator_extra_block *extra;
47
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
48
struct grub_relocator_fw_leftover *pre, *post;
52
struct grub_relocator_chunk
54
struct grub_relocator_chunk *next;
57
grub_phys_addr_t target;
59
struct grub_relocator_subchunk *subchunks;
63
struct grub_relocator_extra_block
65
struct grub_relocator_extra_block *next;
66
struct grub_relocator_extra_block **prev;
67
grub_phys_addr_t start;
71
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
72
struct grub_relocator_fw_leftover
74
struct grub_relocator_fw_leftover *next;
75
struct grub_relocator_fw_leftover **prev;
76
grub_phys_addr_t quantstart;
77
grub_uint8_t freebytes[GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT / 8];
80
struct grub_relocator_fw_leftover *leftovers;
83
struct grub_relocator_extra_block *extra_blocks;
86
get_virtual_current_address (grub_relocator_chunk_t in)
92
get_physical_target_address (grub_relocator_chunk_t in)
97
struct grub_relocator *
98
grub_relocator_new (void)
100
struct grub_relocator *ret;
102
grub_cpu_relocator_init ();
104
ret = grub_zalloc (sizeof (struct grub_relocator));
108
ret->postchunks = ~(grub_phys_addr_t) 0;
109
ret->relocators_size = grub_relocator_jumper_size;
110
grub_dprintf ("relocator", "relocators_size=%lu\n",
111
(unsigned long) ret->relocators_size);
115
#define DIGITSORT_BITS 8
116
#define DIGITSORT_MASK ((1 << DIGITSORT_BITS) - 1)
117
#define BITS_IN_BYTE 8
119
#define max(a, b) (((a) > (b)) ? (a) : (b))
120
#define min(a, b) (((a) < (b)) ? (a) : (b))
125
return !(type & 1) && (type != COLLISION_START);
129
allocate_regstart (grub_phys_addr_t addr, grub_size_t size, grub_mm_region_t rb,
130
grub_mm_region_t *regancestor, grub_mm_header_t hancestor)
132
grub_addr_t newreg_start, newreg_raw_start
133
= (grub_addr_t) rb + (addr - grub_vtop (rb)) + size;
134
grub_addr_t newreg_size, newreg_presize;
135
grub_mm_header_t new_header;
136
grub_mm_header_t hb = (grub_mm_header_t) (rb + 1);
138
grub_dprintf ("relocator", "ra = %p, rb = %p\n", regancestor, rb);
140
newreg_start = ALIGN_UP (newreg_raw_start, GRUB_MM_ALIGN);
141
newreg_presize = newreg_start - newreg_raw_start;
142
newreg_size = rb->size - (newreg_start - (grub_addr_t) rb);
143
if ((hb->size << GRUB_MM_ALIGN_LOG2) >= newreg_start
146
grub_mm_header_t newhnext = hb->next;
147
grub_size_t newhsize = ((hb->size << GRUB_MM_ALIGN_LOG2)
149
- (grub_addr_t) rb)) >> GRUB_MM_ALIGN_LOG2;
150
new_header = (void *) (newreg_start + sizeof (*rb));
152
newhnext = new_header;
153
new_header->next = newhnext;
154
new_header->size = newhsize;
155
new_header->magic = GRUB_MM_FREE_MAGIC;
159
new_header = hb->next;
160
if (new_header == hb)
161
new_header = (void *) (newreg_start + sizeof (*rb));
164
struct grub_mm_header *newregfirst = rb->first;
165
struct grub_mm_region *newregnext = rb->next;
166
struct grub_mm_region *newreg = (void *) newreg_start;
167
hancestor->next = new_header;
168
if (newregfirst == hb)
169
newregfirst = new_header;
170
newreg->first = newregfirst;
171
newreg->next = newregnext;
172
newreg->pre_size = newreg_presize;
173
newreg->size = newreg_size;
174
*regancestor = newreg;
176
grub_mm_header_t h = newreg->first, hp = NULL;
179
if ((void *) h < (void *) (newreg + 1))
180
grub_fatal ("Failed to adjust memory region: %p, %p, %p, %p, %p",
181
newreg, newreg->first, h, hp, hb);
182
if ((void *) h == (void *) (newreg + 1))
183
grub_dprintf ("relocator",
184
"Free start memory region: %p, %p, %p, %p, %p",
185
newreg, newreg->first, h, hp, hb);
190
while (h != newreg->first);
196
allocate_inreg (grub_phys_addr_t paddr, grub_size_t size,
197
grub_mm_header_t hb, grub_mm_header_t hbp,
200
struct grub_mm_header *foll = NULL;
201
grub_addr_t vaddr = (grub_addr_t) hb + (paddr - grub_vtop (hb));
203
grub_dprintf ("relocator", "inreg paddr = 0x%lx, size = %lu,"
204
" hb = %p, hbp = %p, rb = %p, vaddr = 0x%lx\n",
205
(unsigned long) paddr, (unsigned long) size, hb, hbp,
206
rb, (unsigned long) vaddr);
208
if (ALIGN_UP (vaddr + size, GRUB_MM_ALIGN) + GRUB_MM_ALIGN
209
<= (grub_addr_t) (hb + hb->size))
211
foll = (void *) ALIGN_UP (vaddr + size, GRUB_MM_ALIGN);
212
foll->magic = GRUB_MM_FREE_MAGIC;
213
foll->size = hb + hb->size - foll;
214
grub_dprintf ("relocator", "foll = %p, foll->size = %lu\n", foll,
215
(unsigned long) foll->size);
218
if (vaddr - (grub_addr_t) hb >= sizeof (*hb))
220
hb->size = ((vaddr - (grub_addr_t) hb) >> GRUB_MM_ALIGN_LOG2);
235
foll->next = hb->next;
247
rb->first = (void *) (rb + 1);
252
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
254
check_leftover (struct grub_relocator_fw_leftover *lo)
257
for (i = 0; i < sizeof (lo->freebytes); i++)
258
if (lo->freebytes[i] != 0xff)
260
grub_relocator_firmware_free_region (lo->quantstart,
261
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
262
*lo->prev = lo->next;
264
lo->next->prev = lo->prev;
269
free_subchunk (const struct grub_relocator_subchunk *subchu)
271
switch (subchu->type)
273
case CHUNK_TYPE_REGION_START:
275
grub_mm_region_t r1, r2, *rp;
277
grub_size_t pre_size;
279
r2 = (grub_mm_region_t) ALIGN_UP ((grub_addr_t) subchu->reg
280
+ (grub_vtop (subchu->reg)
281
- subchu->start) + subchu->size,
283
for (rp = &grub_mm_base; *rp && *rp != r2; rp = &((*rp)->next));
284
pre_size = subchu->pre_size;
288
grub_mm_header_t h2, *hp;
289
r1->first = r2->first;
291
r1->pre_size = pre_size;
292
r1->size = r2->size + (r2 - r1) * sizeof (*r2);
294
h = (grub_mm_header_t) (r1 + 1);
296
h->magic = GRUB_MM_FREE_MAGIC;
297
h->size = (r2 - r1 - 1);
298
for (hp = &r2->first, h2 = *hp; h2->next != r2->first;
299
hp = &(h2->next), h2 = *hp)
300
if (h2 == (grub_mm_header_t) (r2 + 1))
302
if (h2 == (grub_mm_header_t) (r2 + 1))
304
h->size = h2->size + (h2 - h);
307
if (hp == &r2->first)
309
for (h2 = r2->first; h2->next != r2->first; h2 = h2->next);
320
r1->pre_size = pre_size;
321
r1->size = (r2 - r1) * sizeof (*r2);
322
/* Find where to insert this region.
323
Put a smaller one before bigger ones,
324
to prevent fragmentation. */
325
for (rp = &grub_mm_base; *rp; rp = &((*rp)->next))
326
if ((*rp)->size > r1->size)
330
h = (grub_mm_header_t) (r1 + 1);
333
h->magic = GRUB_MM_FREE_MAGIC;
334
h->size = (r2 - r1 - 1);
336
for (r2 = grub_mm_base; r2; r2 = r2->next)
337
if ((grub_addr_t) r2 + r2->size == (grub_addr_t) r1)
341
grub_mm_header_t hl2, hl, g;
342
g = (grub_mm_header_t) ((grub_addr_t) r2 + r2->size);
343
g->size = (grub_mm_header_t) r1 - g;
344
r2->size += r1->size;
345
for (hl = r2->first; hl->next != r2->first; hl = hl->next);
346
for (hl2 = r1->first; hl2->next != r1->first; hl2 = hl2->next);
347
hl2->next = r2->first;
348
r2->first = r1->first;
349
hl->next = r2->first;
355
case CHUNK_TYPE_IN_REGION:
357
grub_mm_header_t h = (grub_mm_header_t) ALIGN_DOWN ((grub_addr_t) subchu->start,
360
= ((subchu->start + subchu->size + GRUB_MM_ALIGN - 1) / GRUB_MM_ALIGN)
361
- (subchu->start / GRUB_MM_ALIGN) - 1;
363
h->magic = GRUB_MM_ALLOC_MAGIC;
367
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
368
case CHUNK_TYPE_FIRMWARE:
369
case CHUNK_TYPE_LEFTOVER:
371
grub_addr_t fstart, fend;
372
fstart = ALIGN_UP (subchu->start,
373
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
374
fend = ALIGN_DOWN (subchu->start + subchu->size,
375
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
377
grub_relocator_firmware_free_region (fstart, fend - fstart);
378
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
381
int off = subchu->start - fstart
382
- GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT;
383
grub_memset (subchu->pre->freebytes + off / 8 + 1,
384
0xff, sizeof (subchu->pre->freebytes) - off / 8 - 1);
385
subchu->pre->freebytes[off / 8] |= ~((1 << (off % 8)) - 1);
386
check_leftover (subchu->pre);
390
int off = subchu->start + subchu->size - fend;
391
grub_memset (subchu->pre->freebytes,
392
0xff, sizeof (subchu->pre->freebytes) - off / 8);
393
subchu->pre->freebytes[off / 8] |= ((1 << (8 - (off % 8))) - 1);
394
check_leftover (subchu->post);
397
*subchu->extra->prev = subchu->extra->next;
398
grub_free (subchu->extra);
406
malloc_in_range (struct grub_relocator *rel,
407
grub_addr_t start, grub_addr_t end, grub_addr_t align,
408
grub_size_t size, struct grub_relocator_chunk *res,
409
int from_low_priv, int collisioncheck)
411
grub_mm_region_t r, *ra, base_saved;
412
struct grub_relocator_mmap_event *events = NULL, *eventt = NULL, *t;
413
/* 128 is just in case of additional malloc (shouldn't happen). */
414
unsigned maxevents = 2 + 128;
415
grub_mm_header_t p, pa;
419
grub_addr_t target = 0;
421
grub_dprintf ("relocator",
422
"trying to allocate in 0x%lx-0x%lx aligned 0x%lx size 0x%lx\n",
423
(unsigned long) start, (unsigned long) end,
424
(unsigned long) align, (unsigned long) size);
426
start = ALIGN_UP (start, align);
427
end = ALIGN_DOWN (end - size, align) + size;
429
if (end < start + size)
432
/* We have to avoid any allocations when filling scanline events.
435
for (r = grub_mm_base; r; r = r->next)
440
if ((grub_addr_t) p < (grub_addr_t) (r + 1)
441
|| (grub_addr_t) p >= (grub_addr_t) (r + 1) + r->size)
442
grub_fatal ("%d: out of range pointer: %p\n", __LINE__, p);
446
while (p != r->first);
450
if (collisioncheck && rel)
452
struct grub_relocator_chunk *chunk;
453
for (chunk = rel->chunks; chunk; chunk = chunk->next)
457
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
459
struct grub_relocator_extra_block *cur;
460
for (cur = extra_blocks; cur; cur = cur->next)
463
for (r = grub_mm_base; r; r = r->next)
466
maxevents += grub_relocator_firmware_get_max_events ();
469
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
471
struct grub_relocator_fw_leftover *cur;
472
for (cur = leftovers; cur; cur = cur->next)
476
for (i = 0; i < GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT; i++)
478
if (l != ((cur->freebytes[i / 8] >> (i % 8)) & 1))
480
l = ((cur->freebytes[i / 8] >> (i % 8)) & 1);
488
events = grub_malloc (maxevents * sizeof (events[0]));
489
eventt = grub_malloc (maxevents * sizeof (events[0]));
490
counter = grub_malloc ((DIGITSORT_MASK + 2) * sizeof (counter[0]));
491
if (!events || !eventt || !counter)
493
grub_dprintf ("relocator", "events or counter allocation failed %d\n",
501
if (collisioncheck && rel)
503
struct grub_relocator_chunk *chunk;
504
for (chunk = rel->chunks; chunk; chunk = chunk->next)
506
events[N].type = COLLISION_START;
507
events[N].pos = chunk->target;
509
events[N].type = COLLISION_END;
510
events[N].pos = chunk->target + chunk->size;
515
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
516
for (r = grub_mm_base; r; r = r->next)
518
grub_dprintf ("relocator", "Blocking at 0x%lx-0x%lx\n",
519
(unsigned long) r - r->pre_size,
520
(unsigned long) (r + 1) + r->size);
521
events[N].type = FIRMWARE_BLOCK_START;
522
events[N].pos = (grub_addr_t) r - r->pre_size;
524
events[N].type = FIRMWARE_BLOCK_END;
525
events[N].pos = (grub_addr_t) (r + 1) + r->size;
529
struct grub_relocator_extra_block *cur;
530
for (cur = extra_blocks; cur; cur = cur->next)
532
grub_dprintf ("relocator", "Blocking at 0x%lx-0x%lx\n",
533
(unsigned long) cur->start, (unsigned long) cur->end);
534
events[N].type = FIRMWARE_BLOCK_START;
535
events[N].pos = cur->start;
537
events[N].type = FIRMWARE_BLOCK_END;
538
events[N].pos = cur->end;
543
N += grub_relocator_firmware_fill_events (events + N);
545
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
547
struct grub_relocator_fw_leftover *cur;
548
for (cur = leftovers; cur; cur = cur->next)
552
for (i = 0; i < GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT; i++)
554
if (l != ((cur->freebytes[i / 8] >> (i % 8)) & 1))
556
events[N].type = l ? REG_LEFTOVER_END : REG_LEFTOVER_START;
557
events[N].pos = cur->quantstart + i;
558
events[N].leftover = cur;
561
l = ((cur->freebytes[i / 8] >> (i % 8)) & 1);
565
events[N].type = REG_LEFTOVER_END;
566
events[N].pos = cur->quantstart + i;
567
events[N].leftover = cur;
575
/* No malloc from this point. */
576
base_saved = grub_mm_base;
579
for (ra = &base_saved, r = *ra; r; ra = &(r->next), r = *ra)
583
if (p->magic == GRUB_MM_ALLOC_MAGIC)
587
if (p->magic != GRUB_MM_FREE_MAGIC)
588
grub_fatal (__FILE__":%d free magic broken at %p (0x%x)\n",
589
__LINE__, p, p->magic);
590
if (p == (grub_mm_header_t) (r + 1))
592
events[N].type = REG_BEG_START;
593
events[N].pos = grub_vtop (r) - r->pre_size;
595
events[N].regancestor = ra;
597
events[N].hancestor = pa;
599
events[N].type = REG_BEG_END;
600
events[N].pos = grub_vtop (p + p->size) - sizeof (*r)
601
- sizeof (struct grub_mm_header);
606
events[N].type = IN_REG_START;
607
events[N].pos = grub_vtop (p);
609
events[N].hancestor = pa;
612
events[N].type = IN_REG_END;
613
events[N].pos = grub_vtop (p + p->size);
619
while (pa != r->first);
622
/* Put ending events after starting events. */
624
int st = 0, e = N / 2;
625
for (j = 0; j < N; j++)
626
if (is_start (events[j].type) || events[j].type == COLLISION_START)
627
eventt[st++] = events[j];
629
eventt[e++] = events[j];
637
for (i = 0; i < (BITS_IN_BYTE * sizeof (grub_addr_t) / DIGITSORT_BITS);
640
memset (counter, 0, (1 + (1 << DIGITSORT_BITS)) * sizeof (counter[0]));
641
for (j = 0; j < N; j++)
642
counter[((events[j].pos >> (DIGITSORT_BITS * i))
643
& DIGITSORT_MASK) + 1]++;
644
for (j = 0; j <= DIGITSORT_MASK; j++)
645
counter[j+1] += counter[j];
646
for (j = 0; j < N; j++)
647
eventt[counter[((events[j].pos >> (DIGITSORT_BITS * i))
648
& DIGITSORT_MASK)]++] = events[j];
655
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
659
/* Now events are nicely sorted. */
661
int nstarted = 0, ncollisions = 0, nstartedfw = 0, nblockfw = 0;
662
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
665
const int nlefto = 0;
667
grub_addr_t starta = 0;
668
for (j = from_low_priv ? 0 : N - 1; from_low_priv ? j < N : (j + 1);
669
from_low_priv ? j++ : j--)
671
int isinsidebefore, isinsideafter;
672
isinsidebefore = (!ncollisions && (nstarted || (((nlefto || nstartedfw)
674
switch (events[j].type)
676
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
677
case REG_FIRMWARE_START:
681
case REG_FIRMWARE_END:
685
case FIRMWARE_BLOCK_START:
689
case FIRMWARE_BLOCK_END:
694
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
695
case REG_LEFTOVER_START:
699
case REG_LEFTOVER_END:
704
case COLLISION_START:
722
isinsideafter = (!ncollisions && (nstarted || ((nlefto || nstartedfw)
724
if (!isinsidebefore && isinsideafter)
725
starta = from_low_priv ? ALIGN_UP (events[j].pos, align)
726
: ALIGN_DOWN (events[j].pos - size, align) + size;
727
if (isinsidebefore && !isinsideafter && from_low_priv)
732
if (target + size <= end && target + size <= events[j].pos)
733
/* Found an usable address. */
736
if (isinsidebefore && !isinsideafter && !from_low_priv)
738
target = starta - size;
739
if (target > end - size)
741
if (target >= start && target >= events[j].pos)
747
grub_mm_base = base_saved;
755
int inreg = 0, regbeg = 0, ncol = 0;
756
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
757
int fwin = 0, fwb = 0, fwlefto = 0;
760
for (j = 0; j < N; j++)
766
typepre = CHUNK_TYPE_REGION_START;
768
typepre = CHUNK_TYPE_IN_REGION;
769
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
770
else if (fwin && !fwb)
771
typepre = CHUNK_TYPE_FIRMWARE;
772
else if (fwlefto && !fwb)
773
typepre = CHUNK_TYPE_LEFTOVER;
778
if (j != 0 && events[j - 1].pos != events[j].pos)
780
grub_addr_t alloc_start, alloc_end;
781
alloc_start = max (events[j - 1].pos, target);
782
alloc_end = min (events[j].pos, target + size);
783
if (alloc_end > alloc_start)
787
case CHUNK_TYPE_REGION_START:
788
allocate_regstart (alloc_start, alloc_end - alloc_start,
789
events[last_start].reg,
790
events[last_start].regancestor,
791
events[last_start].hancestor);
792
/* TODO: maintain a reverse lookup tree for hancestor. */
795
for (k = 0; k < N; k++)
796
if (events[k].hancestor == events[last_start].head)
797
events[k].hancestor = events[last_start].hancestor;
800
case CHUNK_TYPE_IN_REGION:
801
allocate_inreg (alloc_start, alloc_end - alloc_start,
802
events[last_start].head,
803
events[last_start].hancestor,
804
events[last_start].reg);
807
for (k = 0; k < N; k++)
808
if (events[k].hancestor == events[last_start].head)
809
events[k].hancestor = events[last_start].hancestor;
812
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
813
case CHUNK_TYPE_FIRMWARE:
815
grub_addr_t fstart, fend;
817
= ALIGN_DOWN (alloc_start,
818
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
820
= ALIGN_UP (alloc_end,
821
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
822
grub_dprintf ("relocator", "requesting %lx-%lx\n",
823
(unsigned long) fstart,
824
(unsigned long) fend);
825
/* The failure here can be very expensive. */
826
if (!grub_relocator_firmware_alloc_region (fstart,
839
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
840
case CHUNK_TYPE_LEFTOVER:
842
unsigned offstart = alloc_start
843
% GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT;
844
unsigned offend = alloc_end
845
% GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT;
846
struct grub_relocator_fw_leftover *lo
847
= events[last_start].leftover;
848
lo->freebytes[offstart / 8]
849
&= ((1 << (8 - (start % 8))) - 1);
850
grub_memset (lo->freebytes + (offstart + 7) / 8, 0,
851
offend / 8 - (offstart + 7) / 8);
852
lo->freebytes[offend / 8] &= ~((1 << (offend % 8)) - 1);
861
switch (events[j].type)
865
if (events[j].type == REG_BEG_START &&
866
(grub_addr_t) (events[j].reg + 1) > target)
881
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
882
case REG_FIRMWARE_START:
886
case REG_FIRMWARE_END:
890
case FIRMWARE_BLOCK_START:
894
case FIRMWARE_BLOCK_END:
899
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
900
case REG_LEFTOVER_START:
904
case REG_LEFTOVER_END:
908
case COLLISION_START:
919
/* Malloc is available again. */
920
grub_mm_base = base_saved;
924
int inreg = 0, regbeg = 0, ncol = 0;
925
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
926
int fwin = 0, fwlefto = 0, fwb = 0;
930
res->subchunks = grub_malloc (sizeof (res->subchunks[0]) * nallocs);
933
res->nsubchunks = nallocs;
935
for (j = 0; j < N; j++)
941
typepre = CHUNK_TYPE_REGION_START;
943
typepre = CHUNK_TYPE_IN_REGION;
944
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
945
else if (fwin && !fwb)
946
typepre = CHUNK_TYPE_FIRMWARE;
947
else if (fwlefto && !fwb)
948
typepre = CHUNK_TYPE_LEFTOVER;
953
if (j != 0 && events[j - 1].pos != events[j].pos)
955
grub_addr_t alloc_start, alloc_end;
956
struct grub_relocator_subchunk tofree;
957
struct grub_relocator_subchunk *curschu = &tofree;
959
curschu = &res->subchunks[cural];
960
alloc_start = max (events[j - 1].pos, target);
961
alloc_end = min (events[j].pos, target + size);
962
if (alloc_end > alloc_start)
964
grub_dprintf ("relocator", "subchunk 0x%lx-0x%lx, %d\n",
965
(unsigned long) alloc_start,
966
(unsigned long) alloc_end, typepre);
967
curschu->type = typepre;
968
curschu->start = alloc_start;
969
curschu->size = alloc_end - alloc_start;
970
if (typepre == CHUNK_TYPE_REGION_START
971
|| typepre == CHUNK_TYPE_IN_REGION)
973
curschu->reg = events[last_start].reg;
974
curschu->pre_size = alloc_start - events[j - 1].pos;
976
if (!oom && (typepre == CHUNK_TYPE_REGION_START
977
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
978
|| typepre == CHUNK_TYPE_FIRMWARE
982
struct grub_relocator_extra_block *ne;
983
ne = grub_malloc (sizeof (*ne));
987
grub_memcpy (&tofree, curschu, sizeof (tofree));
991
ne->start = alloc_start;
993
ne->next = extra_blocks;
994
ne->prev = &extra_blocks;
996
extra_blocks->prev = &(ne->next);
1001
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
1002
if (!oom && typepre == CHUNK_TYPE_FIRMWARE)
1004
grub_addr_t fstart, fend;
1007
= ALIGN_DOWN (alloc_start,
1008
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
1010
= ALIGN_UP (alloc_end,
1011
GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT);
1013
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
1015
struct grub_relocator_fw_leftover *lo1 = NULL;
1016
struct grub_relocator_fw_leftover *lo2 = NULL;
1017
if (fstart != alloc_start)
1018
lo1 = grub_malloc (sizeof (*lo1));
1019
if (fend != alloc_end)
1020
lo2 = grub_malloc (sizeof (*lo2));
1021
if ((!lo1 && fstart != alloc_start)
1022
|| (!lo2 && fend != alloc_end))
1024
struct grub_relocator_extra_block *ne;
1030
grub_memcpy (&tofree, curschu, sizeof (tofree));
1032
extra_blocks = extra_blocks->next;
1037
lo1->quantstart = fstart;
1038
grub_memset (lo1->freebytes, 0xff,
1039
(alloc_start - fstart) / 8);
1040
lo1->freebytes[(alloc_start - fstart) / 8]
1041
= (1 << ((alloc_start - fstart) % 8)) - 1;
1042
grub_memset (lo1->freebytes
1043
+ ((alloc_start - fstart) / 8) + 1, 0,
1044
sizeof (lo1->freebytes)
1045
- (alloc_start - fstart) / 8 - 1);
1046
lo1->next = leftovers;
1047
lo1->prev = &leftovers;
1049
leftovers->prev = &lo1->next;
1055
= fend - GRUB_RELOCATOR_FIRMWARE_REQUESTS_QUANT;
1056
grub_memset (lo2->freebytes, 0,
1057
(alloc_end - lo2->quantstart) / 8);
1058
lo2->freebytes[(alloc_end - lo2->quantstart) / 8]
1059
= ~((1 << ((alloc_end - lo2->quantstart) % 8)) - 1);
1060
grub_memset (lo2->freebytes
1061
+ ((alloc_end - lo2->quantstart) / 8)
1062
+ 1, 0, sizeof (lo2->freebytes)
1063
- (alloc_end - lo2->quantstart) / 8 - 1);
1064
lo2->prev = &leftovers;
1066
leftovers->prev = &lo2->next;
1067
lo2->next = leftovers;
1071
curschu->post = lo2;
1076
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
1077
if (typepre == CHUNK_TYPE_LEFTOVER)
1079
curschu->pre = events[last_start].leftover;
1080
curschu->post = events[last_start].leftover;
1088
free_subchunk (&tofree);
1092
switch (events[j].type)
1096
if (events[j].type == REG_BEG_START &&
1097
(grub_addr_t) (events[j].reg + 1) > target)
1109
#if GRUB_RELOCATOR_HAVE_FIRMWARE_REQUESTS
1110
case REG_FIRMWARE_START:
1114
case REG_FIRMWARE_END:
1118
case FIRMWARE_BLOCK_START:
1122
case FIRMWARE_BLOCK_END:
1127
#if GRUB_RELOCATOR_HAVE_LEFTOVERS
1128
case REG_LEFTOVER_START:
1132
case REG_LEFTOVER_END:
1136
case COLLISION_START:
1147
for (i = 0; i < cural; i++)
1148
free_subchunk (&res->subchunks[i]);
1149
grub_free (res->subchunks);
1150
grub_dprintf ("relocator", "allocation failed with out-of-memory\n");
1157
grub_dprintf ("relocator", "allocated: 0x%lx+0x%lx\n", (unsigned long) target,
1158
(unsigned long) size);
1164
adjust_limits (struct grub_relocator *rel,
1165
grub_phys_addr_t *min_addr, grub_phys_addr_t *max_addr,
1166
grub_phys_addr_t in_min, grub_phys_addr_t in_max)
1168
struct grub_relocator_chunk *chunk;
1171
*max_addr = rel->postchunks;
1173
/* Keep chunks in memory in the same order as they'll be after relocation. */
1174
for (chunk = rel->chunks; chunk; chunk = chunk->next)
1176
if (chunk->target > in_max && chunk->src < *max_addr
1177
&& chunk->src < rel->postchunks)
1178
*max_addr = chunk->src;
1179
if (chunk->target + chunk->size <= in_min
1180
&& chunk->src + chunk->size > *min_addr
1181
&& chunk->src < rel->postchunks)
1182
*min_addr = chunk->src + chunk->size;
1187
grub_relocator_alloc_chunk_addr (struct grub_relocator *rel,
1188
grub_relocator_chunk_t *out,
1189
grub_phys_addr_t target, grub_size_t size)
1191
struct grub_relocator_chunk *chunk;
1192
grub_phys_addr_t min_addr = 0, max_addr;
1195
return grub_error (GRUB_ERR_OUT_OF_RANGE, "address is out of range");
1197
adjust_limits (rel, &min_addr, &max_addr, target, target);
1199
for (chunk = rel->chunks; chunk; chunk = chunk->next)
1200
if ((chunk->target <= target && target < chunk->target + chunk->size)
1201
|| (target <= chunk->target && chunk->target < target + size))
1202
return grub_error (GRUB_ERR_BAD_ARGUMENT, "overlap detected");
1204
chunk = grub_malloc (sizeof (struct grub_relocator_chunk));
1208
grub_dprintf ("relocator",
1209
"min_addr = 0x%llx, max_addr = 0x%llx, target = 0x%llx\n",
1210
(unsigned long long) min_addr, (unsigned long long) max_addr,
1211
(unsigned long long) target);
1215
/* A trick to improve Linux allocation. */
1216
#if defined (__i386__) || defined (__x86_64__)
1217
if (target < 0x100000)
1218
if (malloc_in_range (rel, rel->highestnonpostaddr, ~(grub_addr_t)0, 1,
1221
if (rel->postchunks > chunk->src)
1222
rel->postchunks = chunk->src;
1226
if (malloc_in_range (rel, target, max_addr, 1, size, chunk, 1, 0))
1229
if (malloc_in_range (rel, min_addr, target, 1, size, chunk, 0, 0))
1232
if (malloc_in_range (rel, rel->highestnonpostaddr, ~(grub_addr_t)0, 1,
1235
if (rel->postchunks > chunk->src)
1236
rel->postchunks = chunk->src;
1240
grub_dprintf ("relocator", "not allocated\n");
1242
return grub_error (GRUB_ERR_OUT_OF_MEMORY, "out of memory");
1246
grub_dprintf ("relocator", "allocated 0x%llx/0x%llx\n",
1247
(unsigned long long) chunk->src, (unsigned long long) target);
1249
if (rel->highestaddr < target + size)
1250
rel->highestaddr = target + size;
1252
if (rel->highestaddr < chunk->src + size)
1253
rel->highestaddr = chunk->src + size;
1255
if (chunk->src < rel->postchunks)
1257
if (rel->highestnonpostaddr < target + size)
1258
rel->highestnonpostaddr = target + size;
1260
if (rel->highestnonpostaddr < chunk->src + size)
1261
rel->highestnonpostaddr = chunk->src + size;
1264
grub_dprintf ("relocator", "relocators_size=%ld\n",
1265
(unsigned long) rel->relocators_size);
1267
if (chunk->src < target)
1268
rel->relocators_size += grub_relocator_backward_size;
1269
if (chunk->src > target)
1270
rel->relocators_size += grub_relocator_forward_size;
1272
grub_dprintf ("relocator", "relocators_size=%ld\n",
1273
(unsigned long) rel->relocators_size);
1275
chunk->target = target;
1277
chunk->next = rel->chunks;
1278
rel->chunks = chunk;
1279
grub_dprintf ("relocator", "cur = %p, next = %p\n", rel->chunks,
1282
chunk->srcv = grub_map_memory (chunk->src, chunk->size);
1284
#ifdef DEBUG_RELOCATOR
1288
grub_memset (chunk->srcv, 0xfa, chunk->size);
1289
for (r = grub_mm_base; r; r = r->next)
1294
if ((grub_addr_t) p < (grub_addr_t) (r + 1)
1295
|| (grub_addr_t) p >= (grub_addr_t) (r + 1) + r->size)
1296
grub_fatal (__FILE__ ":%d: out of range pointer: %p\n", __LINE__, p);
1299
while (p != r->first);
1303
return GRUB_ERR_NONE;
1307
grub_relocator_alloc_chunk_align (struct grub_relocator *rel,
1308
grub_relocator_chunk_t *out,
1309
grub_phys_addr_t min_addr,
1310
grub_phys_addr_t max_addr,
1311
grub_size_t size, grub_size_t align,
1314
grub_addr_t min_addr2 = 0, max_addr2;
1315
struct grub_relocator_chunk *chunk;
1317
if (max_addr > ~size)
1320
#ifdef GRUB_MACHINE_PCBIOS
1321
if (min_addr < 0x1000)
1325
grub_dprintf ("relocator", "chunks = %p\n", rel->chunks);
1327
chunk = grub_malloc (sizeof (struct grub_relocator_chunk));
1331
if (malloc_in_range (rel, min_addr, max_addr, align,
1333
preference != GRUB_RELOCATOR_PREFERENCE_HIGH, 1))
1335
grub_dprintf ("relocator", "allocated 0x%llx/0x%llx\n",
1336
(unsigned long long) chunk->src,
1337
(unsigned long long) chunk->src);
1338
grub_dprintf ("relocator", "chunks = %p\n", rel->chunks);
1339
chunk->target = chunk->src;
1341
chunk->next = rel->chunks;
1342
rel->chunks = chunk;
1343
chunk->srcv = grub_map_memory (chunk->src, chunk->size);
1345
return GRUB_ERR_NONE;
1348
adjust_limits (rel, &min_addr2, &max_addr2, min_addr, max_addr);
1349
grub_dprintf ("relocator", "Adjusted limits from %lx-%lx to %lx-%lx\n",
1350
(unsigned long) min_addr, (unsigned long) max_addr,
1351
(unsigned long) min_addr2, (unsigned long) max_addr2);
1355
if (malloc_in_range (rel, min_addr2, max_addr2, align,
1359
if (malloc_in_range (rel, rel->highestnonpostaddr, ~(grub_addr_t)0, 1,
1362
if (rel->postchunks > chunk->src)
1363
rel->postchunks = chunk->src;
1367
return grub_error (GRUB_ERR_OUT_OF_MEMORY, "out of memory");
1373
auto int NESTED_FUNC_ATTR hook (grub_uint64_t, grub_uint64_t,
1374
grub_memory_type_t);
1375
int NESTED_FUNC_ATTR hook (grub_uint64_t addr, grub_uint64_t sz,
1376
grub_memory_type_t type)
1378
grub_uint64_t candidate;
1379
if (type != GRUB_MEMORY_AVAILABLE)
1381
candidate = ALIGN_UP (addr, align);
1382
if (candidate < min_addr)
1383
candidate = ALIGN_UP (min_addr, align);
1384
if (candidate + size > addr + sz
1385
|| candidate > ALIGN_DOWN (max_addr, align))
1387
if (preference == GRUB_RELOCATOR_PREFERENCE_HIGH)
1388
candidate = ALIGN_DOWN (min (addr + sz - size, max_addr), align);
1389
if (!found || (preference == GRUB_RELOCATOR_PREFERENCE_HIGH
1390
&& candidate > chunk->target))
1391
chunk->target = candidate;
1392
if (!found || (preference == GRUB_RELOCATOR_PREFERENCE_LOW
1393
&& candidate < chunk->target))
1394
chunk->target = candidate;
1399
grub_machine_mmap_iterate (hook);
1401
return grub_error (GRUB_ERR_BAD_OS, "couldn't find suitable memory target");
1405
struct grub_relocator_chunk *chunk2;
1406
for (chunk2 = rel->chunks; chunk2; chunk2 = chunk2->next)
1407
if ((chunk2->target <= chunk->target
1408
&& chunk->target < chunk2->target + chunk2->size)
1409
|| (chunk2->target <= chunk->target + size
1410
&& chunk->target + size < chunk2->target + chunk2->size)
1411
|| (chunk->target <= chunk2->target && chunk2->target
1412
< chunk->target + size)
1413
|| (chunk->target <= chunk2->target + chunk2->size
1414
&& chunk2->target + chunk2->size < chunk->target + size))
1416
if (preference == GRUB_RELOCATOR_PREFERENCE_HIGH)
1417
chunk->target = ALIGN_DOWN (chunk2->target, align);
1419
chunk->target = ALIGN_UP (chunk2->target + chunk2->size, align);
1426
if (chunk->src < chunk->target)
1427
rel->relocators_size += grub_relocator_backward_size;
1428
if (chunk->src > chunk->target)
1429
rel->relocators_size += grub_relocator_forward_size;
1432
chunk->next = rel->chunks;
1433
rel->chunks = chunk;
1434
grub_dprintf ("relocator", "cur = %p, next = %p\n", rel->chunks,
1436
chunk->srcv = grub_map_memory (chunk->src, chunk->size);
1438
#ifdef DEBUG_RELOCATOR
1443
grub_memset (chunk->srcv, 0xfa, chunk->size);
1444
for (r = grub_mm_base; r; r = r->next)
1449
if ((grub_addr_t) p < (grub_addr_t) (r + 1)
1450
|| (grub_addr_t) p >= (grub_addr_t) (r + 1) + r->size)
1451
grub_fatal (__FILE__ "%d: out of range pointer: %p\n", __LINE__, p);
1454
while (p != r->first);
1458
return GRUB_ERR_NONE;
1462
grub_relocator_unload (struct grub_relocator *rel)
1464
struct grub_relocator_chunk *chunk, *next;
1467
for (chunk = rel->chunks; chunk; chunk = next)
1470
for (i = 0; i < chunk->nsubchunks; i++)
1471
free_subchunk (&chunk->subchunks[i]);
1472
grub_unmap_memory (chunk->srcv, chunk->size);
1474
grub_free (chunk->subchunks);
1481
grub_relocator_prepare_relocs (struct grub_relocator *rel, grub_addr_t addr,
1482
void **relstart, grub_size_t *relsize)
1485
grub_uint8_t *rels0;
1486
struct grub_relocator_chunk *sorted;
1487
grub_size_t nchunks = 0;
1489
struct grub_relocator_chunk movers_chunk;
1491
grub_dprintf ("relocator", "Preparing relocs (size=%ld)\n",
1492
(unsigned long) rel->relocators_size);
1494
if (!malloc_in_range (rel, 0, ~(grub_addr_t)0 - rel->relocators_size + 1,
1495
grub_relocator_align,
1496
rel->relocators_size, &movers_chunk, 1, 1))
1497
return grub_error (GRUB_ERR_OUT_OF_MEMORY, "out of memory");
1498
movers_chunk.srcv = rels = rels0
1499
= grub_map_memory (movers_chunk.src, movers_chunk.size);
1502
*relsize = rel->relocators_size;
1504
grub_dprintf ("relocator", "Relocs allocated at %p\n", movers_chunk.srcv);
1508
grub_size_t count[257];
1509
struct grub_relocator_chunk *from, *to, *tmp;
1511
grub_memset (count, 0, sizeof (count));
1514
struct grub_relocator_chunk *chunk;
1515
for (chunk = rel->chunks; chunk; chunk = chunk->next)
1517
grub_dprintf ("relocator", "chunk %p->%p, 0x%lx\n",
1518
(void *) chunk->src, (void *) chunk->target,
1519
(unsigned long) chunk->size);
1521
count[(chunk->src & 0xff) + 1]++;
1524
from = grub_malloc (nchunks * sizeof (sorted[0]));
1525
to = grub_malloc (nchunks * sizeof (sorted[0]));
1533
for (j = 0; j < 256; j++)
1534
count[j+1] += count[j];
1537
struct grub_relocator_chunk *chunk;
1538
for (chunk = rel->chunks; chunk; chunk = chunk->next)
1539
from[count[chunk->src & 0xff]++] = *chunk;
1542
for (i = 1; i < GRUB_CPU_SIZEOF_VOID_P; i++)
1544
grub_memset (count, 0, sizeof (count));
1545
for (j = 0; j < nchunks; j++)
1546
count[((from[j].src >> (8 * i)) & 0xff) + 1]++;
1547
for (j = 0; j < 256; j++)
1548
count[j+1] += count[j];
1549
for (j = 0; j < nchunks; j++)
1550
to[count[(from[j].src >> (8 * i)) & 0xff]++] = from[j];
1559
for (j = 0; j < nchunks; j++)
1561
grub_dprintf ("relocator", "sorted chunk %p->%p, 0x%lx\n",
1562
(void *) sorted[j].src, (void *) sorted[j].target,
1563
(unsigned long) sorted[j].size);
1564
if (sorted[j].src < sorted[j].target)
1566
grub_cpu_relocator_backward ((void *) rels,
1568
grub_map_memory (sorted[j].target,
1571
rels += grub_relocator_backward_size;
1573
if (sorted[j].src > sorted[j].target)
1575
grub_cpu_relocator_forward ((void *) rels,
1577
grub_map_memory (sorted[j].target,
1580
rels += grub_relocator_forward_size;
1582
if (sorted[j].src == sorted[j].target)
1583
grub_arch_sync_caches (sorted[j].srcv, sorted[j].size);
1585
grub_cpu_relocator_jumper ((void *) rels, (grub_addr_t) addr);
1588
return GRUB_ERR_NONE;
1592
grub_mm_check_real (char *file, int line)
1595
grub_mm_header_t p, pa;
1597
for (r = grub_mm_base; r; r = r->next)
1601
if (p->magic == GRUB_MM_ALLOC_MAGIC)
1605
if ((grub_addr_t) p < (grub_addr_t) (r + 1)
1606
|| (grub_addr_t) p >= (grub_addr_t) (r + 1) + r->size)
1607
grub_fatal ("%s:%d: out of range pointer: %p\n", file, line, p);
1608
if (p->magic != GRUB_MM_FREE_MAGIC)
1609
grub_fatal ("%s:%d free magic broken at %p (0x%x)\n", file,
1614
while (pa != r->first);