/**
* Adjust the profile environment after forking the child process and changing
* the UID.
*
* @returns IRPT status code.
* @param hEnvToUse The environment we're going to use with execve.
* @param fFlags The process creation flags.
* @param hEnv The environment passed in by the user.
*/
static int rtProcPosixAdjustProfileEnvFromChild(RTENV hEnvToUse, uint32_t fFlags, RTENV hEnv)
{
int rc = VINF_SUCCESS;
#ifdef RT_OS_DARWIN
if ( RT_SUCCESS(rc)
&& (!(fFlags & RTPROC_FLAGS_ENV_CHANGE_RECORD) || RTEnvExistEx(hEnv, "TMPDIR")) )
{
char szValue[_4K];
size_t cbNeeded = confstr(_CS_DARWIN_USER_TEMP_DIR, szValue, sizeof(szValue));
if (cbNeeded > 0 && cbNeeded < sizeof(szValue))
{
char *pszTmp;
rc = RTStrCurrentCPToUtf8(&pszTmp, szValue);
if (RT_SUCCESS(rc))
{
rc = RTEnvSetEx(hEnvToUse, "TMPDIR", pszTmp);
RTStrFree(pszTmp);
}
}
else
rc = VERR_BUFFER_OVERFLOW;
}
#endif
return rc;
}
static int tstRTCreateProcEx6Child(int argc, char **argv)
{
int rc = RTR3InitExeNoArguments(0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
int cErrors = 0;
char szValue[_16K];
/*
* Check for the environment variable we've set in the parent process.
*/
if (argc >= 3 && strcmp(argv[2], "inherit") == 0)
{
if (!RTEnvExistEx(RTENV_DEFAULT, "testcase-child-6"))
{
RTStrmPrintf(g_pStdErr, "child6: Env.var. 'testcase-child-6' was not inherited from parent\n");
cErrors++;
}
}
else if (argc >= 3 && strstr(argv[2], "change-record") != NULL)
{
rc = RTEnvGetEx(RTENV_DEFAULT, "testcase-child-6", szValue, sizeof(szValue), NULL);
if (RT_SUCCESS(rc) && strcmp(szValue, "changed"))
{
RTStrmPrintf(g_pStdErr, "child6: Env.var. 'testcase-child-6'='%s', expected 'changed'.\n", szValue);
cErrors++;
}
else if (RT_FAILURE(rc))
{
RTStrmPrintf(g_pStdErr, "child6: RTEnvGetEx(,'testcase-child-6',,) -> %Rrc\n", rc);
cErrors++;
}
}
else
{
if (RTEnvExistEx(RTENV_DEFAULT, "testcase-child-6"))
{
RTStrmPrintf(g_pStdErr, "child6: Env.var. 'testcase-child-6' was inherited from parent\n");
cErrors++;
}
}
/*
* Check the user name if present we didn't inherit from parent.
*/
if ( argc >= 4
&& argv[3][0] != '\0'
&& strstr(argv[2], "noinherit") != NULL)
{
static struct
{
const char *pszVarNm;
bool fReq;
} const s_aVars[] =
{
#ifdef RT_OS_WINDOWS
{ "USERNAME", true },
#else
{ "LOGNAME", true },
{ "USER", false },
#endif
};
for (unsigned i = 0; i < RT_ELEMENTS(s_aVars); i++)
{
rc = RTEnvGetEx(RTENV_DEFAULT, s_aVars[i].pszVarNm, szValue, sizeof(szValue), NULL);
if (RT_SUCCESS(rc))
{
if (strcmp(szValue, argv[3]))
{
RTStrmPrintf(g_pStdErr, "child6: env.var. '%s'='%s', expected '%s'\n",
s_aVars[i].pszVarNm, szValue, argv[3]);
cErrors++;
}
}
else if (rc != VERR_ENV_VAR_NOT_FOUND || s_aVars[i].fReq)
{
RTStrmPrintf(g_pStdErr, "child6: RTGetEnv('%s') -> %Rrc\n", s_aVars[i].pszVarNm, rc);
cErrors++;
}
}
}
#if 1
/* For manual testing. */
if (strcmp(argv[2],"noinherit") == 0)
//if (strcmp(argv[2],"noinherit-change-record") == 0)
{
RTENV hEnv;
rc = RTEnvClone(&hEnv, RTENV_DEFAULT);
if (RT_SUCCESS(rc))
{
uint32_t cVars = RTEnvCountEx(hEnv);
for (uint32_t i = 0; i < cVars; i++)
{
char szVarNm[_1K];
rc = RTEnvGetByIndexEx(hEnv, i, szVarNm, sizeof(szVarNm), szValue, sizeof(szValue));
if (RT_SUCCESS(rc))
RTStrmPrintf(g_pStdErr, "child6: #%u: %s=%s\n", i, szVarNm, szValue);
else
{
RTStrmPrintf(g_pStdErr, "child6: #%u: %Rrc\n", i, rc);
cErrors++;
}
}
RTEnvDestroy(hEnv);
}
else
{
RTStrmPrintf(g_pStdErr, "child6: RTEnvClone failed: %Rrc\n", rc);
cErrors++;
}
}
#endif
return cErrors == 0 ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE;
}
