/* Copyright (C) 2018 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 <errno.h>

#include <stdio.h>

#include <sys/resource.h>

#include <support/check.h>

static int resources[] = {

  /* The following 7 limits are part of POSIX and must exist.  */

  RLIMIT_CORE,

  RLIMIT_CPU,

  RLIMIT_DATA,

  RLIMIT_FSIZE,

  RLIMIT_NOFILE,

  RLIMIT_STACK,

  RLIMIT_AS

};

#define nresources (sizeof (resources) / sizeof (resources[0]))

/* Assume that the prlimit64 function calls the prlimit64 syscall without

   mangling the arguments.  */

#define PRLIMIT64_INFINITY	((rlim64_t) -1)

/* As we don't know which limit will be modified, use a sufficiently high

   value to not shoot ourself in the foot.  Use a 32-bit value to test

   both the 32- and 64-bit versions, and keep the highest bit clear to

   avoid sign extension.  */

#define PRLIMIT64_TESTVAL	((rlim64_t) 0x42420000)

static void

test_getrlimit (int resource, rlim_t exp_cur, rlim_t exp_max)

{

  struct rlimit r;

  TEST_VERIFY_EXIT (getrlimit (resource, &r) == 0);

  TEST_COMPARE (r.rlim_cur, exp_cur);

  TEST_COMPARE (r.rlim_max, exp_max);

}

static void

test_getrlimit64 (int resource, rlim64_t exp_cur, rlim64_t exp_max)

{

  struct rlimit64 r;

  TEST_VERIFY_EXIT (getrlimit64 (resource, &r) == 0);

  TEST_COMPARE (r.rlim_cur, exp_cur);

  TEST_COMPARE (r.rlim_max, exp_max);

}

static void

test_prlimit_get (int resource, rlim_t exp_cur, rlim_t exp_max)

{

  struct rlimit r;

  TEST_VERIFY_EXIT (prlimit (0, resource, NULL, &r) == 0);

  TEST_COMPARE (r.rlim_cur, exp_cur);

  TEST_COMPARE (r.rlim_max, exp_max);

}

static void

test_prlimit64_get (int resource, rlim64_t exp_cur, rlim64_t exp_max)

{

  struct rlimit64 r;

  TEST_COMPARE (prlimit64 (0, resource, NULL, &r), 0);

  TEST_COMPARE (r.rlim_cur, exp_cur);

  TEST_COMPARE (r.rlim_max, exp_max);

}

static void

test_setrlimit (int resource, rlim_t new_cur, rlim_t new_max)

{

  struct rlimit r = { new_cur, new_max };

  TEST_COMPARE (setrlimit (resource, &r), 0);

}

static void

test_setrlimit64 (int resource, rlim64_t new_cur, rlim64_t new_max)

{

  struct rlimit64 r = { new_cur, new_max };

  TEST_COMPARE (setrlimit64 (resource, &r), 0);

}

static void

test_prlimit_set (int resource, rlim_t new_cur, rlim_t new_max)

{

  struct rlimit r = { new_cur, new_max };

  TEST_COMPARE (prlimit (0, resource, &r, NULL), 0);

}

static void

test_prlimit64_set (int resource, rlim64_t new_cur, rlim64_t new_max)

{

  struct rlimit64 r = { new_cur, new_max };

  TEST_COMPARE (prlimit64 (0, resource, &r, NULL), 0);

}

static int

do_test (void)

{

  int resource = -1;

  /* Find a resource with hard limit set to infinity, so that the soft limit

     can be manipulated to any value.  */

  for (int i = 0; i < nresources; ++i)

    {

      struct rlimit64 r64;

      int res = prlimit64 (0, resources[i], NULL, &r64);

      if ((res == 0) && (r64.rlim_max == PRLIMIT64_INFINITY))

	{

	  resource = resources[i];

	  break;

	}

    }

  if (resource == -1)

    FAIL_UNSUPPORTED

      ("Could not find and limit with hard limit set to infinity.");

  /* First check that the get functions work correctly with the test value.  */

  test_prlimit64_set (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY);

  test_getrlimit (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY);

  test_getrlimit64 (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY);

  test_prlimit_get (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY);

  /* Then check that the get functions work correctly with infinity.  */

  test_prlimit64_set (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY);

  test_getrlimit (resource, RLIM_INFINITY, RLIM_INFINITY);

  test_getrlimit64 (resource, RLIM64_INFINITY, RLIM64_INFINITY);

  test_prlimit_get (resource, RLIM_INFINITY, RLIM_INFINITY);

  test_prlimit64_get (resource, RLIM64_INFINITY, RLIM64_INFINITY);

  /* Then check that setrlimit works correctly with the test value.  */

  test_setrlimit (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY);

  /* Then check that setrlimit works correctly with infinity.  */

  test_setrlimit (resource, RLIM_INFINITY, RLIM_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY);

  /* Then check that setrlimit64 works correctly with the test value.  */

  test_setrlimit64 (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY);

  /* Then check that setrlimit64 works correctly with infinity.  */

  test_setrlimit64 (resource, RLIM64_INFINITY, RLIM64_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY);

  /* Then check that prlimit works correctly with the test value.  */

  test_prlimit_set (resource, RLIM_INFINITY, RLIM_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY);

  /* Finally check that prlimit works correctly with infinity.  */

  test_prlimit_set (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY);

  test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY);

  return 0;

}

#include <support/test-driver.c>