wild card socket lookup
This commit is contained in:
+164
-33
@@ -16,74 +16,205 @@ const Host = require('../utils/redis_model')({
|
||||
}
|
||||
});
|
||||
|
||||
Host.lookupDict = {};
|
||||
Host.lookUpObj = {};
|
||||
|
||||
Host.buildLookUpObj = async function(){
|
||||
/*
|
||||
Build a look up tree for domain records in the redis back end to allow
|
||||
complex looks with wildcards.
|
||||
*/
|
||||
|
||||
// Hold lookUp ready while the look up object is being built.
|
||||
this.__lookUpIsReady = false;
|
||||
|
||||
Host.make_lookup = async function(){
|
||||
try{
|
||||
|
||||
// Loop over all the hosts in the redis.
|
||||
for(let host of await this.list()){
|
||||
|
||||
// Spit the hosts on "." into its fragments .
|
||||
let fragments = host.split('.');
|
||||
let place = this.lookupDict;
|
||||
|
||||
while(fragments.length !==0){
|
||||
let current = fragments.pop();
|
||||
if(!place[current]){
|
||||
place[current] = {};
|
||||
// Hold a pointer to the root of the lookup tree.
|
||||
let pointer = this.lookUpObj;
|
||||
|
||||
// Walk over each fragment, popping from right to left.
|
||||
while(fragments.length){
|
||||
let fragment = fragments.pop();
|
||||
|
||||
// Add a branch to the lookup at the current position
|
||||
if(!pointer[fragment]){
|
||||
pointer[fragment] = {};
|
||||
}
|
||||
|
||||
// Add the record(leaf) when we hit the a full host name.
|
||||
// #record denotes a leaf node on this tree.
|
||||
if(fragments.length === 0){
|
||||
place[current]['#record'] = await this.get(host)
|
||||
pointer[fragment]['#record'] = await this.get(host)
|
||||
}
|
||||
place = place[current];
|
||||
|
||||
// Advance the pointer to the next level of the tree.
|
||||
pointer = pointer[fragment];
|
||||
}
|
||||
}
|
||||
|
||||
// When the look up tree is finished, remove the ready hold.
|
||||
this.__lookUpIsReady = true;
|
||||
|
||||
}catch(error){
|
||||
console.error(error);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
Host.lookUp = async function(host){
|
||||
let place = this.lookupDict;
|
||||
Host.lookUp = function(host){
|
||||
/*
|
||||
Perform a complex lookup of @host on the look up tree.
|
||||
*/
|
||||
|
||||
|
||||
// Hold a pointer to the root of the look up tree
|
||||
let place = this.lookUpObj;
|
||||
|
||||
// Hold the last passed long wild card.
|
||||
let last_resort = {};
|
||||
|
||||
for(let fragment of host.split('.').slice().reverse()){
|
||||
// Walk over each fragment of the host, from right to left
|
||||
for(let fragment of host.split('.').reverse()){
|
||||
|
||||
// If a long wild card is found on this level, hold on to it
|
||||
if(place['**']) last_resort = place['**'];
|
||||
|
||||
if({...place, ...last_resort}[fragment]){
|
||||
place = {...place, ...last_resort}[fragment];
|
||||
// If we have a match for the current fragment, update the current pointer
|
||||
// A match in the lookup tree takes priority being a more exact match.
|
||||
if({...last_resort, ...place}[fragment]){
|
||||
place = {...last_resort, ...place}[fragment];
|
||||
// If we have a not exact fragment match, a wild card will do.
|
||||
}else if(place['*']){
|
||||
place = place['*']
|
||||
// If no fragment can be matched, continue with the long wild card branch.
|
||||
}else if(last_resort){
|
||||
place = last_resort;
|
||||
}
|
||||
}
|
||||
|
||||
// After the tree has been traversed, see if we have leaf node to return.
|
||||
if(place && place['#record']) return place['#record'];
|
||||
}
|
||||
};
|
||||
|
||||
Host.__lookUpIsReady = false;
|
||||
|
||||
Host.lookUpReady = async function(){
|
||||
/*
|
||||
Wait for the lookup tree to be built.
|
||||
*/
|
||||
|
||||
// Check every 5ms to see if the look up tree is ready
|
||||
while(!this.__lookUpIsReady) await new Promise(r => setTimeout(r, 5));
|
||||
return true;
|
||||
};
|
||||
|
||||
(async function(){
|
||||
await Host.buildLookUpObj();
|
||||
})()
|
||||
|
||||
|
||||
var net = require('net');
|
||||
var fs = require('fs');
|
||||
|
||||
// This server listens on a Unix socket at /var/run/mysocket
|
||||
var unixServer = net.createServer(function(client) {
|
||||
// Do something with the client connection
|
||||
});
|
||||
|
||||
unixServer.on('connection', function(clientSocket){
|
||||
let buffer = '';
|
||||
|
||||
console.log('server EVENT connection from client:', clientSocket.remoteAddress);
|
||||
|
||||
// When a connection is started, send a message informing the remote
|
||||
// peer of our ID
|
||||
|
||||
clientSocket.on('data', function(data){
|
||||
buffer += data.toString();
|
||||
try{
|
||||
// p2p.__read(JSON.parse(buffer), clientSocket.remoteAddress, clientSocket);
|
||||
console.log('buffer', buffer, Host.lookUp(buffer))
|
||||
clientSocket.write(JSON.stringify(Host.lookUp(buffer)|| {host: 'none'}));
|
||||
}catch(error){
|
||||
;
|
||||
}
|
||||
});
|
||||
|
||||
clientSocket.on('close', function(){
|
||||
console.log('info', `server Peer ${clientSocket.remoteAddress} - ${clientSocket.peerID} droped.`);
|
||||
|
||||
});
|
||||
clientSocket.on('error', function(error){
|
||||
console.log(error)
|
||||
})
|
||||
|
||||
});
|
||||
|
||||
unixServer.on('error', function(error){
|
||||
console.log(error)
|
||||
})
|
||||
|
||||
var SOCKETFILE = '/var/run/mysocket'
|
||||
|
||||
unixServer.on('listening', function(){
|
||||
fs.chmodSync(SOCKETFILE, '777');
|
||||
console.log('info','p2p server listening on')
|
||||
});
|
||||
|
||||
|
||||
|
||||
fs.stat(SOCKETFILE, function (err, stats) {
|
||||
if (err) {
|
||||
// start server
|
||||
console.log('No leftover socket found.');
|
||||
server = createServer(SOCKETFILE); return;
|
||||
}
|
||||
// remove file then start server
|
||||
console.log('Removing leftover socket.')
|
||||
fs.unlink(SOCKETFILE, function(err){
|
||||
if(err){
|
||||
// This should never happen.
|
||||
console.error(err); process.exit(0);
|
||||
}
|
||||
unixServer.listen(SOCKETFILE);
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
module.exports = {Host};
|
||||
|
||||
(async function(){
|
||||
await Host.make_lookup();
|
||||
// console.log(Host.lookupDict)
|
||||
|
||||
// console.log(Host.lookupDict['com']['vm42'])
|
||||
await Host.lookUpReady();
|
||||
|
||||
// console.log('test-res', await Host.lookUp('payments.test.com'))
|
||||
// console.log(Host.lookUpObj)
|
||||
|
||||
let count = 5
|
||||
console.log(count++, (await Host.lookUp('payments.test.com')).host === 'payments.**')
|
||||
console.log(count++, (await Host.lookUp('test.sample.other.exmaple.com')).host === '**.exmaple.com')
|
||||
console.log(count++, (await Host.lookUp('stan.test.vm42.com')).host === 'stan.test.vm42.com')
|
||||
console.log(count++, (await Host.lookUp('test.vm42.com')).host === 'test.vm42.com')
|
||||
console.log(count++, (await Host.lookUp('blah.test.vm42.com')).host === '*.test.vm42.com')
|
||||
console.log(count++, (await Host.lookUp('payments.example.com')).host === 'payments.**')
|
||||
console.log(count++, (await Host.lookUp('info.wma.users.718it.biz')).host === 'info.*.users.718it.biz')
|
||||
console.log(count++, (await Host.lookUp('info.users.718it.biz')) === undefined)
|
||||
console.log(count++, (await Host.lookUp('test.1.2.718it.net')).host === 'test.*.*.718it.net')
|
||||
console.log(count++, (await Host.lookUp('test1.exmaple.com')).host === 'test1.exmaple.com')
|
||||
console.log(count++, (await Host.lookUp('other.exmaple.com')).host === '*.exmaple.com')
|
||||
// console.log(Host.lookUpObj['com']['vm42'])
|
||||
|
||||
// console.log('test-res', await Host.lookUp('payments.718it.biz'))
|
||||
|
||||
let count = 6
|
||||
console.log(count++, Host.lookUp('payments.718it.biz').host === 'payments.718it.biz')
|
||||
console.log(count++, Host.lookUp('sd.blah.test.vm42.com') === undefined)
|
||||
console.log(count++, Host.lookUp('payments.test.com').host === 'payments.**')
|
||||
console.log(count++, Host.lookUp('test.sample.other.exmaple.com').host === '**.exmaple.com')
|
||||
console.log(count++, Host.lookUp('stan.test.vm42.com').host === 'stan.test.vm42.com')
|
||||
console.log(count++, Host.lookUp('test.vm42.com').host === 'test.vm42.com')
|
||||
console.log(count++, Host.lookUp('blah.test.vm42.com').host === '*.test.vm42.com')
|
||||
console.log(count++, Host.lookUp('payments.example.com').host === 'payments.**')
|
||||
console.log(count++, Host.lookUp('info.wma.users.718it.biz').host === 'info.*.users.718it.biz')
|
||||
console.log(count++, Host.lookUp('infof.users.718it.biz') === undefined)
|
||||
console.log(count++, Host.lookUp('blah.biz') === undefined)
|
||||
console.log(count++, Host.lookUp('test.1.2.718it.net').host === 'test.*.*.718it.net')
|
||||
console.log(count++, Host.lookUp('test1.exmaple.com').host === 'test1.exmaple.com')
|
||||
console.log(count++, Host.lookUp('other.exmaple.com').host === '*.exmaple.com')
|
||||
console.log(count++, Host.lookUp('info.payments.example.com').host === 'info.**')
|
||||
console.log(count++, Host.lookUp('718it.biz').host === '718it.biz')
|
||||
|
||||
|
||||
return ;
|
||||
})()
|
||||
|
||||
Reference in New Issue
Block a user