src/third_party/WebKit/Source/modules/encoding/TextEncoder.cpp

   1 /*
   2  * Copyright (C) 2013 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  *     * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *     * Redistributions in binary form must reproduce the above
  11  * copyright notice, this list of conditions and the following disclaimer
  12  * in the documentation and/or other materials provided with the
  13  * distribution.
  14  *     * Neither the name of Google Inc. nor the names of its
  15  * contributors may be used to endorse or promote products derived from
  16  * this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include "config.h"
  32
  33 #include "modules/encoding/TextEncoder.h"
  34
  35 #include "bindings/core/v8/ExceptionState.h"
  36 #include "wtf/text/CString.h"
  37 #include "wtf/text/TextEncodingRegistry.h"
  38
  39 namespace blink {
  40
  41 TextEncoder* TextEncoder::create(const String& utfLabel, ExceptionState& exceptionState)
  42 {
  43     WTF::TextEncoding encoding(utfLabel);
  44     if (!encoding.isValid()) {
  45         exceptionState.throwTypeError("The encoding label provided ('" + utfLabel + "') is invalid.");
  46         return 0;
  47     }
  48
  49     String name(encoding.name());
  50     if (name != "UTF-8" && name != "UTF-16LE" && name != "UTF-16BE") {
  51         exceptionState.throwTypeError("The encoding provided ('" + utfLabel + "') is not one of 'utf-8', 'utf-16', or 'utf-16be'.");
  52         return 0;
  53     }
  54
  55     return new TextEncoder(encoding);
  56 }
  57
  58 TextEncoder::TextEncoder(const WTF::TextEncoding& encoding)
  59     : m_encoding(encoding)
  60     , m_codec(newTextCodec(encoding))
  61 {
  62     ScriptWrappable::init(this);
  63 }
  64
  65 TextEncoder::~TextEncoder()
  66 {
  67 }
  68
  69 String TextEncoder::encoding() const
  70 {
  71     String name = String(m_encoding.name()).lower();
  72     ASSERT(name == "utf-8" || name == "utf-16le" || name == "utf-16be");
  73     return name;
  74 }
  75
  76 PassRefPtr<Uint8Array> TextEncoder::encode(const String& input)
  77 {
  78     CString result;
  79     if (input.is8Bit())
  80         result = m_codec->encode(input.characters8(), input.length(), WTF::QuestionMarksForUnencodables);
  81     else
  82         result = m_codec->encode(input.characters16(), input.length(), WTF::QuestionMarksForUnencodables);
  83
  84     const char* buffer = result.data();
  85     const unsigned char* unsignedBuffer = reinterpret_cast<const unsigned char*>(buffer);
  86
  87     return Uint8Array::create(unsignedBuffer, result.length());
  88 }
  89
  90 } // namespace blink