/* 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:        parent.c
** description: test the process machinery
*/

#include "prmem.h"
#include "prprf.h"
#include "prinit.h"
#include "prproces.h"
#include "prinrval.h"

typedef struct Child {
    const char* name;
    char** argv;
    PRProcess* process;
    PRProcessAttr* attr;
} Child;

/* for the default test 'cvar -c 2000' */
static char* default_argv[] = { "cvar", "-c", "2000", NULL };

static void
PrintUsage(void)
{
    PR_fprintf(PR_GetSpecialFD(PR_StandardError),
               "Usage: parent [-d] child [options]\n");
}

int
main(int argc, char** argv)
{
    PRStatus rv;
    PRInt32 test_status = 1;
    PRIntervalTime t_start, t_elapsed;
    PRFileDesc* debug = NULL;
    Child* child = PR_NEWZAP(Child);

    if (1 == argc) {
        /* no command-line arguments: run the default test */
        child->argv = default_argv;
    } else {
        argv += 1; /* don't care about our program name */
        while (*argv != NULL && argv[0][0] == '-') {
            if (argv[0][1] == 'd') {
                debug = PR_GetSpecialFD(PR_StandardError);
            } else {
                PrintUsage();
                return 2; /* not sufficient */
            }
            argv += 1;
        }
        child->argv = argv;
    }

    if (NULL == *child->argv) {
        PrintUsage();
        return 2;
    }

    child->name = *child->argv;
    if (NULL != debug) {
        PR_fprintf(debug, "Forking %s\n", child->name);
    }

    child->attr = PR_NewProcessAttr();
    PR_ProcessAttrSetStdioRedirect(child->attr, PR_StandardOutput,
                                   PR_GetSpecialFD(PR_StandardOutput));
    PR_ProcessAttrSetStdioRedirect(child->attr, PR_StandardError,
                                   PR_GetSpecialFD(PR_StandardError));

    t_start = PR_IntervalNow();
    child->process =
        PR_CreateProcess(child->name, child->argv, NULL, child->attr);
    t_elapsed = (PRIntervalTime)(PR_IntervalNow() - t_start);

    PR_DestroyProcessAttr(child->attr);

    test_status = (NULL == child->process) ? 1 : 0;
    if (NULL != debug) {
        PR_fprintf(debug, "Child was %sforked\n", (0 == test_status) ? "" : "NOT ");
        if (0 == test_status)
            PR_fprintf(debug, "PR_CreateProcess took %lu microseconds\n",
                       PR_IntervalToMicroseconds(t_elapsed));
    }

    if (0 == test_status) {
        if (NULL != debug) {
            PR_fprintf(debug, "Waiting for child to exit\n");
        }
        rv = PR_WaitProcess(child->process, &test_status);
        if (PR_SUCCESS == rv) {
            if (NULL != debug)
                PR_fprintf(debug, "Child exited %s\n",
                           (0 == test_status) ? "successfully" : "with error");
        } else {
            test_status = 1;
            if (NULL != debug) {
                PR_fprintf(debug, "PR_WaitProcess failed\n");
            }
        }
    }
    PR_DELETE(child);
    PR_Cleanup();
    return test_status;

} /* main */

/* parent.c */