index.js

var express = require('express');
var session = require('express-session');
var bodyParser = require('body-parser');
var app = express();
var PortScan = require("./modules/scanner");
const testFolder = './Public/scan';
var lookups = require("./modules/dnslookup");
var fs = require("fs");
const passport = require('passport');
var cookieParser = require('cookie-parser');
var flash = require('connect-flash');
var util = require('util');
var parser = require('parse-whois');
var whois = require('node-whois');
require('./modules/passport.js')(passport);
var dns = require('native-dns');
var util = require('util');
var sockettool = require("./modules/socket");

app.use(bodyParser());  // BodyParser
app.use(cookieParser()); // read cookies (needed for auth)
app.set('view engine', 'ejs'); // Use view engine ejs
app.use(bodyParser.urlencoded({
	extended: true
}));
app.use(bodyParser.json());

// required for passport
app.use(session({
	secret: 'somethingsomething',
	resave: true,
	saveUninitialized: true
 } )); // session secret
app.use(passport.initialize());
app.use(passport.session()); // persistent login sessions
app.use(flash()); // use connect-flash for flash messages stored in session


app.use(express.static(__dirname + '/Public'))

	//RETURN INDEX IF LOGGED
	app.get('/', isLoggedIn, function(req, res){

	var ip =getClientIP(req);
	res.render("index", { username: req.user.username });
	});



	// =====================================
	// LOGIN ===============================
	// =====================================
	// show the login form
	app.get('/login', function(req, res) {

		// render the page and pass in any flash data if it exists
		res.render('login.ejs', { message: req.flash('loginMessage') });
	});


	// process the login form
	app.post('/login', passport.authenticate('local-login', {
            successRedirect : '/', // redirect to the secure profile section
            failureRedirect : '/login', // redirect back to the signup page if there is an error
            failureFlash : true // allow flash messages
		}),
        function(req, res) {
            console.log("hello");

            if (req.body.remember) {
              req.session.cookie.maxAge = 1000 * 60 * 3;
            } else {
              req.session.cookie.expires = false;
            }
        res.redirect('/');
    });




	// =====================================
	// SIGNUP ==============================
	// =====================================
	// show the signup form
	app.get('/signup', function(req, res) {
		// render the page and pass in any flash data if it exists
		res.render('signup.ejs', { message: req.flash('signupMessage') });
	});

	// process the signup form
	app.post('/signup', passport.authenticate('local-signup', {
		successRedirect : '/', // redirect to the secure profile section
		failureRedirect : '/signup', // redirect back to the signup page if there is an error
		failureFlash : true // allow flash messages
}));

	// =====================================
	// WHOIS ==============================
	// =====================================
	//Whois for url
	app.post('/whois', function(req, res, next) {
		var whoGet = req.body.whois;

		    whois.lookup(whoGet, function(err, data){
				if (err) throw err;
			    
				var info = parser.parseWhoIsData(data);
				//console.log(info);
				res.render("whois", {info: JSON.stringify(info) });
			    
			});
		
	});

	// =====================================
	// SOCKET ==============================
	// =====================================
	//Socket stuff
	app.post('/socket', function(req, res, next) {
		var targetip = req.body.socketip;
        var whoGet = req.body.port;
        var string = req.body.string;
        var timeout = req.body.timeout;
        var radio11 = req.body.radio11;
        var radio33 = req.body.radio33;

        var port = req.body.port;
        // HTTP GET SOCKET
		if(radio33 === "on"){
			var stringhttp = "GET / HTTP/1.0";
			var porthttp = 80;
			sockettool(targetip, porthttp, stringhttp, timeout, function (err, data){
				if (err) return console.log(err);
				console.log(data);
				res.render("layout_lookup", { list: data});

			});
        }else{
            sockettool(targetip, whoGet, string, timeout, function (err, data){
                if (err) return console.log(err);
                console.log(data);
                res.render("layout_socket", { list: data});
            });
		}

    });


	// =====================================
	// IPLOGGER ==============================
	// =====================================
	// Run scanner


	app.post('/iploggin', function(req, res){

	//NAME
	 	var nameR = req.body.nameR;
	//IP & RANGE
		var code = req.body.name2;
		var range = req.body.range;
		var rangef = "full";
	//ARRAY FROM RANGE
		var range2 = range.split("-");
		var list = [];
		for(i = range2[0]; i<= range2[1]; i++){
		list.push(i);
		}
        var listfull = [];
        for(i = 0; i<= 655000; i++){
            rangef.push(i);
        }
	//OPTIONS
		var radioa = req.body.radio1; // ask for banners
		var radiob = req.body.radio2; // most common
		var radioc = req.body.radio3; // quick
        var radiod = req.body.radio4; // full scan
		//FULL SCAN
			if(radiod === "on"){
				var set = PortScan(code, listfull, nameR, range, req.user.username);
				var iphost = getClientIP(req);
			}
		//NORMAL SCAN
		var set = PortScan(code, list, nameR, rangef, req.user.username);
		var iphost = getClientIP(req);



	res.render("index", { username: req.user.username });

});

	// =====================================
	// DNSLOOKUP ==============================
	// =====================================
	// Dnslookup module resolve domain to ip


	app.post('/lookup', function(req, res){
		var url = req.body.dnslookup;

			lookups(url, function (err, data){
				if (err) return console.log(err);
				//console.log(data);
				res.render("layout_lookup", { list: data});

               });

	});


	// =====================================
	// REPORTS ==============================
	// =====================================
	//Bring back scan reports with id from MySQL
	//and loop them

	app.get('/Public/scan/*', function(req, res){

	res.sendFile(__dirname + req.url);



	});


	//DB//
	app.get('/data', isLoggedIn, function (req, res) {
	var mysql = require('mysql');
	var obj = {};
	var connection = mysql.createConnection({
	  host     : 'localhost',
	  user     : 'root',
	  password : '38613861',
	  database : 'ipRes'
	});



	connection.connect()

		//if admin
		if(req.user.username == "admin"){

			connection.query("SELECT * FROM ipRes", function (err, result) {


			if(err){
			    throw err;
			} else {
			    obj = {print: result};
			    res.render('report', obj);                
			}

		//not admin
		}); } else{
		connection.query("SELECT * FROM ipRes where owner = ?", [req.user.username],  function (err, result) {


				if(err){
				    throw err;
				} else {
				    obj = {print: result};
				    res.render('report', obj);
				}
			
			});
		}
		

});

	// =====================================
	// IS LOGGED IN CHECK ====================
	// =====================================
	//Check if logged in


	function isLoggedIn(req, res, next) {

		// if user is authenticated in the session, carry on
		if (req.isAuthenticated())
			return next();

		// if they aren't redirect them to the home page
		res.redirect('/login');
	}

	// =====================================
	// Get client ip   ====================
	// =====================================
	// get client

function getClientIP(req){
    return req.headers['x-forwarded-for'] || req.connection.remoteAddress;
}

app.listen(8080);