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/v8/Dictionary.h"
  36 #include "bindings/v8/ExceptionState.h"
  37 #include "wtf/text/CString.h"
  38 #include "wtf/text/TextEncodingRegistry.h"
  39
  40 namespace WebCore {
  41
  42 DEFINE_GC_INFO(TextEncoder);
  43
  44 PassRefPtrWillBeRawPtr<TextEncoder> TextEncoder::create(const String& utfLabel, ExceptionState& exceptionState)
  45 {
  46     const String& encodingLabel = utfLabel.isNull() ? String("utf-8") : utfLabel;
  47
  48     WTF::TextEncoding encoding(encodingLabel);
  49     if (!encoding.isValid()) {
  50         exceptionState.throwTypeError("The encoding label provided ('" + encodingLabel + "') is invalid.");
  51         return 0;
  52     }
  53
  54     String name(encoding.name());
  55     if (name != "UTF-8" && name != "UTF-16LE" && name != "UTF-16BE") {
  56         exceptionState.throwTypeError("The encoding provided ('" + encodingLabel + "') is not one of 'utf-8', 'utf-16', or 'utf-16be'.");
  57         return 0;
  58     }
  59
  60     return adoptRefWillBeNoop(new TextEncoder(encoding.name()));
  61 }
  62
  63 TextEncoder::TextEncoder(const String& encoding)
  64     : m_encoding(encoding)
  65     , m_codec(newTextCodec(m_encoding))
  66 {
  67 }
  68
  69 TextEncoder::~TextEncoder()
  70 {
  71 }
  72
  73 String TextEncoder::encoding() const
  74 {
  75     String name = String(m_encoding.name()).lower();
  76     ASSERT(name == "utf-8" || name == "utf-16le" || name == "utf-16be");
  77     return name;
  78 }
  79
  80 PassRefPtr<Uint8Array> TextEncoder::encode(const String& input, const Dictionary& options)
  81 {
  82     bool stream = false;
  83     options.get("stream", stream);
  84
  85     // FIXME: Not flushing is not supported by TextCodec for encode; add it or
  86     // handle split surrogates here.
  87
  88     CString result;
  89     if (input.is8Bit())
  90         result = m_codec->encode(input.characters8(), input.length(), WTF::QuestionMarksForUnencodables);
  91     else
  92         result = m_codec->encode(input.characters16(), input.length(), WTF::QuestionMarksForUnencodables);
  93
  94     const char* buffer = result.data();
  95     const unsigned char* unsignedBuffer = reinterpret_cast<const unsigned char*>(buffer);
  96
  97     return Uint8Array::create(unsignedBuffer, result.length());
  98 }
  99
 100 } // namespace WebCore