Revision control
Copy as Markdown
Other Tools
/* -*- 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
// 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 = [];
this.listening = false;
this.dns = Cc["@mozilla.org/network/dns-service;1"].getService(
Ci.nsIDNSService
);
this.parent = parent;
this.eventPump = parent.eventPump;
}
IdentServer.prototype.start = function () {
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 () {
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 (net, serv) {
var addr,
dnsRecord = this.dns.resolve(serv.hostname, 0);
while (dnsRecord.hasMore()) {
addr = dnsRecord.getNextAddrAsString();
this.responses.push({
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 (net) {
var newResponses = [];
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 (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 (
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 (status) {};
IdentListener.prototype.onDataAvailable = function (e) {
var lines = e.line.split(/\r\n/);
if (this.savedLine) {
lines[0] = this.savedLine + lines[0];
this.savedLine = "";
}
if (!e.line.endsWith("\r\n")) {
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 (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 (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();
};