1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- /* Copyright (C) 2014 Stony Brook University
- This file is part of Graphene Library OS.
- Graphene Library OS is free software: you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public License
- as published by the Free Software Foundation, either version 3 of the
- License, or (at your option) any later version.
- Graphene Library OS is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>. */
- #include <api.h>
- #include <pal_error.h>
- #include <pal_internal.h>
- #include <pal_security.h>
- #include <spinlock.h>
- #include "enclave_ocalls.h"
- static spinlock_t malloc_lock = INIT_SPINLOCK_UNLOCKED;
- static size_t g_page_size = PRESET_PAGESIZE;
- #define SYSTEM_LOCK() spinlock_lock(&malloc_lock)
- #define SYSTEM_UNLOCK() spinlock_unlock(&malloc_lock)
- #define SYSTEM_LOCKED() spinlock_is_locked(&malloc_lock)
- #define ALLOC_ALIGNMENT g_page_size
- static inline void* __malloc(int size) {
- void* addr = NULL;
- ocall_mmap_untrusted(-1, 0, size, PROT_READ | PROT_WRITE, &addr);
- return addr;
- }
- #define system_malloc(size) __malloc(size)
- static inline void __free(void* addr, int size) {
- ocall_munmap_untrusted(addr, size);
- }
- #define system_free(addr, size) __free(addr, size)
- #include "slabmgr.h"
- static SLAB_MGR untrusted_slabmgr = NULL;
- void init_untrusted_slab_mgr() {
- if (untrusted_slabmgr)
- return;
- untrusted_slabmgr = create_slab_mgr();
- if (!untrusted_slabmgr)
- INIT_FAIL(PAL_ERROR_NOMEM, "cannot initialize slab manager");
- }
- void* malloc_untrusted(int size) {
- return slab_alloc(untrusted_slabmgr, size);
- }
- void free_untrusted(void* ptr) {
- slab_free(untrusted_slabmgr, ptr);
- }
|