# ResourceLimits.pm                          Copyright(c) 2008       cPanel, Inc.
# copyright@cpanel.net                                         http://cpanel.net
# <@LICENSE>
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to you under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# </@LICENSE>

package Mail::SpamAssassin::Plugin::ResourceLimits;

use Mail::SpamAssassin::Plugin ();
use Mail::SpamAssassin::Logger ();
use Mail::SpamAssassin::Util   ();
use Mail::SpamAssassin::Constants qw(:sa);

use strict;
use warnings;

use BSD::Resource qw(setrlimit RLIM_INFINITY RLIMIT_RSS RLIMIT_AS RLIMIT_CORE RLIMIT_CPU);

use vars qw(@ISA);
@ISA = qw(Mail::SpamAssassin::Plugin);

sub new {
    my $class        = shift;
    my $mailsaobject = shift;

    $class = ref($class) || $class;
    my $self = $class->SUPER::new($mailsaobject);
    bless( $self, $class );

    $self->set_config( $mailsaobject->{conf} );
    return $self;
}

sub set_config {
    my ( $self, $conf ) = @_;
    my @cmds = ();

    push(
        @cmds,
        {
            setting  => 'resource_limit_mem',
            is_admin => 1,
            default  => '0',
            type     => $Mail::SpamAssassin::Conf::CONF_TYPE_NUMERIC
        }
    );

    push(
        @cmds,
        {
            setting  => 'resource_limit_cpu',
            is_admin => 1,
            default  => '0',
            type     => $Mail::SpamAssassin::Conf::CONF_TYPE_NUMERIC
        }
    );

    $conf->{parser}->register_commands( \@cmds );
}

sub spamd_child_init {
    my ($self) = @_;
    Mail::SpamAssassin::Util::dbg("resourcelimitplugin: In spamd_child_init");
    Mail::SpamAssassin::Util::dbg( "resourcelimitplugin: cpu limit: " . $self->{main}->{conf}->{resource_limit_cpu} );
    if ( $self->{main}->{conf}->{resource_limit_cpu} ) {
        BSD::Resource::setrlimit( RLIMIT_CPU, $self->{main}->{conf}->{resource_limit_cpu}, $self->{main}->{conf}->{resource_limit_cpu} )
          || info("resourcelimitplugin: Unable to set RLIMIT_CPU");
    }
    Mail::SpamAssassin::Util::dbg( "resourcelimitplugin: mem limit: " . $self->{main}->{conf}->{resource_limit_mem} );
    if ( $self->{main}->{conf}->{resource_limit_mem} ) {
        BSD::Resource::setrlimit( RLIMIT_RSS, $self->{main}->{conf}->{resource_limit_mem}, $self->{main}->{conf}->{resource_limit_mem} )
          || info("resourcelimitplugin: Unable to set RLIMIT_RSS");
        BSD::Resource::setrlimit( RLIMIT_AS, $self->{main}->{conf}->{resource_limit_mem}, $self->{main}->{conf}->{resource_limit_mem} )
          || info("resourcelimitplugin: Unable to set RLIMIT_AS");
    }
}

1;