2019-03-24 17:57:54 +11:00
|
|
|
#!/usr/bin/node
|
|
|
|
|
|
|
|
// Get some libraries
|
2019-04-10 13:07:31 +10:00
|
|
|
const winston = require("winston");
|
2019-03-30 18:34:19 +11:00
|
|
|
const random_bytes = require("random-bytes");
|
|
|
|
const bsplit = require("buffer-split");
|
|
|
|
const node_rsa = require("node-rsa");
|
2019-04-10 20:23:01 +10:00
|
|
|
const sqlite3 = require('sqlite3');
|
2019-04-09 19:19:02 +10:00
|
|
|
const btoa = require("btoa");
|
|
|
|
const atob = require("atob");
|
2019-03-24 17:57:54 +11:00
|
|
|
const net = require("net");
|
|
|
|
const fs = require("fs");
|
|
|
|
|
2019-03-30 18:34:19 +11:00
|
|
|
// Load the settings
|
|
|
|
var settings = require("./settings.json");
|
|
|
|
|
2019-04-10 20:23:01 +10:00
|
|
|
// Create a sqlite3 database
|
|
|
|
var db = new sqlite3.Database("./app.db", function(err)
|
|
|
|
{
|
|
|
|
// Was there an error
|
|
|
|
if(err)
|
|
|
|
{
|
|
|
|
// Tell the user the error
|
|
|
|
console.log("FATAL: SQlite3 failed to start:", err);
|
|
|
|
}
|
|
|
|
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// Tell the user it worked
|
|
|
|
console.log("INFO: SQlite3 ready.");
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
function sqlite3_createTable(name, callback)
|
|
|
|
{
|
|
|
|
// Run the create table task with the callback
|
|
|
|
db.run('CREATE TABLE IF NOT EXISTS '+name
|
|
|
|
+'(id INTEGER PRIMARY KEY AUTOINCREMENT, data TEXT)', callback);
|
|
|
|
}
|
|
|
|
|
|
|
|
function sqlite3_insertData(name, data)
|
|
|
|
{
|
|
|
|
// Write data to the table
|
|
|
|
db.run('INSERT INTO '+name+' (data) VALUES (?)', [JSON.stringify(data)]);
|
|
|
|
}
|
|
|
|
|
|
|
|
function sqlite3_getData(name, callback)
|
|
|
|
{
|
|
|
|
// Get the sqlite3 data and send it to the callback
|
|
|
|
db.all('SELECT rowid AS id, data FROM '+name, callback);
|
|
|
|
}
|
|
|
|
|
2019-04-12 17:09:24 +10:00
|
|
|
function getrows_sqlite3(name, callback)
|
2019-04-10 20:23:01 +10:00
|
|
|
{
|
2019-04-12 17:09:24 +10:00
|
|
|
// Create the table
|
|
|
|
sqlite3_createTable(name, function()
|
2019-04-10 20:23:01 +10:00
|
|
|
{
|
2019-04-12 17:09:24 +10:00
|
|
|
// Load the table
|
|
|
|
sqlite3_getData(name, function(err, rows)
|
2019-04-10 20:23:01 +10:00
|
|
|
{
|
2019-04-12 17:09:24 +10:00
|
|
|
// Throw an error if there is one
|
|
|
|
if(err) throw err;
|
2019-04-10 20:23:01 +10:00
|
|
|
|
2019-04-12 17:09:24 +10:00
|
|
|
// Loop over the rows
|
|
|
|
for(var i=0;i<rows.length;i++)
|
|
|
|
{
|
|
|
|
// Parse the data
|
|
|
|
var data = JSON.parse(rows[i].data);
|
|
|
|
|
|
|
|
// Call the callback
|
|
|
|
callback(data, i);
|
|
|
|
}
|
|
|
|
});
|
2019-04-10 20:23:01 +10:00
|
|
|
});
|
2019-04-12 17:09:24 +10:00
|
|
|
}
|
|
|
|
|
|
|
|
// Load some databases
|
|
|
|
var users = {};
|
|
|
|
var chats = {};
|
|
|
|
|
|
|
|
// Get the users table
|
|
|
|
getrows_sqlite3("users", function(data)
|
|
|
|
{
|
|
|
|
// Load a user
|
|
|
|
users[data.username] = new Object();
|
|
|
|
users[data.username].password = data.password;
|
|
|
|
});
|
|
|
|
|
|
|
|
// Get the chats table
|
|
|
|
getrows_sqlite3("chats", function(data)
|
|
|
|
{
|
|
|
|
// Create a new object at this chat id
|
|
|
|
chats[data.id] = new Object();
|
|
|
|
chats[data.id].messages = [];
|
|
|
|
chats[data.id].messages = data.name;
|
2019-04-10 20:23:01 +10:00
|
|
|
});
|
2019-03-30 18:34:19 +11:00
|
|
|
|
2019-04-12 17:09:24 +10:00
|
|
|
// Create a new messages table
|
|
|
|
getrows_sqlite3("messages", function(data)
|
|
|
|
{
|
|
|
|
// Push all the messages to the chats
|
|
|
|
chats[data.chat].messages.push(data.message);
|
2019-04-10 20:23:01 +10:00
|
|
|
});
|
|
|
|
|
|
|
|
console.log("Ready.");
|
2019-04-09 19:19:02 +10:00
|
|
|
|
|
|
|
function toBuffer(bytes)
|
|
|
|
{
|
|
|
|
// String
|
|
|
|
if(typeof(bytes) == 'string')
|
|
|
|
{
|
|
|
|
// Create a buffer
|
|
|
|
var buffer = new Buffer.alloc(bytes.length);
|
|
|
|
|
|
|
|
// Loop over the bytes
|
|
|
|
for(var i=0;i<bytes.length;i++)
|
|
|
|
{
|
|
|
|
// Save to the buffer
|
|
|
|
buffer[i] = bytes[i].charCodeAt(0);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the buffer
|
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return if this is anything else
|
|
|
|
return bytes;
|
|
|
|
}
|
|
|
|
|
|
|
|
function toBytes(buffer)
|
|
|
|
{
|
|
|
|
// Make sure the buffer is a buffer
|
|
|
|
buffer = toBuffer(buffer);
|
|
|
|
|
|
|
|
// Create some bytes
|
|
|
|
var bytes = "";
|
|
|
|
|
|
|
|
// Loop over the buffer
|
|
|
|
for(var i=0;i<buffer.length;i++)
|
|
|
|
{
|
|
|
|
// Add to the bytes string
|
|
|
|
bytes += String.fromCharCode(buffer[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the bytes
|
|
|
|
return bytes;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Recieve helper global varibles
|
|
|
|
var recieve_buffer = new Buffer.alloc(0);
|
|
|
|
var recieve_get = false;
|
|
|
|
var recieve_size = 0;
|
|
|
|
|
|
|
|
function recieve_ordered(data, callback)
|
|
|
|
{
|
|
|
|
// Convert the data into a buffer
|
|
|
|
data = toBuffer(data);
|
|
|
|
|
|
|
|
// Loop over the data
|
|
|
|
for(var i=0;i<data.length;i++)
|
|
|
|
{
|
|
|
|
// Add the data to the buffer
|
|
|
|
recieve_buffer = new Buffer.concat([recieve_buffer, Buffer([data[i]])]);
|
|
|
|
|
|
|
|
// Is the buffer getting data
|
|
|
|
if(!recieve_get)
|
|
|
|
{
|
|
|
|
// Does the buffer contain a number
|
|
|
|
if(recieve_buffer.length >= 4)
|
|
|
|
{
|
|
|
|
// Get the number
|
|
|
|
recieve_size = recieve_buffer.readUInt32BE(0);
|
|
|
|
|
|
|
|
// Reset the buffer
|
|
|
|
recieve_buffer = new Buffer.alloc(0);
|
|
|
|
|
|
|
|
// Set the get data mode to true
|
|
|
|
recieve_get = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// Is the recieve buffer as big as the size
|
|
|
|
if(recieve_buffer.length == recieve_size)
|
|
|
|
{
|
|
|
|
// Call the callback
|
|
|
|
callback(recieve_buffer);
|
|
|
|
|
|
|
|
// Reset the buffer and the size
|
|
|
|
recieve_buffer = new Buffer.alloc(0);
|
|
|
|
recieve_size = 0;
|
|
|
|
|
|
|
|
// Set the get data mode to false
|
|
|
|
recieve_get = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function send_ordered(data)
|
|
|
|
{
|
|
|
|
// Convert the data into a buffer
|
|
|
|
data = toBuffer(data);
|
|
|
|
|
|
|
|
// Get the size of the data
|
|
|
|
var size = data.length;
|
|
|
|
|
|
|
|
// Turn it into a 32 bit buffer
|
|
|
|
var bytes = new Buffer.alloc(4);
|
|
|
|
bytes.writeUInt32BE(size, 0);
|
|
|
|
|
|
|
|
// Return the buffer and the data
|
|
|
|
return new Buffer.concat([bytes, data]);
|
|
|
|
}
|
|
|
|
|
2019-03-30 18:34:19 +11:00
|
|
|
function string_decrypt(key, string)
|
|
|
|
{
|
|
|
|
// Convert the string to a buffer
|
2019-04-09 19:19:02 +10:00
|
|
|
var buff = toBuffer(string);
|
2019-03-30 18:34:19 +11:00
|
|
|
|
|
|
|
// Create some new data
|
|
|
|
var decrypted = new Buffer.alloc(buff.length);
|
|
|
|
|
|
|
|
// Iterate over the string
|
|
|
|
for(var i=0;i<string.length;i++)
|
|
|
|
{
|
|
|
|
// Convert the string item to a number
|
|
|
|
var d = buff[i]-key.str[key.at.rx];
|
|
|
|
while(d < 0) d += 256;
|
|
|
|
decrypted[i] = d;
|
|
|
|
|
|
|
|
// Add 1 to the key counter
|
|
|
|
key.at.rx += 1;
|
|
|
|
|
|
|
|
// Is the key counter out of range
|
|
|
|
if(key.at.rx >= key.str.length)
|
|
|
|
{
|
|
|
|
// Set to zero
|
|
|
|
key.at.rx = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the encrypted data
|
|
|
|
return decrypted;
|
|
|
|
}
|
|
|
|
|
|
|
|
function string_encrypt(key, string)
|
|
|
|
{
|
|
|
|
// Convert the string to a buffer
|
2019-04-09 19:19:02 +10:00
|
|
|
var buff = toBuffer(string);
|
2019-03-30 18:34:19 +11:00
|
|
|
|
|
|
|
// Create some new data
|
|
|
|
var encrypted = new Buffer.alloc(buff.length);
|
|
|
|
|
|
|
|
// Iterate over the string
|
|
|
|
for(var i=0;i<string.length;i++)
|
|
|
|
{
|
|
|
|
// Convert the string item to a number
|
|
|
|
var e = buff[i]+key.str[key.at.tx];
|
|
|
|
while(e > 255) e -= 256;
|
|
|
|
encrypted[i] = e;
|
|
|
|
|
|
|
|
// Add 1 to the key counter
|
|
|
|
key.at.tx += 1;
|
|
|
|
|
|
|
|
// Is the key counter out of range
|
|
|
|
if(key.at.tx >= key.str.length)
|
|
|
|
{
|
|
|
|
// Set to zero
|
|
|
|
key.at.tx = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the encrypted data
|
|
|
|
return encrypted;
|
|
|
|
}
|
|
|
|
|
|
|
|
function make_encryption_key(string)
|
|
|
|
{
|
|
|
|
// Make a new key
|
|
|
|
var key = new Object();
|
|
|
|
|
|
|
|
// Set the varibles
|
2019-04-09 21:03:39 +10:00
|
|
|
key.str = toBuffer(string);
|
2019-03-30 18:34:19 +11:00
|
|
|
key.at = new Object();
|
|
|
|
key.at.rx = 0;
|
|
|
|
key.at.tx = 0;
|
|
|
|
|
2019-04-10 13:07:31 +10:00
|
|
|
//console.log("make_encryption_key:", key.str);
|
2019-04-09 21:03:39 +10:00
|
|
|
|
2019-03-30 18:34:19 +11:00
|
|
|
// Return the key
|
|
|
|
return key;
|
|
|
|
}
|
|
|
|
|
|
|
|
function socket_write(socket, data)
|
|
|
|
{
|
|
|
|
// Send the data encrypted with JSON
|
2019-04-09 19:19:02 +10:00
|
|
|
socket.sock.write(send_ordered(string_encrypt(socket.key, JSON.stringify(data)+"\n")));
|
2019-03-30 18:34:19 +11:00
|
|
|
}
|
|
|
|
|
2019-04-10 20:23:01 +10:00
|
|
|
function socket_init(socket, ondata)
|
2019-03-30 18:34:19 +11:00
|
|
|
{
|
|
|
|
// Set the socket
|
|
|
|
var sock = new Object();
|
|
|
|
sock.sock = socket;
|
|
|
|
sock.new = true;
|
|
|
|
|
2019-04-10 13:07:31 +10:00
|
|
|
//console.log("Connection from "+socket.localAdress);
|
2019-03-30 21:27:36 +11:00
|
|
|
|
2019-03-30 18:34:19 +11:00
|
|
|
// Wait for data
|
|
|
|
sock.sock.on('data', function(data)
|
|
|
|
{
|
2019-04-09 19:19:02 +10:00
|
|
|
// Recieve data in order
|
|
|
|
recieve_ordered(data, function(data)
|
2019-03-30 18:34:19 +11:00
|
|
|
{
|
2019-04-10 13:07:31 +10:00
|
|
|
//console.log(data);
|
2019-03-30 18:34:19 +11:00
|
|
|
|
2019-04-09 19:19:02 +10:00
|
|
|
// Is the socket new
|
|
|
|
if(sock.new)
|
2019-03-30 18:34:19 +11:00
|
|
|
{
|
2019-04-09 19:19:02 +10:00
|
|
|
// Set sock new
|
|
|
|
sock.new = false;
|
|
|
|
|
|
|
|
// Convert from JSON
|
|
|
|
data = JSON.parse(data);
|
|
|
|
|
|
|
|
// Is this the public key
|
|
|
|
if(data.mode == 'pubkey')
|
|
|
|
{
|
|
|
|
// Load the key
|
|
|
|
var key = new node_rsa();
|
|
|
|
key.importKey(data.key, 'public');
|
2019-04-10 13:07:31 +10:00
|
|
|
console.log("Loaded the RSA key");
|
2019-04-09 19:19:02 +10:00
|
|
|
|
|
|
|
// Get some random bytes
|
|
|
|
random_bytes(settings.encryption_key_size, function(error,string)
|
|
|
|
{
|
2019-04-10 13:07:31 +10:00
|
|
|
//console.log("btoa(string):",btoa(string));
|
2019-04-09 19:19:02 +10:00
|
|
|
|
|
|
|
// Throw an error if there is one
|
|
|
|
if(error) throw error;
|
|
|
|
|
|
|
|
// Make the key
|
|
|
|
sock.key = make_encryption_key(string);
|
|
|
|
console.log("Created an encryption key")
|
|
|
|
|
|
|
|
// Encrypt the key with RSA
|
2019-04-09 21:03:39 +10:00
|
|
|
var key_encrypted = key.encrypt(string, 'base64');
|
2019-04-10 13:07:31 +10:00
|
|
|
console.log("Encrypted the key");
|
2019-04-09 19:19:02 +10:00
|
|
|
|
|
|
|
// Send the key to the client
|
|
|
|
sock.sock.write(send_ordered(JSON.stringify({
|
|
|
|
key: toBytes(key_encrypted),
|
|
|
|
mode: "encryption_key"
|
|
|
|
})));
|
|
|
|
|
|
|
|
console.log("Sent the key to the client");
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
2019-03-30 18:34:19 +11:00
|
|
|
|
2019-04-09 19:19:02 +10:00
|
|
|
else
|
2019-03-30 18:34:19 +11:00
|
|
|
{
|
2019-04-09 19:19:02 +10:00
|
|
|
// Decrypt the data
|
|
|
|
data = string_decrypt(sock.key, data);
|
|
|
|
|
2019-04-10 13:07:31 +10:00
|
|
|
//console.log(data);
|
2019-04-09 19:19:02 +10:00
|
|
|
|
|
|
|
// Parse the json data
|
|
|
|
var data = JSON.parse(data);
|
|
|
|
|
2019-04-10 20:23:01 +10:00
|
|
|
// Send the parsed data to the callback
|
|
|
|
ondata(data);
|
2019-03-30 18:34:19 +11:00
|
|
|
}
|
2019-04-09 19:19:02 +10:00
|
|
|
});
|
2019-03-30 18:34:19 +11:00
|
|
|
});
|
|
|
|
|
|
|
|
// Return the socket
|
|
|
|
return sock;
|
|
|
|
}
|
|
|
|
|
2019-03-24 17:57:54 +11:00
|
|
|
// Create a server
|
|
|
|
var server = net.createServer(function(socket)
|
|
|
|
{
|
2019-04-10 20:23:01 +10:00
|
|
|
// Setup the sock varible
|
2019-03-30 18:34:19 +11:00
|
|
|
var sock;
|
2019-03-24 17:57:54 +11:00
|
|
|
|
2019-03-30 18:34:19 +11:00
|
|
|
// Initialise the socket and wait for data
|
2019-04-10 20:23:01 +10:00
|
|
|
sock = socket_init(socket, function(data)
|
2019-03-30 18:34:19 +11:00
|
|
|
{
|
2019-04-10 20:23:01 +10:00
|
|
|
// Does the user want to login
|
|
|
|
if(data.mode == "login")
|
|
|
|
{
|
2019-04-12 17:09:24 +10:00
|
|
|
// Set the logged in varible
|
|
|
|
var logged_in = false;
|
|
|
|
|
2019-04-10 20:23:01 +10:00
|
|
|
// Is the user registered here
|
|
|
|
if(users[data.username])
|
|
|
|
{
|
|
|
|
// Is the password correct
|
|
|
|
if(users[data.username].password == data.password)
|
|
|
|
{
|
|
|
|
// Log the user in
|
|
|
|
sock.username = data.username;
|
|
|
|
|
|
|
|
console.log("Logged "+sock.username+" in.");
|
|
|
|
|
2019-04-12 17:09:24 +10:00
|
|
|
// Set logged in
|
|
|
|
logged_in = true;
|
2019-04-10 20:23:01 +10:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-04-12 17:09:24 +10:00
|
|
|
// Is register on fail set
|
|
|
|
else if(settings.register_on_fail)
|
2019-04-10 20:23:01 +10:00
|
|
|
{
|
|
|
|
// Make a new user
|
|
|
|
users[data.username] = new Object();
|
|
|
|
users[data.username].password = data.password;
|
|
|
|
|
|
|
|
// Send it to sqlite3
|
|
|
|
sqlite3_insertData("users", {
|
|
|
|
username: data.username,
|
|
|
|
password: data.password
|
|
|
|
});
|
|
|
|
|
|
|
|
console.log("Creating account "+data.username+".");
|
2019-04-12 17:09:24 +10:00
|
|
|
|
|
|
|
// Set logged in
|
|
|
|
logged_in = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Is the user now logged in
|
|
|
|
if(logged_in)
|
|
|
|
{
|
|
|
|
// Get a list of chat keys
|
|
|
|
var chat_keys = Object.keys(chats);
|
|
|
|
var chats_client = {};
|
|
|
|
|
|
|
|
// Loop over them
|
|
|
|
for(var i=0;i<chat_keys.length;i++)
|
|
|
|
{
|
|
|
|
// Register the key to the temporary varible
|
|
|
|
chats_client[chat_keys[i]] = new Object();
|
|
|
|
|
|
|
|
// Set the chat name
|
|
|
|
chats_client[chat_keys[i]].name = chats[chat_keys[i]].name;
|
|
|
|
chats_client[chat_keys[i]].messages = chat[chat_keys[i]].messages;
|
|
|
|
}
|
|
|
|
|
|
|
|
console.log("Sending list of chats");
|
|
|
|
|
|
|
|
// Send the list of chats
|
|
|
|
socket_write(sock, {
|
|
|
|
mode: "chat_list",
|
|
|
|
chats: chats_client
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
else
|
|
|
|
{
|
|
|
|
console.log("Authentication failed.");
|
|
|
|
|
|
|
|
// Send back an error message
|
|
|
|
socket_write(sock, {
|
|
|
|
mode: "error",
|
|
|
|
error: "auth"
|
|
|
|
});
|
2019-04-10 20:23:01 +10:00
|
|
|
}
|
|
|
|
}
|
2019-03-24 17:57:54 +11:00
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
// Listen for data
|
|
|
|
server.listen(22068, '');
|