vendor/sindrive/sindrive/IpsetBundle/Listener/BanListener.php line 45

Open in your IDE?
  1. <?php
  2. /**
  3.  * Created by PhpStorm.
  4.  * User: bumz
  5.  * Date: 7/8/14
  6.  * Time: 2:49 PM
  7.  */
  9. namespace Sindrive\IpsetBundle\Listener;
  12. use Sindrive\IpsetBundle\Event\BanEvent;
  13. use Sindrive\IpsetBundle\Service\Ban;
  14. use Sindrive\MemcachedBundle\Services\SindriveMemcached;
  16. class BanListener
  17. {
  18.     /**
  19.      * @var string
  20.      */
  21.     private $banKey;
  23.     /**
  24.      * @var \Sindrive\IpsetBundle\Service\Ban
  25.      */
  26.     private $banService;
  28.     /**
  29.      * @var \Memcached|SindriveMemcached
  30.      */
  31.     private $memcached;
  33.     /**
  34.      * @param SindriveMemcached $memcached
  35.      * @param \Sindrive\IpsetBundle\Service\Ban $banService
  36.      * @param string $banKey
  37.      */
  38.     public function __construct(SindriveMemcached $memcached, Ban $banService, string $banKey = 'sindrive-ips-ban-listener')
  39.     {
  40.         $this->memcached = $memcached;
  41.         $this->banService = $banService;
  42.         $this->banKey = $banKey;
  43.     }
  45.     public function banByCount(BanEvent $event): void
  46.     {
  47.         $banKey = sprintf('%s-%s-%s', $this->banKey, 'count', $event->getIp());
  49.         $result = $this->memcached->increment($banKey);
  51.         if (false === $result && $this->memcached->getResultCode() == \Memcached::RES_NOTFOUND) {
  52.             $this->memcached->add($banKey, 1, time() + 30 * 24 * 60 * 60);
  53.             $result = 1;
  54.         }
  56.         if ($result > $event->getCount()) {
  57.             $this->banService->add($event->getIp());
  58.         }
  60.     }
  62.     public function banByTimeSpanCount(BanEvent $event): bool
  63.     {
  64.         $banKey = sprintf('%s-%s-%s', $this->banKey, 'count-time', $event->getIp());
  66.         do {
  68.             $resultStore = $this->memcached->getCas($banKey);
  69.             $result = $resultStore['value'];
  70.             $cas = $resultStore['cas'];
  72.             $currentTime = time();
  73.             $time = $currentTime + $event->getTimeOffset();
  75.             if ($this->memcached->getResultCode() == \Memcached::RES_NOTFOUND) {
  77.                 $result = array($event->getTimeOffset() => array($time));
  78.                 $this->memcached->add($banKey, $result, $time);
  80.             } elseif ($this->memcached->getResultCode() == \Memcached::RES_SUCCESS) {
  82.                 if (isset($result[$event->getTimeOffset()])) {
  83.                     $result[$event->getTimeOffset()] = array_filter($result[$event->getTimeOffset()], function($item) use ($currentTime) {
  84.                         return $item > $currentTime;
  85.                     });
  86.                     // need a clean indexed array
  87.                     $result[$event->getTimeOffset()] = array_values($result[$event->getTimeOffset()]);
  88.                 }
  90.                 $result[$event->getTimeOffset()][] = $time;
  91.                 $this->memcached->cas($cas, $banKey, $result, $time);
  93.             } else {
  94.                 return false;
  95.             }
  99.         } while ($this->memcached->getResultCode() != \Memcached::RES_SUCCESS);
  101.         if (count($result[$event->getTimeOffset()]) > $event->getCount()) {
  102.             $this->banService->add($event->getIp());
  103.         }
  105.         return true;
  106.     }
  108.     public function banImmediately(BanEvent $event): void
  109.     {
  110.         $this->banService->add($event->getIp());
  111.     }
  112. }