ResourceLimit.cpp

/*
** libutil++
** $Id: ResourceLimit.cpp 1653 2016-02-28 19:54:59Z sella $
** Copyright (c) 2011-2016 Digital Genesis, LLC. All Rights Reserved.
** Released under the LGPL Version 2.1 License.
** http://www.digitalgenesis.com
*/

static const char rcsid[] __attribute__((used)) = "$Id: ResourceLimit.cpp 1653 2016-02-28 19:54:59Z sella $";

#include "ResourceLimit.h"

#include <sys/resource.h>

using namespace sella::util;

void ResourceLimit::setCoredumpSize(unsigned long size, Type type) throw (ResourceLimitException) {
        struct rlimit limit;

        if (type == Both) {
                limit.rlim_cur = limit.rlim_max = size;
        } else if (type == Hard) {
                limit.rlim_max = size;
        } else {
                limit.rlim_cur = size;
        }
        
        if (setrlimit(RLIMIT_CORE, &limit) != 0) {
                THROW2(ResourceLimitException, "Failed to adjust coredump size");
        }
}

void ResourceLimit::setFileDescriptors(unsigned long size, Type type) throw (ResourceLimitException) {
        struct rlimit limit;

        if (type == Both) {
                limit.rlim_cur = limit.rlim_max = size;
        } else if (type == Hard) {
                limit.rlim_max = size;
        } else {
                limit.rlim_cur = size;
        }
        
        if (setrlimit(RLIMIT_NOFILE, &limit) != 0) {
                THROW2(ResourceLimitException, "Failed to adjust file descriptor limit");
        }
}

unsigned long ResourceLimit::getCoredumpSize(Type type) throw (ResourceLimitException) {
        struct rlimit limit;
        
        if (getrlimit(RLIMIT_CORE, &limit) != 0) {
                THROW2(ResourceLimitException, "Failed to access coredump size");
        }

        if (type == Hard) {
                return limit.rlim_max;
        }

        return limit.rlim_cur;
}

unsigned long ResourceLimit::getFileDescriptors(Type type) throw (ResourceLimitException) {
        struct rlimit limit;
        
        if (getrlimit(RLIMIT_NOFILE, &limit) != 0) {
                THROW2(ResourceLimitException, "Failed to access file descriptor limit");
        }

        if (type == Hard) {
                return limit.rlim_max;
        }

        return limit.rlim_cur;
}

/*
** vim: noet ts=3 sw=3
*/