Blame

Date:
Message:
add ulimit and umask as builtins
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 39cff6e7 2004-10-17 devnull #include <sys/resource.h>

  4 39cff6e7 2004-10-17 devnull #include <libc.h>

  5 39cff6e7 2004-10-17 devnull #include "rc.h"

  6 39cff6e7 2004-10-17 devnull #include "exec.h"

  7 39cff6e7 2004-10-17 devnull #include "io.h"

  8 39cff6e7 2004-10-17 devnull #include "fns.h"

  9 39cff6e7 2004-10-17 devnull #include "getflags.h"

 10 39cff6e7 2004-10-17 devnull 

 11 39cff6e7 2004-10-17 devnull extern char **mkargv(word*);

 12 39cff6e7 2004-10-17 devnull extern int mapfd(int);

 13 39cff6e7 2004-10-17 devnull 

 14 39cff6e7 2004-10-17 devnull static char *eargs = "cdflmnstuv";

 15 39cff6e7 2004-10-17 devnull static int rlx[] = {

 16 39cff6e7 2004-10-17 devnull 	RLIMIT_CORE,

 17 39cff6e7 2004-10-17 devnull 	RLIMIT_DATA,

 18 39cff6e7 2004-10-17 devnull 	RLIMIT_FSIZE,

 19 39cff6e7 2004-10-17 devnull 	RLIMIT_MEMLOCK,

 20 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,

 21 39cff6e7 2004-10-17 devnull 	RLIMIT_NOFILE,

 22 39cff6e7 2004-10-17 devnull 	RLIMIT_STACK,

 23 39cff6e7 2004-10-17 devnull 	RLIMIT_CPU,

 24 39cff6e7 2004-10-17 devnull 	RLIMIT_NPROC,

 25 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,

 26 39cff6e7 2004-10-17 devnull };

 27 39cff6e7 2004-10-17 devnull 

 28 39cff6e7 2004-10-17 devnull static void

 29 39cff6e7 2004-10-17 devnull eusage(void)

 30 39cff6e7 2004-10-17 devnull {

 31 39cff6e7 2004-10-17 devnull 	fprint(mapfd(2), "usage: ulimit [-SHa%s [limit]]\n", eargs);

 32 39cff6e7 2004-10-17 devnull }

 33 39cff6e7 2004-10-17 devnull 

 34 39cff6e7 2004-10-17 devnull #define Notset -4

 35 39cff6e7 2004-10-17 devnull #define Unlimited -3

 36 39cff6e7 2004-10-17 devnull #define Hard -2

 37 39cff6e7 2004-10-17 devnull #define Soft -1

 38 39cff6e7 2004-10-17 devnull 

 39 39cff6e7 2004-10-17 devnull void

 40 39cff6e7 2004-10-17 devnull execulimit(void)

 41 39cff6e7 2004-10-17 devnull {

 42 39cff6e7 2004-10-17 devnull 	int fd, n, argc, sethard, setsoft, limit;

 43 39cff6e7 2004-10-17 devnull 	int flag[256];

 44 39cff6e7 2004-10-17 devnull 	char **argv, **oargv, *p;

 45 39cff6e7 2004-10-17 devnull 	char *argv0;

 46 39cff6e7 2004-10-17 devnull 	struct rlimit rl;

 47 39cff6e7 2004-10-17 devnull 

 48 39cff6e7 2004-10-17 devnull 	argv0 = nil;

 49 39cff6e7 2004-10-17 devnull 	setstatus("");

 50 39cff6e7 2004-10-17 devnull 	oargv = mkargv(runq->argv->words);

 51 39cff6e7 2004-10-17 devnull 	argv = oargv+1;

 52 39cff6e7 2004-10-17 devnull 	for(argc=0; argv[argc]; argc++)

 53 39cff6e7 2004-10-17 devnull 		;

 54 39cff6e7 2004-10-17 devnull 

 55 39cff6e7 2004-10-17 devnull 	memset(flag, 0, sizeof flag);

 56 39cff6e7 2004-10-17 devnull 	ARGBEGIN{

 57 39cff6e7 2004-10-17 devnull 	default:

 58 39cff6e7 2004-10-17 devnull 		if(strchr(eargs, ARGC()) == nil){

 59 39cff6e7 2004-10-17 devnull 			eusage();

 60 39cff6e7 2004-10-17 devnull 			return;

 61 39cff6e7 2004-10-17 devnull 		}

 62 39cff6e7 2004-10-17 devnull 	case 'S':

 63 39cff6e7 2004-10-17 devnull 	case 'H':

 64 39cff6e7 2004-10-17 devnull 	case 'a':

 65 39cff6e7 2004-10-17 devnull 		flag[ARGC()] = 1;

 66 39cff6e7 2004-10-17 devnull 		break;

 67 39cff6e7 2004-10-17 devnull 	}ARGEND

 68 39cff6e7 2004-10-17 devnull 

 69 39cff6e7 2004-10-17 devnull 	if(argc > 1){

 70 39cff6e7 2004-10-17 devnull 		eusage();

 71 39cff6e7 2004-10-17 devnull 		goto out;

 72 39cff6e7 2004-10-17 devnull 	}

 73 39cff6e7 2004-10-17 devnull 

 74 39cff6e7 2004-10-17 devnull 	fd = mapfd(1);

 75 39cff6e7 2004-10-17 devnull 

 76 39cff6e7 2004-10-17 devnull 	sethard = 1;

 77 39cff6e7 2004-10-17 devnull 	setsoft = 1;

 78 39cff6e7 2004-10-17 devnull 	if(flag['S'] && flag['H'])

 79 39cff6e7 2004-10-17 devnull 		;

 80 39cff6e7 2004-10-17 devnull 	else if(flag['S'])

 81 39cff6e7 2004-10-17 devnull 		sethard = 0;

 82 39cff6e7 2004-10-17 devnull 	else if(flag['H'])

 83 39cff6e7 2004-10-17 devnull 		setsoft = 0;

 84 39cff6e7 2004-10-17 devnull 

 85 39cff6e7 2004-10-17 devnull 	limit = Notset;

 86 39cff6e7 2004-10-17 devnull 	if(argc>0){

 87 39cff6e7 2004-10-17 devnull 		if(strcmp(argv[0], "unlimited") == 0)

 88 39cff6e7 2004-10-17 devnull 			limit = Unlimited;

 89 39cff6e7 2004-10-17 devnull 		else if(strcmp(argv[0], "hard") == 0)

 90 39cff6e7 2004-10-17 devnull 			limit = Hard;

 91 39cff6e7 2004-10-17 devnull 		else if(strcmp(argv[0], "soft") == 0)

 92 39cff6e7 2004-10-17 devnull 			limit = Soft;

 93 39cff6e7 2004-10-17 devnull 		else if((limit = strtol(argv[0], &p, 0)) < 0 || *p != 0){

 94 39cff6e7 2004-10-17 devnull 			eusage();

 95 39cff6e7 2004-10-17 devnull 			goto out;

 96 39cff6e7 2004-10-17 devnull 		}

 97 39cff6e7 2004-10-17 devnull 	}

 98 39cff6e7 2004-10-17 devnull 	if(flag['a']){

 99 39cff6e7 2004-10-17 devnull 		for(p=eargs; *p; p++){

100 39cff6e7 2004-10-17 devnull 			getrlimit(rlx[p-eargs], &rl);

101 39cff6e7 2004-10-17 devnull 			n = flag['H'] ? rl.rlim_max : rl.rlim_cur;

102 39cff6e7 2004-10-17 devnull 			if(n == -1)

103 39cff6e7 2004-10-17 devnull 				fprint(fd, "ulimit -%c unlimited\n", *p);

104 39cff6e7 2004-10-17 devnull 			else

105 39cff6e7 2004-10-17 devnull 				fprint(fd, "ulimit -%c %d\n", *p, n);

106 39cff6e7 2004-10-17 devnull 		}

107 39cff6e7 2004-10-17 devnull 		goto out;

108 39cff6e7 2004-10-17 devnull 	}

109 39cff6e7 2004-10-17 devnull 	for(p=eargs; *p; p++){

110 39cff6e7 2004-10-17 devnull 		if(flag[(uchar)*p]){

111 39cff6e7 2004-10-17 devnull 			n = 0;

112 39cff6e7 2004-10-17 devnull 			getrlimit(rlx[p-eargs], &rl);

113 39cff6e7 2004-10-17 devnull 			switch(limit){

114 39cff6e7 2004-10-17 devnull 			case Notset:

115 39cff6e7 2004-10-17 devnull 				n = flag['H'] ? rl.rlim_max : rl.rlim_cur;

116 39cff6e7 2004-10-17 devnull 				if(n == -1)

117 39cff6e7 2004-10-17 devnull 					fprint(fd, "ulimit -%c unlimited\n", *p);

118 39cff6e7 2004-10-17 devnull 				else

119 39cff6e7 2004-10-17 devnull 					fprint(fd, "ulimit -%c %d\n", *p, n);

120 39cff6e7 2004-10-17 devnull 				break;

121 39cff6e7 2004-10-17 devnull 			case Hard:

122 39cff6e7 2004-10-17 devnull 				n = rl.rlim_max;

123 39cff6e7 2004-10-17 devnull 				goto set;

124 39cff6e7 2004-10-17 devnull 			case Soft:

125 39cff6e7 2004-10-17 devnull 				n = rl.rlim_cur;

126 39cff6e7 2004-10-17 devnull 				goto set;

127 39cff6e7 2004-10-17 devnull 			case Unlimited:

128 39cff6e7 2004-10-17 devnull 				n = -1;

129 39cff6e7 2004-10-17 devnull 				goto set;

130 39cff6e7 2004-10-17 devnull 			default:

131 39cff6e7 2004-10-17 devnull 				n = limit;

132 39cff6e7 2004-10-17 devnull 			set:

133 39cff6e7 2004-10-17 devnull 				if(setsoft)

134 39cff6e7 2004-10-17 devnull 					rl.rlim_cur = n;

135 39cff6e7 2004-10-17 devnull 				if(sethard)

136 39cff6e7 2004-10-17 devnull 					rl.rlim_max = n;

137 39cff6e7 2004-10-17 devnull 				if(setrlimit(rlx[p-eargs], &rl) < 0)

138 39cff6e7 2004-10-17 devnull 					fprint(mapfd(2), "setrlimit: %r\n");

139 39cff6e7 2004-10-17 devnull 			}

140 39cff6e7 2004-10-17 devnull 		}

141 39cff6e7 2004-10-17 devnull 	}

142 39cff6e7 2004-10-17 devnull 

143 39cff6e7 2004-10-17 devnull out:

144 39cff6e7 2004-10-17 devnull 	free(oargv);

145 39cff6e7 2004-10-17 devnull 	poplist();

146 39cff6e7 2004-10-17 devnull 	flush(err);

147 39cff6e7 2004-10-17 devnull }

148 39cff6e7 2004-10-17 devnull 

149 39cff6e7 2004-10-17 devnull void

150 39cff6e7 2004-10-17 devnull execumask(void)

151 39cff6e7 2004-10-17 devnull {

152 39cff6e7 2004-10-17 devnull 	int n, argc;

153 39cff6e7 2004-10-17 devnull 	char **argv, **oargv, *p;

154 39cff6e7 2004-10-17 devnull 	char *argv0;

155 39cff6e7 2004-10-17 devnull 

156 39cff6e7 2004-10-17 devnull 	argv0 = nil;

157 39cff6e7 2004-10-17 devnull 	setstatus("");

158 39cff6e7 2004-10-17 devnull 	oargv = mkargv(runq->argv->words);

159 39cff6e7 2004-10-17 devnull 	argv = oargv+1;

160 39cff6e7 2004-10-17 devnull 	for(argc=0; argv[argc]; argc++)

161 39cff6e7 2004-10-17 devnull 		;

162 39cff6e7 2004-10-17 devnull 

163 39cff6e7 2004-10-17 devnull 	ARGBEGIN{

164 39cff6e7 2004-10-17 devnull 	default:

165 39cff6e7 2004-10-17 devnull 	usage:

166 39cff6e7 2004-10-17 devnull 		fprint(mapfd(2), "usage: umask [mode]\n");

167 39cff6e7 2004-10-17 devnull 		goto out;

168 39cff6e7 2004-10-17 devnull 	}ARGEND

169 39cff6e7 2004-10-17 devnull 

170 39cff6e7 2004-10-17 devnull 	if(argc > 1)

171 39cff6e7 2004-10-17 devnull 		goto usage;

172 39cff6e7 2004-10-17 devnull 

173 39cff6e7 2004-10-17 devnull 	if(argc == 1){

174 39cff6e7 2004-10-17 devnull 		n = strtol(argv[0], &p, 8);

175 39cff6e7 2004-10-17 devnull 		if(*p != 0 || p == argv[0])

176 39cff6e7 2004-10-17 devnull 			goto usage;

177 39cff6e7 2004-10-17 devnull 		if(umask(n) < 0){

178 39cff6e7 2004-10-17 devnull 			fprint(mapfd(2), "umask: %r\n");

179 39cff6e7 2004-10-17 devnull 			goto out;

180 39cff6e7 2004-10-17 devnull 		}

181 39cff6e7 2004-10-17 devnull 		goto out;

182 39cff6e7 2004-10-17 devnull 	}

183 39cff6e7 2004-10-17 devnull 

184 39cff6e7 2004-10-17 devnull 	n = umask(0);

185 39cff6e7 2004-10-17 devnull 	umask(n);

186 39cff6e7 2004-10-17 devnull 	if(n < 0){

187 39cff6e7 2004-10-17 devnull 		fprint(mapfd(2), "umask: %r\n");

188 39cff6e7 2004-10-17 devnull 		goto out;

189 39cff6e7 2004-10-17 devnull 	}

190 39cff6e7 2004-10-17 devnull 

191 39cff6e7 2004-10-17 devnull 	fprint(mapfd(1), "umask %03o\n", n);

192 39cff6e7 2004-10-17 devnull 

193 39cff6e7 2004-10-17 devnull out:

194 39cff6e7 2004-10-17 devnull 	free(oargv);

195 39cff6e7 2004-10-17 devnull 	poplist();

196 39cff6e7 2004-10-17 devnull 	flush(err);

197 39cff6e7 2004-10-17 devnull }