88 lines
2.7 KiB
C
88 lines
2.7 KiB
C
/*
|
|
Copyright 2025 Kamil Kowalczyk
|
|
|
|
Redistribution and use in source and binary forms, with or
|
|
without modification, are permitted provided that the following
|
|
conditions are met:
|
|
|
|
1. Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
2. Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
3. Neither the name of the copyright holder nor the names of its
|
|
contributors may be used to endorse or promote products derived from
|
|
this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
“AS IS” AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
|
|
TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <libk/types.h>
|
|
#include <libk/bitmap.h>
|
|
#include <libk/string.h>
|
|
#include <libk/util.h>
|
|
#include <sync/spinlock.h>
|
|
#include <mm/pmm.h>
|
|
#include <config.h>
|
|
|
|
static struct pmm pmm;
|
|
|
|
void pmm_init(uptr_t baseptr, uptr_t bm_baseptr, usize_t block_count, usize_t block_size) {
|
|
zero(&pmm);
|
|
pmm.block_size = block_size;
|
|
pmm.baseptr = baseptr;
|
|
bitmap_init(&pmm.bm, (byte_t *)bm_baseptr, block_count);
|
|
sl_init(&pmm.sl, "pmm");
|
|
dbgf("pmm_init(): initialized PMM %zu blocks, %zu block size, total=%zu\n",
|
|
block_count, block_size, block_count * block_size);
|
|
}
|
|
|
|
uptr_t pmm_alloc(void) {
|
|
uptr_t ret = 0;
|
|
|
|
sl_lock(&pmm.sl);
|
|
|
|
for (usize_t i = 0; i < pmm.bm.bit_count; i++) {
|
|
if (!bitmap_test(&pmm.bm, i)) {
|
|
ret = (pmm.baseptr + i * pmm.block_size);
|
|
bitmap_set(&pmm.bm, i);
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
done:
|
|
sl_unlock(&pmm.sl);
|
|
return ret;
|
|
}
|
|
|
|
void pmm_free(uptr_t addr) {
|
|
uptr_t aligned_addr = ALIGN_DOWN(addr, pmm.block_size);
|
|
|
|
if (aligned_addr < pmm.baseptr)
|
|
goto done;
|
|
|
|
if (aligned_addr >= pmm.baseptr + pmm.bm.bit_count * pmm.block_size)
|
|
goto done;
|
|
|
|
usize_t bit = (usize_t)(aligned_addr - pmm.baseptr) / pmm.block_size;
|
|
|
|
sl_lock(&pmm.sl);
|
|
bitmap_clear(&pmm.bm, bit);
|
|
sl_unlock(&pmm.sl);
|
|
|
|
done:
|
|
return;
|
|
}
|