|
|
|
|
@ -13,6 +13,7 @@ use OC\Core\Controller\TwoFactorChallengeController;
|
|
|
|
|
use OCP\AppFramework\Controller;
|
|
|
|
|
use OCP\AppFramework\Http\Attribute\PublicPage;
|
|
|
|
|
use OCP\AppFramework\Middleware;
|
|
|
|
|
use OCP\AppFramework\Utility\ITimeFactory;
|
|
|
|
|
use OCP\Authentication\TwoFactorAuth\ALoginSetupController;
|
|
|
|
|
use OCP\ISession;
|
|
|
|
|
use OCP\IUserSession;
|
|
|
|
|
@ -22,19 +23,32 @@ use ReflectionMethod;
|
|
|
|
|
// Will close the session if the user session is ephemeral.
|
|
|
|
|
// Happens when the user logs in via the login flow v2.
|
|
|
|
|
class FlowV2EphemeralSessionsMiddleware extends Middleware {
|
|
|
|
|
|
|
|
|
|
private const EPHEMERAL_SESSION_TTL = 5 * 60; // 5 minutes
|
|
|
|
|
|
|
|
|
|
public function __construct(
|
|
|
|
|
private ISession $session,
|
|
|
|
|
private IUserSession $userSession,
|
|
|
|
|
private ControllerMethodReflector $reflector,
|
|
|
|
|
private LoggerInterface $logger,
|
|
|
|
|
private ITimeFactory $timeFactory,
|
|
|
|
|
) {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public function beforeController(Controller $controller, string $methodName) {
|
|
|
|
|
if (!$this->session->get(ClientFlowLoginV2Controller::EPHEMERAL_NAME)) {
|
|
|
|
|
$sessionCreationTime = $this->session->get(ClientFlowLoginV2Controller::EPHEMERAL_NAME);
|
|
|
|
|
|
|
|
|
|
// Not an ephemeral session.
|
|
|
|
|
if ($sessionCreationTime === null) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Lax enforcement until TTL is reached.
|
|
|
|
|
if ($this->timeFactory->getTime() < $sessionCreationTime + self::EPHEMERAL_SESSION_TTL) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Allow certain controllers/methods to proceed without logging out.
|
|
|
|
|
if (
|
|
|
|
|
$controller instanceof ClientFlowLoginV2Controller
|
|
|
|
|
&& ($methodName === 'grantPage' || $methodName === 'generateAppPassword')
|
|
|
|
|
|