app.js

/***
 *
 * DO NOT MODIFY
 *
 */
var express = require('express');
var path = require('path');
var favicon = require('serve-favicon');
var logger = require('morgan');
var cookieParser = require('cookie-parser');
var bodyParser = require('body-parser');
var fs = require('fs');

var session = require('express-session');
var flashify = require('flashify');

var app = express();
var db = require('./db');

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'jade');

// uncomment after placing your favicon in /public
//app.use(favicon(path.join(__dirname, 'public', 'favicon.ico')));
app.use(logger('dev'));
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: false }));
app.use(session({
  secret: 'ee29cba3758dba63309461d782f8dcae',
  resave: false,
  saveUninitialized: true
}));
app.use(flashify);
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));

app.use(function(req,res,next){
  // Set x-xss-protection to false
  res.header('X-XSS-Protection', '0');
  var pathToRoutes = __dirname + '/routes';
  var categories = require('./helpers/loadModules')(pathToRoutes);
  var modules = [];

  // Get each category
  categories.forEach(function(category){
    var objectives = {};

    var levels = require('./helpers/loadModules')(pathToRoutes + '/'+category);

    // Get each level meta data
    for (var i = 0; i < levels.length; i++) {
      if (fs.existsSync(pathToRoutes + '/' +category + '/' + levels[i] + '/meta.json')) {
        objectives[levels[i]] = require(pathToRoutes + '/' +category + '/' + levels[i] + '/meta.json');
      }
      else {
        objectives[levels[i]] = {};
      }
      objectives[levels[i]].completed = false;
    }

    // Check if we have a config for the category
    if (fs.existsSync(pathToRoutes + '/'+category+'/config.json')){
      var config = require(pathToRoutes + '/'+category+'/config.json');
      modules.push({title: config.title, urlPrefix: config.urlPrefix, levels: levels, objectives: objectives});
    }
    else {
      modules.push({title: category, urlPrefix: category, levels: levels, objectives: objectives, completed: false});
    }
  });
  res.locals.modules = modules;
  next();
});

/**
 * Routes
 */

var home  = require('./routes/index'),
    cmdInj= require('./routes/cmdInj'),
    csrf  = require('./routes/csrf'),
    imajs = require('./routes/imajs'),
    nosql = require('./routes/nosql'),
    sqli  = require('./routes/sqli'),
    ssjs  = require('./routes/ssjs'),
    xss   = require('./routes/xss'),
    xml   = require('./routes/xml');

app.use('/', home);
app.use('/cmdInj', cmdInj);
app.use('/csrf/', csrf);
app.use('/imajs', imajs);
app.use('/nosql', nosql);
app.use('/sqli', sqli);
app.use('/ssjs', ssjs);
app.use('/xss', xss);
app.use('/xml', xml);

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  var err = new Error('Not Found');
  err.status = 404;
  next(err);
});

// error handlers

// development error handler
// will print stacktrace
if (app.get('env') === 'development') {
  app.use(function(err, req, res, next) {
    res.status(err.status || 500);
    res.render('error', {
      message: err.message,
      error: err
    });
  });
}

// production error handler
// no stacktraces leaked to user
app.use(function(err, req, res, next) {
  res.status(err.status || 500);
  res.render('error', {
    message: err.message,
    error: {}
  });
});


module.exports = app;