/* 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 "prio.h"
#include "prprf.h"
#include "prinit.h"
#include "prthread.h"
#include "prproces.h"
#include "prinrval.h"

#include "plgetopt.h"

#include <stdlib.h>

static PRInt32 dally = 0;
static PRFileDesc* err = NULL;
static PRBool verbose = PR_FALSE, force = PR_FALSE;

static void Help(void) {
  PR_fprintf(err, "Usage: [-t s] [-h]\n");
  PR_fprintf(err, "\t-d   Verbose output              (default: FALSE)\n");
  PR_fprintf(err, "\t-x   Forced termination          (default: FALSE)\n");
  PR_fprintf(err, "\t-t   Time for thread to block    (default: 10 seconds)\n");
  PR_fprintf(err, "\t-h   This message and nothing else\n");
} /* Help */

static void Dull(void* arg) {
  PR_Sleep(PR_SecondsToInterval(dally));
  if (verbose && force) {
    PR_fprintf(err, "If you see this, the test failed\n");
  }
} /* Dull */

static PRIntn PR_CALLBACK RealMain(PRIntn argc, char** argv) {
  PLOptStatus os;
  PLOptState* opt = PL_CreateOptState(argc, argv, "ht:dx");

  err = PR_GetSpecialFD(PR_StandardError);

  while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
    if (PL_OPT_BAD == os) {
      continue;
    }
    switch (opt->option) {
      case 'd': /* verbosity */
        verbose = PR_TRUE;
        break;
      case 'x': /* force exit */
        force = PR_TRUE;
        break;
      case 't': /* seconds to dally in child */
        dally = atoi(opt->value);
        break;
      case 'h': /* user wants some guidance */
      default:
        Help();   /* so give him an earful */
        return 2; /* but not a lot else */
    }
  }
  PL_DestroyOptState(opt);

  if (0 == dally) {
    dally = 10;
  }

  /*
   * Create LOCAL and GLOBAL threads
   */
  (void)PR_CreateThread(PR_USER_THREAD, Dull, NULL, PR_PRIORITY_NORMAL,
                        PR_LOCAL_THREAD, PR_UNJOINABLE_THREAD, 0);

  (void)PR_CreateThread(PR_USER_THREAD, Dull, NULL, PR_PRIORITY_NORMAL,
                        PR_GLOBAL_THREAD, PR_UNJOINABLE_THREAD, 0);

  if (verbose)
    PR_fprintf(err, "Main is exiting now. Program should exit %s.\n",
               (force) ? "immediately" : "after child dally time");

  if (force) {
    PR_ProcessExit(0);
    if (verbose) {
      PR_fprintf(err, "You should not have gotten here.\n");
      return 1;
    }
  }
  return 0;
}

int main(int argc, char** argv) {
  PRIntn rv;

  rv = PR_Initialize(RealMain, argc, argv, 0);
  return rv;
} /* main */