Annotation of embedaddon/pcre/sljit/sljitUtils.c, revision 1.1
1.1 ! misho 1: /*
! 2: * Stack-less Just-In-Time compiler
! 3: *
! 4: * Copyright 2009-2010 Zoltan Herczeg (hzmester@freemail.hu). All rights reserved.
! 5: *
! 6: * Redistribution and use in source and binary forms, with or without modification, are
! 7: * permitted provided that the following conditions are met:
! 8: *
! 9: * 1. Redistributions of source code must retain the above copyright notice, this list of
! 10: * conditions and the following disclaimer.
! 11: *
! 12: * 2. Redistributions in binary form must reproduce the above copyright notice, this list
! 13: * of conditions and the following disclaimer in the documentation and/or other materials
! 14: * provided with the distribution.
! 15: *
! 16: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) AND CONTRIBUTORS ``AS IS'' AND ANY
! 17: * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
! 18: * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
! 19: * SHALL THE COPYRIGHT HOLDER(S) OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
! 20: * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
! 21: * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
! 22: * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
! 23: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
! 24: * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
! 25: */
! 26:
! 27: /* ------------------------------------------------------------------------ */
! 28: /* Locks */
! 29: /* ------------------------------------------------------------------------ */
! 30:
! 31: #if (defined SLJIT_EXECUTABLE_ALLOCATOR && SLJIT_EXECUTABLE_ALLOCATOR) || (defined SLJIT_UTIL_GLOBAL_LOCK && SLJIT_UTIL_GLOBAL_LOCK)
! 32:
! 33: #ifdef _WIN32
! 34:
! 35: #include "windows.h"
! 36:
! 37: #if (defined SLJIT_EXECUTABLE_ALLOCATOR && SLJIT_EXECUTABLE_ALLOCATOR)
! 38:
! 39: static HANDLE allocator_mutex = 0;
! 40:
! 41: static SLJIT_INLINE void allocator_grab_lock(void)
! 42: {
! 43: /* No idea what to do if an error occures. Static mutexes should never fail... */
! 44: if (!allocator_mutex)
! 45: allocator_mutex = CreateMutex(NULL, TRUE, NULL);
! 46: else
! 47: WaitForSingleObject(allocator_mutex, INFINITE);
! 48: }
! 49:
! 50: static SLJIT_INLINE void allocator_release_lock(void)
! 51: {
! 52: ReleaseMutex(allocator_mutex);
! 53: }
! 54:
! 55: #endif /* SLJIT_EXECUTABLE_ALLOCATOR */
! 56:
! 57: #if (defined SLJIT_UTIL_GLOBAL_LOCK && SLJIT_UTIL_GLOBAL_LOCK)
! 58:
! 59: static HANDLE global_mutex = 0;
! 60:
! 61: SLJIT_API_FUNC_ATTRIBUTE void SLJIT_CALL sljit_grab_lock(void)
! 62: {
! 63: /* No idea what to do if an error occures. Static mutexes should never fail... */
! 64: if (!global_mutex)
! 65: global_mutex = CreateMutex(NULL, TRUE, NULL);
! 66: else
! 67: WaitForSingleObject(global_mutex, INFINITE);
! 68: }
! 69:
! 70: SLJIT_API_FUNC_ATTRIBUTE void SLJIT_CALL sljit_release_lock(void)
! 71: {
! 72: ReleaseMutex(global_mutex);
! 73: }
! 74:
! 75: #endif /* SLJIT_UTIL_GLOBAL_LOCK */
! 76:
! 77: #else /* _WIN32 */
! 78:
! 79: #include "pthread.h"
! 80:
! 81: #if (defined SLJIT_EXECUTABLE_ALLOCATOR && SLJIT_EXECUTABLE_ALLOCATOR)
! 82:
! 83: static pthread_mutex_t allocator_mutex = PTHREAD_MUTEX_INITIALIZER;
! 84:
! 85: static SLJIT_INLINE void allocator_grab_lock(void)
! 86: {
! 87: pthread_mutex_lock(&allocator_mutex);
! 88: }
! 89:
! 90: static SLJIT_INLINE void allocator_release_lock(void)
! 91: {
! 92: pthread_mutex_unlock(&allocator_mutex);
! 93: }
! 94:
! 95: #endif /* SLJIT_EXECUTABLE_ALLOCATOR */
! 96:
! 97: #if (defined SLJIT_UTIL_GLOBAL_LOCK && SLJIT_UTIL_GLOBAL_LOCK)
! 98:
! 99: static pthread_mutex_t global_mutex = PTHREAD_MUTEX_INITIALIZER;
! 100:
! 101: SLJIT_API_FUNC_ATTRIBUTE void SLJIT_CALL sljit_grab_lock(void)
! 102: {
! 103: pthread_mutex_lock(&global_mutex);
! 104: }
! 105:
! 106: SLJIT_API_FUNC_ATTRIBUTE void SLJIT_CALL sljit_release_lock(void)
! 107: {
! 108: pthread_mutex_unlock(&global_mutex);
! 109: }
! 110:
! 111: #endif /* SLJIT_UTIL_GLOBAL_LOCK */
! 112:
! 113: #endif /* _WIN32 */
! 114:
! 115: /* ------------------------------------------------------------------------ */
! 116: /* Stack */
! 117: /* ------------------------------------------------------------------------ */
! 118:
! 119: #if (defined SLJIT_UTIL_STACK && SLJIT_UTIL_STACK)
! 120:
! 121: #ifdef _WIN32
! 122: #include "windows.h"
! 123: #else
! 124: #include <sys/mman.h>
! 125: #include <unistd.h>
! 126: #endif
! 127:
! 128: /* Planning to make it even more clever in the future. */
! 129: static sljit_w sljit_page_align = 0;
! 130:
! 131: SLJIT_API_FUNC_ATTRIBUTE struct sljit_stack* SLJIT_CALL sljit_allocate_stack(sljit_uw limit, sljit_uw max_limit)
! 132: {
! 133: struct sljit_stack *stack;
! 134: union {
! 135: void *ptr;
! 136: sljit_uw uw;
! 137: } base;
! 138: #ifdef _WIN32
! 139: SYSTEM_INFO si;
! 140: #endif
! 141:
! 142: if (limit > max_limit || limit < 1)
! 143: return NULL;
! 144:
! 145: #ifdef _WIN32
! 146: if (!sljit_page_align) {
! 147: GetSystemInfo(&si);
! 148: sljit_page_align = si.dwPageSize - 1;
! 149: }
! 150: #else
! 151: if (!sljit_page_align) {
! 152: sljit_page_align = sysconf(_SC_PAGESIZE);
! 153: /* Should never happen. */
! 154: if (sljit_page_align < 0)
! 155: sljit_page_align = 4096;
! 156: sljit_page_align--;
! 157: }
! 158: #endif
! 159:
! 160: /* Align limit and max_limit. */
! 161: max_limit = (max_limit + sljit_page_align) & ~sljit_page_align;
! 162:
! 163: stack = (struct sljit_stack*)SLJIT_MALLOC(sizeof(struct sljit_stack));
! 164: if (!stack)
! 165: return NULL;
! 166:
! 167: #ifdef _WIN32
! 168: base.ptr = VirtualAlloc(0, max_limit, MEM_RESERVE, PAGE_READWRITE);
! 169: if (!base.ptr) {
! 170: SLJIT_FREE(stack);
! 171: return NULL;
! 172: }
! 173: stack->base = base.uw;
! 174: stack->limit = stack->base;
! 175: stack->max_limit = stack->base + max_limit;
! 176: if (sljit_stack_resize(stack, stack->base + limit)) {
! 177: sljit_free_stack(stack);
! 178: return NULL;
! 179: }
! 180: #else
! 181: base.ptr = mmap(0, max_limit, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANON, -1, 0);
! 182: if (base.ptr == MAP_FAILED) {
! 183: SLJIT_FREE(stack);
! 184: return NULL;
! 185: }
! 186: stack->base = base.uw;
! 187: stack->limit = stack->base + limit;
! 188: stack->max_limit = stack->base + max_limit;
! 189: #endif
! 190: stack->top = stack->base;
! 191: return stack;
! 192: }
! 193:
! 194: #undef PAGE_ALIGN
! 195:
! 196: SLJIT_API_FUNC_ATTRIBUTE void SLJIT_CALL sljit_free_stack(struct sljit_stack* stack)
! 197: {
! 198: #ifdef _WIN32
! 199: VirtualFree((void*)stack->base, 0, MEM_RELEASE);
! 200: #else
! 201: munmap((void*)stack->base, stack->max_limit - stack->base);
! 202: #endif
! 203: SLJIT_FREE(stack);
! 204: }
! 205:
! 206: SLJIT_API_FUNC_ATTRIBUTE sljit_w SLJIT_CALL sljit_stack_resize(struct sljit_stack* stack, sljit_uw new_limit)
! 207: {
! 208: sljit_uw aligned_old_limit;
! 209: sljit_uw aligned_new_limit;
! 210:
! 211: if ((new_limit > stack->max_limit) || (new_limit < stack->base))
! 212: return -1;
! 213: #ifdef _WIN32
! 214: aligned_new_limit = (new_limit + sljit_page_align) & ~sljit_page_align;
! 215: aligned_old_limit = (stack->limit + sljit_page_align) & ~sljit_page_align;
! 216: if (aligned_new_limit != aligned_old_limit) {
! 217: if (aligned_new_limit > aligned_old_limit) {
! 218: if (!VirtualAlloc((void*)aligned_old_limit, aligned_new_limit - aligned_old_limit, MEM_COMMIT, PAGE_READWRITE))
! 219: return -1;
! 220: }
! 221: else {
! 222: if (!VirtualFree((void*)aligned_new_limit, aligned_old_limit - aligned_new_limit, MEM_DECOMMIT))
! 223: return -1;
! 224: }
! 225: }
! 226: stack->limit = new_limit;
! 227: return 0;
! 228: #else
! 229: if (new_limit >= stack->limit) {
! 230: stack->limit = new_limit;
! 231: return 0;
! 232: }
! 233: aligned_new_limit = (new_limit + sljit_page_align) & ~sljit_page_align;
! 234: aligned_old_limit = (stack->limit + sljit_page_align) & ~sljit_page_align;
! 235: if (aligned_new_limit < aligned_old_limit)
! 236: madvise((void*)aligned_new_limit, aligned_old_limit - aligned_new_limit, MADV_DONTNEED);
! 237: stack->limit = new_limit;
! 238: return 0;
! 239: #endif
! 240: }
! 241:
! 242: #endif /* SLJIT_UTIL_STACK */
! 243:
! 244: #endif
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>