1 /* 2 * Copyright (c) 2014, The Android Open Source Project 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in 12 * the documentation and/or other materials provided with the 13 * distribution. 14 * * Neither the name of Google, Inc. nor the names of its contributors 15 * may be used to endorse or promote products derived from this 16 * software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 21 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 22 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 24 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS 25 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 26 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 27 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 28 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <stdio.h> 33 #include <unistd.h> 34 #include <stdlib.h> 35 #include <sys/time.h> 36 #include <sys/resource.h> 37 38 static void 39 usage(const char *s) 40 { 41 fprintf(stderr, "usage: %s pid resource cur max\n", s); 42 exit(EXIT_FAILURE); 43 } 44 45 int prlimit_main(int argc, char *argv[]) 46 { 47 pid_t pid; 48 struct rlimit64 rl; 49 int resource; 50 int rc; 51 52 if (argc != 5) 53 usage(*argv); 54 55 if (sscanf(argv[1], "%d", &pid) != 1) 56 usage(*argv); 57 58 if (sscanf(argv[2], "%d", &resource) != 1) 59 usage(*argv); 60 61 if (sscanf(argv[3], "%llu", &rl.rlim_cur) != 1) 62 usage(*argv); 63 64 if (sscanf(argv[4], "%llu", &rl.rlim_max) != 1) 65 usage(*argv); 66 67 printf("setting resource %d of pid %d to [%llu,%llu]\n", resource, pid, 68 rl.rlim_cur, rl.rlim_max); 69 rc = prlimit64(pid, resource, &rl, NULL); 70 if (rc < 0) { 71 perror("prlimit"); 72 exit(EXIT_FAILURE); 73 } 74 75 return 0; 76 } 77
