factor/vm/os-unix.cpp

367 lines
8.2 KiB
C++
Raw Normal View History

2009-05-02 05:04:19 -04:00
#include "master.hpp"
2009-05-04 02:46:13 -04:00
namespace factor
{
2009-08-25 03:55:18 -04:00
THREADHANDLE start_thread(void *(*start_routine)(void *),void *args)
2009-05-02 05:04:19 -04:00
{
pthread_attr_t attr;
pthread_t thread;
if (pthread_attr_init (&attr) != 0)
fatal_error("pthread_attr_init() failed",0);
if (pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_JOINABLE) != 0)
fatal_error("pthread_attr_setdetachstate() failed",0);
if (pthread_create (&thread, &attr, start_routine, args) != 0)
fatal_error("pthread_create() failed",0);
2009-05-02 05:04:19 -04:00
pthread_attr_destroy (&attr);
2009-08-25 03:55:18 -04:00
return thread;
2009-05-02 05:04:19 -04:00
}
2009-08-28 16:46:47 -04:00
pthread_key_t tlsKey = 0;
void init_platform_globals()
{
2009-09-25 19:08:00 -04:00
if (pthread_key_create(&tlsKey, NULL) != 0)
2009-08-28 16:46:47 -04:00
fatal_error("pthread_key_create() failed",0);
2009-08-25 03:55:18 -04:00
}
2009-09-23 14:05:46 -04:00
void register_vm_with_thread(factor_vm *vm)
2009-08-28 16:46:47 -04:00
{
pthread_setspecific(tlsKey,vm);
}
2009-09-23 14:05:46 -04:00
factor_vm *tls_vm()
2009-08-28 16:46:47 -04:00
{
2009-10-02 10:37:04 -04:00
factor_vm *vm = (factor_vm*)pthread_getspecific(tlsKey);
assert(vm != NULL);
return vm;
2009-08-28 16:46:47 -04:00
}
2009-08-25 03:55:18 -04:00
2009-05-02 05:04:19 -04:00
static void *null_dll;
2009-05-05 12:33:35 -04:00
s64 current_micros()
2009-05-02 05:04:19 -04:00
{
struct timeval t;
gettimeofday(&t,NULL);
return (s64)t.tv_sec * 1000000 + t.tv_usec;
}
2009-05-04 05:50:24 -04:00
void sleep_micros(cell usec)
2009-05-02 05:04:19 -04:00
{
usleep(usec);
}
2009-09-23 14:05:46 -04:00
void factor_vm::init_ffi()
2009-05-02 05:04:19 -04:00
{
/* NULL_DLL is "libfactor.dylib" for OS X and NULL for generic unix */
null_dll = dlopen(NULL_DLL,RTLD_LAZY);
}
2009-09-23 14:05:46 -04:00
void factor_vm::ffi_dlopen(dll *dll)
2009-05-02 05:04:19 -04:00
{
dll->dll = dlopen(alien_offset(dll->path), RTLD_LAZY);
2009-05-02 05:04:19 -04:00
}
2009-09-23 14:05:46 -04:00
void *factor_vm::ffi_dlsym(dll *dll, symbol_char *symbol)
2009-05-02 05:04:19 -04:00
{
void *handle = (dll == NULL ? null_dll : dll->dll);
return dlsym(handle,symbol);
}
2009-09-23 14:05:46 -04:00
void factor_vm::ffi_dlclose(dll *dll)
2009-05-02 05:04:19 -04:00
{
if(dlclose(dll->dll))
general_error(ERROR_FFI,false_object,false_object,NULL);
2009-05-02 05:04:19 -04:00
dll->dll = NULL;
}
void factor_vm::primitive_existsp()
2009-05-02 05:04:19 -04:00
{
struct stat sb;
char *path = (char *)(untag_check<byte_array>(dpop()) + 1);
box_boolean(stat(path,&sb) >= 0);
2009-05-02 05:04:19 -04:00
}
segment::segment(cell size_, bool executable_p)
2009-05-02 05:04:19 -04:00
{
size = size_;
2009-05-02 05:04:19 -04:00
int pagesize = getpagesize();
int prot;
if(executable_p)
prot = (PROT_READ | PROT_WRITE | PROT_EXEC);
else
prot = (PROT_READ | PROT_WRITE);
2009-05-02 05:04:19 -04:00
char *array = (char *)mmap(NULL,pagesize + size + pagesize,prot,MAP_ANON | MAP_PRIVATE,-1,0);
2009-10-07 09:33:54 -04:00
if(array == (char*)-1) out_of_memory();
2009-05-02 05:04:19 -04:00
if(mprotect(array,pagesize,PROT_NONE) == -1)
fatal_error("Cannot protect low guard page",(cell)array);
2009-05-02 05:04:19 -04:00
if(mprotect(array + pagesize + size,pagesize,PROT_NONE) == -1)
fatal_error("Cannot protect high guard page",(cell)array);
2009-05-02 05:04:19 -04:00
start = (cell)(array + pagesize);
end = start + size;
2009-05-02 05:04:19 -04:00
}
segment::~segment()
2009-05-02 05:04:19 -04:00
{
int pagesize = getpagesize();
int retval = munmap((void*)(start - pagesize),pagesize + size + pagesize);
2009-05-02 05:04:19 -04:00
if(retval)
fatal_error("Segment deallocation failed",0);
2009-05-02 05:04:19 -04:00
}
2009-09-23 14:05:46 -04:00
stack_frame *factor_vm::uap_stack_pointer(void *uap)
2009-05-02 05:04:19 -04:00
{
/* There is a race condition here, but in practice a signal
delivered during stack frame setup/teardown or while transitioning
from Factor to C is a sign of things seriously gone wrong, not just
a divide by zero or stack underflow in the listener */
if(in_code_heap_p(UAP_PROGRAM_COUNTER(uap)))
2009-05-02 05:04:19 -04:00
{
2009-05-04 05:50:24 -04:00
stack_frame *ptr = (stack_frame *)ucontext_stack_pointer(uap);
2009-05-02 05:04:19 -04:00
if(!ptr)
critical_error("Invalid uap",(cell)uap);
2009-05-02 05:04:19 -04:00
return ptr;
}
else
return NULL;
}
2009-09-23 14:05:46 -04:00
void factor_vm::memory_signal_handler(int signal, siginfo_t *siginfo, void *uap)
2009-09-04 14:23:20 -04:00
{
signal_fault_addr = (cell)siginfo->si_addr;
signal_callstack_top = uap_stack_pointer(uap);
UAP_PROGRAM_COUNTER(uap) = (cell)factor::memory_signal_handler_impl;
}
2009-05-02 05:04:19 -04:00
void memory_signal_handler(int signal, siginfo_t *siginfo, void *uap)
{
tls_vm()->memory_signal_handler(signal,siginfo,uap);
2009-09-04 14:23:20 -04:00
}
2009-09-23 14:05:46 -04:00
void factor_vm::misc_signal_handler(int signal, siginfo_t *siginfo, void *uap)
2009-09-04 14:23:20 -04:00
{
signal_number = signal;
signal_callstack_top = uap_stack_pointer(uap);
UAP_PROGRAM_COUNTER(uap) = (cell)factor::misc_signal_handler_impl;
2009-05-02 05:04:19 -04:00
}
void misc_signal_handler(int signal, siginfo_t *siginfo, void *uap)
{
tls_vm()->misc_signal_handler(signal,siginfo,uap);
2009-05-02 05:04:19 -04:00
}
2009-09-23 14:05:46 -04:00
void factor_vm::fpe_signal_handler(int signal, siginfo_t *siginfo, void *uap)
{
signal_number = signal;
signal_callstack_top = uap_stack_pointer(uap);
2009-09-13 16:50:20 -04:00
signal_fpu_status = fpu_status(uap_fpu_status(uap));
uap_clear_fpu_status(uap);
UAP_PROGRAM_COUNTER(uap) =
2009-09-13 16:50:20 -04:00
(siginfo->si_code == FPE_INTDIV || siginfo->si_code == FPE_INTOVF)
? (cell)factor::misc_signal_handler_impl
: (cell)factor::fp_signal_handler_impl;
}
void fpe_signal_handler(int signal, siginfo_t *siginfo, void *uap)
{
tls_vm()->fpe_signal_handler(signal, siginfo, uap);
}
2009-05-02 05:04:19 -04:00
static void sigaction_safe(int signum, const struct sigaction *act, struct sigaction *oldact)
{
int ret;
do
{
ret = sigaction(signum, act, oldact);
}
while(ret == -1 && errno == EINTR);
if(ret == -1)
fatal_error("sigaction failed", 0);
2009-05-02 05:04:19 -04:00
}
2009-05-05 12:33:35 -04:00
void unix_init_signals()
2009-05-02 05:04:19 -04:00
{
struct sigaction memory_sigaction;
struct sigaction misc_sigaction;
struct sigaction fpe_sigaction;
2009-05-02 05:04:19 -04:00
struct sigaction ignore_sigaction;
memset(&memory_sigaction,0,sizeof(struct sigaction));
sigemptyset(&memory_sigaction.sa_mask);
memory_sigaction.sa_sigaction = memory_signal_handler;
memory_sigaction.sa_flags = SA_SIGINFO;
sigaction_safe(SIGBUS,&memory_sigaction,NULL);
sigaction_safe(SIGSEGV,&memory_sigaction,NULL);
memset(&fpe_sigaction,0,sizeof(struct sigaction));
sigemptyset(&fpe_sigaction.sa_mask);
fpe_sigaction.sa_sigaction = fpe_signal_handler;
fpe_sigaction.sa_flags = SA_SIGINFO;
sigaction_safe(SIGFPE,&fpe_sigaction,NULL);
2009-05-02 05:04:19 -04:00
memset(&misc_sigaction,0,sizeof(struct sigaction));
sigemptyset(&misc_sigaction.sa_mask);
misc_sigaction.sa_sigaction = misc_signal_handler;
misc_sigaction.sa_flags = SA_SIGINFO;
sigaction_safe(SIGQUIT,&misc_sigaction,NULL);
sigaction_safe(SIGILL,&misc_sigaction,NULL);
memset(&ignore_sigaction,0,sizeof(struct sigaction));
sigemptyset(&ignore_sigaction.sa_mask);
ignore_sigaction.sa_handler = SIG_IGN;
sigaction_safe(SIGPIPE,&ignore_sigaction,NULL);
}
/* On Unix, shared fds such as stdin cannot be set to non-blocking mode
(http://homepages.tesco.net/J.deBoynePollard/FGA/dont-set-shared-file-descriptors-to-non-blocking-mode.html)
so we kludge around this by spawning a thread, which waits on a control pipe
for a signal, upon receiving this signal it reads one block of data from stdin
and writes it to a data pipe. Upon completion, it writes a 4-byte integer to
the size pipe, indicating how much data was written to the data pipe.
The read end of the size pipe can be set to non-blocking. */
extern "C" {
int stdin_read;
int stdin_write;
int control_read;
int control_write;
int size_read;
int size_write;
}
void safe_close(int fd)
{
if(close(fd) < 0)
fatal_error("error closing fd",errno);
2009-05-02 05:04:19 -04:00
}
bool check_write(int fd, void *data, ssize_t size)
{
if(write(fd,data,size) == size)
return true;
else
{
if(errno == EINTR)
return check_write(fd,data,size);
else
return false;
}
}
void safe_write(int fd, void *data, ssize_t size)
{
if(!check_write(fd,data,size))
fatal_error("error writing fd",errno);
2009-05-02 05:04:19 -04:00
}
bool safe_read(int fd, void *data, ssize_t size)
{
ssize_t bytes = read(fd,data,size);
if(bytes < 0)
{
if(errno == EINTR)
return safe_read(fd,data,size);
else
{
fatal_error("error reading fd",errno);
2009-05-02 05:04:19 -04:00
return false;
}
}
else
return (bytes == size);
}
void *stdin_loop(void *arg)
{
unsigned char buf[4096];
bool loop_running = true;
while(loop_running)
{
if(!safe_read(control_read,buf,1))
break;
if(buf[0] != 'X')
fatal_error("stdin_loop: bad data on control fd",buf[0]);
2009-05-02 05:04:19 -04:00
for(;;)
{
ssize_t bytes = read(0,buf,sizeof(buf));
if(bytes < 0)
{
if(errno == EINTR)
continue;
else
{
loop_running = false;
break;
}
}
else if(bytes >= 0)
{
safe_write(size_write,&bytes,sizeof(bytes));
if(!check_write(stdin_write,buf,bytes))
loop_running = false;
break;
}
}
}
safe_close(stdin_write);
safe_close(control_read);
return NULL;
}
2009-05-05 12:33:35 -04:00
void open_console()
2009-05-02 05:04:19 -04:00
{
int filedes[2];
if(pipe(filedes) < 0)
fatal_error("Error opening control pipe",errno);
2009-05-02 05:04:19 -04:00
control_read = filedes[0];
control_write = filedes[1];
if(pipe(filedes) < 0)
fatal_error("Error opening size pipe",errno);
2009-05-02 05:04:19 -04:00
size_read = filedes[0];
size_write = filedes[1];
if(pipe(filedes) < 0)
fatal_error("Error opening stdin pipe",errno);
2009-05-02 05:04:19 -04:00
stdin_read = filedes[0];
stdin_write = filedes[1];
start_thread(stdin_loop,NULL);
2009-05-02 05:04:19 -04:00
}
2009-05-05 12:33:35 -04:00
VM_C_API void wait_for_stdin()
2009-05-02 05:04:19 -04:00
{
if(write(control_write,"X",1) != 1)
{
if(errno == EINTR)
wait_for_stdin();
else
fatal_error("Error writing control fd",errno);
2009-05-02 05:04:19 -04:00
}
}
2009-05-04 02:46:13 -04:00
}