1 /*
2 * Copyright (C) 2009 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 #include "WebEntities.h"
33
34 #include "WebString.h"
35
36 #include <string.h>
37 #include <wtf/HashMap.h>
38 #include <wtf/text/StringBuilder.h>
39 #include <wtf/text/WTFString.h>
40
41 namespace WebKit {
42
WebEntities(bool xmlEntities)43 WebEntities::WebEntities(bool xmlEntities)
44 {
45 ASSERT(m_entitiesMap.isEmpty());
46 m_entitiesMap.set(0x003c, "lt");
47 m_entitiesMap.set(0x003e, "gt");
48 m_entitiesMap.set(0x0026, "amp");
49 m_entitiesMap.set(0x0027, "apos");
50 m_entitiesMap.set(0x0022, "quot");
51 // We add #39 for test-compatibility reason.
52 if (!xmlEntities)
53 m_entitiesMap.set(0x0027, String("#39"));
54 }
55
entityNameByCode(int code) const56 String WebEntities::entityNameByCode(int code) const
57 {
58 if (m_entitiesMap.contains(code))
59 return m_entitiesMap.get(code);
60 return "";
61 }
62
convertEntitiesInString(const String & value) const63 String WebEntities::convertEntitiesInString(const String& value) const
64 {
65 unsigned len = value.length();
66 const UChar* startPos = value.characters();
67 const UChar* curPos = startPos;
68
69 // FIXME: Optimize - create StringBuilder only if value has any entities.
70 StringBuilder result;
71 while (len--) {
72 if (m_entitiesMap.contains(*curPos)) {
73 // Append content before entity code.
74 if (curPos > startPos)
75 result.append(String(startPos, curPos - startPos));
76 result.append('&');
77 result.append(m_entitiesMap.get(*curPos));
78 result.append(';');
79 startPos = ++curPos;
80 } else
81 curPos++;
82 }
83 // Append the remaining content.
84 if (curPos > startPos)
85 result.append(String(startPos, curPos - startPos));
86
87 return result.toString();
88 }
89
90 } // namespace WebKit
91