static RTEXITCODE tstRTPipe4Child(const char *pszPipe)
{
int rc = RTR3InitExeNoArguments(0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
int64_t iNative;
rc = RTStrToInt64Full(pszPipe, 10, &iNative);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTStrToUInt64Full(%s) -> %Rrc\n", pszPipe, rc);
RTPIPE hPipe;
rc = RTPipeFromNative(&hPipe, (RTHCINTPTR)iNative, RTPIPE_N_WRITE);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeFromNative(,%s,WRITE) -> %Rrc\n", pszPipe, rc);
rc = RTPipeWriteBlocking(hPipe, g_szTest4Message, sizeof(g_szTest4Message) - 1, NULL);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeWriteBlocking() -> %Rrc\n", rc);
rc = RTPipeClose(hPipe);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeClose() -> %Rrc\n", rc);
return RTEXITCODE_SUCCESS;
}
static RTEXITCODE tstRTPipe5Child(const char *pszPipe)
{
int rc = RTR3InitExeNoArguments(0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
int64_t iNative;
rc = RTStrToInt64Full(pszPipe, 10, &iNative);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTStrToUInt64Full(%s) -> %Rrc\n", pszPipe, rc);
RTPIPE hPipe;
rc = RTPipeFromNative(&hPipe, (RTHCINTPTR)iNative, RTPIPE_N_READ);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeFromNative(,%s,READ) -> %Rrc\n", pszPipe, rc);
///
char szTmp[1024];
size_t cbRead = 0;
rc = RTPipeReadBlocking(hPipe, szTmp, sizeof(szTmp) - 1, &cbRead);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeReadBlocking() -> %Rrc\n", rc);
szTmp[cbRead] = '\0';
size_t cbRead2;
char szTmp2[4];
rc = RTPipeReadBlocking(hPipe, szTmp2, sizeof(szTmp2), &cbRead2);
if (rc != VERR_BROKEN_PIPE)
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeReadBlocking() -> %Rrc instead of VERR_BROKEN_PIPE\n", rc);
rc = RTPipeClose(hPipe);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPipeClose() -> %Rrc\n", rc);
if (memcmp(szTmp, g_szTest5Message, sizeof(g_szTest5Message)))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "Message mismatch.\n:Expected '%s'\nGot '%s'\n", g_szTest5Message, szTmp);
return RTEXITCODE_SUCCESS;
}
RTDECL(int) RTHandleGetStandard(RTHANDLESTD enmStdHandle, PRTHANDLE ph)
{
/*
* Validate and convert input.
*/
AssertPtrReturn(ph, VERR_INVALID_POINTER);
int fd;
switch (enmStdHandle)
{
case RTHANDLESTD_INPUT: fd = 0; break;
case RTHANDLESTD_OUTPUT: fd = 1; break;
case RTHANDLESTD_ERROR: fd = 2; break;
default:
AssertFailedReturn(VERR_INVALID_PARAMETER);
}
/*
* Is the requested descriptor valid and which IPRT handle type does it
* best map on to?
*/
struct stat st;
int rc = fstat(fd, &st);
if (rc == -1)
return RTErrConvertFromErrno(errno);
rc = fcntl(fd, F_GETFD, 0);
if (rc == -1)
return RTErrConvertFromErrno(errno);
bool const fInherit = !(rc & FD_CLOEXEC);
RTHANDLE h;
if (S_ISREG(st.st_mode))
h.enmType = RTHANDLETYPE_FILE;
else if ( S_ISFIFO(st.st_mode)
|| (st.st_mode == 0 && st.st_nlink == 0 /*see bugs on bsd manpage*/))
h.enmType = RTHANDLETYPE_PIPE;
else if (S_ISSOCK(st.st_mode))
{
/** @todo check if it's really a socket... IIRC some OSes reports
* anonymouse pips as sockets. */
h.enmType = RTHANDLETYPE_SOCKET;
}
#if 0 /** @todo re-enable this when the VFS pipe has been coded up. */
else if (isatty(fd))
h.enmType = RTHANDLETYPE_PIPE;
#endif
else
h.enmType = RTHANDLETYPE_FILE;
/*
* Create the IPRT handle.
*/
switch (h.enmType)
{
case RTHANDLETYPE_FILE:
rc = RTFileFromNative(&h.u.hFile, fd);
break;
case RTHANDLETYPE_PIPE:
rc = RTPipeFromNative(&h.u.hPipe, fd,
(enmStdHandle == RTHANDLESTD_INPUT ? RTPIPE_N_READ : RTPIPE_N_WRITE)
| (fInherit ? RTPIPE_N_INHERIT : 0));
break;
case RTHANDLETYPE_SOCKET:
rc = rtSocketCreateForNative(&h.u.hSocket, fd);
break;
default: /* shut up gcc */
return VERR_INTERNAL_ERROR;
}
if (RT_SUCCESS(rc))
*ph = h;
return rc;
}
RTDECL(int) RTHandleGetStandard(RTHANDLESTD enmStdHandle, PRTHANDLE ph)
{
/*
* Validate and convert input.
*/
AssertPtrReturn(ph, VERR_INVALID_POINTER);
DWORD dwStdHandle;
switch (enmStdHandle)
{
case RTHANDLESTD_INPUT: dwStdHandle = STD_INPUT_HANDLE; break;
case RTHANDLESTD_OUTPUT: dwStdHandle = STD_OUTPUT_HANDLE; break;
case RTHANDLESTD_ERROR: dwStdHandle = STD_ERROR_HANDLE; break;
default:
AssertFailedReturn(VERR_INVALID_PARAMETER);
}
/*
* Is the requested descriptor valid and which IPRT handle type does it
* best map on to?
*/
HANDLE hNative = GetStdHandle(dwStdHandle);
if (hNative == INVALID_HANDLE_VALUE)
return RTErrConvertFromWin32(GetLastError());
DWORD dwInfo;
if (!GetHandleInformation(hNative, &dwInfo))
return RTErrConvertFromWin32(GetLastError());
bool const fInherit = RT_BOOL(dwInfo & HANDLE_FLAG_INHERIT);
RTHANDLE h;
DWORD dwType = GetFileType(hNative);
switch (dwType & ~FILE_TYPE_REMOTE)
{
default:
case FILE_TYPE_UNKNOWN:
case FILE_TYPE_CHAR:
case FILE_TYPE_DISK:
h.enmType = RTHANDLETYPE_FILE;
break;
case FILE_TYPE_PIPE:
{
DWORD cMaxInstances;
DWORD fInfo;
if (!GetNamedPipeInfo(hNative, &fInfo, NULL, NULL, &cMaxInstances))
h.enmType = RTHANDLETYPE_SOCKET;
else
h.enmType = RTHANDLETYPE_PIPE;
break;
}
}
/*
* Create the IPRT handle.
*/
int rc;
switch (h.enmType)
{
case RTHANDLETYPE_FILE:
rc = RTFileFromNative(&h.u.hFile, (RTHCUINTPTR)hNative);
break;
case RTHANDLETYPE_PIPE:
rc = RTPipeFromNative(&h.u.hPipe, (RTHCUINTPTR)hNative,
(enmStdHandle == RTHANDLESTD_INPUT ? RTPIPE_N_READ : RTPIPE_N_WRITE)
| (fInherit ? RTPIPE_N_INHERIT : 0));
break;
case RTHANDLETYPE_SOCKET:
rc = rtSocketCreateForNative(&h.u.hSocket, (RTHCUINTPTR)hNative);
break;
default: /* shut up gcc */
return VERR_INTERNAL_ERROR;
}
if (RT_SUCCESS(rc))
*ph = h;
return rc;
}
