diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/cmd/rc/mkfile | 1 | ||||
-rw-r--r-- | src/cmd/rc/plan9ish.c | 4 | ||||
-rw-r--r-- | src/cmd/rc/unixcrap.c | 197 |
3 files changed, 202 insertions, 0 deletions
diff --git a/src/cmd/rc/mkfile b/src/cmd/rc/mkfile index f7d50e3f..09478dc0 100644 --- a/src/cmd/rc/mkfile +++ b/src/cmd/rc/mkfile @@ -17,6 +17,7 @@ OFILES=\ subr.$O\ trap.$O\ tree.$O\ + unixcrap.$O\ var.$O\ y.tab.$O\ plan9ish.$O\ diff --git a/src/cmd/rc/plan9ish.c b/src/cmd/rc/plan9ish.c index af1e449d..bd9698c8 100644 --- a/src/cmd/rc/plan9ish.c +++ b/src/cmd/rc/plan9ish.c @@ -34,6 +34,8 @@ char Fdprefix[]="/dev/fd/"; void execfinit(void); void execbind(void); void execmount(void); +void execulimit(void); +void execumask(void); builtin Builtin[]={ "cd", execcd, "whatis", execwhatis, @@ -45,6 +47,8 @@ builtin Builtin[]={ ".", execdot, "finit", execfinit, "flag", execflag, + "ulimit", execulimit, + "umask", execumask, 0 }; #define SEP '\1' diff --git a/src/cmd/rc/unixcrap.c b/src/cmd/rc/unixcrap.c new file mode 100644 index 00000000..57da10ed --- /dev/null +++ b/src/cmd/rc/unixcrap.c @@ -0,0 +1,197 @@ +#include <u.h> +#include <sys/time.h> +#include <sys/resource.h> +#include <libc.h> +#include "rc.h" +#include "exec.h" +#include "io.h" +#include "fns.h" +#include "getflags.h" + +extern char **mkargv(word*); +extern int mapfd(int); + +static char *eargs = "cdflmnstuv"; +static int rlx[] = { + RLIMIT_CORE, + RLIMIT_DATA, + RLIMIT_FSIZE, + RLIMIT_MEMLOCK, + RLIMIT_RSS, + RLIMIT_NOFILE, + RLIMIT_STACK, + RLIMIT_CPU, + RLIMIT_NPROC, + RLIMIT_RSS, +}; + +static void +eusage(void) +{ + fprint(mapfd(2), "usage: ulimit [-SHa%s [limit]]\n", eargs); +} + +#define Notset -4 +#define Unlimited -3 +#define Hard -2 +#define Soft -1 + +void +execulimit(void) +{ + int fd, n, argc, sethard, setsoft, limit; + int flag[256]; + char **argv, **oargv, *p; + char *argv0; + struct rlimit rl; + + argv0 = nil; + setstatus(""); + oargv = mkargv(runq->argv->words); + argv = oargv+1; + for(argc=0; argv[argc]; argc++) + ; + + memset(flag, 0, sizeof flag); + ARGBEGIN{ + default: + if(strchr(eargs, ARGC()) == nil){ + eusage(); + return; + } + case 'S': + case 'H': + case 'a': + flag[ARGC()] = 1; + break; + }ARGEND + + if(argc > 1){ + eusage(); + goto out; + } + + fd = mapfd(1); + + sethard = 1; + setsoft = 1; + if(flag['S'] && flag['H']) + ; + else if(flag['S']) + sethard = 0; + else if(flag['H']) + setsoft = 0; + + limit = Notset; + if(argc>0){ + if(strcmp(argv[0], "unlimited") == 0) + limit = Unlimited; + else if(strcmp(argv[0], "hard") == 0) + limit = Hard; + else if(strcmp(argv[0], "soft") == 0) + limit = Soft; + else if((limit = strtol(argv[0], &p, 0)) < 0 || *p != 0){ + eusage(); + goto out; + } + } + if(flag['a']){ + for(p=eargs; *p; p++){ + getrlimit(rlx[p-eargs], &rl); + n = flag['H'] ? rl.rlim_max : rl.rlim_cur; + if(n == -1) + fprint(fd, "ulimit -%c unlimited\n", *p); + else + fprint(fd, "ulimit -%c %d\n", *p, n); + } + goto out; + } + for(p=eargs; *p; p++){ + if(flag[(uchar)*p]){ + n = 0; + getrlimit(rlx[p-eargs], &rl); + switch(limit){ + case Notset: + n = flag['H'] ? rl.rlim_max : rl.rlim_cur; + if(n == -1) + fprint(fd, "ulimit -%c unlimited\n", *p); + else + fprint(fd, "ulimit -%c %d\n", *p, n); + break; + case Hard: + n = rl.rlim_max; + goto set; + case Soft: + n = rl.rlim_cur; + goto set; + case Unlimited: + n = -1; + goto set; + default: + n = limit; + set: + if(setsoft) + rl.rlim_cur = n; + if(sethard) + rl.rlim_max = n; + if(setrlimit(rlx[p-eargs], &rl) < 0) + fprint(mapfd(2), "setrlimit: %r\n"); + } + } + } + +out: + free(oargv); + poplist(); + flush(err); +} + +void +execumask(void) +{ + int n, argc; + char **argv, **oargv, *p; + char *argv0; + + argv0 = nil; + setstatus(""); + oargv = mkargv(runq->argv->words); + argv = oargv+1; + for(argc=0; argv[argc]; argc++) + ; + + ARGBEGIN{ + default: + usage: + fprint(mapfd(2), "usage: umask [mode]\n"); + goto out; + }ARGEND + + if(argc > 1) + goto usage; + + if(argc == 1){ + n = strtol(argv[0], &p, 8); + if(*p != 0 || p == argv[0]) + goto usage; + if(umask(n) < 0){ + fprint(mapfd(2), "umask: %r\n"); + goto out; + } + goto out; + } + + n = umask(0); + umask(n); + if(n < 0){ + fprint(mapfd(2), "umask: %r\n"); + goto out; + } + + fprint(mapfd(1), "umask %03o\n", n); + +out: + free(oargv); + poplist(); + flush(err); +} |