100 lines
2.5 KiB
PHP
100 lines
2.5 KiB
PHP
<?php
|
|
/**
|
|
* Pterodactyl - Panel
|
|
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
|
|
*
|
|
* This software is licensed under the terms of the MIT license.
|
|
* https://opensource.org/licenses/MIT
|
|
*/
|
|
|
|
namespace Pterodactyl\Http\Middleware;
|
|
|
|
use Closure;
|
|
use Krucas\Settings\Settings;
|
|
use Prologue\Alerts\AlertsMessageBag;
|
|
|
|
class RequireTwoFactorAuthentication
|
|
{
|
|
const LEVEL_NONE = 0;
|
|
const LEVEL_ADMIN = 1;
|
|
const LEVEL_ALL = 2;
|
|
|
|
/**
|
|
* @var \Prologue\Alerts\AlertsMessageBag
|
|
*/
|
|
protected $alert;
|
|
|
|
/**
|
|
* @var \Krucas\Settings\Settings
|
|
*/
|
|
protected $settings;
|
|
|
|
/**
|
|
* All TOTP related routes.
|
|
*
|
|
* @var array
|
|
*/
|
|
protected $ignoreRoutes = [
|
|
'account.security',
|
|
'account.security.revoke',
|
|
'account.security.totp',
|
|
'account.security.totp.set',
|
|
'account.security.totp.disable',
|
|
'auth.totp',
|
|
'auth.logout',
|
|
];
|
|
|
|
/**
|
|
* RequireTwoFactorAuthentication constructor.
|
|
*
|
|
* @param \Prologue\Alerts\AlertsMessageBag $alert
|
|
* @param \Krucas\Settings\Settings $settings
|
|
*/
|
|
public function __construct(AlertsMessageBag $alert, Settings $settings)
|
|
{
|
|
$this->alert = $alert;
|
|
$this->settings = $settings;
|
|
}
|
|
|
|
/**
|
|
* Handle an incoming request.
|
|
*
|
|
* @param \Illuminate\Http\Request $request
|
|
* @param \Closure $next
|
|
* @return mixed
|
|
*/
|
|
public function handle($request, Closure $next)
|
|
{
|
|
// Ignore non-users
|
|
if (! $request->user()) {
|
|
return $next($request);
|
|
}
|
|
|
|
// Skip the 2FA pages
|
|
if (in_array($request->route()->getName(), $this->ignoreRoutes)) {
|
|
return $next($request);
|
|
}
|
|
|
|
// Get the setting
|
|
switch ((int) $this->settings->get('2fa', 0)) {
|
|
case self::LEVEL_NONE:
|
|
return $next($request);
|
|
|
|
case self::LEVEL_ADMIN:
|
|
if (! $request->user()->root_admin) {
|
|
return $next($request);
|
|
}
|
|
break;
|
|
|
|
case self::LEVEL_ALL:
|
|
if ($request->user()->use_totp) {
|
|
return $next($request);
|
|
}
|
|
break;
|
|
}
|
|
|
|
$this->alert->danger('The administrator has required 2FA to be enabled. You must enable it before you can do any other action.')->flash();
|
|
|
|
return redirect()->route('account.security');
|
|
}
|
|
}
|