| /* |
| * |
| * Copyright (c) International Business Machines Corp., 2001 |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program 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 General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| /* |
| * Testcase to check the basic functionality of the setrlimit system call. |
| * Use the different commands like RLIMIT_NOFILE, RLIMIT_CORE, |
| * RLIMIT_FSIZE, and, RLIMIT_NOFILE, and test for different test |
| * conditions. |
| * |
| * 07/2001 Ported by Wayne Boyer |
| */ |
| |
| #include <sys/types.h> |
| #include <sys/resource.h> |
| #include <sys/stat.h> |
| #include <sys/time.h> |
| #include <sys/wait.h> |
| #include <errno.h> |
| #include <fcntl.h> |
| #include <stdlib.h> |
| #include <unistd.h> |
| #include "test.h" |
| |
| char *TCID = "setrlimit01"; |
| int TST_TOTAL = 1; |
| |
| static void setup(void); |
| static void cleanup(void); |
| static void test1(void); |
| static void test2(void); |
| static void test3(void); |
| static void test4(void); |
| static void sighandler(int); |
| |
| static char filename[40] = ""; |
| static struct rlimit save_rlim, rlim, rlim1; |
| static int nofiles, fd, bytes, i, status; |
| static char *buf = "abcdefghijklmnopqrstuvwxyz"; |
| static pid_t pid; |
| |
| int main(int ac, char **av) |
| { |
| int lc; |
| const char *msg; |
| |
| msg = parse_opts(ac, av, NULL, NULL); |
| if (msg != NULL) |
| tst_brkm(TBROK, NULL, "OPTION PARSING ERROR - %s", msg); |
| |
| setup(); |
| |
| for (lc = 0; TEST_LOOPING(lc); lc++) { |
| tst_count = 0; |
| |
| test1(); |
| test2(); |
| test3(); |
| /* reset saved conditions */ |
| if ((setrlimit(RLIMIT_NPROC, &save_rlim)) == -1) { |
| tst_brkm(TBROK, cleanup, "setrlimit failed to reset " |
| "RLIMIT_NPROC, errno = %d", errno); |
| } |
| test4(); |
| } |
| |
| cleanup(); |
| tst_exit(); |
| } |
| |
| /* |
| * test1 - Test for RLIMIT_NOFILE |
| */ |
| static void test1(void) |
| { |
| rlim.rlim_cur = 100; |
| rlim.rlim_max = 100; |
| |
| TEST(setrlimit(RLIMIT_NOFILE, &rlim)); |
| |
| if (TEST_RETURN == -1) { |
| tst_resm(TFAIL, "setrlimit failed to set " |
| "RLIMIT_NOFILE, errno = %d", errno); |
| return; |
| } |
| |
| nofiles = getdtablesize(); |
| |
| if (nofiles != 100) { |
| tst_resm(TFAIL, "setrlimit failed, expected " |
| "100, got %d", nofiles); |
| return; |
| } |
| |
| tst_resm(TPASS, "RLIMIT_NOFILE functionality is correct"); |
| } |
| |
| /* |
| * test2 - Test for RLIMIT_FSIZE |
| */ |
| static void test2(void) |
| { |
| /* |
| * Since we would be altering the filesize in the child, |
| * we need to "sync", ie. fflush the parent's write buffers |
| * here. This is because the child will inherit the parent's |
| * write buffer, and while exitting it would try to fflush it. |
| * Since its filesize is truncated to only 10 bytes, the |
| * fflush attempt would fail, and the child would exit with |
| * an wired value! So, it is essential to fflush the parent's |
| * write buffer HERE |
| */ |
| int pipefd[2]; |
| fflush(stdout); |
| if (pipe(pipefd) == -1) |
| tst_brkm(TBROK | TERRNO, NULL, "pipe creation failed"); |
| |
| /* |
| * Spawn a child process, and reduce the filesize to |
| * 10 by calling setrlimit(). We can't do this in the |
| * parent, because the parent needs a bigger filesize as its |
| * output will be saved to the logfile (instead of stdout) |
| * when the testcase (parent) is run from the driver. |
| */ |
| pid = FORK_OR_VFORK(); |
| if (pid == -1) |
| tst_brkm(TBROK, cleanup, "fork() failed"); |
| |
| if (pid == 0) { |
| close(pipefd[0]); /* close unused read end */ |
| rlim.rlim_cur = 10; |
| rlim.rlim_max = 10; |
| if ((setrlimit(RLIMIT_FSIZE, &rlim)) == -1) |
| exit(1); |
| |
| fd = creat(filename, 0644); |
| if (fd < 0) |
| exit(2); |
| |
| bytes = write(fd, buf, 26); |
| if (bytes != 10) { |
| if (write(pipefd[1], &bytes, sizeof(bytes)) |
| < sizeof(bytes)) { |
| perror("child: write to pipe failed"); |
| } |
| close(pipefd[1]); /* EOF */ |
| exit(3); |
| } |
| exit(0); /* success */ |
| } |
| |
| /* parent */ |
| if (waitpid(pid, &status, 0) == -1) |
| tst_brkm(TBROK, cleanup, "waitpid() failed"); |
| |
| switch (WEXITSTATUS(status)) { |
| case 0: |
| tst_resm(TPASS, "RLIMIT_FSIZE test PASSED"); |
| break; |
| case 1: |
| tst_resm(TFAIL, "setrlimit failed to set " |
| "RLIMIT_FSIZE, errno = %d", errno); |
| break; |
| case 2: |
| tst_resm(TFAIL, "creating testfile failed"); |
| break; |
| case 3: |
| close(pipefd[1]); /* close unused write end */ |
| if (read(pipefd[0], &bytes, sizeof(bytes)) < sizeof(bytes)) |
| tst_resm(TFAIL, "parent: reading pipe failed"); |
| |
| close(pipefd[0]); |
| tst_resm(TFAIL, "setrlimit failed, expected " |
| "10 got %d", bytes); |
| break; |
| default: |
| tst_resm(TFAIL, "child returned bad exit status"); |
| } |
| } |
| |
| /* |
| * test3 - Test for RLIMIT_NPROC |
| */ |
| static void test3(void) |
| { |
| if (getrlimit(RLIMIT_NPROC, &save_rlim) < 0) |
| tst_brkm(TBROK, cleanup, "getrlimit failed, errno: %d", errno); |
| |
| rlim.rlim_cur = 10; |
| rlim.rlim_max = 10; |
| |
| TEST(setrlimit(RLIMIT_NPROC, &rlim)); |
| |
| if (TEST_RETURN == -1) { |
| tst_resm(TFAIL, "setrlimit failed to set " |
| "RLIMIT_NPROC, errno = %d", errno); |
| return; |
| } |
| |
| if ((getrlimit(RLIMIT_NPROC, &rlim1)) == -1) { |
| tst_brkm(TBROK, cleanup, "getrlimit failed to get " |
| "values for RLIMIT_NPROC, errno = %d", errno); |
| } |
| |
| if ((rlim1.rlim_cur != 10) && (rlim1.rlim_max != 10)) { |
| tst_resm(TFAIL, "setrlimit did not set the proc " |
| "limit correctly"); |
| return; |
| } |
| |
| for (i = 0; i < 20; i++) { |
| pid = FORK_OR_VFORK(); |
| if (pid == -1) { |
| if (errno != EAGAIN) { |
| tst_resm(TWARN, "Expected EAGAIN got %d", |
| errno); |
| } |
| } else if (pid == 0) { |
| exit(0); |
| } |
| } |
| waitpid(pid, &status, 0); |
| if (WEXITSTATUS(status) != 0) |
| tst_resm(TFAIL, "RLIMIT_NPROC functionality is not correct"); |
| else |
| tst_resm(TPASS, "RLIMIT_NPROC functionality is correct"); |
| } |
| |
| /* |
| * test4() - Test for RLIMIT_CORE by forking a child and |
| * having it cause a segfault |
| */ |
| static void test4(void) |
| { |
| rlim.rlim_cur = 10; |
| rlim.rlim_max = 10; |
| |
| TEST(setrlimit(RLIMIT_CORE, &rlim)); |
| |
| if (TEST_RETURN == -1) { |
| tst_resm(TFAIL | TERRNO, "setrlimit failed to set RLIMIT_CORE"); |
| return; |
| } |
| |
| pid = FORK_OR_VFORK(); |
| if (pid == -1) |
| tst_brkm(TBROK, cleanup, "fork() failed"); |
| |
| if (pid == 0) { /* child */ |
| char *testbuf = NULL; |
| strcpy(testbuf, "abcd"); |
| exit(0); |
| } |
| wait(&status); |
| |
| if (access("core", F_OK) == 0) { |
| tst_resm(TFAIL, "core dump dumped unexpectedly"); |
| return; |
| } else if (errno != ENOENT) { |
| tst_resm(TFAIL | TERRNO, "access failed unexpectedly"); |
| return; |
| } |
| |
| tst_resm(TPASS, "RLIMIT_CORE functionality is correct"); |
| } |
| |
| /* |
| * sighandler() - catch sigsegv when generated by child in test #4 |
| */ |
| static void sighandler(int sig) |
| { |
| if (sig != SIGSEGV && sig != SIGXFSZ && sig != SIGTERM) |
| tst_brkm(TBROK, NULL, "caught unexpected signal: %d", sig); |
| |
| _exit(0); |
| } |
| |
| static void setup(void) |
| { |
| tst_require_root(NULL); |
| |
| umask(0); |
| |
| tst_sig(FORK, sighandler, cleanup); |
| |
| TEST_PAUSE; |
| |
| tst_tmpdir(); |
| |
| sprintf(filename, "setrlimit1.%d", getpid()); |
| } |
| |
| static void cleanup(void) |
| { |
| unlink(filename); |
| tst_rmdir(); |
| } |