/* Linuxthreads - a simple clone()-based implementation of Posix        */
/* threads for Linux.                                                   */
/* Copyright (C) 1996 Xavier Leroy (Xavier.Leroy@inria.fr)              */
/*                                                                      */
/* This program 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 program 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.                 */

/* Thread cancellation */

#include "pthread.h"
#include "internals.h"
#include "spinlock.h"
#include "restart.h"

int pthread_setcancelstate(int state, int * oldstate)
{
  pthread_descr self = thread_self();
  /* poniższy napis odpowiada (przynajmniej w tej wersji biblioteki) */
  /* if (state != PTHREAD_CANCEL_ENABLE && state != PTHREAD_CANCEL_DISABLE */
  if (state < PTHREAD_CANCEL_ENABLE || state > PTHREAD_CANCEL_DISABLE)
    return EINVAL;
  if (oldstate != NULL) *oldstate = self->p_cancelstate;
  self->p_cancelstate = state;
  if (self->p_canceled && 
      self->p_cancelstate == PTHREAD_CANCEL_ENABLE &&
      self->p_canceltype == PTHREAD_CANCEL_ASYNCHRONOUS)
    pthread_exit(PTHREAD_CANCELED);    /* jeśli można anulować wątek i ktoś już */
                                       /* próbował to zrobić to anulujmy go     */
  return 0;
}

int pthread_setcanceltype(int type, int * oldtype)
{
  pthread_descr self = thread_self();
  /* poniższy napis odpowiada (przynajmniej w tej wersji biblioteki) */
  /* if (state != PTHREAD_CANCEL_DEFERRED && state != PTHREAD_CANCEL_ASYNCHRONOUS */
  if (type < PTHREAD_CANCEL_DEFERRED || type > PTHREAD_CANCEL_ASYNCHRONOUS)
    return EINVAL;
  if (oldtype != NULL) *oldtype = self->p_canceltype;
  self->p_canceltype = type;
  if (self->p_canceled && 
      self->p_cancelstate == PTHREAD_CANCEL_ENABLE &&
      self->p_canceltype == PTHREAD_CANCEL_ASYNCHRONOUS)
    pthread_exit(PTHREAD_CANCELED);    /* jeśli można anulować wątek i ktoś już */
                                       /* próbował to zrobić to anulujmy go     */
  return 0;
}

int pthread_cancel(pthread_t thread)
/* bezpieczne niszczenie wątków (tzn. bez obawy, że zostanie zniszczony */
/* cały proces, jak stałoby się po wysłaniu sygnału SIGKILL)           */
{
  pthread_handle handle = thread_handle(thread);
  int pid;

  acquire(&handle->h_spinlock);
  if (invalid_handle(handle, thread)) {  /* zły identyfikator */
    release(&handle->h_spinlock);
    return ESRCH;
  }
  handle->h_descr->p_canceled = 1;
  pid = handle->h_descr->p_pid;
  release(&handle->h_spinlock);
  kill(pid, PTHREAD_SIG_CANCEL);
  return 0;
}

void pthread_testcancel(void)
/* wyjście z wątku tylko jeśli ktoś próbował go zniszczyć (anulować)        */
/* funkcja ta pozwala wątkowi (po ustawieniu flagi PTHREAD_CANCEL_DEFERRED) */
/* na kontrolę miejsca, w którym może zostać anulowany                      */
/* - nie chcielibyśmy zapewne pozwolić na zniszczenie wątku w środku        */
/* sekcji krytycznej                                                        */
{
  pthread_descr self = thread_self();
  if (self->p_canceled && self->p_cancelstate == PTHREAD_CANCEL_ENABLE)
    pthread_exit(PTHREAD_CANCELED);
}

void _pthread_cleanup_push(struct _pthread_cleanup_buffer * buffer,
                           void (*routine)(void *), void * arg)
/* dołożenie funkcji czyszczącej (wywoływanej przy wyjściu) do listy */
{
  pthread_descr self = thread_self();
  buffer->routine = routine;
  buffer->arg = arg;
  buffer->prev = self->p_cleanup;
  self->p_cleanup = buffer;
}

void _pthread_cleanup_pop(struct _pthread_cleanup_buffer * buffer,
                          int execute)
/* zdjęcie funkcji czyszczącej (wywoływanej przy wyjściu) z listy */
/* z ewentualnym wywołaniem (flaga execute)                       */
{
  pthread_descr self = thread_self();
  if (execute) buffer->routine(buffer->arg);
  self->p_cleanup = buffer->prev;
}

void _pthread_cleanup_push_defer(struct _pthread_cleanup_buffer * buffer,
                                    void (*routine)(void *), void * arg)
/* dołożenie funkcji czyszczącej (wywoływanej przy wyjściu) do listy         */
/* zapamiętanie flagi canceltype i ustawienie jej na PTHREAD_CANCEL_DEFERRED */
/* co oznacza anulowanie na żądanie                                          */
{
  pthread_descr self = thread_self();
  buffer->routine = routine;
  buffer->arg = arg;
  buffer->canceltype = self->p_canceltype;
  buffer->prev = self->p_cleanup;
  self->p_canceltype = PTHREAD_CANCEL_DEFERRED;
  self->p_cleanup = buffer;
}

void _pthread_cleanup_pop_restore(struct _pthread_cleanup_buffer * buffer,
                                     int execute)
/* zdjęcie funkcji czyszczącej (wywoływanej przy wyjściu) z listy           */
/* z ewentualnym wywołaniem (flaga execute) i przywrócenie flagi canceltype */
{
  pthread_descr self = thread_self();
  if (execute) buffer->routine(buffer->arg);
  self->p_cleanup = buffer->prev;
  self->p_canceltype = buffer->canceltype;
  if (self->p_canceled && 
      self->p_cancelstate == PTHREAD_CANCEL_ENABLE &&
      self->p_canceltype == PTHREAD_CANCEL_ASYNCHRONOUS)
    pthread_exit(PTHREAD_CANCELED);    /* jeśli można anulować wątek i ktoś już */
                                       /* próbował to zrobić to anulujmy go     */
}

void __pthread_perform_cleanup(void)
/* wywołanie wszystkich funkcji czyszczących */
{
  pthread_descr self = thread_self();
  struct _pthread_cleanup_buffer * c;
  for (c = self->p_cleanup; c != NULL; c = c->prev) c->routine(c->arg);
}