XmppConnection.java

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