<?php
/*
 * sysPass
 *
 * @author nuxsmin
 * @link https://syspass.org
 * @copyright 2012-2021, Rubén Domínguez nuxsmin@$syspass.org
 *
 * This file is part of sysPass.
 *
 * sysPass is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * sysPass is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with sysPass.  If not, see <http://www.gnu.org/licenses/>.
 */

namespace SP\Core\Crypt;


use SP\Core\Context\ContextInterface;
use SP\Domain\Config\In\ConfigDataInterface;
use SP\Http\Request;
use SP\Http\RequestInterface;

/**
 * Class CSRF
 *
 * @package SP\Core\Crypt
 */
class CSRF
{
    private ContextInterface $context;
    private Request $request;
    private ConfigDataInterface $configData;

    /**
     * CSRF constructor.
     *
     * @param  ContextInterface  $context
     * @param  RequestInterface  $request
     * @param  \SP\Domain\Config\In\ConfigDataInterface  $configData
     */
    public function __construct(
        ContextInterface    $context,
        RequestInterface $request,
        ConfigDataInterface $configData
    )
    {
        $this->context = $context;
        $this->request = $request;
        $this->configData = $configData;
    }

    /**
     * Check for CSRF token on POST requests
     */
    public function check(): bool
    {
        $method = strtoupper($this->request->getMethod());
        $with = $this->request->getHeader('X-Requested-With');

        if ($this->context->isLoggedIn()
            && $this->context->getCSRF() !== null
            && ($method === 'POST'
                || ($method === 'GET' && $with === 'XMLHttpRequest'))
        ) {
            $token = $this->request->getHeader('X-CSRF');

            if (empty($token)
                || !Hash::checkMessage($this->getKey(), $this->configData->getPasswordSalt(), $token)
            ) {
                logger(sprintf('Invalid CSRF token: %s', $token), 'ERROR');

                return false;
            }

            logger('CSRF token OK');
        }

        return true;
    }

    /**
     * Devolver la llave de cifrado para los datos de la cookie
     */
    private function getKey(): string
    {
        return sha1($this->request->getHeader('User-Agent') . $this->request->getClientAddress());
    }

    /**
     * Initialize the CSRF key
     */
    public function initialize(): void
    {
        if ($this->context->isLoggedIn()
            && $this->context->getCSRF() === null
        ) {
            $key = Hash::signMessage($this->getKey(), $this->configData->getPasswordSalt());

            $this->context->setCSRF($key);

            logger(sprintf('CSRF key (set): %s', $this->context->getCSRF()));
        }
    }
}