This commit is contained in:
Tony Brix 2021-05-27 14:16:05 -05:00
parent 9b1a10fff0
commit 6d1f23a831
2 changed files with 106 additions and 108 deletions

View File

@ -1,66 +1,65 @@
const cluster = require('cluster'); const cluster = require('cluster')
const ldap = require('ldapjs'); const ldap = require('ldapjs')
const net = require('net'); const net = require('net')
const os = require('os'); const os = require('os')
const threads = []; const threads = []
threads.getNext = function () { threads.getNext = function () {
return (Math.floor(Math.random() * this.length)); return (Math.floor(Math.random() * this.length))
}; }
const serverOptions = { const serverOptions = {
port: 1389 port: 1389
}; }
if (cluster.isMaster) { if (cluster.isMaster) {
const server = net.createServer(serverOptions, (socket) => { const server = net.createServer(serverOptions, (socket) => {
socket.pause(); socket.pause()
console.log('ldapjs client requesting connection'); console.log('ldapjs client requesting connection')
let routeTo = threads.getNext(); const routeTo = threads.getNext()
threads[routeTo].send({ type: 'connection' }, socket); threads[routeTo].send({ type: 'connection' }, socket)
}); })
for (let i = 0; i < os.cpus().length; i++) { for (let i = 0; i < os.cpus().length; i++) {
let thread = cluster.fork({ const thread = cluster.fork({
'id': i id: i
}); })
thread.id = i; thread.id = i
thread.on('message', function (msg) { thread.on('message', function () {
}); })
threads.push(thread); threads.push(thread)
}
server.listen(serverOptions.port, function () {
console.log('ldapjs listening at ldap://127.0.0.1:' + serverOptions.port)
})
} else {
const server = ldap.createServer(serverOptions)
const threadId = process.env.id
process.on('message', (msg, socket) => {
switch (msg.type) {
case 'connection':
server.newConnection(socket)
socket.resume()
console.log('ldapjs client connection accepted on ' + threadId.toString())
}
})
server.search('dc=example', function (req, res) {
console.log('ldapjs search initiated on ' + threadId.toString())
const obj = {
dn: req.dn.toString(),
attributes: {
objectclass: ['organization', 'top'],
o: 'example'
}
} }
server.listen(serverOptions.port, function () { if (req.filter.matches(obj.attributes)) { res.send(obj) }
console.log('ldapjs listening at ldap://127.0.0.1:' + serverOptions.port);
});
} else {
const server = ldap.createServer(serverOptions);
let threadId = process.env.id; res.end()
})
process.on('message', (msg, socket) => {
switch (msg.type) {
case 'connection':
server.newConnection(socket);
socket.resume();
console.log('ldapjs client connection accepted on ' + threadId.toString());
}
});
server.search('dc=example', function (req, res, next) {
console.log('ldapjs search initiated on ' + threadId.toString());
var obj = {
dn: req.dn.toString(),
attributes: {
objectclass: ['organization', 'top'],
o: 'example'
}
};
if (req.filter.matches(obj.attributes))
res.send(obj);
res.end();
});
} }

View File

@ -1,66 +1,65 @@
const cluster = require('cluster'); const cluster = require('cluster')
const ldap = require('ldapjs'); const ldap = require('ldapjs')
const os = require('os'); const os = require('os')
const threads = []; const threads = []
threads.getNext = function () { threads.getNext = function () {
return (Math.floor(Math.random() * this.length)); return (Math.floor(Math.random() * this.length))
}; }
const serverOptions = { const serverOptions = {
connectionRouter: (socket) => { connectionRouter: (socket) => {
socket.pause(); socket.pause()
console.log('ldapjs client requesting connection'); console.log('ldapjs client requesting connection')
let routeTo = threads.getNext(); const routeTo = threads.getNext()
threads[routeTo].send({ type: 'connection' }, socket); threads[routeTo].send({ type: 'connection' }, socket)
} }
}; }
const server = ldap.createServer(serverOptions); const server = ldap.createServer(serverOptions)
if (cluster.isMaster) { if (cluster.isMaster) {
for (let i = 0; i < os.cpus().length; i++) { for (let i = 0; i < os.cpus().length; i++) {
let thread = cluster.fork({ const thread = cluster.fork({
'id': i id: i
}); })
thread.id = i; thread.id = i
thread.on('message', function (msg) { thread.on('message', function () {
}); })
threads.push(thread); threads.push(thread)
}
server.listen(1389, function () {
console.log('ldapjs listening at ' + server.url)
})
} else {
const threadId = process.env.id
serverOptions.connectionRouter = () => {
console.log('should not be hit')
}
process.on('message', (msg, socket) => {
switch (msg.type) {
case 'connection':
server.newConnection(socket)
socket.resume()
console.log('ldapjs client connection accepted on ' + threadId.toString())
}
})
server.search('dc=example', function (req, res) {
console.log('ldapjs search initiated on ' + threadId.toString())
const obj = {
dn: req.dn.toString(),
attributes: {
objectclass: ['organization', 'top'],
o: 'example'
}
} }
server.listen(1389, function () { if (req.filter.matches(obj.attributes)) { res.send(obj) }
console.log('ldapjs listening at ' + server.url);
});
} else {
let threadId = process.env.id;
serverOptions.connectionRouter = (connection) => {
console.log('should not be hit');
};
process.on('message', (msg, socket) => { res.end()
switch (msg.type) { })
case 'connection':
server.newConnection(socket);
socket.resume();
console.log('ldapjs client connection accepted on ' + threadId.toString());
}
});
server.search('dc=example', function (req, res, next) {
console.log('ldapjs search initiated on ' + threadId.toString());
var obj = {
dn: req.dn.toString(),
attributes: {
objectclass: ['organization', 'top'],
o: 'example'
}
};
if (req.filter.matches(obj.attributes))
res.send(obj);
res.end();
});
} }