Subversion Repositories HelenOS

Rev

Rev 3973 | Rev 4230 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
1 jermar 1
/*
2071 jermar 2
 * Copyright (c) 2001-2005 Jakub Jermar
3
 * Copyright (c) 2005 Sergey Bondari
3972 decky 4
 * Copyright (c) 2009 Martin Decky
1 jermar 5
 * All rights reserved.
6
 *
7
 * Redistribution and use in source and binary forms, with or without
8
 * modification, are permitted provided that the following conditions
9
 * are met:
10
 *
11
 * - Redistributions of source code must retain the above copyright
12
 *   notice, this list of conditions and the following disclaimer.
13
 * - Redistributions in binary form must reproduce the above copyright
14
 *   notice, this list of conditions and the following disclaimer in the
15
 *   documentation and/or other materials provided with the distribution.
16
 * - The name of the author may not be used to endorse or promote products
17
 *   derived from this software without specific prior written permission.
18
 *
19
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29
 */
30
 
1757 jermar 31
/** @addtogroup genericmm
1702 cejka 32
 * @{
33
 */
34
 
1248 jermar 35
/**
1702 cejka 36
 * @file
3972 decky 37
 * @brief Physical frame allocator.
1248 jermar 38
 *
39
 * This file contains the physical frame allocator and memory zone management.
40
 * The frame allocator is built on top of the buddy allocator.
41
 *
42
 * @see buddy.c
43
 */
44
 
1 jermar 45
#include <arch/types.h>
46
#include <mm/frame.h>
703 jermar 47
#include <mm/as.h>
1 jermar 48
#include <panic.h>
367 jermar 49
#include <debug.h>
788 jermar 50
#include <adt/list.h>
3187 jermar 51
#include <synch/mutex.h>
52
#include <synch/condvar.h>
115 jermar 53
#include <arch/asm.h>
195 vana 54
#include <arch.h>
533 bondari 55
#include <print.h>
536 bondari 56
#include <align.h>
814 palkovsky 57
#include <mm/slab.h>
822 palkovsky 58
#include <bitops.h>
1063 palkovsky 59
#include <macros.h>
2128 jermar 60
#include <config.h>
115 jermar 61
 
3973 decky 62
zones_t zones;
367 jermar 63
 
814 palkovsky 64
/*
3187 jermar 65
 * Synchronization primitives used to sleep when there is no memory
66
 * available.
67
 */
3188 jermar 68
mutex_t mem_avail_mtx;
69
condvar_t mem_avail_cv;
3972 decky 70
count_t mem_avail_req = 0;  /**< Number of frames requested. */
71
count_t mem_avail_gen = 0;  /**< Generation counter. */
2121 decky 72
 
2725 decky 73
/********************/
814 palkovsky 74
/* Helper functions */
2725 decky 75
/********************/
76
 
814 palkovsky 77
static inline index_t frame_index(zone_t *zone, frame_t *frame)
762 palkovsky 78
{
2725 decky 79
    return (index_t) (frame - zone->frames);
762 palkovsky 80
}
2725 decky 81
 
814 palkovsky 82
static inline index_t frame_index_abs(zone_t *zone, frame_t *frame)
1 jermar 83
{
2725 decky 84
    return (index_t) (frame - zone->frames) + zone->base;
1 jermar 85
}
2725 decky 86
 
3972 decky 87
static inline bool frame_index_valid(zone_t *zone, index_t index)
814 palkovsky 88
{
2745 decky 89
    return (index < zone->count);
814 palkovsky 90
}
1 jermar 91
 
3972 decky 92
static inline index_t make_frame_index(zone_t *zone, frame_t *frame)
762 palkovsky 93
{
2725 decky 94
    return (frame - zone->frames);
762 palkovsky 95
}
96
 
3185 jermar 97
/** Initialize frame structure.
762 palkovsky 98
 *
3972 decky 99
 * @param frame Frame structure to be initialized.
100
 *
762 palkovsky 101
 */
814 palkovsky 102
static void frame_initialize(frame_t *frame)
762 palkovsky 103
{
814 palkovsky 104
    frame->refcount = 1;
105
    frame->buddy_order = 0;
762 palkovsky 106
}
107
 
3972 decky 108
/*******************/
109
/* Zones functions */
110
/*******************/
762 palkovsky 111
 
3185 jermar 112
/** Insert-sort zone into zones list.
822 palkovsky 113
 *
3972 decky 114
 * Assume interrupts are disabled and zones lock is
115
 * locked.
116
 *
117
 * @param base  Base frame of the newly inserted zone.
118
 * @param count Number of frames of the newly inserted zone.
119
 *
120
 * @return Zone number on success, -1 on error.
121
 *
1 jermar 122
 */
3972 decky 123
static count_t zones_insert_zone(pfn_t base, count_t count)
1 jermar 124
{
2725 decky 125
    if (zones.count + 1 == ZONES_MAX) {
126
        printf("Maximum zone count %u exceeded!\n", ZONES_MAX);
3972 decky 127
        return (count_t) -1;
2725 decky 128
    }
129
 
3972 decky 130
    count_t i;
1037 decky 131
    for (i = 0; i < zones.count; i++) {
3972 decky 132
        /* Check for overlap */
133
        if (overlaps(base, count,
134
            zones.info[i].base, zones.info[i].count)) {
822 palkovsky 135
            printf("Zones overlap!\n");
3972 decky 136
            return (count_t) -1;
814 palkovsky 137
        }
3972 decky 138
        if (base < zones.info[i].base)
822 palkovsky 139
            break;
368 jermar 140
    }
2725 decky 141
 
822 palkovsky 142
    /* Move other zones up */
3972 decky 143
    count_t j;
3973 decky 144
    for (j = zones.count; j > i; j--) {
145
        zones.info[j] = zones.info[j - 1];
146
        zones.info[j].buddy_system->data =
147
            (void *) &zones.info[j - 1];
148
    }
2725 decky 149
 
822 palkovsky 150
    zones.count++;
2725 decky 151
 
822 palkovsky 152
    return i;
1 jermar 153
}
154
 
3972 decky 155
/** Get total available frames.
2725 decky 156
 *
3972 decky 157
 * Assume interrupts are disabled and zones lock is
158
 * locked.
2725 decky 159
 *
3972 decky 160
 * @return Total number of available frames.
161
 *
368 jermar 162
 */
4048 jermar 163
#ifdef CONFIG_DEBUG
3972 decky 164
static count_t total_frames_free(void)
1 jermar 165
{
3972 decky 166
    count_t total = 0;
167
    count_t i;
168
    for (i = 0; i < zones.count; i++)
169
        total += zones.info[i].free_count;
533 bondari 170
 
3972 decky 171
    return total;
172
}
4048 jermar 173
#endif
814 palkovsky 174
 
3973 decky 175
/** Find a zone with a given frames.
3972 decky 176
 *
177
 * Assume interrupts are disabled and zones lock is
178
 * locked.
179
 *
180
 * @param frame Frame number contained in zone.
3973 decky 181
 * @param count Number of frames to look for.
3972 decky 182
 * @param hint  Used as zone hint.
183
 *
184
 * @return Zone index or -1 if not found.
185
 *
186
 */
3973 decky 187
count_t find_zone(pfn_t frame, count_t count, count_t hint)
3972 decky 188
{
2745 decky 189
    if (hint >= zones.count)
814 palkovsky 190
        hint = 0;
533 bondari 191
 
3972 decky 192
    count_t i = hint;
814 palkovsky 193
    do {
3972 decky 194
        if ((zones.info[i].base <= frame)
3973 decky 195
            && (zones.info[i].base + zones.info[i].count >= frame + count))
3972 decky 196
            return i;
197
 
814 palkovsky 198
        i++;
199
        if (i >= zones.count)
200
            i = 0;
3207 jermar 201
    } while (i != hint);
3972 decky 202
 
203
    return (count_t) -1;
533 bondari 204
}
205
 
822 palkovsky 206
/** @return True if zone can allocate specified order */
3972 decky 207
static bool zone_can_alloc(zone_t *zone, uint8_t order)
822 palkovsky 208
{
3972 decky 209
    return (zone_flags_available(zone->flags)
210
        && buddy_system_can_alloc(zone->buddy_system, order));
822 palkovsky 211
}
212
 
3972 decky 213
/** Find a zone that can allocate order frames.
367 jermar 214
 *
3972 decky 215
 * Assume interrupts are disabled and zones lock is
216
 * locked.
367 jermar 217
 *
3972 decky 218
 * @param order Size (2^order) of free space we are trying to find.
219
 * @param flags Required flags of the target zone.
220
 * @param hind  Preferred zone.
221
 *
367 jermar 222
 */
3972 decky 223
static count_t find_free_zone(uint8_t order, zone_flags_t flags, count_t hint)
367 jermar 224
{
814 palkovsky 225
    if (hint >= zones.count)
226
        hint = 0;
3972 decky 227
 
228
    count_t i = hint;
814 palkovsky 229
    do {
3207 jermar 230
        /*
231
         * Check whether the zone meets the search criteria.
232
         */
3972 decky 233
        if ((zones.info[i].flags & flags) == flags) {
3207 jermar 234
            /*
235
             * Check if the zone has 2^order frames area available.
236
             */
3972 decky 237
            if (zone_can_alloc(&zones.info[i], order))
238
                return i;
367 jermar 239
        }
3972 decky 240
 
241
        i++;
242
        if (i >= zones.count)
814 palkovsky 243
            i = 0;
3207 jermar 244
    } while (i != hint);
3972 decky 245
 
246
    return (count_t) -1;
367 jermar 247
}
248
 
2059 jermar 249
/**************************/
814 palkovsky 250
/* Buddy system functions */
2059 jermar 251
/**************************/
814 palkovsky 252
 
3185 jermar 253
/** Buddy system find_block implementation.
367 jermar 254
 *
814 palkovsky 255
 * Find block that is parent of current list.
256
 * That means go to lower addresses, until such block is found
367 jermar 257
 *
3972 decky 258
 * @param order Order of parent must be different then this
259
 *              parameter!!
260
 *
367 jermar 261
 */
3972 decky 262
static link_t *zone_buddy_find_block(buddy_system_t *buddy, link_t *child,
2133 jermar 263
    uint8_t order)
367 jermar 264
{
3972 decky 265
    frame_t *frame = list_get_instance(child, frame_t, buddy_link);
266
    zone_t *zone = (zone_t *) buddy->data;
367 jermar 267
 
3972 decky 268
    index_t index = frame_index(zone, frame);
814 palkovsky 269
    do {
3972 decky 270
        if (zone->frames[index].buddy_order != order)
814 palkovsky 271
            return &zone->frames[index].buddy_link;
3972 decky 272
    } while (index-- > 0);
273
 
814 palkovsky 274
    return NULL;
367 jermar 275
}
479 bondari 276
 
3185 jermar 277
/** Buddy system find_buddy implementation.
489 jermar 278
 *
3972 decky 279
 * @param buddy Buddy system.
280
 * @param block Block for which buddy should be found.
479 bondari 281
 *
3972 decky 282
 * @return Buddy for given block if found.
283
 *
479 bondari 284
 */
3972 decky 285
static link_t *zone_buddy_find_buddy(buddy_system_t *buddy, link_t *block)
814 palkovsky 286
{
3972 decky 287
    frame_t *frame = list_get_instance(block, frame_t, buddy_link);
288
    zone_t *zone = (zone_t *) buddy->data;
2133 jermar 289
    ASSERT(IS_BUDDY_ORDER_OK(frame_index_abs(zone, frame),
290
        frame->buddy_order));
480 bondari 291
 
3972 decky 292
    bool is_left = IS_BUDDY_LEFT_BLOCK_ABS(zone, frame);
293
 
294
    index_t index;
533 bondari 295
    if (is_left) {
3185 jermar 296
        index = (frame_index(zone, frame)) +
297
            (1 << frame->buddy_order);
4048 jermar 298
    } else {    /* is_right */
3185 jermar 299
        index = (frame_index(zone, frame)) -
300
            (1 << frame->buddy_order);
533 bondari 301
    }
302
 
814 palkovsky 303
    if (frame_index_valid(zone, index)) {
3972 decky 304
        if ((zone->frames[index].buddy_order == frame->buddy_order) &&
305
            (zone->frames[index].refcount == 0)) {
533 bondari 306
            return &zone->frames[index].buddy_link;
480 bondari 307
        }
308
    }
3972 decky 309
 
310
    return NULL;
479 bondari 311
}
312
 
3185 jermar 313
/** Buddy system bisect implementation.
479 bondari 314
 *
3972 decky 315
 * @param buddy Buddy system.
316
 * @param block Block to bisect.
480 bondari 317
 *
3972 decky 318
 * @return Right block.
319
 *
479 bondari 320
 */
3972 decky 321
static link_t *zone_buddy_bisect(buddy_system_t *buddy, link_t *block)
3185 jermar 322
{
3972 decky 323
    frame_t *frame_l = list_get_instance(block, frame_t, buddy_link);
324
    frame_t *frame_r = (frame_l + (1 << (frame_l->buddy_order - 1)));
564 jermar 325
 
480 bondari 326
    return &frame_r->buddy_link;
479 bondari 327
}
328
 
3185 jermar 329
/** Buddy system coalesce implementation.
479 bondari 330
 *
3972 decky 331
 * @param buddy   Buddy system.
332
 * @param block_1 First block.
333
 * @param block_2 First block's buddy.
480 bondari 334
 *
3972 decky 335
 * @return Coalesced block (actually block that represents lower
336
 *         address).
337
 *
479 bondari 338
 */
3972 decky 339
static link_t *zone_buddy_coalesce(buddy_system_t *buddy, link_t *block_1,
340
    link_t *block_2)
822 palkovsky 341
{
3972 decky 342
    frame_t *frame1 = list_get_instance(block_1, frame_t, buddy_link);
343
    frame_t *frame2 = list_get_instance(block_2, frame_t, buddy_link);
564 jermar 344
 
3972 decky 345
    return ((frame1 < frame2) ? block_1 : block_2);
479 bondari 346
}
347
 
3185 jermar 348
/** Buddy system set_order implementation.
489 jermar 349
 *
3972 decky 350
 * @param buddy Buddy system.
351
 * @param block Buddy system block.
352
 * @param order Order to set.
353
 *
479 bondari 354
 */
3972 decky 355
static void zone_buddy_set_order(buddy_system_t *buddy, link_t *block,
3185 jermar 356
    uint8_t order)
357
{
3972 decky 358
    list_get_instance(block, frame_t, buddy_link)->buddy_order = order;
479 bondari 359
}
360
 
3185 jermar 361
/** Buddy system get_order implementation.
489 jermar 362
 *
3972 decky 363
 * @param buddy Buddy system.
364
 * @param block Buddy system block.
479 bondari 365
 *
3972 decky 366
 * @return Order of block.
367
 *
479 bondari 368
 */
3972 decky 369
static uint8_t zone_buddy_get_order(buddy_system_t *buddy, link_t *block)
3185 jermar 370
{
3972 decky 371
    return list_get_instance(block, frame_t, buddy_link)->buddy_order;
479 bondari 372
}
533 bondari 373
 
3185 jermar 374
/** Buddy system mark_busy implementation.
533 bondari 375
 *
3972 decky 376
 * @param buddy Buddy system.
377
 * @param block Buddy system block.
378
 *
533 bondari 379
 */
3972 decky 380
static void zone_buddy_mark_busy(buddy_system_t *buddy, link_t * block)
3185 jermar 381
{
3972 decky 382
    list_get_instance(block, frame_t, buddy_link)->refcount = 1;
533 bondari 383
}
676 bondari 384
 
3185 jermar 385
/** Buddy system mark_available implementation.
814 palkovsky 386
 *
3972 decky 387
 * @param buddy Buddy system.
388
 * @param block Buddy system block.
814 palkovsky 389
 */
3972 decky 390
static void zone_buddy_mark_available(buddy_system_t *buddy, link_t *block)
3185 jermar 391
{
3972 decky 392
    list_get_instance(block, frame_t, buddy_link)->refcount = 0;
814 palkovsky 393
}
394
 
2083 decky 395
static buddy_system_operations_t zone_buddy_system_operations = {
814 palkovsky 396
    .find_buddy = zone_buddy_find_buddy,
397
    .bisect = zone_buddy_bisect,
398
    .coalesce = zone_buddy_coalesce,
399
    .set_order = zone_buddy_set_order,
400
    .get_order = zone_buddy_get_order,
401
    .mark_busy = zone_buddy_mark_busy,
402
    .mark_available = zone_buddy_mark_available,
3206 jermar 403
    .find_block = zone_buddy_find_block
814 palkovsky 404
};
405
 
2059 jermar 406
/******************/
814 palkovsky 407
/* Zone functions */
2059 jermar 408
/******************/
814 palkovsky 409
 
3185 jermar 410
/** Allocate frame in particular zone.
814 palkovsky 411
 *
3972 decky 412
 * Assume zone is locked and is available for allocation.
1269 decky 413
 * Panics if allocation is impossible.
814 palkovsky 414
 *
3972 decky 415
 * @param zone  Zone to allocate from.
416
 * @param order Allocate exactly 2^order frames.
1269 decky 417
 *
3972 decky 418
 * @return Frame index in zone.
1269 decky 419
 *
814 palkovsky 420
 */
1780 jermar 421
static pfn_t zone_frame_alloc(zone_t *zone, uint8_t order)
814 palkovsky 422
{
3972 decky 423
    ASSERT(zone_flags_available(zone->flags));
424
 
814 palkovsky 425
    /* Allocate frames from zone buddy system */
3972 decky 426
    link_t *link = buddy_system_alloc(zone->buddy_system, order);
814 palkovsky 427
 
3972 decky 428
    ASSERT(link);
814 palkovsky 429
 
430
    /* Update zone information. */
431
    zone->free_count -= (1 << order);
432
    zone->busy_count += (1 << order);
3972 decky 433
 
814 palkovsky 434
    /* Frame will be actually a first frame of the block. */
3972 decky 435
    frame_t *frame = list_get_instance(link, frame_t, buddy_link);
814 palkovsky 436
 
3972 decky 437
    /* Get frame address */
438
    return make_frame_index(zone, frame);
814 palkovsky 439
}
440
 
3185 jermar 441
/** Free frame from zone.
814 palkovsky 442
 *
3972 decky 443
 * Assume zone is locked and is available for deallocation.
1568 palkovsky 444
 *
3972 decky 445
 * @param zone      Pointer to zone from which the frame is to be freed.
446
 * @param frame_idx Frame index relative to zone.
447
 *
814 palkovsky 448
 */
820 jermar 449
static void zone_frame_free(zone_t *zone, index_t frame_idx)
814 palkovsky 450
{
3972 decky 451
    ASSERT(zone_flags_available(zone->flags));
814 palkovsky 452
 
3972 decky 453
    frame_t *frame = &zone->frames[frame_idx];
454
 
455
    /* Remember frame order */
456
    uint8_t order = frame->buddy_order;
457
 
814 palkovsky 458
    ASSERT(frame->refcount);
3972 decky 459
 
814 palkovsky 460
    if (!--frame->refcount) {
461
        buddy_system_free(zone->buddy_system, &frame->buddy_link);
3972 decky 462
 
946 jermar 463
        /* Update zone information. */
464
        zone->free_count += (1 << order);
465
        zone->busy_count -= (1 << order);
814 palkovsky 466
    }
467
}
468
 
3185 jermar 469
/** Return frame from zone. */
3188 jermar 470
static frame_t *zone_get_frame(zone_t *zone, index_t frame_idx)
814 palkovsky 471
{
472
    ASSERT(frame_idx < zone->count);
473
    return &zone->frames[frame_idx];
474
}
475
 
3185 jermar 476
/** Mark frame in zone unavailable to allocation. */
820 jermar 477
static void zone_mark_unavailable(zone_t *zone, index_t frame_idx)
814 palkovsky 478
{
3972 decky 479
    ASSERT(zone_flags_available(zone->flags));
480
 
481
    frame_t *frame = zone_get_frame(zone, frame_idx);
822 palkovsky 482
    if (frame->refcount)
483
        return;
3972 decky 484
 
4048 jermar 485
    link_t *link __attribute__ ((unused));
486
 
487
    link = buddy_system_alloc_block(zone->buddy_system,
2133 jermar 488
        &frame->buddy_link);
3972 decky 489
 
814 palkovsky 490
    ASSERT(link);
491
    zone->free_count--;
492
}
493
 
3972 decky 494
/** Merge two zones.
822 palkovsky 495
 *
3972 decky 496
 * Expect buddy to point to space at least zone_conf_size large.
497
 * Assume z1 & z2 are locked and compatible and zones lock is
498
 * locked.
822 palkovsky 499
 *
3972 decky 500
 * @param z1     First zone to merge.
501
 * @param z2     Second zone to merge.
502
 * @param old_z1 Original date of the first zone.
503
 * @param buddy  Merged zone buddy.
1568 palkovsky 504
 *
822 palkovsky 505
 */
3972 decky 506
static void zone_merge_internal(count_t z1, count_t z2, zone_t *old_z1, buddy_system_t *buddy)
822 palkovsky 507
{
3972 decky 508
    ASSERT(zone_flags_available(zones.info[z1].flags));
509
    ASSERT(zone_flags_available(zones.info[z2].flags));
510
    ASSERT(zones.info[z1].flags == zones.info[z2].flags);
511
    ASSERT(zones.info[z1].base < zones.info[z2].base);
512
    ASSERT(!overlaps(zones.info[z1].base, zones.info[z1].count,
513
        zones.info[z2].base, zones.info[z2].count));
822 palkovsky 514
 
3972 decky 515
    /* Difference between zone bases */
516
    pfn_t base_diff = zones.info[z2].base - zones.info[z1].base;
517
 
518
    zones.info[z1].count = base_diff + zones.info[z2].count;
519
    zones.info[z1].free_count += zones.info[z2].free_count;
520
    zones.info[z1].busy_count += zones.info[z2].busy_count;
521
    zones.info[z1].buddy_system = buddy;
522
 
523
    uint8_t order = fnzb(zones.info[z1].count);
524
    buddy_system_create(zones.info[z1].buddy_system, order,
525
        &zone_buddy_system_operations, (void *) &zones.info[z1]);
526
 
527
    zones.info[z1].frames =
528
        (frame_t *) ((uint8_t *) zones.info[z1].buddy_system
529
        + buddy_conf_size(order));
530
 
531
    /* This marks all frames busy */
532
    count_t i;
533
    for (i = 0; i < zones.info[z1].count; i++)
534
        frame_initialize(&zones.info[z1].frames[i]);
535
 
822 palkovsky 536
    /* Copy frames from both zones to preserve full frame orders,
3972 decky 537
     * parents etc. Set all free frames with refcount = 0 to 1, because
538
     * we add all free frames to buddy allocator later again, clearing
539
     * order to 0. Don't set busy frames with refcount = 0, as they
1093 palkovsky 540
     * will not be reallocated during merge and it would make later
541
     * problems with allocation/free.
822 palkovsky 542
     */
3972 decky 543
    for (i = 0; i < old_z1->count; i++)
544
        zones.info[z1].frames[i] = old_z1->frames[i];
545
 
546
    for (i = 0; i < zones.info[z2].count; i++)
547
        zones.info[z1].frames[base_diff + i]
548
            = zones.info[z2].frames[i];
549
 
1093 palkovsky 550
    i = 0;
3972 decky 551
    while (i < zones.info[z1].count) {
552
        if (zones.info[z1].frames[i].refcount) {
553
            /* Skip busy frames */
554
            i += 1 << zones.info[z1].frames[i].buddy_order;
555
        } else {
556
            /* Free frames, set refcount = 1
557
             * (all free frames have refcount == 0, we need not
558
             * to check the order)
559
             */
560
            zones.info[z1].frames[i].refcount = 1;
561
            zones.info[z1].frames[i].buddy_order = 0;
1093 palkovsky 562
            i++;
822 palkovsky 563
        }
564
    }
3972 decky 565
 
566
    /* Add free blocks from the original zone z1 */
567
    while (zone_can_alloc(old_z1, 0)) {
568
        /* Allocate from the original zone */
569
        pfn_t frame_idx = zone_frame_alloc(old_z1, 0);
570
 
571
        /* Free the frame from the merged zone */
572
        frame_t *frame = &zones.info[z1].frames[frame_idx];
822 palkovsky 573
        frame->refcount = 0;
3972 decky 574
        buddy_system_free(zones.info[z1].buddy_system, &frame->buddy_link);
822 palkovsky 575
    }
3972 decky 576
 
577
    /* Add free blocks from the original zone z2 */
578
    while (zone_can_alloc(&zones.info[z2], 0)) {
579
        /* Allocate from the original zone */
580
        pfn_t frame_idx = zone_frame_alloc(&zones.info[z2], 0);
581
 
582
        /* Free the frame from the merged zone */
583
        frame_t *frame = &zones.info[z1].frames[base_diff + frame_idx];
822 palkovsky 584
        frame->refcount = 0;
3972 decky 585
        buddy_system_free(zones.info[z1].buddy_system, &frame->buddy_link);
822 palkovsky 586
    }
587
}
588
 
3185 jermar 589
/** Return old configuration frames into the zone.
822 palkovsky 590
 *
3972 decky 591
 * We have two cases:
592
 * - The configuration data is outside the zone
593
 *   -> do nothing (perhaps call frame_free?)
594
 * - The configuration data was created by zone_create
595
 *   or updated by reduce_region -> free every frame
824 palkovsky 596
 *
3972 decky 597
 * @param znum  The actual zone where freeing should occur.
598
 * @param pfn   Old zone configuration frame.
599
 * @param count Old zone frame count.
600
 *
822 palkovsky 601
 */
3972 decky 602
static void return_config_frames(count_t znum, pfn_t pfn, count_t count)
822 palkovsky 603
{
3972 decky 604
    ASSERT(zone_flags_available(zones.info[znum].flags));
822 palkovsky 605
 
3972 decky 606
    count_t cframes = SIZE2FRAMES(zone_conf_size(count));
607
 
608
    if ((pfn < zones.info[znum].base)
609
        || (pfn >= zones.info[znum].base + zones.info[znum].count))
822 palkovsky 610
        return;
3972 decky 611
 
4048 jermar 612
    frame_t *frame __attribute__ ((unused));
613
 
614
    frame = &zones.info[znum].frames[pfn - zones.info[znum].base];
824 palkovsky 615
    ASSERT(!frame->buddy_order);
3972 decky 616
 
617
    count_t i;
2122 decky 618
    for (i = 0; i < cframes; i++) {
3972 decky 619
        zones.info[znum].busy_count++;
620
        zone_frame_free(&zones.info[znum],
621
            pfn - zones.info[znum].base + i);
822 palkovsky 622
    }
623
}
624
 
3185 jermar 625
/** Reduce allocated block to count of order 0 frames.
824 palkovsky 626
 *
3972 decky 627
 * The allocated block needs 2^order frames. Reduce all frames
628
 * in the block to order 0 and free the unneeded frames. This means that
629
 * when freeing the previously allocated block starting with frame_idx,
1568 palkovsky 630
 * you have to free every frame.
824 palkovsky 631
 *
3972 decky 632
 * @param znum      Zone.
633
 * @param frame_idx Index the first frame of the block.
634
 * @param count     Allocated frames in block.
635
 *
824 palkovsky 636
 */
3972 decky 637
static void zone_reduce_region(count_t znum, pfn_t frame_idx, count_t count)
824 palkovsky 638
{
3972 decky 639
    ASSERT(zone_flags_available(zones.info[znum].flags));
640
    ASSERT(frame_idx + count < zones.info[znum].count);
824 palkovsky 641
 
3972 decky 642
    uint8_t order = zones.info[znum].frames[frame_idx].buddy_order;
2122 decky 643
    ASSERT((count_t) (1 << order) >= count);
3972 decky 644
 
824 palkovsky 645
    /* Reduce all blocks to order 0 */
3972 decky 646
    count_t i;
2122 decky 647
    for (i = 0; i < (count_t) (1 << order); i++) {
3972 decky 648
        frame_t *frame = &zones.info[znum].frames[i + frame_idx];
824 palkovsky 649
        frame->buddy_order = 0;
2133 jermar 650
        if (!frame->refcount)
824 palkovsky 651
            frame->refcount = 1;
652
        ASSERT(frame->refcount == 1);
653
    }
3972 decky 654
 
824 palkovsky 655
    /* Free unneeded frames */
3972 decky 656
    for (i = count; i < (count_t) (1 << order); i++)
657
        zone_frame_free(&zones.info[znum], i + frame_idx);
824 palkovsky 658
}
659
 
3185 jermar 660
/** Merge zones z1 and z2.
822 palkovsky 661
 *
3972 decky 662
 * The merged zones must be 2 zones with no zone existing in between
663
 * (which means that z2 = z1 + 1). Both zones must be available zones
664
 * with the same flags.
822 palkovsky 665
 *
3972 decky 666
 * When you create a new zone, the frame allocator configuration does
667
 * not to be 2^order size. Once the allocator is running it is no longer
668
 * possible, merged configuration data occupies more space :-/
669
 *
670
 * The function uses
671
 *
822 palkovsky 672
 */
3972 decky 673
bool zone_merge(count_t z1, count_t z2)
822 palkovsky 674
{
3972 decky 675
    ipl_t ipl = interrupts_disable();
822 palkovsky 676
    spinlock_lock(&zones.lock);
3972 decky 677
 
678
    bool ret = true;
679
 
680
    /* We can join only 2 zones with none existing inbetween,
681
     * the zones have to be available and with the same
682
     * set of flags
683
     */
684
    if ((z1 >= zones.count) || (z2 >= zones.count)
685
        || (z2 - z1 != 1)
686
        || (!zone_flags_available(zones.info[z1].flags))
687
        || (!zone_flags_available(zones.info[z2].flags))
688
        || (zones.info[z1].flags != zones.info[z2].flags)) {
689
        ret = false;
822 palkovsky 690
        goto errout;
3972 decky 691
    }
692
 
693
    pfn_t cframes = SIZE2FRAMES(zone_conf_size(
694
        zones.info[z2].base - zones.info[z1].base
695
        + zones.info[z2].count));
696
 
697
    uint8_t order;
1700 palkovsky 698
    if (cframes == 1)
699
        order = 0;
3972 decky 700
    else
1700 palkovsky 701
        order = fnzb(cframes - 1) + 1;
3972 decky 702
 
703
    /* Allocate merged zone data inside one of the zones */
704
    pfn_t pfn;
705
    if (zone_can_alloc(&zones.info[z1], order)) {
706
        pfn = zones.info[z1].base + zone_frame_alloc(&zones.info[z1], order);
707
    } else if (zone_can_alloc(&zones.info[z2], order)) {
708
        pfn = zones.info[z2].base + zone_frame_alloc(&zones.info[z2], order);
709
    } else {
710
        ret = false;
711
        goto errout;
712
    }
713
 
714
    /* Preserve original data from z1 */
715
    zone_t old_z1 = zones.info[z1];
716
    old_z1.buddy_system->data = (void *) &old_z1;
717
 
718
    /* Do zone merging */
719
    buddy_system_t *buddy = (buddy_system_t *) PA2KA(PFN2ADDR(pfn));
720
    zone_merge_internal(z1, z2, &old_z1, buddy);
721
 
824 palkovsky 722
    /* Free unneeded config frames */
3972 decky 723
    zone_reduce_region(z1, pfn - zones.info[z1].base, cframes);
724
 
822 palkovsky 725
    /* Subtract zone information from busy frames */
3972 decky 726
    zones.info[z1].busy_count -= cframes;
727
 
728
    /* Free old zone information */
729
    return_config_frames(z1,
730
        ADDR2PFN(KA2PA((uintptr_t) old_z1.frames)), old_z1.count);
731
    return_config_frames(z1,
732
        ADDR2PFN(KA2PA((uintptr_t) zones.info[z2].frames)),
733
        zones.info[z2].count);
734
 
3973 decky 735
    /* Move zones down */
3972 decky 736
    count_t i;
3973 decky 737
    for (i = z2 + 1; i < zones.count; i++) {
1037 decky 738
        zones.info[i - 1] = zones.info[i];
3973 decky 739
        zones.info[i - 1].buddy_system->data =
740
            (void *) &zones.info[i - 1];
741
    }
742
 
822 palkovsky 743
    zones.count--;
3972 decky 744
 
822 palkovsky 745
errout:
746
    spinlock_unlock(&zones.lock);
747
    interrupts_restore(ipl);
3972 decky 748
 
749
    return ret;
822 palkovsky 750
}
751
 
3972 decky 752
/** Merge all mergeable zones into one big zone.
822 palkovsky 753
 *
3972 decky 754
 * It is reasonable to do this on systems where
755
 * BIOS reports parts in chunks, so that we could
756
 * have 1 zone (it's faster).
757
 *
822 palkovsky 758
 */
759
void zone_merge_all(void)
760
{
3972 decky 761
    count_t i = 0;
762
    while (i < zones.count) {
763
        if (!zone_merge(i, i + 1))
764
            i++;
822 palkovsky 765
    }
766
}
767
 
3185 jermar 768
/** Create new frame zone.
814 palkovsky 769
 *
3972 decky 770
 * @param zone  Zone to construct.
771
 * @param buddy Address of buddy system configuration information.
772
 * @param start Physical address of the first frame within the zone.
773
 * @param count Count of frames in zone.
774
 * @param flags Zone flags.
814 palkovsky 775
 *
3972 decky 776
 * @return Initialized zone.
777
 *
814 palkovsky 778
 */
3972 decky 779
static void zone_construct(zone_t *zone, buddy_system_t *buddy, pfn_t start, count_t count, zone_flags_t flags)
814 palkovsky 780
{
3972 decky 781
    zone->base = start;
782
    zone->count = count;
783
    zone->flags = flags;
784
    zone->free_count = count;
785
    zone->busy_count = 0;
786
    zone->buddy_system = buddy;
814 palkovsky 787
 
3972 decky 788
    if (zone_flags_available(flags)) {
789
        /*
790
         * Compute order for buddy system and initialize
791
         */
792
        uint8_t order = fnzb(count);
793
        buddy_system_create(zone->buddy_system, order,
794
            &zone_buddy_system_operations, (void *) zone);
795
 
796
        /* Allocate frames _after_ the confframe */
797
 
798
        /* Check sizes */
799
        zone->frames = (frame_t *) ((uint8_t *) zone->buddy_system +
800
            buddy_conf_size(order));
801
 
802
        count_t i;
803
        for (i = 0; i < count; i++)
804
            frame_initialize(&zone->frames[i]);
805
 
806
        /* Stuffing frames */
807
        for (i = 0; i < count; i++) {
808
            zone->frames[i].refcount = 0;
809
            buddy_system_free(zone->buddy_system, &zone->frames[i].buddy_link);
810
        }
811
    } else
812
        zone->frames = NULL;
814 palkovsky 813
}
814
 
3185 jermar 815
/** Compute configuration data size for zone.
1568 palkovsky 816
 *
3972 decky 817
 * @param count Size of zone in frames.
818
 *
819
 * @return Size of zone configuration info (in bytes).
820
 *
1568 palkovsky 821
 */
1780 jermar 822
uintptr_t zone_conf_size(count_t count)
814 palkovsky 823
{
3972 decky 824
    return (count * sizeof(frame_t) + buddy_conf_size(fnzb(count)));
814 palkovsky 825
}
826
 
3185 jermar 827
/** Create and add zone to system.
814 palkovsky 828
 *
3972 decky 829
 * @param start     First frame number (absolute).
830
 * @param count     Size of zone in frames.
831
 * @param confframe Where configuration frames are supposed to be.
832
 *                  Automatically checks, that we will not disturb the
833
 *                  kernel and possibly init. If confframe is given
834
 *                  _outside_ this zone, it is expected, that the area is
835
 *                  already marked BUSY and big enough to contain
836
 *                  zone_conf_size() amount of data. If the confframe is
837
 *                  inside the area, the zone free frame information is
838
 *                  modified not to include it.
822 palkovsky 839
 *
3972 decky 840
 * @return Zone number or -1 on error.
841
 *
814 palkovsky 842
 */
3972 decky 843
count_t zone_create(pfn_t start, count_t count, pfn_t confframe, zone_flags_t flags)
814 palkovsky 844
{
3972 decky 845
    ipl_t ipl = interrupts_disable();
846
    spinlock_lock(&zones.lock);
847
 
848
    if (zone_flags_available(flags)) {  /* Create available zone */
849
        /* Theoretically we could have NULL here, practically make sure
850
         * nobody tries to do that. If some platform requires, remove
851
         * the assert
852
         */
853
        ASSERT(confframe != NULL);
854
 
855
        /* If confframe is supposed to be inside our zone, then make sure
856
         * it does not span kernel & init
857
         */
858
        count_t confcount = SIZE2FRAMES(zone_conf_size(count));
859
        if ((confframe >= start) && (confframe < start + count)) {
860
            for (; confframe < start + count; confframe++) {
861
                uintptr_t addr = PFN2ADDR(confframe);
2133 jermar 862
                if (overlaps(addr, PFN2ADDR(confcount),
3972 decky 863
                    KA2PA(config.base), config.kernel_size))
864
                    continue;
865
 
866
                if (overlaps(addr, PFN2ADDR(confcount),
867
                    KA2PA(config.stack_base), config.stack_size))
868
                    continue;
869
 
870
                bool overlap = false;
871
                count_t i;
872
                for (i = 0; i < init.cnt; i++)
873
                    if (overlaps(addr, PFN2ADDR(confcount),
874
                        KA2PA(init.tasks[i].addr),
875
                        init.tasks[i].size)) {
876
                        overlap = true;
877
                        break;
878
                    }
879
                if (overlap)
880
                    continue;
881
 
882
                break;
883
            }
1037 decky 884
 
3972 decky 885
            if (confframe >= start + count)
886
                panic("Cannot find configuration data for zone.");
814 palkovsky 887
        }
3972 decky 888
 
889
        count_t znum = zones_insert_zone(start, count);
890
        if (znum == (count_t) -1) {
891
            spinlock_unlock(&zones.lock);
892
            interrupts_restore(ipl);
893
            return (count_t) -1;
894
        }
895
 
896
        buddy_system_t *buddy = (buddy_system_t *) PA2KA(PFN2ADDR(confframe));
897
        zone_construct(&zones.info[znum], buddy, start, count, flags);
898
 
899
        /* If confdata in zone, mark as unavailable */
900
        if ((confframe >= start) && (confframe < start + count)) {
901
            count_t i;
902
            for (i = confframe; i < confframe + confcount; i++)
903
                zone_mark_unavailable(&zones.info[znum],
904
                    i - zones.info[znum].base);
905
        }
906
 
907
        spinlock_unlock(&zones.lock);
908
        interrupts_restore(ipl);
909
 
910
        return znum;
814 palkovsky 911
    }
3188 jermar 912
 
3972 decky 913
    /* Non-available zone */
914
    count_t znum = zones_insert_zone(start, count);
915
    if (znum == (count_t) -1) {
916
        spinlock_unlock(&zones.lock);
917
        interrupts_restore(ipl);
918
        return (count_t) -1;
919
    }
920
    zone_construct(&zones.info[znum], NULL, start, count, flags);
921
 
922
    spinlock_unlock(&zones.lock);
923
    interrupts_restore(ipl);
924
 
822 palkovsky 925
    return znum;
814 palkovsky 926
}
927
 
3972 decky 928
/*******************/
814 palkovsky 929
/* Frame functions */
3972 decky 930
/*******************/
814 palkovsky 931
 
3185 jermar 932
/** Set parent of frame. */
3972 decky 933
void frame_set_parent(pfn_t pfn, void *data, count_t hint)
814 palkovsky 934
{
3972 decky 935
    ipl_t ipl = interrupts_disable();
936
    spinlock_lock(&zones.lock);
937
 
3973 decky 938
    count_t znum = find_zone(pfn, 1, hint);
3972 decky 939
 
940
    ASSERT(znum != (count_t) -1);
941
 
942
    zone_get_frame(&zones.info[znum],
943
        pfn - zones.info[znum].base)->parent = data;
944
 
945
    spinlock_unlock(&zones.lock);
946
    interrupts_restore(ipl);
814 palkovsky 947
}
948
 
3972 decky 949
void *frame_get_parent(pfn_t pfn, count_t hint)
814 palkovsky 950
{
3972 decky 951
    ipl_t ipl = interrupts_disable();
952
    spinlock_lock(&zones.lock);
814 palkovsky 953
 
3973 decky 954
    count_t znum = find_zone(pfn, 1, hint);
3972 decky 955
 
956
    ASSERT(znum != (count_t) -1);
957
 
958
    void *res = zone_get_frame(&zones.info[znum],
959
        pfn - zones.info[znum].base)->parent;
960
 
961
    spinlock_unlock(&zones.lock);
962
    interrupts_restore(ipl);
963
 
814 palkovsky 964
    return res;
965
}
966
 
967
/** Allocate power-of-two frames of physical memory.
968
 *
3972 decky 969
 * @param order Allocate exactly 2^order frames.
970
 * @param flags Flags for host zone selection and address processing.
971
 * @param pzone Preferred zone.
814 palkovsky 972
 *
3972 decky 973
 * @return Physical address of the allocated frame.
1269 decky 974
 *
814 palkovsky 975
 */
3972 decky 976
void *frame_alloc_generic(uint8_t order, frame_flags_t flags, count_t *pzone)
814 palkovsky 977
{
3972 decky 978
    count_t size = ((count_t) 1) << order;
814 palkovsky 979
    ipl_t ipl;
3972 decky 980
    count_t hint = pzone ? (*pzone) : 0;
814 palkovsky 981
 
982
loop:
983
    ipl = interrupts_disable();
3972 decky 984
    spinlock_lock(&zones.lock);
1269 decky 985
 
814 palkovsky 986
    /*
987
     * First, find suitable frame zone.
988
     */
3972 decky 989
    count_t znum = find_free_zone(order,
990
        FRAME_TO_ZONE_FLAGS(flags), hint);
1269 decky 991
 
814 palkovsky 992
    /* If no memory, reclaim some slab memory,
993
       if it does not help, reclaim all */
3972 decky 994
    if ((znum == (count_t) -1) && (!(flags & FRAME_NO_RECLAIM))) {
995
        count_t freed = slab_reclaim(0);
996
 
997
        if (freed > 0)
998
            znum = find_free_zone(order,
999
                FRAME_TO_ZONE_FLAGS(flags), hint);
1000
 
1001
        if (znum == (count_t) -1) {
814 palkovsky 1002
            freed = slab_reclaim(SLAB_RECLAIM_ALL);
3972 decky 1003
            if (freed > 0)
1004
                znum = find_free_zone(order,
1005
                    FRAME_TO_ZONE_FLAGS(flags), hint);
814 palkovsky 1006
        }
1007
    }
3972 decky 1008
 
1009
    if (znum == (count_t) -1) {
3187 jermar 1010
        if (flags & FRAME_ATOMIC) {
3972 decky 1011
            spinlock_unlock(&zones.lock);
3187 jermar 1012
            interrupts_restore(ipl);
3972 decky 1013
            return NULL;
3187 jermar 1014
        }
814 palkovsky 1015
 
3187 jermar 1016
#ifdef CONFIG_DEBUG
3972 decky 1017
        count_t avail = total_frames_free();
3187 jermar 1018
#endif
3972 decky 1019
 
1020
        spinlock_unlock(&zones.lock);
1021
        interrupts_restore(ipl);
1022
 
1023
        /*
1024
         * Sleep until some frames are available again.
1025
         */
1026
 
1027
#ifdef CONFIG_DEBUG
1028
        printf("Thread %" PRIu64 " waiting for %" PRIc " frames, "
1029
            "%" PRIc " available.\n", THREAD->tid, size, avail);
1030
#endif
1031
 
3188 jermar 1032
        mutex_lock(&mem_avail_mtx);
3972 decky 1033
 
1034
        if (mem_avail_req > 0)
1035
            mem_avail_req = min(mem_avail_req, size);
1036
        else
1037
            mem_avail_req = size;
1038
        count_t gen = mem_avail_gen;
1039
 
1040
        while (gen == mem_avail_gen)
3188 jermar 1041
            condvar_wait(&mem_avail_cv, &mem_avail_mtx);
3972 decky 1042
 
3188 jermar 1043
        mutex_unlock(&mem_avail_mtx);
3972 decky 1044
 
3187 jermar 1045
#ifdef CONFIG_DEBUG
3972 decky 1046
        printf("Thread %" PRIu64 " woken up.\n", THREAD->tid);
3187 jermar 1047
#endif
3972 decky 1048
 
814 palkovsky 1049
        goto loop;
1050
    }
1269 decky 1051
 
3972 decky 1052
    pfn_t pfn = zone_frame_alloc(&zones.info[znum], order)
1053
        + zones.info[znum].base;
3188 jermar 1054
 
3972 decky 1055
    spinlock_unlock(&zones.lock);
814 palkovsky 1056
    interrupts_restore(ipl);
3972 decky 1057
 
1058
    if (pzone)
1059
        *pzone = znum;
1060
 
1760 palkovsky 1061
    if (flags & FRAME_KA)
3972 decky 1062
        return (void *) PA2KA(PFN2ADDR(pfn));
1063
 
1064
    return (void *) PFN2ADDR(pfn);
814 palkovsky 1065
}
1066
 
1067
/** Free a frame.
1068
 *
1854 jermar 1069
 * Find respective frame structure for supplied physical frame address.
3972 decky 1070
 * Decrement frame reference count. If it drops to zero, move the frame
1071
 * structure to free list.
814 palkovsky 1072
 *
3972 decky 1073
 * @param frame Physical Address of of the frame to be freed.
1074
 *
814 palkovsky 1075
 */
1780 jermar 1076
void frame_free(uintptr_t frame)
814 palkovsky 1077
{
3972 decky 1078
    ipl_t ipl = interrupts_disable();
1079
    spinlock_lock(&zones.lock);
1080
 
814 palkovsky 1081
    /*
1082
     * First, find host frame zone for addr.
1083
     */
3972 decky 1084
    pfn_t pfn = ADDR2PFN(frame);
3973 decky 1085
    count_t znum = find_zone(pfn, 1, NULL);
814 palkovsky 1086
 
3972 decky 1087
    ASSERT(znum != (count_t) -1);
814 palkovsky 1088
 
3972 decky 1089
    zone_frame_free(&zones.info[znum], pfn - zones.info[znum].base);
3187 jermar 1090
 
3972 decky 1091
    spinlock_unlock(&zones.lock);
1092
    interrupts_restore(ipl);
1093
 
3187 jermar 1094
    /*
1095
     * Signal that some memory has been freed.
1096
     */
3188 jermar 1097
    mutex_lock(&mem_avail_mtx);
3972 decky 1098
    if (mem_avail_req > 0)
1099
        mem_avail_req--;
1100
 
1101
    if (mem_avail_req == 0) {
1102
        mem_avail_gen++;
1103
        condvar_broadcast(&mem_avail_cv);
1104
    }
3188 jermar 1105
    mutex_unlock(&mem_avail_mtx);
814 palkovsky 1106
}
1107
 
1236 jermar 1108
/** Add reference to frame.
1109
 *
1110
 * Find respective frame structure for supplied PFN and
1111
 * increment frame reference count.
1112
 *
3972 decky 1113
 * @param pfn Frame number of the frame to be freed.
1114
 *
1236 jermar 1115
 */
1116
void frame_reference_add(pfn_t pfn)
1117
{
3972 decky 1118
    ipl_t ipl = interrupts_disable();
1119
    spinlock_lock(&zones.lock);
1236 jermar 1120
 
1121
    /*
1122
     * First, find host frame zone for addr.
1123
     */
3973 decky 1124
    count_t znum = find_zone(pfn, 1, NULL);
1236 jermar 1125
 
3972 decky 1126
    ASSERT(znum != (count_t) -1);
1236 jermar 1127
 
3972 decky 1128
    zones.info[znum].frames[pfn - zones.info[znum].base].refcount++;
1129
 
1130
    spinlock_unlock(&zones.lock);
1236 jermar 1131
    interrupts_restore(ipl);
1132
}
814 palkovsky 1133
 
3185 jermar 1134
/** Mark given range unavailable in frame zones. */
820 jermar 1135
void frame_mark_unavailable(pfn_t start, count_t count)
814 palkovsky 1136
{
3972 decky 1137
    ipl_t ipl = interrupts_disable();
1138
    spinlock_lock(&zones.lock);
852 palkovsky 1139
 
3972 decky 1140
    count_t i;
2122 decky 1141
    for (i = 0; i < count; i++) {
3973 decky 1142
        count_t znum = find_zone(start + i, 1, 0);
3972 decky 1143
        if (znum == (count_t) -1)  /* PFN not found */
814 palkovsky 1144
            continue;
3972 decky 1145
 
1146
        zone_mark_unavailable(&zones.info[znum],
1147
            start + i - zones.info[znum].base);
814 palkovsky 1148
    }
3972 decky 1149
 
1150
    spinlock_unlock(&zones.lock);
1151
    interrupts_restore(ipl);
814 palkovsky 1152
}
1153
 
3185 jermar 1154
/** Initialize physical memory management. */
814 palkovsky 1155
void frame_init(void)
1156
{
1157
    if (config.cpu_active == 1) {
1158
        zones.count = 0;
1981 jermar 1159
        spinlock_initialize(&zones.lock, "zones.lock");
3188 jermar 1160
        mutex_initialize(&mem_avail_mtx, MUTEX_ACTIVE);
1161
        condvar_initialize(&mem_avail_cv);
814 palkovsky 1162
    }
3972 decky 1163
 
814 palkovsky 1164
    /* Tell the architecture to create some memory */
1165
    frame_arch_init();
1166
    if (config.cpu_active == 1) {
2133 jermar 1167
        frame_mark_unavailable(ADDR2PFN(KA2PA(config.base)),
1168
            SIZE2FRAMES(config.kernel_size));
1169
        frame_mark_unavailable(ADDR2PFN(KA2PA(config.stack_base)),
1170
            SIZE2FRAMES(config.stack_size));
1037 decky 1171
 
1172
        count_t i;
2133 jermar 1173
        for (i = 0; i < init.cnt; i++) {
1174
            pfn_t pfn = ADDR2PFN(KA2PA(init.tasks[i].addr));
1175
            frame_mark_unavailable(pfn,
1176
                SIZE2FRAMES(init.tasks[i].size));
1177
        }
3972 decky 1178
 
1894 jermar 1179
        if (ballocs.size)
2133 jermar 1180
            frame_mark_unavailable(ADDR2PFN(KA2PA(ballocs.base)),
1181
                SIZE2FRAMES(ballocs.size));
3972 decky 1182
 
1599 palkovsky 1183
        /* Black list first frame, as allocating NULL would
3972 decky 1184
         * fail in some places
1185
         */
1599 palkovsky 1186
        frame_mark_unavailable(0, 1);
814 palkovsky 1187
    }
1188
}
1189
 
3185 jermar 1190
/** Return total size of all zones. */
1191
uint64_t zone_total_size(void)
1192
{
3972 decky 1193
    ipl_t ipl = interrupts_disable();
2725 decky 1194
    spinlock_lock(&zones.lock);
1195
 
3972 decky 1196
    uint64_t total = 0;
1197
    count_t i;
1198
    for (i = 0; i < zones.count; i++)
1199
        total += (uint64_t) FRAMES2SIZE(zones.info[i].count);
2725 decky 1200
 
1201
    spinlock_unlock(&zones.lock);
1202
    interrupts_restore(ipl);
1203
 
1204
    return total;
1205
}
1206
 
3185 jermar 1207
/** Prints list of zones. */
1208
void zone_print_list(void)
1209
{
3972 decky 1210
#ifdef __32_BITS__
3973 decky 1211
    printf("#  base address frames       flags    free frames  busy frames\n");
1212
    printf("-- ------------ ------------ -------- ------------ ------------\n");
3057 decky 1213
#endif
1214
 
1215
#ifdef __64_BITS__
3973 decky 1216
    printf("#  base address          frames      flags    free frames  busy frames\n");
1217
    printf("-- -------------------- ------------ -------- ------------ ------------\n");
3057 decky 1218
#endif
2712 decky 1219
 
3205 jermar 1220
    /*
1221
     * Because printing may require allocation of memory, we may not hold
1222
     * the frame allocator locks when printing zone statistics.  Therefore,
1223
     * we simply gather the statistics under the protection of the locks and
1224
     * print the statistics when the locks have been released.
1225
     *
1226
     * When someone adds/removes zones while we are printing the statistics,
1227
     * we may end up with inaccurate output (e.g. a zone being skipped from
1228
     * the listing).
1229
     */
3972 decky 1230
 
1231
    count_t i;
1232
    for (i = 0;; i++) {
1233
        ipl_t ipl = interrupts_disable();
3205 jermar 1234
        spinlock_lock(&zones.lock);
1235
 
1236
        if (i >= zones.count) {
1237
            spinlock_unlock(&zones.lock);
1238
            interrupts_restore(ipl);
1239
            break;
1240
        }
1241
 
3972 decky 1242
        uintptr_t base = PFN2ADDR(zones.info[i].base);
3973 decky 1243
        count_t count = zones.info[i].count;
3972 decky 1244
        zone_flags_t flags = zones.info[i].flags;
1245
        count_t free_count = zones.info[i].free_count;
1246
        count_t busy_count = zones.info[i].busy_count;
1247
 
3205 jermar 1248
        spinlock_unlock(&zones.lock);
1249
        interrupts_restore(ipl);
3972 decky 1250
 
1251
        bool available = zone_flags_available(flags);
1252
 
3973 decky 1253
        printf("%-2" PRIc, i);
1254
 
3057 decky 1255
#ifdef __32_BITS__
3973 decky 1256
        printf("   %10p", base);
3057 decky 1257
#endif
3972 decky 1258
 
3057 decky 1259
#ifdef __64_BITS__
3973 decky 1260
        printf("   %18p", base);
1261
#endif
1262
 
1263
        printf(" %12" PRIc " %c%c%c      ", count,
3972 decky 1264
            available ? 'A' : ' ',
1265
            (flags & ZONE_RESERVED) ? 'R' : ' ',
1266
            (flags & ZONE_FIRMWARE) ? 'F' : ' ');
2712 decky 1267
 
3972 decky 1268
        if (available)
1269
            printf("%12" PRIc " %12" PRIc,
1270
                free_count, busy_count);
3973 decky 1271
 
3972 decky 1272
        printf("\n");
676 bondari 1273
    }
1274
}
1275
 
1708 jermar 1276
/** Prints zone details.
677 bondari 1277
 *
3972 decky 1278
 * @param num Zone base address or zone number.
1279
 *
677 bondari 1280
 */
3972 decky 1281
void zone_print_one(count_t num)
3185 jermar 1282
{
3972 decky 1283
    ipl_t ipl = interrupts_disable();
814 palkovsky 1284
    spinlock_lock(&zones.lock);
3972 decky 1285
    count_t znum = (count_t) -1;
1286
 
1287
    count_t i;
1037 decky 1288
    for (i = 0; i < zones.count; i++) {
3972 decky 1289
        if ((i == num) || (PFN2ADDR(zones.info[i].base) == num)) {
1290
            znum = i;
822 palkovsky 1291
            break;
1292
        }
676 bondari 1293
    }
3972 decky 1294
 
1295
    if (znum == (count_t) -1) {
3206 jermar 1296
        spinlock_unlock(&zones.lock);
1297
        interrupts_restore(ipl);
822 palkovsky 1298
        printf("Zone not found.\n");
3206 jermar 1299
        return;
822 palkovsky 1300
    }
676 bondari 1301
 
3972 decky 1302
    uintptr_t base = PFN2ADDR(zones.info[i].base);
1303
    zone_flags_t flags = zones.info[i].flags;
1304
    count_t count = zones.info[i].count;
1305
    count_t free_count = zones.info[i].free_count;
1306
    count_t busy_count = zones.info[i].busy_count;
1307
 
814 palkovsky 1308
    spinlock_unlock(&zones.lock);
701 jermar 1309
    interrupts_restore(ipl);
3972 decky 1310
 
1311
    bool available = zone_flags_available(flags);
1312
 
1313
    printf("Zone number:       %" PRIc "\n", znum);
3206 jermar 1314
    printf("Zone base address: %p\n", base);
3972 decky 1315
    printf("Zone size:         %" PRIc " frames (%" PRIs " KiB)\n", count,
3206 jermar 1316
        SIZE2KB(FRAMES2SIZE(count)));
3972 decky 1317
    printf("Zone flags:        %c%c%c\n",
1318
        available ? 'A' : ' ',
1319
        (flags & ZONE_RESERVED) ? 'R' : ' ',
1320
        (flags & ZONE_FIRMWARE) ? 'F' : ' ');
1321
 
1322
    if (available) {
1323
        printf("Allocated space:   %" PRIc " frames (%" PRIs " KiB)\n",
1324
            busy_count, SIZE2KB(FRAMES2SIZE(busy_count)));
1325
        printf("Available space:   %" PRIc " frames (%" PRIs " KiB)\n",
1326
            free_count, SIZE2KB(FRAMES2SIZE(free_count)));
1327
    }
676 bondari 1328
}
1329
 
1757 jermar 1330
/** @}
1702 cejka 1331
 */