App\Security\EntityListeners\ElementPermissionListener

Complexity

Total Complexity

19

Size/Duplication

Total Lines	122
Duplicated Lines	0 %

Importance

Changes	5
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
B	preFlushHandler()	0	41	8
A	postLoadHandler()	0	31	6
A	isRunningFromCLI()	0	8	4
A	__construct()	0	7	1

<?php
/**
 * part-db version 0.1
 * Copyright (C) 2005 Christoph Lechner
 * http://www.cl-projects.de/.
 *
 * part-db version 0.2+
 * Copyright (C) 2009 K. Jacobs and others (see authors.php)
 * http://code.google.com/p/part-db/
 *
 * Part-DB Version 0.4+
 * Copyright (C) 2016 - 2019 Jan Böhmer
 * https://github.com/jbtronics
 *
 * This program 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 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

namespace App\Security\EntityListeners;

use App\Entity\Base\DBElement;
use App\Security\Annotations\ColumnSecurity;
use Doctrine\Common\Annotations\Reader;
use Doctrine\Common\Collections\Collection;
use Doctrine\Common\Persistence\Event\LifecycleEventArgs;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Event\PreFlushEventArgs;
use Doctrine\ORM\Event\PreUpdateEventArgs;
use Doctrine\ORM\Mapping as ORM;
use Doctrine\ORM\Mapping\PostLoad;
use Doctrine\ORM\Mapping\PreUpdate;
use ReflectionClass;
use Symfony\Component\HttpKernel\KernelInterface;
use Symfony\Component\Security\Core\Security;

/**
 * The purpose of this class is to hook into the doctrine entity lifecycle and restrict access to entity informations
 * configured by ColoumnSecurity Annotation.
 * If the current programm is running from CLI (like a CLI command), the security checks are disabled.
 * (Commands should be able to do everything they like)
 *
 * If a user does not have access to an coloumn, it will be filled, with a placeholder, after doctrine loading is finished.
 * The edit process is also catched, so that these placeholders, does not get saved to database.
 */
class ElementPermissionListener
{
    protected $security;
    protected $reader;
    protected $em;
    protected $disabled;

    public function __construct(Security $security, Reader $reader, EntityManagerInterface $em, KernelInterface $kernel)

The parameter $kernel is not used and could be removed.

    {
        $this->security = $security;
        $this->reader = $reader;
        $this->em = $em;
        //Disable security when the current program is running from CLI
        $this->disabled = $this->isRunningFromCLI();
    }

    /**
     * This function checks if the current script is run from web or from a terminal.
     * @return bool Returns true if the current programm is running from CLI (terminal)
     */
    protected function isRunningFromCLI()
    {

        if (empty($_SERVER['REMOTE_ADDR']) && !isset($_SERVER['HTTP_USER_AGENT']) && count($_SERVER['argv']) > 0) {
            return true;
        }

        return false;
    }

    /**
     * @PostLoad
     * @ORM\PostUpdate()
     * This function is called after doctrine filled, the entity properties with db values.
     * We use this, to check if the user is allowed to access these properties, and if not, we write a placeholder
     * into the element properties, so that a user only gets non sensitve data.
     *
     * This function is also called after an entity was updated, so we dont show the original data to user,
     * after an update.
     */
    public function postLoadHandler(DBElement $element, LifecycleEventArgs $event)

The parameter $event is not used and could be removed.

    {
        //Do nothing if security is disabled
        if ($this->disabled) {
            return;
        }

        //Read Annotations and properties.
        $reflectionClass = new ReflectionClass($element);
        $properties = $reflectionClass->getProperties();

        foreach ($properties as $property) {
            /**
             * @var ColumnSecurity
             */
            $annotation = $this->reader->getPropertyAnnotation(
                $property,
                ColumnSecurity::class
            );

            //Check if user is allowed to read info, otherwise apply placeholder
            if ((null !== $annotation) && !$this->security->isGranted($annotation->getReadOperationName(), $element)) {
                $property->setAccessible(true);
                $value = $annotation->getPlaceholder();

                //Detach placeholder entities, so we dont get cascade errors
                if ($value instanceof DBElement) {
                    $this->em->detach($value);
                }

                $property->setValue($element, $value);
            }
        }
    }

    /**
     * @ORM\PreFlush()
     * This function is called before flushing. We use it, to remove all placeholders.
     * We do it here and not in preupdate, because this is called before calculating the changeset,
     * and so we dont get problems with orphan removal.
     */
    public function preFlushHandler(DBElement $element, PreFlushEventArgs $eventArgs)
    {
        //Do nothing if security is disabled
        if ($this->disabled) {
            return;
        }

        $em = $eventArgs->getEntityManager();

The assignment to $em is dead and can be removed.

        $unitOfWork = $eventArgs->getEntityManager()->getUnitOfWork();

        $reflectionClass = new ReflectionClass($element);
        $properties = $reflectionClass->getProperties();

        $old_data = $unitOfWork->getOriginalEntityData($element);

        foreach ($properties as $property) {
            $annotation = $this->reader->getPropertyAnnotation(
                $property,
                ColumnSecurity::class
            );

            $changed = false;

            //Only set the field if it has an annotation
            if (null !== $annotation) {
                $property->setAccessible(true);

                //If the user is not allowed to edit or read this property, reset all values.
                if ((!$this->security->isGranted($annotation->getEditOperationName(), $element)
                    || !$this->security->isGranted($annotation->getReadOperationName(), $element))) {
                    //Set value to old value, so that there a no change to this property
                    if (isset($old_data[$property->getName()])) {
                        $property->setValue($element, $old_data[$property->getName()]);
                        $changed = true;
                    }

                }

                if ($changed) {
                    //Schedule for update, so the post update method will be called
                    $unitOfWork->scheduleForUpdate($element);
                }
            }
        }
    }
}