/* 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/. */

#include "nspr.h"
#include "plgetopt.h"

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

int _debug_on = 0;
#define DPRINTF(arg) \
  if (_debug_on) printf arg

#include "obsolete/prsem.h"

PRLock* lock;
PRMonitor* mon;
PRMonitor* mon2;

#define DEFAULT_COUNT 1000

PRInt32 count;

static void nop(int a, int b, int c) {}

static void LocalProcedureCall(void) {
  PRInt32 i;

  for (i = 0; i < count; i++) {
    nop(i, i, 5);
  }
}

static void DLLProcedureCall(void) {
  PRInt32 i;
  PRThreadState state;
  PRThread* self = PR_GetCurrentThread();

  for (i = 0; i < count; i++) {
    state = PR_GetThreadState(self);
  }
}

static void Now(void) {
  PRInt32 i;
  PRTime time;

  for (i = 0; i < count; i++) {
    time = PR_Now();
  }
}

static void Interval(void) {
  PRInt32 i;
  PRIntervalTime time;

  for (i = 0; i < count; i++) {
    time = PR_IntervalNow();
  }
}

static void IdleLock(void) {
  PRInt32 i;

  for (i = 0; i < count; i++) {
    PR_Lock(lock);
    PR_Unlock(lock);
  }
}

static void IdleMonitor(void) {
  PRInt32 i;

  for (i = 0; i < count; i++) {
    PR_EnterMonitor(mon);
    PR_ExitMonitor(mon);
  }
}

static void IdleCMonitor(void) {
  PRInt32 i;

  for (i = 0; i < count; i++) {
    PR_CEnterMonitor((void*)7);
    PR_CExitMonitor((void*)7);
  }
}

/************************************************************************/

static void PR_CALLBACK dull(void* arg) {}

static void CDThread(void) {
  PRInt32 i;
  int num_threads = count;

  /*
   * Cannot create too many threads
   */
  if (num_threads > 1000) {
    num_threads = 1000;
  }

  for (i = 0; i < num_threads; i++) {
    PRThread* t = PR_CreateThread(PR_USER_THREAD, dull, 0, PR_PRIORITY_NORMAL,
                                  PR_LOCAL_THREAD, PR_UNJOINABLE_THREAD, 0);
    if (NULL == t) {
      fprintf(stderr, "CDThread: cannot create thread %3d\n", i);
    } else {
      DPRINTF(("CDThread: created thread %3d \n", i));
    }
    PR_Sleep(0);
  }
}

static int alive;
static int cxq;

static void PR_CALLBACK CXReader(void* arg) {
  PRInt32 i, n;

  PR_EnterMonitor(mon);
  n = count / 2;
  for (i = 0; i < n; i++) {
    while (cxq == 0) {
      DPRINTF(("CXReader: thread = 0x%lx waiting\n", PR_GetCurrentThread()));
      PR_Wait(mon, PR_INTERVAL_NO_TIMEOUT);
    }
    --cxq;
    PR_Notify(mon);
  }
  PR_ExitMonitor(mon);

  PR_EnterMonitor(mon2);
  --alive;
  PR_Notify(mon2);
  PR_ExitMonitor(mon2);
  DPRINTF(("CXReader: thread = 0x%lx exiting\n", PR_GetCurrentThread()));
}

static void PR_CALLBACK CXWriter(void* arg) {
  PRInt32 i, n;

  PR_EnterMonitor(mon);
  n = count / 2;
  for (i = 0; i < n; i++) {
    while (cxq == 1) {
      DPRINTF(("CXWriter: thread = 0x%lx waiting\n", PR_GetCurrentThread()));
      PR_Wait(mon, PR_INTERVAL_NO_TIMEOUT);
    }
    ++cxq;
    PR_Notify(mon);
  }
  PR_ExitMonitor(mon);

  PR_EnterMonitor(mon2);
  --alive;
  PR_Notify(mon2);
  PR_ExitMonitor(mon2);
  DPRINTF(("CXWriter: thread = 0x%lx exiting\n", PR_GetCurrentThread()));
}

static void ContextSwitch(PRThreadScope scope1, PRThreadScope scope2) {
  PRThread *t1, *t2;

  PR_EnterMonitor(mon2);
  alive = 2;
  cxq = 0;

  t1 = PR_CreateThread(PR_USER_THREAD, CXReader, 0, PR_PRIORITY_NORMAL, scope1,
                       PR_UNJOINABLE_THREAD, 0);
  if (NULL == t1) {
    fprintf(stderr, "ContextSwitch: cannot create thread\n");
  } else {
    DPRINTF(
        ("ContextSwitch: created %s thread = 0x%lx\n",
         (scope1 == PR_GLOBAL_THREAD ? "PR_GLOBAL_THREAD" : "PR_LOCAL_THREAD"),
         t1));
  }
  t2 = PR_CreateThread(PR_USER_THREAD, CXWriter, 0, PR_PRIORITY_NORMAL, scope2,
                       PR_UNJOINABLE_THREAD, 0);
  if (NULL == t2) {
    fprintf(stderr, "ContextSwitch: cannot create thread\n");
  } else {
    DPRINTF(
        ("ContextSwitch: created %s thread = 0x%lx\n",
         (scope2 == PR_GLOBAL_THREAD ? "PR_GLOBAL_THREAD" : "PR_LOCAL_THREAD"),
         t2));
  }

  /* Wait for both of the threads to exit */
  while (alive) {
    PR_Wait(mon2, PR_INTERVAL_NO_TIMEOUT);
  }
  PR_ExitMonitor(mon2);
}

static void ContextSwitchUU(void) {
  ContextSwitch(PR_LOCAL_THREAD, PR_LOCAL_THREAD);
}

static void ContextSwitchUK(void) {
  ContextSwitch(PR_LOCAL_THREAD, PR_GLOBAL_THREAD);
}

static void ContextSwitchKU(void) {
  ContextSwitch(PR_GLOBAL_THREAD, PR_LOCAL_THREAD);
}

static void ContextSwitchKK(void) {
  ContextSwitch(PR_GLOBAL_THREAD, PR_GLOBAL_THREAD);
}

/************************************************************************/

static void PR_CALLBACK SemaThread(void* argSema) {
  PRSemaphore** sem = (PRSemaphore**)argSema;
  PRInt32 i, n;

  n = count / 2;
  for (i = 0; i < n; i++) {
    DPRINTF(("SemaThread: thread = 0x%lx waiting on sem = 0x%lx\n",
             PR_GetCurrentThread(), sem[0]));
    PR_WaitSem(sem[0]);
    DPRINTF(("SemaThread: thread = 0x%lx posting on sem = 0x%lx\n",
             PR_GetCurrentThread(), sem[1]));
    PR_PostSem(sem[1]);
  }

  PR_EnterMonitor(mon2);
  --alive;
  PR_Notify(mon2);
  PR_ExitMonitor(mon2);
  DPRINTF(("SemaThread: thread = 0x%lx exiting\n", PR_GetCurrentThread()));
}

static PRSemaphore* sem_set1[2];
static PRSemaphore* sem_set2[2];

static void SemaContextSwitch(PRThreadScope scope1, PRThreadScope scope2) {
  PRThread *t1, *t2;
  sem_set1[0] = PR_NewSem(1);
  sem_set1[1] = PR_NewSem(0);
  sem_set2[0] = sem_set1[1];
  sem_set2[1] = sem_set1[0];

  PR_EnterMonitor(mon2);
  alive = 2;
  cxq = 0;

  t1 = PR_CreateThread(PR_USER_THREAD, SemaThread, sem_set1, PR_PRIORITY_NORMAL,
                       scope1, PR_UNJOINABLE_THREAD, 0);
  if (NULL == t1) {
    fprintf(stderr, "SemaContextSwitch: cannot create thread\n");
  } else {
    DPRINTF(
        ("SemaContextSwitch: created %s thread = 0x%lx\n",
         (scope1 == PR_GLOBAL_THREAD ? "PR_GLOBAL_THREAD" : "PR_LOCAL_THREAD"),
         t1));
  }
  t2 = PR_CreateThread(PR_USER_THREAD, SemaThread, sem_set2, PR_PRIORITY_NORMAL,
                       scope2, PR_UNJOINABLE_THREAD, 0);
  if (NULL == t2) {
    fprintf(stderr, "SemaContextSwitch: cannot create thread\n");
  } else {
    DPRINTF(
        ("SemaContextSwitch: created %s thread = 0x%lx\n",
         (scope2 == PR_GLOBAL_THREAD ? "PR_GLOBAL_THREAD" : "PR_LOCAL_THREAD"),
         t2));
  }

  /* Wait for both of the threads to exit */
  while (alive) {
    PR_Wait(mon2, PR_INTERVAL_NO_TIMEOUT);
  }
  PR_ExitMonitor(mon2);

  PR_DestroySem(sem_set1[0]);
  PR_DestroySem(sem_set1[1]);
}

static void SemaContextSwitchUU(void) {
  SemaContextSwitch(PR_LOCAL_THREAD, PR_LOCAL_THREAD);
}

static void SemaContextSwitchUK(void) {
  SemaContextSwitch(PR_LOCAL_THREAD, PR_GLOBAL_THREAD);
}

static void SemaContextSwitchKU(void) {
  SemaContextSwitch(PR_GLOBAL_THREAD, PR_LOCAL_THREAD);
}

static void SemaContextSwitchKK(void) {
  SemaContextSwitch(PR_GLOBAL_THREAD, PR_GLOBAL_THREAD);
}

/************************************************************************/

static void Measure(void (*func)(void), const char* msg) {
  PRIntervalTime start, stop;
  double d;

  start = PR_IntervalNow();
  (*func)();
  stop = PR_IntervalNow() - start;
  d = (double)PR_IntervalToMicroseconds(stop);

  printf("%40s: %6.2f usec\n", msg, d / count);
}

int main(int argc, char** argv) {
  PLOptStatus os;
  PLOptState* opt = PL_CreateOptState(argc, argv, "dc:");
  while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
    if (PL_OPT_BAD == os) {
      continue;
    }
    switch (opt->option) {
      case 'd': /* debug mode */
        _debug_on = 1;
        break;
      case 'c': /* loop count */
        count = atoi(opt->value);
        break;
      default:
        break;
    }
  }
  PL_DestroyOptState(opt);

  if (0 == count) {
    count = DEFAULT_COUNT;
  }

  PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
  PR_BlockClockInterrupts();
  PR_UnblockClockInterrupts();

  lock = PR_NewLock();
  mon = PR_NewMonitor();
  mon2 = PR_NewMonitor();

  Measure(LocalProcedureCall, "local procedure call overhead");
  Measure(DLLProcedureCall, "DLL procedure call overhead");
  Measure(Now, "current calendar time");
  Measure(Interval, "interval time");
  Measure(IdleLock, "idle lock lock/unlock pair");
  Measure(IdleMonitor, "idle monitor entry/exit pair");
  Measure(IdleCMonitor, "idle cache monitor entry/exit pair");
  Measure(CDThread, "create/destroy thread pair");
  Measure(ContextSwitchUU, "context switch - user/user");
  Measure(ContextSwitchUK, "context switch - user/kernel");
  Measure(ContextSwitchKU, "context switch - kernel/user");
  Measure(ContextSwitchKK, "context switch - kernel/kernel");
  Measure(SemaContextSwitchUU, "sema context switch - user/user");
  Measure(SemaContextSwitchUK, "sema context switch - user/kernel");
  Measure(SemaContextSwitchKU, "sema context switch - kernel/user");
  Measure(SemaContextSwitchKK, "sema context switch - kernel/kernel");

  printf("--------------\n");
  printf("Adding 7 additional CPUs\n");

  PR_SetConcurrency(8);
  printf("--------------\n");

  Measure(LocalProcedureCall, "local procedure call overhead");
  Measure(DLLProcedureCall, "DLL procedure call overhead");
  Measure(Now, "current calendar time");
  Measure(Interval, "interval time");
  Measure(IdleLock, "idle lock lock/unlock pair");
  Measure(IdleMonitor, "idle monitor entry/exit pair");
  Measure(IdleCMonitor, "idle cache monitor entry/exit pair");
  Measure(CDThread, "create/destroy thread pair");
  Measure(ContextSwitchUU, "context switch - user/user");
  Measure(ContextSwitchUK, "context switch - user/kernel");
  Measure(ContextSwitchKU, "context switch - kernel/user");
  Measure(ContextSwitchKK, "context switch - kernel/kernel");
  Measure(SemaContextSwitchUU, "sema context switch - user/user");
  Measure(SemaContextSwitchUK, "sema context switch - user/kernel");
  Measure(SemaContextSwitchKU, "sema context switch - kernel/user");
  Measure(SemaContextSwitchKK, "sema context switch - kernel/kernel");

  PR_DestroyLock(lock);
  PR_DestroyMonitor(mon);
  PR_DestroyMonitor(mon2);

  PR_Cleanup();
  return 0;
}