epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2006 The Android Open Source Project |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 8 | #include "SkXMLWriter.h" |
| 9 | #include "SkStream.h" |
| 10 | |
| 11 | SkXMLWriter::SkXMLWriter(bool doEscapeMarkup) : fDoEscapeMarkup(doEscapeMarkup) |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 12 | {} |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 13 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 14 | SkXMLWriter::~SkXMLWriter() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 15 | SkASSERT(fElems.count() == 0); |
| 16 | } |
| 17 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 18 | void SkXMLWriter::flush() { |
| 19 | while (fElems.count()) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 20 | this->endElement(); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 21 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 22 | } |
| 23 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 24 | void SkXMLWriter::addAttribute(const char name[], const char value[]) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 25 | this->addAttributeLen(name, value, strlen(value)); |
| 26 | } |
| 27 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 28 | void SkXMLWriter::addS32Attribute(const char name[], int32_t value) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 29 | SkString tmp; |
| 30 | tmp.appendS32(value); |
| 31 | this->addAttribute(name, tmp.c_str()); |
| 32 | } |
| 33 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 34 | void SkXMLWriter::addHexAttribute(const char name[], uint32_t value, int minDigits) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 35 | SkString tmp("0x"); |
| 36 | tmp.appendHex(value, minDigits); |
| 37 | this->addAttribute(name, tmp.c_str()); |
| 38 | } |
| 39 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 40 | void SkXMLWriter::addScalarAttribute(const char name[], SkScalar value) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 41 | SkString tmp; |
| 42 | tmp.appendScalar(value); |
| 43 | this->addAttribute(name, tmp.c_str()); |
| 44 | } |
| 45 | |
reed | e73da40 | 2015-02-04 18:29:27 -0800 | [diff] [blame] | 46 | void SkXMLWriter::addText(const char text[], size_t length) { |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 47 | if (fElems.isEmpty()) { |
| 48 | return; |
| 49 | } |
reed | e73da40 | 2015-02-04 18:29:27 -0800 | [diff] [blame] | 50 | |
| 51 | this->onAddText(text, length); |
| 52 | |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 53 | fElems.top()->fHasText = true; |
| 54 | } |
| 55 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 56 | void SkXMLWriter::doEnd(Elem* elem) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 57 | delete elem; |
| 58 | } |
| 59 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 60 | bool SkXMLWriter::doStart(const char name[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 61 | int level = fElems.count(); |
| 62 | bool firstChild = level > 0 && !fElems[level-1]->fHasChildren; |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 63 | if (firstChild) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 64 | fElems[level-1]->fHasChildren = true; |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 65 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 66 | Elem** elem = fElems.push(); |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 67 | *elem = new Elem(name, length); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 68 | return firstChild; |
| 69 | } |
| 70 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 71 | SkXMLWriter::Elem* SkXMLWriter::getEnd() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 72 | Elem* elem; |
| 73 | fElems.pop(&elem); |
| 74 | return elem; |
| 75 | } |
| 76 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 77 | const char* SkXMLWriter::getHeader() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 78 | static const char gHeader[] = "<?xml version=\"1.0\" encoding=\"utf-8\" ?>"; |
| 79 | return gHeader; |
| 80 | } |
| 81 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 82 | void SkXMLWriter::startElement(const char name[]) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 83 | this->startElementLen(name, strlen(name)); |
| 84 | } |
| 85 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 86 | static const char* escape_char(char c, char storage[2]) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 87 | static const char* gEscapeChars[] = { |
| 88 | "<<", |
| 89 | ">>", |
| 90 | //"\""", |
| 91 | //"''", |
| 92 | "&&" |
| 93 | }; |
| 94 | |
| 95 | const char** array = gEscapeChars; |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 96 | for (unsigned i = 0; i < SK_ARRAY_COUNT(gEscapeChars); i++) { |
| 97 | if (array[i][0] == c) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 98 | return &array[i][1]; |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 99 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 100 | } |
| 101 | storage[0] = c; |
| 102 | storage[1] = 0; |
| 103 | return storage; |
| 104 | } |
| 105 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 106 | static size_t escape_markup(char dst[], const char src[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 107 | size_t extra = 0; |
| 108 | const char* stop = src + length; |
| 109 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 110 | while (src < stop) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 111 | char orig[2]; |
| 112 | const char* seq = escape_char(*src, orig); |
| 113 | size_t seqSize = strlen(seq); |
| 114 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 115 | if (dst) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 116 | memcpy(dst, seq, seqSize); |
| 117 | dst += seqSize; |
| 118 | } |
| 119 | |
| 120 | // now record the extra size needed |
| 121 | extra += seqSize - 1; // minus one to subtract the original char |
| 122 | |
| 123 | // bump to the next src char |
| 124 | src += 1; |
| 125 | } |
| 126 | return extra; |
| 127 | } |
| 128 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 129 | void SkXMLWriter::addAttributeLen(const char name[], const char value[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 130 | SkString valueStr; |
| 131 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 132 | if (fDoEscapeMarkup) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 133 | size_t extra = escape_markup(nullptr, value, length); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 134 | if (extra) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 135 | valueStr.resize(length + extra); |
| 136 | (void)escape_markup(valueStr.writable_str(), value, length); |
| 137 | value = valueStr.c_str(); |
| 138 | length += extra; |
| 139 | } |
| 140 | } |
| 141 | this->onAddAttributeLen(name, value, length); |
| 142 | } |
| 143 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 144 | void SkXMLWriter::startElementLen(const char elem[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 145 | this->onStartElementLen(elem, length); |
| 146 | } |
| 147 | |
| 148 | //////////////////////////////////////////////////////////////////////////////////////// |
| 149 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 150 | static void write_dom(const SkDOM& dom, const SkDOM::Node* node, SkXMLWriter* w, bool skipRoot) { |
| 151 | if (!skipRoot) { |
fmalita | 7a04869 | 2015-02-20 13:54:40 -0800 | [diff] [blame] | 152 | const char* elem = dom.getName(node); |
| 153 | if (dom.getType(node) == SkDOM::kText_Type) { |
| 154 | SkASSERT(dom.countChildren(node) == 0); |
| 155 | w->addText(elem, strlen(elem)); |
| 156 | return; |
| 157 | } |
| 158 | |
| 159 | w->startElement(elem); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 160 | |
| 161 | SkDOM::AttrIter iter(dom, node); |
| 162 | const char* name; |
| 163 | const char* value; |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 164 | while ((name = iter.next(&value)) != nullptr) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 165 | w->addAttribute(name, value); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 166 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 167 | } |
| 168 | |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 169 | node = dom.getFirstChild(node, nullptr); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 170 | while (node) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 171 | write_dom(dom, node, w, false); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 172 | node = dom.getNextSibling(node, nullptr); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 173 | } |
| 174 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 175 | if (!skipRoot) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 176 | w->endElement(); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 177 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 178 | } |
| 179 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 180 | void SkXMLWriter::writeDOM(const SkDOM& dom, const SkDOM::Node* node, bool skipRoot) { |
| 181 | if (node) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 182 | write_dom(dom, node, this, skipRoot); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 183 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 184 | } |
| 185 | |
| 186 | void SkXMLWriter::writeHeader() |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 187 | {} |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 188 | |
| 189 | // SkXMLStreamWriter |
| 190 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 191 | static void tab(SkWStream& stream, int level) { |
| 192 | for (int i = 0; i < level; i++) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 193 | stream.writeText("\t"); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 194 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 195 | } |
| 196 | |
| 197 | SkXMLStreamWriter::SkXMLStreamWriter(SkWStream* stream) : fStream(*stream) |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 198 | {} |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 199 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 200 | SkXMLStreamWriter::~SkXMLStreamWriter() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 201 | this->flush(); |
| 202 | } |
| 203 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 204 | void SkXMLStreamWriter::onAddAttributeLen(const char name[], const char value[], size_t length) { |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 205 | SkASSERT(!fElems.top()->fHasChildren && !fElems.top()->fHasText); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 206 | fStream.writeText(" "); |
| 207 | fStream.writeText(name); |
| 208 | fStream.writeText("=\""); |
| 209 | fStream.write(value, length); |
| 210 | fStream.writeText("\""); |
| 211 | } |
| 212 | |
reed | e73da40 | 2015-02-04 18:29:27 -0800 | [diff] [blame] | 213 | void SkXMLStreamWriter::onAddText(const char text[], size_t length) { |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 214 | Elem* elem = fElems.top(); |
| 215 | |
| 216 | if (!elem->fHasChildren && !elem->fHasText) { |
| 217 | fStream.writeText(">"); |
| 218 | fStream.newline(); |
| 219 | } |
| 220 | |
| 221 | tab(fStream, fElems.count() + 1); |
reed | e73da40 | 2015-02-04 18:29:27 -0800 | [diff] [blame] | 222 | fStream.write(text, length); |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 223 | fStream.newline(); |
| 224 | } |
| 225 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 226 | void SkXMLStreamWriter::onEndElement() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 227 | Elem* elem = getEnd(); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 228 | if (elem->fHasChildren || elem->fHasText) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 229 | tab(fStream, fElems.count()); |
| 230 | fStream.writeText("</"); |
| 231 | fStream.writeText(elem->fName.c_str()); |
| 232 | fStream.writeText(">"); |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 233 | } else { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 234 | fStream.writeText("/>"); |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 235 | } |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 236 | fStream.newline(); |
| 237 | doEnd(elem); |
| 238 | } |
| 239 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 240 | void SkXMLStreamWriter::onStartElementLen(const char name[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 241 | int level = fElems.count(); |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 242 | if (this->doStart(name, length)) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 243 | // the first child, need to close with > |
| 244 | fStream.writeText(">"); |
| 245 | fStream.newline(); |
| 246 | } |
| 247 | |
| 248 | tab(fStream, level); |
| 249 | fStream.writeText("<"); |
| 250 | fStream.write(name, length); |
| 251 | } |
| 252 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 253 | void SkXMLStreamWriter::writeHeader() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 254 | const char* header = getHeader(); |
| 255 | fStream.write(header, strlen(header)); |
| 256 | fStream.newline(); |
| 257 | } |
| 258 | |
| 259 | //////////////////////////////////////////////////////////////////////////////////////////////// |
| 260 | |
| 261 | #include "SkXMLParser.h" |
| 262 | |
| 263 | SkXMLParserWriter::SkXMLParserWriter(SkXMLParser* parser) |
| 264 | : SkXMLWriter(false), fParser(*parser) |
| 265 | { |
| 266 | } |
| 267 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 268 | SkXMLParserWriter::~SkXMLParserWriter() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 269 | this->flush(); |
| 270 | } |
| 271 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 272 | void SkXMLParserWriter::onAddAttributeLen(const char name[], const char value[], size_t length) { |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 273 | SkASSERT(fElems.count() == 0 || (!fElems.top()->fHasChildren && !fElems.top()->fHasText)); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 274 | SkString str(value, length); |
| 275 | fParser.addAttribute(name, str.c_str()); |
| 276 | } |
| 277 | |
reed | e73da40 | 2015-02-04 18:29:27 -0800 | [diff] [blame] | 278 | void SkXMLParserWriter::onAddText(const char text[], size_t length) { |
| 279 | fParser.text(text, SkToInt(length)); |
fmalita | fe3f260 | 2015-02-03 17:47:12 -0800 | [diff] [blame] | 280 | } |
| 281 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 282 | void SkXMLParserWriter::onEndElement() { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 283 | Elem* elem = this->getEnd(); |
| 284 | fParser.endElement(elem->fName.c_str()); |
| 285 | this->doEnd(elem); |
| 286 | } |
| 287 | |
Mike Reed | c3063e5 | 2017-01-07 16:16:02 -0500 | [diff] [blame] | 288 | void SkXMLParserWriter::onStartElementLen(const char name[], size_t length) { |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 289 | (void)this->doStart(name, length); |
| 290 | SkString str(name, length); |
| 291 | fParser.startElement(str.c_str()); |
| 292 | } |