/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* vim: set ts=8 sts=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 "CryptoBuffer.h" #include "secitem.h" #include "mozilla/Base64.h" #include "mozilla/dom/UnionTypes.h" namespace mozilla { namespace dom { uint8_t* CryptoBuffer::Assign(const CryptoBuffer& aData) { // Same as in nsTArray_Impl::operator=, but return the value // returned from ReplaceElementsAt to enable OOM detection return ReplaceElementsAt(0, Length(), aData.Elements(), aData.Length(), fallible); } uint8_t* CryptoBuffer::Assign(const uint8_t* aData, uint32_t aLength) { return ReplaceElementsAt(0, Length(), aData, aLength, fallible); } uint8_t* CryptoBuffer::Assign(const nsACString& aString) { return Assign(reinterpret_cast<uint8_t const*>(aString.BeginReading()), aString.Length()); } uint8_t* CryptoBuffer::Assign(const SECItem* aItem) { MOZ_ASSERT(aItem); return Assign(aItem->data, aItem->len); } uint8_t* CryptoBuffer::Assign(const InfallibleTArray<uint8_t>& aData) { return ReplaceElementsAt(0, Length(), aData.Elements(), aData.Length(), fallible); } uint8_t* CryptoBuffer::Assign(const ArrayBuffer& aData) { aData.ComputeLengthAndData(); return Assign(aData.Data(), aData.Length()); } uint8_t* CryptoBuffer::Assign(const ArrayBufferView& aData) { aData.ComputeLengthAndData(); return Assign(aData.Data(), aData.Length()); } uint8_t* CryptoBuffer::Assign(const ArrayBufferViewOrArrayBuffer& aData) { if (aData.IsArrayBufferView()) { return Assign(aData.GetAsArrayBufferView()); } else if (aData.IsArrayBuffer()) { return Assign(aData.GetAsArrayBuffer()); } // If your union is uninitialized, something's wrong MOZ_ASSERT(false); Clear(); return nullptr; } uint8_t* CryptoBuffer::Assign(const OwningArrayBufferViewOrArrayBuffer& aData) { if (aData.IsArrayBufferView()) { return Assign(aData.GetAsArrayBufferView()); } else if (aData.IsArrayBuffer()) { return Assign(aData.GetAsArrayBuffer()); } // If your union is uninitialized, something's wrong MOZ_ASSERT(false); Clear(); return nullptr; } uint8_t* CryptoBuffer::AppendSECItem(const SECItem* aItem) { MOZ_ASSERT(aItem); return AppendElements(aItem->data, aItem->len, fallible); } uint8_t* CryptoBuffer::AppendSECItem(const SECItem& aItem) { return AppendElements(aItem.data, aItem.len, fallible); } // Helpers to encode/decode JWK's special flavor of Base64 // * No whitespace // * No padding // * URL-safe character set nsresult CryptoBuffer::FromJwkBase64(const nsString& aBase64) { NS_ConvertUTF16toUTF8 temp(aBase64); temp.StripWhitespace(); // JWK prohibits padding per RFC 7515, section 2. nsresult rv = Base64URLDecode(temp, Base64URLDecodePaddingPolicy::Reject, *this); NS_ENSURE_SUCCESS(rv, rv); return NS_OK; } nsresult CryptoBuffer::ToJwkBase64(nsString& aBase64) { // Shortcut for the empty octet string if (Length() == 0) { aBase64.Truncate(); return NS_OK; } nsAutoCString base64; nsresult rv = Base64URLEncode(Length(), Elements(), Base64URLEncodePaddingPolicy::Omit, base64); NS_ENSURE_SUCCESS(rv, rv); CopyASCIItoUTF16(base64, aBase64); return NS_OK; } bool CryptoBuffer::ToSECItem(PLArenaPool *aArena, SECItem* aItem) const { aItem->type = siBuffer; aItem->data = nullptr; if (!::SECITEM_AllocItem(aArena, aItem, Length())) { return false; } memcpy(aItem->data, Elements(), Length()); return true; } JSObject* CryptoBuffer::ToUint8Array(JSContext* aCx) const { return Uint8Array::Create(aCx, Length(), Elements()); } bool CryptoBuffer::ToNewUnsignedBuffer(uint8_t** aBuf, uint32_t* aBufLen) const { MOZ_ASSERT(aBuf); MOZ_ASSERT(aBufLen); uint32_t dataLen = Length(); uint8_t* tmp = reinterpret_cast<uint8_t*>(moz_xmalloc(dataLen)); if (NS_WARN_IF(!tmp)) { return false; } memcpy(tmp, Elements(), dataLen); *aBuf = tmp; *aBufLen = dataLen; return true; } // "BigInt" comes from the WebCrypto spec // ("unsigned long" isn't very "big", of course) // Likewise, the spec calls for big-endian ints bool CryptoBuffer::GetBigIntValue(unsigned long& aRetVal) { if (Length() > sizeof(aRetVal)) { return false; } aRetVal = 0; for (size_t i=0; i < Length(); ++i) { aRetVal = (aRetVal << 8) + ElementAt(i); } return true; } } // namespace dom } // namespace mozilla