1 files changed, 203 insertions, 0 deletions
diff --git a/comm/suite/chatzilla/js/lib/ident.js b/comm/suite/chatzilla/js/lib/ident.js
new file mode 100644
index 0000000000..5110d70165
--- /dev/null
+++ b/comm/suite/chatzilla/js/lib/ident.js
@@ -0,0 +1,203 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+// RFC1413-ish identification server for ChatZilla
+// One Ident Server is used for all networks. When multiple networks are in the
+// process of connecting, it won't stop listening until they're all done.
+
+function IdentServer(parent)
+{
+    this.responses = new Array();
+    this.listening = false;
+    this.dns = getService("@mozilla.org/network/dns-service;1","nsIDNSService");
+
+    this.parent = parent;
+    this.eventPump = parent.eventPump;
+}
+
+IdentServer.prototype.start =
+function ident_start()
+{
+    if (this.listening)
+        return true;
+
+    if (!this.socket)
+        this.socket = new CBSConnection();
+
+    if (!this.socket.listen(113, this))
+        return false;
+
+    this.listening = true;
+    return true;
+}
+
+IdentServer.prototype.stop =
+function ident_stop()
+{
+    if (!this.socket || !this.listening)
+        return;
+
+    // No need to destroy socket, listen() will work again.
+    this.socket.close();
+    this.listening = false;
+}
+
+IdentServer.prototype.addNetwork =
+function ident_add(net, serv)
+{
+    var addr, dnsRecord = this.dns.resolve(serv.hostname, 0);
+
+    while (dnsRecord.hasMore())
+    {
+        addr = dnsRecord.getNextAddrAsString();
+        this.responses.push({net: net, ip: addr, port: serv.port,
+                             username: net.INITIAL_NAME || net.INITIAL_NICK});
+    }
+
+    if (this.responses.length == 0)
+        return false;
+
+    // Start the ident server if necessary.
+    if (!this.listening)
+        return this.start();
+
+    return true;
+}
+
+IdentServer.prototype.removeNetwork =
+function ident_remove(net)
+{
+    var newResponses = new Array();
+
+    for (var i = 0; i < this.responses.length; ++i)
+    {
+        if (this.responses[i].net != net)
+            newResponses.push(this.responses[i]);
+    }
+    this.responses = newResponses;
+
+    // Only stop listening if responses is empty - no networks need us anymore.
+    if (this.responses.length == 0)
+        this.stop();
+}
+
+IdentServer.prototype.onSocketAccepted =
+function ident_gotconn(serv, transport)
+{
+    // Using the listening CBSConnection to accept would stop it listening.
+    // A new CBSConnection does exactly what we want.
+    var connection = new CBSConnection();
+    connection.accept(transport);
+
+    connection.startAsyncRead(new IdentListener(this, connection));
+}
+
+function IdentListener(server, connection)
+{
+    this.server = server;
+    this.connection = connection;
+}
+
+IdentListener.prototype.onStreamDataAvailable =
+function ident_listener_sda(request, inStream, sourceOffset, count)
+{
+    var ev = new CEvent("ident-listener", "data-available", this,
+                        "onDataAvailable");
+    ev.line = this.connection.readData(0, count);
+    this.server.eventPump.routeEvent(ev);
+}
+
+IdentListener.prototype.onStreamClose =
+function ident_listener_sclose(status)
+{
+}
+
+IdentListener.prototype.onDataAvailable =
+function ident_listener_dataavailable(e)
+{
+    var incomplete = (e.line.substr(-2) != "\r\n");
+    var lines = e.line.split(/\r\n/);
+
+    if (this.savedLine)
+    {
+        lines[0] = this.savedLine + lines[0];
+        this.savedLine = "";
+    }
+
+    if (incomplete)
+        this.savedLine = lines.pop()
+
+    for (var i in lines)
+    {
+        var ev = new CEvent("ident-listener", "rawdata", this, "onRawData");
+        ev.line = lines[i];
+        this.server.eventPump.routeEvent(ev);
+    }
+}
+
+IdentListener.prototype.onRawData =
+function ident_listener_rawdata(e)
+{
+    var ports = e.line.match(/(\d+) *, *(\d+)/);
+    // <port-on-server> , <port-on-client>
+    // (where "server" is the ident server)
+
+    if (!ports)
+    {
+        this.connection.disconnect(); // same meaning as "ERROR : UNKNOWN-ERROR"
+        return;
+    }
+
+    e.type = "parsedrequest";
+    e.destObject = this;
+    e.destMethod = "onParsedRequest";
+    e.localPort = ports[1];
+    e.remotePort = ports[2];
+}
+
+IdentListener.prototype.onParsedRequest =
+function ident_listener_request(e)
+{
+    function response(str)
+    {
+        return e.localPort + " , " + e.remotePort + " : " + str + "\r\n";
+    };
+
+    function validPort(p)
+    {
+        return (p >= 1) && (p <= 65535);
+    };
+
+    if (!validPort(e.localPort) || !validPort(e.remotePort))
+    {
+        this.connection.sendData(response("ERROR : INVALID-PORT"));
+        this.connection.disconnect();
+        return;
+    }
+
+    var found, responses = this.server.responses;
+    for (var i = 0; i < responses.length; ++i)
+    {
+        if ((e.remotePort == responses[i].port) &&
+            (this.connection._transport.host == responses[i].ip))
+        {
+            // charset defaults to US-ASCII
+            // anything except an OS username should use OTHER
+            // however, ircu sucks, so we can't do that.
+            this.connection.sendData(response("USERID : CHATZILLA :" + 
+                                              responses[i].username));
+            found = true;
+            break;
+        }
+    }
+
+    if (!found)
+        this.connection.sendData(response("ERROR : NO-USER"));
+
+    // Spec gives us a choice: drop the connection, or listen for more queries.
+    // Since IRC servers will only ever want one name, we disconnect.
+    this.connection.disconnect();
+}