misc.c revision 1.22 1 1.22 kamil /* $NetBSD: misc.c,v 1.22 2017/06/22 23:38:49 kamil Exp $ */
2 1.2 tls
3 1.1 jtc /*
4 1.1 jtc * Miscellaneous functions
5 1.1 jtc */
6 1.8 agc #include <sys/cdefs.h>
7 1.8 agc
8 1.8 agc #ifndef lint
9 1.22 kamil __RCSID("$NetBSD: misc.c,v 1.22 2017/06/22 23:38:49 kamil Exp $");
10 1.8 agc #endif
11 1.8 agc
12 1.1 jtc
13 1.1 jtc #include "sh.h"
14 1.1 jtc #include <ctype.h> /* for FILECHCONV */
15 1.22 kamil #include <limits.h>
16 1.1 jtc
17 1.1 jtc #ifndef UCHAR_MAX
18 1.1 jtc # define UCHAR_MAX 0xFF
19 1.1 jtc #endif
20 1.1 jtc
21 1.1 jtc short ctypes [UCHAR_MAX+1]; /* type bits for unsigned char */
22 1.1 jtc
23 1.1 jtc static int do_gmatch ARGS((const unsigned char *s, const unsigned char *p,
24 1.1 jtc const unsigned char *se, const unsigned char *pe,
25 1.1 jtc int isfile));
26 1.1 jtc static const unsigned char *cclass ARGS((const unsigned char *p, int sub));
27 1.1 jtc
28 1.1 jtc /*
29 1.1 jtc * Fast character classes
30 1.1 jtc */
31 1.1 jtc void
32 1.1 jtc setctypes(s, t)
33 1.1 jtc register const char *s;
34 1.1 jtc register int t;
35 1.1 jtc {
36 1.1 jtc register int i;
37 1.1 jtc
38 1.1 jtc if (t & C_IFS) {
39 1.1 jtc for (i = 0; i < UCHAR_MAX+1; i++)
40 1.1 jtc ctypes[i] &= ~C_IFS;
41 1.1 jtc ctypes[0] |= C_IFS; /* include \0 in C_IFS */
42 1.1 jtc }
43 1.1 jtc while (*s != 0)
44 1.1 jtc ctypes[(unsigned char) *s++] |= t;
45 1.1 jtc }
46 1.1 jtc
47 1.1 jtc void
48 1.1 jtc initctypes()
49 1.1 jtc {
50 1.1 jtc register int c;
51 1.1 jtc
52 1.1 jtc for (c = 'a'; c <= 'z'; c++)
53 1.1 jtc ctypes[c] |= C_ALPHA;
54 1.1 jtc for (c = 'A'; c <= 'Z'; c++)
55 1.1 jtc ctypes[c] |= C_ALPHA;
56 1.1 jtc ctypes['_'] |= C_ALPHA;
57 1.1 jtc setctypes("0123456789", C_DIGIT);
58 1.1 jtc setctypes(" \t\n|&;<>()", C_LEX1); /* \0 added automatically */
59 1.1 jtc setctypes("*@#!$-?", C_VAR1);
60 1.1 jtc setctypes(" \t\n", C_IFSWS);
61 1.1 jtc setctypes("=-+?", C_SUBOP1);
62 1.1 jtc setctypes("#%", C_SUBOP2);
63 1.1 jtc setctypes(" \n\t\"#$&'()*;<>?[\\`|", C_QUOTE);
64 1.1 jtc }
65 1.1 jtc
66 1.1 jtc /* convert unsigned long to base N string */
67 1.1 jtc
68 1.1 jtc char *
69 1.1 jtc ulton(n, base)
70 1.1 jtc register unsigned long n;
71 1.1 jtc int base;
72 1.1 jtc {
73 1.1 jtc register char *p;
74 1.1 jtc static char buf [20];
75 1.1 jtc
76 1.1 jtc p = &buf[sizeof(buf)];
77 1.1 jtc *--p = '\0';
78 1.1 jtc do {
79 1.1 jtc *--p = "0123456789ABCDEF"[n%base];
80 1.1 jtc n /= base;
81 1.1 jtc } while (n != 0);
82 1.1 jtc return p;
83 1.1 jtc }
84 1.1 jtc
85 1.1 jtc char *
86 1.1 jtc str_save(s, ap)
87 1.1 jtc register const char *s;
88 1.1 jtc Area *ap;
89 1.1 jtc {
90 1.9 mycroft size_t len;
91 1.9 mycroft char *p;
92 1.9 mycroft
93 1.9 mycroft if (!s)
94 1.9 mycroft return NULL;
95 1.9 mycroft len = strlen(s)+1;
96 1.9 mycroft p = alloc(len, ap);
97 1.14 seanb strlcpy(p, s, len);
98 1.9 mycroft return (p);
99 1.1 jtc }
100 1.1 jtc
101 1.1 jtc /* Allocate a string of size n+1 and copy upto n characters from the possibly
102 1.1 jtc * null terminated string s into it. Always returns a null terminated string
103 1.1 jtc * (unless n < 0).
104 1.1 jtc */
105 1.1 jtc char *
106 1.1 jtc str_nsave(s, n, ap)
107 1.1 jtc register const char *s;
108 1.1 jtc int n;
109 1.1 jtc Area *ap;
110 1.1 jtc {
111 1.1 jtc char *ns;
112 1.1 jtc
113 1.1 jtc if (n < 0)
114 1.1 jtc return 0;
115 1.1 jtc ns = alloc(n + 1, ap);
116 1.1 jtc ns[0] = '\0';
117 1.1 jtc return strncat(ns, s, n);
118 1.1 jtc }
119 1.1 jtc
120 1.1 jtc /* called from expand.h:XcheckN() to grow buffer */
121 1.1 jtc char *
122 1.1 jtc Xcheck_grow_(xsp, xp, more)
123 1.1 jtc XString *xsp;
124 1.1 jtc char *xp;
125 1.1 jtc int more;
126 1.1 jtc {
127 1.1 jtc char *old_beg = xsp->beg;
128 1.1 jtc
129 1.13 lukem xsp->len += (size_t)more > xsp->len ? (size_t)more : xsp->len;
130 1.1 jtc xsp->beg = aresize(xsp->beg, xsp->len + 8, xsp->areap);
131 1.1 jtc xsp->end = xsp->beg + xsp->len;
132 1.1 jtc return xsp->beg + (xp - old_beg);
133 1.1 jtc }
134 1.1 jtc
135 1.12 christos const struct option goptions[] = {
136 1.1 jtc /* Special cases (see parse_args()): -A, -o, -s.
137 1.1 jtc * Options are sorted by their longnames - the order of these
138 1.1 jtc * entries MUST match the order of sh_flag F* enumerations in sh.h.
139 1.1 jtc */
140 1.1 jtc { "allexport", 'a', OF_ANY },
141 1.1 jtc #ifdef BRACE_EXPAND
142 1.1 jtc { "braceexpand", 0, OF_ANY }, /* non-standard */
143 1.1 jtc #endif
144 1.1 jtc { "bgnice", 0, OF_ANY },
145 1.4 hubertf { (char *) 0, 'c', OF_CMDLINE },
146 1.1 jtc #ifdef EMACS
147 1.1 jtc { "emacs", 0, OF_ANY },
148 1.9 mycroft { "emacs-usemeta", 0, OF_ANY }, /* non-standard */
149 1.1 jtc #endif
150 1.1 jtc { "errexit", 'e', OF_ANY },
151 1.1 jtc #ifdef EMACS
152 1.1 jtc { "gmacs", 0, OF_ANY },
153 1.1 jtc #endif
154 1.1 jtc { "ignoreeof", 0, OF_ANY },
155 1.1 jtc { "interactive",'i', OF_CMDLINE },
156 1.1 jtc { "keyword", 'k', OF_ANY },
157 1.1 jtc { "login", 'l', OF_CMDLINE },
158 1.1 jtc { "markdirs", 'X', OF_ANY },
159 1.1 jtc #ifdef JOBS
160 1.1 jtc { "monitor", 'm', OF_ANY },
161 1.1 jtc #else /* JOBS */
162 1.4 hubertf { (char *) 0, 'm', 0 }, /* so FMONITOR not ifdef'd */
163 1.1 jtc #endif /* JOBS */
164 1.1 jtc { "noclobber", 'C', OF_ANY },
165 1.1 jtc { "noexec", 'n', OF_ANY },
166 1.1 jtc { "noglob", 'f', OF_ANY },
167 1.1 jtc { "nohup", 0, OF_ANY },
168 1.1 jtc { "nolog", 0, OF_ANY }, /* no effect */
169 1.1 jtc #ifdef JOBS
170 1.1 jtc { "notify", 'b', OF_ANY },
171 1.1 jtc #endif /* JOBS */
172 1.1 jtc { "nounset", 'u', OF_ANY },
173 1.1 jtc { "physical", 0, OF_ANY }, /* non-standard */
174 1.1 jtc { "posix", 0, OF_ANY }, /* non-standard */
175 1.1 jtc { "privileged", 'p', OF_ANY },
176 1.1 jtc { "restricted", 'r', OF_CMDLINE },
177 1.1 jtc { "stdin", 's', OF_CMDLINE }, /* pseudo non-standard */
178 1.1 jtc { "trackall", 'h', OF_ANY },
179 1.1 jtc { "verbose", 'v', OF_ANY },
180 1.1 jtc #ifdef VI
181 1.1 jtc { "vi", 0, OF_ANY },
182 1.1 jtc { "viraw", 0, OF_ANY }, /* no effect */
183 1.1 jtc { "vi-show8", 0, OF_ANY }, /* non-standard */
184 1.1 jtc { "vi-tabcomplete", 0, OF_ANY }, /* non-standard */
185 1.1 jtc { "vi-esccomplete", 0, OF_ANY }, /* non-standard */
186 1.1 jtc #endif
187 1.1 jtc { "xtrace", 'x', OF_ANY },
188 1.4 hubertf /* Anonymous flags: used internally by shell only
189 1.9 mycroft * (not visible to user)
190 1.4 hubertf */
191 1.4 hubertf { (char *) 0, 0, OF_INTERNAL }, /* FTALKING_I */
192 1.1 jtc };
193 1.1 jtc
194 1.1 jtc /*
195 1.1 jtc * translate -o option into F* constant (also used for test -o option)
196 1.1 jtc */
197 1.1 jtc int
198 1.1 jtc option(n)
199 1.1 jtc const char *n;
200 1.1 jtc {
201 1.1 jtc int i;
202 1.1 jtc
203 1.13 lukem for (i = 0; i < (int)NELEM(goptions); i++)
204 1.12 christos if (goptions[i].name && strcmp(goptions[i].name, n) == 0)
205 1.1 jtc return i;
206 1.1 jtc
207 1.1 jtc return -1;
208 1.1 jtc }
209 1.1 jtc
210 1.1 jtc struct options_info {
211 1.1 jtc int opt_width;
212 1.1 jtc struct {
213 1.1 jtc const char *name;
214 1.1 jtc int flag;
215 1.12 christos } opts[NELEM(goptions)];
216 1.1 jtc };
217 1.1 jtc
218 1.1 jtc static char *options_fmt_entry ARGS((void *arg, int i, char *buf, int buflen));
219 1.1 jtc static void printoptions ARGS((int verbose));
220 1.1 jtc
221 1.1 jtc /* format a single select menu item */
222 1.1 jtc static char *
223 1.1 jtc options_fmt_entry(arg, i, buf, buflen)
224 1.1 jtc void *arg;
225 1.1 jtc int i;
226 1.1 jtc char *buf;
227 1.1 jtc int buflen;
228 1.1 jtc {
229 1.1 jtc struct options_info *oi = (struct options_info *) arg;
230 1.1 jtc
231 1.1 jtc shf_snprintf(buf, buflen, "%-*s %s",
232 1.1 jtc oi->opt_width, oi->opts[i].name,
233 1.1 jtc Flag(oi->opts[i].flag) ? "on" : "off");
234 1.1 jtc return buf;
235 1.1 jtc }
236 1.1 jtc
237 1.1 jtc static void
238 1.1 jtc printoptions(verbose)
239 1.1 jtc int verbose;
240 1.1 jtc {
241 1.1 jtc int i;
242 1.1 jtc
243 1.1 jtc if (verbose) {
244 1.1 jtc struct options_info oi;
245 1.1 jtc int n, len;
246 1.1 jtc
247 1.1 jtc /* verbose version */
248 1.1 jtc shprintf("Current option settings\n");
249 1.1 jtc
250 1.13 lukem for (i = n = oi.opt_width = 0; i < (int)NELEM(goptions); i++)
251 1.12 christos if (goptions[i].name) {
252 1.12 christos len = strlen(goptions[i].name);
253 1.12 christos oi.opts[n].name = goptions[i].name;
254 1.1 jtc oi.opts[n++].flag = i;
255 1.1 jtc if (len > oi.opt_width)
256 1.1 jtc oi.opt_width = len;
257 1.1 jtc }
258 1.1 jtc print_columns(shl_stdout, n, options_fmt_entry, &oi,
259 1.6 provos oi.opt_width + 5, 1);
260 1.1 jtc } else {
261 1.1 jtc /* short version ala ksh93 */
262 1.1 jtc shprintf("set");
263 1.13 lukem for (i = 0; i < (int)NELEM(goptions); i++)
264 1.12 christos if (Flag(i) && goptions[i].name)
265 1.12 christos shprintf(" -o %s", goptions[i].name);
266 1.15 joerg shprintf("%s", newline);
267 1.1 jtc }
268 1.1 jtc }
269 1.1 jtc
270 1.1 jtc char *
271 1.1 jtc getoptions()
272 1.1 jtc {
273 1.13 lukem size_t i;
274 1.4 hubertf char m[(int) FNFLAGS + 1];
275 1.1 jtc register char *cp = m;
276 1.1 jtc
277 1.12 christos for (i = 0; i < NELEM(goptions); i++)
278 1.12 christos if (goptions[i].c && Flag(i))
279 1.12 christos *cp++ = goptions[i].c;
280 1.1 jtc *cp = 0;
281 1.1 jtc return str_save(m, ATEMP);
282 1.1 jtc }
283 1.1 jtc
284 1.1 jtc /* change a Flag(*) value; takes care of special actions */
285 1.1 jtc void
286 1.1 jtc change_flag(f, what, newval)
287 1.1 jtc enum sh_flag f; /* flag to change */
288 1.1 jtc int what; /* what is changing the flag (command line vs set) */
289 1.1 jtc int newval;
290 1.1 jtc {
291 1.1 jtc int oldval;
292 1.1 jtc
293 1.1 jtc oldval = Flag(f);
294 1.1 jtc Flag(f) = newval;
295 1.1 jtc #ifdef JOBS
296 1.1 jtc if (f == FMONITOR) {
297 1.1 jtc if (what != OF_CMDLINE && newval != oldval)
298 1.1 jtc j_change();
299 1.1 jtc } else
300 1.1 jtc #endif /* JOBS */
301 1.1 jtc #ifdef EDIT
302 1.1 jtc if (0
303 1.1 jtc # ifdef VI
304 1.1 jtc || f == FVI
305 1.1 jtc # endif /* VI */
306 1.1 jtc # ifdef EMACS
307 1.1 jtc || f == FEMACS || f == FGMACS
308 1.1 jtc # endif /* EMACS */
309 1.1 jtc )
310 1.1 jtc {
311 1.1 jtc if (newval) {
312 1.1 jtc # ifdef VI
313 1.1 jtc Flag(FVI) = 0;
314 1.1 jtc # endif /* VI */
315 1.1 jtc # ifdef EMACS
316 1.1 jtc Flag(FEMACS) = Flag(FGMACS) = 0;
317 1.1 jtc # endif /* EMACS */
318 1.1 jtc Flag(f) = newval;
319 1.1 jtc }
320 1.1 jtc } else
321 1.1 jtc #endif /* EDIT */
322 1.1 jtc /* Turning off -p? */
323 1.1 jtc if (f == FPRIVILEGED && oldval && !newval) {
324 1.9 mycroft seteuid(ksheuid = getuid());
325 1.9 mycroft setuid(ksheuid);
326 1.9 mycroft setegid(getgid());
327 1.1 jtc setgid(getgid());
328 1.1 jtc } else if (f == FPOSIX && newval) {
329 1.1 jtc #ifdef BRACE_EXPAND
330 1.1 jtc Flag(FBRACEEXPAND) = 0
331 1.1 jtc #endif /* BRACE_EXPAND */
332 1.1 jtc ;
333 1.1 jtc }
334 1.4 hubertf /* Changing interactive flag? */
335 1.4 hubertf if (f == FTALKING) {
336 1.4 hubertf if ((what == OF_CMDLINE || what == OF_SET) && procpid == kshpid)
337 1.4 hubertf Flag(FTALKING_I) = newval;
338 1.4 hubertf }
339 1.1 jtc }
340 1.1 jtc
341 1.1 jtc /* parse command line & set command arguments. returns the index of
342 1.1 jtc * non-option arguments, -1 if there is an error.
343 1.1 jtc */
344 1.1 jtc int
345 1.1 jtc parse_args(argv, what, setargsp)
346 1.1 jtc char **argv;
347 1.1 jtc int what; /* OF_CMDLINE or OF_SET */
348 1.1 jtc int *setargsp;
349 1.1 jtc {
350 1.12 christos static char cmd_opts[NELEM(goptions) + 3]; /* o:\0 */
351 1.12 christos static char set_opts[NELEM(goptions) + 5]; /* Ao;s\0 */
352 1.1 jtc char *opts;
353 1.4 hubertf char *array = (char *) 0;
354 1.1 jtc Getopt go;
355 1.1 jtc int i, optc, set, sortargs = 0, arrayset = 0;
356 1.1 jtc
357 1.1 jtc /* First call? Build option strings... */
358 1.1 jtc if (cmd_opts[0] == '\0') {
359 1.4 hubertf char *p, *q;
360 1.1 jtc
361 1.9 mycroft /* see cmd_opts[] declaration */
362 1.9 mycroft strlcpy(cmd_opts, "o:", sizeof cmd_opts);
363 1.1 jtc p = cmd_opts + strlen(cmd_opts);
364 1.9 mycroft /* see set_opts[] declaration */
365 1.9 mycroft strlcpy(set_opts, "A:o;s", sizeof set_opts);
366 1.4 hubertf q = set_opts + strlen(set_opts);
367 1.13 lukem for (i = 0; i < (int)NELEM(goptions); i++) {
368 1.12 christos if (goptions[i].c) {
369 1.12 christos if (goptions[i].flags & OF_CMDLINE)
370 1.12 christos *p++ = goptions[i].c;
371 1.12 christos if (goptions[i].flags & OF_SET)
372 1.12 christos *q++ = goptions[i].c;
373 1.4 hubertf }
374 1.4 hubertf }
375 1.1 jtc *p = '\0';
376 1.4 hubertf *q = '\0';
377 1.1 jtc }
378 1.1 jtc
379 1.1 jtc if (what == OF_CMDLINE) {
380 1.1 jtc char *p;
381 1.1 jtc /* Set FLOGIN before parsing options so user can clear
382 1.1 jtc * flag using +l.
383 1.1 jtc */
384 1.1 jtc Flag(FLOGIN) = (argv[0][0] == '-'
385 1.1 jtc || ((p = ksh_strrchr_dirsep(argv[0]))
386 1.1 jtc && *++p == '-'));
387 1.1 jtc opts = cmd_opts;
388 1.1 jtc } else
389 1.1 jtc opts = set_opts;
390 1.1 jtc ksh_getopt_reset(&go, GF_ERROR|GF_PLUSOPT);
391 1.1 jtc while ((optc = ksh_getopt(argv, &go, opts)) != EOF) {
392 1.1 jtc set = (go.info & GI_PLUS) ? 0 : 1;
393 1.1 jtc switch (optc) {
394 1.1 jtc case 'A':
395 1.1 jtc arrayset = set ? 1 : -1;
396 1.4 hubertf array = go.optarg;
397 1.1 jtc break;
398 1.1 jtc
399 1.1 jtc case 'o':
400 1.1 jtc if (go.optarg == (char *) 0) {
401 1.1 jtc /* lone -o: print options
402 1.1 jtc *
403 1.1 jtc * Note that on the command line, -o requires
404 1.1 jtc * an option (ie, can't get here if what is
405 1.1 jtc * OF_CMDLINE).
406 1.1 jtc */
407 1.1 jtc printoptions(set);
408 1.1 jtc break;
409 1.1 jtc }
410 1.1 jtc i = option(go.optarg);
411 1.1 jtc if (i >= 0 && set == Flag(i))
412 1.1 jtc /* Don't check the context if the flag
413 1.1 jtc * isn't changing - makes "set -o interactive"
414 1.1 jtc * work if you're already interactive. Needed
415 1.1 jtc * if the output of "set +o" is to be used.
416 1.1 jtc */
417 1.1 jtc ;
418 1.12 christos else if (i >= 0 && (goptions[i].flags & what))
419 1.1 jtc change_flag((enum sh_flag) i, what, set);
420 1.1 jtc else {
421 1.1 jtc bi_errorf("%s: bad option", go.optarg);
422 1.1 jtc return -1;
423 1.1 jtc }
424 1.1 jtc break;
425 1.1 jtc
426 1.1 jtc case '?':
427 1.1 jtc return -1;
428 1.1 jtc
429 1.1 jtc default:
430 1.1 jtc /* -s: sort positional params (at&t ksh stupidity) */
431 1.1 jtc if (what == OF_SET && optc == 's') {
432 1.1 jtc sortargs = 1;
433 1.1 jtc break;
434 1.1 jtc }
435 1.13 lukem for (i = 0; i < (int)NELEM(goptions); i++)
436 1.12 christos if (optc == goptions[i].c
437 1.12 christos && (what & goptions[i].flags))
438 1.1 jtc {
439 1.1 jtc change_flag((enum sh_flag) i, what,
440 1.1 jtc set);
441 1.1 jtc break;
442 1.1 jtc }
443 1.12 christos if (i == NELEM(goptions)) {
444 1.1 jtc internal_errorf(1, "parse_args: `%c'", optc);
445 1.1 jtc return -1; /* not reached */
446 1.1 jtc }
447 1.1 jtc }
448 1.1 jtc }
449 1.1 jtc if (!(go.info & GI_MINUSMINUS) && argv[go.optind]
450 1.1 jtc && (argv[go.optind][0] == '-' || argv[go.optind][0] == '+')
451 1.1 jtc && argv[go.optind][1] == '\0')
452 1.1 jtc {
453 1.1 jtc /* lone - clears -v and -x flags */
454 1.1 jtc if (argv[go.optind][0] == '-' && !Flag(FPOSIX))
455 1.1 jtc Flag(FVERBOSE) = Flag(FXTRACE) = 0;
456 1.1 jtc /* set skips lone - or + option */
457 1.1 jtc go.optind++;
458 1.1 jtc }
459 1.1 jtc if (setargsp)
460 1.1 jtc /* -- means set $#/$* even if there are no arguments */
461 1.1 jtc *setargsp = !arrayset && ((go.info & GI_MINUSMINUS)
462 1.1 jtc || argv[go.optind]);
463 1.1 jtc
464 1.4 hubertf if (arrayset && (!*array || *skip_varname(array, FALSE))) {
465 1.4 hubertf bi_errorf("%s: is not an identifier", array);
466 1.4 hubertf return -1;
467 1.4 hubertf }
468 1.1 jtc if (sortargs) {
469 1.1 jtc for (i = go.optind; argv[i]; i++)
470 1.1 jtc ;
471 1.1 jtc qsortp((void **) &argv[go.optind], (size_t) (i - go.optind),
472 1.1 jtc xstrcmp);
473 1.1 jtc }
474 1.1 jtc if (arrayset) {
475 1.1 jtc set_array(array, arrayset, argv + go.optind);
476 1.1 jtc for (; argv[go.optind]; go.optind++)
477 1.1 jtc ;
478 1.1 jtc }
479 1.1 jtc
480 1.1 jtc return go.optind;
481 1.1 jtc }
482 1.1 jtc
483 1.1 jtc /* parse a decimal number: returns 0 if string isn't a number, 1 otherwise */
484 1.1 jtc int
485 1.1 jtc getn(as, ai)
486 1.1 jtc const char *as;
487 1.1 jtc int *ai;
488 1.1 jtc {
489 1.9 mycroft char *p;
490 1.9 mycroft long n;
491 1.9 mycroft
492 1.9 mycroft n = strtol(as, &p, 10);
493 1.1 jtc
494 1.9 mycroft if (!*as || *p || INT_MIN >= n || n >= INT_MAX)
495 1.1 jtc return 0;
496 1.9 mycroft
497 1.9 mycroft *ai = (int)n;
498 1.1 jtc return 1;
499 1.1 jtc }
500 1.1 jtc
501 1.1 jtc /* getn() that prints error */
502 1.1 jtc int
503 1.1 jtc bi_getn(as, ai)
504 1.1 jtc const char *as;
505 1.1 jtc int *ai;
506 1.1 jtc {
507 1.1 jtc int rv = getn(as, ai);
508 1.1 jtc
509 1.1 jtc if (!rv)
510 1.1 jtc bi_errorf("%s: bad number", as);
511 1.1 jtc return rv;
512 1.1 jtc }
513 1.1 jtc
514 1.1 jtc /* -------- gmatch.c -------- */
515 1.1 jtc
516 1.1 jtc /*
517 1.1 jtc * int gmatch(string, pattern)
518 1.1 jtc * char *string, *pattern;
519 1.1 jtc *
520 1.1 jtc * Match a pattern as in sh(1).
521 1.1 jtc * pattern character are prefixed with MAGIC by expand.
522 1.1 jtc */
523 1.1 jtc
524 1.1 jtc int
525 1.1 jtc gmatch(s, p, isfile)
526 1.1 jtc const char *s, *p;
527 1.1 jtc int isfile;
528 1.1 jtc {
529 1.1 jtc const char *se, *pe;
530 1.1 jtc
531 1.1 jtc if (s == NULL || p == NULL)
532 1.1 jtc return 0;
533 1.1 jtc se = s + strlen(s);
534 1.1 jtc pe = p + strlen(p);
535 1.1 jtc /* isfile is false iff no syntax check has been done on
536 1.1 jtc * the pattern. If check fails, just to a strcmp().
537 1.1 jtc */
538 1.1 jtc if (!isfile && !has_globbing(p, pe)) {
539 1.1 jtc int len = pe - p + 1;
540 1.1 jtc char tbuf[64];
541 1.13 lukem char *t = len <= (int)sizeof(tbuf) ? tbuf
542 1.1 jtc : (char *) alloc(len, ATEMP);
543 1.9 mycroft debunk(t, p, len);
544 1.1 jtc return !strcmp(t, s);
545 1.1 jtc }
546 1.1 jtc return do_gmatch((const unsigned char *) s, (const unsigned char *) se,
547 1.1 jtc (const unsigned char *) p, (const unsigned char *) pe,
548 1.1 jtc isfile);
549 1.1 jtc }
550 1.1 jtc
551 1.1 jtc /* Returns if p is a syntacticly correct globbing pattern, false
552 1.1 jtc * if it contains no pattern characters or if there is a syntax error.
553 1.1 jtc * Syntax errors are:
554 1.1 jtc * - [ with no closing ]
555 1.9 mycroft * - imbalanced $(...) expression
556 1.1 jtc * - [...] and *(...) not nested (eg, [a$(b|]c), *(a[b|c]d))
557 1.1 jtc */
558 1.1 jtc /*XXX
559 1.1 jtc - if no magic,
560 1.1 jtc if dest given, copy to dst
561 1.1 jtc return ?
562 1.1 jtc - if magic && (no globbing || syntax error)
563 1.1 jtc debunk to dst
564 1.1 jtc return ?
565 1.1 jtc - return ?
566 1.1 jtc */
567 1.1 jtc int
568 1.1 jtc has_globbing(xp, xpe)
569 1.1 jtc const char *xp, *xpe;
570 1.1 jtc {
571 1.1 jtc const unsigned char *p = (const unsigned char *) xp;
572 1.1 jtc const unsigned char *pe = (const unsigned char *) xpe;
573 1.1 jtc int c;
574 1.1 jtc int nest = 0, bnest = 0;
575 1.1 jtc int saw_glob = 0;
576 1.1 jtc int in_bracket = 0; /* inside [...] */
577 1.1 jtc
578 1.1 jtc for (; p < pe; p++) {
579 1.1 jtc if (!ISMAGIC(*p))
580 1.1 jtc continue;
581 1.1 jtc if ((c = *++p) == '*' || c == '?')
582 1.1 jtc saw_glob = 1;
583 1.1 jtc else if (c == '[') {
584 1.1 jtc if (!in_bracket) {
585 1.1 jtc saw_glob = 1;
586 1.1 jtc in_bracket = 1;
587 1.1 jtc if (ISMAGIC(p[1]) && p[2] == NOT)
588 1.1 jtc p += 2;
589 1.1 jtc if (ISMAGIC(p[1]) && p[2] == ']')
590 1.1 jtc p += 2;
591 1.1 jtc }
592 1.1 jtc /* XXX Do we need to check ranges here? POSIX Q */
593 1.1 jtc } else if (c == ']') {
594 1.1 jtc if (in_bracket) {
595 1.1 jtc if (bnest) /* [a*(b]) */
596 1.1 jtc return 0;
597 1.1 jtc in_bracket = 0;
598 1.1 jtc }
599 1.4 hubertf } else if ((c & 0x80) && strchr("*+?@! ", c & 0x7f)) {
600 1.1 jtc saw_glob = 1;
601 1.1 jtc if (in_bracket)
602 1.1 jtc bnest++;
603 1.1 jtc else
604 1.1 jtc nest++;
605 1.1 jtc } else if (c == '|') {
606 1.1 jtc if (in_bracket && !bnest) /* *(a[foo|bar]) */
607 1.1 jtc return 0;
608 1.1 jtc } else if (c == /*(*/ ')') {
609 1.1 jtc if (in_bracket) {
610 1.1 jtc if (!bnest--) /* *(a[b)c] */
611 1.1 jtc return 0;
612 1.1 jtc } else if (nest)
613 1.1 jtc nest--;
614 1.1 jtc }
615 1.1 jtc /* else must be a MAGIC-MAGIC, or MAGIC-!, MAGIC--, MAGIC-]
616 1.1 jtc MAGIC-{, MAGIC-,, MAGIC-} */
617 1.1 jtc }
618 1.1 jtc return saw_glob && !in_bracket && !nest;
619 1.1 jtc }
620 1.1 jtc
621 1.1 jtc /* Function must return either 0 or 1 (assumed by code for 0x80|'!') */
622 1.1 jtc static int
623 1.1 jtc do_gmatch(s, se, p, pe, isfile)
624 1.1 jtc const unsigned char *s, *p;
625 1.1 jtc const unsigned char *se, *pe;
626 1.1 jtc int isfile;
627 1.1 jtc {
628 1.16 christos int sc, pc;
629 1.1 jtc const unsigned char *prest, *psub, *pnext;
630 1.1 jtc const unsigned char *srest;
631 1.16 christos const unsigned char *sNext, *pNext, *sStart;
632 1.1 jtc
633 1.1 jtc if (s == NULL || p == NULL)
634 1.1 jtc return 0;
635 1.16 christos sNext = sStart = s;
636 1.16 christos pNext = p;
637 1.16 christos while (p < pe || s < se) {
638 1.16 christos pc = *p;
639 1.1 jtc sc = s < se ? *s : '\0';
640 1.1 jtc if (isfile) {
641 1.11 rillig sc = FILECHCONV((unsigned char)sc);
642 1.11 rillig pc = FILECHCONV((unsigned char)pc);
643 1.1 jtc }
644 1.1 jtc if (!ISMAGIC(pc)) {
645 1.1 jtc if (sc != pc)
646 1.16 christos goto backtrack;
647 1.16 christos p++;
648 1.16 christos s++;
649 1.1 jtc continue;
650 1.16 christos } else
651 1.16 christos pc = *++p;
652 1.16 christos switch (pc) {
653 1.1 jtc case '[':
654 1.16 christos p++;
655 1.16 christos s++;
656 1.1 jtc if (sc == 0 || (p = cclass(p, sc)) == NULL)
657 1.16 christos break;
658 1.16 christos continue;
659 1.1 jtc
660 1.1 jtc case '?':
661 1.1 jtc if (sc == 0)
662 1.16 christos break;
663 1.16 christos p++;
664 1.16 christos s++;
665 1.16 christos continue;
666 1.1 jtc
667 1.1 jtc case '*':
668 1.16 christos pNext = p - 1;
669 1.16 christos sNext = s + 1;
670 1.16 christos p++;
671 1.16 christos continue;
672 1.4 hubertf /*
673 1.4 hubertf * [*+?@!](pattern|pattern|..)
674 1.4 hubertf *
675 1.4 hubertf * Not ifdef'd KSH as this is needed for ${..%..}, etc.
676 1.4 hubertf */
677 1.1 jtc case 0x80|'+': /* matches one or more times */
678 1.1 jtc case 0x80|'*': /* matches zero or more times */
679 1.16 christos if (!(prest = pat_scan(++p, pe, 0)))
680 1.17 christos return 0;
681 1.1 jtc /* take care of zero matches */
682 1.1 jtc if (p[-1] == (0x80 | '*')
683 1.1 jtc && do_gmatch(s, se, prest, pe, isfile))
684 1.17 christos return 1;
685 1.1 jtc for (psub = p; ; psub = pnext) {
686 1.1 jtc pnext = pat_scan(psub, pe, 1);
687 1.1 jtc for (srest = s; srest <= se; srest++) {
688 1.1 jtc if (do_gmatch(s, srest,
689 1.1 jtc psub, pnext - 2, isfile)
690 1.1 jtc && (do_gmatch(srest, se,
691 1.1 jtc prest, pe, isfile)
692 1.1 jtc || (s != srest
693 1.1 jtc && do_gmatch(srest, se,
694 1.1 jtc p - 2, pe, isfile))))
695 1.1 jtc return 1;
696 1.1 jtc }
697 1.1 jtc if (pnext == prest)
698 1.1 jtc break;
699 1.1 jtc }
700 1.17 christos return 0;
701 1.1 jtc
702 1.1 jtc case 0x80|'?': /* matches zero or once */
703 1.1 jtc case 0x80|'@': /* matches one of the patterns */
704 1.4 hubertf case 0x80|' ': /* simile for @ */
705 1.16 christos if (!(prest = pat_scan(++p, pe, 0)))
706 1.17 christos return 0;
707 1.1 jtc /* Take care of zero matches */
708 1.1 jtc if (p[-1] == (0x80 | '?')
709 1.1 jtc && do_gmatch(s, se, prest, pe, isfile))
710 1.17 christos return 1;
711 1.1 jtc for (psub = p; ; psub = pnext) {
712 1.1 jtc pnext = pat_scan(psub, pe, 1);
713 1.1 jtc srest = prest == pe ? se : s;
714 1.1 jtc for (; srest <= se; srest++) {
715 1.1 jtc if (do_gmatch(s, srest,
716 1.1 jtc psub, pnext - 2, isfile)
717 1.1 jtc && do_gmatch(srest, se,
718 1.1 jtc prest, pe, isfile))
719 1.17 christos return 1;
720 1.1 jtc }
721 1.1 jtc if (pnext == prest)
722 1.1 jtc break;
723 1.1 jtc }
724 1.17 christos return 0;
725 1.1 jtc
726 1.1 jtc case 0x80|'!': /* matches none of the patterns */
727 1.16 christos if (!(prest = pat_scan(++p, pe, 0)))
728 1.17 christos return 0;
729 1.1 jtc for (srest = s; srest <= se; srest++) {
730 1.1 jtc int matched = 0;
731 1.1 jtc
732 1.1 jtc for (psub = p; ; psub = pnext) {
733 1.1 jtc pnext = pat_scan(psub, pe, 1);
734 1.1 jtc if (do_gmatch(s, srest,
735 1.1 jtc psub, pnext - 2, isfile))
736 1.1 jtc {
737 1.1 jtc matched = 1;
738 1.1 jtc break;
739 1.1 jtc }
740 1.1 jtc if (pnext == prest)
741 1.1 jtc break;
742 1.1 jtc }
743 1.1 jtc if (!matched && do_gmatch(srest, se,
744 1.1 jtc prest, pe, isfile))
745 1.17 christos return 1;
746 1.1 jtc }
747 1.17 christos return 0;
748 1.1 jtc
749 1.1 jtc default:
750 1.16 christos if (sc != pc)
751 1.16 christos break;
752 1.16 christos p++;
753 1.16 christos s++;
754 1.16 christos continue;
755 1.16 christos }
756 1.16 christos backtrack: if (sNext != sStart && sNext <= se) {
757 1.16 christos p = pNext;
758 1.16 christos s = sNext;
759 1.16 christos continue;
760 1.1 jtc }
761 1.16 christos return 0;
762 1.1 jtc }
763 1.16 christos return 1;
764 1.1 jtc }
765 1.1 jtc
766 1.1 jtc static const unsigned char *
767 1.1 jtc cclass(p, sub)
768 1.1 jtc const unsigned char *p;
769 1.1 jtc register int sub;
770 1.1 jtc {
771 1.1 jtc register int c, d, not, found = 0;
772 1.1 jtc const unsigned char *orig_p = p;
773 1.1 jtc
774 1.1 jtc if ((not = (ISMAGIC(*p) && *++p == NOT)))
775 1.1 jtc p++;
776 1.1 jtc do {
777 1.1 jtc c = *p++;
778 1.1 jtc if (ISMAGIC(c)) {
779 1.1 jtc c = *p++;
780 1.4 hubertf if ((c & 0x80) && !ISMAGIC(c)) {
781 1.1 jtc c &= 0x7f;/* extended pattern matching: *+?@! */
782 1.4 hubertf /* XXX the ( char isn't handled as part of [] */
783 1.4 hubertf if (c == ' ') /* simile for @: plain (..) */
784 1.4 hubertf c = '(' /*)*/;
785 1.4 hubertf }
786 1.1 jtc }
787 1.1 jtc if (c == '\0')
788 1.1 jtc /* No closing ] - act as if the opening [ was quoted */
789 1.1 jtc return sub == '[' ? orig_p : NULL;
790 1.1 jtc if (ISMAGIC(p[0]) && p[1] == '-'
791 1.1 jtc && (!ISMAGIC(p[2]) || p[3] != ']'))
792 1.1 jtc {
793 1.1 jtc p += 2; /* MAGIC- */
794 1.1 jtc d = *p++;
795 1.1 jtc if (ISMAGIC(d)) {
796 1.1 jtc d = *p++;
797 1.1 jtc if ((d & 0x80) && !ISMAGIC(d))
798 1.1 jtc d &= 0x7f;
799 1.1 jtc }
800 1.1 jtc /* POSIX says this is an invalid expression */
801 1.1 jtc if (c > d)
802 1.1 jtc return NULL;
803 1.1 jtc } else
804 1.1 jtc d = c;
805 1.1 jtc if (c == sub || (c <= sub && sub <= d))
806 1.1 jtc found = 1;
807 1.1 jtc } while (!(ISMAGIC(p[0]) && p[1] == ']'));
808 1.1 jtc
809 1.1 jtc return (found != not) ? p+2 : NULL;
810 1.1 jtc }
811 1.1 jtc
812 1.1 jtc /* Look for next ) or | (if match_sep) in *(foo|bar) pattern */
813 1.1 jtc const unsigned char *
814 1.1 jtc pat_scan(p, pe, match_sep)
815 1.1 jtc const unsigned char *p;
816 1.1 jtc const unsigned char *pe;
817 1.1 jtc int match_sep;
818 1.1 jtc {
819 1.1 jtc int nest = 0;
820 1.1 jtc
821 1.1 jtc for (; p < pe; p++) {
822 1.1 jtc if (!ISMAGIC(*p))
823 1.1 jtc continue;
824 1.1 jtc if ((*++p == /*(*/ ')' && nest-- == 0)
825 1.1 jtc || (*p == '|' && match_sep && nest == 0))
826 1.1 jtc return ++p;
827 1.4 hubertf if ((*p & 0x80) && strchr("*+?@! ", *p & 0x7f))
828 1.1 jtc nest++;
829 1.1 jtc }
830 1.1 jtc return (const unsigned char *) 0;
831 1.1 jtc }
832 1.1 jtc
833 1.1 jtc
834 1.1 jtc /* -------- qsort.c -------- */
835 1.1 jtc
836 1.1 jtc /*
837 1.1 jtc * quick sort of array of generic pointers to objects.
838 1.1 jtc */
839 1.1 jtc static void qsort1 ARGS((void **base, void **lim, int (*f)(void *, void *)));
840 1.1 jtc
841 1.1 jtc void
842 1.1 jtc qsortp(base, n, f)
843 1.1 jtc void **base; /* base address */
844 1.1 jtc size_t n; /* elements */
845 1.1 jtc int (*f) ARGS((void *, void *)); /* compare function */
846 1.1 jtc {
847 1.1 jtc qsort1(base, base + n, f);
848 1.1 jtc }
849 1.1 jtc
850 1.1 jtc #define swap2(a, b) {\
851 1.1 jtc register void *t; t = *(a); *(a) = *(b); *(b) = t;\
852 1.1 jtc }
853 1.1 jtc #define swap3(a, b, c) {\
854 1.1 jtc register void *t; t = *(a); *(a) = *(c); *(c) = *(b); *(b) = t;\
855 1.1 jtc }
856 1.1 jtc
857 1.1 jtc static void
858 1.1 jtc qsort1(base, lim, f)
859 1.1 jtc void **base, **lim;
860 1.1 jtc int (*f) ARGS((void *, void *));
861 1.1 jtc {
862 1.1 jtc register void **i, **j;
863 1.1 jtc register void **lptr, **hptr;
864 1.1 jtc size_t n;
865 1.1 jtc int c;
866 1.1 jtc
867 1.1 jtc top:
868 1.1 jtc n = (lim - base) / 2;
869 1.1 jtc if (n == 0)
870 1.1 jtc return;
871 1.1 jtc hptr = lptr = base+n;
872 1.1 jtc i = base;
873 1.1 jtc j = lim - 1;
874 1.1 jtc
875 1.1 jtc for (;;) {
876 1.1 jtc if (i < lptr) {
877 1.1 jtc if ((c = (*f)(*i, *lptr)) == 0) {
878 1.10 simonb lptr--;
879 1.1 jtc swap2(i, lptr);
880 1.1 jtc continue;
881 1.1 jtc }
882 1.1 jtc if (c < 0) {
883 1.1 jtc i += 1;
884 1.1 jtc continue;
885 1.1 jtc }
886 1.1 jtc }
887 1.1 jtc
888 1.1 jtc begin:
889 1.1 jtc if (j > hptr) {
890 1.1 jtc if ((c = (*f)(*hptr, *j)) == 0) {
891 1.10 simonb hptr++;
892 1.1 jtc swap2(hptr, j);
893 1.1 jtc goto begin;
894 1.1 jtc }
895 1.1 jtc if (c > 0) {
896 1.1 jtc if (i == lptr) {
897 1.10 simonb hptr++;
898 1.1 jtc swap3(i, hptr, j);
899 1.1 jtc i = lptr += 1;
900 1.1 jtc goto begin;
901 1.1 jtc }
902 1.1 jtc swap2(i, j);
903 1.1 jtc j -= 1;
904 1.1 jtc i += 1;
905 1.1 jtc continue;
906 1.1 jtc }
907 1.1 jtc j -= 1;
908 1.1 jtc goto begin;
909 1.1 jtc }
910 1.1 jtc
911 1.1 jtc if (i == lptr) {
912 1.1 jtc if (lptr-base >= lim-hptr) {
913 1.1 jtc qsort1(hptr+1, lim, f);
914 1.1 jtc lim = lptr;
915 1.1 jtc } else {
916 1.1 jtc qsort1(base, lptr, f);
917 1.1 jtc base = hptr+1;
918 1.1 jtc }
919 1.1 jtc goto top;
920 1.1 jtc }
921 1.1 jtc
922 1.1 jtc lptr -= 1;
923 1.1 jtc swap3(j, lptr, i);
924 1.1 jtc j = hptr -= 1;
925 1.1 jtc }
926 1.1 jtc }
927 1.1 jtc
928 1.1 jtc int
929 1.1 jtc xstrcmp(p1, p2)
930 1.1 jtc void *p1, *p2;
931 1.1 jtc {
932 1.1 jtc return (strcmp((char *)p1, (char *)p2));
933 1.1 jtc }
934 1.1 jtc
935 1.1 jtc /* Initialize a Getopt structure */
936 1.1 jtc void
937 1.1 jtc ksh_getopt_reset(go, flags)
938 1.1 jtc Getopt *go;
939 1.1 jtc int flags;
940 1.1 jtc {
941 1.1 jtc go->optind = 1;
942 1.1 jtc go->optarg = (char *) 0;
943 1.1 jtc go->p = 0;
944 1.1 jtc go->flags = flags;
945 1.1 jtc go->info = 0;
946 1.1 jtc go->buf[1] = '\0';
947 1.1 jtc }
948 1.1 jtc
949 1.1 jtc
950 1.1 jtc /* getopt() used for shell built-in commands, the getopts command, and
951 1.1 jtc * command line options.
952 1.1 jtc * A leading ':' in options means don't print errors, instead return '?'
953 1.1 jtc * or ':' and set go->optarg to the offending option character.
954 1.1 jtc * If GF_ERROR is set (and option doesn't start with :), errors result in
955 1.1 jtc * a call to bi_errorf().
956 1.1 jtc *
957 1.1 jtc * Non-standard features:
958 1.1 jtc * - ';' is like ':' in options, except the argument is optional
959 1.1 jtc * (if it isn't present, optarg is set to 0).
960 1.1 jtc * Used for 'set -o'.
961 1.1 jtc * - ',' is like ':' in options, except the argument always immediately
962 1.1 jtc * follows the option character (optarg is set to the null string if
963 1.1 jtc * the option is missing).
964 1.1 jtc * Used for 'read -u2', 'print -u2' and fc -40.
965 1.1 jtc * - '#' is like ':' in options, expect that the argument is optional
966 1.1 jtc * and must start with a digit. If the argument doesn't start with a
967 1.1 jtc * digit, it is assumed to be missing and normal option processing
968 1.1 jtc * continues (optarg is set to 0 if the option is missing).
969 1.1 jtc * Used for 'typeset -LZ4'.
970 1.1 jtc * - accepts +c as well as -c IF the GF_PLUSOPT flag is present. If an
971 1.1 jtc * option starting with + is accepted, the GI_PLUS flag will be set
972 1.4 hubertf * in go->info.
973 1.1 jtc */
974 1.1 jtc int
975 1.1 jtc ksh_getopt(argv, go, options)
976 1.1 jtc char **argv;
977 1.1 jtc Getopt *go;
978 1.1 jtc const char *options;
979 1.1 jtc {
980 1.1 jtc char c;
981 1.1 jtc char *o;
982 1.1 jtc
983 1.1 jtc if (go->p == 0 || (c = argv[go->optind - 1][go->p]) == '\0') {
984 1.1 jtc char *arg = argv[go->optind], flag = arg ? *arg : '\0';
985 1.1 jtc
986 1.1 jtc go->p = 1;
987 1.1 jtc if (flag == '-' && arg[1] == '-' && arg[2] == '\0') {
988 1.1 jtc go->optind++;
989 1.1 jtc go->p = 0;
990 1.1 jtc go->info |= GI_MINUSMINUS;
991 1.1 jtc return EOF;
992 1.1 jtc }
993 1.1 jtc if (arg == (char *) 0
994 1.4 hubertf || ((flag != '-' ) /* neither a - nor a + (if + allowed) */
995 1.4 hubertf && (!(go->flags & GF_PLUSOPT) || flag != '+'))
996 1.1 jtc || (c = arg[1]) == '\0')
997 1.1 jtc {
998 1.1 jtc go->p = 0;
999 1.1 jtc return EOF;
1000 1.1 jtc }
1001 1.1 jtc go->optind++;
1002 1.4 hubertf go->info &= ~(GI_MINUS|GI_PLUS);
1003 1.1 jtc go->info |= flag == '-' ? GI_MINUS : GI_PLUS;
1004 1.1 jtc }
1005 1.1 jtc go->p++;
1006 1.1 jtc if (c == '?' || c == ':' || c == ';' || c == ',' || c == '#'
1007 1.1 jtc || !(o = strchr(options, c)))
1008 1.1 jtc {
1009 1.1 jtc if (options[0] == ':') {
1010 1.1 jtc go->buf[0] = c;
1011 1.1 jtc go->optarg = go->buf;
1012 1.1 jtc } else {
1013 1.1 jtc warningf(TRUE, "%s%s-%c: unknown option",
1014 1.1 jtc (go->flags & GF_NONAME) ? "" : argv[0],
1015 1.1 jtc (go->flags & GF_NONAME) ? "" : ": ", c);
1016 1.1 jtc if (go->flags & GF_ERROR)
1017 1.15 joerg bi_errorf("%s", null);
1018 1.1 jtc }
1019 1.1 jtc return '?';
1020 1.1 jtc }
1021 1.1 jtc /* : means argument must be present, may be part of option argument
1022 1.1 jtc * or the next argument
1023 1.1 jtc * ; same as : but argument may be missing
1024 1.1 jtc * , means argument is part of option argument, and may be null.
1025 1.1 jtc */
1026 1.1 jtc if (*++o == ':' || *o == ';') {
1027 1.1 jtc if (argv[go->optind - 1][go->p])
1028 1.1 jtc go->optarg = argv[go->optind - 1] + go->p;
1029 1.1 jtc else if (argv[go->optind])
1030 1.1 jtc go->optarg = argv[go->optind++];
1031 1.1 jtc else if (*o == ';')
1032 1.1 jtc go->optarg = (char *) 0;
1033 1.1 jtc else {
1034 1.1 jtc if (options[0] == ':') {
1035 1.1 jtc go->buf[0] = c;
1036 1.1 jtc go->optarg = go->buf;
1037 1.1 jtc return ':';
1038 1.1 jtc }
1039 1.1 jtc warningf(TRUE, "%s%s-`%c' requires argument",
1040 1.1 jtc (go->flags & GF_NONAME) ? "" : argv[0],
1041 1.1 jtc (go->flags & GF_NONAME) ? "" : ": ", c);
1042 1.1 jtc if (go->flags & GF_ERROR)
1043 1.15 joerg bi_errorf("%s", null);
1044 1.1 jtc return '?';
1045 1.1 jtc }
1046 1.1 jtc go->p = 0;
1047 1.1 jtc } else if (*o == ',') {
1048 1.9 mycroft /* argument is attached to option character, even if null */
1049 1.1 jtc go->optarg = argv[go->optind - 1] + go->p;
1050 1.1 jtc go->p = 0;
1051 1.1 jtc } else if (*o == '#') {
1052 1.9 mycroft /* argument is optional and may be attached or unattached
1053 1.1 jtc * but must start with a digit. optarg is set to 0 if the
1054 1.1 jtc * argument is missing.
1055 1.1 jtc */
1056 1.1 jtc if (argv[go->optind - 1][go->p]) {
1057 1.1 jtc if (digit(argv[go->optind - 1][go->p])) {
1058 1.1 jtc go->optarg = argv[go->optind - 1] + go->p;
1059 1.1 jtc go->p = 0;
1060 1.1 jtc } else
1061 1.7 simonb go->optarg = (char *) 0;
1062 1.1 jtc } else {
1063 1.1 jtc if (argv[go->optind] && digit(argv[go->optind][0])) {
1064 1.1 jtc go->optarg = argv[go->optind++];
1065 1.1 jtc go->p = 0;
1066 1.1 jtc } else
1067 1.7 simonb go->optarg = (char *) 0;
1068 1.1 jtc }
1069 1.1 jtc }
1070 1.1 jtc return c;
1071 1.1 jtc }
1072 1.1 jtc
1073 1.1 jtc /* print variable/alias value using necessary quotes
1074 1.1 jtc * (POSIX says they should be suitable for re-entry...)
1075 1.1 jtc * No trailing newline is printed.
1076 1.1 jtc */
1077 1.1 jtc void
1078 1.1 jtc print_value_quoted(s)
1079 1.1 jtc const char *s;
1080 1.1 jtc {
1081 1.1 jtc const char *p;
1082 1.1 jtc int inquote = 0;
1083 1.1 jtc
1084 1.1 jtc /* Test if any quotes are needed */
1085 1.1 jtc for (p = s; *p; p++)
1086 1.1 jtc if (ctype(*p, C_QUOTE))
1087 1.1 jtc break;
1088 1.1 jtc if (!*p) {
1089 1.1 jtc shprintf("%s", s);
1090 1.1 jtc return;
1091 1.1 jtc }
1092 1.1 jtc for (p = s; *p; p++) {
1093 1.1 jtc if (*p == '\'') {
1094 1.15 joerg shprintf("%s", "'\\'" + 1 - inquote);
1095 1.1 jtc inquote = 0;
1096 1.1 jtc } else {
1097 1.1 jtc if (!inquote) {
1098 1.1 jtc shprintf("'");
1099 1.1 jtc inquote = 1;
1100 1.1 jtc }
1101 1.1 jtc shf_putc(*p, shl_stdout);
1102 1.1 jtc }
1103 1.1 jtc }
1104 1.1 jtc if (inquote)
1105 1.1 jtc shprintf("'");
1106 1.1 jtc }
1107 1.1 jtc
1108 1.1 jtc /* Print things in columns and rows - func() is called to format the ith
1109 1.1 jtc * element
1110 1.1 jtc */
1111 1.1 jtc void
1112 1.6 provos print_columns(shf, n, func, arg, max_width, prefcol)
1113 1.1 jtc struct shf *shf;
1114 1.1 jtc int n;
1115 1.1 jtc char *(*func) ARGS((void *, int, char *, int));
1116 1.1 jtc void *arg;
1117 1.1 jtc int max_width;
1118 1.6 provos int prefcol;
1119 1.1 jtc {
1120 1.1 jtc char *str = (char *) alloc(max_width + 1, ATEMP);
1121 1.1 jtc int i;
1122 1.1 jtc int r, c;
1123 1.1 jtc int rows, cols;
1124 1.1 jtc int nspace;
1125 1.1 jtc
1126 1.1 jtc /* max_width + 1 for the space. Note that no space
1127 1.1 jtc * is printed after the last column to avoid problems
1128 1.1 jtc * with terminals that have auto-wrap.
1129 1.1 jtc */
1130 1.1 jtc cols = x_cols / (max_width + 1);
1131 1.1 jtc if (!cols)
1132 1.1 jtc cols = 1;
1133 1.1 jtc rows = (n + cols - 1) / cols;
1134 1.6 provos if (prefcol && n && cols > rows) {
1135 1.1 jtc int tmp = rows;
1136 1.1 jtc
1137 1.1 jtc rows = cols;
1138 1.1 jtc cols = tmp;
1139 1.1 jtc if (rows > n)
1140 1.1 jtc rows = n;
1141 1.1 jtc }
1142 1.1 jtc
1143 1.1 jtc nspace = (x_cols - max_width * cols) / cols;
1144 1.1 jtc if (nspace <= 0)
1145 1.1 jtc nspace = 1;
1146 1.1 jtc for (r = 0; r < rows; r++) {
1147 1.1 jtc for (c = 0; c < cols; c++) {
1148 1.1 jtc i = c * rows + r;
1149 1.1 jtc if (i < n) {
1150 1.1 jtc shf_fprintf(shf, "%-*s",
1151 1.1 jtc max_width,
1152 1.1 jtc (*func)(arg, i, str, max_width + 1));
1153 1.1 jtc if (c + 1 < cols)
1154 1.1 jtc shf_fprintf(shf, "%*s", nspace, null);
1155 1.1 jtc }
1156 1.1 jtc }
1157 1.1 jtc shf_putchar('\n', shf);
1158 1.1 jtc }
1159 1.1 jtc afree(str, ATEMP);
1160 1.1 jtc }
1161 1.1 jtc
1162 1.1 jtc /* Strip any nul bytes from buf - returns new length (nbytes - # of nuls) */
1163 1.1 jtc int
1164 1.1 jtc strip_nuls(buf, nbytes)
1165 1.1 jtc char *buf;
1166 1.1 jtc int nbytes;
1167 1.1 jtc {
1168 1.1 jtc char *dst;
1169 1.1 jtc
1170 1.1 jtc /* nbytes check because some systems (older freebsd's) have a buggy
1171 1.1 jtc * memchr()
1172 1.1 jtc */
1173 1.1 jtc if (nbytes && (dst = memchr(buf, '\0', nbytes))) {
1174 1.1 jtc char *end = buf + nbytes;
1175 1.1 jtc char *p, *q;
1176 1.1 jtc
1177 1.1 jtc for (p = dst; p < end; p = q) {
1178 1.1 jtc /* skip a block of nulls */
1179 1.1 jtc while (++p < end && *p == '\0')
1180 1.1 jtc ;
1181 1.1 jtc /* find end of non-null block */
1182 1.1 jtc if (!(q = memchr(p, '\0', end - p)))
1183 1.1 jtc q = end;
1184 1.1 jtc memmove(dst, p, q - p);
1185 1.1 jtc dst += q - p;
1186 1.1 jtc }
1187 1.1 jtc *dst = '\0';
1188 1.1 jtc return dst - buf;
1189 1.1 jtc }
1190 1.1 jtc return nbytes;
1191 1.1 jtc }
1192 1.1 jtc
1193 1.1 jtc /* Copy at most dsize-1 bytes from src to dst, ensuring dst is null terminated.
1194 1.1 jtc * Returns dst.
1195 1.1 jtc */
1196 1.1 jtc char *
1197 1.1 jtc str_zcpy(dst, src, dsize)
1198 1.1 jtc char *dst;
1199 1.1 jtc const char *src;
1200 1.1 jtc int dsize;
1201 1.1 jtc {
1202 1.1 jtc if (dsize > 0) {
1203 1.1 jtc int len = strlen(src);
1204 1.1 jtc
1205 1.1 jtc if (len >= dsize)
1206 1.1 jtc len = dsize - 1;
1207 1.1 jtc memcpy(dst, src, len);
1208 1.1 jtc dst[len] = '\0';
1209 1.1 jtc }
1210 1.1 jtc return dst;
1211 1.1 jtc }
1212 1.1 jtc
1213 1.1 jtc /* Like read(2), but if read fails due to non-blocking flag, resets flag
1214 1.1 jtc * and restarts read.
1215 1.1 jtc */
1216 1.1 jtc int
1217 1.1 jtc blocking_read(fd, buf, nbytes)
1218 1.1 jtc int fd;
1219 1.1 jtc char *buf;
1220 1.1 jtc int nbytes;
1221 1.1 jtc {
1222 1.1 jtc int ret;
1223 1.1 jtc int tried_reset = 0;
1224 1.1 jtc
1225 1.1 jtc while ((ret = read(fd, buf, nbytes)) < 0) {
1226 1.1 jtc if (!tried_reset && (errno == EAGAIN
1227 1.1 jtc #ifdef EWOULDBLOCK
1228 1.1 jtc || errno == EWOULDBLOCK
1229 1.1 jtc #endif /* EWOULDBLOCK */
1230 1.1 jtc ))
1231 1.1 jtc {
1232 1.1 jtc int oerrno = errno;
1233 1.1 jtc if (reset_nonblock(fd) > 0) {
1234 1.1 jtc tried_reset = 1;
1235 1.1 jtc continue;
1236 1.1 jtc }
1237 1.1 jtc errno = oerrno;
1238 1.1 jtc }
1239 1.1 jtc break;
1240 1.1 jtc }
1241 1.1 jtc return ret;
1242 1.1 jtc }
1243 1.1 jtc
1244 1.1 jtc /* Reset the non-blocking flag on the specified file descriptor.
1245 1.1 jtc * Returns -1 if there was an error, 0 if non-blocking wasn't set,
1246 1.1 jtc * 1 if it was.
1247 1.1 jtc */
1248 1.1 jtc int
1249 1.1 jtc reset_nonblock(fd)
1250 1.1 jtc int fd;
1251 1.1 jtc {
1252 1.1 jtc int flags;
1253 1.1 jtc int blocking_flags;
1254 1.1 jtc
1255 1.1 jtc if ((flags = fcntl(fd, F_GETFL, 0)) < 0)
1256 1.1 jtc return -1;
1257 1.1 jtc /* With luck, the C compiler will reduce this to a constant */
1258 1.1 jtc blocking_flags = 0;
1259 1.1 jtc #ifdef O_NONBLOCK
1260 1.1 jtc blocking_flags |= O_NONBLOCK;
1261 1.1 jtc #endif /* O_NONBLOCK */
1262 1.1 jtc #ifdef O_NDELAY
1263 1.1 jtc blocking_flags |= O_NDELAY;
1264 1.1 jtc #else /* O_NDELAY */
1265 1.1 jtc # ifndef O_NONBLOCK
1266 1.1 jtc blocking_flags |= FNDELAY; /* hope this exists... */
1267 1.1 jtc # endif /* O_NONBLOCK */
1268 1.1 jtc #endif /* O_NDELAY */
1269 1.1 jtc if (!(flags & blocking_flags))
1270 1.1 jtc return 0;
1271 1.1 jtc flags &= ~blocking_flags;
1272 1.1 jtc if (fcntl(fd, F_SETFL, flags) < 0)
1273 1.1 jtc return -1;
1274 1.1 jtc return 1;
1275 1.1 jtc }
1276 1.1 jtc
1277 1.1 jtc
1278 1.1 jtc #ifdef HAVE_SYS_PARAM_H
1279 1.1 jtc # include <sys/param.h>
1280 1.1 jtc #endif /* HAVE_SYS_PARAM_H */
1281 1.1 jtc #ifndef MAXPATHLEN
1282 1.1 jtc # define MAXPATHLEN PATH
1283 1.1 jtc #endif /* MAXPATHLEN */
1284 1.1 jtc
1285 1.1 jtc /* Like getcwd(), except bsize is ignored if buf is 0 (MAXPATHLEN is used) */
1286 1.1 jtc char *
1287 1.1 jtc ksh_get_wd(buf, bsize)
1288 1.1 jtc char *buf;
1289 1.1 jtc int bsize;
1290 1.1 jtc {
1291 1.4 hubertf #ifdef HAVE_GETCWD
1292 1.4 hubertf char *b;
1293 1.4 hubertf char *ret;
1294 1.4 hubertf
1295 1.4 hubertf /* Assume getcwd() available */
1296 1.4 hubertf if (!buf) {
1297 1.4 hubertf bsize = MAXPATHLEN;
1298 1.4 hubertf b = alloc(MAXPATHLEN + 1, ATEMP);
1299 1.4 hubertf } else
1300 1.4 hubertf b = buf;
1301 1.4 hubertf
1302 1.4 hubertf ret = getcwd(b, bsize);
1303 1.4 hubertf
1304 1.4 hubertf if (!buf) {
1305 1.4 hubertf if (ret)
1306 1.4 hubertf ret = aresize(b, strlen(b) + 1, ATEMP);
1307 1.4 hubertf else
1308 1.4 hubertf afree(b, ATEMP);
1309 1.4 hubertf }
1310 1.4 hubertf
1311 1.4 hubertf return ret;
1312 1.4 hubertf #else /* HAVE_GETCWD */
1313 1.9 mycroft extern char *getwd ARGS((char *));
1314 1.1 jtc char *b;
1315 1.1 jtc int len;
1316 1.1 jtc
1317 1.1 jtc if (buf && bsize > MAXPATHLEN)
1318 1.1 jtc b = buf;
1319 1.1 jtc else
1320 1.1 jtc b = alloc(MAXPATHLEN + 1, ATEMP);
1321 1.3 erh if (!getcwd(b, MAXPATHLEN)) {
1322 1.1 jtc errno = EACCES;
1323 1.1 jtc if (b != buf)
1324 1.1 jtc afree(b, ATEMP);
1325 1.1 jtc return (char *) 0;
1326 1.1 jtc }
1327 1.1 jtc len = strlen(b) + 1;
1328 1.1 jtc if (!buf)
1329 1.1 jtc b = aresize(b, len, ATEMP);
1330 1.1 jtc else if (buf != b) {
1331 1.1 jtc if (len > bsize) {
1332 1.1 jtc errno = ERANGE;
1333 1.1 jtc return (char *) 0;
1334 1.1 jtc }
1335 1.1 jtc memcpy(buf, b, len);
1336 1.1 jtc afree(b, ATEMP);
1337 1.1 jtc b = buf;
1338 1.1 jtc }
1339 1.1 jtc
1340 1.1 jtc return b;
1341 1.4 hubertf #endif /* HAVE_GETCWD */
1342 1.1 jtc }
1343