/chrome/content/libs/xmppsocket.js

/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is OneWeb.
 *
 * The Initial Developer of the Original Code is
 * ProcessOne.
 * Portions created by the Initial Developer are Copyright (C) 2009
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */

var EXPORTED_SYMBOLS = ["XMPPSocket"];

function XMPPSocket(listener, host, port, ssl, domain, authhost)
{
    this.listener = listener;
    this.host = host;
    this.port = port;
    this.ssl = ssl;
    this.domain = domain;
    this.authhost = authhost;
    this.converter = new CharsetConverter("UTF-8");
}

_DECL_(XMPPSocket).prototype =
{
    tlsProblemHandled: {},

    connect: function() {
        if (this.host == this.domain) {
            try {
                var dnsSrv = Components.classes["@process-one.net/dns;1"].
                    getService(Components.interfaces.otIDNSService);
                var mainThread = Components.classes["@mozilla.org/thread-manager;1"].
                    getService(Components.interfaces.nsIThreadManager).mainThread;
                var _this = this;

                dnsSrv.asyncResolveSRV("_xmpp-client._tcp."+this.domain, this,
                                       mainThread);
                return;
            } catch (ex) {}
        }
        this.onLookupComplete();
    },

    onLookupComplete: function(request, response) {
        if (response && response.hasMore())
            [this.host, this.port] = response.getNextAddrAsString().split(":");

        this.doConnect();
        this.listener._handleConnectionEstabilished();
    },

    doConnect: function() {
        var ioSrv = Components.classes["@mozilla.org/network/io-service;1"].
            getService(Components.interfaces.nsIIOService);
        var pps = Components.classes["@mozilla.org/network/protocol-proxy-service;1"].
            getService(Components.interfaces.nsIProtocolProxyService);
        var proxyUri = ioSrv.newURI((this.ssl == "ssl" ? "https://" : "http://")+this.host,
                                    null, null);
        var proxyInfo = pps.resolve(proxyUri, pps.RESOLVE_NON_BLOCKING);

        var mainThread = Components.classes["@mozilla.org/event-queue-service;1"] ?
            Components.classes["@mozilla.org/event-queue-service;1"].
                getService(Components.interfaces.nsIEventQueueService).
                    getSpecialEventQueue(eqs.CURRENT_THREAD_EVENT_QUEUE) :
            Components.classes["@mozilla.org/thread-manager;1"].
                getService().mainThread;

        var stSrv = Components.classes["@mozilla.org/network/socket-transport-service;1"].
          getService(Components.interfaces.nsISocketTransportService);

        this.reset();

        this.transport = stSrv.createTransport([this.ssl ? "ssl" : "starttls"], 1,
                                               this.host, this.port, proxyInfo);
        this.transport.setEventSink(this, mainThread);
        this.is = this.transport.openInputStream(0, 0, 0);
        var pump = Components.classes['@mozilla.org/network/input-stream-pump;1'].
            createInstance(Components.interfaces.nsIInputStreamPump);
        pump.init(this.is, -1, -1, 0, 0, false);
        pump.asyncRead(this, null);

        this.os = this.transport.openOutputStream(1, 0, 0);
        this.bos = Components.classes["@mozilla.org/binaryoutputstream;1"].
          createInstance(Components.interfaces.nsIBinaryOutputStream);
        this.bos.setOutputStream(this.os);
        this._pingInterval = window.setInterval(function(t){t.send(" ")}, 50000, this);
        this.reconnect = false;
    },

    send: function(data) {
        try {
            data = this.converter.encode(data);
            this.bos.writeBytes(data, data.length);
        } catch(ex) {}
    },

    startTLS: function() {
        this.transport.securityInfo.
            QueryInterface(Components.interfaces.nsISSLSocketControl).
            StartTLS();

        // some servers (like gmail.com) have problems with handling TLS1.0
        // (they just stuck on initial hello), to failback to SSL3.0 we needs
        // to reconnect after some time
        if (!(this.domain in this.tlsProblemHandled)) {
            this._sslDowngradeTimeout = setTimeout(function(_this) {
                _this.tlsProblemHandled[_this.domain] = true;
                _this.reconnect = true;
                _this.disconnect();
            }, 1000, this);
        }
    },

    reset: function() {
        if (!this._afterReset) {
            this.saxParser = Components.classes["@mozilla.org/saxparser/xmlreader;1"].
                createInstance(Components.interfaces.nsISAXXMLReader);
            this.saxParser.contentHandler = this;
            this.saxParser.errorHandler = this;
            this.saxParser.parseAsync(null);

            this.parent = null;
            this._afterReset = true;
        }
    },

    disconnect: function() {
        if (this.is)
            this.is.close();
        if (this.bos)
            this.bos.close();
        if (this.transport)
            this.transport.close(0);
        if (this._pingInterval)
            window.clearInterval(this._pingInterval);

        this.is = this.os = this.bos = this.transport = null;
    },

    // nsISAXContentHandler
    startDocument: function() {
        this._afterReset = false;
        this.doc = Components.classes["@mozilla.org/xml/xml-document;1"].
            createInstance(Components.interfaces.nsIDOMXMLDocument);
    },

    endDocument: function() {

    },

    startElement: function(ns, localName, qName, attrs)
    {
        var el = this.doc.createElementNS(ns, localName);
        for (var i = 0; i < attrs.length; i++)
            el.setAttribute(attrs.getQName(i), attrs.getValue(i));

        if (ns == "http://etherx.jabber.org/streams" && qName == "stream:stream") {
            this.listener._handleInitialElement(el);
            this.parent = null;
        } else {
            if (this.parent)
                this.parent.appendChild(el);
            this.parent = el;
        }
    },

    endElement: function(ns, localName, qName)
    {
        var el = this.parent;
        if (this.parent)
            this.parent = this.parent.parentNode;

        if (el && !this.parent)
            this.listener._handleElement(el);
    },

    characters: function(data)
    {
        if (!this.parent)
            return;

        if (this.parent.lastChild && this.parent.lastChild.nodeType == this.parent.TEXT_NODE)
            this.parent.lastChild.appendData(data);
        else
            this.parent.appendChild(this.doc.createTextNode(data));
    },

    processingInstruction: function(target, data) { },
    ignorableWhitespace: function(data) { },
    startPrefixMapping: function(prefix, uri) { },
    endPrefixMapping: function(prefix) { },

    // nsISAXErrorHandler

    error: function(locator, error)
    {
        this.listener._handleError();
    },

    fatalError: function(locator, error)
    {
        if (this.is || this.os)
            this.listener._handleError();
    },

    ignorableWarning: function(locator, error)
    {

    },

    // nsIStreamListener
    onStartRequest: function(request, context)
    {
        this.saxParser.onStartRequest.apply(this.saxParser, arguments);
    },

    onDataAvailable: function(request, context, is, offset, count)
    {
        if (this._sslDowngradeTimeout) {
            clearTimeout(this._sslDowngradeTimeout)
            delete this._sslDowngradeTimeout;
            this.tlsProblemHandled[this.domain] = false;
        }

        if (this._afterReset)
            this.saxParser.onStartRequest(request, context);
        this.saxParser.onDataAvailable.apply(this.saxParser, arguments);

    },

    onStopRequest: function(request, context, status)
    {
        if (this.reconnect) {
            this.disconnect();
            this.doConnect();
            this.listener._handleReconnect();
            return;
        }

        this.listener._handleDisconnect();
        try {
            this.saxParser.onStopRequest.apply(this.saxParser, arguments);
        } catch(ex) { }
    },

    // EventSink

    onTransportStatus: function(transport, status, progress, progressMax)
    {
        if (status != transport.STATUS_CONNECTING_TO)
            return;

        try {
            var si = this.transport.securityInfo.
                QueryInterface(Components.interfaces.nsISSLSocketControl);
            if (si)
                si.notificationCallbacks = {
                    socket: this,
                    notifyCertProblem: function(info, status, host) {
                        var srv = Components.classes["@mozilla.org/security/certoverride;1"].
                            getService(Components.interfaces.nsICertOverrideService);
                        var promptSrv = Components.classes["@mozilla.org/embedcomp/prompt-service;1"].
                            getService(Components.interfaces.nsIPromptService);
                        var flags = 0, msg = "", check = {value: false};

                        status = status.QueryInterface(Components.interfaces.nsISSLStatus);

                        if (status.isUntrusted) {
                            flags |= srv.ERROR_UNTRUSTED;
                            msg += "\n  "+_("Hasn't been verified by recognized authority");
                        }
                        if (status.isDomainMismatch && status.serverCert.commonName != this.socket.domain) {
                            flags |= srv.ERROR_MISMATCH;
                            msg += "\n  "+_("Belongs to different domain");
                        }
                        if (status.isNotValidAtThisTime) {
                            flags |= srv.ERROR_TIME;
                            msg += "\n  "+_("Has been expired");
                        }

                        if (flags == 0)
                            check.value = true;
                        else if (promptSrv.confirmEx(null, _("Invalid certificate"),
                                                     _("Certificate used by server is invalid because:")+msg,
                                                     127+256*2, _("Continue"), "", "",
                                                     _("Always skip this dialog"), check))
                            return true;

                        srv.rememberValidityOverride(this.socket.host, this.socket.port,
                                                     status.serverCert, flags, !check.value);
                        this.socket.reconnect = true;

                        return true;
                    },

                    getInterface: function(iid) {
                        return this.QueryInterface(iid);
                    },

                    QueryInterface: function(iid) {
                        if (!iid.equals(Components.interfaces.nsISupports) &&
                            !iid.equals(Components.interfaces.nsIInterfaceRequestor) &&
                            !iid.equals(Components.interfaces.nsIBadCertListener2))
                            throw Components.results.NS_ERROR_NO_INTERFACE;
                        return this;
                    }
                }
        } catch (ex) {alert(ex)}
    }
};