misc_pterodactyl-panel/app/Http/Middleware/RequireTwoFactorAuthentication.php

61 lines
2.2 KiB
PHP
Raw Normal View History

<?php
namespace Pterodactyl\Http\Middleware;
use Closure;
2018-07-05 02:38:23 +00:00
use Illuminate\Support\Str;
2017-10-29 17:37:25 +00:00
use Illuminate\Http\Request;
use Pterodactyl\Exceptions\Http\TwoFactorAuthRequiredException;
class RequireTwoFactorAuthentication
{
2021-01-23 20:33:34 +00:00
public const LEVEL_NONE = 0;
public const LEVEL_ADMIN = 1;
public const LEVEL_ALL = 2;
2017-10-29 17:37:25 +00:00
/**
* The route to redirect a user to to enable 2FA.
*/
protected string $redirectRoute = '/account';
/**
* Check the user state on the incoming request to determine if they should be allowed to
* proceed or not. This checks if the Panel is configured to require 2FA on an account in
* order to perform actions. If so, we check the level at which it is required (all users
* or just admins) and then check if the user has enabled it for their account.
*
* @return mixed
*
* @throws \Pterodactyl\Exceptions\Http\TwoFactorAuthRequiredException
*/
2017-10-29 17:37:25 +00:00
public function handle(Request $request, Closure $next)
{
/** @var \Pterodactyl\Models\User|null $user */
$user = $request->user();
$uri = rtrim($request->getRequestUri(), '/') . '/';
$current = $request->route()->getName();
2021-01-23 20:33:34 +00:00
if (!$user || Str::startsWith($uri, ['/auth/']) || Str::startsWith($current, ['auth.', 'account.'])) {
return $next($request);
}
$level = (int) config('pterodactyl.auth.2fa_required');
// If this setting is not configured, or the user is already using 2FA then we can just
// send them right through, nothing else needs to be checked.
//
// If the level is set as admin and the user is not an admin, pass them through as well.
if ($level === self::LEVEL_NONE || ($user->use_totp || $user->securityKeys->isNotEmpty())) {
return $next($request);
2021-01-23 20:33:34 +00:00
} elseif ($level === self::LEVEL_ADMIN && !$user->root_admin) {
return $next($request);
}
// For API calls return an exception which gets rendered nicely in the API response.
if ($request->isJson() || Str::startsWith($uri, '/api/')) {
2021-01-23 20:33:34 +00:00
throw new TwoFactorAuthRequiredException();
}
return redirect()->to($this->redirectRoute);
}
}