d41f46294d3604eb604cb747016ce543617edddf
[WebKit-https.git] / Source / WTF / wtf / text / win / WCharStringExtras.h
1 /*
2  * Copyright (C) 2017 Konstantin Tokarev <annulen@yandex.ru>
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #include <wchar.h>
29 #include <wtf/text/WTFString.h>
30
31 namespace WTF {
32
33 inline Vector<wchar_t> stringToNullTerminatedWChar(const String& string)
34 {
35     Vector<wchar_t> result;
36
37     if (!string.isNull()) {
38         result.reserveInitialCapacity(string.impl()->length() + 1);
39
40         if (string.is8Bit()) {
41             const LChar* characters8 = string.impl()->characters8();
42             for (size_t i = 0; i < string.impl()->length(); ++i)
43                 result.uncheckedAppend(characters8[i]);
44         } else {
45             const UChar* characters16 = string.impl()->characters16();
46             result.append(characters16, string.impl()->length());
47         }
48
49         result.append(0);
50     }
51
52     return result;
53 }
54
55 inline String wcharToString(const wchar_t* characters, unsigned length)
56 {
57     static_assert(sizeof(wchar_t) == sizeof(UChar), "We assume wchar_t and UChar have the same size");
58     return String(reinterpret_cast<const UChar*>(characters), length);
59 }
60
61 inline String nullTerminatedWCharToString(const wchar_t* characters)
62 {
63     return wcharToString(characters, wcslen(characters));
64 }
65
66 } // namespace WTF
67
68 using WTF::stringToNullTerminatedWChar;
69 using WTF::wcharToString;
70 using WTF::nullTerminatedWCharToString;