138 lines
3.7 KiB
C
138 lines
3.7 KiB
C
|
/*
|
||
|
* Copyright (c) 2010 Wayne Meissner
|
||
|
*
|
||
|
* Copyright (c) 2008-2013, Ruby FFI project contributors
|
||
|
* All rights reserved.
|
||
|
*
|
||
|
* Redistribution and use in source and binary forms, with or without
|
||
|
* modification, are permitted provided that the following conditions are met:
|
||
|
* * Redistributions of source code must retain the above copyright
|
||
|
* notice, this list of conditions and the following disclaimer.
|
||
|
* * 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.
|
||
|
* * Neither the name of the Ruby FFI project 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 <COPYRIGHT HOLDER> 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 <stddef.h>
|
||
|
#ifndef _MSC_VER
|
||
|
#include <stdbool.h>
|
||
|
#else
|
||
|
# include "win32/stdbool.h"
|
||
|
# include "win32/stdint.h"
|
||
|
#endif
|
||
|
|
||
|
#if defined(__CYGWIN__) || !defined(_WIN32)
|
||
|
# include <pthread.h>
|
||
|
# include <errno.h>
|
||
|
# include <signal.h>
|
||
|
# include <unistd.h>
|
||
|
#else
|
||
|
# include <winsock2.h>
|
||
|
# define _WINSOCKAPI_
|
||
|
# include <windows.h>
|
||
|
#endif
|
||
|
#include <fcntl.h>
|
||
|
#include "Thread.h"
|
||
|
|
||
|
#ifdef _WIN32
|
||
|
static volatile DWORD frame_thread_key = TLS_OUT_OF_INDEXES;
|
||
|
#else
|
||
|
static pthread_key_t thread_data_key;
|
||
|
struct thread_data {
|
||
|
rbffi_frame_t* frame;
|
||
|
};
|
||
|
static inline struct thread_data* thread_data_get(void);
|
||
|
|
||
|
#endif
|
||
|
|
||
|
rbffi_frame_t*
|
||
|
rbffi_frame_current(void)
|
||
|
{
|
||
|
#ifdef _WIN32
|
||
|
return (rbffi_frame_t *) TlsGetValue(frame_thread_key);
|
||
|
#else
|
||
|
struct thread_data* td = (struct thread_data *) pthread_getspecific(thread_data_key);
|
||
|
return td != NULL ? td->frame : NULL;
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
void
|
||
|
rbffi_frame_push(rbffi_frame_t* frame)
|
||
|
{
|
||
|
memset(frame, 0, sizeof(*frame));
|
||
|
#ifndef HAVE_RUBY_THREAD_HAS_GVL_P
|
||
|
frame->has_gvl = true;
|
||
|
#endif
|
||
|
frame->exc = Qnil;
|
||
|
|
||
|
#ifdef _WIN32
|
||
|
frame->prev = TlsGetValue(frame_thread_key);
|
||
|
TlsSetValue(frame_thread_key, frame);
|
||
|
#else
|
||
|
frame->td = thread_data_get();
|
||
|
frame->prev = frame->td->frame;
|
||
|
frame->td->frame = frame;
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
void
|
||
|
rbffi_frame_pop(rbffi_frame_t* frame)
|
||
|
{
|
||
|
#ifdef _WIN32
|
||
|
TlsSetValue(frame_thread_key, frame->prev);
|
||
|
#else
|
||
|
frame->td->frame = frame->prev;
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
#ifndef _WIN32
|
||
|
static struct thread_data* thread_data_init(void);
|
||
|
|
||
|
static inline struct thread_data*
|
||
|
thread_data_get(void)
|
||
|
{
|
||
|
struct thread_data* td = (struct thread_data *) pthread_getspecific(thread_data_key);
|
||
|
return td != NULL ? td : thread_data_init();
|
||
|
}
|
||
|
|
||
|
static struct thread_data*
|
||
|
thread_data_init(void)
|
||
|
{
|
||
|
struct thread_data* td = calloc(1, sizeof(struct thread_data));
|
||
|
|
||
|
pthread_setspecific(thread_data_key, td);
|
||
|
|
||
|
return td;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
thread_data_free(void *ptr)
|
||
|
{
|
||
|
free(ptr);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
void
|
||
|
rbffi_Thread_Init(VALUE moduleFFI)
|
||
|
{
|
||
|
#ifdef _WIN32
|
||
|
frame_thread_key = TlsAlloc();
|
||
|
#else
|
||
|
pthread_key_create(&thread_data_key, thread_data_free);
|
||
|
#endif
|
||
|
}
|