зеркало из https://github.com/mozilla/gecko-dev.git
49 строки
1.4 KiB
C++
49 строки
1.4 KiB
C++
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include "nscore.h"
|
|
#include "nsIInputStream.h"
|
|
#include "nsIOutputStream.h"
|
|
|
|
/*
|
|
* Fully reads the required amount of data. Keeps reading until all the
|
|
* data is retrieved or an error is hit.
|
|
*/
|
|
nsresult ZW_ReadData(nsIInputStream *aStream, char *aBuffer, uint32_t aCount)
|
|
{
|
|
while (aCount > 0) {
|
|
uint32_t read;
|
|
nsresult rv = aStream->Read(aBuffer, aCount, &read);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
aCount -= read;
|
|
aBuffer += read;
|
|
// If we hit EOF before reading the data we need then throw.
|
|
if (read == 0 && aCount > 0)
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
/*
|
|
* Fully writes the required amount of data. Keeps writing until all the
|
|
* data is written or an error is hit.
|
|
*/
|
|
nsresult ZW_WriteData(nsIOutputStream *aStream, const char *aBuffer,
|
|
uint32_t aCount)
|
|
{
|
|
while (aCount > 0) {
|
|
uint32_t written;
|
|
nsresult rv = aStream->Write(aBuffer, aCount, &written);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
if (written <= 0)
|
|
return NS_ERROR_FAILURE;
|
|
aCount -= written;
|
|
aBuffer += written;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|