tstRTProcCreateEx.cpp revision 4ee5a4cd660730c997785c6cbc12881a115079e8
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync/* $Id$ */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync/** @file
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * IPRT Testcase - RTProcCreateEx.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync/*
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * Copyright (C) 2010 Sun Microsystems, Inc.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * available from http://www.virtualbox.org. This file is free software;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * you can redistribute it and/or modify it under the terms of the GNU
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * General Public License (GPL) as published by the Free Software
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * The contents of this file may alternatively be used under the terms
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * of the Common Development and Distribution License Version 1.0
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * VirtualBox OSE distribution, in which case the provisions of the
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * CDDL are applicable instead of those of the GPL.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * You may elect to license modified versions of this file under the
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * terms and conditions of either the GPL or the CDDL or both.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * Clara, CA 95054 USA or visit http://www.sun.com if you need
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * additional information or have any questions.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync/*******************************************************************************
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync* Header Files *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync*******************************************************************************/
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/process.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/assert.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/env.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/err.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/initterm.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/message.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/param.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/pipe.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/string.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/stream.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/test.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync#include <iprt/thread.h>
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync/*******************************************************************************
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync* Global Variables *
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync*******************************************************************************/
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic char g_szExecName[RTPATH_MAX];
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsyncstatic const char * const g_apszArgs4[] =
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync{
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 0 */ "non existing non executable file",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 1 */ "--testcase-child-4",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 2 */ "a b",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 3 */ " cdef",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 4 */ "ghijkl ",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 5 */ "\"",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 6 */ "\\",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 7 */ "\\\"",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 8 */ "\\\"\\",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /* 9 */ "\\\\\"\\",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync NULL
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync};
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsyncstatic int tstRTCreateProcEx4Child(int argc, char **argv)
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync{
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync int rc = RTR3Init();
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync if (rc)
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync return RTMsgInitFailure(rc);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync for (int i = 0; i < argc; i++)
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync if (strcmp(argv[i], g_apszArgs4[i]))
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync {
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTStrmPrintf(g_pStdErr,
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync "child4: argv[%2u]='%s'\n"
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync "child4: expected='%s'\n",
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync i, argv[i], g_apszArgs4[i]);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync rc++;
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync }
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync return rc;
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync}
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsyncstatic void tstRTCreateProcEx4(void)
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync{
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTTestISub("Argument with spaces and stuff");
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTPROCESS hProc;
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTTESTI_CHECK_RC_RETV(RTProcCreateEx(g_szExecName, g_apszArgs4, RTENV_DEFAULT, 0 /*fFlags*/, NULL,
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync NULL, NULL, NULL, &hProc), VINF_SUCCESS);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTPROCSTATUS ProcStatus = { -1, RTPROCEXITREASON_ABEND };
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTTESTI_CHECK_RC(RTProcWait(hProc, RTPROCWAIT_FLAGS_BLOCK, &ProcStatus), VINF_SUCCESS);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync if (ProcStatus.enmReason != RTPROCEXITREASON_NORMAL || ProcStatus.iStatus != 0)
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTTestIFailed("enmReason=%d iStatus=%d", ProcStatus.enmReason, ProcStatus.iStatus);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync else
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync RTTestIPassed(NULL);
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync}
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic int tstRTCreateProcEx3Child(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTR3Init();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (rc)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return RTMsgInitFailure(rc);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdOut, "w"); RTStrmFlush(g_pStdOut);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdErr, "o"); RTStrmFlush(g_pStdErr);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdOut, "r"); RTStrmFlush(g_pStdOut);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdErr, "k"); RTStrmFlush(g_pStdErr);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdOut, "s");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic void tstRTCreateProcEx3(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestISub("Standard Out+Err");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPIPE hPipeR, hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTPipeCreate(&hPipeR, &hPipeW, RTPIPE_C_INHERIT_WRITE), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync const char * apszArgs[3] =
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "non-existing-non-executable-file",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "--testcase-child-3",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync NULL
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTHANDLE Handle;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.enmType = RTHANDLETYPE_PIPE;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.u.hPipe = hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCESS hProc;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTProcCreateEx(g_szExecName, apszArgs, RTENV_DEFAULT, 0 /*fFlags*/, NULL,
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync &Handle, &Handle, NULL, &hProc), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeW), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync char szOutput[_4K];
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t offOutput = 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync for (;;)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbLeft = sizeof(szOutput) - 1 - offOutput;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK(cbLeft > 0);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (cbLeft == 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTPipeReadBlocking(hPipeR, &szOutput[offOutput], cbLeft, &cbRead);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (RT_FAILURE(rc))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(rc, VERR_BROKEN_PIPE);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync offOutput += cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync szOutput[offOutput] = '\0';
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeR), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCSTATUS ProcStatus = { -1, RTPROCEXITREASON_ABEND };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTProcWait(hProc, RTPROCWAIT_FLAGS_BLOCK, &ProcStatus), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTThreadSleep(10);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (ProcStatus.enmReason != RTPROCEXITREASON_NORMAL || ProcStatus.iStatus != 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("enmReason=%d iStatus=%d", ProcStatus.enmReason, ProcStatus.iStatus);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else if ( offOutput != sizeof("works") - 1
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync || strcmp(szOutput, "works"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("wrong output: \"%s\" (len=%u)", szOutput, offOutput);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIPassed(NULL);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic int tstRTCreateProcEx2Child(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTR3Init();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (rc)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return RTMsgInitFailure(rc);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdErr, "howdy");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdOut, "ignore this output\n");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic void tstRTCreateProcEx2(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestISub("Standard Err");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPIPE hPipeR, hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTPipeCreate(&hPipeR, &hPipeW, RTPIPE_C_INHERIT_WRITE), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync const char * apszArgs[3] =
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "non-existing-non-executable-file",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "--testcase-child-2",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync NULL
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTHANDLE Handle;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.enmType = RTHANDLETYPE_PIPE;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.u.hPipe = hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCESS hProc;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTProcCreateEx(g_szExecName, apszArgs, RTENV_DEFAULT, 0 /*fFlags*/, NULL,
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync NULL, &Handle, NULL, &hProc), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeW), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync char szOutput[_4K];
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t offOutput = 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync for (;;)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbLeft = sizeof(szOutput) - 1 - offOutput;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK(cbLeft > 0);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (cbLeft == 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTPipeReadBlocking(hPipeR, &szOutput[offOutput], cbLeft, &cbRead);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (RT_FAILURE(rc))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(rc, VERR_BROKEN_PIPE);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync offOutput += cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync szOutput[offOutput] = '\0';
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeR), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCSTATUS ProcStatus = { -1, RTPROCEXITREASON_ABEND };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTProcWait(hProc, RTPROCWAIT_FLAGS_BLOCK, &ProcStatus), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTThreadSleep(10);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (ProcStatus.enmReason != RTPROCEXITREASON_NORMAL || ProcStatus.iStatus != 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("enmReason=%d iStatus=%d", ProcStatus.enmReason, ProcStatus.iStatus);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else if ( offOutput != sizeof("howdy") - 1
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync || strcmp(szOutput, "howdy"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("wrong output: \"%s\" (len=%u)", szOutput, offOutput);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIPassed(NULL);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic int tstRTCreateProcEx1Child(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTR3Init();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (rc)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return RTMsgInitFailure(rc);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPrintf("it works");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrmPrintf(g_pStdErr, "ignore this output\n");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncstatic void tstRTCreateProcEx1(void)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestISub("Standard Out");
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPIPE hPipeR, hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTPipeCreate(&hPipeR, &hPipeW, RTPIPE_C_INHERIT_WRITE), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync const char * apszArgs[3] =
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "non-existing-non-executable-file",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync "--testcase-child-1",
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync NULL
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTHANDLE Handle;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.enmType = RTHANDLETYPE_PIPE;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync Handle.u.hPipe = hPipeW;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCESS hProc;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC_RETV(RTProcCreateEx(g_szExecName, apszArgs, RTENV_DEFAULT, 0 /*fFlags*/, NULL,
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync &Handle, NULL, NULL, &hProc), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeW), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync char szOutput[_4K];
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t offOutput = 0;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync for (;;)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbLeft = sizeof(szOutput) - 1 - offOutput;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK(cbLeft > 0);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (cbLeft == 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync size_t cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTPipeReadBlocking(hPipeR, &szOutput[offOutput], cbLeft, &cbRead);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (RT_FAILURE(rc))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync {
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(rc, VERR_BROKEN_PIPE);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync break;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync offOutput += cbRead;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync }
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync szOutput[offOutput] = '\0';
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTPipeClose(hPipeR), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTPROCSTATUS ProcStatus = { -1, RTPROCEXITREASON_ABEND };
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTESTI_CHECK_RC(RTProcWait(hProc, RTPROCWAIT_FLAGS_BLOCK, &ProcStatus), VINF_SUCCESS);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (ProcStatus.enmReason != RTPROCEXITREASON_NORMAL || ProcStatus.iStatus != 0)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("enmReason=%d iStatus=%d", ProcStatus.enmReason, ProcStatus.iStatus);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else if ( offOutput != sizeof("it works") - 1
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync || strcmp(szOutput, "it works"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIFailed("wrong output: \"%s\" (len=%u)", szOutput, offOutput);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync else
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestIPassed(NULL);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsyncint main(int argc, char **argv)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync{
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (argc == 2 && !strcmp(argv[1], "--testcase-child-1"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return tstRTCreateProcEx1Child();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (argc == 2 && !strcmp(argv[1], "--testcase-child-2"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return tstRTCreateProcEx2Child();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (argc == 2 && !strcmp(argv[1], "--testcase-child-3"))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return tstRTCreateProcEx3Child();
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync if (argc >= 5 && !strcmp(argv[1], "--testcase-child-4"))
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync return tstRTCreateProcEx4Child(argc, argv);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (argc != 1)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return 99;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTEST hTest;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync int rc = RTTestInitAndCreate("tstRTProcCreateEx", &hTest);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (rc)
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return rc;
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTTestBanner(hTest);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync if (!RTProcGetExecutableName(g_szExecName, sizeof(g_szExecName)))
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync RTStrCopy(g_szExecName, sizeof(g_szExecName), argv[0]);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync /*
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * The tests.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync tstRTCreateProcEx1();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync tstRTCreateProcEx2();
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync tstRTCreateProcEx3();
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync tstRTCreateProcEx4();
4ee5a4cd660730c997785c6cbc12881a115079e8vboxsync /** @todo Cover files, pszAsUser, ++ */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync /*
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync * Summary.
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync */
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync return RTTestSummaryAndDestroy(hTest);
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync}
b1d7d513c459787311cd09c440524044fa7ff8a9vboxsync