/**
* Closes the specified stream.
*
* @returns iprt status code.
* @param pStream The stream to close.
*/
RTR3DECL(int) RTStrmClose(PRTSTREAM pStream)
{
if (!pStream)
return VINF_SUCCESS;
AssertReturn(RT_VALID_PTR(pStream) && pStream->u32Magic == RTSTREAM_MAGIC, VERR_INVALID_PARAMETER);
if (!fclose(pStream->pFile))
{
pStream->u32Magic = 0xdeaddead;
pStream->pFile = NULL;
#ifndef HAVE_FWRITE_UNLOCKED
if (pStream->pCritSect)
{
RTCritSectEnter(pStream->pCritSect);
RTCritSectLeave(pStream->pCritSect);
RTCritSectDelete(pStream->pCritSect);
RTMemFree(pStream->pCritSect);
pStream->pCritSect = NULL;
}
#endif
RTMemFree(pStream);
return VINF_SUCCESS;
}
return RTErrConvertFromErrno(errno);
}
static void doGeneralTests(PCFGMNODE pRoot)
{
/* test multilevel node creation */
PCFGMNODE pChild = NULL;
RTTESTI_CHECK_RC_RETV(CFGMR3InsertNode(pRoot, "First/Second/Third//Final", &pChild), VINF_SUCCESS);
RTTESTI_CHECK_RETV(RT_VALID_PTR(pChild));
RTTESTI_CHECK(CFGMR3GetChild(pRoot, "First/Second/Third/Final") == pChild);
/*
* Boolean queries.
*/
RTTESTI_CHECK_RC(CFGMR3InsertInteger(pChild, "BoolValue", 1), VINF_SUCCESS);
bool f = false;
RTTESTI_CHECK_RC(CFGMR3QueryBool(pChild, "BoolValue", &f), VINF_SUCCESS);
RTTESTI_CHECK(f == true);
RTTESTI_CHECK_RC(CFGMR3QueryBool(pRoot, "BoolValue", &f), VERR_CFGM_VALUE_NOT_FOUND);
RTTESTI_CHECK_RC(CFGMR3QueryBool(NULL, "BoolValue", &f), VERR_CFGM_NO_PARENT);
RTTESTI_CHECK_RC(CFGMR3QueryBoolDef(pChild, "ValueNotFound", &f, true), VINF_SUCCESS);
RTTESTI_CHECK(f == true);
RTTESTI_CHECK_RC(CFGMR3QueryBoolDef(pChild, "ValueNotFound", &f, false), VINF_SUCCESS);
RTTESTI_CHECK(f == false);
RTTESTI_CHECK_RC(CFGMR3QueryBoolDef(NULL, "BoolValue", &f, true), VINF_SUCCESS);
RTTESTI_CHECK(f == true);
RTTESTI_CHECK_RC(CFGMR3QueryBoolDef(NULL, "BoolValue", &f, false), VINF_SUCCESS);
RTTESTI_CHECK(f == false);
}
/**
* Clears stream error condition.
*
* All stream operations save RTStrmClose and this will fail
* while an error is asserted on the stream
*
* @returns iprt status code.
* @param pStream The stream.
*/
RTR3DECL(int) RTStrmClearError(PRTSTREAM pStream)
{
AssertReturn(RT_VALID_PTR(pStream) && pStream->u32Magic == RTSTREAM_MAGIC, VERR_INVALID_PARAMETER);
clearerr(pStream->pFile);
ASMAtomicWriteS32(&pStream->i32Error, VINF_SUCCESS);
return VINF_SUCCESS;
}
/**
* Reads from a file stream.
*
* @returns iprt status code.
* @param pStream The stream.
* @param pvBuf Where to put the read bits.
* Must be cbRead bytes or more.
* @param cbRead Number of bytes to read.
* @param pcbRead Where to store the number of bytes actually read.
* If NULL cbRead bytes are read or an error is returned.
*/
RTR3DECL(int) RTStrmReadEx(PRTSTREAM pStream, void *pvBuf, size_t cbRead, size_t *pcbRead)
{
AssertReturn(RT_VALID_PTR(pStream) && pStream->u32Magic == RTSTREAM_MAGIC, VERR_INVALID_PARAMETER);
int rc = pStream->i32Error;
if (RT_SUCCESS(rc))
{
if (pStream->fRecheckMode)
rtStreamRecheckMode(pStream);
if (pcbRead)
{
/*
* Can do with a partial read.
*/
*pcbRead = fread(pvBuf, 1, cbRead, pStream->pFile);
if ( *pcbRead == cbRead
|| !ferror(pStream->pFile))
return VINF_SUCCESS;
if (feof(pStream->pFile))
{
if (*pcbRead)
return VINF_EOF;
rc = VERR_EOF;
}
else if (ferror(pStream->pFile))
rc = VERR_READ_ERROR;
else
{
AssertMsgFailed(("This shouldn't happen\n"));
rc = VERR_INTERNAL_ERROR;
}
}
else
{
/*
* Must read it all!
*/
if (fread(pvBuf, cbRead, 1, pStream->pFile) == 1)
return VINF_SUCCESS;
/* possible error/eof. */
if (feof(pStream->pFile))
rc = VERR_EOF;
else if (ferror(pStream->pFile))
rc = VERR_READ_ERROR;
else
{
AssertMsgFailed(("This shouldn't happen\n"));
rc = VERR_INTERNAL_ERROR;
}
}
ASMAtomicWriteS32(&pStream->i32Error, rc);
}
return rc;
}
/**
* @note This code is duplicated on other platforms with variations, so please
* keep them all up to date when making changes!
*/
int VBOXCALL VBoxGuestIDC(void *pvSession, uintptr_t uReq, PVBGLREQHDR pReqHdr, size_t cbReq)
{
/*
* Simple request validation (common code does the rest).
*/
int rc;
if ( RT_VALID_PTR(pReqHdr)
&& cbReq >= sizeof(*pReqHdr))
{
/*
* All requests except the connect one requires a valid session.
*/
PVBOXGUESTSESSION pSession = (PVBOXGUESTSESSION)pvSession;
if (pSession)
{
if ( RT_VALID_PTR(pSession)
&& pSession->pDevExt == &g_DevExt)
rc = VGDrvCommonIoCtl(uReq, &g_DevExt, pSession, pReqHdr, cbReq);
else
rc = VERR_INVALID_HANDLE;
}
else if (uReq == VBGL_IOCTL_IDC_CONNECT)
{
rc = VGDrvCommonCreateKernelSession(&g_DevExt, &pSession);
if (RT_SUCCESS(rc))
{
rc = VGDrvCommonIoCtl(uReq, &g_DevExt, pSession, pReqHdr, cbReq);
if (RT_FAILURE(rc))
VGDrvCommonCloseSession(&g_DevExt, pSession);
}
}
else
rc = VERR_INVALID_HANDLE;
}
else
rc = VERR_INVALID_POINTER;
return rc;
}
STDMETHODIMP VirtualBoxSDS::DeregisterVBoxSVC(IVBoxSVCRegistration *aVBoxSVC, LONG aPid)
{
LogRel(("deregisterVBoxSVC: aVBoxSVC=%p aPid=%u (%#x)\n", (IVBoxSVCRegistration *)aVBoxSVC, aPid, aPid));
HRESULT hrc;
if (RT_VALID_PTR(aVBoxSVC))
{
/* Get the client user SID and name. */
com::Utf8Str strSid;
com::Utf8Str strUsername;
if (i_getClientUserSid(&strSid, &strUsername))
{
VBoxSDSPerUserData *pUserData = i_lookupPerUserData(strSid);
if (pUserData)
{
if (aVBoxSVC == (IVBoxSVCRegistration *)pUserData->m_ptrTheChosenOne)
{
LogRel(("deregisterVBoxSVC: It's the chosen one for %s (%s)!\n",
pUserData->m_strUserSid.c_str(), pUserData->m_strUsername.c_str()));
#ifdef WITH_WATCHER
i_stopWatching(pUserData, pUserData->m_pidTheChosenOne);
#endif
pUserData->i_unchooseTheOne(false /*fIrregular*/);
}
else
LogRel(("deregisterVBoxSVC: not the choosen one (%p != %p)\n",
(IVBoxSVCRegistration *)aVBoxSVC, (IVBoxSVCRegistration *)pUserData->m_ptrTheChosenOne));
pUserData->i_unlock();
pUserData->i_release();
hrc = S_OK;
}
else
{
LogRel(("deregisterVBoxSVC: Found no user data for %s (%s) (pid %u)\n",
strSid.c_str(), strUsername.c_str(), aPid));
hrc = S_OK;
}
}
else
hrc = E_FAIL;
}
else
hrc = E_INVALIDARG;
LogRel2(("VirtualBoxSDS::deregisterVBoxSVC: returns %Rhrc\n", hrc));
return hrc;
}
static PPDMNSBWGROUP pdmNsBwGroupFindById(PPDMNETSHAPER pShaper, const char *pcszId)
{
PPDMNSBWGROUP pBwGroup = NULL;
if (RT_VALID_PTR(pcszId))
{
int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
pBwGroup = pShaper->pBwGroupsHead;
while ( pBwGroup
&& RTStrCmp(pBwGroup->pszName, pcszId))
pBwGroup = pBwGroup->pNext;
rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
}
return pBwGroup;
}
static PPDMNSBWGROUP pdmNsBwGroupFindById(PPDMNETSHAPER pShaper, const char *pszId)
{
PPDMNSBWGROUP pBwGroup = NULL;
if (RT_VALID_PTR(pszId))
{
LOCK_NETSHAPER(pShaper);
pBwGroup = pShaper->pBwGroupsHead;
while ( pBwGroup
&& RTStrCmp(pBwGroup->pszNameR3, pszId))
pBwGroup = pBwGroup->pNextR3;
UNLOCK_NETSHAPER(pShaper);
}
return pBwGroup;
}
/**
* Get the pending error of the stream.
*
* @returns iprt status code. of the stream.
* @param pStream The stream.
*/
RTR3DECL(int) RTStrmError(PRTSTREAM pStream)
{
AssertReturn(RT_VALID_PTR(pStream) && pStream->u32Magic == RTSTREAM_MAGIC, VERR_INVALID_PARAMETER);
return pStream->i32Error;
}
/**
* Worker for DBGFR3Info and DBGFR3InfoEx.
*
* @returns VBox status code.
* @param pUVM The user mode VM handle.
* @param idCpu Which CPU to run EMT bound handlers on. VMCPUID_ANY or
* a valid CPU ID.
* @param pszName What to dump.
* @param pszArgs Arguments, optional.
* @param pHlp Output helper, optional.
*/
static DECLCALLBACK(int) dbgfR3Info(PUVM pUVM, VMCPUID idCpu, const char *pszName, const char *pszArgs, PCDBGFINFOHLP pHlp)
{
/*
* Validate input.
*/
AssertPtrReturn(pszName, VERR_INVALID_POINTER);
AssertPtrNullReturn(pszArgs, VERR_INVALID_POINTER);
if (pHlp)
{
AssertPtrReturn(pHlp, VERR_INVALID_PARAMETER);
AssertPtrReturn(pHlp->pfnPrintf, VERR_INVALID_PARAMETER);
AssertPtrReturn(pHlp->pfnPrintfV, VERR_INVALID_PARAMETER);
}
else
pHlp = &g_dbgfR3InfoLogHlp;
/*
* Find the info handler.
*/
size_t cchName = strlen(pszName);
int rc = RTCritSectEnter(&pUVM->dbgf.s.InfoCritSect);
AssertRC(rc);
PDBGFINFO pInfo = pUVM->dbgf.s.pInfoFirst;
for (; pInfo; pInfo = pInfo->pNext)
if ( pInfo->cchName == cchName
&& !memcmp(pInfo->szName, pszName, cchName))
break;
if (pInfo)
{
/*
* Found it.
* Make a copy of it on the stack so we can leave the crit sect.
* Switch on the type and invoke the handler.
*/
DBGFINFO Info = *pInfo;
rc = RTCritSectLeave(&pUVM->dbgf.s.InfoCritSect);
AssertRC(rc);
rc = VINF_SUCCESS;
switch (Info.enmType)
{
case DBGFINFOTYPE_DEV:
if (Info.fFlags & DBGFINFO_FLAGS_RUN_ON_EMT)
rc = VMR3ReqCallVoidWaitU(pUVM, idCpu, (PFNRT)Info.u.Dev.pfnHandler, 3, Info.u.Dev.pDevIns, pHlp, pszArgs);
else
Info.u.Dev.pfnHandler(Info.u.Dev.pDevIns, pHlp, pszArgs);
break;
case DBGFINFOTYPE_DRV:
if (Info.fFlags & DBGFINFO_FLAGS_RUN_ON_EMT)
rc = VMR3ReqCallVoidWaitU(pUVM, idCpu, (PFNRT)Info.u.Drv.pfnHandler, 3, Info.u.Drv.pDrvIns, pHlp, pszArgs);
else
Info.u.Drv.pfnHandler(Info.u.Drv.pDrvIns, pHlp, pszArgs);
break;
case DBGFINFOTYPE_INT:
if (RT_VALID_PTR(pUVM->pVM))
{
if (Info.fFlags & DBGFINFO_FLAGS_RUN_ON_EMT)
rc = VMR3ReqCallVoidWaitU(pUVM, idCpu, (PFNRT)Info.u.Int.pfnHandler, 3, pUVM->pVM, pHlp, pszArgs);
else
Info.u.Int.pfnHandler(pUVM->pVM, pHlp, pszArgs);
}
else
rc = VERR_INVALID_VM_HANDLE;
break;
case DBGFINFOTYPE_EXT:
if (Info.fFlags & DBGFINFO_FLAGS_RUN_ON_EMT)
rc = VMR3ReqCallVoidWaitU(pUVM, idCpu, (PFNRT)Info.u.Ext.pfnHandler, 3, Info.u.Ext.pvUser, pHlp, pszArgs);
else
Info.u.Ext.pfnHandler(Info.u.Ext.pvUser, pHlp, pszArgs);
break;
default:
AssertMsgFailedReturn(("Invalid info type enmType=%d\n", Info.enmType), VERR_IPE_NOT_REACHED_DEFAULT_CASE);
}
}
else
{
rc = RTCritSectLeave(&pUVM->dbgf.s.InfoCritSect);
AssertRC(rc);
rc = VERR_FILE_NOT_FOUND;
}
return rc;
}
/* SDS plan B interfaces: */
STDMETHODIMP VirtualBoxSDS::RegisterVBoxSVC(IVBoxSVCRegistration *aVBoxSVC, LONG aPid, IUnknown **aExistingVirtualBox)
{
LogRel(("registerVBoxSVC: aVBoxSVC=%p aPid=%u (%#x)\n", (IVBoxSVCRegistration *)aVBoxSVC, aPid, aPid));
/*
* Get the caller PID so we can validate the aPid parameter with the other two.
* The V2 structure requires Vista or later, so fake it if older.
*/
RPC_CALL_ATTRIBUTES_V2_W CallAttribs = { RPC_CALL_ATTRIBUTES_VERSION, RPC_QUERY_CLIENT_PID | RPC_QUERY_IS_CLIENT_LOCAL };
RPC_STATUS rcRpc;
if (RTSystemGetNtVersion() >= RTSYSTEM_MAKE_NT_VERSION(6, 0, 0))
rcRpc = RpcServerInqCallAttributesW(NULL, &CallAttribs);
else
{
CallAttribs.ClientPID = (HANDLE)(intptr_t)aPid;
rcRpc = RPC_S_OK;
}
HRESULT hrc;
if ( RT_VALID_PTR(aVBoxSVC)
&& RT_VALID_PTR(aExistingVirtualBox)
&& rcRpc == RPC_S_OK
&& (intptr_t)CallAttribs.ClientPID == aPid)
{
*aExistingVirtualBox = NULL;
/*
* Get the client user SID and name.
*/
com::Utf8Str strSid;
com::Utf8Str strUsername;
if (i_getClientUserSid(&strSid, &strUsername))
{
VBoxSDSPerUserData *pUserData = i_lookupOrCreatePerUserData(strSid, strUsername); /* (returns holding the lock) */
if (pUserData)
{
/*
* If there already is a chosen one, ask it for a IVirtualBox instance
* to return to the caller. Should it be dead or unresponsive, the caller
* takes its place.
*/
if (pUserData->m_ptrTheChosenOne.isNotNull())
{
try
{
hrc = pUserData->m_ptrTheChosenOne->GetVirtualBox(aExistingVirtualBox);
}
catch (...)
{
LogRel(("registerVBoxSVC: Unexpected exception calling GetVirtualBox!!\n"));
hrc = E_FAIL;
}
if (FAILED_DEAD_INTERFACE(hrc))
{
LogRel(("registerVBoxSVC: Seems VBoxSVC instance died. Dropping it and letting caller take over. (hrc=%Rhrc)\n", hrc));
#ifdef WITH_WATCHER
i_stopWatching(pUserData, pUserData->m_pidTheChosenOne);
#endif
pUserData->i_unchooseTheOne(true /*fIrregular*/);
}
}
else
hrc = S_OK;
/*
* No chosen one? Make the caller the new chosen one!
*/
if (pUserData->m_ptrTheChosenOne.isNull())
{
LogRel(("registerVBoxSVC: Making aPid=%u (%#x) the chosen one for user %s (%s)!\n",
aPid, aPid, pUserData->m_strUserSid.c_str(), pUserData->m_strUsername.c_str()));
#ifdef WITH_WATCHER
/* Open the process so we can watch it. */
HANDLE hProcess = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION, FALSE /*fInherit*/, aPid);
if (hProcess == NULL)
hProcess = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_LIMITED_INFORMATION, FALSE /*fInherit*/, aPid);
if (hProcess == NULL)
hProcess = OpenProcess(SYNCHRONIZE, FALSE /*fInherit*/, aPid);
if (hProcess != NULL)
{
if (i_watchIt(pUserData, hProcess, aPid))
#endif
{
/* Make it official... */
pUserData->m_ptrTheChosenOne = aVBoxSVC;
pUserData->m_pidTheChosenOne = aPid;
hrc = S_OK;
}
#ifdef WITH_WATCHER
else
{
LogRel(("registerVBoxSVC: i_watchIt failed!\n"));
hrc = RPC_E_OUT_OF_RESOURCES;
}
}
else
{
LogRel(("registerVBoxSVC: OpenProcess failed: %u\n", GetLastError()));
hrc = E_ACCESSDENIED;
}
#endif
}
pUserData->i_unlock();
pUserData->i_release();
}
else
hrc = E_OUTOFMEMORY;
}
else
hrc = E_FAIL;
}
else if ( !RT_VALID_PTR(aVBoxSVC)
|| !RT_VALID_PTR(aExistingVirtualBox))
hrc = E_INVALIDARG;
else if (rcRpc != RPC_S_OK)
{
LogRel(("registerVBoxSVC: rcRpc=%d (%#x)!\n", rcRpc, rcRpc));
hrc = E_UNEXPECTED;
}
else
{
LogRel(("registerVBoxSVC: Client PID mismatch: aPid=%d (%#x), RPC ClientPID=%zd (%#zx)\n",
aPid, aPid, CallAttribs.ClientPID, CallAttribs.ClientPID));
hrc = E_INVALIDARG;
}
LogRel2(("VirtualBoxSDS::registerVBoxSVC: returns %Rhrc aExistingVirtualBox=%p\n", hrc, (IUnknown *)aExistingVirtualBox));
return hrc;
}