summaryrefslogtreecommitdiffstats
path: root/intl/uconv/nsConverterOutputStream.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'intl/uconv/nsConverterOutputStream.cpp')
-rw-r--r--intl/uconv/nsConverterOutputStream.cpp151
1 files changed, 151 insertions, 0 deletions
diff --git a/intl/uconv/nsConverterOutputStream.cpp b/intl/uconv/nsConverterOutputStream.cpp
new file mode 100644
index 000000000..dae3624a9
--- /dev/null
+++ b/intl/uconv/nsConverterOutputStream.cpp
@@ -0,0 +1,151 @@
+/* vim:set expandtab ts=4 sw=4 sts=4 cin: */
+/* 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 "nsCOMPtr.h"
+
+#include "nsIOutputStream.h"
+#include "nsString.h"
+
+#include "nsConverterOutputStream.h"
+#include "nsIUnicodeEncoder.h"
+#include "mozilla/dom/EncodingUtils.h"
+
+using mozilla::dom::EncodingUtils;
+
+NS_IMPL_ISUPPORTS(nsConverterOutputStream,
+ nsIUnicharOutputStream,
+ nsIConverterOutputStream)
+
+nsConverterOutputStream::~nsConverterOutputStream()
+{
+ Close();
+}
+
+NS_IMETHODIMP
+nsConverterOutputStream::Init(nsIOutputStream* aOutStream,
+ const char* aCharset,
+ uint32_t aBufferSize /* ignored */,
+ char16_t aReplacementChar)
+{
+ NS_PRECONDITION(aOutStream, "Null output stream!");
+
+ nsAutoCString label;
+ if (!aCharset) {
+ label.AssignLiteral("UTF-8");
+ } else {
+ label = aCharset;
+ }
+
+ nsAutoCString encoding;
+ if (label.EqualsLiteral("UTF-16")) {
+ // Make sure to output a BOM when UTF-16 requested
+ encoding.Assign(label);
+ } else if (!EncodingUtils::FindEncodingForLabelNoReplacement(label,
+ encoding)) {
+ return NS_ERROR_UCONV_NOCONV;
+ }
+ mConverter = EncodingUtils::EncoderForEncoding(encoding);
+
+ mOutStream = aOutStream;
+
+ int32_t behaviour = aReplacementChar ? nsIUnicodeEncoder::kOnError_Replace
+ : nsIUnicodeEncoder::kOnError_Signal;
+ return mConverter->
+ SetOutputErrorBehavior(behaviour,
+ nullptr,
+ aReplacementChar);
+}
+
+NS_IMETHODIMP
+nsConverterOutputStream::Write(uint32_t aCount, const char16_t* aChars,
+ bool* aSuccess)
+{
+ if (!mOutStream) {
+ NS_ASSERTION(!mConverter, "Closed streams shouldn't have converters");
+ return NS_BASE_STREAM_CLOSED;
+ }
+ NS_ASSERTION(mConverter, "Must have a converter when not closed");
+
+ int32_t inLen = aCount;
+
+ int32_t maxLen;
+ nsresult rv = mConverter->GetMaxLength(aChars, inLen, &maxLen);
+ NS_ENSURE_SUCCESS(rv, rv);
+
+ nsAutoCString buf;
+ buf.SetLength(maxLen);
+ if (buf.Length() != (uint32_t) maxLen)
+ return NS_ERROR_OUT_OF_MEMORY;
+
+ int32_t outLen = maxLen;
+ rv = mConverter->Convert(aChars, &inLen, buf.BeginWriting(), &outLen);
+ if (NS_FAILED(rv))
+ return rv;
+ if (rv == NS_ERROR_UENC_NOMAPPING) {
+ // Yes, NS_ERROR_UENC_NOMAPPING is a success code
+ return NS_ERROR_LOSS_OF_SIGNIFICANT_DATA;
+ }
+ NS_ASSERTION((uint32_t) inLen == aCount,
+ "Converter didn't consume all the data!");
+
+ uint32_t written;
+ rv = mOutStream->Write(buf.get(), outLen, &written);
+ *aSuccess = NS_SUCCEEDED(rv) && written == uint32_t(outLen);
+ return rv;
+
+}
+
+NS_IMETHODIMP
+nsConverterOutputStream::WriteString(const nsAString& aString, bool* aSuccess)
+{
+ int32_t inLen = aString.Length();
+ nsAString::const_iterator i;
+ aString.BeginReading(i);
+ return Write(inLen, i.get(), aSuccess);
+}
+
+NS_IMETHODIMP
+nsConverterOutputStream::Flush()
+{
+ if (!mOutStream)
+ return NS_OK; // Already closed.
+
+ char buf[1024];
+ int32_t size = sizeof(buf);
+ nsresult rv = mConverter->Finish(buf, &size);
+ NS_ASSERTION(rv != NS_OK_UENC_MOREOUTPUT,
+ "1024 bytes ought to be enough for everyone");
+ if (NS_FAILED(rv))
+ return rv;
+ if (size == 0)
+ return NS_OK;
+
+ uint32_t written;
+ rv = mOutStream->Write(buf, size, &written);
+ if (NS_FAILED(rv)) {
+ NS_WARNING("Flush() lost data!");
+ return rv;
+ }
+ if (written != uint32_t(size)) {
+ NS_WARNING("Flush() lost data!");
+ return NS_ERROR_LOSS_OF_SIGNIFICANT_DATA;
+ }
+ return rv;
+}
+
+NS_IMETHODIMP
+nsConverterOutputStream::Close()
+{
+ if (!mOutStream)
+ return NS_OK; // Already closed.
+
+ nsresult rv1 = Flush();
+
+ nsresult rv2 = mOutStream->Close();
+ mOutStream = nullptr;
+ mConverter = nullptr;
+ return NS_FAILED(rv1) ? rv1 : rv2;
+}
+