Blame

Date:
Message:
success on the sun
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 005a85f3 2005-01-07 devnull #ifdef RLIMIT_MEMLOCK

 21 39cff6e7 2004-10-17 devnull 	RLIMIT_MEMLOCK,

 22 005a85f3 2005-01-07 devnull #else

 23 005a85f3 2005-01-07 devnull 	0,

 24 005a85f3 2005-01-07 devnull #endif

 25 005a85f3 2005-01-07 devnull #ifdef RLIMIT_RSS

 26 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,

 27 005a85f3 2005-01-07 devnull #else

 28 005a85f3 2005-01-07 devnull 	0,

 29 005a85f3 2005-01-07 devnull #endif

 30 39cff6e7 2004-10-17 devnull 	RLIMIT_NOFILE,

 31 39cff6e7 2004-10-17 devnull 	RLIMIT_STACK,

 32 39cff6e7 2004-10-17 devnull 	RLIMIT_CPU,

 33 005a85f3 2005-01-07 devnull #ifdef RLIMIT_NPROC

 34 39cff6e7 2004-10-17 devnull 	RLIMIT_NPROC,

 35 005a85f3 2005-01-07 devnull #else

 36 005a85f3 2005-01-07 devnull 	0,

 37 005a85f3 2005-01-07 devnull #endif

 38 005a85f3 2005-01-07 devnull #ifdef RLIMIT_RSS

 39 39cff6e7 2004-10-17 devnull 	RLIMIT_RSS,

 40 005a85f3 2005-01-07 devnull #else

 41 005a85f3 2005-01-07 devnull 	0,

 42 005a85f3 2005-01-07 devnull #endif

 43 39cff6e7 2004-10-17 devnull };

 44 39cff6e7 2004-10-17 devnull 

 45 39cff6e7 2004-10-17 devnull static void

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

 47 39cff6e7 2004-10-17 devnull {

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

 49 39cff6e7 2004-10-17 devnull }

 50 39cff6e7 2004-10-17 devnull 

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

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

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

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

 55 39cff6e7 2004-10-17 devnull 

 56 39cff6e7 2004-10-17 devnull void

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

 58 39cff6e7 2004-10-17 devnull {

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

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

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

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

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

 64 39cff6e7 2004-10-17 devnull 

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

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

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

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

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

 70 39cff6e7 2004-10-17 devnull 		;

 71 39cff6e7 2004-10-17 devnull 

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

 73 39cff6e7 2004-10-17 devnull 	ARGBEGIN{

 74 39cff6e7 2004-10-17 devnull 	default:

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

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

 77 39cff6e7 2004-10-17 devnull 			return;

 78 39cff6e7 2004-10-17 devnull 		}

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

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

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

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

 83 39cff6e7 2004-10-17 devnull 		break;

 84 39cff6e7 2004-10-17 devnull 	}ARGEND

 85 39cff6e7 2004-10-17 devnull 

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

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

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

 89 39cff6e7 2004-10-17 devnull 	}

 90 39cff6e7 2004-10-17 devnull 

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

 92 39cff6e7 2004-10-17 devnull 

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

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

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

 96 39cff6e7 2004-10-17 devnull 		;

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

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

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

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

101 39cff6e7 2004-10-17 devnull 

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

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

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

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

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

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

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

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

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

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

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

113 39cff6e7 2004-10-17 devnull 		}

114 39cff6e7 2004-10-17 devnull 	}

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

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

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

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

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

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

121 39cff6e7 2004-10-17 devnull 			else

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

123 39cff6e7 2004-10-17 devnull 		}

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

125 39cff6e7 2004-10-17 devnull 	}

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

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

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

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

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

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

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

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

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

135 39cff6e7 2004-10-17 devnull 				else

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

137 39cff6e7 2004-10-17 devnull 				break;

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

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

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

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

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

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

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

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

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

147 39cff6e7 2004-10-17 devnull 			default:

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

149 39cff6e7 2004-10-17 devnull 			set:

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

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

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

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

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

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

156 39cff6e7 2004-10-17 devnull 			}

157 39cff6e7 2004-10-17 devnull 		}

158 39cff6e7 2004-10-17 devnull 	}

159 39cff6e7 2004-10-17 devnull 

160 39cff6e7 2004-10-17 devnull out:

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

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

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

164 39cff6e7 2004-10-17 devnull }

165 39cff6e7 2004-10-17 devnull 

166 39cff6e7 2004-10-17 devnull void

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

168 39cff6e7 2004-10-17 devnull {

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

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

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

172 39cff6e7 2004-10-17 devnull 

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

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

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

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

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

178 39cff6e7 2004-10-17 devnull 		;

179 39cff6e7 2004-10-17 devnull 

180 39cff6e7 2004-10-17 devnull 	ARGBEGIN{

181 39cff6e7 2004-10-17 devnull 	default:

182 39cff6e7 2004-10-17 devnull 	usage:

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

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

185 39cff6e7 2004-10-17 devnull 	}ARGEND

186 39cff6e7 2004-10-17 devnull 

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

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

189 39cff6e7 2004-10-17 devnull 

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

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

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

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

194 f7b74c17 2004-12-28 devnull 		umask(n);

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

196 39cff6e7 2004-10-17 devnull 	}

197 39cff6e7 2004-10-17 devnull 

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

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

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

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

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

203 39cff6e7 2004-10-17 devnull 	}

204 39cff6e7 2004-10-17 devnull 

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

206 39cff6e7 2004-10-17 devnull 

207 39cff6e7 2004-10-17 devnull out:

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

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

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

211 39cff6e7 2004-10-17 devnull }