Subversion Repositories HelenOS

Rev

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

Rev Author Line No. Line
703 jermar 1
/*
2071 jermar 2
 * Copyright (c) 2001-2006 Jakub Jermar
703 jermar 3
 * All rights reserved.
4
 *
5
 * Redistribution and use in source and binary forms, with or without
6
 * modification, are permitted provided that the following conditions
7
 * are met:
8
 *
9
 * - Redistributions of source code must retain the above copyright
10
 *   notice, this list of conditions and the following disclaimer.
11
 * - Redistributions in binary form must reproduce the above copyright
12
 *   notice, this list of conditions and the following disclaimer in the
13
 *   documentation and/or other materials provided with the distribution.
14
 * - The name of the author may not be used to endorse or promote products
15
 *   derived from this software without specific prior written permission.
16
 *
17
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27
 */
28
 
1757 jermar 29
/** @addtogroup genericmm
1702 cejka 30
 * @{
31
 */
32
 
1248 jermar 33
/**
1702 cejka 34
 * @file
1248 jermar 35
 * @brief   Address space related functions.
36
 *
703 jermar 37
 * This file contains address space manipulation functions.
38
 * Roughly speaking, this is a higher-level client of
39
 * Virtual Address Translation (VAT) subsystem.
1248 jermar 40
 *
41
 * Functionality provided by this file allows one to
1757 jermar 42
 * create address spaces and create, resize and share
1248 jermar 43
 * address space areas.
44
 *
45
 * @see page.c
46
 *
703 jermar 47
 */
48
 
49
#include <mm/as.h>
756 jermar 50
#include <arch/mm/as.h>
703 jermar 51
#include <mm/page.h>
52
#include <mm/frame.h>
814 palkovsky 53
#include <mm/slab.h>
703 jermar 54
#include <mm/tlb.h>
55
#include <arch/mm/page.h>
56
#include <genarch/mm/page_pt.h>
1108 jermar 57
#include <genarch/mm/page_ht.h>
727 jermar 58
#include <mm/asid.h>
703 jermar 59
#include <arch/mm/asid.h>
2183 jermar 60
#include <preemption.h>
703 jermar 61
#include <synch/spinlock.h>
1380 jermar 62
#include <synch/mutex.h>
788 jermar 63
#include <adt/list.h>
1147 jermar 64
#include <adt/btree.h>
1235 jermar 65
#include <proc/task.h>
1288 jermar 66
#include <proc/thread.h>
1235 jermar 67
#include <arch/asm.h>
703 jermar 68
#include <panic.h>
69
#include <debug.h>
1235 jermar 70
#include <print.h>
703 jermar 71
#include <memstr.h>
1070 jermar 72
#include <macros.h>
703 jermar 73
#include <arch.h>
1235 jermar 74
#include <errno.h>
75
#include <config.h>
1387 jermar 76
#include <align.h>
1235 jermar 77
#include <arch/types.h>
1288 jermar 78
#include <syscall/copy.h>
79
#include <arch/interrupt.h>
703 jermar 80
 
2009 jermar 81
#ifdef CONFIG_VIRT_IDX_DCACHE
82
#include <arch/mm/cache.h>
83
#endif /* CONFIG_VIRT_IDX_DCACHE */
84
 
2125 decky 85
#ifndef __OBJC__
1757 jermar 86
/**
87
 * Each architecture decides what functions will be used to carry out
88
 * address space operations such as creating or locking page tables.
89
 */
756 jermar 90
as_operations_t *as_operations = NULL;
703 jermar 91
 
1890 jermar 92
/**
93
 * Slab for as_t objects.
94
 */
95
static slab_cache_t *as_slab;
2126 decky 96
#endif
1890 jermar 97
 
2087 jermar 98
/**
2170 jermar 99
 * This lock serializes access to the ASID subsystem.
100
 * It protects:
101
 * - inactive_as_with_asid_head list
102
 * - as->asid for each as of the as_t type
103
 * - asids_allocated counter
2087 jermar 104
 */
2170 jermar 105
SPINLOCK_INITIALIZE(asidlock);
823 jermar 106
 
107
/**
108
 * This list contains address spaces that are not active on any
109
 * processor and that have valid ASID.
110
 */
111
LIST_INITIALIZE(inactive_as_with_asid_head);
112
 
757 jermar 113
/** Kernel address space. */
114
as_t *AS_KERNEL = NULL;
115
 
1235 jermar 116
static int area_flags_to_page_flags(int aflags);
1780 jermar 117
static as_area_t *find_area_and_lock(as_t *as, uintptr_t va);
2087 jermar 118
static bool check_area_conflicts(as_t *as, uintptr_t va, size_t size,
119
    as_area_t *avoid_area);
1409 jermar 120
static void sh_info_remove_reference(share_info_t *sh_info);
703 jermar 121
 
2126 decky 122
#ifndef __OBJC__
1891 jermar 123
static int as_constructor(void *obj, int flags)
124
{
125
    as_t *as = (as_t *) obj;
126
    int rc;
127
 
128
    link_initialize(&as->inactive_as_with_asid_link);
129
    mutex_initialize(&as->lock);   
130
 
131
    rc = as_constructor_arch(as, flags);
132
 
133
    return rc;
134
}
135
 
136
static int as_destructor(void *obj)
137
{
138
    as_t *as = (as_t *) obj;
139
 
140
    return as_destructor_arch(as);
141
}
2126 decky 142
#endif
1891 jermar 143
 
756 jermar 144
/** Initialize address space subsystem. */
145
void as_init(void)
146
{
147
    as_arch_init();
2126 decky 148
 
149
#ifndef __OBJC__
1891 jermar 150
    as_slab = slab_cache_create("as_slab", sizeof(as_t), 0,
2087 jermar 151
        as_constructor, as_destructor, SLAB_CACHE_MAGDEFERRED);
2126 decky 152
#endif
1890 jermar 153
 
789 palkovsky 154
    AS_KERNEL = as_create(FLAG_AS_KERNEL);
1383 decky 155
    if (!AS_KERNEL)
156
        panic("can't create kernel address space\n");
157
 
756 jermar 158
}
159
 
757 jermar 160
/** Create address space.
161
 *
162
 * @param flags Flags that influence way in wich the address space is created.
163
 */
756 jermar 164
as_t *as_create(int flags)
703 jermar 165
{
166
    as_t *as;
167
 
2126 decky 168
#ifdef __OBJC__
169
    as = [as_t new];
170
    link_initialize(&as->inactive_as_with_asid_link);
171
    mutex_initialize(&as->lock);   
172
    (void) as_constructor_arch(as, flags);
173
#else
1890 jermar 174
    as = (as_t *) slab_alloc(as_slab, 0);
2126 decky 175
#endif
1891 jermar 176
    (void) as_create_arch(as, 0);
177
 
1147 jermar 178
    btree_create(&as->as_area_btree);
822 palkovsky 179
 
180
    if (flags & FLAG_AS_KERNEL)
181
        as->asid = ASID_KERNEL;
182
    else
183
        as->asid = ASID_INVALID;
184
 
2183 jermar 185
    atomic_set(&as->refcount, 0);
1415 jermar 186
    as->cpu_refcount = 0;
2089 decky 187
#ifdef AS_PAGE_TABLE
2106 jermar 188
    as->genarch.page_table = page_table_create(flags);
2089 decky 189
#else
190
    page_table_create(flags);
191
#endif
703 jermar 192
 
193
    return as;
194
}
195
 
1468 jermar 196
/** Destroy adress space.
197
 *
2087 jermar 198
 * When there are no tasks referencing this address space (i.e. its refcount is
199
 * zero), the address space can be destroyed.
2183 jermar 200
 *
201
 * We know that we don't hold any spinlock.
1468 jermar 202
 */
203
void as_destroy(as_t *as)
973 palkovsky 204
{
1468 jermar 205
    ipl_t ipl;
1594 jermar 206
    bool cond;
2183 jermar 207
    DEADLOCK_PROBE_INIT(p_asidlock);
973 palkovsky 208
 
2183 jermar 209
    ASSERT(atomic_get(&as->refcount) == 0);
1468 jermar 210
 
211
    /*
212
     * Since there is no reference to this area,
213
     * it is safe not to lock its mutex.
214
     */
2170 jermar 215
 
2183 jermar 216
    /*
217
     * We need to avoid deadlock between TLB shootdown and asidlock.
218
     * We therefore try to take asid conditionally and if we don't succeed,
219
     * we enable interrupts and try again. This is done while preemption is
220
     * disabled to prevent nested context switches. We also depend on the
221
     * fact that so far no spinlocks are held.
222
     */
223
    preemption_disable();
224
    ipl = interrupts_read();
225
retry:
226
    interrupts_disable();
227
    if (!spinlock_trylock(&asidlock)) {
228
        interrupts_enable();
229
        DEADLOCK_PROBE(p_asidlock, DEADLOCK_THRESHOLD);
230
        goto retry;
231
    }
232
    preemption_enable();    /* Interrupts disabled, enable preemption */
1587 jermar 233
    if (as->asid != ASID_INVALID && as != AS_KERNEL) {
1594 jermar 234
        if (as != AS && as->cpu_refcount == 0)
1587 jermar 235
            list_remove(&as->inactive_as_with_asid_link);
1468 jermar 236
        asid_put(as->asid);
237
    }
2170 jermar 238
    spinlock_unlock(&asidlock);
1468 jermar 239
 
240
    /*
241
     * Destroy address space areas of the address space.
1954 jermar 242
     * The B+tree must be walked carefully because it is
1594 jermar 243
     * also being destroyed.
1468 jermar 244
     */
1594 jermar 245
    for (cond = true; cond; ) {
1468 jermar 246
        btree_node_t *node;
1594 jermar 247
 
248
        ASSERT(!list_empty(&as->as_area_btree.leaf_head));
2087 jermar 249
        node = list_get_instance(as->as_area_btree.leaf_head.next,
250
            btree_node_t, leaf_link);
1594 jermar 251
 
252
        if ((cond = node->keys)) {
253
            as_area_destroy(as, node->key[0]);
254
        }
1468 jermar 255
    }
1495 jermar 256
 
1483 jermar 257
    btree_destroy(&as->as_area_btree);
2089 decky 258
#ifdef AS_PAGE_TABLE
2106 jermar 259
    page_table_destroy(as->genarch.page_table);
2089 decky 260
#else
261
    page_table_destroy(NULL);
262
#endif
1468 jermar 263
 
264
    interrupts_restore(ipl);
2126 decky 265
 
266
#ifdef __OBJC__
267
    [as free];
268
#else
1890 jermar 269
    slab_free(as_slab, as);
2126 decky 270
#endif
973 palkovsky 271
}
272
 
703 jermar 273
/** Create address space area of common attributes.
274
 *
275
 * The created address space area is added to the target address space.
276
 *
277
 * @param as Target address space.
1239 jermar 278
 * @param flags Flags of the area memory.
1048 jermar 279
 * @param size Size of area.
703 jermar 280
 * @param base Base address of area.
1239 jermar 281
 * @param attrs Attributes of the area.
1409 jermar 282
 * @param backend Address space area backend. NULL if no backend is used.
283
 * @param backend_data NULL or a pointer to an array holding two void *.
703 jermar 284
 *
285
 * @return Address space area on success or NULL on failure.
286
 */
2069 jermar 287
as_area_t *
288
as_area_create(as_t *as, int flags, size_t size, uintptr_t base, int attrs,
1424 jermar 289
           mem_backend_t *backend, mem_backend_data_t *backend_data)
703 jermar 290
{
291
    ipl_t ipl;
292
    as_area_t *a;
293
 
294
    if (base % PAGE_SIZE)
1048 jermar 295
        return NULL;
296
 
1233 jermar 297
    if (!size)
298
        return NULL;
299
 
1048 jermar 300
    /* Writeable executable areas are not supported. */
301
    if ((flags & AS_AREA_EXEC) && (flags & AS_AREA_WRITE))
302
        return NULL;
703 jermar 303
 
304
    ipl = interrupts_disable();
1380 jermar 305
    mutex_lock(&as->lock);
703 jermar 306
 
1048 jermar 307
    if (!check_area_conflicts(as, base, size, NULL)) {
1380 jermar 308
        mutex_unlock(&as->lock);
1048 jermar 309
        interrupts_restore(ipl);
310
        return NULL;
311
    }
703 jermar 312
 
822 palkovsky 313
    a = (as_area_t *) malloc(sizeof(as_area_t), 0);
703 jermar 314
 
1380 jermar 315
    mutex_initialize(&a->lock);
822 palkovsky 316
 
1424 jermar 317
    a->as = as;
1026 jermar 318
    a->flags = flags;
1239 jermar 319
    a->attributes = attrs;
1048 jermar 320
    a->pages = SIZE2FRAMES(size);
822 palkovsky 321
    a->base = base;
1409 jermar 322
    a->sh_info = NULL;
323
    a->backend = backend;
1424 jermar 324
    if (backend_data)
325
        a->backend_data = *backend_data;
326
    else
2087 jermar 327
        memsetb((uintptr_t) &a->backend_data, sizeof(a->backend_data),
328
            0);
1424 jermar 329
 
1387 jermar 330
    btree_create(&a->used_space);
822 palkovsky 331
 
1147 jermar 332
    btree_insert(&as->as_area_btree, base, (void *) a, NULL);
822 palkovsky 333
 
1380 jermar 334
    mutex_unlock(&as->lock);
703 jermar 335
    interrupts_restore(ipl);
704 jermar 336
 
703 jermar 337
    return a;
338
}
339
 
1235 jermar 340
/** Find address space area and change it.
341
 *
342
 * @param as Address space.
2087 jermar 343
 * @param address Virtual address belonging to the area to be changed. Must be
344
 *     page-aligned.
1235 jermar 345
 * @param size New size of the virtual memory block starting at address.
346
 * @param flags Flags influencing the remap operation. Currently unused.
347
 *
1306 jermar 348
 * @return Zero on success or a value from @ref errno.h otherwise.
1235 jermar 349
 */
1780 jermar 350
int as_area_resize(as_t *as, uintptr_t address, size_t size, int flags)
1235 jermar 351
{
1306 jermar 352
    as_area_t *area;
1235 jermar 353
    ipl_t ipl;
354
    size_t pages;
355
 
356
    ipl = interrupts_disable();
1380 jermar 357
    mutex_lock(&as->lock);
1235 jermar 358
 
359
    /*
360
     * Locate the area.
361
     */
362
    area = find_area_and_lock(as, address);
363
    if (!area) {
1380 jermar 364
        mutex_unlock(&as->lock);
1235 jermar 365
        interrupts_restore(ipl);
1306 jermar 366
        return ENOENT;
1235 jermar 367
    }
368
 
1424 jermar 369
    if (area->backend == &phys_backend) {
1235 jermar 370
        /*
371
         * Remapping of address space areas associated
372
         * with memory mapped devices is not supported.
373
         */
1380 jermar 374
        mutex_unlock(&area->lock);
375
        mutex_unlock(&as->lock);
1235 jermar 376
        interrupts_restore(ipl);
1306 jermar 377
        return ENOTSUP;
1235 jermar 378
    }
1409 jermar 379
    if (area->sh_info) {
380
        /*
381
         * Remapping of shared address space areas
382
         * is not supported.
383
         */
384
        mutex_unlock(&area->lock);
385
        mutex_unlock(&as->lock);
386
        interrupts_restore(ipl);
387
        return ENOTSUP;
388
    }
1235 jermar 389
 
390
    pages = SIZE2FRAMES((address - area->base) + size);
391
    if (!pages) {
392
        /*
393
         * Zero size address space areas are not allowed.
394
         */
1380 jermar 395
        mutex_unlock(&area->lock);
396
        mutex_unlock(&as->lock);
1235 jermar 397
        interrupts_restore(ipl);
1306 jermar 398
        return EPERM;
1235 jermar 399
    }
400
 
401
    if (pages < area->pages) {
1403 jermar 402
        bool cond;
1780 jermar 403
        uintptr_t start_free = area->base + pages*PAGE_SIZE;
1235 jermar 404
 
405
        /*
406
         * Shrinking the area.
407
         * No need to check for overlaps.
408
         */
1403 jermar 409
 
410
        /*
1436 jermar 411
         * Start TLB shootdown sequence.
412
         */
2087 jermar 413
        tlb_shootdown_start(TLB_INVL_PAGES, AS->asid, area->base +
414
            pages * PAGE_SIZE, area->pages - pages);
1436 jermar 415
 
416
        /*
1403 jermar 417
         * Remove frames belonging to used space starting from
418
         * the highest addresses downwards until an overlap with
419
         * the resized address space area is found. Note that this
420
         * is also the right way to remove part of the used_space
421
         * B+tree leaf list.
422
         */    
423
        for (cond = true; cond;) {
424
            btree_node_t *node;
425
 
426
            ASSERT(!list_empty(&area->used_space.leaf_head));
2087 jermar 427
            node =
428
                list_get_instance(area->used_space.leaf_head.prev,
429
                btree_node_t, leaf_link);
1403 jermar 430
            if ((cond = (bool) node->keys)) {
1780 jermar 431
                uintptr_t b = node->key[node->keys - 1];
2087 jermar 432
                count_t c =
433
                    (count_t) node->value[node->keys - 1];
2745 decky 434
                unsigned int i = 0;
1235 jermar 435
 
2087 jermar 436
                if (overlaps(b, c * PAGE_SIZE, area->base,
2133 jermar 437
                    pages * PAGE_SIZE)) {
1403 jermar 438
 
2087 jermar 439
                    if (b + c * PAGE_SIZE <= start_free) {
1403 jermar 440
                        /*
2087 jermar 441
                         * The whole interval fits
442
                         * completely in the resized
443
                         * address space area.
1403 jermar 444
                         */
445
                        break;
446
                    }
447
 
448
                    /*
2087 jermar 449
                     * Part of the interval corresponding
450
                     * to b and c overlaps with the resized
451
                     * address space area.
1403 jermar 452
                     */
453
 
454
                    cond = false;   /* we are almost done */
455
                    i = (start_free - b) >> PAGE_WIDTH;
2087 jermar 456
                    if (!used_space_remove(area, start_free,
457
                        c - i))
458
                        panic("Could not remove used "
459
                            "space.\n");
1403 jermar 460
                } else {
461
                    /*
2087 jermar 462
                     * The interval of used space can be
463
                     * completely removed.
1403 jermar 464
                     */
465
                    if (!used_space_remove(area, b, c))
2087 jermar 466
                        panic("Could not remove used "
467
                            "space.\n");
1403 jermar 468
                }
469
 
470
                for (; i < c; i++) {
471
                    pte_t *pte;
472
 
473
                    page_table_lock(as, false);
2087 jermar 474
                    pte = page_mapping_find(as, b +
475
                        i * PAGE_SIZE);
476
                    ASSERT(pte && PTE_VALID(pte) &&
477
                        PTE_PRESENT(pte));
478
                    if (area->backend &&
479
                        area->backend->frame_free) {
1424 jermar 480
                        area->backend->frame_free(area,
2087 jermar 481
                            b + i * PAGE_SIZE,
482
                            PTE_GET_FRAME(pte));
1409 jermar 483
                    }
2087 jermar 484
                    page_mapping_remove(as, b +
485
                        i * PAGE_SIZE);
1403 jermar 486
                    page_table_unlock(as, false);
487
                }
1235 jermar 488
            }
489
        }
1436 jermar 490
 
1235 jermar 491
        /*
1436 jermar 492
         * Finish TLB shootdown sequence.
1235 jermar 493
         */
2183 jermar 494
 
2087 jermar 495
        tlb_invalidate_pages(as->asid, area->base + pages * PAGE_SIZE,
496
            area->pages - pages);
1889 jermar 497
        /*
498
         * Invalidate software translation caches (e.g. TSB on sparc64).
499
         */
2087 jermar 500
        as_invalidate_translation_cache(as, area->base +
501
            pages * PAGE_SIZE, area->pages - pages);
2183 jermar 502
        tlb_shootdown_finalize();
503
 
1235 jermar 504
    } else {
505
        /*
506
         * Growing the area.
507
         * Check for overlaps with other address space areas.
508
         */
2087 jermar 509
        if (!check_area_conflicts(as, address, pages * PAGE_SIZE,
510
            area)) {
1380 jermar 511
            mutex_unlock(&area->lock);
512
            mutex_unlock(&as->lock);       
1235 jermar 513
            interrupts_restore(ipl);
1306 jermar 514
            return EADDRNOTAVAIL;
1235 jermar 515
        }
516
    }
517
 
518
    area->pages = pages;
519
 
1380 jermar 520
    mutex_unlock(&area->lock);
521
    mutex_unlock(&as->lock);
1235 jermar 522
    interrupts_restore(ipl);
523
 
1306 jermar 524
    return 0;
1235 jermar 525
}
526
 
1306 jermar 527
/** Destroy address space area.
528
 *
529
 * @param as Address space.
530
 * @param address Address withing the area to be deleted.
531
 *
532
 * @return Zero on success or a value from @ref errno.h on failure.
533
 */
1780 jermar 534
int as_area_destroy(as_t *as, uintptr_t address)
1306 jermar 535
{
536
    as_area_t *area;
1780 jermar 537
    uintptr_t base;
1495 jermar 538
    link_t *cur;
1306 jermar 539
    ipl_t ipl;
540
 
541
    ipl = interrupts_disable();
1380 jermar 542
    mutex_lock(&as->lock);
1306 jermar 543
 
544
    area = find_area_and_lock(as, address);
545
    if (!area) {
1380 jermar 546
        mutex_unlock(&as->lock);
1306 jermar 547
        interrupts_restore(ipl);
548
        return ENOENT;
549
    }
550
 
1403 jermar 551
    base = area->base;
552
 
1411 jermar 553
    /*
1436 jermar 554
     * Start TLB shootdown sequence.
555
     */
1889 jermar 556
    tlb_shootdown_start(TLB_INVL_PAGES, as->asid, area->base, area->pages);
1436 jermar 557
 
558
    /*
1411 jermar 559
     * Visit only the pages mapped by used_space B+tree.
560
     */
2087 jermar 561
    for (cur = area->used_space.leaf_head.next;
562
        cur != &area->used_space.leaf_head; cur = cur->next) {
1411 jermar 563
        btree_node_t *node;
2745 decky 564
        unsigned int i;
1403 jermar 565
 
1495 jermar 566
        node = list_get_instance(cur, btree_node_t, leaf_link);
567
        for (i = 0; i < node->keys; i++) {
1780 jermar 568
            uintptr_t b = node->key[i];
1495 jermar 569
            count_t j;
1411 jermar 570
            pte_t *pte;
1403 jermar 571
 
1495 jermar 572
            for (j = 0; j < (count_t) node->value[i]; j++) {
1411 jermar 573
                page_table_lock(as, false);
2087 jermar 574
                pte = page_mapping_find(as, b + j * PAGE_SIZE);
575
                ASSERT(pte && PTE_VALID(pte) &&
576
                    PTE_PRESENT(pte));
577
                if (area->backend &&
578
                    area->backend->frame_free) {
579
                    area->backend->frame_free(area, b +
2133 jermar 580
                        j * PAGE_SIZE, PTE_GET_FRAME(pte));
1403 jermar 581
                }
2087 jermar 582
                page_mapping_remove(as, b + j * PAGE_SIZE);            
1411 jermar 583
                page_table_unlock(as, false);
1306 jermar 584
            }
585
        }
586
    }
1403 jermar 587
 
1306 jermar 588
    /*
1436 jermar 589
     * Finish TLB shootdown sequence.
1306 jermar 590
     */
2183 jermar 591
 
1889 jermar 592
    tlb_invalidate_pages(as->asid, area->base, area->pages);
593
    /*
2087 jermar 594
     * Invalidate potential software translation caches (e.g. TSB on
595
     * sparc64).
1889 jermar 596
     */
597
    as_invalidate_translation_cache(as, area->base, area->pages);
2183 jermar 598
    tlb_shootdown_finalize();
1889 jermar 599
 
1436 jermar 600
    btree_destroy(&area->used_space);
1306 jermar 601
 
1309 jermar 602
    area->attributes |= AS_AREA_ATTR_PARTIAL;
1409 jermar 603
 
604
    if (area->sh_info)
605
        sh_info_remove_reference(area->sh_info);
606
 
1380 jermar 607
    mutex_unlock(&area->lock);
1306 jermar 608
 
609
    /*
610
     * Remove the empty area from address space.
611
     */
1889 jermar 612
    btree_remove(&as->as_area_btree, base, NULL);
1306 jermar 613
 
1309 jermar 614
    free(area);
615
 
1889 jermar 616
    mutex_unlock(&as->lock);
1306 jermar 617
    interrupts_restore(ipl);
618
    return 0;
619
}
620
 
1413 jermar 621
/** Share address space area with another or the same address space.
1235 jermar 622
 *
1424 jermar 623
 * Address space area mapping is shared with a new address space area.
624
 * If the source address space area has not been shared so far,
625
 * a new sh_info is created. The new address space area simply gets the
626
 * sh_info of the source area. The process of duplicating the
627
 * mapping is done through the backend share function.
1413 jermar 628
 *
1417 jermar 629
 * @param src_as Pointer to source address space.
1239 jermar 630
 * @param src_base Base address of the source address space area.
1417 jermar 631
 * @param acc_size Expected size of the source area.
1428 palkovsky 632
 * @param dst_as Pointer to destination address space.
1417 jermar 633
 * @param dst_base Target base address.
634
 * @param dst_flags_mask Destination address space area flags mask.
1235 jermar 635
 *
2007 jermar 636
 * @return Zero on success or ENOENT if there is no such task or if there is no
637
 * such address space area, EPERM if there was a problem in accepting the area
638
 * or ENOMEM if there was a problem in allocating destination address space
639
 * area. ENOTSUP is returned if the address space area backend does not support
2141 jermar 640
 * sharing.
1235 jermar 641
 */
1780 jermar 642
int as_area_share(as_t *src_as, uintptr_t src_base, size_t acc_size,
2647 jermar 643
    as_t *dst_as, uintptr_t dst_base, int dst_flags_mask)
1235 jermar 644
{
645
    ipl_t ipl;
1239 jermar 646
    int src_flags;
647
    size_t src_size;
648
    as_area_t *src_area, *dst_area;
1413 jermar 649
    share_info_t *sh_info;
1424 jermar 650
    mem_backend_t *src_backend;
651
    mem_backend_data_t src_backend_data;
1434 palkovsky 652
 
1235 jermar 653
    ipl = interrupts_disable();
1380 jermar 654
    mutex_lock(&src_as->lock);
1329 palkovsky 655
    src_area = find_area_and_lock(src_as, src_base);
1239 jermar 656
    if (!src_area) {
1238 jermar 657
        /*
658
         * Could not find the source address space area.
659
         */
1380 jermar 660
        mutex_unlock(&src_as->lock);
1238 jermar 661
        interrupts_restore(ipl);
662
        return ENOENT;
663
    }
2007 jermar 664
 
1424 jermar 665
    if (!src_area->backend || !src_area->backend->share) {
1413 jermar 666
        /*
1851 jermar 667
         * There is no backend or the backend does not
1424 jermar 668
         * know how to share the area.
1413 jermar 669
         */
670
        mutex_unlock(&src_area->lock);
671
        mutex_unlock(&src_as->lock);
672
        interrupts_restore(ipl);
673
        return ENOTSUP;
674
    }
675
 
1239 jermar 676
    src_size = src_area->pages * PAGE_SIZE;
677
    src_flags = src_area->flags;
1424 jermar 678
    src_backend = src_area->backend;
679
    src_backend_data = src_area->backend_data;
1544 palkovsky 680
 
681
    /* Share the cacheable flag from the original mapping */
682
    if (src_flags & AS_AREA_CACHEABLE)
683
        dst_flags_mask |= AS_AREA_CACHEABLE;
684
 
2087 jermar 685
    if (src_size != acc_size ||
686
        (src_flags & dst_flags_mask) != dst_flags_mask) {
1413 jermar 687
        mutex_unlock(&src_area->lock);
688
        mutex_unlock(&src_as->lock);
1235 jermar 689
        interrupts_restore(ipl);
690
        return EPERM;
691
    }
1413 jermar 692
 
1235 jermar 693
    /*
1413 jermar 694
     * Now we are committed to sharing the area.
1954 jermar 695
     * First, prepare the area for sharing.
1413 jermar 696
     * Then it will be safe to unlock it.
697
     */
698
    sh_info = src_area->sh_info;
699
    if (!sh_info) {
700
        sh_info = (share_info_t *) malloc(sizeof(share_info_t), 0);
701
        mutex_initialize(&sh_info->lock);
702
        sh_info->refcount = 2;
703
        btree_create(&sh_info->pagemap);
704
        src_area->sh_info = sh_info;
2647 jermar 705
        /*
706
         * Call the backend to setup sharing.
707
         */
708
        src_area->backend->share(src_area);
1413 jermar 709
    } else {
710
        mutex_lock(&sh_info->lock);
711
        sh_info->refcount++;
712
        mutex_unlock(&sh_info->lock);
713
    }
714
 
715
    mutex_unlock(&src_area->lock);
716
    mutex_unlock(&src_as->lock);
717
 
718
    /*
1239 jermar 719
     * Create copy of the source address space area.
720
     * The destination area is created with AS_AREA_ATTR_PARTIAL
721
     * attribute set which prevents race condition with
722
     * preliminary as_page_fault() calls.
1417 jermar 723
     * The flags of the source area are masked against dst_flags_mask
724
     * to support sharing in less privileged mode.
1235 jermar 725
     */
1461 palkovsky 726
    dst_area = as_area_create(dst_as, dst_flags_mask, src_size, dst_base,
2087 jermar 727
        AS_AREA_ATTR_PARTIAL, src_backend, &src_backend_data);
1239 jermar 728
    if (!dst_area) {
1235 jermar 729
        /*
730
         * Destination address space area could not be created.
731
         */
1413 jermar 732
        sh_info_remove_reference(sh_info);
733
 
1235 jermar 734
        interrupts_restore(ipl);
735
        return ENOMEM;
736
    }
2009 jermar 737
 
1235 jermar 738
    /*
1239 jermar 739
     * Now the destination address space area has been
740
     * fully initialized. Clear the AS_AREA_ATTR_PARTIAL
1413 jermar 741
     * attribute and set the sh_info.
1239 jermar 742
     */
2009 jermar 743
    mutex_lock(&dst_as->lock); 
1380 jermar 744
    mutex_lock(&dst_area->lock);
1239 jermar 745
    dst_area->attributes &= ~AS_AREA_ATTR_PARTIAL;
1413 jermar 746
    dst_area->sh_info = sh_info;
1380 jermar 747
    mutex_unlock(&dst_area->lock);
2009 jermar 748
    mutex_unlock(&dst_as->lock);   
749
 
1235 jermar 750
    interrupts_restore(ipl);
751
 
752
    return 0;
753
}
754
 
1423 jermar 755
/** Check access mode for address space area.
756
 *
757
 * The address space area must be locked prior to this call.
758
 *
759
 * @param area Address space area.
760
 * @param access Access mode.
761
 *
762
 * @return False if access violates area's permissions, true otherwise.
763
 */
764
bool as_area_check_access(as_area_t *area, pf_access_t access)
765
{
766
    int flagmap[] = {
767
        [PF_ACCESS_READ] = AS_AREA_READ,
768
        [PF_ACCESS_WRITE] = AS_AREA_WRITE,
769
        [PF_ACCESS_EXEC] = AS_AREA_EXEC
770
    };
771
 
772
    if (!(area->flags & flagmap[access]))
773
        return false;
774
 
775
    return true;
776
}
777
 
2984 svoboda 778
/** Change adress area flags.
779
 *
780
 * The idea is to have the same data, but with a different access mode.
781
 * This is needed e.g. for writing code into memory and then executing it.
782
 * In order for this to work properly, this may copy the data
783
 * into private anonymous memory (unless it's already there).
784
 *
785
 * @param as Address space.
786
 * @param flags Flags of the area memory.
787
 * @param address Address withing the area to be changed.
788
 *
789
 * @return Zero on success or a value from @ref errno.h on failure.
790
 */
791
int as_area_change_flags(as_t *as, int flags, uintptr_t address)
792
{
793
    as_area_t *area;
794
    uintptr_t base;
795
    link_t *cur;
796
    ipl_t ipl;
797
 
798
    ipl = interrupts_disable();
799
    mutex_lock(&as->lock);
800
 
801
    area = find_area_and_lock(as, address);
802
    if (!area) {
803
        mutex_unlock(&as->lock);
804
        interrupts_restore(ipl);
805
        return ENOENT;
806
    }
807
 
808
    if (area->sh_info || area->backend != &anon_backend) {
809
        /* Copying shared areas not supported yet */
810
        /* Copying non-anonymous memory not supported yet */
811
        mutex_unlock(&as->lock);
812
        interrupts_restore(ipl);
813
        return ENOTSUP;
814
    }
815
 
816
    base = area->base;
817
 
818
    /*
819
     * Start TLB shootdown sequence.
820
     */
821
    tlb_shootdown_start(TLB_INVL_PAGES, as->asid, area->base, area->pages);
822
 
823
    /*
824
     * Visit only the pages mapped by used_space B+tree.
825
     */
826
    for (cur = area->used_space.leaf_head.next;
827
        cur != &area->used_space.leaf_head; cur = cur->next) {
828
        btree_node_t *node;
829
        unsigned int i;
830
 
831
        node = list_get_instance(cur, btree_node_t, leaf_link);
832
        for (i = 0; i < node->keys; i++) {
833
            uintptr_t b = node->key[i];
834
            count_t j;
835
            pte_t *pte;
836
 
837
            for (j = 0; j < (count_t) node->value[i]; j++) {
838
                page_table_lock(as, false);
839
                pte = page_mapping_find(as, b + j * PAGE_SIZE);
840
                ASSERT(pte && PTE_VALID(pte) &&
841
                    PTE_PRESENT(pte));
842
 
843
                /* Remove old mapping and insert the new one */
844
                page_mapping_remove(as, b + j * PAGE_SIZE);
845
                page_mapping_insert(as, b + j * PAGE_SIZE,
846
                    PTE_GET_FRAME(pte), flags);
847
                page_table_unlock(as, false);
848
            }
849
        }
850
    }
851
 
852
    /*
853
     * Finish TLB shootdown sequence.
854
     */
855
 
856
    tlb_invalidate_pages(as->asid, area->base, area->pages);
857
    /*
858
     * Invalidate potential software translation caches (e.g. TSB on
859
     * sparc64).
860
     */
861
    as_invalidate_translation_cache(as, area->base, area->pages);
862
    tlb_shootdown_finalize();
863
 
864
    mutex_unlock(&area->lock);
865
    mutex_unlock(&as->lock);
866
    interrupts_restore(ipl);
867
 
868
    return 0;
869
}
870
 
871
 
703 jermar 872
/** Handle page fault within the current address space.
873
 *
1409 jermar 874
 * This is the high-level page fault handler. It decides
875
 * whether the page fault can be resolved by any backend
876
 * and if so, it invokes the backend to resolve the page
877
 * fault.
878
 *
703 jermar 879
 * Interrupts are assumed disabled.
880
 *
881
 * @param page Faulting page.
1411 jermar 882
 * @param access Access mode that caused the fault (i.e. read/write/exec).
1288 jermar 883
 * @param istate Pointer to interrupted state.
703 jermar 884
 *
1409 jermar 885
 * @return AS_PF_FAULT on page fault, AS_PF_OK on success or AS_PF_DEFER if the
886
 *     fault was caused by copy_to_uspace() or copy_from_uspace().
703 jermar 887
 */
1780 jermar 888
int as_page_fault(uintptr_t page, pf_access_t access, istate_t *istate)
703 jermar 889
{
1044 jermar 890
    pte_t *pte;
977 jermar 891
    as_area_t *area;
703 jermar 892
 
1380 jermar 893
    if (!THREAD)
1409 jermar 894
        return AS_PF_FAULT;
1380 jermar 895
 
703 jermar 896
    ASSERT(AS);
1044 jermar 897
 
1380 jermar 898
    mutex_lock(&AS->lock);
977 jermar 899
    area = find_area_and_lock(AS, page);   
703 jermar 900
    if (!area) {
901
        /*
902
         * No area contained mapping for 'page'.
903
         * Signal page fault to low-level handler.
904
         */
1380 jermar 905
        mutex_unlock(&AS->lock);
1288 jermar 906
        goto page_fault;
703 jermar 907
    }
908
 
1239 jermar 909
    if (area->attributes & AS_AREA_ATTR_PARTIAL) {
910
        /*
911
         * The address space area is not fully initialized.
912
         * Avoid possible race by returning error.
913
         */
1380 jermar 914
        mutex_unlock(&area->lock);
915
        mutex_unlock(&AS->lock);
1288 jermar 916
        goto page_fault;       
1239 jermar 917
    }
918
 
1424 jermar 919
    if (!area->backend || !area->backend->page_fault) {
1409 jermar 920
        /*
921
         * The address space area is not backed by any backend
922
         * or the backend cannot handle page faults.
923
         */
924
        mutex_unlock(&area->lock);
925
        mutex_unlock(&AS->lock);
926
        goto page_fault;       
927
    }
1179 jermar 928
 
1044 jermar 929
    page_table_lock(AS, false);
930
 
703 jermar 931
    /*
1044 jermar 932
     * To avoid race condition between two page faults
933
     * on the same address, we need to make sure
934
     * the mapping has not been already inserted.
935
     */
936
    if ((pte = page_mapping_find(AS, page))) {
937
        if (PTE_PRESENT(pte)) {
1423 jermar 938
            if (((access == PF_ACCESS_READ) && PTE_READABLE(pte)) ||
2087 jermar 939
                (access == PF_ACCESS_WRITE && PTE_WRITABLE(pte)) ||
940
                (access == PF_ACCESS_EXEC && PTE_EXECUTABLE(pte))) {
1423 jermar 941
                page_table_unlock(AS, false);
942
                mutex_unlock(&area->lock);
943
                mutex_unlock(&AS->lock);
944
                return AS_PF_OK;
945
            }
1044 jermar 946
        }
947
    }
1409 jermar 948
 
1044 jermar 949
    /*
1409 jermar 950
     * Resort to the backend page fault handler.
703 jermar 951
     */
1424 jermar 952
    if (area->backend->page_fault(area, page, access) != AS_PF_OK) {
1409 jermar 953
        page_table_unlock(AS, false);
954
        mutex_unlock(&area->lock);
955
        mutex_unlock(&AS->lock);
956
        goto page_fault;
957
    }
703 jermar 958
 
1044 jermar 959
    page_table_unlock(AS, false);
1380 jermar 960
    mutex_unlock(&area->lock);
961
    mutex_unlock(&AS->lock);
1288 jermar 962
    return AS_PF_OK;
963
 
964
page_fault:
965
    if (THREAD->in_copy_from_uspace) {
966
        THREAD->in_copy_from_uspace = false;
2087 jermar 967
        istate_set_retaddr(istate,
968
            (uintptr_t) &memcpy_from_uspace_failover_address);
1288 jermar 969
    } else if (THREAD->in_copy_to_uspace) {
970
        THREAD->in_copy_to_uspace = false;
2087 jermar 971
        istate_set_retaddr(istate,
972
            (uintptr_t) &memcpy_to_uspace_failover_address);
1288 jermar 973
    } else {
974
        return AS_PF_FAULT;
975
    }
976
 
977
    return AS_PF_DEFER;
703 jermar 978
}
979
 
823 jermar 980
/** Switch address spaces.
703 jermar 981
 *
1380 jermar 982
 * Note that this function cannot sleep as it is essentially a part of
2170 jermar 983
 * scheduling. Sleeping here would lead to deadlock on wakeup. Another
984
 * thing which is forbidden in this context is locking the address space.
1380 jermar 985
 *
2183 jermar 986
 * When this function is enetered, no spinlocks may be held.
987
 *
823 jermar 988
 * @param old Old address space or NULL.
989
 * @param new New address space.
703 jermar 990
 */
2106 jermar 991
void as_switch(as_t *old_as, as_t *new_as)
703 jermar 992
{
2183 jermar 993
    DEADLOCK_PROBE_INIT(p_asidlock);
994
    preemption_disable();
995
retry:
996
    (void) interrupts_disable();
997
    if (!spinlock_trylock(&asidlock)) {
998
        /*
999
         * Avoid deadlock with TLB shootdown.
1000
         * We can enable interrupts here because
1001
         * preemption is disabled. We should not be
1002
         * holding any other lock.
1003
         */
1004
        (void) interrupts_enable();
1005
        DEADLOCK_PROBE(p_asidlock, DEADLOCK_THRESHOLD);
1006
        goto retry;
1007
    }
1008
    preemption_enable();
703 jermar 1009
 
1010
    /*
823 jermar 1011
     * First, take care of the old address space.
1012
     */
2106 jermar 1013
    if (old_as) {
1014
        ASSERT(old_as->cpu_refcount);
1015
        if((--old_as->cpu_refcount == 0) && (old_as != AS_KERNEL)) {
823 jermar 1016
            /*
1017
             * The old address space is no longer active on
1018
             * any processor. It can be appended to the
1019
             * list of inactive address spaces with assigned
1020
             * ASID.
1021
             */
2141 jermar 1022
            ASSERT(old_as->asid != ASID_INVALID);
1023
            list_append(&old_as->inactive_as_with_asid_link,
1024
                &inactive_as_with_asid_head);
823 jermar 1025
        }
1890 jermar 1026
 
1027
        /*
1028
         * Perform architecture-specific tasks when the address space
1029
         * is being removed from the CPU.
1030
         */
2106 jermar 1031
        as_deinstall_arch(old_as);
823 jermar 1032
    }
1033
 
1034
    /*
1035
     * Second, prepare the new address space.
1036
     */
2106 jermar 1037
    if ((new_as->cpu_refcount++ == 0) && (new_as != AS_KERNEL)) {
2170 jermar 1038
        if (new_as->asid != ASID_INVALID)
2106 jermar 1039
            list_remove(&new_as->inactive_as_with_asid_link);
2170 jermar 1040
        else
1041
            new_as->asid = asid_get();
823 jermar 1042
    }
2106 jermar 1043
#ifdef AS_PAGE_TABLE
1044
    SET_PTL0_ADDRESS(new_as->genarch.page_table);
1045
#endif
823 jermar 1046
 
1047
    /*
703 jermar 1048
     * Perform architecture-specific steps.
727 jermar 1049
     * (e.g. write ASID to hardware register etc.)
703 jermar 1050
     */
2106 jermar 1051
    as_install_arch(new_as);
2170 jermar 1052
 
1053
    spinlock_unlock(&asidlock);
703 jermar 1054
 
2106 jermar 1055
    AS = new_as;
703 jermar 1056
}
754 jermar 1057
 
1235 jermar 1058
/** Convert address space area flags to page flags.
754 jermar 1059
 *
1235 jermar 1060
 * @param aflags Flags of some address space area.
754 jermar 1061
 *
1235 jermar 1062
 * @return Flags to be passed to page_mapping_insert().
754 jermar 1063
 */
1235 jermar 1064
int area_flags_to_page_flags(int aflags)
754 jermar 1065
{
1066
    int flags;
1067
 
1178 jermar 1068
    flags = PAGE_USER | PAGE_PRESENT;
754 jermar 1069
 
1235 jermar 1070
    if (aflags & AS_AREA_READ)
1026 jermar 1071
        flags |= PAGE_READ;
1072
 
1235 jermar 1073
    if (aflags & AS_AREA_WRITE)
1026 jermar 1074
        flags |= PAGE_WRITE;
1075
 
1235 jermar 1076
    if (aflags & AS_AREA_EXEC)
1026 jermar 1077
        flags |= PAGE_EXEC;
1078
 
1424 jermar 1079
    if (aflags & AS_AREA_CACHEABLE)
1178 jermar 1080
        flags |= PAGE_CACHEABLE;
1081
 
754 jermar 1082
    return flags;
1083
}
756 jermar 1084
 
1235 jermar 1085
/** Compute flags for virtual address translation subsytem.
1086
 *
1087
 * The address space area must be locked.
1088
 * Interrupts must be disabled.
1089
 *
1090
 * @param a Address space area.
1091
 *
1092
 * @return Flags to be used in page_mapping_insert().
1093
 */
1409 jermar 1094
int as_area_get_flags(as_area_t *a)
1235 jermar 1095
{
1096
    return area_flags_to_page_flags(a->flags);
1097
}
1098
 
756 jermar 1099
/** Create page table.
1100
 *
1101
 * Depending on architecture, create either address space
1102
 * private or global page table.
1103
 *
1104
 * @param flags Flags saying whether the page table is for kernel address space.
1105
 *
1106
 * @return First entry of the page table.
1107
 */
1108
pte_t *page_table_create(int flags)
1109
{
2125 decky 1110
#ifdef __OBJC__
1111
    return [as_t page_table_create: flags];
1112
#else
1113
    ASSERT(as_operations);
1114
    ASSERT(as_operations->page_table_create);
1115
 
1116
    return as_operations->page_table_create(flags);
1117
#endif
756 jermar 1118
}
977 jermar 1119
 
1468 jermar 1120
/** Destroy page table.
1121
 *
1122
 * Destroy page table in architecture specific way.
1123
 *
1124
 * @param page_table Physical address of PTL0.
1125
 */
1126
void page_table_destroy(pte_t *page_table)
1127
{
2125 decky 1128
#ifdef __OBJC__
1129
    return [as_t page_table_destroy: page_table];
1130
#else
1131
    ASSERT(as_operations);
1132
    ASSERT(as_operations->page_table_destroy);
1133
 
1134
    as_operations->page_table_destroy(page_table);
1135
#endif
1468 jermar 1136
}
1137
 
1044 jermar 1138
/** Lock page table.
1139
 *
1140
 * This function should be called before any page_mapping_insert(),
1141
 * page_mapping_remove() and page_mapping_find().
1142
 *
1143
 * Locking order is such that address space areas must be locked
1144
 * prior to this call. Address space can be locked prior to this
1145
 * call in which case the lock argument is false.
1146
 *
1147
 * @param as Address space.
1248 jermar 1148
 * @param lock If false, do not attempt to lock as->lock.
1044 jermar 1149
 */
1150
void page_table_lock(as_t *as, bool lock)
1151
{
2125 decky 1152
#ifdef __OBJC__
1153
    [as page_table_lock: lock];
1154
#else
1044 jermar 1155
    ASSERT(as_operations);
1156
    ASSERT(as_operations->page_table_lock);
2125 decky 1157
 
1044 jermar 1158
    as_operations->page_table_lock(as, lock);
2125 decky 1159
#endif
1044 jermar 1160
}
1161
 
1162
/** Unlock page table.
1163
 *
1164
 * @param as Address space.
1248 jermar 1165
 * @param unlock If false, do not attempt to unlock as->lock.
1044 jermar 1166
 */
1167
void page_table_unlock(as_t *as, bool unlock)
1168
{
2125 decky 1169
#ifdef __OBJC__
1170
    [as page_table_unlock: unlock];
1171
#else
1044 jermar 1172
    ASSERT(as_operations);
1173
    ASSERT(as_operations->page_table_unlock);
2125 decky 1174
 
1044 jermar 1175
    as_operations->page_table_unlock(as, unlock);
2125 decky 1176
#endif
1044 jermar 1177
}
1178
 
977 jermar 1179
 
1180
/** Find address space area and lock it.
1181
 *
1182
 * The address space must be locked and interrupts must be disabled.
1183
 *
1184
 * @param as Address space.
1185
 * @param va Virtual address.
1186
 *
2087 jermar 1187
 * @return Locked address space area containing va on success or NULL on
1188
 *     failure.
977 jermar 1189
 */
1780 jermar 1190
as_area_t *find_area_and_lock(as_t *as, uintptr_t va)
977 jermar 1191
{
1192
    as_area_t *a;
1147 jermar 1193
    btree_node_t *leaf, *lnode;
2745 decky 1194
    unsigned int i;
977 jermar 1195
 
1147 jermar 1196
    a = (as_area_t *) btree_search(&as->as_area_btree, va, &leaf);
1197
    if (a) {
1198
        /* va is the base address of an address space area */
1380 jermar 1199
        mutex_lock(&a->lock);
1147 jermar 1200
        return a;
1201
    }
1202
 
1203
    /*
1150 jermar 1204
     * Search the leaf node and the righmost record of its left neighbour
1147 jermar 1205
     * to find out whether this is a miss or va belongs to an address
1206
     * space area found there.
1207
     */
1208
 
1209
    /* First, search the leaf node itself. */
1210
    for (i = 0; i < leaf->keys; i++) {
1211
        a = (as_area_t *) leaf->value[i];
1380 jermar 1212
        mutex_lock(&a->lock);
1147 jermar 1213
        if ((a->base <= va) && (va < a->base + a->pages * PAGE_SIZE)) {
1214
            return a;
1215
        }
1380 jermar 1216
        mutex_unlock(&a->lock);
1147 jermar 1217
    }
977 jermar 1218
 
1147 jermar 1219
    /*
1150 jermar 1220
     * Second, locate the left neighbour and test its last record.
1148 jermar 1221
     * Because of its position in the B+tree, it must have base < va.
1147 jermar 1222
     */
2087 jermar 1223
    lnode = btree_leaf_node_left_neighbour(&as->as_area_btree, leaf);
1224
    if (lnode) {
1147 jermar 1225
        a = (as_area_t *) lnode->value[lnode->keys - 1];
1380 jermar 1226
        mutex_lock(&a->lock);
1147 jermar 1227
        if (va < a->base + a->pages * PAGE_SIZE) {
1048 jermar 1228
            return a;
1147 jermar 1229
        }
1380 jermar 1230
        mutex_unlock(&a->lock);
977 jermar 1231
    }
1232
 
1233
    return NULL;
1234
}
1048 jermar 1235
 
1236
/** Check area conflicts with other areas.
1237
 *
1238
 * The address space must be locked and interrupts must be disabled.
1239
 *
1240
 * @param as Address space.
1241
 * @param va Starting virtual address of the area being tested.
1242
 * @param size Size of the area being tested.
1243
 * @param avoid_area Do not touch this area.
1244
 *
1245
 * @return True if there is no conflict, false otherwise.
1246
 */
2087 jermar 1247
bool check_area_conflicts(as_t *as, uintptr_t va, size_t size,
1248
              as_area_t *avoid_area)
1048 jermar 1249
{
1250
    as_area_t *a;
1147 jermar 1251
    btree_node_t *leaf, *node;
2745 decky 1252
    unsigned int i;
1048 jermar 1253
 
1070 jermar 1254
    /*
1255
     * We don't want any area to have conflicts with NULL page.
1256
     */
1257
    if (overlaps(va, size, NULL, PAGE_SIZE))
1258
        return false;
1259
 
1147 jermar 1260
    /*
1261
     * The leaf node is found in O(log n), where n is proportional to
1262
     * the number of address space areas belonging to as.
1263
     * The check for conflicts is then attempted on the rightmost
1150 jermar 1264
     * record in the left neighbour, the leftmost record in the right
1265
     * neighbour and all records in the leaf node itself.
1147 jermar 1266
     */
1048 jermar 1267
 
1147 jermar 1268
    if ((a = (as_area_t *) btree_search(&as->as_area_btree, va, &leaf))) {
1269
        if (a != avoid_area)
1270
            return false;
1271
    }
1272
 
1273
    /* First, check the two border cases. */
1150 jermar 1274
    if ((node = btree_leaf_node_left_neighbour(&as->as_area_btree, leaf))) {
1147 jermar 1275
        a = (as_area_t *) node->value[node->keys - 1];
1380 jermar 1276
        mutex_lock(&a->lock);
1147 jermar 1277
        if (overlaps(va, size, a->base, a->pages * PAGE_SIZE)) {
1380 jermar 1278
            mutex_unlock(&a->lock);
1147 jermar 1279
            return false;
1280
        }
1380 jermar 1281
        mutex_unlock(&a->lock);
1147 jermar 1282
    }
2087 jermar 1283
    node = btree_leaf_node_right_neighbour(&as->as_area_btree, leaf);
1284
    if (node) {
1147 jermar 1285
        a = (as_area_t *) node->value[0];
1380 jermar 1286
        mutex_lock(&a->lock);
1147 jermar 1287
        if (overlaps(va, size, a->base, a->pages * PAGE_SIZE)) {
1380 jermar 1288
            mutex_unlock(&a->lock);
1147 jermar 1289
            return false;
1290
        }
1380 jermar 1291
        mutex_unlock(&a->lock);
1147 jermar 1292
    }
1293
 
1294
    /* Second, check the leaf node. */
1295
    for (i = 0; i < leaf->keys; i++) {
1296
        a = (as_area_t *) leaf->value[i];
1297
 
1048 jermar 1298
        if (a == avoid_area)
1299
            continue;
1147 jermar 1300
 
1380 jermar 1301
        mutex_lock(&a->lock);
1147 jermar 1302
        if (overlaps(va, size, a->base, a->pages * PAGE_SIZE)) {
1380 jermar 1303
            mutex_unlock(&a->lock);
1147 jermar 1304
            return false;
1305
        }
1380 jermar 1306
        mutex_unlock(&a->lock);
1048 jermar 1307
    }
1308
 
1070 jermar 1309
    /*
1310
     * So far, the area does not conflict with other areas.
1311
     * Check if it doesn't conflict with kernel address space.
1312
     */  
1313
    if (!KERNEL_ADDRESS_SPACE_SHADOWED) {
1314
        return !overlaps(va, size,
2087 jermar 1315
            KERNEL_ADDRESS_SPACE_START,
1316
            KERNEL_ADDRESS_SPACE_END - KERNEL_ADDRESS_SPACE_START);
1070 jermar 1317
    }
1318
 
1048 jermar 1319
    return true;
1320
}
1235 jermar 1321
 
2556 jermar 1322
/** Return size of the address space area with given base.
1323
 *
1324
 * @param base      Arbitrary address insede the address space area.
1325
 *
1326
 * @return      Size of the address space area in bytes or zero if it
1327
 *          does not exist.
1328
 */
1329
size_t as_area_get_size(uintptr_t base)
1329 palkovsky 1330
{
1331
    ipl_t ipl;
1332
    as_area_t *src_area;
1333
    size_t size;
1334
 
1335
    ipl = interrupts_disable();
1336
    src_area = find_area_and_lock(AS, base);
1337
    if (src_area){
1338
        size = src_area->pages * PAGE_SIZE;
1380 jermar 1339
        mutex_unlock(&src_area->lock);
1329 palkovsky 1340
    } else {
1341
        size = 0;
1342
    }
1343
    interrupts_restore(ipl);
1344
    return size;
1345
}
1346
 
1387 jermar 1347
/** Mark portion of address space area as used.
1348
 *
1349
 * The address space area must be already locked.
1350
 *
1351
 * @param a Address space area.
1352
 * @param page First page to be marked.
1353
 * @param count Number of page to be marked.
1354
 *
1355
 * @return 0 on failure and 1 on success.
1356
 */
1780 jermar 1357
int used_space_insert(as_area_t *a, uintptr_t page, count_t count)
1387 jermar 1358
{
1359
    btree_node_t *leaf, *node;
1360
    count_t pages;
2745 decky 1361
    unsigned int i;
1387 jermar 1362
 
1363
    ASSERT(page == ALIGN_DOWN(page, PAGE_SIZE));
1364
    ASSERT(count);
1365
 
1366
    pages = (count_t) btree_search(&a->used_space, page, &leaf);
1367
    if (pages) {
1368
        /*
1369
         * We hit the beginning of some used space.
1370
         */
1371
        return 0;
1372
    }
1373
 
1437 jermar 1374
    if (!leaf->keys) {
1375
        btree_insert(&a->used_space, page, (void *) count, leaf);
1376
        return 1;
1377
    }
1378
 
1387 jermar 1379
    node = btree_leaf_node_left_neighbour(&a->used_space, leaf);
1380
    if (node) {
2087 jermar 1381
        uintptr_t left_pg = node->key[node->keys - 1];
1382
        uintptr_t right_pg = leaf->key[0];
1383
        count_t left_cnt = (count_t) node->value[node->keys - 1];
1384
        count_t right_cnt = (count_t) leaf->value[0];
1387 jermar 1385
 
1386
        /*
1387
         * Examine the possibility that the interval fits
1388
         * somewhere between the rightmost interval of
1389
         * the left neigbour and the first interval of the leaf.
1390
         */
1391
 
1392
        if (page >= right_pg) {
1393
            /* Do nothing. */
2087 jermar 1394
        } else if (overlaps(page, count * PAGE_SIZE, left_pg,
1395
            left_cnt * PAGE_SIZE)) {
1387 jermar 1396
            /* The interval intersects with the left interval. */
1397
            return 0;
2087 jermar 1398
        } else if (overlaps(page, count * PAGE_SIZE, right_pg,
1399
            right_cnt * PAGE_SIZE)) {
1387 jermar 1400
            /* The interval intersects with the right interval. */
1401
            return 0;          
2087 jermar 1402
        } else if ((page == left_pg + left_cnt * PAGE_SIZE) &&
1403
            (page + count * PAGE_SIZE == right_pg)) {
1404
            /*
1405
             * The interval can be added by merging the two already
1406
             * present intervals.
1407
             */
1403 jermar 1408
            node->value[node->keys - 1] += count + right_cnt;
1387 jermar 1409
            btree_remove(&a->used_space, right_pg, leaf);
1410
            return 1;
2087 jermar 1411
        } else if (page == left_pg + left_cnt * PAGE_SIZE) {
1412
            /*
1413
             * The interval can be added by simply growing the left
1414
             * interval.
1415
             */
1403 jermar 1416
            node->value[node->keys - 1] += count;
1387 jermar 1417
            return 1;
2087 jermar 1418
        } else if (page + count * PAGE_SIZE == right_pg) {
1387 jermar 1419
            /*
2087 jermar 1420
             * The interval can be addded by simply moving base of
1421
             * the right interval down and increasing its size
1422
             * accordingly.
1387 jermar 1423
             */
1403 jermar 1424
            leaf->value[0] += count;
1387 jermar 1425
            leaf->key[0] = page;
1426
            return 1;
1427
        } else {
1428
            /*
1429
             * The interval is between both neigbouring intervals,
1430
             * but cannot be merged with any of them.
1431
             */
2087 jermar 1432
            btree_insert(&a->used_space, page, (void *) count,
1433
                leaf);
1387 jermar 1434
            return 1;
1435
        }
1436
    } else if (page < leaf->key[0]) {
1780 jermar 1437
        uintptr_t right_pg = leaf->key[0];
1387 jermar 1438
        count_t right_cnt = (count_t) leaf->value[0];
1439
 
1440
        /*
2087 jermar 1441
         * Investigate the border case in which the left neighbour does
1442
         * not exist but the interval fits from the left.
1387 jermar 1443
         */
1444
 
2087 jermar 1445
        if (overlaps(page, count * PAGE_SIZE, right_pg,
1446
            right_cnt * PAGE_SIZE)) {
1387 jermar 1447
            /* The interval intersects with the right interval. */
1448
            return 0;
2087 jermar 1449
        } else if (page + count * PAGE_SIZE == right_pg) {
1387 jermar 1450
            /*
2087 jermar 1451
             * The interval can be added by moving the base of the
1452
             * right interval down and increasing its size
1453
             * accordingly.
1387 jermar 1454
             */
1455
            leaf->key[0] = page;
1403 jermar 1456
            leaf->value[0] += count;
1387 jermar 1457
            return 1;
1458
        } else {
1459
            /*
1460
             * The interval doesn't adjoin with the right interval.
1461
             * It must be added individually.
1462
             */
2087 jermar 1463
            btree_insert(&a->used_space, page, (void *) count,
1464
                leaf);
1387 jermar 1465
            return 1;
1466
        }
1467
    }
1468
 
1469
    node = btree_leaf_node_right_neighbour(&a->used_space, leaf);
1470
    if (node) {
2087 jermar 1471
        uintptr_t left_pg = leaf->key[leaf->keys - 1];
1472
        uintptr_t right_pg = node->key[0];
1473
        count_t left_cnt = (count_t) leaf->value[leaf->keys - 1];
1474
        count_t right_cnt = (count_t) node->value[0];
1387 jermar 1475
 
1476
        /*
1477
         * Examine the possibility that the interval fits
1478
         * somewhere between the leftmost interval of
1479
         * the right neigbour and the last interval of the leaf.
1480
         */
1481
 
1482
        if (page < left_pg) {
1483
            /* Do nothing. */
2087 jermar 1484
        } else if (overlaps(page, count * PAGE_SIZE, left_pg,
1485
            left_cnt * PAGE_SIZE)) {
1387 jermar 1486
            /* The interval intersects with the left interval. */
1487
            return 0;
2087 jermar 1488
        } else if (overlaps(page, count * PAGE_SIZE, right_pg,
1489
            right_cnt * PAGE_SIZE)) {
1387 jermar 1490
            /* The interval intersects with the right interval. */
1491
            return 0;          
2087 jermar 1492
        } else if ((page == left_pg + left_cnt * PAGE_SIZE) &&
1493
            (page + count * PAGE_SIZE == right_pg)) {
1494
            /*
1495
             * The interval can be added by merging the two already
1496
             * present intervals.
1497
             * */
1403 jermar 1498
            leaf->value[leaf->keys - 1] += count + right_cnt;
1387 jermar 1499
            btree_remove(&a->used_space, right_pg, node);
1500
            return 1;
2087 jermar 1501
        } else if (page == left_pg + left_cnt * PAGE_SIZE) {
1502
            /*
1503
             * The interval can be added by simply growing the left
1504
             * interval.
1505
             * */
1403 jermar 1506
            leaf->value[leaf->keys - 1] +=  count;
1387 jermar 1507
            return 1;
2087 jermar 1508
        } else if (page + count * PAGE_SIZE == right_pg) {
1387 jermar 1509
            /*
2087 jermar 1510
             * The interval can be addded by simply moving base of
1511
             * the right interval down and increasing its size
1512
             * accordingly.
1387 jermar 1513
             */
1403 jermar 1514
            node->value[0] += count;
1387 jermar 1515
            node->key[0] = page;
1516
            return 1;
1517
        } else {
1518
            /*
1519
             * The interval is between both neigbouring intervals,
1520
             * but cannot be merged with any of them.
1521
             */
2087 jermar 1522
            btree_insert(&a->used_space, page, (void *) count,
1523
                leaf);
1387 jermar 1524
            return 1;
1525
        }
1526
    } else if (page >= leaf->key[leaf->keys - 1]) {
1780 jermar 1527
        uintptr_t left_pg = leaf->key[leaf->keys - 1];
1387 jermar 1528
        count_t left_cnt = (count_t) leaf->value[leaf->keys - 1];
1529
 
1530
        /*
2087 jermar 1531
         * Investigate the border case in which the right neighbour
1532
         * does not exist but the interval fits from the right.
1387 jermar 1533
         */
1534
 
2087 jermar 1535
        if (overlaps(page, count * PAGE_SIZE, left_pg,
1536
            left_cnt * PAGE_SIZE)) {
1403 jermar 1537
            /* The interval intersects with the left interval. */
1387 jermar 1538
            return 0;
2087 jermar 1539
        } else if (left_pg + left_cnt * PAGE_SIZE == page) {
1540
            /*
1541
             * The interval can be added by growing the left
1542
             * interval.
1543
             */
1403 jermar 1544
            leaf->value[leaf->keys - 1] += count;
1387 jermar 1545
            return 1;
1546
        } else {
1547
            /*
1548
             * The interval doesn't adjoin with the left interval.
1549
             * It must be added individually.
1550
             */
2087 jermar 1551
            btree_insert(&a->used_space, page, (void *) count,
1552
                leaf);
1387 jermar 1553
            return 1;
1554
        }
1555
    }
1556
 
1557
    /*
2087 jermar 1558
     * Note that if the algorithm made it thus far, the interval can fit
1559
     * only between two other intervals of the leaf. The two border cases
1560
     * were already resolved.
1387 jermar 1561
     */
1562
    for (i = 1; i < leaf->keys; i++) {
1563
        if (page < leaf->key[i]) {
2087 jermar 1564
            uintptr_t left_pg = leaf->key[i - 1];
1565
            uintptr_t right_pg = leaf->key[i];
1566
            count_t left_cnt = (count_t) leaf->value[i - 1];
1567
            count_t right_cnt = (count_t) leaf->value[i];
1387 jermar 1568
 
1569
            /*
1570
             * The interval fits between left_pg and right_pg.
1571
             */
1572
 
2087 jermar 1573
            if (overlaps(page, count * PAGE_SIZE, left_pg,
1574
                left_cnt * PAGE_SIZE)) {
1575
                /*
1576
                 * The interval intersects with the left
1577
                 * interval.
1578
                 */
1387 jermar 1579
                return 0;
2087 jermar 1580
            } else if (overlaps(page, count * PAGE_SIZE, right_pg,
1581
                right_cnt * PAGE_SIZE)) {
1582
                /*
1583
                 * The interval intersects with the right
1584
                 * interval.
1585
                 */
1387 jermar 1586
                return 0;          
2087 jermar 1587
            } else if ((page == left_pg + left_cnt * PAGE_SIZE) &&
1588
                (page + count * PAGE_SIZE == right_pg)) {
1589
                /*
1590
                 * The interval can be added by merging the two
1591
                 * already present intervals.
1592
                 */
1403 jermar 1593
                leaf->value[i - 1] += count + right_cnt;
1387 jermar 1594
                btree_remove(&a->used_space, right_pg, leaf);
1595
                return 1;
2087 jermar 1596
            } else if (page == left_pg + left_cnt * PAGE_SIZE) {
1597
                /*
1598
                 * The interval can be added by simply growing
1599
                 * the left interval.
1600
                 */
1403 jermar 1601
                leaf->value[i - 1] += count;
1387 jermar 1602
                return 1;
2087 jermar 1603
            } else if (page + count * PAGE_SIZE == right_pg) {
1387 jermar 1604
                /*
2087 jermar 1605
                     * The interval can be addded by simply moving
1606
                 * base of the right interval down and
1607
                 * increasing its size accordingly.
1387 jermar 1608
                 */
1403 jermar 1609
                leaf->value[i] += count;
1387 jermar 1610
                leaf->key[i] = page;
1611
                return 1;
1612
            } else {
1613
                /*
2087 jermar 1614
                 * The interval is between both neigbouring
1615
                 * intervals, but cannot be merged with any of
1616
                 * them.
1387 jermar 1617
                 */
2087 jermar 1618
                btree_insert(&a->used_space, page,
1619
                    (void *) count, leaf);
1387 jermar 1620
                return 1;
1621
            }
1622
        }
1623
    }
1624
 
2087 jermar 1625
    panic("Inconsistency detected while adding %d pages of used space at "
1626
        "%p.\n", count, page);
1387 jermar 1627
}
1628
 
1629
/** Mark portion of address space area as unused.
1630
 *
1631
 * The address space area must be already locked.
1632
 *
1633
 * @param a Address space area.
1634
 * @param page First page to be marked.
1635
 * @param count Number of page to be marked.
1636
 *
1637
 * @return 0 on failure and 1 on success.
1638
 */
1780 jermar 1639
int used_space_remove(as_area_t *a, uintptr_t page, count_t count)
1387 jermar 1640
{
1641
    btree_node_t *leaf, *node;
1642
    count_t pages;
2745 decky 1643
    unsigned int i;
1387 jermar 1644
 
1645
    ASSERT(page == ALIGN_DOWN(page, PAGE_SIZE));
1646
    ASSERT(count);
1647
 
1648
    pages = (count_t) btree_search(&a->used_space, page, &leaf);
1649
    if (pages) {
1650
        /*
1651
         * We are lucky, page is the beginning of some interval.
1652
         */
1653
        if (count > pages) {
1654
            return 0;
1655
        } else if (count == pages) {
1656
            btree_remove(&a->used_space, page, leaf);
1403 jermar 1657
            return 1;
1387 jermar 1658
        } else {
1659
            /*
1660
             * Find the respective interval.
1661
             * Decrease its size and relocate its start address.
1662
             */
1663
            for (i = 0; i < leaf->keys; i++) {
1664
                if (leaf->key[i] == page) {
2087 jermar 1665
                    leaf->key[i] += count * PAGE_SIZE;
1403 jermar 1666
                    leaf->value[i] -= count;
1387 jermar 1667
                    return 1;
1668
                }
1669
            }
1670
            goto error;
1671
        }
1672
    }
1673
 
1674
    node = btree_leaf_node_left_neighbour(&a->used_space, leaf);
1675
    if (node && page < leaf->key[0]) {
1780 jermar 1676
        uintptr_t left_pg = node->key[node->keys - 1];
1387 jermar 1677
        count_t left_cnt = (count_t) node->value[node->keys - 1];
1678
 
2087 jermar 1679
        if (overlaps(left_pg, left_cnt * PAGE_SIZE, page,
1680
            count * PAGE_SIZE)) {
1681
            if (page + count * PAGE_SIZE ==
1682
                left_pg + left_cnt * PAGE_SIZE) {
1387 jermar 1683
                /*
2087 jermar 1684
                 * The interval is contained in the rightmost
1685
                 * interval of the left neighbour and can be
1686
                 * removed by updating the size of the bigger
1687
                 * interval.
1387 jermar 1688
                 */
1403 jermar 1689
                node->value[node->keys - 1] -= count;
1387 jermar 1690
                return 1;
2087 jermar 1691
            } else if (page + count * PAGE_SIZE <
1692
                left_pg + left_cnt*PAGE_SIZE) {
1403 jermar 1693
                count_t new_cnt;
1387 jermar 1694
 
1695
                /*
2087 jermar 1696
                 * The interval is contained in the rightmost
1697
                 * interval of the left neighbour but its
1698
                 * removal requires both updating the size of
1699
                 * the original interval and also inserting a
1700
                 * new interval.
1387 jermar 1701
                 */
2087 jermar 1702
                new_cnt = ((left_pg + left_cnt * PAGE_SIZE) -
1703
                    (page + count*PAGE_SIZE)) >> PAGE_WIDTH;
1403 jermar 1704
                node->value[node->keys - 1] -= count + new_cnt;
2087 jermar 1705
                btree_insert(&a->used_space, page +
1706
                    count * PAGE_SIZE, (void *) new_cnt, leaf);
1387 jermar 1707
                return 1;
1708
            }
1709
        }
1710
        return 0;
1711
    } else if (page < leaf->key[0]) {
1712
        return 0;
1713
    }
1714
 
1715
    if (page > leaf->key[leaf->keys - 1]) {
1780 jermar 1716
        uintptr_t left_pg = leaf->key[leaf->keys - 1];
1387 jermar 1717
        count_t left_cnt = (count_t) leaf->value[leaf->keys - 1];
1718
 
2087 jermar 1719
        if (overlaps(left_pg, left_cnt * PAGE_SIZE, page,
1720
            count * PAGE_SIZE)) {
1721
            if (page + count * PAGE_SIZE ==
1722
                left_pg + left_cnt * PAGE_SIZE) {
1387 jermar 1723
                /*
2087 jermar 1724
                 * The interval is contained in the rightmost
1725
                 * interval of the leaf and can be removed by
1726
                 * updating the size of the bigger interval.
1387 jermar 1727
                 */
1403 jermar 1728
                leaf->value[leaf->keys - 1] -= count;
1387 jermar 1729
                return 1;
2087 jermar 1730
            } else if (page + count * PAGE_SIZE < left_pg +
1731
                left_cnt * PAGE_SIZE) {
1403 jermar 1732
                count_t new_cnt;
1387 jermar 1733
 
1734
                /*
2087 jermar 1735
                 * The interval is contained in the rightmost
1736
                 * interval of the leaf but its removal
1737
                 * requires both updating the size of the
1738
                 * original interval and also inserting a new
1739
                 * interval.
1387 jermar 1740
                 */
2087 jermar 1741
                new_cnt = ((left_pg + left_cnt * PAGE_SIZE) -
1742
                    (page + count * PAGE_SIZE)) >> PAGE_WIDTH;
1403 jermar 1743
                leaf->value[leaf->keys - 1] -= count + new_cnt;
2087 jermar 1744
                btree_insert(&a->used_space, page +
1745
                    count * PAGE_SIZE, (void *) new_cnt, leaf);
1387 jermar 1746
                return 1;
1747
            }
1748
        }
1749
        return 0;
1750
    }  
1751
 
1752
    /*
1753
     * The border cases have been already resolved.
1754
     * Now the interval can be only between intervals of the leaf.
1755
     */
1756
    for (i = 1; i < leaf->keys - 1; i++) {
1757
        if (page < leaf->key[i]) {
1780 jermar 1758
            uintptr_t left_pg = leaf->key[i - 1];
1387 jermar 1759
            count_t left_cnt = (count_t) leaf->value[i - 1];
1760
 
1761
            /*
2087 jermar 1762
             * Now the interval is between intervals corresponding
1763
             * to (i - 1) and i.
1387 jermar 1764
             */
2087 jermar 1765
            if (overlaps(left_pg, left_cnt * PAGE_SIZE, page,
1766
                count * PAGE_SIZE)) {
1767
                if (page + count * PAGE_SIZE ==
1768
                    left_pg + left_cnt*PAGE_SIZE) {
1387 jermar 1769
                    /*
2087 jermar 1770
                     * The interval is contained in the
1771
                     * interval (i - 1) of the leaf and can
1772
                     * be removed by updating the size of
1773
                     * the bigger interval.
1387 jermar 1774
                     */
1403 jermar 1775
                    leaf->value[i - 1] -= count;
1387 jermar 1776
                    return 1;
2087 jermar 1777
                } else if (page + count * PAGE_SIZE <
1778
                    left_pg + left_cnt * PAGE_SIZE) {
1403 jermar 1779
                    count_t new_cnt;
1387 jermar 1780
 
1781
                    /*
2087 jermar 1782
                     * The interval is contained in the
1783
                     * interval (i - 1) of the leaf but its
1784
                     * removal requires both updating the
1785
                     * size of the original interval and
1387 jermar 1786
                     * also inserting a new interval.
1787
                     */
2087 jermar 1788
                    new_cnt = ((left_pg +
1789
                        left_cnt * PAGE_SIZE) -
1790
                        (page + count * PAGE_SIZE)) >>
1791
                        PAGE_WIDTH;
1403 jermar 1792
                    leaf->value[i - 1] -= count + new_cnt;
2087 jermar 1793
                    btree_insert(&a->used_space, page +
1794
                        count * PAGE_SIZE, (void *) new_cnt,
1795
                        leaf);
1387 jermar 1796
                    return 1;
1797
                }
1798
            }
1799
            return 0;
1800
        }
1801
    }
1802
 
1803
error:
2087 jermar 1804
    panic("Inconsistency detected while removing %d pages of used space "
1805
        "from %p.\n", count, page);
1387 jermar 1806
}
1807
 
1409 jermar 1808
/** Remove reference to address space area share info.
1809
 *
1810
 * If the reference count drops to 0, the sh_info is deallocated.
1811
 *
1812
 * @param sh_info Pointer to address space area share info.
1813
 */
1814
void sh_info_remove_reference(share_info_t *sh_info)
1815
{
1816
    bool dealloc = false;
1817
 
1818
    mutex_lock(&sh_info->lock);
1819
    ASSERT(sh_info->refcount);
1820
    if (--sh_info->refcount == 0) {
1821
        dealloc = true;
1495 jermar 1822
        link_t *cur;
1409 jermar 1823
 
1824
        /*
1825
         * Now walk carefully the pagemap B+tree and free/remove
1826
         * reference from all frames found there.
1827
         */
2087 jermar 1828
        for (cur = sh_info->pagemap.leaf_head.next;
1829
            cur != &sh_info->pagemap.leaf_head; cur = cur->next) {
1409 jermar 1830
            btree_node_t *node;
2745 decky 1831
            unsigned int i;
1409 jermar 1832
 
1495 jermar 1833
            node = list_get_instance(cur, btree_node_t, leaf_link);
1834
            for (i = 0; i < node->keys; i++)
1780 jermar 1835
                frame_free((uintptr_t) node->value[i]);
1409 jermar 1836
        }
1837
 
1838
    }
1839
    mutex_unlock(&sh_info->lock);
1840
 
1841
    if (dealloc) {
1842
        btree_destroy(&sh_info->pagemap);
1843
        free(sh_info);
1844
    }
1845
}
1846
 
1235 jermar 1847
/*
1848
 * Address space related syscalls.
1849
 */
1850
 
1851
/** Wrapper for as_area_create(). */
1780 jermar 1852
unative_t sys_as_area_create(uintptr_t address, size_t size, int flags)
1235 jermar 1853
{
2087 jermar 1854
    if (as_area_create(AS, flags | AS_AREA_CACHEABLE, size, address,
1855
        AS_AREA_ATTR_NONE, &anon_backend, NULL))
1780 jermar 1856
        return (unative_t) address;
1235 jermar 1857
    else
1780 jermar 1858
        return (unative_t) -1;
1235 jermar 1859
}
1860
 
1793 jermar 1861
/** Wrapper for as_area_resize(). */
1780 jermar 1862
unative_t sys_as_area_resize(uintptr_t address, size_t size, int flags)
1235 jermar 1863
{
1780 jermar 1864
    return (unative_t) as_area_resize(AS, address, size, 0);
1235 jermar 1865
}
1866
 
2984 svoboda 1867
/** Wrapper for as_area_change_flags(). */
1868
unative_t sys_as_area_change_flags(uintptr_t address, int flags)
1869
{
1870
    return (unative_t) as_area_change_flags(AS, flags, address);
1871
}
1872
 
1793 jermar 1873
/** Wrapper for as_area_destroy(). */
1780 jermar 1874
unative_t sys_as_area_destroy(uintptr_t address)
1306 jermar 1875
{
1780 jermar 1876
    return (unative_t) as_area_destroy(AS, address);
1306 jermar 1877
}
1702 cejka 1878
 
1914 jermar 1879
/** Print out information about address space.
1880
 *
1881
 * @param as Address space.
1882
 */
1883
void as_print(as_t *as)
1884
{
1885
    ipl_t ipl;
1886
 
1887
    ipl = interrupts_disable();
1888
    mutex_lock(&as->lock);
1889
 
1890
    /* print out info about address space areas */
1891
    link_t *cur;
2087 jermar 1892
    for (cur = as->as_area_btree.leaf_head.next;
1893
        cur != &as->as_area_btree.leaf_head; cur = cur->next) {
1894
        btree_node_t *node;
1914 jermar 1895
 
2087 jermar 1896
        node = list_get_instance(cur, btree_node_t, leaf_link);
1897
 
2745 decky 1898
        unsigned int i;
1914 jermar 1899
        for (i = 0; i < node->keys; i++) {
1915 jermar 1900
            as_area_t *area = node->value[i];
1914 jermar 1901
 
1902
            mutex_lock(&area->lock);
1903
            printf("as_area: %p, base=%p, pages=%d (%p - %p)\n",
2087 jermar 1904
                area, area->base, area->pages, area->base,
1905
                area->base + area->pages*PAGE_SIZE);
1914 jermar 1906
            mutex_unlock(&area->lock);
1907
        }
1908
    }
1909
 
1910
    mutex_unlock(&as->lock);
1911
    interrupts_restore(ipl);
1912
}
1913
 
1757 jermar 1914
/** @}
1702 cejka 1915
 */