1 564ca709 2004-04-19 devnull #include <u.h>
  2 564ca709 2004-04-19 devnull #include <libc.h>
  3 564ca709 2004-04-19 devnull #include <bio.h>
  4 564ca709 2004-04-19 devnull #include <mach.h>
  5 564ca709 2004-04-19 devnull #define Extern
  6 564ca709 2004-04-19 devnull #include "acid.h"
  7 564ca709 2004-04-19 devnull #include "y.tab.h"
  8 564ca709 2004-04-19 devnull 
  9 564ca709 2004-04-19 devnull extern int __ifmt(Fmt*);
 10 564ca709 2004-04-19 devnull 
 11 564ca709 2004-04-19 devnull static Biobuf	bioout;
 12 564ca709 2004-04-19 devnull static char*	lm[16];
 13 564ca709 2004-04-19 devnull static int	nlm;
 14 564ca709 2004-04-19 devnull static char*	mtype;
 15 564ca709 2004-04-19 devnull 
 16 564ca709 2004-04-19 devnull static	int attachfiles(int, char**);
 17 564ca709 2004-04-19 devnull int	xfmt(Fmt*);
 18 564ca709 2004-04-19 devnull int	isnumeric(char*);
 19 564ca709 2004-04-19 devnull void	die(void);
 20 564ca709 2004-04-19 devnull void	setcore(Fhdr*);
 21 564ca709 2004-04-19 devnull 
 22 564ca709 2004-04-19 devnull void
 23 564ca709 2004-04-19 devnull usage(void)
 24 564ca709 2004-04-19 devnull {
 25 564ca709 2004-04-19 devnull 	fprint(2, "usage: acid [-c core] [-l module] [-m machine] [-qrw] [-k] [pid] [file]\n");
 26 564ca709 2004-04-19 devnull 	exits("usage");
 27 f9aa99c7 2004-06-09 devnull }
 28 f9aa99c7 2004-06-09 devnull 
 29 f9aa99c7 2004-06-09 devnull Map*
 30 f9aa99c7 2004-06-09 devnull dumbmap(int fd)
 31 f9aa99c7 2004-06-09 devnull {
 32 f9aa99c7 2004-06-09 devnull 	Map *dumb;
 33 f9aa99c7 2004-06-09 devnull 	Seg s;
 34 f9aa99c7 2004-06-09 devnull 
 35 f9aa99c7 2004-06-09 devnull 	dumb = allocmap();
 36 f9aa99c7 2004-06-09 devnull 	memset(&s, 0, sizeof s);
 37 f9aa99c7 2004-06-09 devnull 	s.fd = fd;
 38 f9aa99c7 2004-06-09 devnull 	s.base = 0;
 39 f9aa99c7 2004-06-09 devnull 	s.offset = 0;
 40 f9aa99c7 2004-06-09 devnull 	s.size = 0xFFFFFFFF;
 41 f9aa99c7 2004-06-09 devnull 	s.name = "data";
 42 f9aa99c7 2004-06-09 devnull 	s.file = "<dumb>";
 43 f9aa99c7 2004-06-09 devnull 	if(addseg(dumb, s) < 0){
 44 f9aa99c7 2004-06-09 devnull 		freemap(dumb);
 45 f9aa99c7 2004-06-09 devnull 		return nil;
 46 f9aa99c7 2004-06-09 devnull 	}
 47 f9aa99c7 2004-06-09 devnull 	if(mach == nil)
 48 f9aa99c7 2004-06-09 devnull 		mach = machcpu;
 49 f9aa99c7 2004-06-09 devnull 	return dumb;
 50 564ca709 2004-04-19 devnull }
 51 564ca709 2004-04-19 devnull 
 52 564ca709 2004-04-19 devnull void
 53 564ca709 2004-04-19 devnull main(int argc, char *argv[])
 54 564ca709 2004-04-19 devnull {
 55 f7b74c17 2004-12-28 devnull 	Lsym *volatile l;
 56 564ca709 2004-04-19 devnull 	Node *n;
 57 564ca709 2004-04-19 devnull 	char buf[128], *s;
 58 564ca709 2004-04-19 devnull 	int pid, i;
 59 564ca709 2004-04-19 devnull 
 60 564ca709 2004-04-19 devnull 	argv0 = argv[0];
 61 564ca709 2004-04-19 devnull 	pid = 0;
 62 564ca709 2004-04-19 devnull 	quiet = 1;
 63 564ca709 2004-04-19 devnull 
 64 564ca709 2004-04-19 devnull 	mtype = 0;
 65 564ca709 2004-04-19 devnull 	ARGBEGIN{
 66 564ca709 2004-04-19 devnull 	case 'A':
 67 564ca709 2004-04-19 devnull 		abort();
 68 564ca709 2004-04-19 devnull 		break;
 69 564ca709 2004-04-19 devnull 	case 'm':
 70 564ca709 2004-04-19 devnull 		mtype = ARGF();
 71 564ca709 2004-04-19 devnull 		break;
 72 564ca709 2004-04-19 devnull 	case 'w':
 73 564ca709 2004-04-19 devnull 		wtflag = 1;
 74 564ca709 2004-04-19 devnull 		break;
 75 564ca709 2004-04-19 devnull 	case 'l':
 76 564ca709 2004-04-19 devnull 		s = ARGF();
 77 564ca709 2004-04-19 devnull 		if(s == 0)
 78 564ca709 2004-04-19 devnull 			usage();
 79 564ca709 2004-04-19 devnull 		lm[nlm++] = s;
 80 564ca709 2004-04-19 devnull 		break;
 81 564ca709 2004-04-19 devnull 	case 'k':
 82 564ca709 2004-04-19 devnull 		kernel++;
 83 564ca709 2004-04-19 devnull 		break;
 84 564ca709 2004-04-19 devnull 	case 'q':
 85 564ca709 2004-04-19 devnull 		quiet = 0;
 86 564ca709 2004-04-19 devnull 		break;
 87 564ca709 2004-04-19 devnull 	case 'r':
 88 564ca709 2004-04-19 devnull 		pid = 1;
 89 564ca709 2004-04-19 devnull 		remote++;
 90 564ca709 2004-04-19 devnull 		kernel++;
 91 564ca709 2004-04-19 devnull 		break;
 92 564ca709 2004-04-19 devnull 	default:
 93 564ca709 2004-04-19 devnull 		usage();
 94 564ca709 2004-04-19 devnull 	}ARGEND
 95 564ca709 2004-04-19 devnull 
 96 564ca709 2004-04-19 devnull 	fmtinstall('x', xfmt);
 97 564ca709 2004-04-19 devnull 	fmtinstall('Z', Zfmt);
 98 564ca709 2004-04-19 devnull 	fmtinstall('L', locfmt);
 99 564ca709 2004-04-19 devnull 	Binit(&bioout, 1, OWRITE);
100 564ca709 2004-04-19 devnull 	bout = &bioout;
101 564ca709 2004-04-19 devnull 
102 e637c944 2004-04-20 devnull 	initexpr();
103 d2cbcb77 2004-04-20 devnull 	initprint();
104 564ca709 2004-04-19 devnull 	kinit();
105 564ca709 2004-04-19 devnull 	initialising = 1;
106 564ca709 2004-04-19 devnull 	pushfile(0);
107 564ca709 2004-04-19 devnull 	loadvars();
108 564ca709 2004-04-19 devnull 	installbuiltin();
109 564ca709 2004-04-19 devnull 
110 564ca709 2004-04-19 devnull 	if(mtype && machbyname(mtype) == 0)
111 564ca709 2004-04-19 devnull 		print("unknown machine %s", mtype);
112 564ca709 2004-04-19 devnull 
113 564ca709 2004-04-19 devnull 	if (attachfiles(argc, argv) < 0)
114 564ca709 2004-04-19 devnull 		varreg();		/* use default register set on error */
115 564ca709 2004-04-19 devnull 	if(mach == nil)
116 564ca709 2004-04-19 devnull 		mach = machcpu;
117 564ca709 2004-04-19 devnull 
118 564ca709 2004-04-19 devnull 	symhdr = nil;	/* not supposed to use this anymore */
119 564ca709 2004-04-19 devnull 
120 564ca709 2004-04-19 devnull 	l = mkvar("acid");
121 564ca709 2004-04-19 devnull 	l->v->set = 1;
122 564ca709 2004-04-19 devnull 	l->v->type = TLIST;
123 564ca709 2004-04-19 devnull 	l->v->store.u.l = nil;
124 564ca709 2004-04-19 devnull 
125 13e66070 2004-04-20 devnull 	loadmodule(unsharp("#9/acid/port"));
126 564ca709 2004-04-19 devnull 	for(i = 0; i < nlm; i++) {
127 564ca709 2004-04-19 devnull 		if(access(lm[i], AREAD) >= 0)
128 564ca709 2004-04-19 devnull 			loadmodule(lm[i]);
129 564ca709 2004-04-19 devnull 		else {
130 13e66070 2004-04-20 devnull 			sprint(buf, "#9/acid/%s", lm[i]);
131 13e66070 2004-04-20 devnull 			loadmodule(unsharp(buf));
132 564ca709 2004-04-19 devnull 		}
133 564ca709 2004-04-19 devnull 	}
134 564ca709 2004-04-19 devnull 
135 564ca709 2004-04-19 devnull 	userinit();
136 564ca709 2004-04-19 devnull 	varsym();
137 564ca709 2004-04-19 devnull 
138 564ca709 2004-04-19 devnull 	l = look("acidmap");
139 564ca709 2004-04-19 devnull 	if(l && l->proc) {
140 b4a659b6 2004-04-19 devnull 		if(setjmp(err) == 0){
141 b4a659b6 2004-04-19 devnull 			n = an(ONAME, ZN, ZN);
142 b4a659b6 2004-04-19 devnull 			n->sym = l;
143 b4a659b6 2004-04-19 devnull 			n = an(OCALL, n, ZN);
144 b4a659b6 2004-04-19 devnull 			execute(n);
145 b4a659b6 2004-04-19 devnull 		}
146 564ca709 2004-04-19 devnull 	}
147 564ca709 2004-04-19 devnull 
148 564ca709 2004-04-19 devnull 	interactive = 1;
149 564ca709 2004-04-19 devnull 	initialising = 0;
150 564ca709 2004-04-19 devnull 	line = 1;
151 564ca709 2004-04-19 devnull 
152 564ca709 2004-04-19 devnull 	notify(catcher);
153 564ca709 2004-04-19 devnull 
154 564ca709 2004-04-19 devnull 	for(;;) {
155 564ca709 2004-04-19 devnull 		if(setjmp(err)) {
156 564ca709 2004-04-19 devnull 			Binit(&bioout, 1, OWRITE);
157 564ca709 2004-04-19 devnull 			unwind();
158 564ca709 2004-04-19 devnull 		}
159 564ca709 2004-04-19 devnull 		stacked = 0;
160 564ca709 2004-04-19 devnull 
161 564ca709 2004-04-19 devnull 		Bprint(bout, "acid; ");
162 564ca709 2004-04-19 devnull 
163 564ca709 2004-04-19 devnull 		if(yyparse() != 1)
164 564ca709 2004-04-19 devnull 			die();
165 564ca709 2004-04-19 devnull 		restartio();
166 564ca709 2004-04-19 devnull 
167 564ca709 2004-04-19 devnull 		unwind();
168 564ca709 2004-04-19 devnull 	}
169 564ca709 2004-04-19 devnull 	Bputc(bout, '\n');
170 564ca709 2004-04-19 devnull 	exits(0);
171 564ca709 2004-04-19 devnull }
172 564ca709 2004-04-19 devnull 
173 564ca709 2004-04-19 devnull static int
174 564ca709 2004-04-19 devnull attachfiles(int argc, char **argv)
175 564ca709 2004-04-19 devnull {
176 f7b74c17 2004-12-28 devnull 	int fd;
177 f7b74c17 2004-12-28 devnull 	volatile int pid;
178 564ca709 2004-04-19 devnull 	char *s;
179 564ca709 2004-04-19 devnull 	int i, omode;
180 564ca709 2004-04-19 devnull 	Fhdr *hdr;
181 564ca709 2004-04-19 devnull 	Lsym *l;
182 564ca709 2004-04-19 devnull 	Value *v;
183 564ca709 2004-04-19 devnull 
184 b4a659b6 2004-04-19 devnull 	pid = 0;
185 564ca709 2004-04-19 devnull 	interactive = 0;
186 564ca709 2004-04-19 devnull 	if(setjmp(err))
187 564ca709 2004-04-19 devnull 		return -1;
188 564ca709 2004-04-19 devnull 
189 564ca709 2004-04-19 devnull 	/*
190 564ca709 2004-04-19 devnull 	 * Unix and Plan 9 differ on what the right order of pid, text, and core is.
191 564ca709 2004-04-19 devnull 	 * I never remember anyway.  Let's just accept them in any order.
192 564ca709 2004-04-19 devnull 	 */
193 564ca709 2004-04-19 devnull 	omode = wtflag ? ORDWR : OREAD;
194 564ca709 2004-04-19 devnull 	for(i=0; i<argc; i++){
195 564ca709 2004-04-19 devnull 		if(isnumeric(argv[i])){
196 564ca709 2004-04-19 devnull 			if(pid){
197 564ca709 2004-04-19 devnull 				fprint(2, "already have pid %d; ignoring pid %d\n", pid, argv[i]);
198 564ca709 2004-04-19 devnull 				continue;
199 564ca709 2004-04-19 devnull 			}
200 564ca709 2004-04-19 devnull 			if(corhdr){
201 564ca709 2004-04-19 devnull 				fprint(2, "already have core %s; ignoring pid %d\n", corfil, pid);
202 564ca709 2004-04-19 devnull 				continue;
203 564ca709 2004-04-19 devnull 			}
204 564ca709 2004-04-19 devnull 			pid = atoi(argv[i]);
205 564ca709 2004-04-19 devnull 			continue;
206 564ca709 2004-04-19 devnull 		}
207 564ca709 2004-04-19 devnull 		if((hdr = crackhdr(argv[i], omode)) == nil){
208 564ca709 2004-04-19 devnull 			fprint(2, "crackhdr %s: %r\n", argv[i]);
209 f9aa99c7 2004-06-09 devnull 			if(argc == 1 && (fd = open(argv[i], omode)) > 0){
210 f9aa99c7 2004-06-09 devnull 				fprint(2, "loading %s direct mapped\n", argv[i]);
211 f9aa99c7 2004-06-09 devnull 				symmap = dumbmap(fd);
212 f9aa99c7 2004-06-09 devnull 				cormap = dumbmap(fd);
213 f9aa99c7 2004-06-09 devnull 				symfil = argv[i];
214 f9aa99c7 2004-06-09 devnull 				corfil = argv[i];
215 f9aa99c7 2004-06-09 devnull 				goto Run;
216 f9aa99c7 2004-06-09 devnull 			}
217 564ca709 2004-04-19 devnull 			continue;
218 564ca709 2004-04-19 devnull 		}
219 564ca709 2004-04-19 devnull 		fprint(2, "%s: %s %s %s\n", argv[i], hdr->aname, hdr->mname, hdr->fname);
220 564ca709 2004-04-19 devnull 		if(hdr->ftype == FCORE){
221 564ca709 2004-04-19 devnull 			if(pid){
222 564ca709 2004-04-19 devnull 				fprint(2, "already have pid %d; ignoring core %s\n", pid, argv[i]);
223 564ca709 2004-04-19 devnull 				uncrackhdr(hdr);
224 564ca709 2004-04-19 devnull 				continue;
225 564ca709 2004-04-19 devnull 			}
226 564ca709 2004-04-19 devnull 			if(corhdr){
227 564ca709 2004-04-19 devnull 				fprint(2, "already have core %s; ignoring core %s\n", corfil, argv[i]);
228 564ca709 2004-04-19 devnull 				uncrackhdr(hdr);
229 564ca709 2004-04-19 devnull 				continue;
230 564ca709 2004-04-19 devnull 			}
231 564ca709 2004-04-19 devnull 			corhdr = hdr;
232 564ca709 2004-04-19 devnull 			corfil = argv[i];
233 564ca709 2004-04-19 devnull 		}else{
234 564ca709 2004-04-19 devnull 			if(symhdr){
235 564ca709 2004-04-19 devnull 				fprint(2, "already have text %s; ignoring text %s\n", symfil, argv[i]);
236 564ca709 2004-04-19 devnull 				uncrackhdr(hdr);
237 564ca709 2004-04-19 devnull 				continue;
238 564ca709 2004-04-19 devnull 			}
239 564ca709 2004-04-19 devnull 			symhdr = hdr;
240 564ca709 2004-04-19 devnull 			symfil = argv[i];
241 564ca709 2004-04-19 devnull 		}
242 564ca709 2004-04-19 devnull 	}
243 564ca709 2004-04-19 devnull 
244 564ca709 2004-04-19 devnull 	if(symhdr==nil){
245 564ca709 2004-04-19 devnull 		symfil = "a.out";
246 564ca709 2004-04-19 devnull 		if(pid){
247 564ca709 2004-04-19 devnull 			if((s = proctextfile(pid)) != nil){
248 564ca709 2004-04-19 devnull 				fprint(2, "pid %d: text %s\n", pid, s);
249 564ca709 2004-04-19 devnull 				symfil = s;
250 564ca709 2004-04-19 devnull 			}
251 564ca709 2004-04-19 devnull 		}
252 564ca709 2004-04-19 devnull 		/* XXX pull command from core */
253 564ca709 2004-04-19 devnull 
254 c2a9c3bb 2004-04-19 devnull 		if((symhdr = crackhdr(symfil, omode)) == nil){
255 564ca709 2004-04-19 devnull 			fprint(2, "crackhdr %s: %r\n", symfil);
256 564ca709 2004-04-19 devnull 			symfil = nil;
257 564ca709 2004-04-19 devnull 		}
258 564ca709 2004-04-19 devnull 	}
259 564ca709 2004-04-19 devnull 
260 564ca709 2004-04-19 devnull 	if(symhdr)
261 8b549a62 2005-01-07 devnull 		symopen(symhdr);
262 564ca709 2004-04-19 devnull 
263 564ca709 2004-04-19 devnull 	if(!mach)
264 564ca709 2004-04-19 devnull 		mach = machcpu;
265 564ca709 2004-04-19 devnull 
266 564ca709 2004-04-19 devnull 	/*
267 564ca709 2004-04-19 devnull 	 * Set up maps.
268 564ca709 2004-04-19 devnull 	 */
269 564ca709 2004-04-19 devnull 	symmap = allocmap();
270 564ca709 2004-04-19 devnull 	cormap = allocmap();
271 564ca709 2004-04-19 devnull 	if(symmap == nil || cormap == nil)
272 564ca709 2004-04-19 devnull 		sysfatal("allocating maps: %r");
273 564ca709 2004-04-19 devnull 
274 564ca709 2004-04-19 devnull 	if(symhdr){
275 564ca709 2004-04-19 devnull 		if(mapfile(symhdr, 0, symmap, nil) < 0)
276 564ca709 2004-04-19 devnull 			fprint(2, "mapping %s: %r\n", symfil);
277 564ca709 2004-04-19 devnull 		mapfile(symhdr, 0, cormap, nil);
278 564ca709 2004-04-19 devnull 	}
279 564ca709 2004-04-19 devnull 
280 f9aa99c7 2004-06-09 devnull Run:
281 564ca709 2004-04-19 devnull 	l = mkvar("objtype");
282 564ca709 2004-04-19 devnull 	v = l->v;
283 564ca709 2004-04-19 devnull 	v->store.fmt = 's';
284 564ca709 2004-04-19 devnull 	v->set = 1;
285 564ca709 2004-04-19 devnull 	v->store.u.string = strnode(mach->name);
286 564ca709 2004-04-19 devnull 	v->type = TSTRING;
287 564ca709 2004-04-19 devnull 
288 564ca709 2004-04-19 devnull 	l = mkvar("textfile");
289 564ca709 2004-04-19 devnull 	v = l->v;
290 564ca709 2004-04-19 devnull 	v->store.fmt = 's';
291 564ca709 2004-04-19 devnull 	v->set = 1;
292 564ca709 2004-04-19 devnull 	v->store.u.string = strnode(symfil ? symfil : "");
293 564ca709 2004-04-19 devnull 	v->type = TSTRING;
294 564ca709 2004-04-19 devnull 
295 564ca709 2004-04-19 devnull 	l = mkvar("systype");
296 564ca709 2004-04-19 devnull 	v = l->v;
297 564ca709 2004-04-19 devnull 	v->store.fmt = 's';
298 564ca709 2004-04-19 devnull 	v->set = 1;
299 564ca709 2004-04-19 devnull 	v->store.u.string = strnode(symhdr ? symhdr->aname : "");
300 564ca709 2004-04-19 devnull 	v->type = TSTRING;
301 564ca709 2004-04-19 devnull 
302 564ca709 2004-04-19 devnull 	l = mkvar("corefile");
303 564ca709 2004-04-19 devnull 	v = l->v;
304 564ca709 2004-04-19 devnull 	v->store.fmt = 's';
305 564ca709 2004-04-19 devnull 	v->set = 1;
306 564ca709 2004-04-19 devnull 	v->store.u.string = strnode(corfil ? corfil : "");
307 564ca709 2004-04-19 devnull 	v->type = TSTRING;
308 564ca709 2004-04-19 devnull 
309 564ca709 2004-04-19 devnull 	if(pid)
310 564ca709 2004-04-19 devnull 		sproc(pid);
311 564ca709 2004-04-19 devnull 	if(corhdr)
312 564ca709 2004-04-19 devnull 		setcore(corhdr);
313 564ca709 2004-04-19 devnull 	varreg();
314 564ca709 2004-04-19 devnull 	return 0;
315 564ca709 2004-04-19 devnull }
316 564ca709 2004-04-19 devnull 
317 564ca709 2004-04-19 devnull void
318 564ca709 2004-04-19 devnull setcore(Fhdr *hdr)
319 564ca709 2004-04-19 devnull {
320 564ca709 2004-04-19 devnull 	unmapproc(cormap);
321 564ca709 2004-04-19 devnull 	unmapfile(corhdr, cormap);
322 564ca709 2004-04-19 devnull 	free(correg);
323 564ca709 2004-04-19 devnull 	correg = nil;
324 564ca709 2004-04-19 devnull 
325 564ca709 2004-04-19 devnull 	if(hdr == nil)
326 564ca709 2004-04-19 devnull 		error("no core");
327 564ca709 2004-04-19 devnull 	if(mapfile(hdr, 0, cormap, &correg) < 0)
328 564ca709 2004-04-19 devnull 		error("mapfile %s: %r", hdr->filename);
329 564ca709 2004-04-19 devnull 	corhdr = hdr;
330 564ca709 2004-04-19 devnull 	corfil = hdr->filename;
331 564ca709 2004-04-19 devnull }
332 564ca709 2004-04-19 devnull 
333 564ca709 2004-04-19 devnull void
334 564ca709 2004-04-19 devnull die(void)
335 564ca709 2004-04-19 devnull {
336 564ca709 2004-04-19 devnull 	Lsym *s;
337 564ca709 2004-04-19 devnull 	List *f;
338 564ca709 2004-04-19 devnull 
339 564ca709 2004-04-19 devnull 	Bprint(bout, "\n");
340 564ca709 2004-04-19 devnull 
341 564ca709 2004-04-19 devnull 	s = look("proclist");
342 564ca709 2004-04-19 devnull 	if(s && s->v->type == TLIST) {
343 564ca709 2004-04-19 devnull 		for(f = s->v->store.u.l; f; f = f->next){
344 564ca709 2004-04-19 devnull 			detachproc((int)f->store.u.ival);
345 564ca709 2004-04-19 devnull 			Bprint(bout, "/bin/kill -9 %d\n", (int)f->store.u.ival);
346 564ca709 2004-04-19 devnull 		}
347 564ca709 2004-04-19 devnull 	}
348 564ca709 2004-04-19 devnull 	exits(0);
349 564ca709 2004-04-19 devnull }
350 564ca709 2004-04-19 devnull 
351 564ca709 2004-04-19 devnull void
352 564ca709 2004-04-19 devnull userinit(void)
353 564ca709 2004-04-19 devnull {
354 564ca709 2004-04-19 devnull 	Lsym *l;
355 564ca709 2004-04-19 devnull 	Node *n;
356 564ca709 2004-04-19 devnull 	char buf[128], *p;
357 564ca709 2004-04-19 devnull 
358 13e66070 2004-04-20 devnull 	sprint(buf, "#9/acid/%s", mach->name);
359 13e66070 2004-04-20 devnull 	loadmodule(unsharp(buf));
360 c8b6342d 2005-01-13 devnull 	p = getenv("HOME");
361 564ca709 2004-04-19 devnull 	if(p != 0) {
362 564ca709 2004-04-19 devnull 		sprint(buf, "%s/lib/acid", p);
363 564ca709 2004-04-19 devnull 		silent = 1;
364 564ca709 2004-04-19 devnull 		loadmodule(buf);
365 564ca709 2004-04-19 devnull 	}
366 564ca709 2004-04-19 devnull 
367 564ca709 2004-04-19 devnull 	interactive = 0;
368 564ca709 2004-04-19 devnull 	if(setjmp(err)) {
369 564ca709 2004-04-19 devnull 		unwind();
370 564ca709 2004-04-19 devnull 		return;
371 564ca709 2004-04-19 devnull 	}
372 564ca709 2004-04-19 devnull 	l = look("acidinit");
373 564ca709 2004-04-19 devnull 	if(l && l->proc) {
374 564ca709 2004-04-19 devnull 		n = an(ONAME, ZN, ZN);
375 564ca709 2004-04-19 devnull 		n->sym = l;
376 564ca709 2004-04-19 devnull 		n = an(OCALL, n, ZN);
377 564ca709 2004-04-19 devnull 		execute(n);
378 564ca709 2004-04-19 devnull 	}
379 564ca709 2004-04-19 devnull }
380 564ca709 2004-04-19 devnull 
381 564ca709 2004-04-19 devnull void
382 564ca709 2004-04-19 devnull loadmodule(char *s)
383 564ca709 2004-04-19 devnull {
384 564ca709 2004-04-19 devnull 	interactive = 0;
385 564ca709 2004-04-19 devnull 	if(setjmp(err)) {
386 564ca709 2004-04-19 devnull 		unwind();
387 564ca709 2004-04-19 devnull 		return;
388 564ca709 2004-04-19 devnull 	}
389 564ca709 2004-04-19 devnull 	pushfile(s);
390 564ca709 2004-04-19 devnull 	silent = 0;
391 564ca709 2004-04-19 devnull 	yyparse();
392 564ca709 2004-04-19 devnull 	popio();
393 564ca709 2004-04-19 devnull 	return;
394 564ca709 2004-04-19 devnull }
395 564ca709 2004-04-19 devnull 
396 564ca709 2004-04-19 devnull Node*
397 564ca709 2004-04-19 devnull an(int op, Node *l, Node *r)
398 564ca709 2004-04-19 devnull {
399 564ca709 2004-04-19 devnull 	Node *n;
400 564ca709 2004-04-19 devnull 
401 564ca709 2004-04-19 devnull 	n = gmalloc(sizeof(Node));
402 564ca709 2004-04-19 devnull 	memset(n, 0, sizeof(Node));
403 564ca709 2004-04-19 devnull 	n->gc.gclink = gcl;
404 564ca709 2004-04-19 devnull 	gcl = (Gc*)n;
405 564ca709 2004-04-19 devnull 	n->op = op;
406 564ca709 2004-04-19 devnull 	n->left = l;
407 564ca709 2004-04-19 devnull 	n->right = r;
408 564ca709 2004-04-19 devnull 	return n;
409 564ca709 2004-04-19 devnull }
410 564ca709 2004-04-19 devnull 
411 564ca709 2004-04-19 devnull List*
412 564ca709 2004-04-19 devnull al(int t)
413 564ca709 2004-04-19 devnull {
414 564ca709 2004-04-19 devnull 	List *l;
415 564ca709 2004-04-19 devnull 
416 564ca709 2004-04-19 devnull 	l = gmalloc(sizeof(List));
417 564ca709 2004-04-19 devnull 	memset(l, 0, sizeof(List));
418 564ca709 2004-04-19 devnull 	l->type = t;
419 564ca709 2004-04-19 devnull 	l->gc.gclink = gcl;
420 564ca709 2004-04-19 devnull 	gcl = (Gc*)l;
421 564ca709 2004-04-19 devnull 	return l;
422 564ca709 2004-04-19 devnull }
423 564ca709 2004-04-19 devnull 
424 564ca709 2004-04-19 devnull Node*
425 564ca709 2004-04-19 devnull con(int v)
426 564ca709 2004-04-19 devnull {
427 564ca709 2004-04-19 devnull 	Node *n;
428 564ca709 2004-04-19 devnull 
429 564ca709 2004-04-19 devnull 	n = an(OCONST, ZN, ZN);
430 564ca709 2004-04-19 devnull 	n->store.u.ival = v;
431 564ca709 2004-04-19 devnull 	n->store.fmt = 'X';
432 564ca709 2004-04-19 devnull 	n->type = TINT;
433 564ca709 2004-04-19 devnull 	return n;
434 564ca709 2004-04-19 devnull }
435 564ca709 2004-04-19 devnull 
436 564ca709 2004-04-19 devnull void
437 564ca709 2004-04-19 devnull fatal(char *fmt, ...)
438 564ca709 2004-04-19 devnull {
439 564ca709 2004-04-19 devnull 	char buf[128];
440 564ca709 2004-04-19 devnull 	va_list arg;
441 564ca709 2004-04-19 devnull 
442 564ca709 2004-04-19 devnull 	va_start(arg, fmt);
443 564ca709 2004-04-19 devnull 	vseprint(buf, buf+sizeof(buf), fmt, arg);
444 564ca709 2004-04-19 devnull 	va_end(arg);
445 564ca709 2004-04-19 devnull 	fprint(2, "%s: %Z (fatal problem) %s\n", argv0, buf);
446 564ca709 2004-04-19 devnull 	exits(buf);
447 564ca709 2004-04-19 devnull }
448 564ca709 2004-04-19 devnull 
449 564ca709 2004-04-19 devnull void
450 564ca709 2004-04-19 devnull yyerror(char *fmt, ...)
451 564ca709 2004-04-19 devnull {
452 564ca709 2004-04-19 devnull 	char buf[128];
453 564ca709 2004-04-19 devnull 	va_list arg;
454 564ca709 2004-04-19 devnull 
455 564ca709 2004-04-19 devnull 	if(strcmp(fmt, "syntax error") == 0) {
456 564ca709 2004-04-19 devnull 		yyerror("syntax error, near symbol '%s'", symbol);
457 564ca709 2004-04-19 devnull 		return;
458 564ca709 2004-04-19 devnull 	}
459 564ca709 2004-04-19 devnull 	va_start(arg, fmt);
460 564ca709 2004-04-19 devnull 	vseprint(buf, buf+sizeof(buf), fmt, arg);
461 564ca709 2004-04-19 devnull 	va_end(arg);
462 564ca709 2004-04-19 devnull 	print("%Z: %s\n", buf);
463 564ca709 2004-04-19 devnull }
464 564ca709 2004-04-19 devnull 
465 564ca709 2004-04-19 devnull void
466 564ca709 2004-04-19 devnull marktree(Node *n)
467 564ca709 2004-04-19 devnull {
468 564ca709 2004-04-19 devnull 
469 564ca709 2004-04-19 devnull 	if(n == 0)
470 564ca709 2004-04-19 devnull 		return;
471 564ca709 2004-04-19 devnull 
472 564ca709 2004-04-19 devnull 	marktree(n->left);
473 564ca709 2004-04-19 devnull 	marktree(n->right);
474 564ca709 2004-04-19 devnull 
475 564ca709 2004-04-19 devnull 	n->gc.gcmark = 1;
476 564ca709 2004-04-19 devnull 	if(n->op != OCONST)
477 564ca709 2004-04-19 devnull 		return;
478 564ca709 2004-04-19 devnull 
479 564ca709 2004-04-19 devnull 	switch(n->type) {
480 564ca709 2004-04-19 devnull 	case TSTRING:
481 564ca709 2004-04-19 devnull 		n->store.u.string->gc.gcmark = 1;
482 564ca709 2004-04-19 devnull 		break;
483 564ca709 2004-04-19 devnull 	case TLIST:
484 564ca709 2004-04-19 devnull 		marklist(n->store.u.l);
485 564ca709 2004-04-19 devnull 		break;
486 564ca709 2004-04-19 devnull 	case TCODE:
487 564ca709 2004-04-19 devnull 		marktree(n->store.u.cc);
488 564ca709 2004-04-19 devnull 		break;
489 564ca709 2004-04-19 devnull 	}
490 564ca709 2004-04-19 devnull }
491 564ca709 2004-04-19 devnull 
492 564ca709 2004-04-19 devnull void
493 564ca709 2004-04-19 devnull marklist(List *l)
494 564ca709 2004-04-19 devnull {
495 564ca709 2004-04-19 devnull 	while(l) {
496 564ca709 2004-04-19 devnull 		l->gc.gcmark = 1;
497 564ca709 2004-04-19 devnull 		switch(l->type) {
498 564ca709 2004-04-19 devnull 		case TSTRING:
499 564ca709 2004-04-19 devnull 			l->store.u.string->gc.gcmark = 1;
500 564ca709 2004-04-19 devnull 			break;
501 564ca709 2004-04-19 devnull 		case TLIST:
502 564ca709 2004-04-19 devnull 			marklist(l->store.u.l);
503 564ca709 2004-04-19 devnull 			break;
504 564ca709 2004-04-19 devnull 		case TCODE:
505 564ca709 2004-04-19 devnull 			marktree(l->store.u.cc);
506 564ca709 2004-04-19 devnull 			break;
507 564ca709 2004-04-19 devnull 		}
508 564ca709 2004-04-19 devnull 		l = l->next;
509 564ca709 2004-04-19 devnull 	}
510 564ca709 2004-04-19 devnull }
511 564ca709 2004-04-19 devnull 
512 564ca709 2004-04-19 devnull void
513 564ca709 2004-04-19 devnull gc(void)
514 564ca709 2004-04-19 devnull {
515 564ca709 2004-04-19 devnull 	int i;
516 564ca709 2004-04-19 devnull 	Lsym *f;
517 564ca709 2004-04-19 devnull 	Value *v;
518 564ca709 2004-04-19 devnull 	Gc *m, **p, *next;
519 564ca709 2004-04-19 devnull 
520 564ca709 2004-04-19 devnull 	if(dogc < Mempergc)
521 564ca709 2004-04-19 devnull 		return;
522 564ca709 2004-04-19 devnull 	dogc = 0;
523 564ca709 2004-04-19 devnull 
524 564ca709 2004-04-19 devnull 	/* Mark */
525 564ca709 2004-04-19 devnull 	for(m = gcl; m; m = m->gclink)
526 564ca709 2004-04-19 devnull 		m->gcmark = 0;
527 564ca709 2004-04-19 devnull 
528 564ca709 2004-04-19 devnull 	/* Scan */
529 564ca709 2004-04-19 devnull 	for(i = 0; i < Hashsize; i++) {
530 564ca709 2004-04-19 devnull 		for(f = hash[i]; f; f = f->hash) {
531 564ca709 2004-04-19 devnull 			marktree(f->proc);
532 564ca709 2004-04-19 devnull 			if(f->lexval != Tid)
533 564ca709 2004-04-19 devnull 				continue;
534 564ca709 2004-04-19 devnull 			for(v = f->v; v; v = v->pop) {
535 564ca709 2004-04-19 devnull 				switch(v->type) {
536 564ca709 2004-04-19 devnull 				case TSTRING:
537 564ca709 2004-04-19 devnull 					v->store.u.string->gc.gcmark = 1;
538 564ca709 2004-04-19 devnull 					break;
539 564ca709 2004-04-19 devnull 				case TLIST:
540 564ca709 2004-04-19 devnull 					marklist(v->store.u.l);
541 564ca709 2004-04-19 devnull 					break;
542 564ca709 2004-04-19 devnull 				case TCODE:
543 564ca709 2004-04-19 devnull 					marktree(v->store.u.cc);
544 564ca709 2004-04-19 devnull 					break;
545 564ca709 2004-04-19 devnull 				}
546 564ca709 2004-04-19 devnull 			}
547 564ca709 2004-04-19 devnull 		}
548 564ca709 2004-04-19 devnull 	}
549 564ca709 2004-04-19 devnull 
550 564ca709 2004-04-19 devnull 	/* Free */
551 564ca709 2004-04-19 devnull 	p = &gcl;
552 564ca709 2004-04-19 devnull 	for(m = gcl; m; m = next) {
553 564ca709 2004-04-19 devnull 		next = m->gclink;
554 564ca709 2004-04-19 devnull 		if(m->gcmark == 0) {
555 564ca709 2004-04-19 devnull 			*p = next;
556 564ca709 2004-04-19 devnull 			free(m);	/* Sleazy reliance on my malloc */
557 564ca709 2004-04-19 devnull 		}
558 564ca709 2004-04-19 devnull 		else
559 564ca709 2004-04-19 devnull 			p = &m->gclink;
560 564ca709 2004-04-19 devnull 	}
561 564ca709 2004-04-19 devnull }
562 564ca709 2004-04-19 devnull 
563 564ca709 2004-04-19 devnull void*
564 564ca709 2004-04-19 devnull gmalloc(long l)
565 564ca709 2004-04-19 devnull {
566 564ca709 2004-04-19 devnull 	void *p;
567 564ca709 2004-04-19 devnull 
568 564ca709 2004-04-19 devnull 	dogc += l;
569 564ca709 2004-04-19 devnull 	p = malloc(l);
570 564ca709 2004-04-19 devnull 	if(p == 0)
571 564ca709 2004-04-19 devnull 		fatal("out of memory");
572 564ca709 2004-04-19 devnull 	return p;
573 564ca709 2004-04-19 devnull }
574 564ca709 2004-04-19 devnull 
575 564ca709 2004-04-19 devnull void
576 564ca709 2004-04-19 devnull checkqid(int f1, int pid)
577 564ca709 2004-04-19 devnull {
578 564ca709 2004-04-19 devnull 	int fd;
579 564ca709 2004-04-19 devnull 	Dir *d1, *d2;
580 564ca709 2004-04-19 devnull 	char buf[128];
581 564ca709 2004-04-19 devnull 
582 564ca709 2004-04-19 devnull 	if(kernel)
583 564ca709 2004-04-19 devnull 		return;
584 564ca709 2004-04-19 devnull 
585 564ca709 2004-04-19 devnull 	d1 = dirfstat(f1);
586 564ca709 2004-04-19 devnull 	if(d1 == nil){
587 564ca709 2004-04-19 devnull 		print("checkqid: (qid not checked) dirfstat: %r\n");
588 564ca709 2004-04-19 devnull 		return;
589 564ca709 2004-04-19 devnull 	}
590 564ca709 2004-04-19 devnull 
591 564ca709 2004-04-19 devnull 	sprint(buf, "/proc/%d/text", pid);
592 564ca709 2004-04-19 devnull 	fd = open(buf, OREAD);
593 564ca709 2004-04-19 devnull 	if(fd < 0 || (d2 = dirfstat(fd)) == nil){
594 564ca709 2004-04-19 devnull 		print("checkqid: (qid not checked) dirstat %s: %r\n", buf);
595 564ca709 2004-04-19 devnull 		free(d1);
596 564ca709 2004-04-19 devnull 		if(fd >= 0)
597 564ca709 2004-04-19 devnull 			close(fd);
598 564ca709 2004-04-19 devnull 		return;
599 564ca709 2004-04-19 devnull 	}
600 564ca709 2004-04-19 devnull 
601 564ca709 2004-04-19 devnull 	close(fd);
602 564ca709 2004-04-19 devnull 
603 564ca709 2004-04-19 devnull 	if(d1->qid.path != d2->qid.path || d1->qid.vers != d2->qid.vers || d1->qid.type != d2->qid.type){
604 564ca709 2004-04-19 devnull 		print("path %llux %llux vers %lud %lud type %d %d\n",
605 564ca709 2004-04-19 devnull 			d1->qid.path, d2->qid.path, d1->qid.vers, d2->qid.vers, d1->qid.type, d2->qid.type);
606 564ca709 2004-04-19 devnull 		print("warning: image does not match text for pid %d\n", pid);
607 564ca709 2004-04-19 devnull 	}
608 564ca709 2004-04-19 devnull 	free(d1);
609 564ca709 2004-04-19 devnull 	free(d2);
610 564ca709 2004-04-19 devnull }
611 564ca709 2004-04-19 devnull 
612 564ca709 2004-04-19 devnull void
613 564ca709 2004-04-19 devnull catcher(void *junk, char *s)
614 564ca709 2004-04-19 devnull {
615 564ca709 2004-04-19 devnull 	USED(junk);
616 564ca709 2004-04-19 devnull 
617 564ca709 2004-04-19 devnull 	if(strstr(s, "interrupt")) {
618 564ca709 2004-04-19 devnull 		gotint = 1;
619 564ca709 2004-04-19 devnull 		noted(NCONT);
620 564ca709 2004-04-19 devnull 	}
621 564ca709 2004-04-19 devnull 	if(strstr(s, "child"))
622 564ca709 2004-04-19 devnull 		noted(NCONT);
623 564ca709 2004-04-19 devnull fprint(2, "note: %s\n", s);
624 564ca709 2004-04-19 devnull 	noted(NDFLT);
625 564ca709 2004-04-19 devnull }
626 564ca709 2004-04-19 devnull 
627 564ca709 2004-04-19 devnull char*
628 564ca709 2004-04-19 devnull system(void)
629 564ca709 2004-04-19 devnull {
630 564ca709 2004-04-19 devnull 	char *cpu, *p, *q;
631 564ca709 2004-04-19 devnull 	static char kernel[128];
632 564ca709 2004-04-19 devnull 
633 564ca709 2004-04-19 devnull 	cpu = getenv("cputype");
634 564ca709 2004-04-19 devnull 	if(cpu == 0) {
635 564ca709 2004-04-19 devnull 		cpu = "mips";
636 564ca709 2004-04-19 devnull 		print("$cputype not set; assuming %s\n", cpu);
637 564ca709 2004-04-19 devnull 	}
638 564ca709 2004-04-19 devnull 	p = getenv("terminal");
639 564ca709 2004-04-19 devnull 	if(p == 0 || (p=strchr(p, ' ')) == 0 || p[1] == ' ' || p[1] == 0) {
640 564ca709 2004-04-19 devnull 		p = "9power";
641 564ca709 2004-04-19 devnull 		print("missing or bad $terminal; assuming %s\n", p);
642 564ca709 2004-04-19 devnull 	}
643 564ca709 2004-04-19 devnull 	else{
644 564ca709 2004-04-19 devnull 		p++;
645 564ca709 2004-04-19 devnull 		q = strchr(p, ' ');
646 564ca709 2004-04-19 devnull 		if(q)
647 564ca709 2004-04-19 devnull 			*q = 0;
648 564ca709 2004-04-19 devnull 		sprint(kernel, "/%s/9%s", cpu, p);
649 564ca709 2004-04-19 devnull 	}
650 564ca709 2004-04-19 devnull 	return kernel;
651 564ca709 2004-04-19 devnull }
652 564ca709 2004-04-19 devnull 
653 564ca709 2004-04-19 devnull int
654 564ca709 2004-04-19 devnull isnumeric(char *s)
655 564ca709 2004-04-19 devnull {
656 564ca709 2004-04-19 devnull 	while(*s) {
657 564ca709 2004-04-19 devnull 		if(*s < '0' || *s > '9')
658 564ca709 2004-04-19 devnull 			return 0;
659 564ca709 2004-04-19 devnull 		s++;
660 564ca709 2004-04-19 devnull 	}
661 564ca709 2004-04-19 devnull 	return 1;
662 564ca709 2004-04-19 devnull }
663 564ca709 2004-04-19 devnull 
664 564ca709 2004-04-19 devnull int
665 564ca709 2004-04-19 devnull xfmt(Fmt *f)
666 564ca709 2004-04-19 devnull {
667 564ca709 2004-04-19 devnull 	f->flags ^= FmtSharp;
668 564ca709 2004-04-19 devnull 	return __ifmt(f);
669 564ca709 2004-04-19 devnull }