diff options
Diffstat (limited to 'dom/localstorage/LSValue.cpp')
| -rw-r--r-- | dom/localstorage/LSValue.cpp | 202 |
1 files changed, 202 insertions, 0 deletions
diff --git a/dom/localstorage/LSValue.cpp b/dom/localstorage/LSValue.cpp new file mode 100644 index 0000000000..03637fdeba --- /dev/null +++ b/dom/localstorage/LSValue.cpp @@ -0,0 +1,202 @@ +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set ts=2 et sw=2 tw=80: */ +/* 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 "mozilla/dom/LSValue.h" + +#include "mozIStorageStatement.h" +#include "mozilla/dom/SnappyUtils.h" +#include "mozilla/fallible.h" +#include "mozilla/TextUtils.h" +#include "nsDebug.h" +#include "nsError.h" + +namespace mozilla::dom { + +namespace { + +bool PutStringBytesToCString(const nsAString& aSrc, nsCString& aDest) { + const char16_t* bufferData; + const size_t byteLength = sizeof(char16_t) * aSrc.GetData(&bufferData); + + char* destDataPtr; + const auto newLength = aDest.GetMutableData(&destDataPtr, byteLength); + if (newLength != byteLength) { + return false; + } + std::memcpy(static_cast<void*>(destDataPtr), + static_cast<const void*>(bufferData), byteLength); + + return true; +} + +template <class T> +using TypeBufferResult = Result<std::pair<T, nsCString>, nsresult>; + +} // namespace + +bool PutCStringBytesToString(const nsACString& aSrc, nsString& aDest) { + const char* bufferData; + const size_t byteLength = aSrc.GetData(&bufferData); + const size_t shortLength = byteLength / sizeof(char16_t); + + char16_t* destDataPtr; + const auto newLength = aDest.GetMutableData(&destDataPtr, shortLength); + if (newLength != shortLength) { + return false; + } + + std::memcpy(static_cast<void*>(destDataPtr), + static_cast<const void*>(bufferData), byteLength); + return true; +} + +LSValue::Converter::Converter(const LSValue& aValue) { + using ConversionType = LSValue::ConversionType; + using CompressionType = LSValue::CompressionType; + + if (aValue.mBuffer.IsVoid()) { + mBuffer.SetIsVoid(true); + return; + } + + const CompressionType compressionType = aValue.GetCompressionType(); + const ConversionType conversionType = aValue.GetConversionType(); + + const nsCString uncompressed = [compressionType, &aValue]() { + if (CompressionType::UNCOMPRESSED != compressionType) { + nsCString buffer; + MOZ_ASSERT(CompressionType::SNAPPY == compressionType); + if (NS_WARN_IF(!SnappyUncompress(aValue.mBuffer, buffer))) { + buffer.Truncate(); + } + return buffer; + } + + return aValue.mBuffer; + }(); + + if (ConversionType::NONE != conversionType) { + MOZ_ASSERT(ConversionType::UTF16_UTF8 == conversionType); + if (NS_WARN_IF(!CopyUTF8toUTF16(uncompressed, mBuffer, fallible))) { + mBuffer.SetIsVoid(true); + } + return; + } + + if (NS_WARN_IF(!PutCStringBytesToString(uncompressed, mBuffer))) { + mBuffer.SetIsVoid(true); + } +} + +bool LSValue::InitFromString(const nsAString& aBuffer) { + MOZ_ASSERT(mBuffer.IsVoid()); + MOZ_ASSERT(!mUTF16Length); + MOZ_ASSERT(ConversionType::NONE == mConversionType); + MOZ_ASSERT(CompressionType::UNCOMPRESSED == mCompressionType); + + if (aBuffer.IsVoid()) { + return true; + } + + const uint32_t utf16Length = aBuffer.Length(); + + const auto conversionRes = [&aBuffer]() -> TypeBufferResult<ConversionType> { + nsCString converted; + + if (Utf16ValidUpTo(aBuffer) == aBuffer.Length()) { + if (NS_WARN_IF(!CopyUTF16toUTF8(aBuffer, converted, fallible))) { + return Err(NS_ERROR_OUT_OF_MEMORY); + } + return std::pair{ConversionType::UTF16_UTF8, std::move(converted)}; + } + + if (NS_WARN_IF(!PutStringBytesToCString(aBuffer, converted))) { + return Err(NS_ERROR_OUT_OF_MEMORY); + } + return std::pair{ConversionType::NONE, std::move(converted)}; + }(); + + if (conversionRes.isErr()) { + return false; + } + + const auto& [conversionType, converted] = conversionRes.inspect(); + + const auto compressionRes = + [&converted = converted]() -> TypeBufferResult<CompressionType> { + nsCString compressed; + if (NS_WARN_IF(!SnappyCompress(converted, compressed))) { + return Err(NS_ERROR_OUT_OF_MEMORY); + } + if (!compressed.IsVoid()) { + return std::pair{CompressionType::SNAPPY, std::move(compressed)}; + } + + compressed = converted; + return std::pair{CompressionType::UNCOMPRESSED, std::move(compressed)}; + }(); + + if (compressionRes.isErr()) { + return false; + } + + const auto& [compressionType, compressed] = compressionRes.inspect(); + + mBuffer = compressed; + mUTF16Length = utf16Length; + mConversionType = conversionType; + mCompressionType = compressionType; + + return true; +} + +nsresult LSValue::InitFromStatement(mozIStorageStatement* aStatement, + uint32_t aIndex) { + MOZ_ASSERT(aStatement); + MOZ_ASSERT(mBuffer.IsVoid()); + MOZ_ASSERT(!mUTF16Length); + MOZ_ASSERT(ConversionType::NONE == mConversionType); + MOZ_ASSERT(CompressionType::UNCOMPRESSED == mCompressionType); + + int32_t utf16Length; + nsresult rv = aStatement->GetInt32(aIndex, &utf16Length); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + + int32_t conversionType; + rv = aStatement->GetInt32(aIndex + 1, &conversionType); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + + int32_t compressionType; + rv = aStatement->GetInt32(aIndex + 2, &compressionType); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + + nsCString buffer; + rv = aStatement->GetBlobAsUTF8String(aIndex + 3, buffer); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + + mBuffer = buffer; + mUTF16Length = static_cast<uint32_t>(utf16Length); + mConversionType = static_cast<decltype(mConversionType)>(conversionType); + mCompressionType = static_cast<decltype(mCompressionType)>(compressionType); + + return NS_OK; +} + +const LSValue& VoidLSValue() { + static const LSValue sVoidLSValue; + + return sVoidLSValue; +} + +} // namespace mozilla::dom |