/* -*- Mode: C; c-basic-offset: 2; indent-tabs-mode: nil -*- */
/* vim: set filetype=c tabstop=2 shiftwidth=2 expandtab: */

/*
 * thread.d - native threads
 *
 * Copyright (c) 2003 Juan Jose Garcia Ripoll
 *
 * See file 'LICENSE' for the copyright details.
 *
 */

#define ECL_INCLUDE_MATH_H
#include <ecl/ecl.h>            /* includes ECL_WINDOWS_THREADS */
#include <ecl/internal.h>
#include <ecl/ecl-inl.h>

#include <errno.h>
#include <time.h>
#include <signal.h>
#include <string.h>
#ifdef ECL_WINDOWS_THREADS
# include <windows.h>
#else
# include <pthread.h>
#endif
#ifdef HAVE_GETTIMEOFDAY
# include <sys/time.h>
#endif
#ifdef HAVE_SCHED_H
# include <sched.h>
#endif

/* -- Core ---------------------------------------------------------- */

static cl_object
ecl_process_list()
{
  cl_env_ptr the_env = ecl_process_env();
  cl_object output = ECL_NIL;
  ECL_WITH_NATIVE_LOCK_BEGIN(the_env, &ecl_core.processes_lock) {
    loop_across_stack_fifo(_env, ecl_core.threads) {
      cl_env_ptr env = ecl_cast_ptr(cl_env_ptr, _env);
      cl_object p = env->own_process;
      output = ecl_cons(p, output);
    } end_loop_across_stack();
  } ECL_WITH_NATIVE_LOCK_END;
  return output;
}

/* -- Environment --------------------------------------------------- */

cl_object
mp_current_process(void)
{
  return ecl_process_env()->own_process;
}

/* -- Thread object ------------------------------------------------- */

static void
assert_type_process(cl_object o)
{
  if (ecl_t_of(o) != t_process)
    FEwrong_type_argument(@[mp::process], o);
}

static cl_object
run_process(cl_narg narg, ...)
{
  /* Upon entering this routine the process environment is set up, the process
   * phase is ECL_PROCESS_BOOTING, signals are disabled in the environment and
   * the communication interrupt is disabled (sigmasked).
   *
   * This process will not receive signals that originate from other processes.
   * Furthermore, we expect not to get any other interrupts (SIGSEGV, SIGFPE) if
   * we do things right.
   */
  cl_env_ptr the_env = ecl_process_env();
  cl_object process = the_env->own_process;
  cl_object fun = process->process.function;
  cl_object args = process->process.args;
  cl_object output = ECL_NIL;
  /* Entry barrier. enable_process releases this lock before exit. */
  ecl_mutex_lock(&process->process.start_stop_lock);

  /* Execute the code. The CATCH_ALL point is the destination provides us with
   * an elegant way to exit the thread: we just do an unwind up to frs_top. */
  ECL_CATCH_ALL_BEGIN(the_env) {
#ifdef HAVE_SIGPROCMASK
    {
      sigset_t *new = (sigset_t*)the_env->default_sigmask;
      ecl_sigmask(SIG_SETMASK, new, NULL);
    }
#endif
    process->process.phase = ECL_PROCESS_ACTIVE;
    ecl_mutex_unlock(&process->process.start_stop_lock);
    si_trap_fpe(@'last', ECL_T);

    ecl_enable_interrupts_env(the_env);
    ecl_bds_bind(the_env, @'mp::*current-process*', process);

    ECL_RESTART_CASE_BEGIN(the_env, @'abort') {
      the_env->values[0] = cl_apply(2, fun, args);
      int i = the_env->nvalues;
      while (i--) {
        output = CONS(the_env->values[i], output);
      }
      process->process.exit_values = output;
    } ECL_RESTART_CASE(1,args) {
      /* ABORT restart. */
      process->process.exit_values = args;
    } ECL_RESTART_CASE_END;
    /* This routine performs some cleanup before a thread is finished. Note that
       it does not provide enough protection -- in order to ensure that all
       UNWIND-PROTECT forms are properly executed, enver use the function
       pthread_cancel() to kill a process, but rather use the lisp functions
       mp_interrupt_process() and mp_process_kill(). */
    ecl_bds_unwind1(the_env);
  } ECL_CATCH_ALL_END;

  ecl_disable_interrupts_env(the_env);

  ecl_mutex_lock(&process->process.start_stop_lock);
  process->process.phase = ECL_PROCESS_EXITING;
#ifdef HAVE_SIGPROCMASK
  /* ...but we might get stray signals. */
  {
    sigset_t new[1];
    sigemptyset(new);
    sigaddset(new, ecl_option_values[ECL_OPT_THREAD_INTERRUPT_SIGNAL]);
    ecl_sigmask(SIG_BLOCK, new, NULL);
  }
#endif
  process->process.env = NULL;
  process->process.phase = ECL_PROCESS_INACTIVE;
  ecl_cond_var_broadcast(&process->process.exit_barrier);
  ecl_mutex_unlock(&process->process.start_stop_lock);

  return the_env->values[0];
}

static cl_object
alloc_process(cl_object name, cl_object initial_bindings_p)
{
  cl_env_ptr env = ecl_process_env();
  cl_object process = ecl_alloc_object(t_process);
  process->process.phase = ECL_PROCESS_INACTIVE;
  process->process.exit_values = ECL_NIL;
  process->process.entry = run_process;
  process->process.name = name;
  process->process.function = ECL_NIL;
  process->process.args = ECL_NIL;
  process->process.woken_up = ECL_NIL;
  process->process.inherit_bindings_p = Null(initial_bindings_p)? ECL_T : ECL_NIL;
  ecl_disable_interrupts_env(env);
  ecl_mutex_init(&process->process.start_stop_lock, TRUE);
  ecl_cond_var_init(&process->process.exit_barrier);
  ecl_set_finalizer_unprotected(process, ECL_T);
  ecl_enable_interrupts_env(env);
  return process;
}

bool
ecl_import_current_thread(cl_object name, cl_object bindings)
{
  cl_object process;
  cl_env_ptr the_env;
  if (ecl_process_env_unsafe() != NULL)
    return 0;
  ecl_module_gc->module.disable();
  the_env = ecl_adopt_cpu();
  ecl_module_gc->module.enable();
  ecl_enable_interrupts_env(the_env);

  process = alloc_process(name, ECL_NIL);
  process->process.env = the_env;
  process->process.phase = ECL_PROCESS_BOOTING;
  the_env->own_process = process;
  process->process.phase = ECL_PROCESS_ACTIVE;

  ecl_bds_bind(the_env, @'mp::*current-process*', process);
  return 1;
}

void
ecl_release_current_thread(void)
{
  cl_object process;
  cl_env_ptr the_env = ecl_process_env_unsafe();
  if (the_env == NULL)
    return;
  process = the_env->own_process;
  ecl_mutex_lock(&process->process.start_stop_lock);
  process->process.env = NULL;
  process->process.phase = ECL_PROCESS_EXITING;
  ecl_disown_cpu();
#ifdef HAVE_SIGPROCMASK
  /* ...but we might get stray signals. */
  {
    sigset_t new[1];
    sigemptyset(new);
    sigaddset(new, ecl_option_values[ECL_OPT_THREAD_INTERRUPT_SIGNAL]);
    ecl_sigmask(SIG_BLOCK, new, NULL);
  }
#endif
  process->process.phase = ECL_PROCESS_INACTIVE;
  ecl_cond_var_broadcast(&process->process.exit_barrier);
  ecl_mutex_unlock(&process->process.start_stop_lock);
}

@(defun mp::make-process (&key name ((:initial-bindings initial_bindings_p) ECL_T))
  cl_object process;
  @
  process = alloc_process(name, initial_bindings_p);
  @(return process);
  @)

cl_object
mp_process_preset(cl_narg narg, cl_object process, cl_object function, ...)
{
  ecl_va_list args;
  if (narg < 2)
    FEwrong_num_arguments(@[mp::process-preset]);
  ecl_va_start(args, function, narg, 2);
  assert_type_process(process);
  process->process.function = function;
  process->process.args = cl_grab_rest_args(args);
  ecl_va_end(args);
  @(return process);
}

cl_object
mp_interrupt_process(cl_object process, cl_object function)
{
  cl_env_ptr env = ecl_process_env();
  /* Make sure we don't interrupt an exiting process */
  ECL_WITH_NATIVE_LOCK_BEGIN(env, &process->process.start_stop_lock) {
    unlikely_if (mp_process_active_p(process) == ECL_NIL)
      FEerror("Cannot interrupt the inactive process ~A", 1, process);
    ecl_interrupt_process(process, function);
  } ECL_WITH_NATIVE_LOCK_END;
  @(return ECL_T);
}

cl_object
mp_suspend_loop()
{
  cl_env_ptr env = ecl_process_env();
  ECL_CATCH_BEGIN(env,@'mp::suspend-loop') {
    for ( ; ; ) {
      cl_sleep(ecl_make_fixnum(100));
    }
  } ECL_CATCH_END;
  ecl_return0(env);
}

cl_object
mp_break_suspend_loop()
{
  cl_env_ptr the_env = ecl_process_env();
  if (frs_sch(@'mp::suspend-loop')) {
    cl_throw(@'mp::suspend-loop');
  }
  ecl_return0(the_env);
}

cl_object
mp_process_suspend(cl_object process)
{
  return mp_interrupt_process(process, @'mp::suspend-loop');
}

cl_object
mp_process_resume(cl_object process)
{
  return mp_interrupt_process(process, @'mp::break-suspend-loop');
}

cl_object
mp_process_kill(cl_object process)
{
  return mp_interrupt_process(process, @'mp::exit-process');
}

cl_object
mp_process_yield(void)
{
#if defined(ECL_WINDOWS_THREADS)
  Sleep(0);
#elif defined(HAVE_SCHED_H)
  sched_yield();
#else
  ecl_musleep(0.0);
#endif
  @(return);
}

cl_object
mp_process_enable(cl_object process)
{
  cl_env_ptr the_env = ecl_process_env();
  cl_env_ptr process_env = NULL;
  /* Try to gain exclusive access to the process. This prevents two concurrent
   * calls to process-enable from different threads on the same process */
  ecl_mutex_lock(&process->process.start_stop_lock);
  /* Ensure that the process is inactive. */
  if (process->process.phase != ECL_PROCESS_INACTIVE) {
    ecl_mutex_unlock(&process->process.start_stop_lock);
    FEerror("Cannot enable the running process ~A.", 1, process);
  }
  process->process.phase = ECL_PROCESS_BOOTING;
  /* Spawn the thread (allocates the environment)*/
  process_env = ecl_spawn_cpu(process);
  if (process_env == NULL) {
    process->process.phase = ECL_PROCESS_INACTIVE;
    ecl_cond_var_broadcast(&process->process.exit_barrier);
  }
  /* Unleash the thread */
  ecl_mutex_unlock(&process->process.start_stop_lock);
  ecl_return1(the_env, (process_env ? process : ECL_NIL));
}

cl_object
mp_exit_process(void)
{
  /* We simply undo the whole of the frame stack. This brings up back to the
     thread entry point, going through all possible UNWIND-PROTECT. */
  const cl_env_ptr the_env = ecl_process_env();
  ecl_unwind(the_env, the_env->frs_stack.org);
  /* Never reached */
}

cl_object
mp_all_processes(void)
{
  /* No race condition here because this list is never destructively
   * modified. When we add or remove processes, we create new lists. */
  @(return ecl_process_list());
}

cl_object
mp_process_name(cl_object process)
{
  assert_type_process(process);
  @(return process->process.name);
}

cl_object
mp_process_active_p(cl_object process)
{
  assert_type_process(process);
  @(return (process->process.phase? ECL_T : ECL_NIL));
}

cl_object
mp_process_whostate(cl_object process)
{
  const cl_env_ptr the_env = ecl_process_env();
  assert_type_process(process);
  ecl_return1(the_env, ecl_ct_null_string);
}

cl_object
mp_process_join(cl_object process)
{
  cl_env_ptr the_env = ecl_process_env();
  volatile cl_object values;

  assert_type_process(process);
  ECL_UNWIND_PROTECT_BEGIN(the_env) {
    ecl_mutex_lock(&process->process.start_stop_lock);
    while (process->process.phase != ECL_PROCESS_INACTIVE) {
      ecl_cond_var_wait(&process->process.exit_barrier, &process->process.start_stop_lock);
    }
    values = cl_values_list(process->process.exit_values);
  } ECL_UNWIND_PROTECT_THREAD_SAFE_EXIT {
    ecl_mutex_unlock(&process->process.start_stop_lock);
  } ECL_UNWIND_PROTECT_THREAD_SAFE_END;
  return values;
}

cl_object
mp_process_run_function(cl_narg narg, cl_object name, cl_object function, ...)
{
  cl_object process;
  cl_object rest;
  ecl_va_list args;
  if (narg < 2)
    FEwrong_num_arguments(@[mp::process-run-function]);
  if (CONSP(name)) {
    process = cl_apply(2, @'mp::make-process', name);
  } else {
    process = mp_make_process(2, @':name', name);
  }
  ecl_va_start(args, function, narg, 2);
  rest = cl_grab_rest_args(args);
  ecl_va_end(args);
  cl_apply(4, @'mp::process-preset', process, function, rest);
  return mp_process_enable(process);
}

cl_object
mp_process_run_function_wait(cl_narg narg, ...)
{
  cl_object process;
  ecl_va_list args;
  ecl_va_start(args, narg, narg, 0);
  process = cl_apply(2, @'mp::process-run-function',
                     cl_grab_rest_args(args));
  if (!Null(process)) {
    while (process->process.phase < ECL_PROCESS_ACTIVE) {
      ecl_musleep(0.001);
    }
  }
  ecl_va_end(args);
  @(return process);
}

/*----------------------------------------------------------------------
 * INTERRUPTS
 */

#ifndef ECL_WINDOWS_THREADS
static cl_object
mp_get_sigmask(void)
{
  cl_object data = ecl_alloc_simple_vector(sizeof(sigset_t), ecl_aet_b8);
  sigset_t *mask_ptr = (sigset_t*)data->vector.self.b8;
  sigset_t no_signals;
  sigemptyset(&no_signals);
  if (ecl_sigmask(SIG_BLOCK, &no_signals, mask_ptr))
    FElibc_error("MP:GET-SIGMASK failed in a call to ecl_sigmask", 0);
  @(return data);
}

static cl_object
mp_set_sigmask(cl_object data)
{
  sigset_t *mask_ptr = (sigset_t*)data->vector.self.b8;
  if (ecl_sigmask(SIG_SETMASK, mask_ptr, NULL))
    FElibc_error("MP:SET-SIGMASK failed in a call to ecl_sigmask", 0);
  @(return data);
}
#endif

cl_object
mp_block_signals(void)
{
#ifdef ECL_WINDOWS_THREADS
  cl_env_ptr the_env = ecl_process_env();
  cl_object previous = ecl_cmp_symbol_value(the_env, ECL_INTERRUPTS_ENABLED);
  ECL_SETQ(the_env, ECL_INTERRUPTS_ENABLED, ECL_NIL);
  @(return previous);
#else
  cl_object previous = mp_get_sigmask();
  sigset_t all_signals;
  sigfillset(&all_signals);
  /* SIGSEGV or SIGBUS are needed by the gc in incremental mode and
   * can thus never be blocked */
  sigdelset(&all_signals, SIGSEGV);
  sigdelset(&all_signals, SIGBUS);
  if (ecl_sigmask(SIG_SETMASK, &all_signals, NULL))
    FElibc_error("MP:BLOCK-SIGNALS failed in a call to ecl_sigmask",0);
  @(return previous);
#endif
}

cl_object
mp_restore_signals(cl_object sigmask)
{
#ifdef ECL_WINDOWS_THREADS
  cl_env_ptr the_env = ecl_process_env();
  ECL_SETQ(the_env, ECL_INTERRUPTS_ENABLED, sigmask);
  ecl_check_pending_interrupts(the_env);
  @(return sigmask);
#else
  return mp_set_sigmask(sigmask);
#endif
}

/* -- Module definition --------------------------------------------- */

void
init_threads()
{
}

static cl_object
create_thread()
{
  cl_env_ptr the_env = ecl_core.first_env;
  cl_object process, _env = ecl_cast_ptr(cl_object,the_env);
  /* We have to set the environment before any allocation takes place,
   * so that the interrupt handling code works. */
  process = ecl_alloc_object(t_process);
  process->process.phase = ECL_PROCESS_ACTIVE;
  process->process.name = @'si::top-level';
  process->process.function = ECL_NIL;
  process->process.args = ECL_NIL;
  process->process.env = the_env;
  ecl_mutex_init(&process->process.start_stop_lock, TRUE);
  ecl_cond_var_init(&process->process.exit_barrier);
  the_env->own_process = process;
  ecl_stack_push(ecl_core.threads, _env);
  return ECL_NIL;
}

static cl_object
enable_thread()
{
  return ECL_NIL;
}

static cl_object
init_env_thread(cl_env_ptr the_env)
{
  return ECL_NIL;
}

static cl_object
init_cpu_thread(cl_env_ptr the_env)
{
  return ECL_NIL;
}

static cl_object
free_cpu_thread(cl_env_ptr the_env)
{
  return ECL_NIL;
}

static cl_object
free_env_thread(cl_env_ptr the_env)
{
  return ECL_NIL;
}

ecl_def_ct_base_string(str_thread, "THREAD", 6, static, const);

static struct ecl_module module_thread = {
  .name = str_thread,
  .create = create_thread,
  .enable = enable_thread,
  .init_env = init_env_thread,
  .init_cpu = init_cpu_thread,
  .free_cpu = free_cpu_thread,
  .free_env = free_env_thread,
  .disable = ecl_module_no_op,
  .destroy = ecl_module_no_op
};

cl_object ecl_module_thread = (cl_object)&module_thread;