Log what is retrieved from headers to help debugging.

scripts/authelia-scripts-suites-test

@@ -19,7 +19,7 @@ async function runTests(suite, withEnv) {
   return new Promise((resolve, reject) => {
     let mochaArgs = ['--exit', '--colors', '--require', 'ts-node/register', 'test/suites/' + suite + '/test.ts']
     if (program.forbidOnly) {
-      mochaArgs = ['--forbid-only', '--forbid-pending'] + mochaArgs;
+      mochaArgs = ['--forbid-only', '--forbid-pending'].concat(mochaArgs);
     }
   
     const mochaCommand = './node_modules/.bin/mocha ' + mochaArgs.join(' ');

scripts/authelia-scripts-unittest

@@ -4,11 +4,17 @@ var program = require('commander');
 var spawn = require('child_process').spawn;
 
 program
+  .option('--forbid-only', 'Forbid only and pending filters.')
   .parse(process.argv);
 
-async function runTests(pattern) {
+async function runTests(patterns) {
   return new Promise((resolve, reject) => {
-    mocha = spawn('./node_modules/.bin/mocha', ['--colors', '--require', 'ts-node/register', pattern], {
+    let mochaArgs = ['--require', 'ts-node/register', '--require', './spec-helper.js', '--colors'];
+    if (program.forbidOnly) {
+      mochaArgs = ['--forbid-only', '--forbid-pending'].concat(mochaArgs);
+    }
+
+    const mocha = spawn('./node_modules/.bin/mocha', mochaArgs.concat(patterns), {
       env: {
         ...process.env,
         'TS_NODE_PROJECT': './server/tsconfig.json'
@@ -27,8 +33,10 @@ async function runTests(pattern) {
 }
 
 async function test() {
-  await runTests('server/src/**/*.spec.ts');
-  await runTests('shared/**/*.spec.ts');
+  await runTests([
+    'server/src/**/*.spec.ts',
+    'shared/**/*.spec.ts'
+  ]);
 }
 
 test()

server/src/lib/ErrorReplies.ts

@@ -1,25 +1,18 @@
-import express = require("express");
+import * as Express from "express";
 import { IRequestLogger } from "./logging/IRequestLogger";
 
-function replyWithError(req: express.Request, res: express.Response,
+function replyWithError(req: Express.Request, res: Express.Response,
   code: number, logger: IRequestLogger, body?: Object): (err: Error) => void {
   return function (err: Error): void {
-    if (req.originalUrl.startsWith("/api/") || code == 200) {
-      logger.error(req, "Reply with error %d: %s", code, err.message);
-      logger.debug(req, "%s", err.stack);
-      res.status(code);
-      res.send(body);
-    }
-    else {
-      logger.error(req, "Redirect to error %d: %s", code, err.message);
-      logger.debug(req, "%s", err.stack);
-      res.redirect("/error/" + code);
-    }
+    logger.error(req, "Reply with error %d: %s", code, err.message);
+    logger.debug(req, "%s", err.stack);
+    res.status(code);
+    res.send(body);
   };
 }
 
-export function redirectTo(redirectUrl: string, req: express.Request,
-  res: express.Response, logger: IRequestLogger) {
+export function redirectTo(redirectUrl: string, req: Express.Request,
+  res: Express.Response, logger: IRequestLogger) {
   return function(err: Error) {
     logger.error(req, "Error: %s", err.message);
     logger.debug(req, "Redirecting to %s", redirectUrl);
@@ -27,22 +20,22 @@ export function redirectTo(redirectUrl: string, req: express.Request,
   };
 }
 
-export function replyWithError400(req: express.Request,
-  res: express.Response, logger: IRequestLogger) {
+export function replyWithError400(req: Express.Request,
+  res: Express.Response, logger: IRequestLogger) {
   return replyWithError(req, res, 400, logger);
 }
 
-export function replyWithError401(req: express.Request,
-  res: express.Response, logger: IRequestLogger) {
+export function replyWithError401(req: Express.Request,
+  res: Express.Response, logger: IRequestLogger) {
   return replyWithError(req, res, 401, logger);
 }
 
-export function replyWithError403(req: express.Request,
-  res: express.Response, logger: IRequestLogger) {
+export function replyWithError403(req: Express.Request,
+  res: Express.Response, logger: IRequestLogger) {
   return replyWithError(req, res, 403, logger);
 }
 
-export function replyWithError200(req: express.Request,
-  res: express.Response, logger: IRequestLogger, message: string) {
+export function replyWithError200(req: Express.Request,
+  res: Express.Response, logger: IRequestLogger, message: string) {
   return replyWithError(req, res, 200, logger, { error: message });
 }

server/src/lib/IdentityCheckMiddleware.spec.ts

@@ -1,10 +1,9 @@
 
-import sinon = require("sinon");
+import * as Sinon from "sinon";
 import * as IdentityCheckMiddleware from "./IdentityCheckMiddleware";
-import exceptions = require("./Exceptions");
 import { ServerVariables } from "./ServerVariables";
-import Assert = require("assert");
-import express = require("express");
+import * as Assert from "assert";
+import * as Express from "express";
 import BluebirdPromise = require("bluebird");
 import ExpressMock = require("./stubs/express.spec");
 import { IdentityValidableStub } from "./IdentityValidableStub.spec";
@@ -13,11 +12,11 @@ import { ServerVariablesMock, ServerVariablesMockBuilder }
 import { OPERATION_FAILED } from "../../../shared/UserMessages";
 
 describe("IdentityCheckMiddleware", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
-  let app: express.Application;
-  let app_get: sinon.SinonStub;
-  let app_post: sinon.SinonStub;
+  let app: Express.Application;
+  let app_get: Sinon.SinonStub;
+  let app_post: Sinon.SinonStub;
   let identityValidable: IdentityValidableStub;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -29,14 +28,6 @@ describe("IdentityCheckMiddleware", function () {
 
     req = ExpressMock.RequestMock();
     res = ExpressMock.ResponseMock();
-
-    req.headers = {};
-    req.originalUrl = "/non-api/xxx";
-    req.session = {};
-
-    req.query = {};
-    req.app = {};
-
     identityValidable = new IdentityValidableStub();
 
     mocks.notifier.notifyStub.returns(BluebirdPromise.resolve());
@@ -45,9 +36,9 @@ describe("IdentityCheckMiddleware", function () {
     mocks.userDataStore.consumeIdentityValidationTokenStub
       .returns(BluebirdPromise.resolve({ userId: "user" }));
 
-    app = express();
-    app_get = sinon.stub(app, "get");
-    app_post = sinon.stub(app, "post");
+    app = Express();
+    app_get = Sinon.stub(app, "get");
+    app_post = Sinon.stub(app, "post");
   });
 
   afterEach(function () {
@@ -56,22 +47,8 @@ describe("IdentityCheckMiddleware", function () {
   });
 
   describe("test start GET", function () {
-    it("should redirect to error 401 if pre validation initialization \
-throws a first factor error", function () {
-        identityValidable.preValidationInitStub.returns(BluebirdPromise.reject(
-          new exceptions.FirstFactorValidationError(
-            "Error during prevalidation")));
-        const callback = IdentityCheckMiddleware.post_start_validation(
-          identityValidable, vars);
-
-        return callback(req as any, res as any, undefined)
-          .then(() => {
-            Assert(res.redirect.calledWith("/error/401"));
-          });
-      });
-
     // In that case we answer with 200 to avoid user enumeration.
-    it("should send 200 if email is missing in provided identity", function () {
+    it("should send 200 if email is missing in provided identity", async function () {
       const identity = { userid: "abc" };
 
       identityValidable.preValidationInitStub
@@ -79,31 +56,26 @@ throws a first factor error", function () {
       const callback = IdentityCheckMiddleware
         .post_start_validation(identityValidable, vars);
 
-      return callback(req as any, res as any, undefined)
-        .then(function () {
-          Assert(identityValidable.preValidationResponseStub.called);
-        });
+      await callback(req as any, res as any, undefined)
+      Assert(identityValidable.preValidationResponseStub.called);
     });
 
     // In that case we answer with 200 to avoid user enumeration.
-    it("should send 200 if userid is missing in provided identity",
-      function () {
-        const identity = { email: "abc@example.com" };
+    it("should send 200 if userid is missing in provided identity", async function () {
+      const identity = { email: "abc@example.com" };
 
-        identityValidable.preValidationInitStub
-          .returns(BluebirdPromise.resolve(identity));
-        const callback = IdentityCheckMiddleware
-          .post_start_validation(identityValidable, vars);
+      identityValidable.preValidationInitStub
+        .returns(BluebirdPromise.resolve(identity));
+      const callback = IdentityCheckMiddleware
+        .post_start_validation(identityValidable, vars);
 
-        return callback(req as any, res as any, undefined)
-          .then(function () {
-            Assert(identityValidable.preValidationResponseStub.called);
-          });
-      });
+      await callback(req as any, res as any, undefined)
+      Assert(identityValidable.preValidationResponseStub.called);
+    });
 
     it("should issue a token, send an email and return 204", async function () {
       const identity = { userid: "user", email: "abc@example.com" };
-      req.get = sinon.stub().withArgs("Host").returns("localhost");
+      req.get = Sinon.stub().withArgs("Host").returns("localhost");
 
       identityValidable.preValidationInitStub
         .returns(BluebirdPromise.resolve(identity));
@@ -124,42 +96,36 @@ throws a first factor error", function () {
 
 
   describe("test finish GET", function () {
-    it("should return an error if no identity_token is provided", () => {
+    it("should return an error if no identity_token is provided", async () => {
       const callback = IdentityCheckMiddleware
         .post_finish_validation(identityValidable, vars);
 
-      return callback(req as any, res as any, undefined)
-        .then(function () {
-          Assert(res.status.calledWith(200));
-          Assert(res.send.calledWith({'error': OPERATION_FAILED}));
-        });
+      await callback(req as any, res as any, undefined)
+      Assert(res.status.calledWith(200));
+      Assert(res.send.calledWith({'error': OPERATION_FAILED}));
     });
 
-    it("should call postValidation if identity_token is provided and still \
-valid", function () {
-        req.query.identity_token = "token";
-        const callback = IdentityCheckMiddleware
-          .post_finish_validation(identityValidable, vars);
-        return callback(req as any, res as any, undefined);
-      });
+    it("should call postValidation if identity_token is provided and still valid", async function () {
+      req.query.identity_token = "token";
+      const callback = IdentityCheckMiddleware
+        .post_finish_validation(identityValidable, vars);
+      await callback(req as any, res as any, undefined);
+    });
 
-    it("should return an error if identity_token is provided but invalid",
-      function () {
-        req.query.identity_token = "token";
+    it("should return an error if identity_token is provided but invalid", async function () {
+      req.query.identity_token = "token";
 
-        identityValidable.postValidationInitStub
-          .returns(BluebirdPromise.resolve());
-        mocks.userDataStore.consumeIdentityValidationTokenStub.reset();
-        mocks.userDataStore.consumeIdentityValidationTokenStub
-          .returns(BluebirdPromise.reject(new Error("Invalid token")));
+      identityValidable.postValidationInitStub
+        .returns(BluebirdPromise.resolve());
+      mocks.userDataStore.consumeIdentityValidationTokenStub.reset();
+      mocks.userDataStore.consumeIdentityValidationTokenStub
+        .returns(() => Promise.reject(new Error("Invalid token")));
 
-        const callback = IdentityCheckMiddleware
-          .post_finish_validation(identityValidable, vars);
-        return callback(req as any, res as any, undefined)
-          .then(() => {
-            Assert(res.status.calledWith(200));
-            Assert(res.send.calledWith({'error': OPERATION_FAILED}));
-          });
-      });
+      const callback = IdentityCheckMiddleware
+        .post_finish_validation(identityValidable, vars);
+      await callback(req as any, res as any, undefined)
+      Assert(res.status.calledWith(200));
+      Assert(res.send.calledWith({'error': OPERATION_FAILED}));
+    });
   });
 });

server/src/lib/Server.ts

@@ -1,4 +1,6 @@
-import BluebirdPromise = require("bluebird");
+import * as Bluebird from "bluebird";
+import * as Express from "express";
+import * as http from "http";
 
 import { Configuration } from "./configuration/schema/Configuration";
 import { GlobalDependencies } from "../../types/Dependencies";
@@ -9,8 +11,7 @@ import { ServerVariables } from "./ServerVariables";
 import { ServerVariablesInitializer } from "./ServerVariablesInitializer";
 import { Configurator } from "./web_server/Configurator";
 
-import * as Express from "express";
-import * as http from "http";
+import { GET_VARIABLE_KEY } from "../../../shared/constants";
 
 function clone(obj: any) {
   return JSON.parse(JSON.stringify(obj));
@@ -44,11 +45,11 @@ export default class Server {
       JSON.stringify(displayableConfiguration, undefined, 2));
   }
 
-  private setup(config: Configuration, app: Express.Application, deps: GlobalDependencies): BluebirdPromise<void> {
-    const that = this;
+  private setup(config: Configuration, app: Express.Application, deps: GlobalDependencies): Bluebird<void> {
     return ServerVariablesInitializer.initialize(
       config, this.globalLogger, this.requestLogger, deps)
       .then(function (vars: ServerVariables) {
+        app.set(GET_VARIABLE_KEY, vars);
         return Configurator.configure(config, app, vars, deps);
       });
   }
@@ -56,7 +57,7 @@ export default class Server {
   private startServer(app: Express.Application, port: number) {
     const that = this;
     that.globalLogger.info("Starting Authelia...");
-    return new BluebirdPromise<void>((resolve, reject) => {
+    return new Bluebird<void>((resolve, reject) => {
       this.httpServer = app.listen(port, function (err: string) {
         that.globalLogger.info("Listening on port %d...", port);
         resolve();
@@ -65,7 +66,7 @@ export default class Server {
   }
 
   start(configuration: Configuration, deps: GlobalDependencies)
-    : BluebirdPromise<void> {
+    : Bluebird<void> {
     const that = this;
     const app = Express();
 

server/src/lib/routes/firstfactor/post.spec.ts

@@ -1,16 +1,17 @@
+import * as Express from 'express';
 import BluebirdPromise = require("bluebird");
 import Assert = require("assert");
 import FirstFactorPost = require("./post");
 import exceptions = require("../../Exceptions");
 import { AuthenticationSessionHandler } from "../../AuthenticationSessionHandler";
 import { AuthenticationSession } from "../../../../types/AuthenticationSession";
-import ExpressMock = require("../../stubs/express.spec");
+import * as ExpressMock from "../../stubs/express.spec";
 import { ServerVariablesMock, ServerVariablesMockBuilder } from "../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../ServerVariables";
 import AuthenticationError from "../../authentication/AuthenticationError";
 
 describe("routes/firstfactor/post", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let emails: string[];
   let groups: string[];
@@ -29,23 +30,11 @@ describe("routes/firstfactor/post", function () {
     mocks.regulator.regulateStub.returns(BluebirdPromise.resolve());
     mocks.regulator.markStub.returns(BluebirdPromise.resolve());
 
-    req = {
-      originalUrl: "/api/firstfactor",
-      body: {
-        username: "username",
-        password: "password"
-      },
-      query: {
-        redirect: "http://redirect.url"
-      },
-      session: {
-        cookie: {}
-      },
-      headers: {
-        host: "home.example.com"
-      }
-    };
-
+    req = ExpressMock.RequestMock();
+    req.body = {
+      username: "username",
+      password: "password"
+    }
     res = ExpressMock.ResponseMock();
     authSession = AuthenticationSessionHandler.get(req as any, vars.logger);
   });

server/src/lib/routes/firstfactor/post.ts

@@ -1,4 +1,3 @@
-import * as ObjectPath from "object-path";
 import BluebirdPromise = require("bluebird");
 import express = require("express");
 import ErrorReplies = require("../../ErrorReplies");
@@ -16,6 +15,7 @@ import { Subject } from "../../../lib/authorization/Subject";
 import AuthenticationError from "../../../lib/authentication/AuthenticationError";
 import IsRedirectionSafe from "../../../lib/utils/IsRedirectionSafe";
 import * as URLParse from "url-parse";
+import GetHeader from "../../utils/GetHeader";
 
 export default function (vars: ServerVariables) {
   return function (req: express.Request, res: express.Response)
@@ -63,7 +63,7 @@ export default function (vars: ServerVariables) {
         vars.regulator.mark(username, true);
       })
       .then(function() {
-        const targetUrl = ObjectPath.get<Express.Request, string>(req, "headers.x-target-url", undefined);
+        const targetUrl = GetHeader(req, "x-target-url");
 
         if (!targetUrl) {
           res.status(204);
@@ -85,7 +85,7 @@ export default function (vars: ServerVariables) {
 
           const authorizationLevel = vars.authorizer.authorization(resObject, subject);
           if (authorizationLevel <= AuthorizationLevel.ONE_FACTOR) {
-            if (IsRedirectionSafe(vars, authSession, new URLParse(targetUrl))) {
+            if (IsRedirectionSafe(vars, new URLParse(targetUrl))) {
               res.json({redirect: targetUrl});
               return BluebirdPromise.resolve();
             } else {

server/src/lib/routes/password-reset/form/post.spec.ts

@@ -1,9 +1,7 @@
-
+import * as Express from "express";
 import PasswordResetFormPost = require("./post");
 import { AuthenticationSessionHandler } from "../../../AuthenticationSessionHandler";
 import { AuthenticationSession } from "../../../../../types/AuthenticationSession";
-import { UserDataStore } from "../../../storage/UserDataStore";
-import Sinon = require("sinon");
 import Assert = require("assert");
 import BluebirdPromise = require("bluebird");
 import ExpressMock = require("../../../stubs/express.spec");
@@ -12,32 +10,19 @@ import { ServerVariables } from "../../../ServerVariables";
 import { Level } from "../../../authentication/Level";
 
 describe("routes/password-reset/form/post", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let vars: ServerVariables;
   let mocks: ServerVariablesMock;
   let authSession: AuthenticationSession;
 
   beforeEach(function () {
-    req = {
-      originalUrl: "/api/password-reset",
-      body: {
-        userid: "user"
-      },
-      session: {},
-      headers: {
-        host: "localhost"
-      }
-    };
+    req = ExpressMock.RequestMock();
 
     const s = ServerVariablesMockBuilder.build();
     mocks = s.mocks;
     vars = s.variables;
 
-    const options = {
-      inMemoryOnly: true
-    };
-
     mocks.userDataStore.saveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
     mocks.userDataStore.retrieveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
     mocks.userDataStore.produceIdentityValidationTokenStub.returns(BluebirdPromise.resolve({}));
@@ -65,7 +50,6 @@ describe("routes/password-reset/form/post", function () {
 
   describe("test reset password post", () => {
     it("should update the password and reset auth_session for reauthentication", function () {
-      req.body = {};
       req.body.password = "new-password";
 
       mocks.usersDatabase.updatePasswordStub.returns(BluebirdPromise.resolve());
@@ -99,7 +83,6 @@ describe("routes/password-reset/form/post", function () {
     });
 
     it("should fail when ldap fails", function () {
-      req.body = {};
       req.body.password = "new-password";
 
       mocks.usersDatabase.updatePasswordStub

server/src/lib/routes/password-reset/identity/PasswordResetHandler.spec.ts

@@ -1,4 +1,4 @@
-
+import * as Express from "express";
 import PasswordResetHandler
   from "./PasswordResetHandler";
 import BluebirdPromise = require("bluebird");
@@ -8,28 +8,20 @@ import { ServerVariablesMock, ServerVariablesMockBuilder }
 import { ServerVariables } from "../../../ServerVariables";
 
 describe("routes/password-reset/identity/PasswordResetHandler", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
 
   beforeEach(function () {
-    req = {
-      originalUrl: "/non-api/xxx",
-      body: {
-        username: "user"
-      },
-      session: {
-        auth: {
-          userid: "user",
-          email: "user@example.com",
-          first_factor: true,
-          second_factor: false
-        }
-      },
-      headers: {
-        host: "localhost"
-      }
+    req = ExpressMock.RequestMock();
+    req.body.username = "user";
+    req.session.auth = {
+      userid: "user",
+      email: "user@example.com",
+      first_factor: true,
+      second_factor: false
     };
+    req.headers.host = "localhost";
 
     const s = ServerVariablesMockBuilder.build();
     mocks = s.mocks;

server/src/lib/routes/secondfactor/redirect.spec.ts

@@ -1,41 +1,44 @@
+import * as Express from "express";
 import Redirect from "./redirect";
 import ExpressMock = require("../../stubs/express.spec");
-import { ServerVariablesMockBuilder, ServerVariablesMock }
+import { ServerVariablesMockBuilder }
 from "../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../ServerVariables";
 import Assert = require("assert");
+import { HEADER_X_TARGET_URL } from "../../../../../shared/constants";
 
 describe("routes/secondfactor/redirect", function() {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
-  let mocks: ServerVariablesMock;
   let vars: ServerVariables;
 
   beforeEach(function () {
     const s = ServerVariablesMockBuilder.build();
-    mocks = s.mocks;
     vars = s.variables;
 
     req = ExpressMock.RequestMock();
     res = ExpressMock.ResponseMock();
+    vars.config.session.domain = 'example.com';
   });
 
-  it("should redirect to default_redirection_url", function() {
-    vars.config.default_redirection_url = "http://default_redirection_url";
-    Redirect(vars)(req as any, res as any)
-    .then(function() {
-      Assert(res.json.calledWith({
-        redirect: "http://default_redirection_url"
-      }));
+  describe('redirect to default url if no target provided', () => {
+    it("should redirect to default url", async () => {
+      vars.config.default_redirection_url = "https://home.example.com";
+      await Redirect(vars)(req, res as any)
+      Assert(res.json.calledWith({redirect: "https://home.example.com"}));
     });
   });
 
-  it("should redirect to /", function() {
-    Redirect(vars)(req as any, res as any)
-    .then(function() {
-      Assert(res.json.calledWith({
-        redirect: "/"
-      }));
-    });
+  it("should redirect to safe url https://test.example.com/", async () => {
+    req.headers[HEADER_X_TARGET_URL] = "https://test.example.com/";
+    await Redirect(vars)(req, res as any);
+    Assert(res.json.calledWith({redirect: "https://test.example.com/"}));
   });
+
+  it('should not redirect to unsafe target url', async () => {
+    vars.config.default_redirection_url = "https://home.example.com";
+    req.headers[HEADER_X_TARGET_URL] = "http://test.example.com/";
+    await Redirect(vars)(req, res as any);
+    Assert(res.status.calledWith(204));
+  })
 });

server/src/lib/routes/secondfactor/redirect.ts

@@ -1,39 +1,27 @@
 
-import express = require("express");
+import * as Express from "express";
 import * as URLParse from "url-parse";
-import * as ObjectPath from "object-path";
 import { ServerVariables } from "../../ServerVariables";
-import BluebirdPromise = require("bluebird");
-import ErrorReplies = require("../../ErrorReplies");
-import UserMessages = require("../../../../../shared/UserMessages");
 import IsRedirectionSafe from "../../../lib/utils/IsRedirectionSafe";
-import { AuthenticationSessionHandler } from "../../../lib/AuthenticationSessionHandler";
+import GetHeader from "../../utils/GetHeader";
+import { HEADER_X_TARGET_URL } from "../../../../../shared/constants";
 
 
 export default function (vars: ServerVariables) {
-  return function (req: express.Request, res: express.Response)
-    : BluebirdPromise<void> {
+  return async function (req: Express.Request, res: Express.Response): Promise<void> {
+    let redirectUrl = GetHeader(req, HEADER_X_TARGET_URL);
 
-    return new BluebirdPromise<void>(function (resolve, reject) {
-      let redirectUrl: string = ObjectPath.get<Express.Request, string>(
-        req, "headers.x-target-url", undefined);
+    if (!redirectUrl && vars.config.default_redirection_url) {
+      redirectUrl = vars.config.default_redirection_url;
+    }
 
-      if (!redirectUrl && vars.config.default_redirection_url) {
-        redirectUrl = vars.config.default_redirection_url;
-      }
+    if ((redirectUrl && !IsRedirectionSafe(vars, new URLParse(redirectUrl)))
+        || !redirectUrl) {
+      res.status(204);
+      res.send();
+      return;
+    }
 
-      const authSession = AuthenticationSessionHandler.get(req, vars.logger);
-      if ((redirectUrl && !IsRedirectionSafe(vars, authSession, new URLParse(redirectUrl)))
-          || !redirectUrl) {
-        res.status(204);
-        res.send();
-        return resolve();
-      }
-
-      res.json({redirect: redirectUrl});
-      return resolve();
-    })
-      .catch(ErrorReplies.replyWithError200(req, res, vars.logger,
-        UserMessages.OPERATION_FAILED));
+    res.json({redirect: redirectUrl});
   };
 }

server/src/lib/routes/secondfactor/totp/identity/RegistrationHandler.spec.ts

@@ -1,7 +1,4 @@
-import Sinon = require("sinon");
 import RegistrationHandler from "./RegistrationHandler";
-import { Identity } from "../../../../../../types/Identity";
-import { UserDataStore } from "../../../../storage/UserDataStore";
 import BluebirdPromise = require("bluebird");
 import ExpressMock = require("../../../../stubs/express.spec");
 import { ServerVariablesMock, ServerVariablesMockBuilder }
@@ -10,7 +7,7 @@ import { ServerVariables } from "../../../../ServerVariables";
 import Assert = require("assert");
 
 describe("routes/secondfactor/totp/identity/RegistrationHandler", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -22,6 +19,7 @@ describe("routes/secondfactor/totp/identity/RegistrationHandler", function () {
 
     req = ExpressMock.RequestMock();
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         email: "user@example.com",
@@ -33,12 +31,6 @@ describe("routes/secondfactor/totp/identity/RegistrationHandler", function () {
         }
       }
     };
-    req.headers = {};
-    req.headers.host = "localhost";
-
-    const options = {
-      inMemoryOnly: true
-    };
 
     mocks.userDataStore.saveU2FRegistrationStub
       .returns(BluebirdPromise.resolve({}));

server/src/lib/routes/secondfactor/totp/identity/RegistrationHandler.ts

@@ -1,5 +1,5 @@
 
-import express = require("express");
+import * as Express from "express";
 import BluebirdPromise = require("bluebird");
 
 import { Identity } from "../../../../../../types/Identity";
@@ -35,7 +35,7 @@ export default class RegistrationHandler implements IdentityValidable {
     return Constants.CHALLENGE;
   }
 
-  private retrieveIdentity(req: express.Request): BluebirdPromise<Identity> {
+  private retrieveIdentity(req: Express.Request): BluebirdPromise<Identity> {
     const that = this;
     return new BluebirdPromise(function (resolve, reject) {
       const authSession = AuthenticationSessionHandler.get(req, that.logger);
@@ -54,7 +54,7 @@ export default class RegistrationHandler implements IdentityValidable {
     });
   }
 
-  preValidationInit(req: express.Request): BluebirdPromise<Identity> {
+  preValidationInit(req: Express.Request): BluebirdPromise<Identity> {
     const that = this;
     return FirstFactorValidator.validate(req, this.logger)
       .then(function () {
@@ -62,16 +62,16 @@ export default class RegistrationHandler implements IdentityValidable {
       });
   }
 
-  preValidationResponse(req: express.Request, res: express.Response) {
+  preValidationResponse(req: Express.Request, res: Express.Response) {
     res.status(204);
     res.send();
   }
 
-  postValidationInit(req: express.Request) {
+  postValidationInit(req: Express.Request) {
     return FirstFactorValidator.validate(req, this.logger);
   }
 
-  postValidationResponse(req: express.Request, res: express.Response)
+  postValidationResponse(req: Express.Request, res: Express.Response)
     : BluebirdPromise<void> {
     const that = this;
     let secret: TOTPSecret;

server/src/lib/routes/secondfactor/totp/sign/post.spec.ts

@@ -1,20 +1,17 @@
 
 import BluebirdPromise = require("bluebird");
-import Sinon = require("sinon");
 import Assert = require("assert");
-import Exceptions = require("../../../../Exceptions");
+import * as Express from "express";
 import { AuthenticationSessionHandler } from "../../../../AuthenticationSessionHandler";
 import { AuthenticationSession } from "../../../../../../types/AuthenticationSession";
 import SignPost = require("./post");
 import { ServerVariables } from "../../../../ServerVariables";
-
 import ExpressMock = require("../../../../stubs/express.spec");
-import { UserDataStoreStub } from "../../../../storage/UserDataStoreStub.spec";
 import { ServerVariablesMock, ServerVariablesMockBuilder } from "../../../../ServerVariablesMockBuilder.spec";
 import { Level } from "../../../../authentication/Level";
 
 describe("routes/secondfactor/totp/sign/post", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let authSession: AuthenticationSession;
   let vars: ServerVariables;
@@ -24,17 +21,9 @@ describe("routes/secondfactor/totp/sign/post", function () {
     const s = ServerVariablesMockBuilder.build();
     vars = s.variables;
     mocks = s.mocks;
-    const app_get = Sinon.stub();
-    req = {
-      originalUrl: "/api/totp-register",
-      app: {},
-      body: {
-        token: "abc"
-      },
-      session: {},
-      query: {
-        redirect: "http://redirect"
-      }
+    req = ExpressMock.RequestMock();
+    req.body = {
+      token: "abc",
     };
     res = ExpressMock.ResponseMock();
 

server/src/lib/routes/secondfactor/u2f/identity/RegistrationHandler.spec.ts

@@ -1,16 +1,13 @@
+import * as Express from "express";
 import Sinon = require("sinon");
-import Assert = require("assert");
 import BluebirdPromise = require("bluebird");
-
-import { Identity } from "../../../../../../types/Identity";
 import RegistrationHandler from "./RegistrationHandler";
 import ExpressMock = require("../../../../stubs/express.spec");
-import { UserDataStoreStub } from "../../../../storage/UserDataStoreStub.spec";
 import { ServerVariablesMock, ServerVariablesMockBuilder } from "../../../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../../../ServerVariables";
 
 describe("routes/secondfactor/u2f/identity/RegistrationHandler", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -21,8 +18,8 @@ describe("routes/secondfactor/u2f/identity/RegistrationHandler", function () {
     vars = s.variables;
 
     req = ExpressMock.RequestMock();
-    req.app = {};
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         email: "user@example.com",
@@ -33,10 +30,6 @@ describe("routes/secondfactor/u2f/identity/RegistrationHandler", function () {
     req.headers = {};
     req.headers.host = "localhost";
 
-    const options = {
-      inMemoryOnly: true
-    };
-
     mocks.userDataStore.saveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
     mocks.userDataStore.retrieveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
     mocks.userDataStore.produceIdentityValidationTokenStub.returns(BluebirdPromise.resolve({}));

server/src/lib/routes/secondfactor/u2f/register/post.spec.ts

@@ -1,4 +1,4 @@
-
+import * as Express from "express";
 import sinon = require("sinon");
 import BluebirdPromise = require("bluebird");
 import assert = require("assert");
@@ -6,13 +6,12 @@ import U2FRegisterPost = require("./post");
 import { AuthenticationSessionHandler } from "../../../../AuthenticationSessionHandler";
 import { AuthenticationSession } from "../../../../../../types/AuthenticationSession";
 import ExpressMock = require("../../../../stubs/express.spec");
-import { UserDataStoreStub } from "../../../../storage/UserDataStoreStub.spec";
 import { ServerVariablesMockBuilder, ServerVariablesMock } from "../../../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../../../ServerVariables";
 
 
 describe("routes/secondfactor/u2f/register/post", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -21,8 +20,8 @@ describe("routes/secondfactor/u2f/register/post", function () {
   beforeEach(function () {
     req = ExpressMock.RequestMock();
     req.originalUrl = "/api/xxxx";
-    req.app = {};
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         first_factor: true,
@@ -40,10 +39,6 @@ describe("routes/secondfactor/u2f/register/post", function () {
     mocks = s.mocks;
     vars = s.variables;
 
-    const options = {
-      inMemoryOnly: true
-    };
-
     mocks.userDataStore.saveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
     mocks.userDataStore.retrieveU2FRegistrationStub.returns(BluebirdPromise.resolve({}));
 

server/src/lib/routes/secondfactor/u2f/register_request/get.spec.ts

@@ -1,15 +1,14 @@
-
+import * as Express from "express";
 import sinon = require("sinon");
 import BluebirdPromise = require("bluebird");
 import Assert = require("assert");
 import U2FRegisterRequestGet = require("./get");
 import ExpressMock = require("../../../../stubs/express.spec");
-import { UserDataStoreStub } from "../../../../storage/UserDataStoreStub.spec";
 import { ServerVariablesMockBuilder, ServerVariablesMock } from "../../../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../../../ServerVariables";
 
 describe("routes/secondfactor/u2f/register_request/get", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -17,8 +16,8 @@ describe("routes/secondfactor/u2f/register_request/get", function () {
   beforeEach(function () {
     req = ExpressMock.RequestMock();
     req.originalUrl = "/api/xxxx";
-    req.app = {};
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         first_factor: true,

server/src/lib/routes/secondfactor/u2f/sign/post.spec.ts

@@ -1,4 +1,4 @@
-
+import * as Express from "express";
 import sinon = require("sinon");
 import BluebirdPromise = require("bluebird");
 import Assert = require("assert");
@@ -10,14 +10,13 @@ import ExpressMock = require("../../../../stubs/express.spec");
 import { Level } from "../../../../authentication/Level";
 
 describe("routes/secondfactor/u2f/sign/post", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
 
   beforeEach(function () {
     req = ExpressMock.RequestMock();
-    req.app = {};
     req.originalUrl = "/api/xxxx";
 
     const s = ServerVariablesMockBuilder.build();
@@ -25,6 +24,7 @@ describe("routes/secondfactor/u2f/sign/post", function () {
     vars = s.variables;
 
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         authentication_level: Level.ONE_FACTOR,

server/src/lib/routes/secondfactor/u2f/sign_request/get.spec.ts

@@ -1,4 +1,4 @@
-
+import * as Express from "express";
 import sinon = require("sinon");
 import BluebirdPromise = require("bluebird");
 import assert = require("assert");
@@ -8,10 +8,8 @@ import { Request } from "u2f";
 import { ServerVariablesMock, ServerVariablesMockBuilder } from "../../../../ServerVariablesMockBuilder.spec";
 import { ServerVariables } from "../../../../ServerVariables";
 
-import { SignMessage } from "../../../../../../../shared/SignMessage";
-
 describe("routes/secondfactor/u2f/sign_request/get", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -19,8 +17,8 @@ describe("routes/secondfactor/u2f/sign_request/get", function () {
   beforeEach(function () {
     req = ExpressMock.RequestMock();
     req.originalUrl = "/api/xxxx";
-    req.app = {};
     req.session = {
+      ...req.session,
       auth: {
         userid: "user",
         first_factor: true,

server/src/lib/routes/verify/get.spec.ts

@@ -1,9 +1,7 @@
 
-import Assert = require("assert");
-import BluebirdPromise = require("bluebird");
-import Express = require("express");
-import Sinon = require("sinon");
-import winston = require("winston");
+import * as Assert from "assert";
+import * as Express from "express";
+import * as Sinon from "sinon";
 
 import VerifyGet = require("./get");
 import { AuthenticationSessionHandler } from "../../AuthenticationSessionHandler";
@@ -13,9 +11,10 @@ import { ServerVariables } from "../../ServerVariables";
 import { ServerVariablesMockBuilder, ServerVariablesMock } from "../../ServerVariablesMockBuilder.spec";
 import { Level } from "../../authentication/Level";
 import { Level as AuthorizationLevel } from "../../authorization/Level";
+import { HEADER_X_ORIGINAL_URL } from "../../../../../shared/constants";
 
 describe("routes/verify/get", function () {
-  let req: ExpressMock.RequestMock;
+  let req: Express.Request;
   let res: ExpressMock.ResponseMock;
   let mocks: ServerVariablesMock;
   let vars: ServerVariables;
@@ -29,7 +28,7 @@ describe("routes/verify/get", function () {
       redirect: "undefined"
     };
     AuthenticationSessionHandler.reset(req as any);
-    req.headers["x-original-url"] = "https://secret.example.com/";
+    req.headers[HEADER_X_ORIGINAL_URL] = "https://secret.example.com/";
     const s = ServerVariablesMockBuilder.build(false);
     mocks = s.mocks;
     vars = s.variables;
@@ -37,40 +36,37 @@ describe("routes/verify/get", function () {
   });
 
   describe("with session cookie", function () {
-    it("should be already authenticated", function () {
+    it("should be already authenticated", async function () {
       mocks.authorizer.authorizationMock.returns(AuthorizationLevel.TWO_FACTOR);
       authSession.authentication_level = Level.TWO_FACTOR;
       authSession.userid = "myuser";
       authSession.groups = ["mygroup", "othergroup"];
-      return VerifyGet.default(vars)(req as Express.Request, res as any)
-        .then(function () {
-          Sinon.assert.calledWithExactly(res.setHeader, "Remote-User", "myuser");
-          Sinon.assert.calledWithExactly(res.setHeader, "Remote-Groups", "mygroup,othergroup");
-          Assert.equal(204, res.status.getCall(0).args[0]);
-        });
+      await VerifyGet.default(vars)(req as Express.Request, res as any);
+      res.setHeader.calledWith("Remote-User", "myuser");
+      res.setHeader.calledWith("Remote-Groups", "mygroup,othergroup");
+      Assert.equal(204, res.status.getCall(0).args[0]);
     });
 
     function test_session(_authSession: AuthenticationSession, status_code: number) {
+      const GetMock = Sinon.stub(AuthenticationSessionHandler, 'get');
+      GetMock.returns(_authSession);
       return VerifyGet.default(vars)(req as Express.Request, res as any)
         .then(function () {
           Assert.equal(status_code, res.status.getCall(0).args[0]);
-        });
+          GetMock.restore();
+        })
     }
 
-    function test_non_authenticated_401(authSession: AuthenticationSession) {
-      return test_session(authSession, 401);
+    function test_non_authenticated_401(_authSession: AuthenticationSession) {
+      return test_session(_authSession, 401);
     }
 
-    function test_unauthorized_403(authSession: AuthenticationSession) {
-      return test_session(authSession, 403);
-    }
-
-    function test_authorized(authSession: AuthenticationSession) {
-      return test_session(authSession, 204);
+    function test_unauthorized_403(_authSession: AuthenticationSession) {
+      return test_session(_authSession, 403);
     }
 
     describe("given user tries to access a 2-factor endpoint", function () {
-      before(function () {
+      beforeEach(function () {
         mocks.authorizer.authorizationMock.returns(AuthorizationLevel.TWO_FACTOR);
       });
 
@@ -115,7 +111,7 @@ describe("routes/verify/get", function () {
         it("should not be authenticated when domain is not allowed for user", function () {
           authSession.authentication_level = Level.TWO_FACTOR;
           authSession.userid = "myuser";
-          req.headers["x-original-url"] = "https://test.example.com/";
+          req.headers[HEADER_X_ORIGINAL_URL] = "https://test.example.com/";
           mocks.authorizer.authorizationMock.returns(AuthorizationLevel.DENY);
 
           return test_unauthorized_403({
@@ -132,7 +128,7 @@ describe("routes/verify/get", function () {
 
     describe("given user tries to access a single factor endpoint", function () {
       beforeEach(function () {
-        req.headers["x-original-url"] = "https://redirect.url/";
+        req.headers[HEADER_X_ORIGINAL_URL] = "https://redirect.url/";
       });
 
       it("should be authenticated when first factor is validated", function () {
@@ -227,7 +223,7 @@ describe("routes/verify/get", function () {
   });
 
   describe("with basic auth", function () {
-    it("should authenticate correctly", function () {
+    it("should authenticate correctly", async function () {
       mocks.authorizer.authorizationMock.returns(AuthorizationLevel.ONE_FACTOR);
       mocks.config.access_control.default_policy = "one_factor";
       mocks.usersDatabase.checkUserPasswordStub.returns({
@@ -235,12 +231,10 @@ describe("routes/verify/get", function () {
       });
       req.headers["proxy-authorization"] = "Basic am9objpwYXNzd29yZA==";
 
-      return VerifyGet.default(vars)(req as Express.Request, res as any)
-        .then(function () {
-          Sinon.assert.calledWithExactly(res.setHeader, "Remote-User", "john");
-          Sinon.assert.calledWithExactly(res.setHeader, "Remote-Groups", "mygroup,othergroup");
-          Assert.equal(204, res.status.getCall(0).args[0]);
-        });
+      await VerifyGet.default(vars)(req as Express.Request, res as any)
+      res.setHeader.calledWith("Remote-User", "john");
+      res.setHeader.calledWith("Remote-Groups", "mygroup,othergroup");
+      Assert.equal(204, res.status.getCall(0).args[0]);
     });
 
     it("should fail when endpoint is protected by two factors", function () {

server/src/lib/routes/verify/get.ts

@@ -6,34 +6,44 @@ import { ServerVariables } from "../../ServerVariables";
 import GetWithSessionCookieMethod from "./get_session_cookie";
 import GetWithBasicAuthMethod from "./get_basic_auth";
 import Constants = require("../../../../../shared/constants");
-import ObjectPath = require("object-path");
-
 import { AuthenticationSessionHandler }
   from "../../AuthenticationSessionHandler";
 import { AuthenticationSession }
   from "../../../../types/AuthenticationSession";
+import GetHeader from "../../utils/GetHeader";
 
 const REMOTE_USER = "Remote-User";
 const REMOTE_GROUPS = "Remote-Groups";
 
 
 function verifyWithSelectedMethod(req: Express.Request, res: Express.Response,
-  vars: ServerVariables, authSession: AuthenticationSession)
+  vars: ServerVariables, authSession: AuthenticationSession | undefined)
   : () => BluebirdPromise<{ username: string, groups: string[] }> {
   return function () {
-    const authorization: string = "" + req.headers["proxy-authorization"];
-    if (authorization && authorization.startsWith("Basic "))
-      return GetWithBasicAuthMethod(req, res, vars, authorization);
-
-    return GetWithSessionCookieMethod(req, res, vars, authSession);
+    const authorization = GetHeader(req, Constants.HEADER_PROXY_AUTHORIZATION);
+    if (authorization) {
+      if (authorization.startsWith("Basic ")) {
+        return GetWithBasicAuthMethod(req, res, vars, authorization);
+      }
+      else {
+        throw new Error("The authorization header should be of the form 'Basic XXXXXX'");
+      }
+    }
+    else {
+      if (authSession) {
+        return GetWithSessionCookieMethod(req, res, vars, authSession);
+      }
+      else {
+        throw new Error("No cookie detected.");
+      }
+    }
   };
 }
 
 function setRedirectHeader(req: Express.Request, res: Express.Response) {
   return function () {
-    const originalUrl = ObjectPath.get<Express.Request, string>(
-      req, "headers.x-original-url");
-    res.set("Redirect", originalUrl);
+    const originalUrl = GetHeader(req, Constants.HEADER_X_ORIGINAL_URL);
+    res.set(Constants.HEADER_REDIRECT, originalUrl);
     return BluebirdPromise.resolve();
   };
 }
@@ -62,7 +72,7 @@ function getRedirectParam(req: Express.Request) {
 export default function (vars: ServerVariables) {
   return function (req: Express.Request, res: Express.Response)
     : BluebirdPromise<void> {
-    let authSession: AuthenticationSession;
+    let authSession: AuthenticationSession | undefined;
     return new BluebirdPromise(function (resolve, reject) {
       authSession = AuthenticationSessionHandler.get(req, vars.logger);
       resolve();
@@ -78,6 +88,7 @@ export default function (vars: ServerVariables) {
         ErrorReplies.replyWithError401(req, res, vars.logger))
       // The user is not yet authenticated -> 401
       .catch((err) => {
+        console.error(err);
         // This redirect parameter is used in Kubernetes to annotate the ingress with
         // the url to the authentication portal.
         const redirectUrl = getRedirectParam(req);

server/src/lib/routes/verify/get_basic_auth.ts

@@ -1,18 +1,17 @@
 import Express = require("express");
 import BluebirdPromise = require("bluebird");
-import ObjectPath = require("object-path");
 import { ServerVariables } from "../../ServerVariables";
-import { AuthenticationSession }
-  from "../../../../types/AuthenticationSession";
 import AccessControl from "./access_control";
 import { URLDecomposer } from "../../utils/URLDecomposer";
 import { Level } from "../../authentication/Level";
+import GetHeader from "../../utils/GetHeader";
+import { HEADER_X_ORIGINAL_URL } from "../../../../../shared/constants";
 
 export default function (req: Express.Request, res: Express.Response,
   vars: ServerVariables, authorizationHeader: string)
   : BluebirdPromise<{ username: string, groups: string[] }> {
   let username: string;
-  const uri = ObjectPath.get<Express.Request, string>(req, "headers.x-original-url");
+  const uri = GetHeader(req, HEADER_X_ORIGINAL_URL);
   const urlDecomposition = URLDecomposer.fromUrl(uri);
 
   return BluebirdPromise.resolve()

server/src/lib/routes/verify/get_session_cookie.ts

@@ -1,8 +1,5 @@
 import Express = require("express");
 import BluebirdPromise = require("bluebird");
-import Util = require("util");
-import ObjectPath = require("object-path");
-
 import Exceptions = require("../../Exceptions");
 import { Configuration } from "../../configuration/schema/Configuration";
 import { ServerVariables } from "../../ServerVariables";
@@ -13,6 +10,8 @@ import { AuthenticationSessionHandler }
   from "../../AuthenticationSessionHandler";
 import AccessControl from "./access_control";
 import { URLDecomposer } from "../../utils/URLDecomposer";
+import GetHeader from "../../utils/GetHeader";
+import { HEADER_X_ORIGINAL_URL } from "../../../../../shared/constants";
 
 function verify_inactivity(req: Express.Request,
   authSession: AuthenticationSession,
@@ -54,8 +53,7 @@ export default function (req: Express.Request, res: Express.Response,
         "userid is missing"));
     }
 
-    const originalUrl = ObjectPath.get<Express.Request, string>(
-      req, "headers.x-original-url");
+    const originalUrl = GetHeader(req, HEADER_X_ORIGINAL_URL);
 
     const d = URLDecomposer.fromUrl(originalUrl);
     vars.logger.debug(req, "domain=%s, path=%s, user=%s, groups=%s", d.domain,

server/src/lib/stubs/express.spec.ts

@@ -1,103 +1,117 @@
 
-import sinon = require("sinon");
-import express = require("express");
-
-export interface RequestMock {
-    app?: any;
-    body?: any;
-    session?: any;
-    headers?: any;
-    get?: any;
-    query?: any;
-    originalUrl: string;
-}
+import * as Sinon from "sinon";
+import * as Express from "express";
+import { GET_VARIABLE_KEY } from "../../../../shared/constants";
+import { RequestLoggerStub } from "../logging/RequestLoggerStub.spec";
 
 export interface ResponseMock {
-    send: sinon.SinonStub | sinon.SinonSpy;
-    sendStatus: sinon.SinonStub;
-    sendFile: sinon.SinonStub;
-    sendfile: sinon.SinonStub;
-    status: sinon.SinonStub | sinon.SinonSpy;
-    json: sinon.SinonStub | sinon.SinonSpy;
-    links: sinon.SinonStub;
-    jsonp: sinon.SinonStub;
-    download: sinon.SinonStub;
-    contentType: sinon.SinonStub;
-    type: sinon.SinonStub;
-    format: sinon.SinonStub;
-    attachment: sinon.SinonStub;
-    set: sinon.SinonStub;
-    header: sinon.SinonStub;
+    send: Sinon.SinonStub | Sinon.SinonSpy;
+    sendStatus: Sinon.SinonStub;
+    sendFile: Sinon.SinonStub;
+    sendfile: Sinon.SinonStub;
+    status: Sinon.SinonStub | Sinon.SinonSpy;
+    json: Sinon.SinonStub | Sinon.SinonSpy;
+    links: Sinon.SinonStub;
+    jsonp: Sinon.SinonStub;
+    download: Sinon.SinonStub;
+    contentType: Sinon.SinonStub;
+    type: Sinon.SinonStub;
+    format: Sinon.SinonStub;
+    attachment: Sinon.SinonStub;
+    set: Sinon.SinonStub;
+    header: Sinon.SinonStub;
     headersSent: boolean;
-    get: sinon.SinonStub;
-    clearCookie: sinon.SinonStub;
-    cookie: sinon.SinonStub;
-    location: sinon.SinonStub;
-    redirect: sinon.SinonStub | sinon.SinonSpy;
-    render: sinon.SinonStub | sinon.SinonSpy;
-    locals: sinon.SinonStub;
+    get: Sinon.SinonStub;
+    clearCookie: Sinon.SinonStub;
+    cookie: Sinon.SinonStub;
+    location: Sinon.SinonStub;
+    redirect: Sinon.SinonStub | Sinon.SinonSpy;
+    render: Sinon.SinonStub | Sinon.SinonSpy;
+    locals: Sinon.SinonStub;
     charset: string;
-    vary: sinon.SinonStub;
+    vary: Sinon.SinonStub;
     app: any;
-    write: sinon.SinonStub;
-    writeContinue: sinon.SinonStub;
-    writeHead: sinon.SinonStub;
+    write: Sinon.SinonStub;
+    writeContinue: Sinon.SinonStub;
+    writeHead: Sinon.SinonStub;
     statusCode: number;
     statusMessage: string;
-    setHeader: sinon.SinonStub;
-    setTimeout: sinon.SinonStub;
+    setHeader: Sinon.SinonStub;
+    setTimeout: Sinon.SinonStub;
     sendDate: boolean;
-    getHeader: sinon.SinonStub;
+    getHeader: Sinon.SinonStub;
 }
 
-export function RequestMock(): RequestMock {
+export function RequestMock(): Express.Request {
+    const getMock = Sinon.mock()
+      .withArgs(GET_VARIABLE_KEY).atLeast(0).returns({
+          logger: new RequestLoggerStub()
+      });
     return {
-        originalUrl: "/non-api/xxx",
+        id: '1234',
+        headers: {},
         app: {
-            get: sinon.stub()
+            get: getMock,
+            set: Sinon.mock(),
         },
-        headers: {
-            "x-forwarded-for": "127.0.0.1"
-        },
-        session: {}
-    };
+        body: {},
+        query: {},
+        session: {
+            id: '1234',
+            regenerate: function() {},
+            reload: function() {},
+            destroy: function() {},
+            save: function() {},
+            touch: function() {},
+            cookie: {
+                domain: 'example.com',
+                expires: true,
+                httpOnly: true,
+                maxAge: 36000,
+                originalMaxAge: 36000,
+                path: '/',
+                secure: true,
+                serialize: () => '',
+            }
+        }
+    } as any;
 }
 export function ResponseMock(): ResponseMock {
     return {
-        send: sinon.stub(),
-        status: sinon.stub(),
-        json: sinon.stub(),
-        sendStatus: sinon.stub(),
-        links: sinon.stub(),
-        jsonp: sinon.stub(),
-        sendFile: sinon.stub(),
-        sendfile: sinon.stub(),
-        download: sinon.stub(),
-        contentType: sinon.stub(),
-        type: sinon.stub(),
-        format: sinon.stub(),
-        attachment: sinon.stub(),
-        set: sinon.stub(),
-        header: sinon.stub(),
+        send: Sinon.stub(),
+        status: Sinon.stub(),
+        json: Sinon.stub(),
+        sendStatus: Sinon.stub(),
+        links: Sinon.stub(),
+        jsonp: Sinon.stub(),
+        sendFile: Sinon.stub(),
+        sendfile: Sinon.stub(),
+        download: Sinon.stub(),
+        contentType: Sinon.stub(),
+        type: Sinon.stub(),
+        format: Sinon.stub(),
+        attachment: Sinon.stub(),
+        set: Sinon.stub(),
+        header: Sinon.stub(),
         headersSent: true,
-        get: sinon.stub(),
-        clearCookie: sinon.stub(),
-        cookie: sinon.stub(),
-        location: sinon.stub(),
-        redirect: sinon.stub(),
-        render: sinon.stub(),
-        locals: sinon.stub(),
+        get: Sinon.stub(),
+        clearCookie: Sinon.stub(),
+        cookie: Sinon.stub(),
+        location: Sinon.stub(),
+        redirect: Sinon.stub(),
+        render: Sinon.stub(),
+        locals: Sinon.stub(),
         charset: "utf-8",
-        vary: sinon.stub(),
-        app: sinon.stub(),
-        write: sinon.stub(),
-        writeContinue: sinon.stub(),
-        writeHead: sinon.stub(),
+        vary: Sinon.stub(),
+        app: Sinon.stub(),
+        write: Sinon.stub(),
+        writeContinue: Sinon.stub(),
+        writeHead: Sinon.stub(),
         statusCode: 200,
         statusMessage: "message",
-        setHeader: sinon.stub(),
-        setTimeout: sinon.stub(),
+        setHeader: Sinon.stub(),
+        setTimeout: Sinon.stub(),
         sendDate: true,
-        getHeader: sinon.stub()
+        getHeader: Sinon.stub()
     };
 }

server/src/lib/utils/GetHeader.spec.ts

@@ -0,0 +1,20 @@
+import * as Express from "express";
+import GetHeader from "./GetHeader";
+import { RequestMock } from "../stubs/express.spec";
+import * as Assert from "assert";
+
+describe('GetHeader', function() {
+  let req: Express.Request;
+  beforeEach(() => {
+    req = RequestMock();
+  });
+
+  it('should return the header if it exists', function() {
+    req.headers["x-target-url"] = 'www.example.com';
+    Assert.equal(GetHeader(req, 'x-target-url'), 'www.example.com');
+  });
+
+  it('should return undefined if header does not exist', function() {
+    Assert.equal(GetHeader(req, 'x-target-url'), undefined);
+  });
+});

server/src/lib/utils/GetHeader.ts

@@ -0,0 +1,19 @@
+import * as Express from "express";
+import * as ObjectPath from "object-path";
+import { ServerVariables } from "../ServerVariables";
+import { GET_VARIABLE_KEY } from "../../../../shared/constants";
+
+/**
+ *
+ * @param req The express request to extract headers from
+ * @param header The name of the header to extract in lowercase.
+ * @returns The header if found, otherwise undefined.
+ */
+export default function(req: Express.Request, header: string): string | undefined {
+  const variables: ServerVariables = req.app.get(GET_VARIABLE_KEY);
+  if (!variables) return undefined;
+
+  const value = ObjectPath.get<Express.Request, string>(req, "headers." + header, undefined);
+  variables.logger.debug(req, "Header %s is set to %s", header, value);
+  return value;
+}

server/src/lib/utils/IsRedirectionSafe.ts

@@ -1,11 +1,8 @@
 import { ServerVariables } from "../ServerVariables";
-import { Level } from "../authentication/Level";
 import * as URLParse from "url-parse";
-import { AuthenticationSession } from "AuthenticationSession";
 
 export default function IsRedirectionSafe(
   vars: ServerVariables,
-  authSession: AuthenticationSession,
   url: URLParse): boolean {
 
   const urlInDomain = url.hostname.endsWith(vars.config.session.domain);

shared/DomainExtractor.spec.ts

@@ -1,7 +1,7 @@
 import { DomainExtractor } from "./DomainExtractor";
 import Assert = require("assert");
 
-describe.only("shared/DomainExtractor", function () {
+describe("shared/DomainExtractor", function () {
   describe("test fromUrl", function () {
     it("should return domain from https url", function () {
       const domain = DomainExtractor.fromUrl("https://www.example.com/test/abc");

shared/constants.ts

@@ -1 +1,11 @@
-export const REDIRECT_QUERY_PARAM = "rd";
+export const REDIRECT_QUERY_PARAM = "rd";
+
+// Used as a first factor script parameter for knowing the authorizations
+// related to the domain and resource.
+export const HEADER_X_TARGET_URL = "x-target-url";
+
+export const HEADER_X_ORIGINAL_URL = "x-original-url";
+export const HEADER_PROXY_AUTHORIZATION = "proxy-authorization";
+export const HEADER_REDIRECT = "redirect";
+
+export const GET_VARIABLE_KEY = "variables";

spec-helper.js

@@ -0,0 +1,3 @@
+var colors = require('mocha/lib/reporters/base').colors;
+    colors['pass'] = 32;
+    colors['error stack'] = 34;