/* 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;

}