op public repos

Blame

Date:: Tue Dec 28 17:34:05 2004 UTC
Message:: FreeBSD tweaks
Actions:: History | Blob | Raw File
  1 39cff6e7 2004-10-17 devnull #include <u.h>
  2 39cff6e7 2004-10-17 devnull #include <sys/time.h>
  3 f7b74c17 2004-12-28 devnull #include <sys/stat.h>
  4 39cff6e7 2004-10-17 devnull #include <sys/resource.h>
  5 39cff6e7 2004-10-17 devnull #include <libc.h>
  6 39cff6e7 2004-10-17 devnull #include "rc.h"
  7 39cff6e7 2004-10-17 devnull #include "exec.h"
  8 39cff6e7 2004-10-17 devnull #include "io.h"
  9 39cff6e7 2004-10-17 devnull #include "fns.h"
 10 39cff6e7 2004-10-17 devnull #include "getflags.h"
 11 39cff6e7 2004-10-17 devnull 
 12 39cff6e7 2004-10-17 devnull extern char **mkargv(word*);
 13 39cff6e7 2004-10-17 devnull extern int mapfd(int);
 14 39cff6e7 2004-10-17 devnull 
 15 39cff6e7 2004-10-17 devnull static char *eargs = "cdflmnstuv";
 16 39cff6e7 2004-10-17 devnull static int rlx[] = {
 17 39cff6e7 2004-10-17 devnull 	RLIMIT_CORE,
 18 39cff6e7 2004-10-17 devnull 	RLIMIT_DATA,
 19 39cff6e7 2004-10-17 devnull 	RLIMIT_FSIZE,
 20 39cff6e7 2004-10-17 devnull 	RLIMIT_MEMLOCK,
 21 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,
 22 39cff6e7 2004-10-17 devnull 	RLIMIT_NOFILE,
 23 39cff6e7 2004-10-17 devnull 	RLIMIT_STACK,
 24 39cff6e7 2004-10-17 devnull 	RLIMIT_CPU,
 25 39cff6e7 2004-10-17 devnull 	RLIMIT_NPROC,
 26 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,
 27 39cff6e7 2004-10-17 devnull };
 28 39cff6e7 2004-10-17 devnull 
 29 39cff6e7 2004-10-17 devnull static void
 30 39cff6e7 2004-10-17 devnull eusage(void)
 31 39cff6e7 2004-10-17 devnull {
 32 39cff6e7 2004-10-17 devnull 	fprint(mapfd(2), "usage: ulimit [-SHa%s [limit]]\n", eargs);
 33 39cff6e7 2004-10-17 devnull }
 34 39cff6e7 2004-10-17 devnull 
 35 39cff6e7 2004-10-17 devnull #define Notset -4
 36 39cff6e7 2004-10-17 devnull #define Unlimited -3
 37 39cff6e7 2004-10-17 devnull #define Hard -2
 38 39cff6e7 2004-10-17 devnull #define Soft -1
 39 39cff6e7 2004-10-17 devnull 
 40 39cff6e7 2004-10-17 devnull void
 41 39cff6e7 2004-10-17 devnull execulimit(void)
 42 39cff6e7 2004-10-17 devnull {
 43 39cff6e7 2004-10-17 devnull 	int fd, n, argc, sethard, setsoft, limit;
 44 39cff6e7 2004-10-17 devnull 	int flag[256];
 45 39cff6e7 2004-10-17 devnull 	char **argv, **oargv, *p;
 46 39cff6e7 2004-10-17 devnull 	char *argv0;
 47 39cff6e7 2004-10-17 devnull 	struct rlimit rl;
 48 39cff6e7 2004-10-17 devnull 
 49 39cff6e7 2004-10-17 devnull 	argv0 = nil;
 50 39cff6e7 2004-10-17 devnull 	setstatus("");
 51 39cff6e7 2004-10-17 devnull 	oargv = mkargv(runq->argv->words);
 52 39cff6e7 2004-10-17 devnull 	argv = oargv+1;
 53 39cff6e7 2004-10-17 devnull 	for(argc=0; argv[argc]; argc++)
 54 39cff6e7 2004-10-17 devnull 		;
 55 39cff6e7 2004-10-17 devnull 
 56 39cff6e7 2004-10-17 devnull 	memset(flag, 0, sizeof flag);
 57 39cff6e7 2004-10-17 devnull 	ARGBEGIN{
 58 39cff6e7 2004-10-17 devnull 	default:
 59 39cff6e7 2004-10-17 devnull 		if(strchr(eargs, ARGC()) == nil){
 60 39cff6e7 2004-10-17 devnull 			eusage();
 61 39cff6e7 2004-10-17 devnull 			return;
 62 39cff6e7 2004-10-17 devnull 		}
 63 39cff6e7 2004-10-17 devnull 	case 'S':
 64 39cff6e7 2004-10-17 devnull 	case 'H':
 65 39cff6e7 2004-10-17 devnull 	case 'a':
 66 39cff6e7 2004-10-17 devnull 		flag[ARGC()] = 1;
 67 39cff6e7 2004-10-17 devnull 		break;
 68 39cff6e7 2004-10-17 devnull 	}ARGEND
 69 39cff6e7 2004-10-17 devnull 
 70 39cff6e7 2004-10-17 devnull 	if(argc > 1){
 71 39cff6e7 2004-10-17 devnull 		eusage();
 72 39cff6e7 2004-10-17 devnull 		goto out;
 73 39cff6e7 2004-10-17 devnull 	}
 74 39cff6e7 2004-10-17 devnull 
 75 39cff6e7 2004-10-17 devnull 	fd = mapfd(1);
 76 39cff6e7 2004-10-17 devnull 
 77 39cff6e7 2004-10-17 devnull 	sethard = 1;
 78 39cff6e7 2004-10-17 devnull 	setsoft = 1;
 79 39cff6e7 2004-10-17 devnull 	if(flag['S'] && flag['H'])
 80 39cff6e7 2004-10-17 devnull 		;
 81 39cff6e7 2004-10-17 devnull 	else if(flag['S'])
 82 39cff6e7 2004-10-17 devnull 		sethard = 0;
 83 39cff6e7 2004-10-17 devnull 	else if(flag['H'])
 84 39cff6e7 2004-10-17 devnull 		setsoft = 0;
 85 39cff6e7 2004-10-17 devnull 
 86 39cff6e7 2004-10-17 devnull 	limit = Notset;
 87 39cff6e7 2004-10-17 devnull 	if(argc>0){
 88 39cff6e7 2004-10-17 devnull 		if(strcmp(argv[0], "unlimited") == 0)
 89 39cff6e7 2004-10-17 devnull 			limit = Unlimited;
 90 39cff6e7 2004-10-17 devnull 		else if(strcmp(argv[0], "hard") == 0)
 91 39cff6e7 2004-10-17 devnull 			limit = Hard;
 92 39cff6e7 2004-10-17 devnull 		else if(strcmp(argv[0], "soft") == 0)
 93 39cff6e7 2004-10-17 devnull 			limit = Soft;
 94 39cff6e7 2004-10-17 devnull 		else if((limit = strtol(argv[0], &p, 0)) < 0 || *p != 0){
 95 39cff6e7 2004-10-17 devnull 			eusage();
 96 39cff6e7 2004-10-17 devnull 			goto out;
 97 39cff6e7 2004-10-17 devnull 		}
 98 39cff6e7 2004-10-17 devnull 	}
 99 39cff6e7 2004-10-17 devnull 	if(flag['a']){
100 39cff6e7 2004-10-17 devnull 		for(p=eargs; *p; p++){
101 39cff6e7 2004-10-17 devnull 			getrlimit(rlx[p-eargs], &rl);
102 39cff6e7 2004-10-17 devnull 			n = flag['H'] ? rl.rlim_max : rl.rlim_cur;
103 39cff6e7 2004-10-17 devnull 			if(n == -1)
104 39cff6e7 2004-10-17 devnull 				fprint(fd, "ulimit -%c unlimited\n", *p);
105 39cff6e7 2004-10-17 devnull 			else
106 39cff6e7 2004-10-17 devnull 				fprint(fd, "ulimit -%c %d\n", *p, n);
107 39cff6e7 2004-10-17 devnull 		}
108 39cff6e7 2004-10-17 devnull 		goto out;
109 39cff6e7 2004-10-17 devnull 	}
110 39cff6e7 2004-10-17 devnull 	for(p=eargs; *p; p++){
111 39cff6e7 2004-10-17 devnull 		if(flag[(uchar)*p]){
112 39cff6e7 2004-10-17 devnull 			n = 0;
113 39cff6e7 2004-10-17 devnull 			getrlimit(rlx[p-eargs], &rl);
114 39cff6e7 2004-10-17 devnull 			switch(limit){
115 39cff6e7 2004-10-17 devnull 			case Notset:
116 39cff6e7 2004-10-17 devnull 				n = flag['H'] ? rl.rlim_max : rl.rlim_cur;
117 39cff6e7 2004-10-17 devnull 				if(n == -1)
118 39cff6e7 2004-10-17 devnull 					fprint(fd, "ulimit -%c unlimited\n", *p);
119 39cff6e7 2004-10-17 devnull 				else
120 39cff6e7 2004-10-17 devnull 					fprint(fd, "ulimit -%c %d\n", *p, n);
121 39cff6e7 2004-10-17 devnull 				break;
122 39cff6e7 2004-10-17 devnull 			case Hard:
123 39cff6e7 2004-10-17 devnull 				n = rl.rlim_max;
124 39cff6e7 2004-10-17 devnull 				goto set;
125 39cff6e7 2004-10-17 devnull 			case Soft:
126 39cff6e7 2004-10-17 devnull 				n = rl.rlim_cur;
127 39cff6e7 2004-10-17 devnull 				goto set;
128 39cff6e7 2004-10-17 devnull 			case Unlimited:
129 39cff6e7 2004-10-17 devnull 				n = -1;
130 39cff6e7 2004-10-17 devnull 				goto set;
131 39cff6e7 2004-10-17 devnull 			default:
132 39cff6e7 2004-10-17 devnull 				n = limit;
133 39cff6e7 2004-10-17 devnull 			set:
134 39cff6e7 2004-10-17 devnull 				if(setsoft)
135 39cff6e7 2004-10-17 devnull 					rl.rlim_cur = n;
136 39cff6e7 2004-10-17 devnull 				if(sethard)
137 39cff6e7 2004-10-17 devnull 					rl.rlim_max = n;
138 39cff6e7 2004-10-17 devnull 				if(setrlimit(rlx[p-eargs], &rl) < 0)
139 39cff6e7 2004-10-17 devnull 					fprint(mapfd(2), "setrlimit: %r\n");
140 39cff6e7 2004-10-17 devnull 			}
141 39cff6e7 2004-10-17 devnull 		}
142 39cff6e7 2004-10-17 devnull 	}
143 39cff6e7 2004-10-17 devnull 
144 39cff6e7 2004-10-17 devnull out:
145 39cff6e7 2004-10-17 devnull 	free(oargv);
146 39cff6e7 2004-10-17 devnull 	poplist();
147 39cff6e7 2004-10-17 devnull 	flush(err);
148 39cff6e7 2004-10-17 devnull }
149 39cff6e7 2004-10-17 devnull 
150 39cff6e7 2004-10-17 devnull void
151 39cff6e7 2004-10-17 devnull execumask(void)
152 39cff6e7 2004-10-17 devnull {
153 39cff6e7 2004-10-17 devnull 	int n, argc;
154 39cff6e7 2004-10-17 devnull 	char **argv, **oargv, *p;
155 39cff6e7 2004-10-17 devnull 	char *argv0;
156 39cff6e7 2004-10-17 devnull 
157 39cff6e7 2004-10-17 devnull 	argv0 = nil;
158 39cff6e7 2004-10-17 devnull 	setstatus("");
159 39cff6e7 2004-10-17 devnull 	oargv = mkargv(runq->argv->words);
160 39cff6e7 2004-10-17 devnull 	argv = oargv+1;
161 39cff6e7 2004-10-17 devnull 	for(argc=0; argv[argc]; argc++)
162 39cff6e7 2004-10-17 devnull 		;
163 39cff6e7 2004-10-17 devnull 
164 39cff6e7 2004-10-17 devnull 	ARGBEGIN{
165 39cff6e7 2004-10-17 devnull 	default:
166 39cff6e7 2004-10-17 devnull 	usage:
167 39cff6e7 2004-10-17 devnull 		fprint(mapfd(2), "usage: umask [mode]\n");
168 39cff6e7 2004-10-17 devnull 		goto out;
169 39cff6e7 2004-10-17 devnull 	}ARGEND
170 39cff6e7 2004-10-17 devnull 
171 39cff6e7 2004-10-17 devnull 	if(argc > 1)
172 39cff6e7 2004-10-17 devnull 		goto usage;
173 39cff6e7 2004-10-17 devnull 
174 39cff6e7 2004-10-17 devnull 	if(argc == 1){
175 39cff6e7 2004-10-17 devnull 		n = strtol(argv[0], &p, 8);
176 39cff6e7 2004-10-17 devnull 		if(*p != 0 || p == argv[0])
177 39cff6e7 2004-10-17 devnull 			goto usage;
178 f7b74c17 2004-12-28 devnull 		umask(n);
179 39cff6e7 2004-10-17 devnull 		goto out;
180 39cff6e7 2004-10-17 devnull 	}
181 39cff6e7 2004-10-17 devnull 
182 39cff6e7 2004-10-17 devnull 	n = umask(0);
183 39cff6e7 2004-10-17 devnull 	umask(n);
184 39cff6e7 2004-10-17 devnull 	if(n < 0){
185 39cff6e7 2004-10-17 devnull 		fprint(mapfd(2), "umask: %r\n");
186 39cff6e7 2004-10-17 devnull 		goto out;
187 39cff6e7 2004-10-17 devnull 	}
188 39cff6e7 2004-10-17 devnull 
189 39cff6e7 2004-10-17 devnull 	fprint(mapfd(1), "umask %03o\n", n);
190 39cff6e7 2004-10-17 devnull 
191 39cff6e7 2004-10-17 devnull out:
192 39cff6e7 2004-10-17 devnull 	free(oargv);
193 39cff6e7 2004-10-17 devnull 	poplist();
194 39cff6e7 2004-10-17 devnull 	flush(err);
195 39cff6e7 2004-10-17 devnull }