Bugzilla/Quantum/Plugin/BlockIP.pm


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

package Bugzilla::Quantum::Plugin::BlockIP;
use 5.10.1;
use Mojo::Base 'Mojolicious::Plugin';

use Bugzilla::Memcached;

use constant BLOCK_TIMEOUT => 60 * 60;

my $MEMCACHED = Bugzilla::Memcached->new()->{memcached};

sub register {
  my ($self, $app, $conf) = @_;

  $app->hook(before_routes => \&_before_routes);
  $app->helper(block_ip   => \&_block_ip);
  $app->helper(unblock_ip => \&_unblock_ip);
}

sub _block_ip {
  my ($class, $ip) = @_;
  $MEMCACHED->set("block_ip:$ip" => 1, BLOCK_TIMEOUT) if $MEMCACHED;
}

sub _unblock_ip {
  my ($class, $ip) = @_;
  $MEMCACHED->delete("block_ip:$ip") if $MEMCACHED;
}

sub _before_routes {
  my ($c) = @_;
  return if $c->stash->{'mojo.static'};

  my $ip = $c->tx->remote_address;
  if ($MEMCACHED && $MEMCACHED->get("block_ip:$ip")) {
    $c->block_ip($ip);
    $c->res->code(429);
    $c->res->message('Too Many Requests');
    $c->res->body('Too Many Requests');
    $c->finish;
  }
}

1;