1package de.gultsch.chat.xmpp;
2
3import java.io.IOException;
4import java.io.InputStream;
5import java.io.OutputStream;
6import java.math.BigInteger;
7import java.net.Socket;
8import java.net.UnknownHostException;
9import java.security.SecureRandom;
10import java.util.Hashtable;
11
12import javax.net.ssl.SSLSocket;
13import javax.net.ssl.SSLSocketFactory;
14
15import org.xmlpull.v1.XmlPullParserException;
16
17import android.os.PowerManager;
18import android.util.Log;
19import de.gultsch.chat.entities.Account;
20import de.gultsch.chat.utils.SASL;
21import de.gultsch.chat.xml.Element;
22import de.gultsch.chat.xml.Tag;
23import de.gultsch.chat.xml.XmlReader;
24import de.gultsch.chat.xml.TagWriter;
25
26public class XmppConnection implements Runnable {
27
28 protected Account account;
29 private static final String LOGTAG = "xmppService";
30
31 private PowerManager.WakeLock wakeLock;
32
33 private SecureRandom random = new SecureRandom();
34
35 private Socket socket;
36 private XmlReader tagReader;
37 private TagWriter tagWriter;
38
39 private boolean isTlsEncrypted = false;
40 private boolean isAuthenticated = false;
41 private boolean shouldUseTLS = false;
42 private boolean shouldReConnect = true;
43 private boolean shouldBind = true;
44 private boolean shouldAuthenticate = true;
45 private Element streamFeatures;
46
47 private static final int PACKET_IQ = 0;
48 private static final int PACKET_MESSAGE = 1;
49 private static final int PACKET_PRESENCE = 2;
50
51 private Hashtable<String, OnIqPacketReceived> iqPacketCallbacks = new Hashtable<String, OnIqPacketReceived>();
52
53 public XmppConnection(Account account, PowerManager pm) {
54 this.account = account;
55 wakeLock = pm.newWakeLock(PowerManager.PARTIAL_WAKE_LOCK,
56 "XmppConnection");
57 tagReader = new XmlReader(wakeLock);
58 tagWriter = new TagWriter();
59 }
60
61 protected void connect() {
62 try {
63 socket = new Socket(account.getServer(), 5222);
64 Log.d(LOGTAG, "starting new socket");
65 OutputStream out = socket.getOutputStream();
66 tagWriter.setOutputStream(out);
67 InputStream in = socket.getInputStream();
68 tagReader.setInputStream(in);
69 tagWriter.beginDocument();
70 sendStartStream();
71 Tag nextTag;
72 while ((nextTag = tagReader.readTag()) != null) {
73 if (nextTag.isStart("stream")) {
74 processStream(nextTag);
75 } else {
76 Log.d(LOGTAG, "found unexpected tag: " + nextTag.getName());
77 return;
78 }
79 }
80 } catch (UnknownHostException e) {
81 Log.d(LOGTAG, "error during connect. unknown host");
82 return;
83 } catch (IOException e) {
84 Log.d(LOGTAG, "error during connect. io exception. falscher port?");
85 return;
86 } catch (XmlPullParserException e) {
87 Log.d(LOGTAG,"xml exception "+e.getMessage());
88 return;
89 }
90 }
91
92 @Override
93 public void run() {
94 while(shouldReConnect) {
95 connect();
96 }
97 }
98
99 private void processStream(Tag currentTag) throws XmlPullParserException,
100 IOException {
101 Log.d(LOGTAG, "process Stream");
102 Tag nextTag;
103 while (!(nextTag = tagReader.readTag()).isEnd("stream")) {
104 if (nextTag.isStart("error")) {
105 processStreamError(nextTag);
106 } else if (nextTag.isStart("features")) {
107 processStreamFeatures(nextTag);
108 } else if (nextTag.isStart("proceed")) {
109 switchOverToTls(nextTag);
110 } else if (nextTag.isStart("success")) {
111 isAuthenticated = true;
112 Log.d(LOGTAG,"read success tag in stream. reset again");
113 tagReader.readTag();
114 tagReader.reset();
115 sendStartStream();
116 processStream(tagReader.readTag());
117 } else if (nextTag.isStart("iq")) {
118 Log.d(LOGTAG,processIq(nextTag).toString());
119 } else if (nextTag.isStart("message")) {
120 Log.d(LOGTAG,processMessage(nextTag).toString());
121 } else if (nextTag.isStart("presence")) {
122 Log.d(LOGTAG,processPresence(nextTag).toString());
123 } else {
124 Log.d(LOGTAG, "found unexpected tag: " + nextTag.getName()
125 + " as child of " + currentTag.getName());
126 }
127 }
128 }
129
130 private Element processPacket(Tag currentTag, int packetType) throws XmlPullParserException, IOException {
131 Element element;
132 switch (packetType) {
133 case PACKET_IQ:
134 element = new IqPacket();
135 break;
136 case PACKET_MESSAGE:
137 element = new MessagePacket();
138 break;
139 case PACKET_PRESENCE:
140 element = new PresencePacket();
141 break;
142 default:
143 return null;
144 }
145 element.setAttributes(currentTag.getAttributes());
146 Tag nextTag = tagReader.readTag();
147 while(!nextTag.isEnd(element.getName())) {
148 if (!nextTag.isNo()) {
149 Element child = tagReader.readElement(nextTag);
150 element.addChild(child);
151 }
152 nextTag = tagReader.readTag();
153 }
154 return element;
155 }
156
157
158 private IqPacket processIq(Tag currentTag) throws XmlPullParserException, IOException {
159 IqPacket packet = (IqPacket) processPacket(currentTag,PACKET_IQ);
160 if (iqPacketCallbacks.containsKey(packet.getId())) {
161 iqPacketCallbacks.get(packet.getId()).onIqPacketReceived(packet);
162 iqPacketCallbacks.remove(packet.getId());
163 }
164 return packet;
165 }
166
167 private MessagePacket processMessage(Tag currentTag) throws XmlPullParserException, IOException {
168 return (MessagePacket) processPacket(currentTag, PACKET_MESSAGE);
169 }
170
171 private PresencePacket processPresence(Tag currentTag) throws XmlPullParserException, IOException {
172 return (PresencePacket) processPacket(currentTag, PACKET_PRESENCE);
173 }
174
175 private void sendStartTLS() throws XmlPullParserException, IOException {
176 Tag startTLS = Tag.empty("starttls");
177 startTLS.setAttribute("xmlns", "urn:ietf:params:xml:ns:xmpp-tls");
178 Log.d(LOGTAG, "sending starttls");
179 tagWriter.writeTag(startTLS).flush();
180 }
181
182 private void switchOverToTls(Tag currentTag) throws XmlPullParserException,
183 IOException {
184 Tag nextTag = tagReader.readTag(); // should be proceed end tag
185 Log.d(LOGTAG, "now switch to ssl");
186 SSLSocket sslSocket;
187 try {
188 sslSocket = (SSLSocket) ((SSLSocketFactory) SSLSocketFactory
189 .getDefault()).createSocket(socket, socket.getInetAddress()
190 .getHostAddress(), socket.getPort(), true);
191 tagReader.setInputStream(sslSocket.getInputStream());
192 Log.d(LOGTAG, "reset inputstream");
193 tagWriter.setOutputStream(sslSocket.getOutputStream());
194 Log.d(LOGTAG, "switch over seemed to work");
195 isTlsEncrypted = true;
196 sendStartStream();
197 processStream(tagReader.readTag());
198 } catch (IOException e) {
199 Log.d(LOGTAG, "error on ssl" + e.getMessage());
200 }
201 }
202
203 private void sendSaslAuth() throws IOException, XmlPullParserException {
204 String saslString = SASL.plain(account.getUsername(),
205 account.getPassword());
206 Element auth = new Element("auth");
207 auth.setAttribute("xmlns", "urn:ietf:params:xml:ns:xmpp-sasl");
208 auth.setAttribute("mechanism", "PLAIN");
209 auth.setContent(saslString);
210 Log.d(LOGTAG,"sending sasl "+auth.toString());
211 tagWriter.writeElement(auth);
212 tagWriter.flush();
213 }
214
215 private void processStreamFeatures(Tag currentTag)
216 throws XmlPullParserException, IOException {
217 this.streamFeatures = tagReader.readElement(currentTag);
218 Log.d(LOGTAG,"process stream features "+streamFeatures);
219 if (this.streamFeatures.hasChild("starttls")&&shouldUseTLS) {
220 sendStartTLS();
221 }
222 if (this.streamFeatures.hasChild("mechanisms")&&shouldAuthenticate) {
223 sendSaslAuth();
224 }
225 if (this.streamFeatures.hasChild("bind")&&shouldBind) {
226 sendBindRequest();
227 if (this.streamFeatures.hasChild("session")) {
228 IqPacket startSession = new IqPacket(IqPacket.TYPE_SET);
229 Element session = new Element("session");
230 session.setAttribute("xmlns","urn:ietf:params:xml:ns:xmpp-session");
231 session.setContent("");
232 startSession.addChild(session);
233 sendIqPacket(startSession, null);
234 tagWriter.writeElement(startSession);
235 tagWriter.flush();
236 }
237 Element presence = new Element("presence");
238
239 tagWriter.writeElement(presence);
240 tagWriter.flush();
241 }
242 }
243
244 private void sendBindRequest() throws IOException {
245 IqPacket iq = new IqPacket(IqPacket.TYPE_SET);
246 Element bind = new Element("bind");
247 bind.setAttribute("xmlns","urn:ietf:params:xml:ns:xmpp-bind");
248 iq.addChild(bind);
249 this.sendIqPacket(iq, new OnIqPacketReceived() {
250 @Override
251 public void onIqPacketReceived(IqPacket packet) {
252 Log.d(LOGTAG,"answer for our bind was: "+packet.toString());
253 }
254 });
255 }
256
257 private void processStreamError(Tag currentTag) {
258 Log.d(LOGTAG, "processStreamError");
259 }
260
261 private void sendStartStream() throws IOException {
262 Tag stream = Tag.start("stream");
263 stream.setAttribute("from", account.getJid());
264 stream.setAttribute("to", account.getServer());
265 stream.setAttribute("version", "1.0");
266 stream.setAttribute("xml:lang", "en");
267 stream.setAttribute("xmlns", "jabber:client");
268 stream.setAttribute("xmlns:stream", "http://etherx.jabber.org/streams");
269 tagWriter.writeTag(stream).flush();
270 }
271
272 private String nextRandomId() {
273 return new BigInteger(50, random).toString(32);
274 }
275
276 public void sendIqPacket(IqPacket packet, OnIqPacketReceived callback) throws IOException {
277 String id = nextRandomId();
278 packet.setAttribute("id",id);
279 tagWriter.writeElement(packet);
280 tagWriter.flush();
281 if (callback != null) {
282 iqPacketCallbacks.put(id, callback);
283 }
284 Log.d(LOGTAG,"sending: "+packet.toString());
285 }
286}