-
Notifications
You must be signed in to change notification settings - Fork 36
/
main.c
2245 lines (2027 loc) · 50.8 KB
/
main.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
/* $OpenBSD: main.c,v 1.57 2015/09/10 22:48:58 nicm Exp $ */
/* $OpenBSD: tty.c,v 1.10 2014/08/10 02:44:26 guenther Exp $ */
/* $OpenBSD: io.c,v 1.26 2015/09/11 08:00:27 guenther Exp $ */
/* $OpenBSD: table.c,v 1.16 2015/09/01 13:12:31 tedu Exp $ */
/*-
* Copyright (c) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010,
* 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018,
* 2019, 2020, 2021, 2022, 2023
* mirabilos <[email protected]>
*
* Provided that these terms and disclaimer and all copyright notices
* are retained or reproduced in an accompanying document, permission
* is granted to deal in this work without restriction, including un-
* limited rights to use, publicly perform, distribute, sell, modify,
* merge, give away, or sublicence.
*
* This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to
* the utmost extent permitted by applicable law, neither express nor
* implied; without malicious intent or gross negligence. In no event
* may a licensor, author or contributor be held liable for indirect,
* direct, other damage, loss, or other issues arising in any way out
* of dealing in the work, even if advised of the possibility of such
* damage or existence of a defect, except proven that it results out
* of said person's immediate fault when using the work as intended.
*/
#define EXTERN
#include "sh.h"
__RCSID("$MirOS: src/bin/mksh/main.c,v 1.438 2023/10/06 21:56:49 tg Exp $");
__IDSTRING(mbsdcc_h_rcsid, SYSKERN_MBSDCC_H);
__IDSTRING(mbsdint_h_rcsid, SYSKERN_MBSDINT_H);
__IDSTRING(sh_h_rcsid, MKSH_SH_H_ID);
#ifndef MKSHRC_PATH
#define MKSHRC_PATH "~/.mkshrc"
#endif
#ifndef MKSH_DEFAULT_TMPDIR
#define MKSH_DEFAULT_TMPDIR MKSH_UNIXROOT "/tmp"
#endif
#if !HAVE_POSIX_UTF8_LOCALE
/* this is the “implementation-defined default locale” */
#ifdef MKSH_DEFAULT_UTFLOC
#define MKSH_DEFAULT_LOCALE "UTF-8"
#else
#define MKSH_DEFAULT_LOCALE "C"
#endif
#endif
static void main_init(int, const char *[], Source **);
void chvt_reinit(void);
static void reclaim(void);
static void remove_temps(struct temp *);
static void rndsetup(void);
static void init_environ(void);
#ifdef SIGWINCH
static void x_sigwinch(int);
#endif
#ifdef DEBUG
static void reclim_trace(void);
#else
#define reclim_trace() /* nothing */
#endif
static const char initsubs[] =
"${PS2=> }"
"${PS3=#? }"
"${PS4=+ }"
"${SECONDS=0}"
"${TMOUT=0}"
"${EPOCHREALTIME=}";
static const char *initcoms[] = {
Ttypeset, Tdr, initvsn, NULL,
Ttypeset, Tdx, "HOME", TPATH, TSHELL, NULL,
Ttypeset, "-i10", "COLUMNS", "LINES", "SECONDS", "TMOUT", NULL,
Talias,
"integer=\\\\builtin typeset -i",
"local=\\\\builtin typeset",
/* not "alias -t --": hash -r needs to work */
"hash=\\\\builtin alias -t",
"type=\\\\builtin whence -v",
"autoload=\\\\builtin typeset -fu",
"functions=\\\\builtin typeset -f",
"history=\\\\builtin fc -l",
"nameref=\\\\builtin typeset -n",
"nohup=nohup ",
"r=\\\\builtin fc -e -",
"login=\\\\builtin exec login",
NULL,
/* this is what AT&T ksh seems to track, with the addition of emacs */
Talias, "-tU",
"cat", "cc", "chmod", "cp", "date", "ed", "emacs", "grep", "ls",
"make", "mv", "pr", "rm", "sed", Tsh, "vi", "who", NULL,
NULL
};
static const char *restr_com[] = {
Ttypeset, Tdr, TENV, "HISTFILE", TPATH, TSHELL, NULL
};
extern const char Tpipest[];
/* top-level parsing and execution environment */
static struct env env;
struct env *e = &env;
/* buffer */
#if !HAVE_GET_CURRENT_DIR_NAME
static size_t getwd_bufsz = 448U;
#endif
static char *getwd_bufp = NULL;
/* many compile-time assertions */
mbCTA_BEG(main_c);
/* require char to be 8 bit long */
mbCTA(char_8bit, (CHAR_BIT) == 8 &&
(((unsigned int)(unsigned char)255U) == 255U) &&
(((unsigned int)(unsigned char)256U) == 0U) &&
mbiTYPE_UBITS(unsigned char) == 8U &&
mbiMASK_BITS(SCHAR_MAX) == 7U);
/* the next assertion is probably not really needed */
mbCTA(short_is_2_char, sizeof(short) == 2);
/* the next assertion is probably not really needed */
mbCTA(int_is_4_char, sizeof(int) == 4);
#ifndef MKSH_LEGACY_MODE
mbiCTA_TYPE_MBIT(sari, mksh_ari_t);
mbiCTA_TYPE_MBIT(uari, mksh_uari_t);
mbCTA(basic_int32_smask, mbiMASK_CHK(INT32_MAX));
mbCTA(basic_int32_umask, mbiMASK_CHK(UINT32_MAX));
mbCTA(basic_int32_ari,
mbiTYPE_UMAX(mksh_uari_t) == (UINT32_MAX) &&
/* require two’s complement */
((INT32_MIN)+1 == -(INT32_MAX)));
/* the next assertion is probably not really needed */
mbCTA(ari_is_4_char, sizeof(mksh_ari_t) == 4);
/* but this is */
mbCTA(ari_has_31_bit, mbiMASK_BITS(INT32_MAX) == 31);
/* the next assertion is probably not really needed */
mbCTA(uari_is_4_char, sizeof(mksh_uari_t) == 4);
mbCTA(uari_is_32_bit, mbiTYPE_UBITS(mksh_uari_t) == 32);
#else
mbCTA(long_complement, (LONG_MIN)+1 == -(LONG_MAX));
#endif
/* these are always required */
mbCTA(ari_is_signed, !mbiTYPE_ISU(mksh_ari_t));
mbCTA(uari_is_unsigned, mbiTYPE_ISU(mksh_uari_t));
/* we require these to have the precisely same size and assume 2s complement */
mbCTA(ari_size_no_matter_of_signedness,
sizeof(mksh_ari_t) == sizeof(mksh_uari_t));
/* our formatting routines assume this */
mbCTA(ari_fits_in_long, sizeof(mksh_ari_t) <= sizeof(long));
mbCTA_END(main_c);
/* end of compile-time asserts */
static void
rndsetup(void)
{
struct {
ALLOC_ITEM alloc_INT;
void *bssptr, *dataptr, *stkptr, *mallocptr;
kshjmp_buf jbuf;
struct timeval tv;
} *bufptr;
char *cp;
cp = alloc(sizeof(*bufptr) - sizeof(ALLOC_ITEM), APERM);
/* clear the allocated space, for valgrind and to avoid UB */
memset(cp, 0, sizeof(*bufptr) - sizeof(ALLOC_ITEM));
/* undo what alloc() did to the malloc result address */
bufptr = (void *)(cp - sizeof(ALLOC_ITEM));
/* PIE or something similar provides us with deltas here */
bufptr->bssptr = &rndsetupstate;
bufptr->dataptr = &e;
/* ASLR in at least Windows, Linux, some BSDs */
bufptr->stkptr = &bufptr;
/* randomised malloc in BSD (and possibly others) */
bufptr->mallocptr = bufptr;
/* glibc pointer guard */
#ifdef MKSH_NO_SIGSETJMP
setjmp(bufptr->jbuf);
#else
sigsetjmp(bufptr->jbuf, 1);
#endif
/* introduce variation (cannot use gettimeofday *tzp portably) */
mksh_TIME(bufptr->tv);
#ifdef MKSH_ALLOC_CATCH_UNDERRUNS
mprotect(((char *)bufptr) + 4096, 4096, PROT_READ | PROT_WRITE);
#endif
chvt_rndsetup(bufptr, sizeof(*bufptr));
afree(cp, APERM);
}
/* pre-initio() */
void
chvt_reinit(void)
{
kshpid = procpid = getpid();
ksheuid = geteuid();
kshpgrp = getpgrp();
kshppid = getppid();
}
static const char *empty_argv[] = {
Tmksh, NULL
};
#ifndef MKSH_EARLY_LOCALE_TRACKING
static kby
isuc(const char *cx) {
const char *cp;
if (!cx || !*cx)
return (0);
if ((cp = cstrchr(cx, '.')))
++cp;
else
cp = cx;
if (!isCh(cp[0], 'U', 'u') ||
!isCh(cp[1], 'T', 't') ||
!isCh(cp[2], 'F', 'f'))
return (0);
cp += isch(cp[3], '-') ? 4 : 3;
return (isch(*cp, '8') && (isch(cp[1], '@') || !cp[1]));
}
#endif
kby
kshname_islogin(const char **kshbasenamep)
{
const char *cp;
size_t o;
kby rv;
/* determine the basename (without '-' or path) of the executable */
cp = kshname;
o = 0;
while ((rv = cp[o++])) {
if (mksh_cdirsep(rv)) {
cp += o;
o = 0;
}
}
rv = isch(*cp, '-') || isch(*kshname, '-');
if (isch(*cp, '-'))
++cp;
if (!*cp)
cp = empty_argv[0];
*kshbasenamep = cp;
return (rv);
}
/* pre-initio() */
static void
main_init(int argc, const char *argv[], Source **sp)
{
int argi = 1, i;
Source *s = NULL;
unsigned char restricted_shell = 0, errexit, utf_flag;
char *cp;
const char *ccp, **wp;
struct tbl *vp;
struct stat s_stdin;
#if !defined(_PATH_DEFPATH) && defined(_CS_PATH)
ssize_t k;
#endif
#if defined(MKSH_EBCDIC) || defined(MKSH_FAUX_EBCDIC)
/* this must come *really* early due to locale use */
ebcdic_init();
#endif
set_ifs(TC_IFSWS);
#ifdef __OS2__
os2_init(&argc, &argv);
#define builtin_name_cmp stricmp
#else
#define builtin_name_cmp strcmp
#endif
/* do things like getpgrp() et al. */
chvt_reinit();
/* make sure argv[] is sane, for weird OSes */
if (!*argv) {
argv = empty_argv;
argc = 1;
}
kshname = argv[0];
/* initialise permanent Area */
ainit(&aperm);
/* max. name length: -2147483648 = 11 (+ NUL) */
vtemp = alloc(mbccFAMSZ(struct tbl, name, 12), APERM);
#if !HAVE_GET_CURRENT_DIR_NAME
getwd_bufp = alloc(getwd_bufsz + 1U, APERM);
getwd_bufp[getwd_bufsz] = '\0';
#endif
/* set up base environment */
env.type = E_NONE;
ainit(&env.area);
/* set up global l->vars and l->funs */
newblock();
/* Do this first so output routines (eg. kwarnf, shellf) can work */
initio();
/* check kshname: leading dash, determine basename */
Flag(FLOGIN) = kshname_islogin(&ccp);
/*
* Turn on nohup by default. (AT&T ksh does not have a nohup
* option - it always sends the hup).
*/
Flag(FNOHUP) = 1;
/*
* Turn on brace expansion by default. AT&T kshs that have
* alternation always have it on.
*/
Flag(FBRACEEXPAND) = 1;
/*
* Turn on "set -x" inheritance by default.
*/
Flag(FXTRACEREC) = 1;
/* define built-in commands and see if we were called as one */
ktinit(APERM, &builtins,
/* currently up to 50 builtins: 75% of 128 = 2^7 */
7);
for (i = 0; mkshbuiltins[i].name != NULL; ++i) {
const char *builtin_name;
builtin_name = builtin(mkshbuiltins[i].name,
mkshbuiltins[i].func);
if (!builtin_name_cmp(ccp, builtin_name)) {
/* canonicalise argv[0] */
ccp = builtin_name;
as_builtin = Ja;
}
}
if (!as_builtin) {
/* check for -T option early */
argi = parse_args(argv, OF_FIRSTTIME, NULL);
if (argi < 0)
unwind(LERROR);
/* called as rsh, rmksh, -rsh, RKSH.EXE, etc.? */
if (isCh(*ccp, 'R', 'r')) {
++ccp;
++restricted_shell;
}
#if defined(MKSH_BINSHPOSIX) || defined(MKSH_BINSHREDUCED)
/* are we called as -rsh or /bin/sh or SH.EXE or so? */
if (isCh(ccp[0], 'S', 's') &&
isCh(ccp[1], 'H', 'h')) {
/* either also turns off braceexpand */
#ifdef MKSH_BINSHPOSIX
/* enable better POSIX conformance */
change_flag(FPOSIX, OF_FIRSTTIME, Ja);
#endif
#ifdef MKSH_BINSHREDUCED
/* enable kludge/compat mode */
change_flag(FSH, OF_FIRSTTIME, Ja);
#endif
}
#endif
}
initvar();
inittraps();
coproc_init();
/* set up variable and command dictionaries */
ktinit(APERM, &taliases, 0);
ktinit(APERM, &aliases, 0);
#ifndef MKSH_NOPWNAM
ktinit(APERM, &homedirs, 0);
#endif
/* define shell keywords */
initkeywords();
init_histvec();
/* initialise tty size before importing environment */
change_winsz();
#ifdef _PATH_DEFPATH
def_path = _PATH_DEFPATH;
#else
#ifdef _CS_PATH
if ((k = confstr(_CS_PATH, NULL, 0)) > 0 &&
confstr(_CS_PATH, cp = alloc(k + 1, APERM), k + 1) == k + 1)
def_path = cp;
else
#endif
/*
* this is uniform across all OSes unless it
* breaks somewhere hard; don't try to optimise,
* e.g. add stuff for Interix or remove /usr
* for HURD, because e.g. Debian GNU/HURD is
* "keeping a regular /usr"; this is supposed
* to be a sane 'basic' default PATH
*/
def_path = MKSH_UNIXROOT "/bin" MKSH_PATHSEPS
MKSH_UNIXROOT "/usr/bin" MKSH_PATHSEPS
MKSH_UNIXROOT "/sbin" MKSH_PATHSEPS
MKSH_UNIXROOT "/usr/sbin";
#endif
/*
* Set PATH to def_path (will set the path global variable).
* Import of environment below will probably change this setting;
* the EXPORT flag is only added via initcoms for this to work.
*/
vp = global(TPATH);
/* setstr can't fail here */
setstr(vp, def_path, KSH_RETURN_ERROR);
#ifndef MKSH_NO_CMDLINE_EDITING
/*
* Set edit mode to emacs by default, may be overridden
* by the environment or the user. Also, we want tab completion
* on in vi by default.
*/
change_flag(FEMACS, OF_INTERNAL, Ja);
#if !MKSH_S_NOVI
Flag(FVITABCOMPLETE) = 1;
#endif
#endif
/* import environment */
init_environ();
/* for security */
typeset(TinitIFS, 0, 0, 0, 0);
/* assign default shell variable values */
typeset("PATHSEP=" MKSH_PATHSEPS, 0, 0, 0, 0);
substitute(initsubs, 0);
/* Figure out the current working directory and set $PWD */
vp = global(TPWD);
cp = str_val(vp);
/* Try to use existing $PWD if it is valid */
set_current_wd((mksh_abspath(cp) && test_eval(NULL, TO_FILEQ, cp,
Tdot, Ja)) ? cp : NULL);
if (current_wd[0])
simplify_path(current_wd);
/* Only set pwd if we know where we are or if it had a bogus value */
if (current_wd[0] || *cp)
/* setstr can't fail here */
setstr(vp, current_wd, KSH_RETURN_ERROR);
for (wp = initcoms; *wp != NULL; wp++) {
c_builtin(wp);
while (*wp != NULL)
wp++;
}
setint_n(global("OPTIND"), 1, 10);
kshuid = getuid();
kshgid = getgid();
kshegid = getegid();
rndsetup();
safe_prompt = ksheuid ? "$ " : "# ";
vp = global("PS1");
/* Set PS1 if unset or we are root and prompt doesn't contain a # */
if (!(vp->flag & ISSET) ||
(!ksheuid && !vstrchr(str_val(vp), '#')))
/* setstr can't fail here */
setstr(vp, safe_prompt, KSH_RETURN_ERROR);
/* ensure several variables will be (unsigned) integers */
setint_n((vp = global("BASHPID")), 0, 10);
vp->flag |= INT_U;
setint_n((vp = global("PGRP")), (mksh_uari_t)kshpgrp, 10);
vp->flag |= INT_U;
setint_n((vp = global("PPID")), (mksh_uari_t)kshppid, 10);
vp->flag |= INT_U;
setint_n((vp = global("USER_ID")), (mksh_uari_t)ksheuid, 10);
vp->flag |= INT_U;
setint_n((vp = global("KSHUID")), (mksh_uari_t)kshuid, 10);
vp->flag |= INT_U;
setint_n((vp = global("KSHEGID")), (mksh_uari_t)kshegid, 10);
vp->flag |= INT_U;
setint_n((vp = global("KSHGID")), (mksh_uari_t)kshgid, 10);
vp->flag |= INT_U;
/* this is needed globally */
setint_n((vp_pipest = global(Tpipest)), 0, 10);
/* unsigned integer, but avoid rndset call: done farther below */
vp = global("RANDOM");
vp->flag &= ~SPECIAL;
setint_n(vp, 0, 10);
vp->flag |= SPECIAL | INT_U;
/* Set this before parsing arguments */
Flag(FPRIVILEGED) = (kshuid != ksheuid || kshgid != kshegid) ? 2 : 0;
/* record if monitor is set on command line (see j_init() in jobs.c) */
#ifndef MKSH_UNEMPLOYED
Flag(FMONITOR) = 127;
#endif
/* this to note if utf-8 mode is set on command line (see below) */
/*XXX this and the below can probably go away */
/*XXX UTFMODE should be set from env here if locale tracking, just keep it */
UTFMODE = 2;
if (!as_builtin) {
argi = parse_args(argv, OF_CMDLINE, NULL);
if (argi < 0)
unwind(LERROR);
}
/*XXX drop this and the entire cases below */
/* process this later only, default to off (hysterical raisins) */
utf_flag = UTFMODE;
UTFMODE = 0;
if (as_builtin) {
/* auto-detect from environment variables, always */
utf_flag = 3;
} else if (Flag(FCOMMAND)) {
s = pushs(SSTRINGCMDLINE, ATEMP);
if (!(s->start = s->str = argv[argi++]))
kerrf(KWF_ERR(1) | KWF_PREFIX | KWF_FILELINE |
KWF_TWOMSG | KWF_NOERRNO, Tdc, Treq_arg);
while (*s->str) {
if (ctype(*s->str, C_QUOTE))
break;
s->str++;
}
if (!*s->str)
s->flags |= SF_MAYEXEC;
s->str = s->start;
#ifdef MKSH_MIDNIGHTBSD01ASH_COMPAT
/* compatibility to MidnightBSD 0.1 /bin/sh (kludge) */
if (Flag(FSH) && argv[argi] && !strcmp(argv[argi], "--"))
++argi;
#endif
if (argv[argi])
kshname = argv[argi++];
} else if (argi < argc && !Flag(FSTDIN)) {
s = pushs(SFILE, ATEMP);
#ifdef __OS2__
/*
* A bug in OS/2 extproc (like shebang) handling makes
* it not pass the full pathname of a script, so we need
* to search for it. This changes the behaviour of a
* simple "mksh foo", but can't be helped.
*/
s->file = argv[argi++];
if (search_access(s->file, X_OK) != 0)
s->file = search_path(s->file, path, X_OK, NULL);
if (!s->file || !*s->file)
s->file = argv[argi - 1];
#else
s->file = argv[argi++];
#endif
s->u.shf = shf_open(s->file, O_RDONLY | O_MAYEXEC, 0,
SHF_MAPHI | SHF_CLEXEC);
if (s->u.shf == NULL) {
shl_stdout_ok = Nee;
kwarnf(KWF_ERR(127) |
KWF_PREFIX | KWF_FILELINE | KWF_ONEMSG, s->file);
unwind(LERROR);
}
kshname = s->file;
} else {
Flag(FSTDIN) = 1;
s = pushs(SSTDIN, ATEMP);
s->file = "<stdin>";
s->u.shf = shf_fdopen(0, SHF_RD | can_seek(0),
NULL);
if (isatty(0) && isatty(2)) {
Flag(FTALKING) = Flag(FTALKING_I) = 1;
/* The following only if isatty(0) */
s->flags |= SF_TTY;
s->u.shf->flags |= SHF_INTERRUPT;
s->file = NULL;
}
}
/* this bizarreness is mandated by POSIX */
if (Flag(FTALKING) && fstat(0, &s_stdin) >= 0 &&
(S_ISCHR(s_stdin.st_mode) || S_ISFIFO(s_stdin.st_mode)))
reset_nonblock(0);
/* initialise job control */
j_init();
/* do this after j_init() which calls tty_init_state() */
if (Flag(FTALKING)) {
if (utf_flag == 2) {
#ifndef MKSH_ASSUME_UTF8
/* auto-detect from locale or environment */
utf_flag = 4;
#else /* this may not be an #elif */
#if MKSH_ASSUME_UTF8
utf_flag = 1;
#else
/* always disable UTF-8 (for interactive) */
utf_flag = 0;
#endif
#endif
}
#ifndef MKSH_NO_CMDLINE_EDITING
x_init();
#endif
}
#ifdef SIGWINCH
sigtraps[SIGWINCH].flags |= TF_SHELL_USES;
setsig(&sigtraps[SIGWINCH], x_sigwinch,
SS_RESTORE_ORIG|SS_FORCE|SS_SHTRAP);
#endif
/*
* allocate a new array because otherwise, when we modify
* it in-place, ps(1) output changes; the meaning of argc
* here is slightly different as it excludes kshname, and
* we add a trailing NULL sentinel as well
*/
e->loc->argc = argc - argi;
e->loc->argv = alloc2(e->loc->argc + 2, sizeof(char *), APERM);
memcpy(&e->loc->argv[1], &argv[argi], e->loc->argc * sizeof(char *));
e->loc->argv[e->loc->argc + 1] = NULL;
if (as_builtin)
e->loc->argv[0] = ccp;
else {
e->loc->argv[0] = kshname;
getopts_reset(1);
}
/*XXX to go away entirely with locale tracking */
/* divine the initial state of the utf8-mode Flag */
ccp = null;
switch (utf_flag) {
#ifdef MKSH_EARLY_LOCALE_TRACKING
/* not set on command line, not FTALKING */
case 2:
#endif
/* auto-detect from locale or environment */
case 4:
#ifndef MKSH_EARLY_LOCALE_TRACKING
#if HAVE_POSIX_UTF8_LOCALE
ccp = setlocale(LC_CTYPE, "");
if (!isuc(ccp))
ccp = nl_langinfo(CODESET);
if (!isuc(ccp))
ccp = null;
#endif
#endif
/* FALLTHROUGH */
/* auto-detect from environment */
case 3:
#ifndef MKSH_EARLY_LOCALE_TRACKING
/* these were imported from environ earlier */
if (ccp == null)
ccp = str_val(global("LC_ALL"));
if (ccp == null)
ccp = str_val(global("LC_CTYPE"));
if (ccp == null)
ccp = str_val(global("LANG"));
UTFMODE = isuc(ccp);
#else
recheck_ctype(); /*XXX probably unnecessary here */
#endif
break;
#ifndef MKSH_EARLY_LOCALE_TRACKING
/* not set on command line, not FTALKING */
case 2:
#endif
/* unknown values */
default:
utf_flag = 0;
/* FALLTHROUGH */
/* known values */
case 1:
case 0:
UTFMODE = utf_flag;
break;
}
/* Disable during .profile/ENV reading */
restricted_shell |= Flag(FRESTRICTED);
Flag(FRESTRICTED) = 0;
errexit = Flag(FERREXIT);
Flag(FERREXIT) = 0;
/* save flags for "set +o" handling */
memcpy(baseline_flags, shell_flags, sizeof(shell_flags));
/* disable these because they have special handling */
baseline_flags[(int)FPOSIX] = 0;
baseline_flags[(int)FSH] = 0;
/* ensure these always show up setting, for FPOSIX/FSH */
baseline_flags[(int)FBRACEEXPAND] = 0;
baseline_flags[(int)FUNNYCODE] = 0;
#if !defined(MKSH_SMALL) || defined(DEBUG)
/* mark as initialised */
baseline_flags[(int)FNFLAGS] = 1;
#endif
rndpush(shell_flags, sizeof(shell_flags));
rndset(hash(/*=current_wd*/ cp) ^ hash(argv[0]));
if (as_builtin)
goto skip_startup_files;
/*
* Do this before profile/$ENV so that if it causes problems in them,
* user will know why things broke.
*/
if (!current_wd[0] && Flag(FTALKING))
kwarnf(KWF_PREFIX | KWF_ONEMSG | KWF_NOERRNO,
"can't determine current directory");
if (Flag(FLOGIN))
include(MKSH_SYSTEM_PROFILE, NULL, Ja);
if (Flag(FPRIVILEGED)) {
include(MKSH_SUID_PROFILE, NULL, Ja);
/* note whether -p was enabled during startup */
if (Flag(FPRIVILEGED) == 1)
/* allow set -p to setuid() later */
Flag(FPRIVILEGED) = 3;
else
/* turn off -p if not set explicitly */
change_flag(FPRIVILEGED, OF_INTERNAL, Nee);
/* track shell-imposed changes */
baseline_flags[(int)FPRIVILEGED] = Flag(FPRIVILEGED);
} else {
if (Flag(FLOGIN))
include(substitute("$HOME/.profile", 0), NULL, Ja);
if (Flag(FTALKING)) {
cp = substitute("${ENV:-" MKSHRC_PATH "}", DOTILDE);
if (cp[0] != '\0')
include(cp, NULL, Ja);
}
}
if (restricted_shell) {
c_builtin(restr_com);
/* After typeset command... */
Flag(FRESTRICTED) = 1;
/* track shell-imposed changes */
baseline_flags[(int)FRESTRICTED] = 1;
}
Flag(FERREXIT) = errexit;
if (Flag(FTALKING) && s)
hist_init(s);
else {
/* set after ENV */
skip_startup_files:
Flag(FTRACKALL) = 1;
/* track shell-imposed change (might lower surprise) */
baseline_flags[(int)FTRACKALL] = 1;
}
alarm_init();
*sp = s;
}
/* this indirection barrier reduces stack usage during normal operation */
int
main(int argc, const char *argv[])
{
int rv;
Source *s;
main_init(argc, argv, &s);
if (as_builtin) {
rv = c_builtin(e->loc->argv) & 0xFF;
exstat = rv;
unwind(LEXIT);
/* NOTREACHED */
} else {
rv = shell(s, 0) & 0xFF;
/* NOTREACHED */
}
return (rv);
}
int
include(const char *name, const char **argv, Wahr intr_ok)
{
Source *volatile s = NULL;
struct shf *shf;
const char **volatile old_argv;
volatile int old_argc;
int i;
shf = shf_open(name, O_RDONLY | O_MAYEXEC, 0, SHF_MAPHI | SHF_CLEXEC);
if (shf == NULL)
return (-1);
if (argv) {
old_argv = e->loc->argv;
old_argc = e->loc->argc;
} else {
old_argv = NULL;
old_argc = 0;
}
newenv(E_INCL);
if ((i = kshsetjmp(e->jbuf))) {
quitenv(s ? s->u.shf : NULL);
if (old_argv) {
e->loc->argv = old_argv;
e->loc->argc = old_argc;
}
switch (i) {
case LRETURN:
case LERROR:
case LERREXT:
/* see below */
return (exstat & 0xFF);
case LINTR:
/*
* intr_ok is set if we are including .profile or $ENV.
* If user ^Cs out, we don't want to kill the shell...
*/
if (intr_ok && ((exstat & 0xFF) - 128) != SIGTERM)
return (1);
/* FALLTHROUGH */
case LEXIT:
case LLEAVE:
case LSHELL:
unwind(i);
/* NOTREACHED */
default:
kerrf0(KWF_INTERNAL | KWF_ERR(0xFF) | KWF_NOERRNO,
Tunexpected_type, Tunwind, Tsource, i);
/* NOTREACHED */
}
}
if (argv) {
e->loc->argv = argv;
e->loc->argc = 0;
while (argv[e->loc->argc + 1])
++e->loc->argc;
}
s = pushs(SFILE, ATEMP);
s->u.shf = shf;
strdupx(s->file, name, ATEMP);
i = shell(s, 1);
quitenv(s->u.shf);
if (old_argv) {
e->loc->argv = old_argv;
e->loc->argc = old_argc;
}
/* & 0xff to ensure value not -1 */
return (i & 0xFF);
}
/* spawn a command into a shell optionally keeping track of the line number */
int
command(const char *comm, int line)
{
Source *s, *sold = source;
int rv;
s = pushs(SSTRING, ATEMP);
s->start = s->str = comm;
s->line = line;
rv = shell(s, 1);
source = sold;
return (rv);
}
/*
* run the commands from the input source, returning status.
*/
int
shell(Source * volatile s, volatile int level)
{
struct op *t;
volatile Wahr wastty = isWahr(s->flags & SF_TTY);
volatile kby attempts = 13;
volatile Wahr interactive = (level == 0) && Flag(FTALKING);
Source *volatile old_source = source;
int i;
newenv(level == 2 ? E_EVAL : E_PARSE);
if (level == 2)
e->flags |= EF_IN_EVAL;
if (interactive)
really_exit = Nee;
switch ((i = kshsetjmp(e->jbuf))) {
case 0:
break;
case LBREAK:
case LCONTIN:
/* assert: interactive == Nee */
source = old_source;
quitenv(NULL);
if (level == 2) {
/* keep on going */
unwind(i);
/* NOTREACHED */
}
kerrf0(KWF_INTERNAL | KWF_ERR(0xFF) | KWF_NOERRNO,
Tf_cant_s, Tshell, i == LBREAK ? Tbreak : Tcontinue);
/* NOTREACHED */
case LINTR:
/* we get here if SIGINT not caught or ignored */
case LERROR:
case LERREXT:
case LSHELL:
if (interactive) {
if (i == LINTR)
shellf(Tnl);
/*
* Reset any eof that was read as part of a
* multiline command.
*/
if (Flag(FIGNOREEOF) && s->type == SEOF && wastty)
s->type = SSTDIN;
/*
* Used by exit command to get back to
* top level shell. Kind of strange since
* interactive is set if we are reading from
* a tty, but to have stopped jobs, one only
* needs FMONITOR set (not FTALKING/SF_TTY)...
*/
/* toss any input we have so far */
yyrecursive_pop(Ja);
s->start = s->str = null;
retrace_info = NULL;
herep = heres;
break;
}
/* FALLTHROUGH */
case LEXIT:
case LLEAVE:
case LRETURN:
source = old_source;
quitenv(NULL);
if (i == LERREXT && level == 2)
return (exstat & 0xFF);
/* keep on going */
unwind(i);
/* NOTREACHED */
default:
source = old_source;
quitenv(NULL);
kerrf0(KWF_INTERNAL | KWF_ERR(0xFF) | KWF_NOERRNO,
Tunexpected_type, Tunwind, Tshell, i);
/* NOTREACHED */
}
while (/* CONSTCOND */ 1) {
if (trap)
runtraps(0);
if (s->next == NULL) {
if (Flag(FVERBOSE))
s->flags |= SF_ECHO;
else
s->flags &= ~SF_ECHO;
}
if (interactive) {
j_notify();
set_prompt(PS1, s);
}
t = compile(s, Ja);
if (interactive)
histsave(&s->line, NULL, HIST_FLUSH, Ja);
if (!t)
goto source_no_tree;
if (t->type == TEOF) {
if (wastty && Flag(FIGNOREEOF) && --attempts > 0) {
shellf("Use 'exit' to leave mksh\n");
s->type = SSTDIN;
} else if (wastty && !really_exit &&
j_stopped_running()) {
really_exit = Ja;
s->type = SSTDIN;
} else {
/*
* this for POSIX which says EXIT traps
* shall be taken in the environment
* immediately after the last command
* executed.