/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

/*
 * File:        intrupt.c
 * Purpose:     testing thread interrupts
 */

#include "plgetopt.h"
#include "prcvar.h"
#include "prerror.h"
#include "prinit.h"
#include "prinrval.h"
#include "prio.h"
#include "prlock.h"
#include "prlog.h"
#include "prthread.h"
#include "prtypes.h"
#include "prnetdb.h"

#include <stdio.h>
#include <string.h>

#define DEFAULT_TCP_PORT 12500

static PRLock* ml = NULL;
static PRCondVar* cv = NULL;

static PRBool passed = PR_TRUE;
static PRBool debug_mode = PR_FALSE;
static PRThreadScope thread_scope = PR_LOCAL_THREAD;

static void PR_CALLBACK AbortCV(void* arg) {
  PRStatus rv;
  PRThread* me = PR_GetCurrentThread();

  /* some other thread (main) is doing the interrupt */
  PR_Lock(ml);
  rv = PR_WaitCondVar(cv, PR_INTERVAL_NO_TIMEOUT);
  if (debug_mode) {
    printf("Expected interrupt on wait CV and ");
  }
  if (PR_FAILURE == rv) {
    if (PR_PENDING_INTERRUPT_ERROR == PR_GetError()) {
      if (debug_mode) {
        printf("got it\n");
      }
    } else {
      if (debug_mode) {
        printf("got random error\n");
      }
      passed = PR_FALSE;
    }
  } else {
    if (debug_mode) {
      printf("got a successful completion\n");
    }
    passed = PR_FALSE;
  }

  rv = PR_WaitCondVar(cv, 10);
  if (debug_mode) {
    printf("Expected success on wait CV and %s\n",
           (PR_SUCCESS == rv) ? "got it" : "failed");
  }
  passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;

  /* interrupt myself, then clear */
  PR_Interrupt(me);
  PR_ClearInterrupt();
  rv = PR_WaitCondVar(cv, 10);
  if (debug_mode) {
    printf("Expected success on wait CV and ");
    if (PR_FAILURE == rv) {
      printf("%s\n", (PR_PENDING_INTERRUPT_ERROR == PR_GetError())
                         ? "got interrupted"
                         : "a random failure");
    }
    printf("got it\n");
  }
  passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;

  /* set, then wait - interrupt - then wait again */
  PR_Interrupt(me);
  rv = PR_WaitCondVar(cv, 10);
  if (debug_mode) {
    printf("Expected interrupt on wait CV and ");
  }
  if (PR_FAILURE == rv) {
    if (PR_PENDING_INTERRUPT_ERROR == PR_GetError()) {
      if (debug_mode) {
        printf("got it\n");
      }
    } else {
      if (debug_mode) {
        printf("failed\n");
      }
      passed = PR_FALSE;
    }
  } else {
    if (debug_mode) {
      printf("got a successful completion\n");
    }
    passed = PR_FALSE;
  }

  rv = PR_WaitCondVar(cv, 10);
  if (debug_mode) {
    printf("Expected success on wait CV and %s\n",
           (PR_SUCCESS == rv) ? "got it" : "failed");
  }
  passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;

  PR_Unlock(ml);

} /* AbortCV */

static void PR_CALLBACK AbortIO(void* arg) {
  PRStatus rv;
  PR_Sleep(PR_SecondsToInterval(2));
  rv = PR_Interrupt((PRThread*)arg);
  PR_ASSERT(PR_SUCCESS == rv);
} /* AbortIO */

static void PR_CALLBACK AbortJoin(void* arg) {} /* AbortJoin */

static void setup_listen_socket(PRFileDesc** listner, PRNetAddr* netaddr) {
  PRStatus rv;
  PRInt16 port = DEFAULT_TCP_PORT;

  *listner = PR_NewTCPSocket();
  PR_ASSERT(*listner != NULL);
  memset(netaddr, 0, sizeof(*netaddr));
  (*netaddr).inet.ip = PR_htonl(PR_INADDR_ANY);
  (*netaddr).inet.family = PR_AF_INET;
  do {
    (*netaddr).inet.port = PR_htons(port);
    rv = PR_Bind(*listner, netaddr);
    port += 1;
    PR_ASSERT(port < (DEFAULT_TCP_PORT + 10));
  } while (PR_FAILURE == rv);

  rv = PR_Listen(*listner, 5);

  if (PR_GetSockName(*listner, netaddr) < 0) {
    if (debug_mode) {
      printf("intrupt: ERROR - PR_GetSockName failed\n");
    }
    passed = PR_FALSE;
    return;
  }
}

static void PR_CALLBACK IntrBlock(void* arg) {
  PRStatus rv;
  PRNetAddr netaddr;
  PRFileDesc* listner;

  /* some other thread (main) is doing the interrupt */
  /* block the interrupt */
  PR_BlockInterrupt();
  PR_Lock(ml);
  rv = PR_WaitCondVar(cv, PR_SecondsToInterval(4));
  PR_Unlock(ml);
  if (debug_mode) {
    printf("Expected success on wait CV and ");
    if (PR_FAILURE == rv) {
      printf("%s\n", (PR_PENDING_INTERRUPT_ERROR == PR_GetError())
                         ? "got interrupted"
                         : "got a random failure");
    } else {
      printf("got it\n");
    }
  }
  passed = ((PR_TRUE == passed) && (PR_SUCCESS == rv)) ? PR_TRUE : PR_FALSE;

  setup_listen_socket(&listner, &netaddr);
  PR_UnblockInterrupt();
  if (PR_Accept(listner, &netaddr, PR_INTERVAL_NO_TIMEOUT) == NULL) {
    PRInt32 error = PR_GetError();
    if (debug_mode) {
      printf("Expected interrupt on PR_Accept() and ");
    }
    if (PR_PENDING_INTERRUPT_ERROR == error) {
      if (debug_mode) {
        printf("got it\n");
      }
    } else {
      if (debug_mode) {
        printf("failed\n");
      }
      passed = PR_FALSE;
    }
  } else {
    if (debug_mode) {
      printf("Failed to interrupt PR_Accept()\n");
    }
    passed = PR_FALSE;
  }

  (void)PR_Close(listner);
  listner = NULL;
} /* TestIntrBlock */

void PR_CALLBACK Intrupt(void* arg) {
  PRStatus rv;
  PRNetAddr netaddr;
  PRFileDesc* listner;
  PRThread *abortCV, *abortIO, *abortJoin, *intrBlock;

  ml = PR_NewLock();
  cv = PR_NewCondVar(ml);

  /* Part I */
  if (debug_mode) {
    printf("Part I\n");
  }
  abortCV = PR_CreateThread(PR_USER_THREAD, AbortCV, 0, PR_PRIORITY_NORMAL,
                            thread_scope, PR_JOINABLE_THREAD, 0);

  PR_Sleep(PR_SecondsToInterval(2));
  rv = PR_Interrupt(abortCV);
  PR_ASSERT(PR_SUCCESS == rv);
  rv = PR_JoinThread(abortCV);
  PR_ASSERT(PR_SUCCESS == rv);

  /* Part II */
  if (debug_mode) {
    printf("Part II\n");
  }
  abortJoin = PR_CreateThread(PR_USER_THREAD, AbortJoin, 0, PR_PRIORITY_NORMAL,
                              thread_scope, PR_JOINABLE_THREAD, 0);
  PR_Sleep(PR_SecondsToInterval(2));
  if (debug_mode) {
    printf("Expecting to interrupt an exited thread ");
  }
  rv = PR_Interrupt(abortJoin);
  PR_ASSERT(PR_SUCCESS == rv);
  rv = PR_JoinThread(abortJoin);
  PR_ASSERT(PR_SUCCESS == rv);
  if (debug_mode) {
    printf("and succeeded\n");
  }

  /* Part III */
  if (debug_mode) {
    printf("Part III\n");
  }
  setup_listen_socket(&listner, &netaddr);
  abortIO =
      PR_CreateThread(PR_USER_THREAD, AbortIO, PR_GetCurrentThread(),
                      PR_PRIORITY_NORMAL, thread_scope, PR_JOINABLE_THREAD, 0);

  if (PR_Accept(listner, &netaddr, PR_INTERVAL_NO_TIMEOUT) == NULL) {
    PRInt32 error = PR_GetError();
    if (debug_mode) {
      printf("Expected interrupt on PR_Accept() and ");
    }
    if (PR_PENDING_INTERRUPT_ERROR == error) {
      if (debug_mode) {
        printf("got it\n");
      }
    } else {
      if (debug_mode) {
        printf("failed\n");
      }
      passed = PR_FALSE;
    }
  } else {
    if (debug_mode) {
      printf("Failed to interrupt PR_Accept()\n");
    }
    passed = PR_FALSE;
  }

  (void)PR_Close(listner);
  listner = NULL;

  rv = PR_JoinThread(abortIO);
  PR_ASSERT(PR_SUCCESS == rv);
  /* Part VI */
  if (debug_mode) {
    printf("Part VI\n");
  }
  intrBlock = PR_CreateThread(PR_USER_THREAD, IntrBlock, 0, PR_PRIORITY_NORMAL,
                              thread_scope, PR_JOINABLE_THREAD, 0);

  PR_Sleep(PR_SecondsToInterval(2));
  rv = PR_Interrupt(intrBlock);
  PR_ASSERT(PR_SUCCESS == rv);
  rv = PR_JoinThread(intrBlock);
  PR_ASSERT(PR_SUCCESS == rv);

  PR_DestroyCondVar(cv);
  PR_DestroyLock(ml);
} /* Intrupt */

int main(int argc, char** argv) {
  PRThread* intrupt;
  PLOptStatus os;
  PLOptState* opt = PL_CreateOptState(argc, argv, "dG");
  while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
    if (PL_OPT_BAD == os) {
      continue;
    }
    switch (opt->option) {
      case 'd': /* debug mode */
        debug_mode = PR_TRUE;
        break;
      case 'G': /* use global threads */
        thread_scope = PR_GLOBAL_THREAD;
        break;
    }
  }
  PL_DestroyOptState(opt);
  intrupt = PR_CreateThread(PR_USER_THREAD, Intrupt, NULL, PR_PRIORITY_NORMAL,
                            thread_scope, PR_JOINABLE_THREAD, 0);
  if (intrupt == NULL) {
    fprintf(stderr, "cannot create thread\n");
    passed = PR_FALSE;
  } else {
    PRStatus rv;
    rv = PR_JoinThread(intrupt);
    PR_ASSERT(rv == PR_SUCCESS);
  }
  printf("%s\n", ((passed) ? "PASSED" : "FAILED"));
  return ((passed) ? 0 : 1);
} /* main */

/* intrupt.c */