1*b30d1939SAndy Fiddaman /***********************************************************************
2*b30d1939SAndy Fiddaman * *
3*b30d1939SAndy Fiddaman * This software is part of the ast package *
4*b30d1939SAndy Fiddaman * Copyright (c) 1982-2012 AT&T Intellectual Property *
5*b30d1939SAndy Fiddaman * and is licensed under the *
6*b30d1939SAndy Fiddaman * Eclipse Public License, Version 1.0 *
7*b30d1939SAndy Fiddaman * by AT&T Intellectual Property *
8*b30d1939SAndy Fiddaman * *
9*b30d1939SAndy Fiddaman * A copy of the License is available at *
10*b30d1939SAndy Fiddaman * http://www.eclipse.org/org/documents/epl-v10.html *
11*b30d1939SAndy Fiddaman * (with md5 checksum b35adb5213ca9657e911e9befb180842) *
12*b30d1939SAndy Fiddaman * *
13*b30d1939SAndy Fiddaman * Information and Software Systems Research *
14*b30d1939SAndy Fiddaman * AT&T Research *
15*b30d1939SAndy Fiddaman * Florham Park NJ *
16*b30d1939SAndy Fiddaman * *
17*b30d1939SAndy Fiddaman * David Korn <dgk@research.att.com> *
18*b30d1939SAndy Fiddaman * *
19*b30d1939SAndy Fiddaman ***********************************************************************/
20*b30d1939SAndy Fiddaman #pragma prototyped
21*b30d1939SAndy Fiddaman /*
22*b30d1939SAndy Fiddaman * ulimit [-HSacdfmnstuv] [limit]
23*b30d1939SAndy Fiddaman *
24*b30d1939SAndy Fiddaman * David Korn
25*b30d1939SAndy Fiddaman * AT&T Labs
26*b30d1939SAndy Fiddaman *
27*b30d1939SAndy Fiddaman */
28*b30d1939SAndy Fiddaman
29*b30d1939SAndy Fiddaman #include <ast.h>
30*b30d1939SAndy Fiddaman #include <sfio.h>
31*b30d1939SAndy Fiddaman #include <error.h>
32*b30d1939SAndy Fiddaman #include "defs.h"
33*b30d1939SAndy Fiddaman #include "builtins.h"
34*b30d1939SAndy Fiddaman #include "name.h"
35*b30d1939SAndy Fiddaman #include "ulimit.h"
36*b30d1939SAndy Fiddaman #ifndef SH_DICT
37*b30d1939SAndy Fiddaman # define SH_DICT "libshell"
38*b30d1939SAndy Fiddaman #endif
39*b30d1939SAndy Fiddaman
40*b30d1939SAndy Fiddaman #ifdef _no_ulimit
b_ulimit(int argc,char * argv[],Shbltin_t * context)41*b30d1939SAndy Fiddaman int b_ulimit(int argc,char *argv[],Shbltin_t *context)
42*b30d1939SAndy Fiddaman {
43*b30d1939SAndy Fiddaman NOT_USED(argc);
44*b30d1939SAndy Fiddaman NOT_USED(argv);
45*b30d1939SAndy Fiddaman NOT_USED(context);
46*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_exit(2),e_nosupport);
47*b30d1939SAndy Fiddaman return(0);
48*b30d1939SAndy Fiddaman }
49*b30d1939SAndy Fiddaman #else
50*b30d1939SAndy Fiddaman
infof(Opt_t * op,Sfio_t * sp,const char * s,Optdisc_t * dp)51*b30d1939SAndy Fiddaman static int infof(Opt_t* op, Sfio_t* sp, const char* s, Optdisc_t* dp)
52*b30d1939SAndy Fiddaman {
53*b30d1939SAndy Fiddaman register const Limit_t* tp;
54*b30d1939SAndy Fiddaman
55*b30d1939SAndy Fiddaman for (tp = shtab_limits; tp->option; tp++)
56*b30d1939SAndy Fiddaman {
57*b30d1939SAndy Fiddaman sfprintf(sp, "[%c=%d:%s?The %s", tp->option, tp - shtab_limits + 1, tp->name, tp->description);
58*b30d1939SAndy Fiddaman if(tp->type != LIM_COUNT)
59*b30d1939SAndy Fiddaman sfprintf(sp, " in %ss", e_units[tp->type]);
60*b30d1939SAndy Fiddaman sfprintf(sp, ".]");
61*b30d1939SAndy Fiddaman }
62*b30d1939SAndy Fiddaman return(1);
63*b30d1939SAndy Fiddaman }
64*b30d1939SAndy Fiddaman
65*b30d1939SAndy Fiddaman #define HARD 2
66*b30d1939SAndy Fiddaman #define SOFT 4
67*b30d1939SAndy Fiddaman
b_ulimit(int argc,char * argv[],Shbltin_t * context)68*b30d1939SAndy Fiddaman int b_ulimit(int argc,char *argv[],Shbltin_t *context)
69*b30d1939SAndy Fiddaman {
70*b30d1939SAndy Fiddaman register char *limit;
71*b30d1939SAndy Fiddaman register int mode=0, n;
72*b30d1939SAndy Fiddaman register unsigned long hit = 0;
73*b30d1939SAndy Fiddaman Shell_t *shp = context->shp;
74*b30d1939SAndy Fiddaman #ifdef _lib_getrlimit
75*b30d1939SAndy Fiddaman struct rlimit rlp;
76*b30d1939SAndy Fiddaman #endif /* _lib_getrlimit */
77*b30d1939SAndy Fiddaman const Limit_t* tp;
78*b30d1939SAndy Fiddaman char* conf;
79*b30d1939SAndy Fiddaman int label, unit, nosupport;
80*b30d1939SAndy Fiddaman rlim_t i;
81*b30d1939SAndy Fiddaman char tmp[32];
82*b30d1939SAndy Fiddaman Optdisc_t disc;
83*b30d1939SAndy Fiddaman memset(&disc, 0, sizeof(disc));
84*b30d1939SAndy Fiddaman disc.version = OPT_VERSION;
85*b30d1939SAndy Fiddaman disc.infof = infof;
86*b30d1939SAndy Fiddaman opt_info.disc = &disc;
87*b30d1939SAndy Fiddaman while((n = optget(argv,sh_optulimit))) switch(n)
88*b30d1939SAndy Fiddaman {
89*b30d1939SAndy Fiddaman case 'H':
90*b30d1939SAndy Fiddaman mode |= HARD;
91*b30d1939SAndy Fiddaman continue;
92*b30d1939SAndy Fiddaman case 'S':
93*b30d1939SAndy Fiddaman mode |= SOFT;
94*b30d1939SAndy Fiddaman continue;
95*b30d1939SAndy Fiddaman case 'a':
96*b30d1939SAndy Fiddaman hit = ~0;
97*b30d1939SAndy Fiddaman break;
98*b30d1939SAndy Fiddaman default:
99*b30d1939SAndy Fiddaman if(n < 0)
100*b30d1939SAndy Fiddaman hit |= (1L<<(-(n+1)));
101*b30d1939SAndy Fiddaman else
102*b30d1939SAndy Fiddaman errormsg(SH_DICT,2, e_notimp, opt_info.name);
103*b30d1939SAndy Fiddaman break;
104*b30d1939SAndy Fiddaman case ':':
105*b30d1939SAndy Fiddaman errormsg(SH_DICT,2, "%s", opt_info.arg);
106*b30d1939SAndy Fiddaman break;
107*b30d1939SAndy Fiddaman case '?':
108*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_usage(2), "%s", opt_info.arg);
109*b30d1939SAndy Fiddaman break;
110*b30d1939SAndy Fiddaman }
111*b30d1939SAndy Fiddaman opt_info.disc = 0;
112*b30d1939SAndy Fiddaman /* default to -f */
113*b30d1939SAndy Fiddaman limit = argv[opt_info.index];
114*b30d1939SAndy Fiddaman if(hit==0)
115*b30d1939SAndy Fiddaman for(n=0; shtab_limits[n].option; n++)
116*b30d1939SAndy Fiddaman if(shtab_limits[n].index == RLIMIT_FSIZE)
117*b30d1939SAndy Fiddaman {
118*b30d1939SAndy Fiddaman hit |= (1L<<n);
119*b30d1939SAndy Fiddaman break;
120*b30d1939SAndy Fiddaman }
121*b30d1939SAndy Fiddaman /* only one option at a time for setting */
122*b30d1939SAndy Fiddaman label = (hit&(hit-1));
123*b30d1939SAndy Fiddaman if(error_info.errors || (limit && label) || argc>opt_info.index+1)
124*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_usage(2),optusage((char*)0));
125*b30d1939SAndy Fiddaman if(mode==0)
126*b30d1939SAndy Fiddaman mode = (HARD|SOFT);
127*b30d1939SAndy Fiddaman for(tp = shtab_limits; tp->option && hit; tp++,hit>>=1)
128*b30d1939SAndy Fiddaman {
129*b30d1939SAndy Fiddaman if(!(hit&1))
130*b30d1939SAndy Fiddaman continue;
131*b30d1939SAndy Fiddaman nosupport = (n = tp->index) == RLIMIT_UNKNOWN;
132*b30d1939SAndy Fiddaman unit = shtab_units[tp->type];
133*b30d1939SAndy Fiddaman if(limit)
134*b30d1939SAndy Fiddaman {
135*b30d1939SAndy Fiddaman if(shp->subshell && !shp->subshare)
136*b30d1939SAndy Fiddaman sh_subfork();
137*b30d1939SAndy Fiddaman if(strcmp(limit,e_unlimited)==0)
138*b30d1939SAndy Fiddaman i = INFINITY;
139*b30d1939SAndy Fiddaman else
140*b30d1939SAndy Fiddaman {
141*b30d1939SAndy Fiddaman char *last;
142*b30d1939SAndy Fiddaman /* an explicit suffix unit overrides the default */
143*b30d1939SAndy Fiddaman if((i=strtol(limit,&last,0))!=INFINITY && !*last)
144*b30d1939SAndy Fiddaman i *= unit;
145*b30d1939SAndy Fiddaman else if((i=strton(limit,&last,NiL,0))==INFINITY || *last)
146*b30d1939SAndy Fiddaman {
147*b30d1939SAndy Fiddaman if((i=sh_strnum(limit,&last,2))==INFINITY || *last)
148*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_number,limit);
149*b30d1939SAndy Fiddaman i *= unit;
150*b30d1939SAndy Fiddaman }
151*b30d1939SAndy Fiddaman }
152*b30d1939SAndy Fiddaman if(nosupport)
153*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_readonly,tp->name);
154*b30d1939SAndy Fiddaman else
155*b30d1939SAndy Fiddaman {
156*b30d1939SAndy Fiddaman #ifdef _lib_getrlimit
157*b30d1939SAndy Fiddaman if(getrlimit(n,&rlp) <0)
158*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_number,limit);
159*b30d1939SAndy Fiddaman if(mode&HARD)
160*b30d1939SAndy Fiddaman rlp.rlim_max = i;
161*b30d1939SAndy Fiddaman if(mode&SOFT)
162*b30d1939SAndy Fiddaman rlp.rlim_cur = i;
163*b30d1939SAndy Fiddaman if(setrlimit(n,&rlp) <0)
164*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_overlimit,limit);
165*b30d1939SAndy Fiddaman #else
166*b30d1939SAndy Fiddaman if((i=vlimit(n,i)) < 0)
167*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_number,limit);
168*b30d1939SAndy Fiddaman #endif /* _lib_getrlimit */
169*b30d1939SAndy Fiddaman }
170*b30d1939SAndy Fiddaman }
171*b30d1939SAndy Fiddaman else
172*b30d1939SAndy Fiddaman {
173*b30d1939SAndy Fiddaman if(!nosupport)
174*b30d1939SAndy Fiddaman {
175*b30d1939SAndy Fiddaman #ifdef _lib_getrlimit
176*b30d1939SAndy Fiddaman if(getrlimit(n,&rlp) <0)
177*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_number,limit);
178*b30d1939SAndy Fiddaman if(mode&HARD)
179*b30d1939SAndy Fiddaman i = rlp.rlim_max;
180*b30d1939SAndy Fiddaman if(mode&SOFT)
181*b30d1939SAndy Fiddaman i = rlp.rlim_cur;
182*b30d1939SAndy Fiddaman #else
183*b30d1939SAndy Fiddaman # ifdef _lib_ulimit
184*b30d1939SAndy Fiddaman n--;
185*b30d1939SAndy Fiddaman # endif /* _lib_ulimit */
186*b30d1939SAndy Fiddaman i = -1;
187*b30d1939SAndy Fiddaman if((i=vlimit(n,i)) < 0)
188*b30d1939SAndy Fiddaman errormsg(SH_DICT,ERROR_system(1),e_number,limit);
189*b30d1939SAndy Fiddaman #endif /* _lib_getrlimit */
190*b30d1939SAndy Fiddaman }
191*b30d1939SAndy Fiddaman if(label)
192*b30d1939SAndy Fiddaman {
193*b30d1939SAndy Fiddaman if(tp->type != LIM_COUNT)
194*b30d1939SAndy Fiddaman sfsprintf(tmp,sizeof(tmp),"%s (%ss)", tp->description, e_units[tp->type]);
195*b30d1939SAndy Fiddaman else
196*b30d1939SAndy Fiddaman sfsprintf(tmp,sizeof(tmp),"%s", tp->name);
197*b30d1939SAndy Fiddaman sfprintf(sfstdout,"%-30s (-%c) ",tmp,tp->option);
198*b30d1939SAndy Fiddaman }
199*b30d1939SAndy Fiddaman if(nosupport)
200*b30d1939SAndy Fiddaman {
201*b30d1939SAndy Fiddaman if(!tp->conf || !*(conf = astconf(tp->conf, NiL, NiL)))
202*b30d1939SAndy Fiddaman conf = (char*)e_nosupport;
203*b30d1939SAndy Fiddaman sfputr(sfstdout,conf,'\n');
204*b30d1939SAndy Fiddaman }
205*b30d1939SAndy Fiddaman else if(i!=INFINITY)
206*b30d1939SAndy Fiddaman {
207*b30d1939SAndy Fiddaman i += (unit-1);
208*b30d1939SAndy Fiddaman sfprintf(sfstdout,"%I*d\n",sizeof(i),i/unit);
209*b30d1939SAndy Fiddaman }
210*b30d1939SAndy Fiddaman else
211*b30d1939SAndy Fiddaman sfputr(sfstdout,e_unlimited,'\n');
212*b30d1939SAndy Fiddaman }
213*b30d1939SAndy Fiddaman }
214*b30d1939SAndy Fiddaman return(0);
215*b30d1939SAndy Fiddaman }
216*b30d1939SAndy Fiddaman #endif /* _no_ulimit */
217