cli_hush.c 94 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * sh.c -- a prototype Bourne shell grammar parser
  4. * Intended to follow the original Thompson and Ritchie
  5. * "small and simple is beautiful" philosophy, which
  6. * incidentally is a good match to today's BusyBox.
  7. *
  8. * Copyright (C) 2000,2001 Larry Doolittle <larry@doolittle.boa.org>
  9. *
  10. * Credits:
  11. * The parser routines proper are all original material, first
  12. * written Dec 2000 and Jan 2001 by Larry Doolittle.
  13. * The execution engine, the builtins, and much of the underlying
  14. * support has been adapted from busybox-0.49pre's lash,
  15. * which is Copyright (C) 2000 by Lineo, Inc., and
  16. * written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>.
  17. * That, in turn, is based in part on ladsh.c, by Michael K. Johnson and
  18. * Erik W. Troan, which they placed in the public domain. I don't know
  19. * how much of the Johnson/Troan code has survived the repeated rewrites.
  20. * Other credits:
  21. * b_addchr() derived from similar w_addchar function in glibc-2.2
  22. * setup_redirect(), redirect_opt_num(), and big chunks of main()
  23. * and many builtins derived from contributions by Erik Andersen
  24. * miscellaneous bugfixes from Matt Kraai
  25. *
  26. * There are two big (and related) architecture differences between
  27. * this parser and the lash parser. One is that this version is
  28. * actually designed from the ground up to understand nearly all
  29. * of the Bourne grammar. The second, consequential change is that
  30. * the parser and input reader have been turned inside out. Now,
  31. * the parser is in control, and asks for input as needed. The old
  32. * way had the input reader in control, and it asked for parsing to
  33. * take place as needed. The new way makes it much easier to properly
  34. * handle the recursion implicit in the various substitutions, especially
  35. * across continuation lines.
  36. *
  37. * Bash grammar not implemented: (how many of these were in original sh?)
  38. * $@ (those sure look like weird quoting rules)
  39. * $_
  40. * ! negation operator for pipes
  41. * &> and >& redirection of stdout+stderr
  42. * Brace Expansion
  43. * Tilde Expansion
  44. * fancy forms of Parameter Expansion
  45. * aliases
  46. * Arithmetic Expansion
  47. * <(list) and >(list) Process Substitution
  48. * reserved words: case, esac, select, function
  49. * Here Documents ( << word )
  50. * Functions
  51. * Major bugs:
  52. * job handling woefully incomplete and buggy
  53. * reserved word execution woefully incomplete and buggy
  54. * to-do:
  55. * port selected bugfixes from post-0.49 busybox lash - done?
  56. * finish implementing reserved words: for, while, until, do, done
  57. * change { and } from special chars to reserved words
  58. * builtins: break, continue, eval, return, set, trap, ulimit
  59. * test magic exec
  60. * handle children going into background
  61. * clean up recognition of null pipes
  62. * check setting of global_argc and global_argv
  63. * control-C handling, probably with longjmp
  64. * follow IFS rules more precisely, including update semantics
  65. * figure out what to do with backslash-newline
  66. * explain why we use signal instead of sigaction
  67. * propagate syntax errors, die on resource errors?
  68. * continuation lines, both explicit and implicit - done?
  69. * memory leak finding and plugging - done?
  70. * more testing, especially quoting rules and redirection
  71. * document how quoting rules not precisely followed for variable assignments
  72. * maybe change map[] to use 2-bit entries
  73. * (eventually) remove all the printf's
  74. */
  75. #define __U_BOOT__
  76. #ifdef __U_BOOT__
  77. #include <common.h> /* readline */
  78. #include <env.h>
  79. #include <malloc.h> /* malloc, free, realloc*/
  80. #include <linux/ctype.h> /* isalpha, isdigit */
  81. #include <console.h>
  82. #include <bootretry.h>
  83. #include <cli.h>
  84. #include <cli_hush.h>
  85. #include <command.h> /* find_cmd */
  86. #ifndef CONFIG_SYS_PROMPT_HUSH_PS2
  87. #define CONFIG_SYS_PROMPT_HUSH_PS2 "> "
  88. #endif
  89. #endif
  90. #ifndef __U_BOOT__
  91. #include <ctype.h> /* isalpha, isdigit */
  92. #include <unistd.h> /* getpid */
  93. #include <stdlib.h> /* getenv, atoi */
  94. #include <string.h> /* strchr */
  95. #include <stdio.h> /* popen etc. */
  96. #include <glob.h> /* glob, of course */
  97. #include <stdarg.h> /* va_list */
  98. #include <errno.h>
  99. #include <fcntl.h>
  100. #include <getopt.h> /* should be pretty obvious */
  101. #include <sys/stat.h> /* ulimit */
  102. #include <sys/types.h>
  103. #include <sys/wait.h>
  104. #include <signal.h>
  105. /* #include <dmalloc.h> */
  106. #if 1
  107. #include "busybox.h"
  108. #include "cmdedit.h"
  109. #else
  110. #define applet_name "hush"
  111. #include "standalone.h"
  112. #define hush_main main
  113. #undef CONFIG_FEATURE_SH_FANCY_PROMPT
  114. #define BB_BANNER
  115. #endif
  116. #endif
  117. #define SPECIAL_VAR_SYMBOL 03
  118. #define SUBSTED_VAR_SYMBOL 04
  119. #ifndef __U_BOOT__
  120. #define FLAG_EXIT_FROM_LOOP 1
  121. #define FLAG_PARSE_SEMICOLON (1 << 1) /* symbol ';' is special for parser */
  122. #define FLAG_REPARSING (1 << 2) /* >= 2nd pass */
  123. #endif
  124. #ifdef __U_BOOT__
  125. DECLARE_GLOBAL_DATA_PTR;
  126. #define EXIT_SUCCESS 0
  127. #define EOF -1
  128. #define syntax() syntax_err()
  129. #define xstrdup strdup
  130. #define error_msg printf
  131. #else
  132. typedef enum {
  133. REDIRECT_INPUT = 1,
  134. REDIRECT_OVERWRITE = 2,
  135. REDIRECT_APPEND = 3,
  136. REDIRECT_HEREIS = 4,
  137. REDIRECT_IO = 5
  138. } redir_type;
  139. /* The descrip member of this structure is only used to make debugging
  140. * output pretty */
  141. struct {int mode; int default_fd; char *descrip;} redir_table[] = {
  142. { 0, 0, "()" },
  143. { O_RDONLY, 0, "<" },
  144. { O_CREAT|O_TRUNC|O_WRONLY, 1, ">" },
  145. { O_CREAT|O_APPEND|O_WRONLY, 1, ">>" },
  146. { O_RDONLY, -1, "<<" },
  147. { O_RDWR, 1, "<>" }
  148. };
  149. #endif
  150. typedef enum {
  151. PIPE_SEQ = 1,
  152. PIPE_AND = 2,
  153. PIPE_OR = 3,
  154. PIPE_BG = 4,
  155. } pipe_style;
  156. /* might eventually control execution */
  157. typedef enum {
  158. RES_NONE = 0,
  159. RES_IF = 1,
  160. RES_THEN = 2,
  161. RES_ELIF = 3,
  162. RES_ELSE = 4,
  163. RES_FI = 5,
  164. RES_FOR = 6,
  165. RES_WHILE = 7,
  166. RES_UNTIL = 8,
  167. RES_DO = 9,
  168. RES_DONE = 10,
  169. RES_XXXX = 11,
  170. RES_IN = 12,
  171. RES_SNTX = 13
  172. } reserved_style;
  173. #define FLAG_END (1<<RES_NONE)
  174. #define FLAG_IF (1<<RES_IF)
  175. #define FLAG_THEN (1<<RES_THEN)
  176. #define FLAG_ELIF (1<<RES_ELIF)
  177. #define FLAG_ELSE (1<<RES_ELSE)
  178. #define FLAG_FI (1<<RES_FI)
  179. #define FLAG_FOR (1<<RES_FOR)
  180. #define FLAG_WHILE (1<<RES_WHILE)
  181. #define FLAG_UNTIL (1<<RES_UNTIL)
  182. #define FLAG_DO (1<<RES_DO)
  183. #define FLAG_DONE (1<<RES_DONE)
  184. #define FLAG_IN (1<<RES_IN)
  185. #define FLAG_START (1<<RES_XXXX)
  186. /* This holds pointers to the various results of parsing */
  187. struct p_context {
  188. struct child_prog *child;
  189. struct pipe *list_head;
  190. struct pipe *pipe;
  191. #ifndef __U_BOOT__
  192. struct redir_struct *pending_redirect;
  193. #endif
  194. reserved_style w;
  195. int old_flag; /* for figuring out valid reserved words */
  196. struct p_context *stack;
  197. int type; /* define type of parser : ";$" common or special symbol */
  198. /* How about quoting status? */
  199. };
  200. #ifndef __U_BOOT__
  201. struct redir_struct {
  202. redir_type type; /* type of redirection */
  203. int fd; /* file descriptor being redirected */
  204. int dup; /* -1, or file descriptor being duplicated */
  205. struct redir_struct *next; /* pointer to the next redirect in the list */
  206. glob_t word; /* *word.gl_pathv is the filename */
  207. };
  208. #endif
  209. struct child_prog {
  210. #ifndef __U_BOOT__
  211. pid_t pid; /* 0 if exited */
  212. #endif
  213. char **argv; /* program name and arguments */
  214. /* was quoted when parsed; copy of struct o_string.nonnull field */
  215. int *argv_nonnull;
  216. #ifdef __U_BOOT__
  217. int argc; /* number of program arguments */
  218. #endif
  219. struct pipe *group; /* if non-NULL, first in group or subshell */
  220. #ifndef __U_BOOT__
  221. int subshell; /* flag, non-zero if group must be forked */
  222. struct redir_struct *redirects; /* I/O redirections */
  223. glob_t glob_result; /* result of parameter globbing */
  224. int is_stopped; /* is the program currently running? */
  225. struct pipe *family; /* pointer back to the child's parent pipe */
  226. #endif
  227. int sp; /* number of SPECIAL_VAR_SYMBOL */
  228. int type;
  229. };
  230. struct pipe {
  231. #ifndef __U_BOOT__
  232. int jobid; /* job number */
  233. #endif
  234. int num_progs; /* total number of programs in job */
  235. #ifndef __U_BOOT__
  236. int running_progs; /* number of programs running */
  237. char *text; /* name of job */
  238. char *cmdbuf; /* buffer various argv's point into */
  239. pid_t pgrp; /* process group ID for the job */
  240. #endif
  241. struct child_prog *progs; /* array of commands in pipe */
  242. struct pipe *next; /* to track background commands */
  243. #ifndef __U_BOOT__
  244. int stopped_progs; /* number of programs alive, but stopped */
  245. int job_context; /* bitmask defining current context */
  246. #endif
  247. pipe_style followup; /* PIPE_BG, PIPE_SEQ, PIPE_OR, PIPE_AND */
  248. reserved_style r_mode; /* supports if, for, while, until */
  249. };
  250. #ifndef __U_BOOT__
  251. struct close_me {
  252. int fd;
  253. struct close_me *next;
  254. };
  255. #endif
  256. struct variables {
  257. char *name;
  258. char *value;
  259. int flg_export;
  260. int flg_read_only;
  261. struct variables *next;
  262. };
  263. /* globals, connect us to the outside world
  264. * the first three support $?, $#, and $1 */
  265. #ifndef __U_BOOT__
  266. char **global_argv;
  267. unsigned int global_argc;
  268. #endif
  269. static unsigned int last_return_code;
  270. #ifndef __U_BOOT__
  271. extern char **environ; /* This is in <unistd.h>, but protected with __USE_GNU */
  272. #endif
  273. /* "globals" within this file */
  274. static uchar *ifs;
  275. static char map[256];
  276. #ifndef __U_BOOT__
  277. static int fake_mode;
  278. static int interactive;
  279. static struct close_me *close_me_head;
  280. static const char *cwd;
  281. static struct pipe *job_list;
  282. static unsigned int last_bg_pid;
  283. static unsigned int last_jobid;
  284. static unsigned int shell_terminal;
  285. static char *PS1;
  286. static char *PS2;
  287. struct variables shell_ver = { "HUSH_VERSION", "0.01", 1, 1, 0 };
  288. struct variables *top_vars = &shell_ver;
  289. #else
  290. static int flag_repeat = 0;
  291. static int do_repeat = 0;
  292. static struct variables *top_vars = NULL ;
  293. #endif /*__U_BOOT__ */
  294. #define B_CHUNK (100)
  295. #define B_NOSPAC 1
  296. typedef struct {
  297. char *data;
  298. int length;
  299. int maxlen;
  300. int quote;
  301. int nonnull;
  302. } o_string;
  303. #define NULL_O_STRING {NULL,0,0,0,0}
  304. /* used for initialization:
  305. o_string foo = NULL_O_STRING; */
  306. /* I can almost use ordinary FILE *. Is open_memstream() universally
  307. * available? Where is it documented? */
  308. struct in_str {
  309. const char *p;
  310. #ifndef __U_BOOT__
  311. char peek_buf[2];
  312. #endif
  313. int __promptme;
  314. int promptmode;
  315. #ifndef __U_BOOT__
  316. FILE *file;
  317. #endif
  318. int (*get) (struct in_str *);
  319. int (*peek) (struct in_str *);
  320. };
  321. #define b_getch(input) ((input)->get(input))
  322. #define b_peek(input) ((input)->peek(input))
  323. #ifndef __U_BOOT__
  324. #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
  325. struct built_in_command {
  326. char *cmd; /* name */
  327. char *descr; /* description */
  328. int (*function) (struct child_prog *); /* function ptr */
  329. };
  330. #endif
  331. /* define DEBUG_SHELL for debugging output (obviously ;-)) */
  332. #if 0
  333. #define DEBUG_SHELL
  334. #endif
  335. /* This should be in utility.c */
  336. #ifdef DEBUG_SHELL
  337. #ifndef __U_BOOT__
  338. static void debug_printf(const char *format, ...)
  339. {
  340. va_list args;
  341. va_start(args, format);
  342. vfprintf(stderr, format, args);
  343. va_end(args);
  344. }
  345. #else
  346. #define debug_printf(fmt,args...) printf (fmt ,##args)
  347. #endif
  348. #else
  349. static inline void debug_printf(const char *format, ...) { }
  350. #endif
  351. #define final_printf debug_printf
  352. #ifdef __U_BOOT__
  353. static void syntax_err(void) {
  354. printf("syntax error\n");
  355. }
  356. #else
  357. static void __syntax(char *file, int line) {
  358. error_msg("syntax error %s:%d", file, line);
  359. }
  360. #define syntax() __syntax(__FILE__, __LINE__)
  361. #endif
  362. #ifdef __U_BOOT__
  363. static void *xmalloc(size_t size);
  364. static void *xrealloc(void *ptr, size_t size);
  365. #else
  366. /* Index of subroutines: */
  367. /* function prototypes for builtins */
  368. static int builtin_cd(struct child_prog *child);
  369. static int builtin_env(struct child_prog *child);
  370. static int builtin_eval(struct child_prog *child);
  371. static int builtin_exec(struct child_prog *child);
  372. static int builtin_exit(struct child_prog *child);
  373. static int builtin_export(struct child_prog *child);
  374. static int builtin_fg_bg(struct child_prog *child);
  375. static int builtin_help(struct child_prog *child);
  376. static int builtin_jobs(struct child_prog *child);
  377. static int builtin_pwd(struct child_prog *child);
  378. static int builtin_read(struct child_prog *child);
  379. static int builtin_set(struct child_prog *child);
  380. static int builtin_shift(struct child_prog *child);
  381. static int builtin_source(struct child_prog *child);
  382. static int builtin_umask(struct child_prog *child);
  383. static int builtin_unset(struct child_prog *child);
  384. static int builtin_not_written(struct child_prog *child);
  385. #endif
  386. /* o_string manipulation: */
  387. static int b_check_space(o_string *o, int len);
  388. static int b_addchr(o_string *o, int ch);
  389. static void b_reset(o_string *o);
  390. static int b_addqchr(o_string *o, int ch, int quote);
  391. #ifndef __U_BOOT__
  392. static int b_adduint(o_string *o, unsigned int i);
  393. #endif
  394. /* in_str manipulations: */
  395. static int static_get(struct in_str *i);
  396. static int static_peek(struct in_str *i);
  397. static int file_get(struct in_str *i);
  398. static int file_peek(struct in_str *i);
  399. #ifndef __U_BOOT__
  400. static void setup_file_in_str(struct in_str *i, FILE *f);
  401. #else
  402. static void setup_file_in_str(struct in_str *i);
  403. #endif
  404. static void setup_string_in_str(struct in_str *i, const char *s);
  405. #ifndef __U_BOOT__
  406. /* close_me manipulations: */
  407. static void mark_open(int fd);
  408. static void mark_closed(int fd);
  409. static void close_all(void);
  410. #endif
  411. /* "run" the final data structures: */
  412. static char *indenter(int i);
  413. static int free_pipe_list(struct pipe *head, int indent);
  414. static int free_pipe(struct pipe *pi, int indent);
  415. /* really run the final data structures: */
  416. #ifndef __U_BOOT__
  417. static int setup_redirects(struct child_prog *prog, int squirrel[]);
  418. #endif
  419. static int run_list_real(struct pipe *pi);
  420. #ifndef __U_BOOT__
  421. static void pseudo_exec(struct child_prog *child) __attribute__ ((noreturn));
  422. #endif
  423. static int run_pipe_real(struct pipe *pi);
  424. /* extended glob support: */
  425. #ifndef __U_BOOT__
  426. static int globhack(const char *src, int flags, glob_t *pglob);
  427. static int glob_needed(const char *s);
  428. static int xglob(o_string *dest, int flags, glob_t *pglob);
  429. #endif
  430. /* variable assignment: */
  431. static int is_assignment(const char *s);
  432. /* data structure manipulation: */
  433. #ifndef __U_BOOT__
  434. static int setup_redirect(struct p_context *ctx, int fd, redir_type style, struct in_str *input);
  435. #endif
  436. static void initialize_context(struct p_context *ctx);
  437. static int done_word(o_string *dest, struct p_context *ctx);
  438. static int done_command(struct p_context *ctx);
  439. static int done_pipe(struct p_context *ctx, pipe_style type);
  440. /* primary string parsing: */
  441. #ifndef __U_BOOT__
  442. static int redirect_dup_num(struct in_str *input);
  443. static int redirect_opt_num(o_string *o);
  444. static int process_command_subs(o_string *dest, struct p_context *ctx, struct in_str *input, int subst_end);
  445. static int parse_group(o_string *dest, struct p_context *ctx, struct in_str *input, int ch);
  446. #endif
  447. static char *lookup_param(char *src);
  448. static char *make_string(char **inp, int *nonnull);
  449. static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input);
  450. #ifndef __U_BOOT__
  451. static int parse_string(o_string *dest, struct p_context *ctx, const char *src);
  452. #endif
  453. static int parse_stream(o_string *dest, struct p_context *ctx, struct in_str *input0, int end_trigger);
  454. /* setup: */
  455. static int parse_stream_outer(struct in_str *inp, int flag);
  456. #ifndef __U_BOOT__
  457. static int parse_string_outer(const char *s, int flag);
  458. static int parse_file_outer(FILE *f);
  459. #endif
  460. #ifndef __U_BOOT__
  461. /* job management: */
  462. static int checkjobs(struct pipe* fg_pipe);
  463. static void insert_bg_job(struct pipe *pi);
  464. static void remove_bg_job(struct pipe *pi);
  465. #endif
  466. /* local variable support */
  467. static char **make_list_in(char **inp, char *name);
  468. static char *insert_var_value(char *inp);
  469. static char *insert_var_value_sub(char *inp, int tag_subst);
  470. #ifndef __U_BOOT__
  471. /* Table of built-in functions. They can be forked or not, depending on
  472. * context: within pipes, they fork. As simple commands, they do not.
  473. * When used in non-forking context, they can change global variables
  474. * in the parent shell process. If forked, of course they can not.
  475. * For example, 'unset foo | whatever' will parse and run, but foo will
  476. * still be set at the end. */
  477. static struct built_in_command bltins[] = {
  478. {"bg", "Resume a job in the background", builtin_fg_bg},
  479. {"break", "Exit for, while or until loop", builtin_not_written},
  480. {"cd", "Change working directory", builtin_cd},
  481. {"continue", "Continue for, while or until loop", builtin_not_written},
  482. {"env", "Print all environment variables", builtin_env},
  483. {"eval", "Construct and run shell command", builtin_eval},
  484. {"exec", "Exec command, replacing this shell with the exec'd process",
  485. builtin_exec},
  486. {"exit", "Exit from shell()", builtin_exit},
  487. {"export", "Set environment variable", builtin_export},
  488. {"fg", "Bring job into the foreground", builtin_fg_bg},
  489. {"jobs", "Lists the active jobs", builtin_jobs},
  490. {"pwd", "Print current directory", builtin_pwd},
  491. {"read", "Input environment variable", builtin_read},
  492. {"return", "Return from a function", builtin_not_written},
  493. {"set", "Set/unset shell local variables", builtin_set},
  494. {"shift", "Shift positional parameters", builtin_shift},
  495. {"trap", "Trap signals", builtin_not_written},
  496. {"ulimit","Controls resource limits", builtin_not_written},
  497. {"umask","Sets file creation mask", builtin_umask},
  498. {"unset", "Unset environment variable", builtin_unset},
  499. {".", "Source-in and run commands in a file", builtin_source},
  500. {"help", "List shell built-in commands", builtin_help},
  501. {NULL, NULL, NULL}
  502. };
  503. static const char *set_cwd(void)
  504. {
  505. if(cwd==unknown)
  506. cwd = NULL; /* xgetcwd(arg) called free(arg) */
  507. cwd = xgetcwd((char *)cwd);
  508. if (!cwd)
  509. cwd = unknown;
  510. return cwd;
  511. }
  512. /* built-in 'eval' handler */
  513. static int builtin_eval(struct child_prog *child)
  514. {
  515. char *str = NULL;
  516. int rcode = EXIT_SUCCESS;
  517. if (child->argv[1]) {
  518. str = make_string(child->argv + 1);
  519. parse_string_outer(str, FLAG_EXIT_FROM_LOOP |
  520. FLAG_PARSE_SEMICOLON);
  521. free(str);
  522. rcode = last_return_code;
  523. }
  524. return rcode;
  525. }
  526. /* built-in 'cd <path>' handler */
  527. static int builtin_cd(struct child_prog *child)
  528. {
  529. char *newdir;
  530. if (child->argv[1] == NULL)
  531. newdir = env_get("HOME");
  532. else
  533. newdir = child->argv[1];
  534. if (chdir(newdir)) {
  535. printf("cd: %s: %s\n", newdir, strerror(errno));
  536. return EXIT_FAILURE;
  537. }
  538. set_cwd();
  539. return EXIT_SUCCESS;
  540. }
  541. /* built-in 'env' handler */
  542. static int builtin_env(struct child_prog *dummy)
  543. {
  544. char **e = environ;
  545. if (e == NULL) return EXIT_FAILURE;
  546. for (; *e; e++) {
  547. puts(*e);
  548. }
  549. return EXIT_SUCCESS;
  550. }
  551. /* built-in 'exec' handler */
  552. static int builtin_exec(struct child_prog *child)
  553. {
  554. if (child->argv[1] == NULL)
  555. return EXIT_SUCCESS; /* Really? */
  556. child->argv++;
  557. pseudo_exec(child);
  558. /* never returns */
  559. }
  560. /* built-in 'exit' handler */
  561. static int builtin_exit(struct child_prog *child)
  562. {
  563. if (child->argv[1] == NULL)
  564. exit(last_return_code);
  565. exit (atoi(child->argv[1]));
  566. }
  567. /* built-in 'export VAR=value' handler */
  568. static int builtin_export(struct child_prog *child)
  569. {
  570. int res = 0;
  571. char *name = child->argv[1];
  572. if (name == NULL) {
  573. return (builtin_env(child));
  574. }
  575. name = strdup(name);
  576. if(name) {
  577. char *value = strchr(name, '=');
  578. if (!value) {
  579. char *tmp;
  580. /* They are exporting something without an =VALUE */
  581. value = get_local_var(name);
  582. if (value) {
  583. size_t ln = strlen(name);
  584. tmp = realloc(name, ln+strlen(value)+2);
  585. if(tmp==NULL)
  586. res = -1;
  587. else {
  588. sprintf(tmp+ln, "=%s", value);
  589. name = tmp;
  590. }
  591. } else {
  592. /* bash does not return an error when trying to export
  593. * an undefined variable. Do likewise. */
  594. res = 1;
  595. }
  596. }
  597. }
  598. if (res<0)
  599. perror_msg("export");
  600. else if(res==0)
  601. res = set_local_var(name, 1);
  602. else
  603. res = 0;
  604. free(name);
  605. return res;
  606. }
  607. /* built-in 'fg' and 'bg' handler */
  608. static int builtin_fg_bg(struct child_prog *child)
  609. {
  610. int i, jobnum;
  611. struct pipe *pi=NULL;
  612. if (!interactive)
  613. return EXIT_FAILURE;
  614. /* If they gave us no args, assume they want the last backgrounded task */
  615. if (!child->argv[1]) {
  616. for (pi = job_list; pi; pi = pi->next) {
  617. if (pi->jobid == last_jobid) {
  618. break;
  619. }
  620. }
  621. if (!pi) {
  622. error_msg("%s: no current job", child->argv[0]);
  623. return EXIT_FAILURE;
  624. }
  625. } else {
  626. if (sscanf(child->argv[1], "%%%d", &jobnum) != 1) {
  627. error_msg("%s: bad argument '%s'", child->argv[0], child->argv[1]);
  628. return EXIT_FAILURE;
  629. }
  630. for (pi = job_list; pi; pi = pi->next) {
  631. if (pi->jobid == jobnum) {
  632. break;
  633. }
  634. }
  635. if (!pi) {
  636. error_msg("%s: %d: no such job", child->argv[0], jobnum);
  637. return EXIT_FAILURE;
  638. }
  639. }
  640. if (*child->argv[0] == 'f') {
  641. /* Put the job into the foreground. */
  642. tcsetpgrp(shell_terminal, pi->pgrp);
  643. }
  644. /* Restart the processes in the job */
  645. for (i = 0; i < pi->num_progs; i++)
  646. pi->progs[i].is_stopped = 0;
  647. if ( (i=kill(- pi->pgrp, SIGCONT)) < 0) {
  648. if (i == ESRCH) {
  649. remove_bg_job(pi);
  650. } else {
  651. perror_msg("kill (SIGCONT)");
  652. }
  653. }
  654. pi->stopped_progs = 0;
  655. return EXIT_SUCCESS;
  656. }
  657. /* built-in 'help' handler */
  658. static int builtin_help(struct child_prog *dummy)
  659. {
  660. struct built_in_command *x;
  661. printf("\nBuilt-in commands:\n");
  662. printf("-------------------\n");
  663. for (x = bltins; x->cmd; x++) {
  664. if (x->descr==NULL)
  665. continue;
  666. printf("%s\t%s\n", x->cmd, x->descr);
  667. }
  668. printf("\n\n");
  669. return EXIT_SUCCESS;
  670. }
  671. /* built-in 'jobs' handler */
  672. static int builtin_jobs(struct child_prog *child)
  673. {
  674. struct pipe *job;
  675. char *status_string;
  676. for (job = job_list; job; job = job->next) {
  677. if (job->running_progs == job->stopped_progs)
  678. status_string = "Stopped";
  679. else
  680. status_string = "Running";
  681. printf(JOB_STATUS_FORMAT, job->jobid, status_string, job->text);
  682. }
  683. return EXIT_SUCCESS;
  684. }
  685. /* built-in 'pwd' handler */
  686. static int builtin_pwd(struct child_prog *dummy)
  687. {
  688. puts(set_cwd());
  689. return EXIT_SUCCESS;
  690. }
  691. /* built-in 'read VAR' handler */
  692. static int builtin_read(struct child_prog *child)
  693. {
  694. int res;
  695. if (child->argv[1]) {
  696. char string[BUFSIZ];
  697. char *var = 0;
  698. string[0] = 0; /* In case stdin has only EOF */
  699. /* read string */
  700. fgets(string, sizeof(string), stdin);
  701. chomp(string);
  702. var = malloc(strlen(child->argv[1])+strlen(string)+2);
  703. if(var) {
  704. sprintf(var, "%s=%s", child->argv[1], string);
  705. res = set_local_var(var, 0);
  706. } else
  707. res = -1;
  708. if (res)
  709. fprintf(stderr, "read: %m\n");
  710. free(var); /* So not move up to avoid breaking errno */
  711. return res;
  712. } else {
  713. do res=getchar(); while(res!='\n' && res!=EOF);
  714. return 0;
  715. }
  716. }
  717. /* built-in 'set VAR=value' handler */
  718. static int builtin_set(struct child_prog *child)
  719. {
  720. char *temp = child->argv[1];
  721. struct variables *e;
  722. if (temp == NULL)
  723. for(e = top_vars; e; e=e->next)
  724. printf("%s=%s\n", e->name, e->value);
  725. else
  726. set_local_var(temp, 0);
  727. return EXIT_SUCCESS;
  728. }
  729. /* Built-in 'shift' handler */
  730. static int builtin_shift(struct child_prog *child)
  731. {
  732. int n=1;
  733. if (child->argv[1]) {
  734. n=atoi(child->argv[1]);
  735. }
  736. if (n>=0 && n<global_argc) {
  737. /* XXX This probably breaks $0 */
  738. global_argc -= n;
  739. global_argv += n;
  740. return EXIT_SUCCESS;
  741. } else {
  742. return EXIT_FAILURE;
  743. }
  744. }
  745. /* Built-in '.' handler (read-in and execute commands from file) */
  746. static int builtin_source(struct child_prog *child)
  747. {
  748. FILE *input;
  749. int status;
  750. if (child->argv[1] == NULL)
  751. return EXIT_FAILURE;
  752. /* XXX search through $PATH is missing */
  753. input = fopen(child->argv[1], "r");
  754. if (!input) {
  755. error_msg("Couldn't open file '%s'", child->argv[1]);
  756. return EXIT_FAILURE;
  757. }
  758. /* Now run the file */
  759. /* XXX argv and argc are broken; need to save old global_argv
  760. * (pointer only is OK!) on this stack frame,
  761. * set global_argv=child->argv+1, recurse, and restore. */
  762. mark_open(fileno(input));
  763. status = parse_file_outer(input);
  764. mark_closed(fileno(input));
  765. fclose(input);
  766. return (status);
  767. }
  768. static int builtin_umask(struct child_prog *child)
  769. {
  770. mode_t new_umask;
  771. const char *arg = child->argv[1];
  772. char *end;
  773. if (arg) {
  774. new_umask=strtoul(arg, &end, 8);
  775. if (*end!='\0' || end == arg) {
  776. return EXIT_FAILURE;
  777. }
  778. } else {
  779. printf("%.3o\n", (unsigned int) (new_umask=umask(0)));
  780. }
  781. umask(new_umask);
  782. return EXIT_SUCCESS;
  783. }
  784. /* built-in 'unset VAR' handler */
  785. static int builtin_unset(struct child_prog *child)
  786. {
  787. /* bash returned already true */
  788. unset_local_var(child->argv[1]);
  789. return EXIT_SUCCESS;
  790. }
  791. static int builtin_not_written(struct child_prog *child)
  792. {
  793. printf("builtin_%s not written\n",child->argv[0]);
  794. return EXIT_FAILURE;
  795. }
  796. #endif
  797. static int b_check_space(o_string *o, int len)
  798. {
  799. /* It would be easy to drop a more restrictive policy
  800. * in here, such as setting a maximum string length */
  801. if (o->length + len > o->maxlen) {
  802. char *old_data = o->data;
  803. /* assert (data == NULL || o->maxlen != 0); */
  804. o->maxlen += max(2*len, B_CHUNK);
  805. o->data = realloc(o->data, 1 + o->maxlen);
  806. if (o->data == NULL) {
  807. free(old_data);
  808. }
  809. }
  810. return o->data == NULL;
  811. }
  812. static int b_addchr(o_string *o, int ch)
  813. {
  814. debug_printf("b_addchr: %c %d %p\n", ch, o->length, o);
  815. if (b_check_space(o, 1)) return B_NOSPAC;
  816. o->data[o->length] = ch;
  817. o->length++;
  818. o->data[o->length] = '\0';
  819. return 0;
  820. }
  821. static void b_reset(o_string *o)
  822. {
  823. o->length = 0;
  824. o->nonnull = 0;
  825. if (o->data != NULL) *o->data = '\0';
  826. }
  827. static void b_free(o_string *o)
  828. {
  829. b_reset(o);
  830. free(o->data);
  831. o->data = NULL;
  832. o->maxlen = 0;
  833. }
  834. /* My analysis of quoting semantics tells me that state information
  835. * is associated with a destination, not a source.
  836. */
  837. static int b_addqchr(o_string *o, int ch, int quote)
  838. {
  839. if (quote && strchr("*?[\\",ch)) {
  840. int rc;
  841. rc = b_addchr(o, '\\');
  842. if (rc) return rc;
  843. }
  844. return b_addchr(o, ch);
  845. }
  846. #ifndef __U_BOOT__
  847. static int b_adduint(o_string *o, unsigned int i)
  848. {
  849. int r;
  850. char *p = simple_itoa(i);
  851. /* no escape checking necessary */
  852. do r=b_addchr(o, *p++); while (r==0 && *p);
  853. return r;
  854. }
  855. #endif
  856. static int static_get(struct in_str *i)
  857. {
  858. int ch = *i->p++;
  859. if (ch=='\0') return EOF;
  860. return ch;
  861. }
  862. static int static_peek(struct in_str *i)
  863. {
  864. return *i->p;
  865. }
  866. #ifndef __U_BOOT__
  867. static inline void cmdedit_set_initial_prompt(void)
  868. {
  869. #ifndef CONFIG_FEATURE_SH_FANCY_PROMPT
  870. PS1 = NULL;
  871. #else
  872. PS1 = env_get("PS1");
  873. if(PS1==0)
  874. PS1 = "\\w \\$ ";
  875. #endif
  876. }
  877. static inline void setup_prompt_string(int promptmode, char **prompt_str)
  878. {
  879. debug_printf("setup_prompt_string %d ",promptmode);
  880. #ifndef CONFIG_FEATURE_SH_FANCY_PROMPT
  881. /* Set up the prompt */
  882. if (promptmode == 1) {
  883. free(PS1);
  884. PS1=xmalloc(strlen(cwd)+4);
  885. sprintf(PS1, "%s %s", cwd, ( geteuid() != 0 ) ? "$ ":"# ");
  886. *prompt_str = PS1;
  887. } else {
  888. *prompt_str = PS2;
  889. }
  890. #else
  891. *prompt_str = (promptmode==1)? PS1 : PS2;
  892. #endif
  893. debug_printf("result %s\n",*prompt_str);
  894. }
  895. #endif
  896. #ifdef __U_BOOT__
  897. static int uboot_cli_readline(struct in_str *i)
  898. {
  899. char *prompt;
  900. char __maybe_unused *ps_prompt = NULL;
  901. if (i->promptmode == 1)
  902. prompt = CONFIG_SYS_PROMPT;
  903. else
  904. prompt = CONFIG_SYS_PROMPT_HUSH_PS2;
  905. #ifdef CONFIG_CMDLINE_PS_SUPPORT
  906. if (i->promptmode == 1)
  907. ps_prompt = env_get("PS1");
  908. else
  909. ps_prompt = env_get("PS2");
  910. if (ps_prompt)
  911. prompt = ps_prompt;
  912. #endif
  913. return cli_readline(prompt);
  914. }
  915. #endif
  916. static void get_user_input(struct in_str *i)
  917. {
  918. #ifndef __U_BOOT__
  919. char *prompt_str;
  920. static char the_command[BUFSIZ];
  921. setup_prompt_string(i->promptmode, &prompt_str);
  922. #ifdef CONFIG_FEATURE_COMMAND_EDITING
  923. /*
  924. ** enable command line editing only while a command line
  925. ** is actually being read; otherwise, we'll end up bequeathing
  926. ** atexit() handlers and other unwanted stuff to our
  927. ** child processes (rob@sysgo.de)
  928. */
  929. cmdedit_read_input(prompt_str, the_command);
  930. #else
  931. fputs(prompt_str, stdout);
  932. fflush(stdout);
  933. the_command[0]=fgetc(i->file);
  934. the_command[1]='\0';
  935. #endif
  936. fflush(stdout);
  937. i->p = the_command;
  938. #else
  939. int n;
  940. static char the_command[CONFIG_SYS_CBSIZE + 1];
  941. bootretry_reset_cmd_timeout();
  942. i->__promptme = 1;
  943. n = uboot_cli_readline(i);
  944. #ifdef CONFIG_BOOT_RETRY_TIME
  945. if (n == -2) {
  946. puts("\nTimeout waiting for command\n");
  947. # ifdef CONFIG_RESET_TO_RETRY
  948. do_reset(NULL, 0, 0, NULL);
  949. # else
  950. # error "This currently only works with CONFIG_RESET_TO_RETRY enabled"
  951. # endif
  952. }
  953. #endif
  954. if (n == -1 ) {
  955. flag_repeat = 0;
  956. i->__promptme = 0;
  957. }
  958. n = strlen(console_buffer);
  959. console_buffer[n] = '\n';
  960. console_buffer[n+1]= '\0';
  961. if (had_ctrlc()) flag_repeat = 0;
  962. clear_ctrlc();
  963. do_repeat = 0;
  964. if (i->promptmode == 1) {
  965. if (console_buffer[0] == '\n'&& flag_repeat == 0) {
  966. strcpy(the_command,console_buffer);
  967. }
  968. else {
  969. if (console_buffer[0] != '\n') {
  970. strcpy(the_command,console_buffer);
  971. flag_repeat = 1;
  972. }
  973. else {
  974. do_repeat = 1;
  975. }
  976. }
  977. i->p = the_command;
  978. }
  979. else {
  980. if (console_buffer[0] != '\n') {
  981. if (strlen(the_command) + strlen(console_buffer)
  982. < CONFIG_SYS_CBSIZE) {
  983. n = strlen(the_command);
  984. the_command[n-1] = ' ';
  985. strcpy(&the_command[n],console_buffer);
  986. }
  987. else {
  988. the_command[0] = '\n';
  989. the_command[1] = '\0';
  990. flag_repeat = 0;
  991. }
  992. }
  993. if (i->__promptme == 0) {
  994. the_command[0] = '\n';
  995. the_command[1] = '\0';
  996. }
  997. i->p = console_buffer;
  998. }
  999. #endif
  1000. }
  1001. /* This is the magic location that prints prompts
  1002. * and gets data back from the user */
  1003. static int file_get(struct in_str *i)
  1004. {
  1005. int ch;
  1006. ch = 0;
  1007. /* If there is data waiting, eat it up */
  1008. if (i->p && *i->p) {
  1009. ch = *i->p++;
  1010. } else {
  1011. /* need to double check i->file because we might be doing something
  1012. * more complicated by now, like sourcing or substituting. */
  1013. #ifndef __U_BOOT__
  1014. if (i->__promptme && interactive && i->file == stdin) {
  1015. while(! i->p || (interactive && strlen(i->p)==0) ) {
  1016. #else
  1017. while(! i->p || strlen(i->p)==0 ) {
  1018. #endif
  1019. get_user_input(i);
  1020. }
  1021. i->promptmode=2;
  1022. #ifndef __U_BOOT__
  1023. i->__promptme = 0;
  1024. #endif
  1025. if (i->p && *i->p) {
  1026. ch = *i->p++;
  1027. }
  1028. #ifndef __U_BOOT__
  1029. } else {
  1030. ch = fgetc(i->file);
  1031. }
  1032. #endif
  1033. debug_printf("b_getch: got a %d\n", ch);
  1034. }
  1035. #ifndef __U_BOOT__
  1036. if (ch == '\n') i->__promptme=1;
  1037. #endif
  1038. return ch;
  1039. }
  1040. /* All the callers guarantee this routine will never be
  1041. * used right after a newline, so prompting is not needed.
  1042. */
  1043. static int file_peek(struct in_str *i)
  1044. {
  1045. #ifndef __U_BOOT__
  1046. if (i->p && *i->p) {
  1047. #endif
  1048. return *i->p;
  1049. #ifndef __U_BOOT__
  1050. } else {
  1051. i->peek_buf[0] = fgetc(i->file);
  1052. i->peek_buf[1] = '\0';
  1053. i->p = i->peek_buf;
  1054. debug_printf("b_peek: got a %d\n", *i->p);
  1055. return *i->p;
  1056. }
  1057. #endif
  1058. }
  1059. #ifndef __U_BOOT__
  1060. static void setup_file_in_str(struct in_str *i, FILE *f)
  1061. #else
  1062. static void setup_file_in_str(struct in_str *i)
  1063. #endif
  1064. {
  1065. i->peek = file_peek;
  1066. i->get = file_get;
  1067. i->__promptme=1;
  1068. i->promptmode=1;
  1069. #ifndef __U_BOOT__
  1070. i->file = f;
  1071. #endif
  1072. i->p = NULL;
  1073. }
  1074. static void setup_string_in_str(struct in_str *i, const char *s)
  1075. {
  1076. i->peek = static_peek;
  1077. i->get = static_get;
  1078. i->__promptme=1;
  1079. i->promptmode=1;
  1080. i->p = s;
  1081. }
  1082. #ifndef __U_BOOT__
  1083. static void mark_open(int fd)
  1084. {
  1085. struct close_me *new = xmalloc(sizeof(struct close_me));
  1086. new->fd = fd;
  1087. new->next = close_me_head;
  1088. close_me_head = new;
  1089. }
  1090. static void mark_closed(int fd)
  1091. {
  1092. struct close_me *tmp;
  1093. if (close_me_head == NULL || close_me_head->fd != fd)
  1094. error_msg_and_die("corrupt close_me");
  1095. tmp = close_me_head;
  1096. close_me_head = close_me_head->next;
  1097. free(tmp);
  1098. }
  1099. static void close_all(void)
  1100. {
  1101. struct close_me *c;
  1102. for (c=close_me_head; c; c=c->next) {
  1103. close(c->fd);
  1104. }
  1105. close_me_head = NULL;
  1106. }
  1107. /* squirrel != NULL means we squirrel away copies of stdin, stdout,
  1108. * and stderr if they are redirected. */
  1109. static int setup_redirects(struct child_prog *prog, int squirrel[])
  1110. {
  1111. int openfd, mode;
  1112. struct redir_struct *redir;
  1113. for (redir=prog->redirects; redir; redir=redir->next) {
  1114. if (redir->dup == -1 && redir->word.gl_pathv == NULL) {
  1115. /* something went wrong in the parse. Pretend it didn't happen */
  1116. continue;
  1117. }
  1118. if (redir->dup == -1) {
  1119. mode=redir_table[redir->type].mode;
  1120. openfd = open(redir->word.gl_pathv[0], mode, 0666);
  1121. if (openfd < 0) {
  1122. /* this could get lost if stderr has been redirected, but
  1123. bash and ash both lose it as well (though zsh doesn't!) */
  1124. perror_msg("error opening %s", redir->word.gl_pathv[0]);
  1125. return 1;
  1126. }
  1127. } else {
  1128. openfd = redir->dup;
  1129. }
  1130. if (openfd != redir->fd) {
  1131. if (squirrel && redir->fd < 3) {
  1132. squirrel[redir->fd] = dup(redir->fd);
  1133. }
  1134. if (openfd == -3) {
  1135. close(openfd);
  1136. } else {
  1137. dup2(openfd, redir->fd);
  1138. if (redir->dup == -1)
  1139. close (openfd);
  1140. }
  1141. }
  1142. }
  1143. return 0;
  1144. }
  1145. static void restore_redirects(int squirrel[])
  1146. {
  1147. int i, fd;
  1148. for (i=0; i<3; i++) {
  1149. fd = squirrel[i];
  1150. if (fd != -1) {
  1151. /* No error checking. I sure wouldn't know what
  1152. * to do with an error if I found one! */
  1153. dup2(fd, i);
  1154. close(fd);
  1155. }
  1156. }
  1157. }
  1158. /* never returns */
  1159. /* XXX no exit() here. If you don't exec, use _exit instead.
  1160. * The at_exit handlers apparently confuse the calling process,
  1161. * in particular stdin handling. Not sure why? */
  1162. static void pseudo_exec(struct child_prog *child)
  1163. {
  1164. int i, rcode;
  1165. char *p;
  1166. struct built_in_command *x;
  1167. if (child->argv) {
  1168. for (i=0; is_assignment(child->argv[i]); i++) {
  1169. debug_printf("pid %d environment modification: %s\n",getpid(),child->argv[i]);
  1170. p = insert_var_value(child->argv[i]);
  1171. putenv(strdup(p));
  1172. if (p != child->argv[i]) free(p);
  1173. }
  1174. child->argv+=i; /* XXX this hack isn't so horrible, since we are about
  1175. to exit, and therefore don't need to keep data
  1176. structures consistent for free() use. */
  1177. /* If a variable is assigned in a forest, and nobody listens,
  1178. * was it ever really set?
  1179. */
  1180. if (child->argv[0] == NULL) {
  1181. _exit(EXIT_SUCCESS);
  1182. }
  1183. /*
  1184. * Check if the command matches any of the builtins.
  1185. * Depending on context, this might be redundant. But it's
  1186. * easier to waste a few CPU cycles than it is to figure out
  1187. * if this is one of those cases.
  1188. */
  1189. for (x = bltins; x->cmd; x++) {
  1190. if (strcmp(child->argv[0], x->cmd) == 0 ) {
  1191. debug_printf("builtin exec %s\n", child->argv[0]);
  1192. rcode = x->function(child);
  1193. fflush(stdout);
  1194. _exit(rcode);
  1195. }
  1196. }
  1197. /* Check if the command matches any busybox internal commands
  1198. * ("applets") here.
  1199. * FIXME: This feature is not 100% safe, since
  1200. * BusyBox is not fully reentrant, so we have no guarantee the things
  1201. * from the .bss are still zeroed, or that things from .data are still
  1202. * at their defaults. We could exec ourself from /proc/self/exe, but I
  1203. * really dislike relying on /proc for things. We could exec ourself
  1204. * from global_argv[0], but if we are in a chroot, we may not be able
  1205. * to find ourself... */
  1206. #ifdef CONFIG_FEATURE_SH_STANDALONE_SHELL
  1207. {
  1208. int argc_l;
  1209. char** argv_l=child->argv;
  1210. char *name = child->argv[0];
  1211. #ifdef CONFIG_FEATURE_SH_APPLETS_ALWAYS_WIN
  1212. /* Following discussions from November 2000 on the busybox mailing
  1213. * list, the default configuration, (without
  1214. * get_last_path_component()) lets the user force use of an
  1215. * external command by specifying the full (with slashes) filename.
  1216. * If you enable CONFIG_FEATURE_SH_APPLETS_ALWAYS_WIN then applets
  1217. * _aways_ override external commands, so if you want to run
  1218. * /bin/cat, it will use BusyBox cat even if /bin/cat exists on the
  1219. * filesystem and is _not_ busybox. Some systems may want this,
  1220. * most do not. */
  1221. name = get_last_path_component(name);
  1222. #endif
  1223. /* Count argc for use in a second... */
  1224. for(argc_l=0;*argv_l!=NULL; argv_l++, argc_l++);
  1225. optind = 1;
  1226. debug_printf("running applet %s\n", name);
  1227. run_applet_by_name(name, argc_l, child->argv);
  1228. }
  1229. #endif
  1230. debug_printf("exec of %s\n",child->argv[0]);
  1231. execvp(child->argv[0],child->argv);
  1232. perror_msg("couldn't exec: %s",child->argv[0]);
  1233. _exit(1);
  1234. } else if (child->group) {
  1235. debug_printf("runtime nesting to group\n");
  1236. interactive=0; /* crucial!!!! */
  1237. rcode = run_list_real(child->group);
  1238. /* OK to leak memory by not calling free_pipe_list,
  1239. * since this process is about to exit */
  1240. _exit(rcode);
  1241. } else {
  1242. /* Can happen. See what bash does with ">foo" by itself. */
  1243. debug_printf("trying to pseudo_exec null command\n");
  1244. _exit(EXIT_SUCCESS);
  1245. }
  1246. }
  1247. static void insert_bg_job(struct pipe *pi)
  1248. {
  1249. struct pipe *thejob;
  1250. /* Linear search for the ID of the job to use */
  1251. pi->jobid = 1;
  1252. for (thejob = job_list; thejob; thejob = thejob->next)
  1253. if (thejob->jobid >= pi->jobid)
  1254. pi->jobid = thejob->jobid + 1;
  1255. /* add thejob to the list of running jobs */
  1256. if (!job_list) {
  1257. thejob = job_list = xmalloc(sizeof(*thejob));
  1258. } else {
  1259. for (thejob = job_list; thejob->next; thejob = thejob->next) /* nothing */;
  1260. thejob->next = xmalloc(sizeof(*thejob));
  1261. thejob = thejob->next;
  1262. }
  1263. /* physically copy the struct job */
  1264. memcpy(thejob, pi, sizeof(struct pipe));
  1265. thejob->next = NULL;
  1266. thejob->running_progs = thejob->num_progs;
  1267. thejob->stopped_progs = 0;
  1268. thejob->text = xmalloc(BUFSIZ); /* cmdedit buffer size */
  1269. /*if (pi->progs[0] && pi->progs[0].argv && pi->progs[0].argv[0]) */
  1270. {
  1271. char *bar=thejob->text;
  1272. char **foo=pi->progs[0].argv;
  1273. while(foo && *foo) {
  1274. bar += sprintf(bar, "%s ", *foo++);
  1275. }
  1276. }
  1277. /* we don't wait for background thejobs to return -- append it
  1278. to the list of backgrounded thejobs and leave it alone */
  1279. printf("[%d] %d\n", thejob->jobid, thejob->progs[0].pid);
  1280. last_bg_pid = thejob->progs[0].pid;
  1281. last_jobid = thejob->jobid;
  1282. }
  1283. /* remove a backgrounded job */
  1284. static void remove_bg_job(struct pipe *pi)
  1285. {
  1286. struct pipe *prev_pipe;
  1287. if (pi == job_list) {
  1288. job_list = pi->next;
  1289. } else {
  1290. prev_pipe = job_list;
  1291. while (prev_pipe->next != pi)
  1292. prev_pipe = prev_pipe->next;
  1293. prev_pipe->next = pi->next;
  1294. }
  1295. if (job_list)
  1296. last_jobid = job_list->jobid;
  1297. else
  1298. last_jobid = 0;
  1299. pi->stopped_progs = 0;
  1300. free_pipe(pi, 0);
  1301. free(pi);
  1302. }
  1303. /* Checks to see if any processes have exited -- if they
  1304. have, figure out why and see if a job has completed */
  1305. static int checkjobs(struct pipe* fg_pipe)
  1306. {
  1307. int attributes;
  1308. int status;
  1309. int prognum = 0;
  1310. struct pipe *pi;
  1311. pid_t childpid;
  1312. attributes = WUNTRACED;
  1313. if (fg_pipe==NULL) {
  1314. attributes |= WNOHANG;
  1315. }
  1316. while ((childpid = waitpid(-1, &status, attributes)) > 0) {
  1317. if (fg_pipe) {
  1318. int i, rcode = 0;
  1319. for (i=0; i < fg_pipe->num_progs; i++) {
  1320. if (fg_pipe->progs[i].pid == childpid) {
  1321. if (i==fg_pipe->num_progs-1)
  1322. rcode=WEXITSTATUS(status);
  1323. (fg_pipe->num_progs)--;
  1324. return(rcode);
  1325. }
  1326. }
  1327. }
  1328. for (pi = job_list; pi; pi = pi->next) {
  1329. prognum = 0;
  1330. while (prognum < pi->num_progs && pi->progs[prognum].pid != childpid) {
  1331. prognum++;
  1332. }
  1333. if (prognum < pi->num_progs)
  1334. break;
  1335. }
  1336. if(pi==NULL) {
  1337. debug_printf("checkjobs: pid %d was not in our list!\n", childpid);
  1338. continue;
  1339. }
  1340. if (WIFEXITED(status) || WIFSIGNALED(status)) {
  1341. /* child exited */
  1342. pi->running_progs--;
  1343. pi->progs[prognum].pid = 0;
  1344. if (!pi->running_progs) {
  1345. printf(JOB_STATUS_FORMAT, pi->jobid, "Done", pi->text);
  1346. remove_bg_job(pi);
  1347. }
  1348. } else {
  1349. /* child stopped */
  1350. pi->stopped_progs++;
  1351. pi->progs[prognum].is_stopped = 1;
  1352. #if 0
  1353. /* Printing this stuff is a pain, since it tends to
  1354. * overwrite the prompt an inconveinient moments. So
  1355. * don't do that. */
  1356. if (pi->stopped_progs == pi->num_progs) {
  1357. printf("\n"JOB_STATUS_FORMAT, pi->jobid, "Stopped", pi->text);
  1358. }
  1359. #endif
  1360. }
  1361. }
  1362. if (childpid == -1 && errno != ECHILD)
  1363. perror_msg("waitpid");
  1364. /* move the shell to the foreground */
  1365. /*if (interactive && tcsetpgrp(shell_terminal, getpgid(0))) */
  1366. /* perror_msg("tcsetpgrp-2"); */
  1367. return -1;
  1368. }
  1369. /* Figure out our controlling tty, checking in order stderr,
  1370. * stdin, and stdout. If check_pgrp is set, also check that
  1371. * we belong to the foreground process group associated with
  1372. * that tty. The value of shell_terminal is needed in order to call
  1373. * tcsetpgrp(shell_terminal, ...); */
  1374. void controlling_tty(int check_pgrp)
  1375. {
  1376. pid_t curpgrp;
  1377. if ((curpgrp = tcgetpgrp(shell_terminal = 2)) < 0
  1378. && (curpgrp = tcgetpgrp(shell_terminal = 0)) < 0
  1379. && (curpgrp = tcgetpgrp(shell_terminal = 1)) < 0)
  1380. goto shell_terminal_error;
  1381. if (check_pgrp && curpgrp != getpgid(0))
  1382. goto shell_terminal_error;
  1383. return;
  1384. shell_terminal_error:
  1385. shell_terminal = -1;
  1386. return;
  1387. }
  1388. #endif
  1389. /* run_pipe_real() starts all the jobs, but doesn't wait for anything
  1390. * to finish. See checkjobs().
  1391. *
  1392. * return code is normally -1, when the caller has to wait for children
  1393. * to finish to determine the exit status of the pipe. If the pipe
  1394. * is a simple builtin command, however, the action is done by the
  1395. * time run_pipe_real returns, and the exit code is provided as the
  1396. * return value.
  1397. *
  1398. * The input of the pipe is always stdin, the output is always
  1399. * stdout. The outpipe[] mechanism in BusyBox-0.48 lash is bogus,
  1400. * because it tries to avoid running the command substitution in
  1401. * subshell, when that is in fact necessary. The subshell process
  1402. * now has its stdout directed to the input of the appropriate pipe,
  1403. * so this routine is noticeably simpler.
  1404. */
  1405. static int run_pipe_real(struct pipe *pi)
  1406. {
  1407. int i;
  1408. #ifndef __U_BOOT__
  1409. int nextin, nextout;
  1410. int pipefds[2]; /* pipefds[0] is for reading */
  1411. struct child_prog *child;
  1412. struct built_in_command *x;
  1413. char *p;
  1414. # if __GNUC__
  1415. /* Avoid longjmp clobbering */
  1416. (void) &i;
  1417. (void) &nextin;
  1418. (void) &nextout;
  1419. (void) &child;
  1420. # endif
  1421. #else
  1422. int nextin;
  1423. int flag = do_repeat ? CMD_FLAG_REPEAT : 0;
  1424. struct child_prog *child;
  1425. char *p;
  1426. # if __GNUC__
  1427. /* Avoid longjmp clobbering */
  1428. (void) &i;
  1429. (void) &nextin;
  1430. (void) &child;
  1431. # endif
  1432. #endif /* __U_BOOT__ */
  1433. nextin = 0;
  1434. #ifndef __U_BOOT__
  1435. pi->pgrp = -1;
  1436. #endif
  1437. /* Check if this is a simple builtin (not part of a pipe).
  1438. * Builtins within pipes have to fork anyway, and are handled in
  1439. * pseudo_exec. "echo foo | read bar" doesn't work on bash, either.
  1440. */
  1441. if (pi->num_progs == 1) child = & (pi->progs[0]);
  1442. #ifndef __U_BOOT__
  1443. if (pi->num_progs == 1 && child->group && child->subshell == 0) {
  1444. int squirrel[] = {-1, -1, -1};
  1445. int rcode;
  1446. debug_printf("non-subshell grouping\n");
  1447. setup_redirects(child, squirrel);
  1448. /* XXX could we merge code with following builtin case,
  1449. * by creating a pseudo builtin that calls run_list_real? */
  1450. rcode = run_list_real(child->group);
  1451. restore_redirects(squirrel);
  1452. #else
  1453. if (pi->num_progs == 1 && child->group) {
  1454. int rcode;
  1455. debug_printf("non-subshell grouping\n");
  1456. rcode = run_list_real(child->group);
  1457. #endif
  1458. return rcode;
  1459. } else if (pi->num_progs == 1 && pi->progs[0].argv != NULL) {
  1460. for (i=0; is_assignment(child->argv[i]); i++) { /* nothing */ }
  1461. if (i!=0 && child->argv[i]==NULL) {
  1462. /* assignments, but no command: set the local environment */
  1463. for (i=0; child->argv[i]!=NULL; i++) {
  1464. /* Ok, this case is tricky. We have to decide if this is a
  1465. * local variable, or an already exported variable. If it is
  1466. * already exported, we have to export the new value. If it is
  1467. * not exported, we need only set this as a local variable.
  1468. * This junk is all to decide whether or not to export this
  1469. * variable. */
  1470. int export_me=0;
  1471. char *name, *value;
  1472. name = xstrdup(child->argv[i]);
  1473. debug_printf("Local environment set: %s\n", name);
  1474. value = strchr(name, '=');
  1475. if (value)
  1476. *value=0;
  1477. #ifndef __U_BOOT__
  1478. if ( get_local_var(name)) {
  1479. export_me=1;
  1480. }
  1481. #endif
  1482. free(name);
  1483. p = insert_var_value(child->argv[i]);
  1484. set_local_var(p, export_me);
  1485. if (p != child->argv[i]) free(p);
  1486. }
  1487. return EXIT_SUCCESS; /* don't worry about errors in set_local_var() yet */
  1488. }
  1489. for (i = 0; is_assignment(child->argv[i]); i++) {
  1490. p = insert_var_value(child->argv[i]);
  1491. #ifndef __U_BOOT__
  1492. putenv(strdup(p));
  1493. #else
  1494. set_local_var(p, 0);
  1495. #endif
  1496. if (p != child->argv[i]) {
  1497. child->sp--;
  1498. free(p);
  1499. }
  1500. }
  1501. if (child->sp) {
  1502. char * str = NULL;
  1503. str = make_string(child->argv + i,
  1504. child->argv_nonnull + i);
  1505. parse_string_outer(str, FLAG_EXIT_FROM_LOOP | FLAG_REPARSING);
  1506. free(str);
  1507. return last_return_code;
  1508. }
  1509. #ifndef __U_BOOT__
  1510. for (x = bltins; x->cmd; x++) {
  1511. if (strcmp(child->argv[i], x->cmd) == 0 ) {
  1512. int squirrel[] = {-1, -1, -1};
  1513. int rcode;
  1514. if (x->function == builtin_exec && child->argv[i+1]==NULL) {
  1515. debug_printf("magic exec\n");
  1516. setup_redirects(child,NULL);
  1517. return EXIT_SUCCESS;
  1518. }
  1519. debug_printf("builtin inline %s\n", child->argv[0]);
  1520. /* XXX setup_redirects acts on file descriptors, not FILEs.
  1521. * This is perfect for work that comes after exec().
  1522. * Is it really safe for inline use? Experimentally,
  1523. * things seem to work with glibc. */
  1524. setup_redirects(child, squirrel);
  1525. child->argv += i; /* XXX horrible hack */
  1526. rcode = x->function(child);
  1527. /* XXX restore hack so free() can work right */
  1528. child->argv -= i;
  1529. restore_redirects(squirrel);
  1530. }
  1531. return rcode;
  1532. }
  1533. #else
  1534. /* check ";", because ,example , argv consist from
  1535. * "help;flinfo" must not execute
  1536. */
  1537. if (strchr(child->argv[i], ';')) {
  1538. printf("Unknown command '%s' - try 'help' or use "
  1539. "'run' command\n", child->argv[i]);
  1540. return -1;
  1541. }
  1542. /* Process the command */
  1543. return cmd_process(flag, child->argc, child->argv,
  1544. &flag_repeat, NULL);
  1545. #endif
  1546. }
  1547. #ifndef __U_BOOT__
  1548. for (i = 0; i < pi->num_progs; i++) {
  1549. child = & (pi->progs[i]);
  1550. /* pipes are inserted between pairs of commands */
  1551. if ((i + 1) < pi->num_progs) {
  1552. if (pipe(pipefds)<0) perror_msg_and_die("pipe");
  1553. nextout = pipefds[1];
  1554. } else {
  1555. nextout=1;
  1556. pipefds[0] = -1;
  1557. }
  1558. /* XXX test for failed fork()? */
  1559. if (!(child->pid = fork())) {
  1560. /* Set the handling for job control signals back to the default. */
  1561. signal(SIGINT, SIG_DFL);
  1562. signal(SIGQUIT, SIG_DFL);
  1563. signal(SIGTERM, SIG_DFL);
  1564. signal(SIGTSTP, SIG_DFL);
  1565. signal(SIGTTIN, SIG_DFL);
  1566. signal(SIGTTOU, SIG_DFL);
  1567. signal(SIGCHLD, SIG_DFL);
  1568. close_all();
  1569. if (nextin != 0) {
  1570. dup2(nextin, 0);
  1571. close(nextin);
  1572. }
  1573. if (nextout != 1) {
  1574. dup2(nextout, 1);
  1575. close(nextout);
  1576. }
  1577. if (pipefds[0]!=-1) {
  1578. close(pipefds[0]); /* opposite end of our output pipe */
  1579. }
  1580. /* Like bash, explicit redirects override pipes,
  1581. * and the pipe fd is available for dup'ing. */
  1582. setup_redirects(child,NULL);
  1583. if (interactive && pi->followup!=PIPE_BG) {
  1584. /* If we (the child) win the race, put ourselves in the process
  1585. * group whose leader is the first process in this pipe. */
  1586. if (pi->pgrp < 0) {
  1587. pi->pgrp = getpid();
  1588. }
  1589. if (setpgid(0, pi->pgrp) == 0) {
  1590. tcsetpgrp(2, pi->pgrp);
  1591. }
  1592. }
  1593. pseudo_exec(child);
  1594. }
  1595. /* put our child in the process group whose leader is the
  1596. first process in this pipe */
  1597. if (pi->pgrp < 0) {
  1598. pi->pgrp = child->pid;
  1599. }
  1600. /* Don't check for errors. The child may be dead already,
  1601. * in which case setpgid returns error code EACCES. */
  1602. setpgid(child->pid, pi->pgrp);
  1603. if (nextin != 0)
  1604. close(nextin);
  1605. if (nextout != 1)
  1606. close(nextout);
  1607. /* If there isn't another process, nextin is garbage
  1608. but it doesn't matter */
  1609. nextin = pipefds[0];
  1610. }
  1611. #endif
  1612. return -1;
  1613. }
  1614. static int run_list_real(struct pipe *pi)
  1615. {
  1616. char *save_name = NULL;
  1617. char **list = NULL;
  1618. char **save_list = NULL;
  1619. struct pipe *rpipe;
  1620. int flag_rep = 0;
  1621. #ifndef __U_BOOT__
  1622. int save_num_progs;
  1623. #endif
  1624. int rcode=0, flag_skip=1;
  1625. int flag_restore = 0;
  1626. int if_code=0, next_if_code=0; /* need double-buffer to handle elif */
  1627. reserved_style rmode, skip_more_in_this_rmode=RES_XXXX;
  1628. /* check syntax for "for" */
  1629. for (rpipe = pi; rpipe; rpipe = rpipe->next) {
  1630. if ((rpipe->r_mode == RES_IN ||
  1631. rpipe->r_mode == RES_FOR) &&
  1632. (rpipe->next == NULL)) {
  1633. syntax();
  1634. #ifdef __U_BOOT__
  1635. flag_repeat = 0;
  1636. #endif
  1637. return 1;
  1638. }
  1639. if ((rpipe->r_mode == RES_IN &&
  1640. (rpipe->next->r_mode == RES_IN &&
  1641. rpipe->next->progs->argv != NULL))||
  1642. (rpipe->r_mode == RES_FOR &&
  1643. rpipe->next->r_mode != RES_IN)) {
  1644. syntax();
  1645. #ifdef __U_BOOT__
  1646. flag_repeat = 0;
  1647. #endif
  1648. return 1;
  1649. }
  1650. }
  1651. for (; pi; pi = (flag_restore != 0) ? rpipe : pi->next) {
  1652. if (pi->r_mode == RES_WHILE || pi->r_mode == RES_UNTIL ||
  1653. pi->r_mode == RES_FOR) {
  1654. #ifdef __U_BOOT__
  1655. /* check Ctrl-C */
  1656. ctrlc();
  1657. if ((had_ctrlc())) {
  1658. return 1;
  1659. }
  1660. #endif
  1661. flag_restore = 0;
  1662. if (!rpipe) {
  1663. flag_rep = 0;
  1664. rpipe = pi;
  1665. }
  1666. }
  1667. rmode = pi->r_mode;
  1668. debug_printf("rmode=%d if_code=%d next_if_code=%d skip_more=%d\n", rmode, if_code, next_if_code, skip_more_in_this_rmode);
  1669. if (rmode == skip_more_in_this_rmode && flag_skip) {
  1670. if (pi->followup == PIPE_SEQ) flag_skip=0;
  1671. continue;
  1672. }
  1673. flag_skip = 1;
  1674. skip_more_in_this_rmode = RES_XXXX;
  1675. if (rmode == RES_THEN || rmode == RES_ELSE) if_code = next_if_code;
  1676. if (rmode == RES_THEN && if_code) continue;
  1677. if (rmode == RES_ELSE && !if_code) continue;
  1678. if (rmode == RES_ELIF && !if_code) break;
  1679. if (rmode == RES_FOR && pi->num_progs) {
  1680. if (!list) {
  1681. /* if no variable values after "in" we skip "for" */
  1682. if (!pi->next->progs->argv) continue;
  1683. /* create list of variable values */
  1684. list = make_list_in(pi->next->progs->argv,
  1685. pi->progs->argv[0]);
  1686. save_list = list;
  1687. save_name = pi->progs->argv[0];
  1688. pi->progs->argv[0] = NULL;
  1689. flag_rep = 1;
  1690. }
  1691. if (!(*list)) {
  1692. free(pi->progs->argv[0]);
  1693. free(save_list);
  1694. list = NULL;
  1695. flag_rep = 0;
  1696. pi->progs->argv[0] = save_name;
  1697. #ifndef __U_BOOT__
  1698. pi->progs->glob_result.gl_pathv[0] =
  1699. pi->progs->argv[0];
  1700. #endif
  1701. continue;
  1702. } else {
  1703. /* insert new value from list for variable */
  1704. free(pi->progs->argv[0]);
  1705. pi->progs->argv[0] = *list++;
  1706. #ifndef __U_BOOT__
  1707. pi->progs->glob_result.gl_pathv[0] =
  1708. pi->progs->argv[0];
  1709. #endif
  1710. }
  1711. }
  1712. if (rmode == RES_IN) continue;
  1713. if (rmode == RES_DO) {
  1714. if (!flag_rep) continue;
  1715. }
  1716. if (rmode == RES_DONE) {
  1717. if (flag_rep) {
  1718. flag_restore = 1;
  1719. } else {
  1720. rpipe = NULL;
  1721. }
  1722. }
  1723. if (pi->num_progs == 0) continue;
  1724. #ifndef __U_BOOT__
  1725. save_num_progs = pi->num_progs; /* save number of programs */
  1726. #endif
  1727. rcode = run_pipe_real(pi);
  1728. debug_printf("run_pipe_real returned %d\n",rcode);
  1729. #ifndef __U_BOOT__
  1730. if (rcode!=-1) {
  1731. /* We only ran a builtin: rcode was set by the return value
  1732. * of run_pipe_real(), and we don't need to wait for anything. */
  1733. } else if (pi->followup==PIPE_BG) {
  1734. /* XXX check bash's behavior with nontrivial pipes */
  1735. /* XXX compute jobid */
  1736. /* XXX what does bash do with attempts to background builtins? */
  1737. insert_bg_job(pi);
  1738. rcode = EXIT_SUCCESS;
  1739. } else {
  1740. if (interactive) {
  1741. /* move the new process group into the foreground */
  1742. if (tcsetpgrp(shell_terminal, pi->pgrp) && errno != ENOTTY)
  1743. perror_msg("tcsetpgrp-3");
  1744. rcode = checkjobs(pi);
  1745. /* move the shell to the foreground */
  1746. if (tcsetpgrp(shell_terminal, getpgid(0)) && errno != ENOTTY)
  1747. perror_msg("tcsetpgrp-4");
  1748. } else {
  1749. rcode = checkjobs(pi);
  1750. }
  1751. debug_printf("checkjobs returned %d\n",rcode);
  1752. }
  1753. last_return_code=rcode;
  1754. #else
  1755. if (rcode < -1) {
  1756. last_return_code = -rcode - 2;
  1757. return -2; /* exit */
  1758. }
  1759. last_return_code=(rcode == 0) ? 0 : 1;
  1760. #endif
  1761. #ifndef __U_BOOT__
  1762. pi->num_progs = save_num_progs; /* restore number of programs */
  1763. #endif
  1764. if ( rmode == RES_IF || rmode == RES_ELIF )
  1765. next_if_code=rcode; /* can be overwritten a number of times */
  1766. if (rmode == RES_WHILE)
  1767. flag_rep = !last_return_code;
  1768. if (rmode == RES_UNTIL)
  1769. flag_rep = last_return_code;
  1770. if ( (rcode==EXIT_SUCCESS && pi->followup==PIPE_OR) ||
  1771. (rcode!=EXIT_SUCCESS && pi->followup==PIPE_AND) )
  1772. skip_more_in_this_rmode=rmode;
  1773. #ifndef __U_BOOT__
  1774. checkjobs(NULL);
  1775. #endif
  1776. }
  1777. return rcode;
  1778. }
  1779. /* broken, of course, but OK for testing */
  1780. static char *indenter(int i)
  1781. {
  1782. static char blanks[]=" ";
  1783. return &blanks[sizeof(blanks)-i-1];
  1784. }
  1785. /* return code is the exit status of the pipe */
  1786. static int free_pipe(struct pipe *pi, int indent)
  1787. {
  1788. char **p;
  1789. struct child_prog *child;
  1790. #ifndef __U_BOOT__
  1791. struct redir_struct *r, *rnext;
  1792. #endif
  1793. int a, i, ret_code=0;
  1794. char *ind = indenter(indent);
  1795. #ifndef __U_BOOT__
  1796. if (pi->stopped_progs > 0)
  1797. return ret_code;
  1798. final_printf("%s run pipe: (pid %d)\n",ind,getpid());
  1799. #endif
  1800. for (i=0; i<pi->num_progs; i++) {
  1801. child = &pi->progs[i];
  1802. final_printf("%s command %d:\n",ind,i);
  1803. if (child->argv) {
  1804. for (a=0,p=child->argv; *p; a++,p++) {
  1805. final_printf("%s argv[%d] = %s\n",ind,a,*p);
  1806. }
  1807. #ifndef __U_BOOT__
  1808. globfree(&child->glob_result);
  1809. #else
  1810. for (a = 0; a < child->argc; a++) {
  1811. free(child->argv[a]);
  1812. }
  1813. free(child->argv);
  1814. free(child->argv_nonnull);
  1815. child->argc = 0;
  1816. #endif
  1817. child->argv=NULL;
  1818. } else if (child->group) {
  1819. #ifndef __U_BOOT__
  1820. final_printf("%s begin group (subshell:%d)\n",ind, child->subshell);
  1821. #endif
  1822. ret_code = free_pipe_list(child->group,indent+3);
  1823. final_printf("%s end group\n",ind);
  1824. } else {
  1825. final_printf("%s (nil)\n",ind);
  1826. }
  1827. #ifndef __U_BOOT__
  1828. for (r=child->redirects; r; r=rnext) {
  1829. final_printf("%s redirect %d%s", ind, r->fd, redir_table[r->type].descrip);
  1830. if (r->dup == -1) {
  1831. /* guard against the case >$FOO, where foo is unset or blank */
  1832. if (r->word.gl_pathv) {
  1833. final_printf(" %s\n", *r->word.gl_pathv);
  1834. globfree(&r->word);
  1835. }
  1836. } else {
  1837. final_printf("&%d\n", r->dup);
  1838. }
  1839. rnext=r->next;
  1840. free(r);
  1841. }
  1842. child->redirects=NULL;
  1843. #endif
  1844. }
  1845. free(pi->progs); /* children are an array, they get freed all at once */
  1846. pi->progs=NULL;
  1847. return ret_code;
  1848. }
  1849. static int free_pipe_list(struct pipe *head, int indent)
  1850. {
  1851. int rcode=0; /* if list has no members */
  1852. struct pipe *pi, *next;
  1853. char *ind = indenter(indent);
  1854. for (pi=head; pi; pi=next) {
  1855. final_printf("%s pipe reserved mode %d\n", ind, pi->r_mode);
  1856. rcode = free_pipe(pi, indent);
  1857. final_printf("%s pipe followup code %d\n", ind, pi->followup);
  1858. next=pi->next;
  1859. pi->next=NULL;
  1860. free(pi);
  1861. }
  1862. return rcode;
  1863. }
  1864. /* Select which version we will use */
  1865. static int run_list(struct pipe *pi)
  1866. {
  1867. int rcode=0;
  1868. #ifndef __U_BOOT__
  1869. if (fake_mode==0) {
  1870. #endif
  1871. rcode = run_list_real(pi);
  1872. #ifndef __U_BOOT__
  1873. }
  1874. #endif
  1875. /* free_pipe_list has the side effect of clearing memory
  1876. * In the long run that function can be merged with run_list_real,
  1877. * but doing that now would hobble the debugging effort. */
  1878. free_pipe_list(pi,0);
  1879. return rcode;
  1880. }
  1881. /* The API for glob is arguably broken. This routine pushes a non-matching
  1882. * string into the output structure, removing non-backslashed backslashes.
  1883. * If someone can prove me wrong, by performing this function within the
  1884. * original glob(3) api, feel free to rewrite this routine into oblivion.
  1885. * Return code (0 vs. GLOB_NOSPACE) matches glob(3).
  1886. * XXX broken if the last character is '\\', check that before calling.
  1887. */
  1888. #ifndef __U_BOOT__
  1889. static int globhack(const char *src, int flags, glob_t *pglob)
  1890. {
  1891. int cnt=0, pathc;
  1892. const char *s;
  1893. char *dest;
  1894. for (cnt=1, s=src; s && *s; s++) {
  1895. if (*s == '\\') s++;
  1896. cnt++;
  1897. }
  1898. dest = malloc(cnt);
  1899. if (!dest) return GLOB_NOSPACE;
  1900. if (!(flags & GLOB_APPEND)) {
  1901. pglob->gl_pathv=NULL;
  1902. pglob->gl_pathc=0;
  1903. pglob->gl_offs=0;
  1904. pglob->gl_offs=0;
  1905. }
  1906. pathc = ++pglob->gl_pathc;
  1907. pglob->gl_pathv = realloc(pglob->gl_pathv, (pathc+1)*sizeof(*pglob->gl_pathv));
  1908. if (pglob->gl_pathv == NULL) return GLOB_NOSPACE;
  1909. pglob->gl_pathv[pathc-1]=dest;
  1910. pglob->gl_pathv[pathc]=NULL;
  1911. for (s=src; s && *s; s++, dest++) {
  1912. if (*s == '\\') s++;
  1913. *dest = *s;
  1914. }
  1915. *dest='\0';
  1916. return 0;
  1917. }
  1918. /* XXX broken if the last character is '\\', check that before calling */
  1919. static int glob_needed(const char *s)
  1920. {
  1921. for (; *s; s++) {
  1922. if (*s == '\\') s++;
  1923. if (strchr("*[?",*s)) return 1;
  1924. }
  1925. return 0;
  1926. }
  1927. #if 0
  1928. static void globprint(glob_t *pglob)
  1929. {
  1930. int i;
  1931. debug_printf("glob_t at %p:\n", pglob);
  1932. debug_printf(" gl_pathc=%d gl_pathv=%p gl_offs=%d gl_flags=%d\n",
  1933. pglob->gl_pathc, pglob->gl_pathv, pglob->gl_offs, pglob->gl_flags);
  1934. for (i=0; i<pglob->gl_pathc; i++)
  1935. debug_printf("pglob->gl_pathv[%d] = %p = %s\n", i,
  1936. pglob->gl_pathv[i], pglob->gl_pathv[i]);
  1937. }
  1938. #endif
  1939. static int xglob(o_string *dest, int flags, glob_t *pglob)
  1940. {
  1941. int gr;
  1942. /* short-circuit for null word */
  1943. /* we can code this better when the debug_printf's are gone */
  1944. if (dest->length == 0) {
  1945. if (dest->nonnull) {
  1946. /* bash man page calls this an "explicit" null */
  1947. gr = globhack(dest->data, flags, pglob);
  1948. debug_printf("globhack returned %d\n",gr);
  1949. } else {
  1950. return 0;
  1951. }
  1952. } else if (glob_needed(dest->data)) {
  1953. gr = glob(dest->data, flags, NULL, pglob);
  1954. debug_printf("glob returned %d\n",gr);
  1955. if (gr == GLOB_NOMATCH) {
  1956. /* quote removal, or more accurately, backslash removal */
  1957. gr = globhack(dest->data, flags, pglob);
  1958. debug_printf("globhack returned %d\n",gr);
  1959. }
  1960. } else {
  1961. gr = globhack(dest->data, flags, pglob);
  1962. debug_printf("globhack returned %d\n",gr);
  1963. }
  1964. if (gr == GLOB_NOSPACE)
  1965. error_msg_and_die("out of memory during glob");
  1966. if (gr != 0) { /* GLOB_ABORTED ? */
  1967. error_msg("glob(3) error %d",gr);
  1968. }
  1969. /* globprint(glob_target); */
  1970. return gr;
  1971. }
  1972. #endif
  1973. #ifdef __U_BOOT__
  1974. static char *get_dollar_var(char ch);
  1975. #endif
  1976. /* This is used to get/check local shell variables */
  1977. char *get_local_var(const char *s)
  1978. {
  1979. struct variables *cur;
  1980. if (!s)
  1981. return NULL;
  1982. #ifdef __U_BOOT__
  1983. if (*s == '$')
  1984. return get_dollar_var(s[1]);
  1985. #endif
  1986. for (cur = top_vars; cur; cur=cur->next)
  1987. if(strcmp(cur->name, s)==0)
  1988. return cur->value;
  1989. return NULL;
  1990. }
  1991. /* This is used to set local shell variables
  1992. flg_export==0 if only local (not exporting) variable
  1993. flg_export==1 if "new" exporting environ
  1994. flg_export>1 if current startup environ (not call putenv()) */
  1995. int set_local_var(const char *s, int flg_export)
  1996. {
  1997. char *name, *value;
  1998. int result=0;
  1999. struct variables *cur;
  2000. #ifdef __U_BOOT__
  2001. /* might be possible! */
  2002. if (!isalpha(*s))
  2003. return -1;
  2004. #endif
  2005. name=strdup(s);
  2006. #ifdef __U_BOOT__
  2007. if (env_get(name) != NULL) {
  2008. printf ("ERROR: "
  2009. "There is a global environment variable with the same name.\n");
  2010. free(name);
  2011. return -1;
  2012. }
  2013. #endif
  2014. /* Assume when we enter this function that we are already in
  2015. * NAME=VALUE format. So the first order of business is to
  2016. * split 's' on the '=' into 'name' and 'value' */
  2017. value = strchr(name, '=');
  2018. if (value == NULL || *(value + 1) == 0) {
  2019. free(name);
  2020. return -1;
  2021. }
  2022. *value++ = 0;
  2023. for(cur = top_vars; cur; cur = cur->next) {
  2024. if(strcmp(cur->name, name)==0)
  2025. break;
  2026. }
  2027. if(cur) {
  2028. if(strcmp(cur->value, value)==0) {
  2029. if(flg_export>0 && cur->flg_export==0)
  2030. cur->flg_export=flg_export;
  2031. else
  2032. result++;
  2033. } else {
  2034. if(cur->flg_read_only) {
  2035. error_msg("%s: readonly variable", name);
  2036. result = -1;
  2037. } else {
  2038. if(flg_export>0 || cur->flg_export>1)
  2039. cur->flg_export=1;
  2040. free(cur->value);
  2041. cur->value = strdup(value);
  2042. }
  2043. }
  2044. } else {
  2045. cur = malloc(sizeof(struct variables));
  2046. if(!cur) {
  2047. result = -1;
  2048. } else {
  2049. cur->name = strdup(name);
  2050. if (cur->name == NULL) {
  2051. free(cur);
  2052. result = -1;
  2053. } else {
  2054. struct variables *bottom = top_vars;
  2055. cur->value = strdup(value);
  2056. cur->next = NULL;
  2057. cur->flg_export = flg_export;
  2058. cur->flg_read_only = 0;
  2059. while(bottom->next) bottom=bottom->next;
  2060. bottom->next = cur;
  2061. }
  2062. }
  2063. }
  2064. #ifndef __U_BOOT__
  2065. if(result==0 && cur->flg_export==1) {
  2066. *(value-1) = '=';
  2067. result = putenv(name);
  2068. } else {
  2069. #endif
  2070. free(name);
  2071. #ifndef __U_BOOT__
  2072. if(result>0) /* equivalent to previous set */
  2073. result = 0;
  2074. }
  2075. #endif
  2076. return result;
  2077. }
  2078. void unset_local_var(const char *name)
  2079. {
  2080. struct variables *cur;
  2081. if (name) {
  2082. for (cur = top_vars; cur; cur=cur->next) {
  2083. if(strcmp(cur->name, name)==0)
  2084. break;
  2085. }
  2086. if (cur != NULL) {
  2087. struct variables *next = top_vars;
  2088. if(cur->flg_read_only) {
  2089. error_msg("%s: readonly variable", name);
  2090. return;
  2091. } else {
  2092. #ifndef __U_BOOT__
  2093. if(cur->flg_export)
  2094. unenv_set(cur->name);
  2095. #endif
  2096. free(cur->name);
  2097. free(cur->value);
  2098. while (next->next != cur)
  2099. next = next->next;
  2100. next->next = cur->next;
  2101. }
  2102. free(cur);
  2103. }
  2104. }
  2105. }
  2106. static int is_assignment(const char *s)
  2107. {
  2108. if (s == NULL)
  2109. return 0;
  2110. if (!isalpha(*s)) return 0;
  2111. ++s;
  2112. while(isalnum(*s) || *s=='_') ++s;
  2113. return *s=='=';
  2114. }
  2115. #ifndef __U_BOOT__
  2116. /* the src parameter allows us to peek forward to a possible &n syntax
  2117. * for file descriptor duplication, e.g., "2>&1".
  2118. * Return code is 0 normally, 1 if a syntax error is detected in src.
  2119. * Resource errors (in xmalloc) cause the process to exit */
  2120. static int setup_redirect(struct p_context *ctx, int fd, redir_type style,
  2121. struct in_str *input)
  2122. {
  2123. struct child_prog *child=ctx->child;
  2124. struct redir_struct *redir = child->redirects;
  2125. struct redir_struct *last_redir=NULL;
  2126. /* Create a new redir_struct and drop it onto the end of the linked list */
  2127. while(redir) {
  2128. last_redir=redir;
  2129. redir=redir->next;
  2130. }
  2131. redir = xmalloc(sizeof(struct redir_struct));
  2132. redir->next=NULL;
  2133. redir->word.gl_pathv=NULL;
  2134. if (last_redir) {
  2135. last_redir->next=redir;
  2136. } else {
  2137. child->redirects=redir;
  2138. }
  2139. redir->type=style;
  2140. redir->fd= (fd==-1) ? redir_table[style].default_fd : fd ;
  2141. debug_printf("Redirect type %d%s\n", redir->fd, redir_table[style].descrip);
  2142. /* Check for a '2>&1' type redirect */
  2143. redir->dup = redirect_dup_num(input);
  2144. if (redir->dup == -2) return 1; /* syntax error */
  2145. if (redir->dup != -1) {
  2146. /* Erik had a check here that the file descriptor in question
  2147. * is legit; I postpone that to "run time"
  2148. * A "-" representation of "close me" shows up as a -3 here */
  2149. debug_printf("Duplicating redirect '%d>&%d'\n", redir->fd, redir->dup);
  2150. } else {
  2151. /* We do _not_ try to open the file that src points to,
  2152. * since we need to return and let src be expanded first.
  2153. * Set ctx->pending_redirect, so we know what to do at the
  2154. * end of the next parsed word.
  2155. */
  2156. ctx->pending_redirect = redir;
  2157. }
  2158. return 0;
  2159. }
  2160. #endif
  2161. static struct pipe *new_pipe(void)
  2162. {
  2163. struct pipe *pi;
  2164. pi = xmalloc(sizeof(struct pipe));
  2165. pi->num_progs = 0;
  2166. pi->progs = NULL;
  2167. pi->next = NULL;
  2168. pi->followup = 0; /* invalid */
  2169. pi->r_mode = RES_NONE;
  2170. return pi;
  2171. }
  2172. static void initialize_context(struct p_context *ctx)
  2173. {
  2174. ctx->pipe=NULL;
  2175. #ifndef __U_BOOT__
  2176. ctx->pending_redirect=NULL;
  2177. #endif
  2178. ctx->child=NULL;
  2179. ctx->list_head=new_pipe();
  2180. ctx->pipe=ctx->list_head;
  2181. ctx->w=RES_NONE;
  2182. ctx->stack=NULL;
  2183. #ifdef __U_BOOT__
  2184. ctx->old_flag=0;
  2185. #endif
  2186. done_command(ctx); /* creates the memory for working child */
  2187. }
  2188. /* normal return is 0
  2189. * if a reserved word is found, and processed, return 1
  2190. * should handle if, then, elif, else, fi, for, while, until, do, done.
  2191. * case, function, and select are obnoxious, save those for later.
  2192. */
  2193. struct reserved_combo {
  2194. char *literal;
  2195. int code;
  2196. long flag;
  2197. };
  2198. /* Mostly a list of accepted follow-up reserved words.
  2199. * FLAG_END means we are done with the sequence, and are ready
  2200. * to turn the compound list into a command.
  2201. * FLAG_START means the word must start a new compound list.
  2202. */
  2203. static struct reserved_combo reserved_list[] = {
  2204. { "if", RES_IF, FLAG_THEN | FLAG_START },
  2205. { "then", RES_THEN, FLAG_ELIF | FLAG_ELSE | FLAG_FI },
  2206. { "elif", RES_ELIF, FLAG_THEN },
  2207. { "else", RES_ELSE, FLAG_FI },
  2208. { "fi", RES_FI, FLAG_END },
  2209. { "for", RES_FOR, FLAG_IN | FLAG_START },
  2210. { "while", RES_WHILE, FLAG_DO | FLAG_START },
  2211. { "until", RES_UNTIL, FLAG_DO | FLAG_START },
  2212. { "in", RES_IN, FLAG_DO },
  2213. { "do", RES_DO, FLAG_DONE },
  2214. { "done", RES_DONE, FLAG_END }
  2215. };
  2216. #define NRES (sizeof(reserved_list)/sizeof(struct reserved_combo))
  2217. static int reserved_word(o_string *dest, struct p_context *ctx)
  2218. {
  2219. struct reserved_combo *r;
  2220. for (r=reserved_list;
  2221. r<reserved_list+NRES; r++) {
  2222. if (strcmp(dest->data, r->literal) == 0) {
  2223. debug_printf("found reserved word %s, code %d\n",r->literal,r->code);
  2224. if (r->flag & FLAG_START) {
  2225. struct p_context *new = xmalloc(sizeof(struct p_context));
  2226. debug_printf("push stack\n");
  2227. if (ctx->w == RES_IN || ctx->w == RES_FOR) {
  2228. syntax();
  2229. free(new);
  2230. ctx->w = RES_SNTX;
  2231. b_reset(dest);
  2232. return 1;
  2233. }
  2234. *new = *ctx; /* physical copy */
  2235. initialize_context(ctx);
  2236. ctx->stack=new;
  2237. } else if ( ctx->w == RES_NONE || ! (ctx->old_flag & (1<<r->code))) {
  2238. syntax();
  2239. ctx->w = RES_SNTX;
  2240. b_reset(dest);
  2241. return 1;
  2242. }
  2243. ctx->w=r->code;
  2244. ctx->old_flag = r->flag;
  2245. if (ctx->old_flag & FLAG_END) {
  2246. struct p_context *old;
  2247. debug_printf("pop stack\n");
  2248. done_pipe(ctx,PIPE_SEQ);
  2249. old = ctx->stack;
  2250. old->child->group = ctx->list_head;
  2251. #ifndef __U_BOOT__
  2252. old->child->subshell = 0;
  2253. #endif
  2254. *ctx = *old; /* physical copy */
  2255. free(old);
  2256. }
  2257. b_reset (dest);
  2258. return 1;
  2259. }
  2260. }
  2261. return 0;
  2262. }
  2263. /* normal return is 0.
  2264. * Syntax or xglob errors return 1. */
  2265. static int done_word(o_string *dest, struct p_context *ctx)
  2266. {
  2267. struct child_prog *child=ctx->child;
  2268. #ifndef __U_BOOT__
  2269. glob_t *glob_target;
  2270. int gr, flags = 0;
  2271. #else
  2272. char *str, *s;
  2273. int argc, cnt;
  2274. #endif
  2275. debug_printf("done_word: %s %p\n", dest->data, child);
  2276. if (dest->length == 0 && !dest->nonnull) {
  2277. debug_printf(" true null, ignored\n");
  2278. return 0;
  2279. }
  2280. #ifndef __U_BOOT__
  2281. if (ctx->pending_redirect) {
  2282. glob_target = &ctx->pending_redirect->word;
  2283. } else {
  2284. #endif
  2285. if (child->group) {
  2286. syntax();
  2287. return 1; /* syntax error, groups and arglists don't mix */
  2288. }
  2289. if (!child->argv && (ctx->type & FLAG_PARSE_SEMICOLON)) {
  2290. debug_printf("checking %s for reserved-ness\n",dest->data);
  2291. if (reserved_word(dest,ctx)) return ctx->w==RES_SNTX;
  2292. }
  2293. #ifndef __U_BOOT__
  2294. glob_target = &child->glob_result;
  2295. if (child->argv) flags |= GLOB_APPEND;
  2296. #else
  2297. for (cnt = 1, s = dest->data; s && *s; s++) {
  2298. if (*s == '\\') s++;
  2299. cnt++;
  2300. }
  2301. str = malloc(cnt);
  2302. if (!str) return 1;
  2303. if ( child->argv == NULL) {
  2304. child->argc=0;
  2305. }
  2306. argc = ++child->argc;
  2307. child->argv = realloc(child->argv, (argc+1)*sizeof(*child->argv));
  2308. if (child->argv == NULL) {
  2309. free(str);
  2310. return 1;
  2311. }
  2312. child->argv_nonnull = realloc(child->argv_nonnull,
  2313. (argc+1)*sizeof(*child->argv_nonnull));
  2314. if (child->argv_nonnull == NULL) {
  2315. free(str);
  2316. return 1;
  2317. }
  2318. child->argv[argc-1]=str;
  2319. child->argv_nonnull[argc-1] = dest->nonnull;
  2320. child->argv[argc]=NULL;
  2321. child->argv_nonnull[argc] = 0;
  2322. for (s = dest->data; s && *s; s++,str++) {
  2323. if (*s == '\\') s++;
  2324. *str = *s;
  2325. }
  2326. *str = '\0';
  2327. #endif
  2328. #ifndef __U_BOOT__
  2329. }
  2330. gr = xglob(dest, flags, glob_target);
  2331. if (gr != 0) return 1;
  2332. #endif
  2333. b_reset(dest);
  2334. #ifndef __U_BOOT__
  2335. if (ctx->pending_redirect) {
  2336. ctx->pending_redirect=NULL;
  2337. if (glob_target->gl_pathc != 1) {
  2338. error_msg("ambiguous redirect");
  2339. return 1;
  2340. }
  2341. } else {
  2342. child->argv = glob_target->gl_pathv;
  2343. }
  2344. #endif
  2345. if (ctx->w == RES_FOR) {
  2346. done_word(dest,ctx);
  2347. done_pipe(ctx,PIPE_SEQ);
  2348. }
  2349. return 0;
  2350. }
  2351. /* The only possible error here is out of memory, in which case
  2352. * xmalloc exits. */
  2353. static int done_command(struct p_context *ctx)
  2354. {
  2355. /* The child is really already in the pipe structure, so
  2356. * advance the pipe counter and make a new, null child.
  2357. * Only real trickiness here is that the uncommitted
  2358. * child structure, to which ctx->child points, is not
  2359. * counted in pi->num_progs. */
  2360. struct pipe *pi=ctx->pipe;
  2361. struct child_prog *prog=ctx->child;
  2362. if (prog && prog->group == NULL
  2363. && prog->argv == NULL
  2364. #ifndef __U_BOOT__
  2365. && prog->redirects == NULL) {
  2366. #else
  2367. ) {
  2368. #endif
  2369. debug_printf("done_command: skipping null command\n");
  2370. return 0;
  2371. } else if (prog) {
  2372. pi->num_progs++;
  2373. debug_printf("done_command: num_progs incremented to %d\n",pi->num_progs);
  2374. } else {
  2375. debug_printf("done_command: initializing\n");
  2376. }
  2377. pi->progs = xrealloc(pi->progs, sizeof(*pi->progs) * (pi->num_progs+1));
  2378. prog = pi->progs + pi->num_progs;
  2379. #ifndef __U_BOOT__
  2380. prog->redirects = NULL;
  2381. #endif
  2382. prog->argv = NULL;
  2383. prog->argv_nonnull = NULL;
  2384. #ifndef __U_BOOT__
  2385. prog->is_stopped = 0;
  2386. #endif
  2387. prog->group = NULL;
  2388. #ifndef __U_BOOT__
  2389. prog->glob_result.gl_pathv = NULL;
  2390. prog->family = pi;
  2391. #endif
  2392. prog->sp = 0;
  2393. ctx->child = prog;
  2394. prog->type = ctx->type;
  2395. /* but ctx->pipe and ctx->list_head remain unchanged */
  2396. return 0;
  2397. }
  2398. static int done_pipe(struct p_context *ctx, pipe_style type)
  2399. {
  2400. struct pipe *new_p;
  2401. done_command(ctx); /* implicit closure of previous command */
  2402. debug_printf("done_pipe, type %d\n", type);
  2403. ctx->pipe->followup = type;
  2404. ctx->pipe->r_mode = ctx->w;
  2405. new_p=new_pipe();
  2406. ctx->pipe->next = new_p;
  2407. ctx->pipe = new_p;
  2408. ctx->child = NULL;
  2409. done_command(ctx); /* set up new pipe to accept commands */
  2410. return 0;
  2411. }
  2412. #ifndef __U_BOOT__
  2413. /* peek ahead in the in_str to find out if we have a "&n" construct,
  2414. * as in "2>&1", that represents duplicating a file descriptor.
  2415. * returns either -2 (syntax error), -1 (no &), or the number found.
  2416. */
  2417. static int redirect_dup_num(struct in_str *input)
  2418. {
  2419. int ch, d=0, ok=0;
  2420. ch = b_peek(input);
  2421. if (ch != '&') return -1;
  2422. b_getch(input); /* get the & */
  2423. ch=b_peek(input);
  2424. if (ch == '-') {
  2425. b_getch(input);
  2426. return -3; /* "-" represents "close me" */
  2427. }
  2428. while (isdigit(ch)) {
  2429. d = d*10+(ch-'0');
  2430. ok=1;
  2431. b_getch(input);
  2432. ch = b_peek(input);
  2433. }
  2434. if (ok) return d;
  2435. error_msg("ambiguous redirect");
  2436. return -2;
  2437. }
  2438. /* If a redirect is immediately preceded by a number, that number is
  2439. * supposed to tell which file descriptor to redirect. This routine
  2440. * looks for such preceding numbers. In an ideal world this routine
  2441. * needs to handle all the following classes of redirects...
  2442. * echo 2>foo # redirects fd 2 to file "foo", nothing passed to echo
  2443. * echo 49>foo # redirects fd 49 to file "foo", nothing passed to echo
  2444. * echo -2>foo # redirects fd 1 to file "foo", "-2" passed to echo
  2445. * echo 49x>foo # redirects fd 1 to file "foo", "49x" passed to echo
  2446. * A -1 output from this program means no valid number was found, so the
  2447. * caller should use the appropriate default for this redirection.
  2448. */
  2449. static int redirect_opt_num(o_string *o)
  2450. {
  2451. int num;
  2452. if (o->length==0) return -1;
  2453. for(num=0; num<o->length; num++) {
  2454. if (!isdigit(*(o->data+num))) {
  2455. return -1;
  2456. }
  2457. }
  2458. /* reuse num (and save an int) */
  2459. num=atoi(o->data);
  2460. b_reset(o);
  2461. return num;
  2462. }
  2463. FILE *generate_stream_from_list(struct pipe *head)
  2464. {
  2465. FILE *pf;
  2466. #if 1
  2467. int pid, channel[2];
  2468. if (pipe(channel)<0) perror_msg_and_die("pipe");
  2469. pid=fork();
  2470. if (pid<0) {
  2471. perror_msg_and_die("fork");
  2472. } else if (pid==0) {
  2473. close(channel[0]);
  2474. if (channel[1] != 1) {
  2475. dup2(channel[1],1);
  2476. close(channel[1]);
  2477. }
  2478. #if 0
  2479. #define SURROGATE "surrogate response"
  2480. write(1,SURROGATE,sizeof(SURROGATE));
  2481. _exit(run_list(head));
  2482. #else
  2483. _exit(run_list_real(head)); /* leaks memory */
  2484. #endif
  2485. }
  2486. debug_printf("forked child %d\n",pid);
  2487. close(channel[1]);
  2488. pf = fdopen(channel[0],"r");
  2489. debug_printf("pipe on FILE *%p\n",pf);
  2490. #else
  2491. free_pipe_list(head,0);
  2492. pf=popen("echo surrogate response","r");
  2493. debug_printf("started fake pipe on FILE *%p\n",pf);
  2494. #endif
  2495. return pf;
  2496. }
  2497. /* this version hacked for testing purposes */
  2498. /* return code is exit status of the process that is run. */
  2499. static int process_command_subs(o_string *dest, struct p_context *ctx, struct in_str *input, int subst_end)
  2500. {
  2501. int retcode;
  2502. o_string result=NULL_O_STRING;
  2503. struct p_context inner;
  2504. FILE *p;
  2505. struct in_str pipe_str;
  2506. initialize_context(&inner);
  2507. /* recursion to generate command */
  2508. retcode = parse_stream(&result, &inner, input, subst_end);
  2509. if (retcode != 0) return retcode; /* syntax error or EOF */
  2510. done_word(&result, &inner);
  2511. done_pipe(&inner, PIPE_SEQ);
  2512. b_free(&result);
  2513. p=generate_stream_from_list(inner.list_head);
  2514. if (p==NULL) return 1;
  2515. mark_open(fileno(p));
  2516. setup_file_in_str(&pipe_str, p);
  2517. /* now send results of command back into original context */
  2518. retcode = parse_stream(dest, ctx, &pipe_str, '\0');
  2519. /* XXX In case of a syntax error, should we try to kill the child?
  2520. * That would be tough to do right, so just read until EOF. */
  2521. if (retcode == 1) {
  2522. while (b_getch(&pipe_str)!=EOF) { /* discard */ };
  2523. }
  2524. debug_printf("done reading from pipe, pclose()ing\n");
  2525. /* This is the step that wait()s for the child. Should be pretty
  2526. * safe, since we just read an EOF from its stdout. We could try
  2527. * to better, by using wait(), and keeping track of background jobs
  2528. * at the same time. That would be a lot of work, and contrary
  2529. * to the KISS philosophy of this program. */
  2530. mark_closed(fileno(p));
  2531. retcode=pclose(p);
  2532. free_pipe_list(inner.list_head,0);
  2533. debug_printf("pclosed, retcode=%d\n",retcode);
  2534. /* XXX this process fails to trim a single trailing newline */
  2535. return retcode;
  2536. }
  2537. static int parse_group(o_string *dest, struct p_context *ctx,
  2538. struct in_str *input, int ch)
  2539. {
  2540. int rcode, endch=0;
  2541. struct p_context sub;
  2542. struct child_prog *child = ctx->child;
  2543. if (child->argv) {
  2544. syntax();
  2545. return 1; /* syntax error, groups and arglists don't mix */
  2546. }
  2547. initialize_context(&sub);
  2548. switch(ch) {
  2549. case '(': endch=')'; child->subshell=1; break;
  2550. case '{': endch='}'; break;
  2551. default: syntax(); /* really logic error */
  2552. }
  2553. rcode=parse_stream(dest,&sub,input,endch);
  2554. done_word(dest,&sub); /* finish off the final word in the subcontext */
  2555. done_pipe(&sub, PIPE_SEQ); /* and the final command there, too */
  2556. child->group = sub.list_head;
  2557. return rcode;
  2558. /* child remains "open", available for possible redirects */
  2559. }
  2560. #endif
  2561. /* basically useful version until someone wants to get fancier,
  2562. * see the bash man page under "Parameter Expansion" */
  2563. static char *lookup_param(char *src)
  2564. {
  2565. char *p;
  2566. char *sep;
  2567. char *default_val = NULL;
  2568. int assign = 0;
  2569. int expand_empty = 0;
  2570. if (!src)
  2571. return NULL;
  2572. sep = strchr(src, ':');
  2573. if (sep) {
  2574. *sep = '\0';
  2575. if (*(sep + 1) == '-')
  2576. default_val = sep+2;
  2577. if (*(sep + 1) == '=') {
  2578. default_val = sep+2;
  2579. assign = 1;
  2580. }
  2581. if (*(sep + 1) == '+') {
  2582. default_val = sep+2;
  2583. expand_empty = 1;
  2584. }
  2585. }
  2586. p = env_get(src);
  2587. if (!p)
  2588. p = get_local_var(src);
  2589. if (!p || strlen(p) == 0) {
  2590. p = default_val;
  2591. if (assign) {
  2592. char *var = malloc(strlen(src)+strlen(default_val)+2);
  2593. if (var) {
  2594. sprintf(var, "%s=%s", src, default_val);
  2595. set_local_var(var, 0);
  2596. }
  2597. free(var);
  2598. }
  2599. } else if (expand_empty) {
  2600. p += strlen(p);
  2601. }
  2602. if (sep)
  2603. *sep = ':';
  2604. return p;
  2605. }
  2606. #ifdef __U_BOOT__
  2607. static char *get_dollar_var(char ch)
  2608. {
  2609. static char buf[40];
  2610. buf[0] = '\0';
  2611. switch (ch) {
  2612. case '?':
  2613. sprintf(buf, "%u", (unsigned int)last_return_code);
  2614. break;
  2615. default:
  2616. return NULL;
  2617. }
  2618. return buf;
  2619. }
  2620. #endif
  2621. /* return code: 0 for OK, 1 for syntax error */
  2622. static int handle_dollar(o_string *dest, struct p_context *ctx, struct in_str *input)
  2623. {
  2624. #ifndef __U_BOOT__
  2625. int i, advance=0;
  2626. #else
  2627. int advance=0;
  2628. #endif
  2629. #ifndef __U_BOOT__
  2630. char sep[]=" ";
  2631. #endif
  2632. int ch = input->peek(input); /* first character after the $ */
  2633. debug_printf("handle_dollar: ch=%c\n",ch);
  2634. if (isalpha(ch)) {
  2635. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2636. ctx->child->sp++;
  2637. while(ch=b_peek(input),isalnum(ch) || ch=='_') {
  2638. b_getch(input);
  2639. b_addchr(dest,ch);
  2640. }
  2641. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2642. #ifndef __U_BOOT__
  2643. } else if (isdigit(ch)) {
  2644. i = ch-'0'; /* XXX is $0 special? */
  2645. if (i<global_argc) {
  2646. parse_string(dest, ctx, global_argv[i]); /* recursion */
  2647. }
  2648. advance = 1;
  2649. #endif
  2650. } else switch (ch) {
  2651. #ifndef __U_BOOT__
  2652. case '$':
  2653. b_adduint(dest,getpid());
  2654. advance = 1;
  2655. break;
  2656. case '!':
  2657. if (last_bg_pid > 0) b_adduint(dest, last_bg_pid);
  2658. advance = 1;
  2659. break;
  2660. #endif
  2661. case '?':
  2662. #ifndef __U_BOOT__
  2663. b_adduint(dest,last_return_code);
  2664. #else
  2665. ctx->child->sp++;
  2666. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2667. b_addchr(dest, '$');
  2668. b_addchr(dest, '?');
  2669. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2670. #endif
  2671. advance = 1;
  2672. break;
  2673. #ifndef __U_BOOT__
  2674. case '#':
  2675. b_adduint(dest,global_argc ? global_argc-1 : 0);
  2676. advance = 1;
  2677. break;
  2678. #endif
  2679. case '{':
  2680. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2681. ctx->child->sp++;
  2682. b_getch(input);
  2683. /* XXX maybe someone will try to escape the '}' */
  2684. while(ch=b_getch(input),ch!=EOF && ch!='}') {
  2685. b_addchr(dest,ch);
  2686. }
  2687. if (ch != '}') {
  2688. syntax();
  2689. return 1;
  2690. }
  2691. b_addchr(dest, SPECIAL_VAR_SYMBOL);
  2692. break;
  2693. #ifndef __U_BOOT__
  2694. case '(':
  2695. b_getch(input);
  2696. process_command_subs(dest, ctx, input, ')');
  2697. break;
  2698. case '*':
  2699. sep[0]=ifs[0];
  2700. for (i=1; i<global_argc; i++) {
  2701. parse_string(dest, ctx, global_argv[i]);
  2702. if (i+1 < global_argc) parse_string(dest, ctx, sep);
  2703. }
  2704. break;
  2705. case '@':
  2706. case '-':
  2707. case '_':
  2708. /* still unhandled, but should be eventually */
  2709. error_msg("unhandled syntax: $%c",ch);
  2710. return 1;
  2711. break;
  2712. #endif
  2713. default:
  2714. b_addqchr(dest,'$',dest->quote);
  2715. }
  2716. /* Eat the character if the flag was set. If the compiler
  2717. * is smart enough, we could substitute "b_getch(input);"
  2718. * for all the "advance = 1;" above, and also end up with
  2719. * a nice size-optimized program. Hah! That'll be the day.
  2720. */
  2721. if (advance) b_getch(input);
  2722. return 0;
  2723. }
  2724. #ifndef __U_BOOT__
  2725. int parse_string(o_string *dest, struct p_context *ctx, const char *src)
  2726. {
  2727. struct in_str foo;
  2728. setup_string_in_str(&foo, src);
  2729. return parse_stream(dest, ctx, &foo, '\0');
  2730. }
  2731. #endif
  2732. /* return code is 0 for normal exit, 1 for syntax error */
  2733. static int parse_stream(o_string *dest, struct p_context *ctx,
  2734. struct in_str *input, int end_trigger)
  2735. {
  2736. unsigned int ch, m;
  2737. #ifndef __U_BOOT__
  2738. int redir_fd;
  2739. redir_type redir_style;
  2740. #endif
  2741. int next;
  2742. /* Only double-quote state is handled in the state variable dest->quote.
  2743. * A single-quote triggers a bypass of the main loop until its mate is
  2744. * found. When recursing, quote state is passed in via dest->quote. */
  2745. debug_printf("parse_stream, end_trigger=%d\n",end_trigger);
  2746. while ((ch=b_getch(input))!=EOF) {
  2747. m = map[ch];
  2748. #ifdef __U_BOOT__
  2749. if (input->__promptme == 0) return 1;
  2750. #endif
  2751. next = (ch == '\n') ? 0 : b_peek(input);
  2752. debug_printf("parse_stream: ch=%c (%d) m=%d quote=%d - %c\n",
  2753. ch >= ' ' ? ch : '.', ch, m,
  2754. dest->quote, ctx->stack == NULL ? '*' : '.');
  2755. if (m==0 || ((m==1 || m==2) && dest->quote)) {
  2756. b_addqchr(dest, ch, dest->quote);
  2757. } else {
  2758. if (m==2) { /* unquoted IFS */
  2759. if (done_word(dest, ctx)) {
  2760. return 1;
  2761. }
  2762. /* If we aren't performing a substitution, treat a newline as a
  2763. * command separator. */
  2764. if (end_trigger != '\0' && ch=='\n')
  2765. done_pipe(ctx,PIPE_SEQ);
  2766. }
  2767. if (ch == end_trigger && !dest->quote && ctx->w==RES_NONE) {
  2768. debug_printf("leaving parse_stream (triggered)\n");
  2769. return 0;
  2770. }
  2771. #if 0
  2772. if (ch=='\n') {
  2773. /* Yahoo! Time to run with it! */
  2774. done_pipe(ctx,PIPE_SEQ);
  2775. run_list(ctx->list_head);
  2776. initialize_context(ctx);
  2777. }
  2778. #endif
  2779. if (m!=2) switch (ch) {
  2780. case '#':
  2781. if (dest->length == 0 && !dest->quote) {
  2782. while(ch=b_peek(input),ch!=EOF && ch!='\n') { b_getch(input); }
  2783. } else {
  2784. b_addqchr(dest, ch, dest->quote);
  2785. }
  2786. break;
  2787. case '\\':
  2788. if (next == EOF) {
  2789. syntax();
  2790. return 1;
  2791. }
  2792. b_addqchr(dest, '\\', dest->quote);
  2793. b_addqchr(dest, b_getch(input), dest->quote);
  2794. break;
  2795. case '$':
  2796. if (handle_dollar(dest, ctx, input)!=0) return 1;
  2797. break;
  2798. case '\'':
  2799. dest->nonnull = 1;
  2800. while(ch=b_getch(input),ch!=EOF && ch!='\'') {
  2801. #ifdef __U_BOOT__
  2802. if(input->__promptme == 0) return 1;
  2803. #endif
  2804. b_addchr(dest,ch);
  2805. }
  2806. if (ch==EOF) {
  2807. syntax();
  2808. return 1;
  2809. }
  2810. break;
  2811. case '"':
  2812. dest->nonnull = 1;
  2813. dest->quote = !dest->quote;
  2814. break;
  2815. #ifndef __U_BOOT__
  2816. case '`':
  2817. process_command_subs(dest, ctx, input, '`');
  2818. break;
  2819. case '>':
  2820. redir_fd = redirect_opt_num(dest);
  2821. done_word(dest, ctx);
  2822. redir_style=REDIRECT_OVERWRITE;
  2823. if (next == '>') {
  2824. redir_style=REDIRECT_APPEND;
  2825. b_getch(input);
  2826. } else if (next == '(') {
  2827. syntax(); /* until we support >(list) Process Substitution */
  2828. return 1;
  2829. }
  2830. setup_redirect(ctx, redir_fd, redir_style, input);
  2831. break;
  2832. case '<':
  2833. redir_fd = redirect_opt_num(dest);
  2834. done_word(dest, ctx);
  2835. redir_style=REDIRECT_INPUT;
  2836. if (next == '<') {
  2837. redir_style=REDIRECT_HEREIS;
  2838. b_getch(input);
  2839. } else if (next == '>') {
  2840. redir_style=REDIRECT_IO;
  2841. b_getch(input);
  2842. } else if (next == '(') {
  2843. syntax(); /* until we support <(list) Process Substitution */
  2844. return 1;
  2845. }
  2846. setup_redirect(ctx, redir_fd, redir_style, input);
  2847. break;
  2848. #endif
  2849. case ';':
  2850. done_word(dest, ctx);
  2851. done_pipe(ctx,PIPE_SEQ);
  2852. break;
  2853. case '&':
  2854. done_word(dest, ctx);
  2855. if (next=='&') {
  2856. b_getch(input);
  2857. done_pipe(ctx,PIPE_AND);
  2858. } else {
  2859. #ifndef __U_BOOT__
  2860. done_pipe(ctx,PIPE_BG);
  2861. #else
  2862. syntax_err();
  2863. return 1;
  2864. #endif
  2865. }
  2866. break;
  2867. case '|':
  2868. done_word(dest, ctx);
  2869. if (next=='|') {
  2870. b_getch(input);
  2871. done_pipe(ctx,PIPE_OR);
  2872. } else {
  2873. /* we could pick up a file descriptor choice here
  2874. * with redirect_opt_num(), but bash doesn't do it.
  2875. * "echo foo 2| cat" yields "foo 2". */
  2876. #ifndef __U_BOOT__
  2877. done_command(ctx);
  2878. #else
  2879. syntax_err();
  2880. return 1;
  2881. #endif
  2882. }
  2883. break;
  2884. #ifndef __U_BOOT__
  2885. case '(':
  2886. case '{':
  2887. if (parse_group(dest, ctx, input, ch)!=0) return 1;
  2888. break;
  2889. case ')':
  2890. case '}':
  2891. syntax(); /* Proper use of this character caught by end_trigger */
  2892. return 1;
  2893. break;
  2894. #endif
  2895. case SUBSTED_VAR_SYMBOL:
  2896. dest->nonnull = 1;
  2897. while (ch = b_getch(input), ch != EOF &&
  2898. ch != SUBSTED_VAR_SYMBOL) {
  2899. debug_printf("subst, pass=%d\n", ch);
  2900. if (input->__promptme == 0)
  2901. return 1;
  2902. b_addchr(dest, ch);
  2903. }
  2904. debug_printf("subst, term=%d\n", ch);
  2905. if (ch == EOF) {
  2906. syntax();
  2907. return 1;
  2908. }
  2909. break;
  2910. default:
  2911. syntax(); /* this is really an internal logic error */
  2912. return 1;
  2913. }
  2914. }
  2915. }
  2916. /* complain if quote? No, maybe we just finished a command substitution
  2917. * that was quoted. Example:
  2918. * $ echo "`cat foo` plus more"
  2919. * and we just got the EOF generated by the subshell that ran "cat foo"
  2920. * The only real complaint is if we got an EOF when end_trigger != '\0',
  2921. * that is, we were really supposed to get end_trigger, and never got
  2922. * one before the EOF. Can't use the standard "syntax error" return code,
  2923. * so that parse_stream_outer can distinguish the EOF and exit smoothly. */
  2924. debug_printf("leaving parse_stream (EOF)\n");
  2925. if (end_trigger != '\0') return -1;
  2926. return 0;
  2927. }
  2928. static void mapset(const unsigned char *set, int code)
  2929. {
  2930. const unsigned char *s;
  2931. for (s=set; *s; s++) map[*s] = code;
  2932. }
  2933. static void update_ifs_map(void)
  2934. {
  2935. /* char *ifs and char map[256] are both globals. */
  2936. ifs = (uchar *)env_get("IFS");
  2937. if (ifs == NULL) ifs=(uchar *)" \t\n";
  2938. /* Precompute a list of 'flow through' behavior so it can be treated
  2939. * quickly up front. Computation is necessary because of IFS.
  2940. * Special case handling of IFS == " \t\n" is not implemented.
  2941. * The map[] array only really needs two bits each, and on most machines
  2942. * that would be faster because of the reduced L1 cache footprint.
  2943. */
  2944. memset(map,0,sizeof(map)); /* most characters flow through always */
  2945. #ifndef __U_BOOT__
  2946. mapset((uchar *)"\\$'\"`", 3); /* never flow through */
  2947. mapset((uchar *)"<>;&|(){}#", 1); /* flow through if quoted */
  2948. #else
  2949. {
  2950. uchar subst[2] = {SUBSTED_VAR_SYMBOL, 0};
  2951. mapset(subst, 3); /* never flow through */
  2952. }
  2953. mapset((uchar *)"\\$'\"", 3); /* never flow through */
  2954. mapset((uchar *)";&|#", 1); /* flow through if quoted */
  2955. #endif
  2956. mapset(ifs, 2); /* also flow through if quoted */
  2957. }
  2958. /* most recursion does not come through here, the exeception is
  2959. * from builtin_source() */
  2960. static int parse_stream_outer(struct in_str *inp, int flag)
  2961. {
  2962. struct p_context ctx;
  2963. o_string temp=NULL_O_STRING;
  2964. int rcode;
  2965. #ifdef __U_BOOT__
  2966. int code = 1;
  2967. #endif
  2968. do {
  2969. ctx.type = flag;
  2970. initialize_context(&ctx);
  2971. update_ifs_map();
  2972. if (!(flag & FLAG_PARSE_SEMICOLON) || (flag & FLAG_REPARSING)) mapset((uchar *)";$&|", 0);
  2973. inp->promptmode=1;
  2974. rcode = parse_stream(&temp, &ctx, inp,
  2975. flag & FLAG_CONT_ON_NEWLINE ? -1 : '\n');
  2976. #ifdef __U_BOOT__
  2977. if (rcode == 1) flag_repeat = 0;
  2978. #endif
  2979. if (rcode != 1 && ctx.old_flag != 0) {
  2980. syntax();
  2981. #ifdef __U_BOOT__
  2982. flag_repeat = 0;
  2983. #endif
  2984. }
  2985. if (rcode != 1 && ctx.old_flag == 0) {
  2986. done_word(&temp, &ctx);
  2987. done_pipe(&ctx,PIPE_SEQ);
  2988. #ifndef __U_BOOT__
  2989. run_list(ctx.list_head);
  2990. #else
  2991. code = run_list(ctx.list_head);
  2992. if (code == -2) { /* exit */
  2993. b_free(&temp);
  2994. code = 0;
  2995. /* XXX hackish way to not allow exit from main loop */
  2996. if (inp->peek == file_peek) {
  2997. printf("exit not allowed from main input shell.\n");
  2998. continue;
  2999. }
  3000. break;
  3001. }
  3002. if (code == -1)
  3003. flag_repeat = 0;
  3004. #endif
  3005. } else {
  3006. if (ctx.old_flag != 0) {
  3007. free(ctx.stack);
  3008. b_reset(&temp);
  3009. }
  3010. #ifdef __U_BOOT__
  3011. if (inp->__promptme == 0) printf("<INTERRUPT>\n");
  3012. inp->__promptme = 1;
  3013. #endif
  3014. temp.nonnull = 0;
  3015. temp.quote = 0;
  3016. inp->p = NULL;
  3017. free_pipe_list(ctx.list_head,0);
  3018. }
  3019. b_free(&temp);
  3020. /* loop on syntax errors, return on EOF */
  3021. } while (rcode != -1 && !(flag & FLAG_EXIT_FROM_LOOP) &&
  3022. (inp->peek != static_peek || b_peek(inp)));
  3023. #ifndef __U_BOOT__
  3024. return 0;
  3025. #else
  3026. return (code != 0) ? 1 : 0;
  3027. #endif /* __U_BOOT__ */
  3028. }
  3029. #ifndef __U_BOOT__
  3030. static int parse_string_outer(const char *s, int flag)
  3031. #else
  3032. int parse_string_outer(const char *s, int flag)
  3033. #endif /* __U_BOOT__ */
  3034. {
  3035. struct in_str input;
  3036. #ifdef __U_BOOT__
  3037. char *p = NULL;
  3038. int rcode;
  3039. if (!s)
  3040. return 1;
  3041. if (!*s)
  3042. return 0;
  3043. if (!(p = strchr(s, '\n')) || *++p) {
  3044. p = xmalloc(strlen(s) + 2);
  3045. strcpy(p, s);
  3046. strcat(p, "\n");
  3047. setup_string_in_str(&input, p);
  3048. rcode = parse_stream_outer(&input, flag);
  3049. free(p);
  3050. return rcode;
  3051. } else {
  3052. #endif
  3053. setup_string_in_str(&input, s);
  3054. return parse_stream_outer(&input, flag);
  3055. #ifdef __U_BOOT__
  3056. }
  3057. #endif
  3058. }
  3059. #ifndef __U_BOOT__
  3060. static int parse_file_outer(FILE *f)
  3061. #else
  3062. int parse_file_outer(void)
  3063. #endif
  3064. {
  3065. int rcode;
  3066. struct in_str input;
  3067. #ifndef __U_BOOT__
  3068. setup_file_in_str(&input, f);
  3069. #else
  3070. setup_file_in_str(&input);
  3071. #endif
  3072. rcode = parse_stream_outer(&input, FLAG_PARSE_SEMICOLON);
  3073. return rcode;
  3074. }
  3075. #ifdef __U_BOOT__
  3076. #ifdef CONFIG_NEEDS_MANUAL_RELOC
  3077. static void u_boot_hush_reloc(void)
  3078. {
  3079. unsigned long addr;
  3080. struct reserved_combo *r;
  3081. for (r=reserved_list; r<reserved_list+NRES; r++) {
  3082. addr = (ulong) (r->literal) + gd->reloc_off;
  3083. r->literal = (char *)addr;
  3084. }
  3085. }
  3086. #endif
  3087. int u_boot_hush_start(void)
  3088. {
  3089. if (top_vars == NULL) {
  3090. top_vars = malloc(sizeof(struct variables));
  3091. top_vars->name = "HUSH_VERSION";
  3092. top_vars->value = "0.01";
  3093. top_vars->next = NULL;
  3094. top_vars->flg_export = 0;
  3095. top_vars->flg_read_only = 1;
  3096. #ifdef CONFIG_NEEDS_MANUAL_RELOC
  3097. u_boot_hush_reloc();
  3098. #endif
  3099. }
  3100. return 0;
  3101. }
  3102. static void *xmalloc(size_t size)
  3103. {
  3104. void *p = NULL;
  3105. if (!(p = malloc(size))) {
  3106. printf("ERROR : memory not allocated\n");
  3107. for(;;);
  3108. }
  3109. return p;
  3110. }
  3111. static void *xrealloc(void *ptr, size_t size)
  3112. {
  3113. void *p = NULL;
  3114. if (!(p = realloc(ptr, size))) {
  3115. printf("ERROR : memory not allocated\n");
  3116. for(;;);
  3117. }
  3118. return p;
  3119. }
  3120. #endif /* __U_BOOT__ */
  3121. #ifndef __U_BOOT__
  3122. /* Make sure we have a controlling tty. If we get started under a job
  3123. * aware app (like bash for example), make sure we are now in charge so
  3124. * we don't fight over who gets the foreground */
  3125. static void setup_job_control(void)
  3126. {
  3127. static pid_t shell_pgrp;
  3128. /* Loop until we are in the foreground. */
  3129. while (tcgetpgrp (shell_terminal) != (shell_pgrp = getpgrp ()))
  3130. kill (- shell_pgrp, SIGTTIN);
  3131. /* Ignore interactive and job-control signals. */
  3132. signal(SIGINT, SIG_IGN);
  3133. signal(SIGQUIT, SIG_IGN);
  3134. signal(SIGTERM, SIG_IGN);
  3135. signal(SIGTSTP, SIG_IGN);
  3136. signal(SIGTTIN, SIG_IGN);
  3137. signal(SIGTTOU, SIG_IGN);
  3138. signal(SIGCHLD, SIG_IGN);
  3139. /* Put ourselves in our own process group. */
  3140. setsid();
  3141. shell_pgrp = getpid ();
  3142. setpgid (shell_pgrp, shell_pgrp);
  3143. /* Grab control of the terminal. */
  3144. tcsetpgrp(shell_terminal, shell_pgrp);
  3145. }
  3146. int hush_main(int argc, char * const *argv)
  3147. {
  3148. int opt;
  3149. FILE *input;
  3150. char **e = environ;
  3151. /* XXX what should these be while sourcing /etc/profile? */
  3152. global_argc = argc;
  3153. global_argv = argv;
  3154. /* (re?) initialize globals. Sometimes hush_main() ends up calling
  3155. * hush_main(), therefore we cannot rely on the BSS to zero out this
  3156. * stuff. Reset these to 0 every time. */
  3157. ifs = NULL;
  3158. /* map[] is taken care of with call to update_ifs_map() */
  3159. fake_mode = 0;
  3160. interactive = 0;
  3161. close_me_head = NULL;
  3162. last_bg_pid = 0;
  3163. job_list = NULL;
  3164. last_jobid = 0;
  3165. /* Initialize some more globals to non-zero values */
  3166. set_cwd();
  3167. #ifdef CONFIG_FEATURE_COMMAND_EDITING
  3168. cmdedit_set_initial_prompt();
  3169. #else
  3170. PS1 = NULL;
  3171. #endif
  3172. PS2 = "> ";
  3173. /* initialize our shell local variables with the values
  3174. * currently living in the environment */
  3175. if (e) {
  3176. for (; *e; e++)
  3177. set_local_var(*e, 2); /* without call putenv() */
  3178. }
  3179. last_return_code=EXIT_SUCCESS;
  3180. if (argv[0] && argv[0][0] == '-') {
  3181. debug_printf("\nsourcing /etc/profile\n");
  3182. if ((input = fopen("/etc/profile", "r")) != NULL) {
  3183. mark_open(fileno(input));
  3184. parse_file_outer(input);
  3185. mark_closed(fileno(input));
  3186. fclose(input);
  3187. }
  3188. }
  3189. input=stdin;
  3190. while ((opt = getopt(argc, argv, "c:xif")) > 0) {
  3191. switch (opt) {
  3192. case 'c':
  3193. {
  3194. global_argv = argv+optind;
  3195. global_argc = argc-optind;
  3196. opt = parse_string_outer(optarg, FLAG_PARSE_SEMICOLON);
  3197. goto final_return;
  3198. }
  3199. break;
  3200. case 'i':
  3201. interactive++;
  3202. break;
  3203. case 'f':
  3204. fake_mode++;
  3205. break;
  3206. default:
  3207. #ifndef BB_VER
  3208. fprintf(stderr, "Usage: sh [FILE]...\n"
  3209. " or: sh -c command [args]...\n\n");
  3210. exit(EXIT_FAILURE);
  3211. #else
  3212. show_usage();
  3213. #endif
  3214. }
  3215. }
  3216. /* A shell is interactive if the `-i' flag was given, or if all of
  3217. * the following conditions are met:
  3218. * no -c command
  3219. * no arguments remaining or the -s flag given
  3220. * standard input is a terminal
  3221. * standard output is a terminal
  3222. * Refer to Posix.2, the description of the `sh' utility. */
  3223. if (argv[optind]==NULL && input==stdin &&
  3224. isatty(fileno(stdin)) && isatty(fileno(stdout))) {
  3225. interactive++;
  3226. }
  3227. debug_printf("\ninteractive=%d\n", interactive);
  3228. if (interactive) {
  3229. /* Looks like they want an interactive shell */
  3230. #ifndef CONFIG_FEATURE_SH_EXTRA_QUIET
  3231. printf( "\n\n" BB_BANNER " hush - the humble shell v0.01 (testing)\n");
  3232. printf( "Enter 'help' for a list of built-in commands.\n\n");
  3233. #endif
  3234. setup_job_control();
  3235. }
  3236. if (argv[optind]==NULL) {
  3237. opt=parse_file_outer(stdin);
  3238. goto final_return;
  3239. }
  3240. debug_printf("\nrunning script '%s'\n", argv[optind]);
  3241. global_argv = argv+optind;
  3242. global_argc = argc-optind;
  3243. input = xfopen(argv[optind], "r");
  3244. opt = parse_file_outer(input);
  3245. #ifdef CONFIG_FEATURE_CLEAN_UP
  3246. fclose(input);
  3247. if (cwd && cwd != unknown)
  3248. free((char*)cwd);
  3249. {
  3250. struct variables *cur, *tmp;
  3251. for(cur = top_vars; cur; cur = tmp) {
  3252. tmp = cur->next;
  3253. if (!cur->flg_read_only) {
  3254. free(cur->name);
  3255. free(cur->value);
  3256. free(cur);
  3257. }
  3258. }
  3259. }
  3260. #endif
  3261. final_return:
  3262. return(opt?opt:last_return_code);
  3263. }
  3264. #endif
  3265. static char *insert_var_value(char *inp)
  3266. {
  3267. return insert_var_value_sub(inp, 0);
  3268. }
  3269. static char *insert_var_value_sub(char *inp, int tag_subst)
  3270. {
  3271. int res_str_len = 0;
  3272. int len;
  3273. int done = 0;
  3274. char *p, *p1, *res_str = NULL;
  3275. while ((p = strchr(inp, SPECIAL_VAR_SYMBOL))) {
  3276. /* check the beginning of the string for normal characters */
  3277. if (p != inp) {
  3278. /* copy any characters to the result string */
  3279. len = p - inp;
  3280. res_str = xrealloc(res_str, (res_str_len + len));
  3281. strncpy((res_str + res_str_len), inp, len);
  3282. res_str_len += len;
  3283. }
  3284. inp = ++p;
  3285. /* find the ending marker */
  3286. p = strchr(inp, SPECIAL_VAR_SYMBOL);
  3287. *p = '\0';
  3288. /* look up the value to substitute */
  3289. if ((p1 = lookup_param(inp))) {
  3290. if (tag_subst)
  3291. len = res_str_len + strlen(p1) + 2;
  3292. else
  3293. len = res_str_len + strlen(p1);
  3294. res_str = xrealloc(res_str, (1 + len));
  3295. if (tag_subst) {
  3296. /*
  3297. * copy the variable value to the result
  3298. * string
  3299. */
  3300. strcpy((res_str + res_str_len + 1), p1);
  3301. /*
  3302. * mark the replaced text to be accepted as
  3303. * is
  3304. */
  3305. res_str[res_str_len] = SUBSTED_VAR_SYMBOL;
  3306. res_str[res_str_len + 1 + strlen(p1)] =
  3307. SUBSTED_VAR_SYMBOL;
  3308. } else
  3309. /*
  3310. * copy the variable value to the result
  3311. * string
  3312. */
  3313. strcpy((res_str + res_str_len), p1);
  3314. res_str_len = len;
  3315. }
  3316. *p = SPECIAL_VAR_SYMBOL;
  3317. inp = ++p;
  3318. done = 1;
  3319. }
  3320. if (done) {
  3321. res_str = xrealloc(res_str, (1 + res_str_len + strlen(inp)));
  3322. strcpy((res_str + res_str_len), inp);
  3323. while ((p = strchr(res_str, '\n'))) {
  3324. *p = ' ';
  3325. }
  3326. }
  3327. return (res_str == NULL) ? inp : res_str;
  3328. }
  3329. static char **make_list_in(char **inp, char *name)
  3330. {
  3331. int len, i;
  3332. int name_len = strlen(name);
  3333. int n = 0;
  3334. char **list;
  3335. char *p1, *p2, *p3;
  3336. /* create list of variable values */
  3337. list = xmalloc(sizeof(*list));
  3338. for (i = 0; inp[i]; i++) {
  3339. p3 = insert_var_value(inp[i]);
  3340. p1 = p3;
  3341. while (*p1) {
  3342. if (*p1 == ' ') {
  3343. p1++;
  3344. continue;
  3345. }
  3346. if ((p2 = strchr(p1, ' '))) {
  3347. len = p2 - p1;
  3348. } else {
  3349. len = strlen(p1);
  3350. p2 = p1 + len;
  3351. }
  3352. /* we use n + 2 in realloc for list,because we add
  3353. * new element and then we will add NULL element */
  3354. list = xrealloc(list, sizeof(*list) * (n + 2));
  3355. list[n] = xmalloc(2 + name_len + len);
  3356. strcpy(list[n], name);
  3357. strcat(list[n], "=");
  3358. strncat(list[n], p1, len);
  3359. list[n++][name_len + len + 1] = '\0';
  3360. p1 = p2;
  3361. }
  3362. if (p3 != inp[i]) free(p3);
  3363. }
  3364. list[n] = NULL;
  3365. return list;
  3366. }
  3367. /*
  3368. * Make new string for parser
  3369. * inp - array of argument strings to flatten
  3370. * nonnull - indicates argument was quoted when originally parsed
  3371. */
  3372. static char *make_string(char **inp, int *nonnull)
  3373. {
  3374. char *p;
  3375. char *str = NULL;
  3376. int n;
  3377. int len = 2;
  3378. char *noeval_str;
  3379. int noeval = 0;
  3380. noeval_str = get_local_var("HUSH_NO_EVAL");
  3381. if (noeval_str != NULL && *noeval_str != '0' && *noeval_str != '\0')
  3382. noeval = 1;
  3383. for (n = 0; inp[n]; n++) {
  3384. p = insert_var_value_sub(inp[n], noeval);
  3385. str = xrealloc(str, (len + strlen(p) + (2 * nonnull[n])));
  3386. if (n) {
  3387. strcat(str, " ");
  3388. } else {
  3389. *str = '\0';
  3390. }
  3391. if (nonnull[n])
  3392. strcat(str, "'");
  3393. strcat(str, p);
  3394. if (nonnull[n])
  3395. strcat(str, "'");
  3396. len = strlen(str) + 3;
  3397. if (p != inp[n]) free(p);
  3398. }
  3399. len = strlen(str);
  3400. *(str + len) = '\n';
  3401. *(str + len + 1) = '\0';
  3402. return str;
  3403. }
  3404. #ifdef __U_BOOT__
  3405. static int do_showvar(struct cmd_tbl *cmdtp, int flag, int argc,
  3406. char *const argv[])
  3407. {
  3408. int i, k;
  3409. int rcode = 0;
  3410. struct variables *cur;
  3411. if (argc == 1) { /* Print all env variables */
  3412. for (cur = top_vars; cur; cur = cur->next) {
  3413. printf ("%s=%s\n", cur->name, cur->value);
  3414. if (ctrlc ()) {
  3415. puts ("\n ** Abort\n");
  3416. return 1;
  3417. }
  3418. }
  3419. return 0;
  3420. }
  3421. for (i = 1; i < argc; ++i) { /* print single env variables */
  3422. char *name = argv[i];
  3423. k = -1;
  3424. for (cur = top_vars; cur; cur = cur->next) {
  3425. if(strcmp (cur->name, name) == 0) {
  3426. k = 0;
  3427. printf ("%s=%s\n", cur->name, cur->value);
  3428. }
  3429. if (ctrlc ()) {
  3430. puts ("\n ** Abort\n");
  3431. return 1;
  3432. }
  3433. }
  3434. if (k < 0) {
  3435. printf ("## Error: \"%s\" not defined\n", name);
  3436. rcode ++;
  3437. }
  3438. }
  3439. return rcode;
  3440. }
  3441. U_BOOT_CMD(
  3442. showvar, CONFIG_SYS_MAXARGS, 1, do_showvar,
  3443. "print local hushshell variables",
  3444. "\n - print values of all hushshell variables\n"
  3445. "showvar name ...\n"
  3446. " - print value of hushshell variable 'name'"
  3447. );
  3448. #endif
  3449. /****************************************************************************/