/**
* @interface_method_impl{RTVFSIOSTREAMOPS,pfnWrite}
*/
static DECLCALLBACK(int) rtManifestPtIos_Write(void *pvThis, RTFOFF off, PCRTSGBUF pSgBuf, bool fBlocking, size_t *pcbWritten)
{
PRTMANIFESTPTIOS pThis = (PRTMANIFESTPTIOS)pvThis;
int rc = RTVfsIoStrmSgWrite(pThis->hVfsIos, pSgBuf, fBlocking, pcbWritten);
if (RT_SUCCESS(rc))
rtManifestPtIos_UpdateHashes(pThis, pSgBuf, pcbWritten ? *pcbWritten : ~(size_t)0);
Assert(off == -1); NOREF(off);
return rc;
}
/**
* Internal helper for rtZipGzip_Write, rtZipGzip_Flush and rtZipGzip_Close.
*
* @returns IPRT status code.
* @retval VINF_SUCCESS
* @retval VINF_TRY_AGAIN - the only informational status.
* @retval VERR_INTERRUPTED - call again.
*
* @param pThis The gzip I/O stream instance data.
* @param fBlocking Whether to block or not.
*/
static int rtZipGzip_WriteOutputBuffer(PRTZIPGZIPSTREAM pThis, bool fBlocking)
{
/*
* Anything to write? No, then just return immediately.
*/
size_t cbToWrite = sizeof(pThis->abBuffer) - pThis->Zlib.avail_out;
if (cbToWrite == 0)
{
Assert(pThis->Zlib.next_out == &pThis->abBuffer[0]);
return VINF_SUCCESS;
}
Assert(cbToWrite <= sizeof(pThis->abBuffer));
/*
* Loop write on VERR_INTERRUPTED.
*
* Note! Asserting a bit extra here to make sure the
* RTVfsIoStrmSgWrite works correctly.
*/
int rc;
size_t cbWrittenOut;
for (;;)
{
/* Set up the buffer. */
pThis->SgSeg.cbSeg = cbToWrite;
Assert(pThis->SgSeg.pvSeg == &pThis->abBuffer[0]);
RTSgBufReset(&pThis->SgBuf);
cbWrittenOut = ~(size_t)0;
rc = RTVfsIoStrmSgWrite(pThis->hVfsIos, &pThis->SgBuf, fBlocking, &cbWrittenOut);
if (rc != VINF_SUCCESS)
{
AssertMsg(RT_FAILURE(rc) || rc == VINF_TRY_AGAIN, ("%Rrc\n", rc));
if (rc == VERR_INTERRUPTED)
{
Assert(cbWrittenOut == 0);
continue;
}
if (RT_FAILURE(rc) || rc == VINF_TRY_AGAIN || cbWrittenOut == 0)
{
AssertReturn(cbWrittenOut == 0, VERR_INTERNAL_ERROR_3);
AssertReturn(rc != VINF_SUCCESS, VERR_IPE_UNEXPECTED_INFO_STATUS);
return rc;
}
}
break;
}
AssertMsgReturn(cbWrittenOut > 0 && cbWrittenOut <= sizeof(pThis->abBuffer),
("%zu %Rrc\n", cbWrittenOut, rc),
VERR_INTERNAL_ERROR_4);
/*
* Adjust the Zlib output buffer members.
*/
if (cbWrittenOut == pThis->SgBuf.paSegs[0].cbSeg)
{
pThis->Zlib.avail_out = sizeof(pThis->abBuffer);
pThis->Zlib.next_out = &pThis->abBuffer[0];
}
else
{
Assert(cbWrittenOut <= pThis->SgBuf.paSegs[0].cbSeg);
size_t cbLeft = pThis->SgBuf.paSegs[0].cbSeg - cbWrittenOut;
memmove(&pThis->abBuffer[0], &pThis->abBuffer[cbWrittenOut], cbLeft);
pThis->Zlib.avail_out += (uInt)cbWrittenOut;
pThis->Zlib.next_out = &pThis->abBuffer[cbWrittenOut];
}
return VINF_SUCCESS;
}
