/* Pthread-friendly coroutines with pth
 * Copyright (C) 2002 Andy Wingo <wingo@pobox.com>
 *
 * cothreads.c: public API implementation
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#include "cothreads-private.h"

#define HAVE_LINUXTHREADS

#ifdef HAVE_LINUXTHREADS
static cothread_attr cothread_attr_default = 
{
  COTHREAD_ATTR_METHOD_LINUXTHREADS,  /* use the linuxthreads hack */
  0x200000,                           /* 2 MB */
  8,                                  /* for a stack size of 256 KB */
  TRUE                                /* set up the first chunk */
};
#else
static cothread_attr cothread_attr_default = 
{
  COTHREAD_ATTR_METHOD_GTHREAD_STACK, /* this is what the old cothreads code does */
  0x100000,                           /* only 1 MB due the the FreeBSD defaults */
  8,                                  /* for a stack size of 128 KB */
  TRUE                                /* set up the first chunk */
};
#endif

cothread_attr *_cothread_attr_global = NULL;

static gboolean (*stack_alloc_func) (char**, char**);

static void	cothread_private_set	(char *sp, void *priv, size_t size);
static void	cothread_private_get	(char *sp, void *priv, size_t size);
static void	cothread_stub		(void);


/**
 * cothreads_initialized:
 *
 * Query the state of the cothreads system.
 *
 * Returns: TRUE if cothreads_init() has already been called, FALSE otherwise
 */
gboolean
cothreads_initialized (void) 
{
  return (_cothread_attr_global != NULL);
}

/**
 * cothreads_init:
 * @attr: attributes for creation of cothread stacks
 *
 * Initialize the cothreads system. If @attr is NULL, use the default parameters
 * detected at compile-time.
 */
void
cothreads_init (cothread_attr *attr)
{
  static cothread_attr _attr;
  
  if (cothreads_initialized()) {
    g_warning ("cothread system has already been initialized");
    return;
  }
  
  if (!attr)
    _attr = cothread_attr_default;
  else 
    _attr = *attr;
  
  _cothread_attr_global = &_attr;
  
  switch (_cothread_attr_global->method) {
  case COTHREAD_ATTR_METHOD_MALLOC:
    stack_alloc_func = cothread_stack_alloc_on_heap;
    break;
  case COTHREAD_ATTR_METHOD_GTHREAD_STACK:
    stack_alloc_func = cothread_stack_alloc_on_gthread_stack;
    break;
  case COTHREAD_ATTR_METHOD_LINUXTHREADS:
    stack_alloc_func = cothread_stack_alloc_linuxthreads;
    break;
  default:
    g_error ("unexpected value for attr method %d", _cothread_attr_global->method);
  }
}

/**
 * cothread_create:
 * @func: function to start with this cothread
 * @argc: argument count
 * @argv: argument vector
 *
 * Create a new cothread running a given function. You must explictly switch
 * into this cothread to give it control. If @func is NULL, a cothread is
 * created on the current stack with the current stack pointer.
 *
 * Returns: A pointer to the new cothread
 */
cothread*
cothread_create (void (*func)(int, void **), int argc, void **argv)
{
  char *low, *high;
  cothread_private priv;
  cothread *ret = g_new0 (cothread, 1);
  
  if (!func) {
    /* we are being asked to save the current thread into a new cothread. this
     * only happens for the first cothread. */
    if (_cothread_attr_global->alloc_cothread_0)
      if (!stack_alloc_func (&low, &high))
        g_error ("couldn't create cothread 0");
      else
        g_message ("created cothread 0 with low=%p, high=%p", low, high);
    else
      g_message ("created cothread 0");
    
    pth_mctx_save (ret);
    return ret;
  }
  
  if (!stack_alloc_func (&low, &high))
    g_error ("could not allocate a new cothread stack");
  
  g_message ("created a cothread with low=%p, high=%p", low, high);
  
  pth_mctx_set (ret, cothread_stub, low, high);
  
  priv.argc = argc;
  priv.argv = argv;
  priv.func = func;
  cothread_private_set (low, &priv, sizeof(priv));
  
  return ret;
}

/**
 * cothread_destroy:
 * @thread: cothread to destroy
 *
 * Deallocate any memory used by the cothread data structures.
 */
void
cothread_destroy (cothread *thread)
{
  /* FIXME: have method-specific destroy functions. */
  
  g_free (thread);
}

/* the whole 'page size' thing is to allow for the last page of a stack or chunk
 * to be mmap'd as a boundary page */

static void
cothread_private_set (char *sp, void *priv, size_t size)
{
  char *dest;
  
#if PTH_STACK_GROWTH > 0
  dest = ((gulong)sp | (_cothread_attr_global->chunk_size / _cothread_attr_global->blocks_per_chunk - 1))
    - size + 1 - getpagesize();
#else
  dest = ((gulong)sp &~ (_cothread_attr_global->chunk_size / _cothread_attr_global->blocks_per_chunk - 1))
    + getpagesize();
#endif
  
  memcpy (dest, priv, size);
}

static void
cothread_private_get (char *sp, void *priv, size_t size)
{
  char *src;
  
#if PTH_STACK_GROWTH > 0
  src = ((gulong)sp | (_cothread_attr_global->chunk_size / _cothread_attr_global->blocks_per_chunk - 1))
    - size + 1 - getpagesize();
#else
  src = ((gulong)sp &~ (_cothread_attr_global->chunk_size / _cothread_attr_global->blocks_per_chunk - 1))
    + getpagesize();
#endif
  
  memcpy (priv, src, size);
}

static void
cothread_stub (void)
{
  cothread_private priv;
  
  cothread_private_get (CURRENT_STACK_FRAME, &priv, sizeof (priv));
  
  priv.func (priv.argc, priv.argv);
  
  g_warning ("we really shouldn't get here");
}