1 /***********************************************************************
2 * *
3 * This software is part of the ast package *
4 * Copyright (c) 1982-2011 AT&T Intellectual Property *
5 * and is licensed under the *
6 * Eclipse Public License, Version 1.0 *
7 * by AT&T Intellectual Property *
8 * *
9 * A copy of the License is available at *
10 * http://www.eclipse.org/org/documents/epl-v10.html *
11 * (with md5 checksum b35adb5213ca9657e911e9befb180842) *
12 * *
13 * Information and Software Systems Research *
14 * AT&T Research *
15 * Florham Park NJ *
16 * *
17 * David Korn <dgk@research.att.com> *
18 * *
19 ***********************************************************************/
20 #pragma prototyped
21 /*
22 * David Korn
23 * AT&T Labs
24 *
25 * shell parse tree dump
26 *
27 */
28
29 #include "defs.h"
30 #include "shnodes.h"
31 #include "path.h"
32 #include "io.h"
33 #include <ccode.h>
34
35 static int p_comlist(const struct dolnod*);
36 static int p_arg(const struct argnod*);
37 static int p_comarg(const struct comnod*);
38 static int p_redirect(const struct ionod*);
39 static int p_switch(const struct regnod*);
40 static int p_tree(const Shnode_t*);
41 static int p_string(const char*);
42
43 static Sfio_t *outfile;
44
sh_tdump(Sfio_t * out,const Shnode_t * t)45 int sh_tdump(Sfio_t *out, const Shnode_t *t)
46 {
47 outfile = out;
48 return(p_tree(t));
49 }
50
51 /*
52 * convert to ASCII to write and back again if needed
53 */
outstring(Sfio_t * out,const char * string,int n)54 static int outstring(Sfio_t *out, const char *string, int n)
55 {
56 int r;
57 char *cp = (char*)string;
58 ccmaps(cp, n, CC_NATIVE, CC_ASCII);
59 r = sfwrite(out,cp,n);
60 ccmaps(cp, n, CC_ASCII, CC_NATIVE);
61 return(r);
62 }
63
64 /*
65 * print script corresponding to shell tree <t>
66 */
p_tree(register const Shnode_t * t)67 static int p_tree(register const Shnode_t *t)
68 {
69 if(!t)
70 return(sfputl(outfile,-1));
71 if(sfputl(outfile,t->tre.tretyp)<0)
72 return(-1);
73 switch(t->tre.tretyp&COMMSK)
74 {
75 case TTIME:
76 case TPAR:
77 return(p_tree(t->par.partre));
78 case TCOM:
79 return(p_comarg((struct comnod*)t));
80 case TSETIO:
81 case TFORK:
82 if(sfputu(outfile,t->fork.forkline)<0)
83 return(-1);
84 if(p_tree(t->fork.forktre)<0)
85 return(-1);
86 return(p_redirect(t->fork.forkio));
87 case TIF:
88 if(p_tree(t->if_.iftre)<0)
89 return(-1);
90 if(p_tree(t->if_.thtre)<0)
91 return(-1);
92 return(p_tree(t->if_.eltre));
93 case TWH:
94 if(t->wh.whinc)
95 {
96 if(p_tree((Shnode_t*)(t->wh.whinc))<0)
97 return(-1);
98 }
99 else
100 {
101 if(sfputl(outfile,-1)<0)
102 return(-1);
103 }
104 if(p_tree(t->wh.whtre)<0)
105 return(-1);
106 return(p_tree(t->wh.dotre));
107 case TLST:
108 case TAND:
109 case TORF:
110 case TFIL:
111 if(p_tree(t->lst.lstlef)<0)
112 return(-1);
113 return(p_tree(t->lst.lstrit));
114 case TARITH:
115 if(sfputu(outfile,t->ar.arline)<0)
116 return(-1);
117 return(p_arg(t->ar.arexpr));
118 case TFOR:
119 if(sfputu(outfile,t->for_.forline)<0)
120 return(-1);
121 if(p_tree(t->for_.fortre)<0)
122 return(-1);
123 if(p_string(t->for_.fornam)<0)
124 return(-1);
125 return(p_tree((Shnode_t*)t->for_.forlst));
126 case TSW:
127 if(sfputu(outfile,t->sw.swline)<0)
128 return(-1);
129 if(p_arg(t->sw.swarg)<0)
130 return(-1);
131 return(p_switch(t->sw.swlst));
132 case TFUN:
133 if(sfputu(outfile,t->funct.functline)<0)
134 return(-1);
135 if(p_string(t->funct.functnam)<0)
136 return(-1);
137 if(p_tree(t->funct.functtre)<0)
138 return(-1);
139 return(p_tree((Shnode_t*)t->funct.functargs));
140 case TTST:
141 if(sfputu(outfile,t->tst.tstline)<0)
142 return(-1);
143 if((t->tre.tretyp&TPAREN)==TPAREN)
144 return(p_tree(t->lst.lstlef));
145 else
146 {
147 if(p_arg(&(t->lst.lstlef->arg))<0)
148 return(-1);
149 if((t->tre.tretyp&TBINARY))
150 return(p_arg(&(t->lst.lstrit->arg)));
151 return(0);
152 }
153 }
154 return(-1);
155 }
156
p_arg(register const struct argnod * arg)157 static int p_arg(register const struct argnod *arg)
158 {
159 register int n;
160 struct fornod *fp;
161 while(arg)
162 {
163 if((n = strlen(arg->argval)) || (arg->argflag&~(ARG_APPEND|ARG_MESSAGE|ARG_QUOTED)))
164 fp=0;
165 else
166 {
167 fp=(struct fornod*)arg->argchn.ap;
168 n = strlen(fp->fornam)+1;
169 }
170 sfputu(outfile,n+1);
171 if(fp)
172 {
173 sfputc(outfile,0);
174 outstring(outfile,fp->fornam,n-1);
175 }
176 else
177 outstring(outfile,arg->argval,n);
178 sfputc(outfile,arg->argflag);
179 if(fp)
180 {
181 sfputu(outfile,fp->fortyp);
182 p_tree(fp->fortre);
183 }
184 else if(n==0 && (arg->argflag&ARG_EXP) && arg->argchn.ap)
185 p_tree((Shnode_t*)arg->argchn.ap);
186 arg = arg->argnxt.ap;
187 }
188 return(sfputu(outfile,0));
189 }
190
p_redirect(register const struct ionod * iop)191 static int p_redirect(register const struct ionod *iop)
192 {
193 while(iop)
194 {
195 if(iop->iovname)
196 sfputl(outfile,iop->iofile|IOVNM);
197 else
198 sfputl(outfile,iop->iofile);
199 p_string(iop->ioname);
200 if(iop->iodelim)
201 {
202 p_string(iop->iodelim);
203 sfputl(outfile,iop->iosize);
204 sfseek(sh.heredocs,iop->iooffset,SEEK_SET);
205 sfmove(sh.heredocs,outfile, iop->iosize,-1);
206 }
207 else
208 sfputu(outfile,0);
209 if(iop->iovname)
210 p_string(iop->iovname);
211 iop = iop->ionxt;
212 }
213 return(sfputl(outfile,-1));
214 }
215
p_comarg(register const struct comnod * com)216 static int p_comarg(register const struct comnod *com)
217 {
218 p_redirect(com->comio);
219 p_arg(com->comset);
220 if(!com->comarg)
221 sfputl(outfile,-1);
222 else if(com->comtyp&COMSCAN)
223 p_arg(com->comarg);
224 else
225 p_comlist((struct dolnod*)com->comarg);
226 return(sfputu(outfile,com->comline));
227 }
228
p_comlist(const struct dolnod * dol)229 static int p_comlist(const struct dolnod *dol)
230 {
231 register char *cp, *const*argv;
232 register int n;
233 argv = dol->dolval+ARG_SPARE;
234 while(cp = *argv)
235 argv++;
236 n = argv - (dol->dolval+1);
237 sfputl(outfile,n);
238 argv = dol->dolval+ARG_SPARE;
239 while(cp = *argv++)
240 p_string(cp);
241 return(sfputu(outfile,0));
242 }
243
p_switch(register const struct regnod * reg)244 static int p_switch(register const struct regnod *reg)
245 {
246 while(reg)
247 {
248 sfputl(outfile,reg->regflag);
249 p_arg(reg->regptr);
250 p_tree(reg->regcom);
251 reg = reg->regnxt;
252 }
253 return(sfputl(outfile,-1));
254 }
255
p_string(register const char * string)256 static int p_string(register const char *string)
257 {
258 register size_t n=strlen(string);
259 if(sfputu(outfile,n+1)<0)
260 return(-1);
261 return(outstring(outfile,string,n));
262 }
263