uclibc-ng/libc/sysdeps/linux/common/setrlimit64.c

/* Copyright (C) 1991,1995,1996,1997,1998,2000 Free Software Foundation, Inc.
   This file is part of the GNU C Library.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, see
   <http://www.gnu.org/licenses/>.  */

#include <_lfs_64.h>
#include <bits/wordsize.h>
#include <sys/resource.h>
#include <sys/syscall.h>
#include <stddef.h> // needed for NULL to be defined


#if defined(__NR_prlimit64)

int setrlimit64 (__rlimit_resource_t resource, const struct rlimit64 *rlimits)
{
	return INLINE_SYSCALL (prlimit64, 4, 0, resource, rlimits, NULL);
}

# if !defined(__NR_usetrlimit) && (__WORDSIZE == 64 || defined (__USE_FILE_OFFSET64))
/* If setrlimit is not implemented through the __NR_usetrlimit and size of
   rlimit_t == rlimit64_t then use setrlimit as an alias to setrlimit64 */
strong_alias_untyped(setrlimit64, setrlimit)
libc_hidden_def(setrlimit)
# endif

#else

/* Set the soft and hard limits for RESOURCE to *RLIMITS.
   Only the super-user can increase hard limits.
   Return 0 if successful, -1 if not (and sets errno).
   The regular setrlimit will work just fine for 64bit users  */
int setrlimit64 (__rlimit_resource_t resource, const struct rlimit64 *rlimits)
{
    struct rlimit rlimits32;

    if (rlimits->rlim_cur >= RLIM_INFINITY)
	rlimits32.rlim_cur = RLIM_INFINITY;
    else
	rlimits32.rlim_cur = rlimits->rlim_cur;
    if (rlimits->rlim_max >= RLIM_INFINITY)
	rlimits32.rlim_max = RLIM_INFINITY;
    else
	rlimits32.rlim_max = rlimits->rlim_max;

    return setrlimit (resource, &rlimits32);
}
#endif