summaryrefslogtreecommitdiffstats
path: root/modules/libjar/zipwriter/StreamFunctions.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'modules/libjar/zipwriter/StreamFunctions.cpp')
-rw-r--r--modules/libjar/zipwriter/StreamFunctions.cpp44
1 files changed, 44 insertions, 0 deletions
diff --git a/modules/libjar/zipwriter/StreamFunctions.cpp b/modules/libjar/zipwriter/StreamFunctions.cpp
new file mode 100644
index 0000000000..a16db28844
--- /dev/null
+++ b/modules/libjar/zipwriter/StreamFunctions.cpp
@@ -0,0 +1,44 @@
+/* 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;
+}