1package eu.siacs.conversations.xml;
2
3import android.util.Log;
4import android.util.Xml;
5
6import eu.siacs.conversations.Config;
7
8import im.conversations.android.xmpp.ExtensionFactory;
9import im.conversations.android.xmpp.model.Extension;
10import im.conversations.android.xmpp.model.StreamElement;
11
12import org.xmlpull.v1.XmlPullParser;
13import org.xmlpull.v1.XmlPullParserException;
14
15import java.io.Closeable;
16import java.io.IOException;
17import java.io.InputStream;
18import java.io.InputStreamReader;
19
20public class XmlReader implements Closeable {
21 private final XmlPullParser parser;
22 private InputStream is;
23
24 public XmlReader() {
25 this.parser = Xml.newPullParser();
26 try {
27 this.parser.setFeature(XmlPullParser.FEATURE_PROCESS_NAMESPACES, true);
28 } catch (XmlPullParserException e) {
29 Log.d(Config.LOGTAG, "error setting namespace feature on parser");
30 }
31 }
32
33 public void setInputStream(InputStream inputStream) throws IOException {
34 if (inputStream == null) {
35 throw new IOException();
36 }
37 this.is = inputStream;
38 try {
39 parser.setInput(new InputStreamReader(this.is));
40 } catch (XmlPullParserException e) {
41 throw new IOException("error resetting parser");
42 }
43 }
44
45 public void reset() throws IOException {
46 if (this.is == null) {
47 throw new IOException();
48 }
49 try {
50 parser.setInput(new InputStreamReader(this.is));
51 } catch (XmlPullParserException e) {
52 throw new IOException("error resetting parser");
53 }
54 }
55
56 @Override
57 public void close() {
58 this.is = null;
59 }
60
61 public Tag readTag() throws IOException {
62 try {
63 while (this.is != null && parser.next() != XmlPullParser.END_DOCUMENT) {
64 if (parser.getEventType() == XmlPullParser.START_TAG) {
65 Tag tag = Tag.start(parser.getName());
66 final String xmlns = parser.getNamespace();
67 for (int i = 0; i < parser.getAttributeCount(); ++i) {
68 final var prefix = parser.getAttributePrefix(i);
69 final var ns = parser.getAttributeNamespace(i);
70 String name;
71 if ("xml".equals(prefix)) {
72 name = "xml:" + parser.getAttributeName(i);
73 } else if (ns != null && !ns.isEmpty()) {
74 name = "{" + ns + "}" + parser.getAttributeName(i);
75 } else {
76 name = parser.getAttributeName(i);
77 }
78 tag.setAttribute(name,parser.getAttributeValue(i));
79 }
80 if (xmlns != null) {
81 tag.setAttribute("xmlns", xmlns);
82 }
83 return tag;
84 } else if (parser.getEventType() == XmlPullParser.END_TAG) {
85 return Tag.end(parser.getName());
86 } else if (parser.getEventType() == XmlPullParser.TEXT) {
87 return Tag.no(parser.getText());
88 }
89 }
90
91 } catch (Throwable throwable) {
92 throw new IOException("xml parser mishandled "+throwable.getClass().getSimpleName()+"("+throwable.getMessage()+")", throwable);
93 }
94 return null;
95 }
96
97 public <T extends StreamElement> T readElement(final Tag current, final Class<T> clazz)
98 throws IOException {
99 final Element element = readElement(current);
100 if (clazz.isInstance(element)) {
101 return clazz.cast(element);
102 }
103 throw new IOException(
104 String.format("Read unexpected {%s}%s", element.getNamespace(), element.getName()));
105 }
106
107 public Element readElement(final Tag currentTag) throws IOException {
108 final var attributes = currentTag.getAttributes();
109 final var namespace = attributes.get("xmlns");
110 final var name = currentTag.getName();
111 final Element element = ExtensionFactory.create(name, namespace);
112 element.setAttributes(currentTag.getAttributes());
113 Tag nextTag = this.readTag();
114 if (nextTag == null) {
115 throw new IOException("interrupted mid tag");
116 }
117 while (!nextTag.isEnd(element.getName())) {
118 if (nextTag.isNo()) {
119 if (nextTag.getName() != null) element.addChild(new TextNode(nextTag.getName()));
120 } else {
121 Element child = this.readElement(nextTag);
122 element.addChild(child);
123 }
124 nextTag = this.readTag();
125 if (nextTag == null) {
126 throw new IOException("interrupted mid tag");
127 }
128 }
129 return element;
130 }
131}