1package eu.siacs.conversations.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.KeyManagementException;
10import java.security.NoSuchAlgorithmException;
11import java.security.SecureRandom;
12import java.util.ArrayList;
13import java.util.HashMap;
14import java.util.Hashtable;
15import java.util.List;
16import java.util.Map.Entry;
17
18import javax.net.ssl.HostnameVerifier;
19import javax.net.ssl.SSLContext;
20import javax.net.ssl.SSLSocket;
21import javax.net.ssl.SSLSocketFactory;
22
23import javax.net.ssl.X509TrustManager;
24
25import org.xmlpull.v1.XmlPullParserException;
26
27import de.duenndns.ssl.MemorizingTrustManager;
28
29import android.os.Bundle;
30import android.os.PowerManager;
31import android.os.PowerManager.WakeLock;
32import android.os.SystemClock;
33import android.util.Log;
34import eu.siacs.conversations.entities.Account;
35import eu.siacs.conversations.services.XmppConnectionService;
36import eu.siacs.conversations.utils.CryptoHelper;
37import eu.siacs.conversations.utils.DNSHelper;
38import eu.siacs.conversations.utils.zlib.ZLibOutputStream;
39import eu.siacs.conversations.utils.zlib.ZLibInputStream;
40import eu.siacs.conversations.xml.Element;
41import eu.siacs.conversations.xml.Tag;
42import eu.siacs.conversations.xml.TagWriter;
43import eu.siacs.conversations.xml.XmlReader;
44import eu.siacs.conversations.xmpp.jingle.OnJinglePacketReceived;
45import eu.siacs.conversations.xmpp.jingle.stanzas.JinglePacket;
46import eu.siacs.conversations.xmpp.stanzas.AbstractStanza;
47import eu.siacs.conversations.xmpp.stanzas.IqPacket;
48import eu.siacs.conversations.xmpp.stanzas.MessagePacket;
49import eu.siacs.conversations.xmpp.stanzas.PresencePacket;
50import eu.siacs.conversations.xmpp.stanzas.streammgmt.AckPacket;
51import eu.siacs.conversations.xmpp.stanzas.streammgmt.EnablePacket;
52import eu.siacs.conversations.xmpp.stanzas.streammgmt.RequestPacket;
53import eu.siacs.conversations.xmpp.stanzas.streammgmt.ResumePacket;
54
55public class XmppConnection implements Runnable {
56
57 protected Account account;
58 private static final String LOGTAG = "xmppService";
59
60 private WakeLock wakeLock;
61
62 private SecureRandom mRandom;
63
64 private Socket socket;
65 private XmlReader tagReader;
66 private TagWriter tagWriter;
67
68 private Features features = new Features(this);
69
70 private boolean shouldBind = true;
71 private boolean shouldAuthenticate = true;
72 private Element streamFeatures;
73 private HashMap<String, List<String>> disco = new HashMap<String, List<String>>();
74
75 private String streamId = null;
76 private int smVersion = 3;
77
78 private int stanzasReceived = 0;
79 private int stanzasSent = 0;
80
81 public long lastPaketReceived = 0;
82 public long lastPingSent = 0;
83 public long lastConnect = 0;
84 public long lastSessionStarted = 0;
85
86 private int attempt = 0;
87
88 private static final int PACKET_IQ = 0;
89 private static final int PACKET_MESSAGE = 1;
90 private static final int PACKET_PRESENCE = 2;
91
92 private Hashtable<String, PacketReceived> packetCallbacks = new Hashtable<String, PacketReceived>();
93 private OnPresencePacketReceived presenceListener = null;
94 private OnJinglePacketReceived jingleListener = null;
95 private OnIqPacketReceived unregisteredIqListener = null;
96 private OnMessagePacketReceived messageListener = null;
97 private OnStatusChanged statusListener = null;
98 private OnBindListener bindListener = null;
99 private MemorizingTrustManager mMemorizingTrustManager;
100
101 public XmppConnection(Account account, XmppConnectionService service) {
102 this.mRandom = service.getRNG();
103 this.mMemorizingTrustManager = service.getMemorizingTrustManager();
104 this.account = account;
105 this.wakeLock = service.getPowerManager().newWakeLock(PowerManager.PARTIAL_WAKE_LOCK,
106 account.getJid());
107 tagWriter = new TagWriter();
108 }
109
110 protected void changeStatus(int nextStatus) {
111 if (account.getStatus() != nextStatus) {
112 if ((nextStatus == Account.STATUS_OFFLINE)
113 && (account.getStatus() != Account.STATUS_CONNECTING)
114 && (account.getStatus() != Account.STATUS_ONLINE)
115 && (account.getStatus() != Account.STATUS_DISABLED)) {
116 return;
117 }
118 if (nextStatus == Account.STATUS_ONLINE) {
119 this.attempt = 0;
120 }
121 account.setStatus(nextStatus);
122 if (statusListener != null) {
123 statusListener.onStatusChanged(account);
124 }
125 }
126 }
127
128 protected void connect() {
129 Log.d(LOGTAG, account.getJid() + ": connecting");
130 lastConnect = SystemClock.elapsedRealtime();
131 this.attempt++;
132 try {
133 shouldAuthenticate = shouldBind = !account
134 .isOptionSet(Account.OPTION_REGISTER);
135 tagReader = new XmlReader(wakeLock);
136 tagWriter = new TagWriter();
137 packetCallbacks.clear();
138 this.changeStatus(Account.STATUS_CONNECTING);
139 Bundle namePort = DNSHelper.getSRVRecord(account.getServer());
140 if ("timeout".equals(namePort.getString("error"))) {
141 Log.d(LOGTAG, account.getJid() + ": dns timeout");
142 this.changeStatus(Account.STATUS_OFFLINE);
143 return;
144 }
145 String srvRecordServer = namePort.getString("name");
146 String srvIpServer = namePort.getString("ipv4");
147 int srvRecordPort = namePort.getInt("port");
148 if (srvRecordServer != null) {
149 if (srvIpServer != null) {
150 Log.d(LOGTAG, account.getJid() + ": using values from dns "
151 + srvRecordServer + "[" + srvIpServer + "]:"
152 + srvRecordPort);
153 socket = new Socket(srvIpServer, srvRecordPort);
154 } else {
155 Log.d(LOGTAG, account.getJid() + ": using values from dns "
156 + srvRecordServer + ":" + srvRecordPort);
157 socket = new Socket(srvRecordServer, srvRecordPort);
158 }
159 } else {
160 socket = new Socket(account.getServer(), 5222);
161 }
162 OutputStream out = socket.getOutputStream();
163 tagWriter.setOutputStream(out);
164 InputStream in = socket.getInputStream();
165 tagReader.setInputStream(in);
166 tagWriter.beginDocument();
167 sendStartStream();
168 Tag nextTag;
169 while ((nextTag = tagReader.readTag()) != null) {
170 if (nextTag.isStart("stream")) {
171 processStream(nextTag);
172 break;
173 } else {
174 Log.d(LOGTAG, "found unexpected tag: " + nextTag.getName());
175 return;
176 }
177 }
178 if (socket.isConnected()) {
179 socket.close();
180 }
181 } catch (UnknownHostException e) {
182 this.changeStatus(Account.STATUS_SERVER_NOT_FOUND);
183 if (wakeLock.isHeld()) {
184 try { wakeLock.release();} catch (RuntimeException re) {}
185 }
186 return;
187 } catch (IOException e) {
188 this.changeStatus(Account.STATUS_OFFLINE);
189 if (wakeLock.isHeld()) {
190 try { wakeLock.release();} catch (RuntimeException re) {}
191 }
192 return;
193 } catch (NoSuchAlgorithmException e) {
194 this.changeStatus(Account.STATUS_OFFLINE);
195 Log.d(LOGTAG, "compression exception " + e.getMessage());
196 if (wakeLock.isHeld()) {
197 try { wakeLock.release();} catch (RuntimeException re) {}
198 }
199 return;
200 } catch (XmlPullParserException e) {
201 this.changeStatus(Account.STATUS_OFFLINE);
202 Log.d(LOGTAG, "xml exception " + e.getMessage());
203 if (wakeLock.isHeld()) {
204 try { wakeLock.release();} catch (RuntimeException re) {}
205 }
206 return;
207 }
208
209 }
210
211 @Override
212 public void run() {
213 connect();
214 }
215
216 private void processStream(Tag currentTag) throws XmlPullParserException,
217 IOException, NoSuchAlgorithmException {
218 Tag nextTag = tagReader.readTag();
219 while ((nextTag != null) && (!nextTag.isEnd("stream"))) {
220 if (nextTag.isStart("error")) {
221 processStreamError(nextTag);
222 } else if (nextTag.isStart("features")) {
223 processStreamFeatures(nextTag);
224 } else if (nextTag.isStart("proceed")) {
225 switchOverToTls(nextTag);
226 } else if (nextTag.isStart("compressed")) {
227 switchOverToZLib(nextTag);
228 } else if (nextTag.isStart("success")) {
229 Log.d(LOGTAG, account.getJid() + ": logged in");
230 tagReader.readTag();
231 tagReader.reset();
232 sendStartStream();
233 processStream(tagReader.readTag());
234 break;
235 } else if (nextTag.isStart("failure")) {
236 tagReader.readElement(nextTag);
237 changeStatus(Account.STATUS_UNAUTHORIZED);
238 } else if (nextTag.isStart("challenge")) {
239 String challange = tagReader.readElement(nextTag).getContent();
240 Element response = new Element("response");
241 response.setAttribute("xmlns",
242 "urn:ietf:params:xml:ns:xmpp-sasl");
243 response.setContent(CryptoHelper.saslDigestMd5(account,
244 challange,mRandom));
245 tagWriter.writeElement(response);
246 } else if (nextTag.isStart("enabled")) {
247 this.stanzasSent = 0;
248 Element enabled = tagReader.readElement(nextTag);
249 if ("true".equals(enabled.getAttribute("resume"))) {
250 this.streamId = enabled.getAttribute("id");
251 Log.d(LOGTAG, account.getJid() + ": stream managment("
252 + smVersion + ") enabled (resumable)");
253 } else {
254 Log.d(LOGTAG, account.getJid() + ": stream managment("
255 + smVersion + ") enabled");
256 }
257 this.lastSessionStarted = SystemClock.elapsedRealtime();
258 this.stanzasReceived = 0;
259 RequestPacket r = new RequestPacket(smVersion);
260 tagWriter.writeStanzaAsync(r);
261 } else if (nextTag.isStart("resumed")) {
262 lastPaketReceived = SystemClock.elapsedRealtime();
263 Log.d(LOGTAG, account.getJid() + ": session resumed");
264 tagReader.readElement(nextTag);
265 sendPing();
266 changeStatus(Account.STATUS_ONLINE);
267 } else if (nextTag.isStart("r")) {
268 tagReader.readElement(nextTag);
269 AckPacket ack = new AckPacket(this.stanzasReceived, smVersion);
270 tagWriter.writeStanzaAsync(ack);
271 } else if (nextTag.isStart("a")) {
272 Element ack = tagReader.readElement(nextTag);
273 lastPaketReceived = SystemClock.elapsedRealtime();
274 int serverSequence = Integer.parseInt(ack.getAttribute("h"));
275 if (serverSequence > this.stanzasSent) {
276 this.stanzasSent = serverSequence;
277 }
278 } else if (nextTag.isStart("failed")) {
279 tagReader.readElement(nextTag);
280 Log.d(LOGTAG, account.getJid() + ": resumption failed");
281 streamId = null;
282 if (account.getStatus() != Account.STATUS_ONLINE) {
283 sendBindRequest();
284 }
285 } else if (nextTag.isStart("iq")) {
286 processIq(nextTag);
287 } else if (nextTag.isStart("message")) {
288 processMessage(nextTag);
289 } else if (nextTag.isStart("presence")) {
290 processPresence(nextTag);
291 }
292 nextTag = tagReader.readTag();
293 }
294 if (account.getStatus() == Account.STATUS_ONLINE) {
295 account.setStatus(Account.STATUS_OFFLINE);
296 if (statusListener != null) {
297 statusListener.onStatusChanged(account);
298 }
299 }
300 }
301
302 private Element processPacket(Tag currentTag, int packetType)
303 throws XmlPullParserException, IOException {
304 Element element;
305 switch (packetType) {
306 case PACKET_IQ:
307 element = new IqPacket();
308 break;
309 case PACKET_MESSAGE:
310 element = new MessagePacket();
311 break;
312 case PACKET_PRESENCE:
313 element = new PresencePacket();
314 break;
315 default:
316 return null;
317 }
318 element.setAttributes(currentTag.getAttributes());
319 Tag nextTag = tagReader.readTag();
320 if (nextTag==null) {
321 throw new IOException("interrupted mid tag");
322 }
323 while (!nextTag.isEnd(element.getName())) {
324 if (!nextTag.isNo()) {
325 Element child = tagReader.readElement(nextTag);
326 if ((packetType == PACKET_IQ)
327 && ("jingle".equals(child.getName()))) {
328 element = new JinglePacket();
329 element.setAttributes(currentTag.getAttributes());
330 }
331 element.addChild(child);
332 }
333 nextTag = tagReader.readTag();
334 if (nextTag==null) {
335 throw new IOException("interrupted mid tag");
336 }
337 }
338 ++stanzasReceived;
339 lastPaketReceived = SystemClock.elapsedRealtime();
340 return element;
341 }
342
343 private void processIq(Tag currentTag) throws XmlPullParserException,
344 IOException {
345 IqPacket packet = (IqPacket) processPacket(currentTag, PACKET_IQ);
346
347 if (packet.getId() == null) {
348 return; // an iq packet without id is definitely invalid
349 }
350
351 if (packet instanceof JinglePacket) {
352 if (this.jingleListener != null) {
353 this.jingleListener.onJinglePacketReceived(account,
354 (JinglePacket) packet);
355 }
356 } else {
357 if (packetCallbacks.containsKey(packet.getId())) {
358 if (packetCallbacks.get(packet.getId()) instanceof OnIqPacketReceived) {
359 ((OnIqPacketReceived) packetCallbacks.get(packet.getId()))
360 .onIqPacketReceived(account, packet);
361 }
362
363 packetCallbacks.remove(packet.getId());
364 } else if (this.unregisteredIqListener != null) {
365 this.unregisteredIqListener.onIqPacketReceived(account, packet);
366 }
367 }
368 }
369
370 private void processMessage(Tag currentTag) throws XmlPullParserException,
371 IOException {
372 MessagePacket packet = (MessagePacket) processPacket(currentTag,
373 PACKET_MESSAGE);
374 String id = packet.getAttribute("id");
375 if ((id != null) && (packetCallbacks.containsKey(id))) {
376 if (packetCallbacks.get(id) instanceof OnMessagePacketReceived) {
377 ((OnMessagePacketReceived) packetCallbacks.get(id))
378 .onMessagePacketReceived(account, packet);
379 }
380 packetCallbacks.remove(id);
381 } else if (this.messageListener != null) {
382 this.messageListener.onMessagePacketReceived(account, packet);
383 }
384 }
385
386 private void processPresence(Tag currentTag) throws XmlPullParserException,
387 IOException {
388 PresencePacket packet = (PresencePacket) processPacket(currentTag,
389 PACKET_PRESENCE);
390 String id = packet.getAttribute("id");
391 if ((id != null) && (packetCallbacks.containsKey(id))) {
392 if (packetCallbacks.get(id) instanceof OnPresencePacketReceived) {
393 ((OnPresencePacketReceived) packetCallbacks.get(id))
394 .onPresencePacketReceived(account, packet);
395 }
396 packetCallbacks.remove(id);
397 } else if (this.presenceListener != null) {
398 this.presenceListener.onPresencePacketReceived(account, packet);
399 }
400 }
401
402 private void sendCompressionZlib() throws IOException {
403 Element compress = new Element("compress");
404 compress.setAttribute("xmlns", "http://jabber.org/protocol/compress");
405 compress.addChild("method").setContent("zlib");
406 tagWriter.writeElement(compress);
407 }
408
409 private void switchOverToZLib(Tag currentTag)
410 throws XmlPullParserException, IOException,
411 NoSuchAlgorithmException {
412 tagReader.readTag(); // read tag close
413 tagWriter.setOutputStream(new ZLibOutputStream(tagWriter
414 .getOutputStream()));
415 tagReader
416 .setInputStream(new ZLibInputStream(tagReader.getInputStream()));
417
418 sendStartStream();
419 Log.d(LOGTAG, account.getJid() + ": compression enabled");
420 processStream(tagReader.readTag());
421 }
422
423 private void sendStartTLS() throws IOException {
424 Tag startTLS = Tag.empty("starttls");
425 startTLS.setAttribute("xmlns", "urn:ietf:params:xml:ns:xmpp-tls");
426 tagWriter.writeTag(startTLS);
427 }
428
429 private void switchOverToTls(Tag currentTag) throws XmlPullParserException,
430 IOException {
431 tagReader.readTag();
432 try {
433 SSLContext sc = SSLContext.getInstance("TLS");
434 sc.init(null, new X509TrustManager[] { this.mMemorizingTrustManager }, mRandom);
435 SSLSocketFactory factory = sc.getSocketFactory();
436
437 HostnameVerifier verifier = this.mMemorizingTrustManager.wrapHostnameVerifier(new org.apache.http.conn.ssl.StrictHostnameVerifier());
438 SSLSocket sslSocket = (SSLSocket) factory.createSocket(socket,
439 socket.getInetAddress().getHostAddress(), socket.getPort(),
440 true);
441
442 if (verifier != null && !verifier.verify(account.getServer(), sslSocket.getSession())) {
443 Log.d(LOGTAG, account.getJid() + ": host mismatch in TLS connection");
444 sslSocket.close();
445 throw new IOException();
446 }
447 tagReader.setInputStream(sslSocket.getInputStream());
448 tagWriter.setOutputStream(sslSocket.getOutputStream());
449 sendStartStream();
450 Log.d(LOGTAG, account.getJid() + ": TLS connection established");
451 processStream(tagReader.readTag());
452 sslSocket.close();
453 } catch (NoSuchAlgorithmException e1) {
454 e1.printStackTrace();
455 } catch (KeyManagementException e) {
456 e.printStackTrace();
457 }
458 }
459
460 private void sendSaslAuthPlain() throws IOException {
461 String saslString = CryptoHelper.saslPlain(account.getUsername(),
462 account.getPassword());
463 Element auth = new Element("auth");
464 auth.setAttribute("xmlns", "urn:ietf:params:xml:ns:xmpp-sasl");
465 auth.setAttribute("mechanism", "PLAIN");
466 auth.setContent(saslString);
467 tagWriter.writeElement(auth);
468 }
469
470 private void sendSaslAuthDigestMd5() throws IOException {
471 Element auth = new Element("auth");
472 auth.setAttribute("xmlns", "urn:ietf:params:xml:ns:xmpp-sasl");
473 auth.setAttribute("mechanism", "DIGEST-MD5");
474 tagWriter.writeElement(auth);
475 }
476
477 private void processStreamFeatures(Tag currentTag)
478 throws XmlPullParserException, IOException {
479 this.streamFeatures = tagReader.readElement(currentTag);
480 if (this.streamFeatures.hasChild("starttls")
481 && account.isOptionSet(Account.OPTION_USETLS)) {
482 sendStartTLS();
483 } else if (compressionAvailable()) {
484 sendCompressionZlib();
485 } else if (this.streamFeatures.hasChild("register")
486 && (account.isOptionSet(Account.OPTION_REGISTER))) {
487 sendRegistryRequest();
488 } else if (!this.streamFeatures.hasChild("register")
489 && (account.isOptionSet(Account.OPTION_REGISTER))) {
490 changeStatus(Account.STATUS_REGISTRATION_NOT_SUPPORTED);
491 disconnect(true);
492 } else if (this.streamFeatures.hasChild("mechanisms")
493 && shouldAuthenticate) {
494 List<String> mechanisms = extractMechanisms(streamFeatures
495 .findChild("mechanisms"));
496 if (mechanisms.contains("PLAIN")) {
497 sendSaslAuthPlain();
498 } else if (mechanisms.contains("DIGEST-MD5")) {
499 sendSaslAuthDigestMd5();
500 }
501 } else if (this.streamFeatures.hasChild("sm", "urn:xmpp:sm:"
502 + smVersion)
503 && streamId != null) {
504 ResumePacket resume = new ResumePacket(this.streamId,
505 stanzasReceived, smVersion);
506 this.tagWriter.writeStanzaAsync(resume);
507 } else if (this.streamFeatures.hasChild("bind") && shouldBind) {
508 sendBindRequest();
509 }
510 }
511
512 private boolean compressionAvailable() {
513 if (!this.streamFeatures.hasChild("compression",
514 "http://jabber.org/features/compress"))
515 return false;
516 if (!ZLibOutputStream.SUPPORTED)
517 return false;
518 if (!account.isOptionSet(Account.OPTION_USECOMPRESSION))
519 return false;
520
521 Element compression = this.streamFeatures.findChild("compression",
522 "http://jabber.org/features/compress");
523 for (Element child : compression.getChildren()) {
524 if (!"method".equals(child.getName()))
525 continue;
526
527 if ("zlib".equalsIgnoreCase(child.getContent())) {
528 return true;
529 }
530 }
531 return false;
532 }
533
534 private List<String> extractMechanisms(Element stream) {
535 ArrayList<String> mechanisms = new ArrayList<String>(stream
536 .getChildren().size());
537 for (Element child : stream.getChildren()) {
538 mechanisms.add(child.getContent());
539 }
540 return mechanisms;
541 }
542
543 private void sendRegistryRequest() {
544 IqPacket register = new IqPacket(IqPacket.TYPE_GET);
545 register.query("jabber:iq:register");
546 register.setTo(account.getServer());
547 sendIqPacket(register, new OnIqPacketReceived() {
548
549 @Override
550 public void onIqPacketReceived(Account account, IqPacket packet) {
551 Element instructions = packet.query().findChild("instructions");
552 if (packet.query().hasChild("username")
553 && (packet.query().hasChild("password"))) {
554 IqPacket register = new IqPacket(IqPacket.TYPE_SET);
555 Element username = new Element("username")
556 .setContent(account.getUsername());
557 Element password = new Element("password")
558 .setContent(account.getPassword());
559 register.query("jabber:iq:register").addChild(username);
560 register.query().addChild(password);
561 sendIqPacket(register, new OnIqPacketReceived() {
562
563 @Override
564 public void onIqPacketReceived(Account account,
565 IqPacket packet) {
566 if (packet.getType() == IqPacket.TYPE_RESULT) {
567 account.setOption(Account.OPTION_REGISTER,
568 false);
569 changeStatus(Account.STATUS_REGISTRATION_SUCCESSFULL);
570 } else if (packet.hasChild("error")
571 && (packet.findChild("error")
572 .hasChild("conflict"))) {
573 changeStatus(Account.STATUS_REGISTRATION_CONFLICT);
574 } else {
575 changeStatus(Account.STATUS_REGISTRATION_FAILED);
576 Log.d(LOGTAG, packet.toString());
577 }
578 disconnect(true);
579 }
580 });
581 } else {
582 changeStatus(Account.STATUS_REGISTRATION_FAILED);
583 disconnect(true);
584 Log.d(LOGTAG, account.getJid()
585 + ": could not register. instructions are"
586 + instructions.getContent());
587 }
588 }
589 });
590 }
591
592 private void sendBindRequest() throws IOException {
593 IqPacket iq = new IqPacket(IqPacket.TYPE_SET);
594 iq.addChild("bind", "urn:ietf:params:xml:ns:xmpp-bind")
595 .addChild("resource").setContent(account.getResource());
596 this.sendUnboundIqPacket(iq, new OnIqPacketReceived() {
597 @Override
598 public void onIqPacketReceived(Account account, IqPacket packet) {
599 Element bind = packet.findChild("bind");
600 if (bind!=null) {
601 Element jid = bind.findChild("jid");
602 if (jid!=null) {
603 account.setResource(jid.getContent().split("/")[1]);
604 if (streamFeatures.hasChild("sm", "urn:xmpp:sm:3")) {
605 smVersion = 3;
606 EnablePacket enable = new EnablePacket(smVersion);
607 tagWriter.writeStanzaAsync(enable);
608 } else if (streamFeatures.hasChild("sm", "urn:xmpp:sm:2")) {
609 smVersion = 2;
610 EnablePacket enable = new EnablePacket(smVersion);
611 tagWriter.writeStanzaAsync(enable);
612 }
613 sendServiceDiscoveryInfo(account.getServer());
614 sendServiceDiscoveryItems(account.getServer());
615 if (bindListener != null) {
616 bindListener.onBind(account);
617 }
618 changeStatus(Account.STATUS_ONLINE);
619 } else {
620 disconnect(true);
621 }
622 } else {
623 disconnect(true);
624 }
625 }
626 });
627 if (this.streamFeatures.hasChild("session")) {
628 Log.d(LOGTAG, account.getJid() + ": sending deprecated session");
629 IqPacket startSession = new IqPacket(IqPacket.TYPE_SET);
630 startSession.addChild("session",
631 "urn:ietf:params:xml:ns:xmpp-session");
632 this.sendUnboundIqPacket(startSession, null);
633 }
634 }
635
636 private void sendServiceDiscoveryInfo(final String server) {
637 IqPacket iq = new IqPacket(IqPacket.TYPE_GET);
638 iq.setTo(server);
639 iq.query("http://jabber.org/protocol/disco#info");
640 this.sendIqPacket(iq, new OnIqPacketReceived() {
641
642 @Override
643 public void onIqPacketReceived(Account account, IqPacket packet) {
644 List<Element> elements = packet.query().getChildren();
645 List<String> features = new ArrayList<String>();
646 for (int i = 0; i < elements.size(); ++i) {
647 if (elements.get(i).getName().equals("feature")) {
648 features.add(elements.get(i).getAttribute("var"));
649 }
650 }
651 disco.put(server, features);
652
653 if (account.getServer().equals(server)) {
654 enableAdvancedStreamFeatures();
655 }
656 }
657 });
658 }
659
660 private void enableAdvancedStreamFeatures() {
661 if (getFeatures().carbons()) {
662 sendEnableCarbons();
663 }
664 }
665
666 private void sendServiceDiscoveryItems(final String server) {
667 IqPacket iq = new IqPacket(IqPacket.TYPE_GET);
668 iq.setTo(server);
669 iq.query("http://jabber.org/protocol/disco#items");
670 this.sendIqPacket(iq, new OnIqPacketReceived() {
671
672 @Override
673 public void onIqPacketReceived(Account account, IqPacket packet) {
674 List<Element> elements = packet.query().getChildren();
675 for (int i = 0; i < elements.size(); ++i) {
676 if (elements.get(i).getName().equals("item")) {
677 String jid = elements.get(i).getAttribute("jid");
678 sendServiceDiscoveryInfo(jid);
679 }
680 }
681 }
682 });
683 }
684
685 private void sendEnableCarbons() {
686 IqPacket iq = new IqPacket(IqPacket.TYPE_SET);
687 iq.addChild("enable", "urn:xmpp:carbons:2");
688 this.sendIqPacket(iq, new OnIqPacketReceived() {
689
690 @Override
691 public void onIqPacketReceived(Account account, IqPacket packet) {
692 if (!packet.hasChild("error")) {
693 Log.d(LOGTAG, account.getJid()
694 + ": successfully enabled carbons");
695 } else {
696 Log.d(LOGTAG, account.getJid()
697 + ": error enableing carbons " + packet.toString());
698 }
699 }
700 });
701 }
702
703 private void processStreamError(Tag currentTag) {
704 Log.d(LOGTAG, "processStreamError");
705 }
706
707 private void sendStartStream() throws IOException {
708 Tag stream = Tag.start("stream:stream");
709 stream.setAttribute("from", account.getJid());
710 stream.setAttribute("to", account.getServer());
711 stream.setAttribute("version", "1.0");
712 stream.setAttribute("xml:lang", "en");
713 stream.setAttribute("xmlns", "jabber:client");
714 stream.setAttribute("xmlns:stream", "http://etherx.jabber.org/streams");
715 tagWriter.writeTag(stream);
716 }
717
718 private String nextRandomId() {
719 return new BigInteger(50, mRandom).toString(32);
720 }
721
722 public void sendIqPacket(IqPacket packet, OnIqPacketReceived callback) {
723 if (packet.getId() == null) {
724 String id = nextRandomId();
725 packet.setAttribute("id", id);
726 }
727 packet.setFrom(account.getFullJid());
728 this.sendPacket(packet, callback);
729 }
730
731 public void sendUnboundIqPacket(IqPacket packet, OnIqPacketReceived callback) {
732 if (packet.getId() == null) {
733 String id = nextRandomId();
734 packet.setAttribute("id", id);
735 }
736 this.sendPacket(packet, callback);
737 }
738
739 public void sendMessagePacket(MessagePacket packet) {
740 this.sendPacket(packet, null);
741 }
742
743 public void sendPresencePacket(PresencePacket packet) {
744 this.sendPacket(packet, null);
745 }
746
747 private synchronized void sendPacket(final AbstractStanza packet,
748 PacketReceived callback) {
749 // TODO dont increment stanza count if packet = request packet or ack;
750 ++stanzasSent;
751 tagWriter.writeStanzaAsync(packet);
752 if (callback != null) {
753 if (packet.getId() == null) {
754 packet.setId(nextRandomId());
755 }
756 packetCallbacks.put(packet.getId(), callback);
757 }
758 }
759
760 public void sendPing() {
761 if (streamFeatures.hasChild("sm")) {
762 tagWriter.writeStanzaAsync(new RequestPacket(smVersion));
763 } else {
764 IqPacket iq = new IqPacket(IqPacket.TYPE_GET);
765 iq.setFrom(account.getFullJid());
766 iq.addChild("ping", "urn:xmpp:ping");
767 this.sendIqPacket(iq, null);
768 }
769 }
770
771 public void setOnMessagePacketReceivedListener(
772 OnMessagePacketReceived listener) {
773 this.messageListener = listener;
774 }
775
776 public void setOnUnregisteredIqPacketReceivedListener(
777 OnIqPacketReceived listener) {
778 this.unregisteredIqListener = listener;
779 }
780
781 public void setOnPresencePacketReceivedListener(
782 OnPresencePacketReceived listener) {
783 this.presenceListener = listener;
784 }
785
786 public void setOnJinglePacketReceivedListener(
787 OnJinglePacketReceived listener) {
788 this.jingleListener = listener;
789 }
790
791 public void setOnStatusChangedListener(OnStatusChanged listener) {
792 this.statusListener = listener;
793 }
794
795 public void setOnBindListener(OnBindListener listener) {
796 this.bindListener = listener;
797 }
798
799 public void disconnect(boolean force) {
800 changeStatus(Account.STATUS_OFFLINE);
801 Log.d(LOGTAG, "disconnecting");
802 try {
803 if (force) {
804 socket.close();
805 return;
806 }
807 new Thread(new Runnable() {
808
809 @Override
810 public void run() {
811 if (tagWriter.isActive()) {
812 tagWriter.finish();
813 try {
814 while (!tagWriter.finished()) {
815 Log.d(LOGTAG, "not yet finished");
816 Thread.sleep(100);
817 }
818 tagWriter.writeTag(Tag.end("stream:stream"));
819 } catch (IOException e) {
820 Log.d(LOGTAG, "io exception during disconnect");
821 } catch (InterruptedException e) {
822 Log.d(LOGTAG, "interrupted");
823 }
824 }
825 }
826 }).start();
827 } catch (IOException e) {
828 Log.d(LOGTAG, "io exception during disconnect");
829 }
830 }
831
832 public List<String> findDiscoItemsByFeature(String feature) {
833 List<String> items = new ArrayList<String>();
834 for (Entry<String, List<String>> cursor : disco.entrySet()) {
835 if (cursor.getValue().contains(feature)) {
836 items.add(cursor.getKey());
837 }
838 }
839 return items;
840 }
841
842 public String findDiscoItemByFeature(String feature) {
843 List<String> items = findDiscoItemsByFeature(feature);
844 if (items.size()>=1) {
845 return items.get(0);
846 }
847 return null;
848 }
849
850 public void r() {
851 this.tagWriter.writeStanzaAsync(new RequestPacket(smVersion));
852 }
853
854 public int getReceivedStanzas() {
855 return this.stanzasReceived;
856 }
857
858 public int getSentStanzas() {
859 return this.stanzasSent;
860 }
861
862 public String getMucServer() {
863 return findDiscoItemByFeature("http://jabber.org/protocol/muc");
864 }
865
866 public int getTimeToNextAttempt() {
867 int interval = (int) (25 * Math.pow(1.5, attempt));
868 int secondsSinceLast = (int) ((SystemClock.elapsedRealtime() - this.lastConnect) / 1000);
869 return interval - secondsSinceLast;
870 }
871
872 public int getAttempt() {
873 return this.attempt;
874 }
875
876 public Features getFeatures() {
877 return this.features;
878 }
879
880 public class Features {
881 XmppConnection connection;
882 public Features(XmppConnection connection) {
883 this.connection = connection;
884 }
885
886 private boolean hasDiscoFeature(String server, String feature) {
887 if (!connection.disco.containsKey(server)) {
888 return false;
889 }
890 return connection.disco.get(server).contains(feature);
891 }
892
893 public boolean carbons() {
894 return hasDiscoFeature(account.getServer(), "urn:xmpp:carbons:2");
895 }
896
897 public boolean sm() {
898 if (connection.streamFeatures == null) {
899 return false;
900 } else {
901 return connection.streamFeatures.hasChild("sm");
902 }
903 }
904
905 public boolean pubsub() {
906 return hasDiscoFeature(account.getServer(), "http://jabber.org/protocol/pubsub#publish");
907 }
908
909 public boolean rosterVersioning() {
910 if (connection.streamFeatures == null) {
911 return false;
912 } else {
913 return connection.streamFeatures.hasChild("ver");
914 }
915 }
916 }
917}