/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * Copyright (c) 2007      Cisco Systems, Inc.  All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 *
 * This file is only here because some platforms have a broken strncpy
 * (e.g., Itanium with RedHat Advanced Server glibc).
 */

#include "opal_config.h"

#include <errno.h>
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#ifdef HAVE_SYS_RESOURCE_H
#include <sys/resource.h>
#endif

#include "opal/mca/base/mca_base_param.h"

#include "opal/util/sys_limits.h"
#include "opal/util/output.h"


/*
 * Create and initialize storage for the system limits
 */
OPAL_DECLSPEC opal_sys_limits_t opal_sys_limits = {
    /* initialized = */     false,
    /* num_files   = */     -1,
    /* num_procs   = */     -1,
    /* file_size   = */      0
};

int opal_util_init_sys_limits(void)
{
#ifndef __WINDOWS__    
    struct rlimit rlim;
#endif
    int value;
    bool set_lims;

    mca_base_param_reg_int_name("opal", "set_max_sys_limits", 
                                "Set to non-zero to automatically set any system-imposed limits to the maximum allowed",
                                false, false, (int)false, &value);
    set_lims = OPAL_INT_TO_BOOL(value);

#ifdef __WINDOWS__
    /* George: please insert whatever is needed here someday */
#else
    /* get/set the system limits on number of files we can have open */
    if (getrlimit (RLIMIT_NOFILE, &rlim) < 0) {
        opal_output(0, "getrlimit (RLIMIT_NOFILE) failed: %s\n", strerror(errno));
    } else {
        if (set_lims) {
            rlim.rlim_cur = rlim.rlim_max;
            if (setrlimit (RLIMIT_NOFILE, &rlim) < 0) {
                opal_output(0, "setrlimit (RLIMIT_NOFILE) failed: %s\n", strerror(errno));
            }
        }
        opal_sys_limits.num_files = rlim.rlim_cur;
    }

#if HAVE_DECL_RLIMIT_NPROC
    /* get/set the system limits on number of child procs we can have open */
    if (getrlimit (RLIMIT_NPROC, &rlim) < 0) {
        opal_output(0, "getrlimit (RLIMIT_NPROC) failed: %s\n", strerror(errno));
    } else {
        if (set_lims) {
            rlim.rlim_cur = rlim.rlim_max;
            if (setrlimit (RLIMIT_NPROC, &rlim) < 0) {
                opal_output(0, "setrlimit (RLIMIT_NPROC) failed: %s\n", strerror(errno));
            }
        }
        opal_sys_limits.num_procs = rlim.rlim_cur;
    }
#endif
    
    /* get/set the system limits on max file size we can create */
    if (getrlimit (RLIMIT_FSIZE, &rlim) < 0) {
        opal_output(0, "getrlimit (RLIMIT_FSIZE) failed: %s\n", strerror(errno));
    } else {
        if (set_lims) {
            rlim.rlim_cur = rlim.rlim_max;
            if (setrlimit (RLIMIT_FSIZE, &rlim) < 0) {
                opal_output(0, "setrlimit (RLIMIT_FSIZE) failed: %s\n", strerror(errno));
            }
        }
        opal_sys_limits.file_size = rlim.rlim_cur;
    }
    
    /* indicate we initialized the limits structure */
    opal_sys_limits.initialized = true;
#endif

  return OPAL_SUCCESS;
}