1 6f4d00ee 2013-09-23 0intro #include "stdinc.h"
3 6f4d00ee 2013-09-23 0intro #include "9.h"
6 6f4d00ee 2013-09-23 0intro Nl = 256, /* max. command line length */
7 6f4d00ee 2013-09-23 0intro Nq = 8*1024, /* amount of I/O buffered */
10 6f4d00ee 2013-09-23 0intro typedef struct Q {
11 4b576658 2013-09-23 0intro QLock lock;
12 4b576658 2013-09-23 0intro Rendez full;
13 4b576658 2013-09-23 0intro Rendez empty;
15 6f4d00ee 2013-09-23 0intro char q[Nq];
21 6f4d00ee 2013-09-23 0intro typedef struct Cons {
22 4b576658 2013-09-23 0intro QLock lock;
24 6f4d00ee 2013-09-23 0intro int closed;
26 6f4d00ee 2013-09-23 0intro int srvfd;
27 6f4d00ee 2013-09-23 0intro int ctlfd;
28 6f4d00ee 2013-09-23 0intro Q* iq; /* points to console.iq */
29 6f4d00ee 2013-09-23 0intro Q* oq; /* points to console.oq */
32 6f4d00ee 2013-09-23 0intro char *currfsysname;
34 6f4d00ee 2013-09-23 0intro static struct {
35 6f4d00ee 2013-09-23 0intro Q* iq; /* input */
36 6f4d00ee 2013-09-23 0intro Q* oq; /* output */
37 6f4d00ee 2013-09-23 0intro char l[Nl]; /* command line assembly */
38 6f4d00ee 2013-09-23 0intro int nl; /* current line length */
39 6f4d00ee 2013-09-23 0intro int nopens;
41 6f4d00ee 2013-09-23 0intro char* prompt;
43 6f4d00ee 2013-09-23 0intro } console;
45 6f4d00ee 2013-09-23 0intro static void
46 6f4d00ee 2013-09-23 0intro consClose(Cons* cons)
48 4b576658 2013-09-23 0intro qlock(&cons->lock);
49 6f4d00ee 2013-09-23 0intro cons->closed = 1;
51 6f4d00ee 2013-09-23 0intro cons->ref--;
52 6f4d00ee 2013-09-23 0intro if(cons->ref > 0){
53 4b576658 2013-09-23 0intro qlock(&cons->iq->lock);
54 4b576658 2013-09-23 0intro rwakeup(&cons->iq->full);
55 4b576658 2013-09-23 0intro qunlock(&cons->iq->lock);
56 4b576658 2013-09-23 0intro qlock(&cons->oq->lock);
57 4b576658 2013-09-23 0intro rwakeup(&cons->oq->empty);
58 4b576658 2013-09-23 0intro qunlock(&cons->oq->lock);
59 4b576658 2013-09-23 0intro qunlock(&cons->lock);
63 6f4d00ee 2013-09-23 0intro if(cons->ctlfd != -1){
64 6f4d00ee 2013-09-23 0intro close(cons->ctlfd);
65 6f4d00ee 2013-09-23 0intro cons->srvfd = -1;
67 6f4d00ee 2013-09-23 0intro if(cons->srvfd != -1){
68 6f4d00ee 2013-09-23 0intro close(cons->srvfd);
69 6f4d00ee 2013-09-23 0intro cons->srvfd = -1;
71 6f4d00ee 2013-09-23 0intro if(cons->fd != -1){
72 6f4d00ee 2013-09-23 0intro close(cons->fd);
73 6f4d00ee 2013-09-23 0intro cons->fd = -1;
75 4b576658 2013-09-23 0intro qunlock(&cons->lock);
76 4b576658 2013-09-23 0intro vtfree(cons);
77 6f4d00ee 2013-09-23 0intro console.nopens--;
80 6f4d00ee 2013-09-23 0intro static void
81 6f4d00ee 2013-09-23 0intro consIProc(void* v)
84 6f4d00ee 2013-09-23 0intro Cons *cons;
86 6f4d00ee 2013-09-23 0intro char buf[Nq/4];
88 4b576658 2013-09-23 0intro threadsetname("consI");
91 6f4d00ee 2013-09-23 0intro q = cons->iq;
94 6f4d00ee 2013-09-23 0intro * Can't tell the difference between zero-length read
95 6f4d00ee 2013-09-23 0intro * and eof, so keep calling read until we get an error.
97 6f4d00ee 2013-09-23 0intro if(cons->closed || (n = read(cons->fd, buf, Nq/4)) < 0)
99 4b576658 2013-09-23 0intro qlock(&q->lock);
100 6f4d00ee 2013-09-23 0intro while(Nq - q->n < n && !cons->closed)
101 4b576658 2013-09-23 0intro rsleep(&q->full);
102 6f4d00ee 2013-09-23 0intro w = Nq - q->w;
103 6f4d00ee 2013-09-23 0intro if(w < n){
104 6f4d00ee 2013-09-23 0intro memmove(&q->q[q->w], buf, w);
105 6f4d00ee 2013-09-23 0intro memmove(&q->q[0], buf + w, n - w);
108 6f4d00ee 2013-09-23 0intro memmove(&q->q[q->w], buf, n);
109 6f4d00ee 2013-09-23 0intro q->w = (q->w + n) % Nq;
110 6f4d00ee 2013-09-23 0intro q->n += n;
111 4b576658 2013-09-23 0intro rwakeup(&q->empty);
112 4b576658 2013-09-23 0intro qunlock(&q->lock);
114 6f4d00ee 2013-09-23 0intro consClose(cons);
117 6f4d00ee 2013-09-23 0intro static void
118 6f4d00ee 2013-09-23 0intro consOProc(void* v)
121 6f4d00ee 2013-09-23 0intro Cons *cons;
122 6f4d00ee 2013-09-23 0intro char buf[Nq];
123 6f4d00ee 2013-09-23 0intro int lastn, n, r;
125 4b576658 2013-09-23 0intro threadsetname("consO");
127 6f4d00ee 2013-09-23 0intro cons = v;
128 6f4d00ee 2013-09-23 0intro q = cons->oq;
129 4b576658 2013-09-23 0intro qlock(&q->lock);
130 6f4d00ee 2013-09-23 0intro lastn = 0;
132 6f4d00ee 2013-09-23 0intro while(lastn == q->n && !cons->closed)
133 4b576658 2013-09-23 0intro rsleep(&q->empty);
134 6f4d00ee 2013-09-23 0intro if((n = q->n - lastn) > Nq)
136 6f4d00ee 2013-09-23 0intro if(n > q->w){
137 6f4d00ee 2013-09-23 0intro r = n - q->w;
138 6f4d00ee 2013-09-23 0intro memmove(buf, &q->q[Nq - r], r);
139 6f4d00ee 2013-09-23 0intro memmove(buf+r, &q->q[0], n - r);
142 6f4d00ee 2013-09-23 0intro memmove(buf, &q->q[q->w - n], n);
143 6f4d00ee 2013-09-23 0intro lastn = q->n;
144 4b576658 2013-09-23 0intro qunlock(&q->lock);
145 6f4d00ee 2013-09-23 0intro if(cons->closed || write(cons->fd, buf, n) < 0)
147 4b576658 2013-09-23 0intro qlock(&q->lock);
148 4b576658 2013-09-23 0intro rwakeup(&q->empty);
150 6f4d00ee 2013-09-23 0intro consClose(cons);
154 6f4d00ee 2013-09-23 0intro consOpen(int fd, int srvfd, int ctlfd)
156 6f4d00ee 2013-09-23 0intro Cons *cons;
158 4b576658 2013-09-23 0intro cons = vtmallocz(sizeof(Cons));
159 6f4d00ee 2013-09-23 0intro cons->fd = fd;
160 6f4d00ee 2013-09-23 0intro cons->srvfd = srvfd;
161 6f4d00ee 2013-09-23 0intro cons->ctlfd = ctlfd;
162 6f4d00ee 2013-09-23 0intro cons->iq = console.iq;
163 6f4d00ee 2013-09-23 0intro cons->oq = console.oq;
164 6f4d00ee 2013-09-23 0intro console.nopens++;
166 4b576658 2013-09-23 0intro qlock(&cons->lock);
167 6f4d00ee 2013-09-23 0intro cons->ref = 2;
168 6f4d00ee 2013-09-23 0intro cons->closed = 0;
169 4b576658 2013-09-23 0intro if(proccreate(consOProc, cons, STACK) < 0){
170 6f4d00ee 2013-09-23 0intro cons->ref--;
171 4b576658 2013-09-23 0intro qunlock(&cons->lock);
172 6f4d00ee 2013-09-23 0intro consClose(cons);
173 6f4d00ee 2013-09-23 0intro return 0;
175 4b576658 2013-09-23 0intro qunlock(&cons->lock);
177 6f4d00ee 2013-09-23 0intro if(ctlfd >= 0)
178 6f4d00ee 2013-09-23 0intro consIProc(cons);
179 4b576658 2013-09-23 0intro else if(proccreate(consIProc, cons, STACK) < 0){
180 6f4d00ee 2013-09-23 0intro consClose(cons);
181 6f4d00ee 2013-09-23 0intro return 0;
184 6f4d00ee 2013-09-23 0intro return 1;
187 6f4d00ee 2013-09-23 0intro static int
188 6f4d00ee 2013-09-23 0intro qWrite(Q* q, char* p, int n)
192 4b576658 2013-09-23 0intro qlock(&q->lock);
193 6f4d00ee 2013-09-23 0intro if(n > Nq - q->w){
194 6f4d00ee 2013-09-23 0intro w = Nq - q->w;
195 6f4d00ee 2013-09-23 0intro memmove(&q->q[q->w], p, w);
196 6f4d00ee 2013-09-23 0intro memmove(&q->q[0], p + w, n - w);
197 6f4d00ee 2013-09-23 0intro q->w = n - w;
200 6f4d00ee 2013-09-23 0intro memmove(&q->q[q->w], p, n);
201 6f4d00ee 2013-09-23 0intro q->w += n;
203 6f4d00ee 2013-09-23 0intro q->n += n;
204 4b576658 2013-09-23 0intro rwakeup(&q->empty);
205 4b576658 2013-09-23 0intro qunlock(&q->lock);
207 6f4d00ee 2013-09-23 0intro return n;
210 6f4d00ee 2013-09-23 0intro static Q*
211 6f4d00ee 2013-09-23 0intro qAlloc(void)
215 4b576658 2013-09-23 0intro q = vtmallocz(sizeof(Q));
216 4b576658 2013-09-23 0intro q->full.l = &q->lock;
217 4b576658 2013-09-23 0intro q->empty.l = &q->lock;
218 6f4d00ee 2013-09-23 0intro q->n = q->r = q->w = 0;
220 6f4d00ee 2013-09-23 0intro return q;
223 6f4d00ee 2013-09-23 0intro static void
224 b32de4ae 2013-09-26 0intro consProc(void* v)
228 6f4d00ee 2013-09-23 0intro int argc, i, n, r;
229 6f4d00ee 2013-09-23 0intro char *argv[20], buf[Nq], *lp, *wbuf;
230 6f4d00ee 2013-09-23 0intro char procname[64];
232 6f4d00ee 2013-09-23 0intro snprint(procname, sizeof procname, "cons %s", currfsysname);
233 4b576658 2013-09-23 0intro threadsetname(procname);
235 6f4d00ee 2013-09-23 0intro q = console.iq;
236 6f4d00ee 2013-09-23 0intro qWrite(console.oq, console.prompt, console.np);
237 4b576658 2013-09-23 0intro qlock(&q->lock);
239 6f4d00ee 2013-09-23 0intro while((n = q->n) == 0)
240 4b576658 2013-09-23 0intro rsleep(&q->empty);
241 6f4d00ee 2013-09-23 0intro r = Nq - q->r;
242 6f4d00ee 2013-09-23 0intro if(r < n){
243 6f4d00ee 2013-09-23 0intro memmove(buf, &q->q[q->r], r);
244 6f4d00ee 2013-09-23 0intro memmove(buf + r, &q->q[0], n - r);
247 6f4d00ee 2013-09-23 0intro memmove(buf, &q->q[q->r], n);
248 6f4d00ee 2013-09-23 0intro q->r = (q->r + n) % Nq;
249 6f4d00ee 2013-09-23 0intro q->n -= n;
250 4b576658 2013-09-23 0intro rwakeup(&q->full);
251 4b576658 2013-09-23 0intro qunlock(&q->lock);
253 6f4d00ee 2013-09-23 0intro for(i = 0; i < n; i++){
254 6f4d00ee 2013-09-23 0intro switch(buf[i]){
255 6f4d00ee 2013-09-23 0intro case '\004': /* ^D */
256 6f4d00ee 2013-09-23 0intro if(console.nl == 0){
257 6f4d00ee 2013-09-23 0intro qWrite(console.oq, "\n", 1);
260 6f4d00ee 2013-09-23 0intro /*FALLTHROUGH*/
262 6f4d00ee 2013-09-23 0intro if(console.nl < Nl-1){
263 6f4d00ee 2013-09-23 0intro qWrite(console.oq, &buf[i], 1);
264 6f4d00ee 2013-09-23 0intro console.l[console.nl++] = buf[i];
266 6f4d00ee 2013-09-23 0intro continue;
267 6f4d00ee 2013-09-23 0intro case '\b':
268 6f4d00ee 2013-09-23 0intro if(console.nl != 0){
269 6f4d00ee 2013-09-23 0intro qWrite(console.oq, &buf[i], 1);
270 6f4d00ee 2013-09-23 0intro console.nl--;
272 6f4d00ee 2013-09-23 0intro continue;
273 6f4d00ee 2013-09-23 0intro case '\n':
274 6f4d00ee 2013-09-23 0intro qWrite(console.oq, &buf[i], 1);
276 6f4d00ee 2013-09-23 0intro case '\025': /* ^U */
277 6f4d00ee 2013-09-23 0intro qWrite(console.oq, "^U\n", 3);
278 6f4d00ee 2013-09-23 0intro console.nl = 0;
280 6f4d00ee 2013-09-23 0intro case '\027': /* ^W */
281 6f4d00ee 2013-09-23 0intro console.l[console.nl] = '\0';
282 4b576658 2013-09-23 0intro wbuf = vtmalloc(console.nl+1);
283 6f4d00ee 2013-09-23 0intro memmove(wbuf, console.l, console.nl+1);
284 6f4d00ee 2013-09-23 0intro argc = tokenize(wbuf, argv, nelem(argv));
285 6f4d00ee 2013-09-23 0intro if(argc > 0)
287 6f4d00ee 2013-09-23 0intro console.nl = 0;
288 6f4d00ee 2013-09-23 0intro lp = console.l;
289 6f4d00ee 2013-09-23 0intro for(i = 0; i < argc; i++)
290 6f4d00ee 2013-09-23 0intro lp += sprint(lp, "%q ", argv[i]);
291 6f4d00ee 2013-09-23 0intro console.nl = lp - console.l;
292 4b576658 2013-09-23 0intro vtfree(wbuf);
293 6f4d00ee 2013-09-23 0intro qWrite(console.oq, "^W\n", 3);
294 6f4d00ee 2013-09-23 0intro if(console.nl == 0)
296 6f4d00ee 2013-09-23 0intro qWrite(console.oq, console.l, console.nl);
297 6f4d00ee 2013-09-23 0intro continue;
298 6f4d00ee 2013-09-23 0intro case '\177':
299 6f4d00ee 2013-09-23 0intro qWrite(console.oq, "\n", 1);
300 6f4d00ee 2013-09-23 0intro console.nl = 0;
304 6f4d00ee 2013-09-23 0intro console.l[console.nl] = '\0';
305 6f4d00ee 2013-09-23 0intro if(console.nl != 0)
306 6f4d00ee 2013-09-23 0intro cliExec(console.l);
308 6f4d00ee 2013-09-23 0intro console.nl = 0;
309 6f4d00ee 2013-09-23 0intro qWrite(console.oq, console.prompt, console.np);
312 4b576658 2013-09-23 0intro qlock(&q->lock);
317 6f4d00ee 2013-09-23 0intro consWrite(char* buf, int len)
319 6f4d00ee 2013-09-23 0intro if(console.oq == nil)
320 6f4d00ee 2013-09-23 0intro return write(2, buf, len);
321 6f4d00ee 2013-09-23 0intro if(console.nopens == 0)
322 6f4d00ee 2013-09-23 0intro write(2, buf, len);
323 6f4d00ee 2013-09-23 0intro return qWrite(console.oq, buf, len);
327 6f4d00ee 2013-09-23 0intro consPrompt(char* prompt)
329 6f4d00ee 2013-09-23 0intro char buf[ERRMAX];
331 6f4d00ee 2013-09-23 0intro if(prompt == nil)
332 6f4d00ee 2013-09-23 0intro prompt = "prompt";
334 4b576658 2013-09-23 0intro vtfree(console.prompt);
335 6f4d00ee 2013-09-23 0intro console.np = snprint(buf, sizeof(buf), "%s: ", prompt);
336 4b576658 2013-09-23 0intro console.prompt = vtstrdup(buf);
338 6f4d00ee 2013-09-23 0intro return console.np;
342 6f4d00ee 2013-09-23 0intro consTTY(void)
344 6f4d00ee 2013-09-23 0intro int ctl, fd;
345 6f4d00ee 2013-09-23 0intro char *name, *p;
347 6f4d00ee 2013-09-23 0intro name = "/dev/cons";
348 6f4d00ee 2013-09-23 0intro if((fd = open(name, ORDWR)) < 0){
349 b32de4ae 2013-09-26 0intro #ifdef PLAN9PORT
350 b32de4ae 2013-09-26 0intro name = "/dev/tty";
352 6f4d00ee 2013-09-23 0intro name = "#c/cons";
354 6f4d00ee 2013-09-23 0intro if((fd = open(name, ORDWR)) < 0){
355 4b576658 2013-09-23 0intro werrstr("consTTY: open %s: %r", name);
356 6f4d00ee 2013-09-23 0intro return 0;
360 b32de4ae 2013-09-26 0intro #ifdef PLAN9PORT
364 6f4d00ee 2013-09-23 0intro p = smprint("%sctl", name);
365 6f4d00ee 2013-09-23 0intro if((ctl = open(p, OWRITE)) < 0){
366 6f4d00ee 2013-09-23 0intro close(fd);
367 4b576658 2013-09-23 0intro werrstr("consTTY: open %s: %r", p);
369 6f4d00ee 2013-09-23 0intro return 0;
371 6f4d00ee 2013-09-23 0intro if(write(ctl, "rawon", 5) < 0){
372 6f4d00ee 2013-09-23 0intro close(ctl);
373 6f4d00ee 2013-09-23 0intro close(fd);
374 4b576658 2013-09-23 0intro werrstr("consTTY: write %s: %r", p);
376 6f4d00ee 2013-09-23 0intro return 0;
381 6f4d00ee 2013-09-23 0intro if(consOpen(fd, fd, ctl) == 0){
382 6f4d00ee 2013-09-23 0intro close(ctl);
383 6f4d00ee 2013-09-23 0intro close(fd);
384 6f4d00ee 2013-09-23 0intro return 0;
387 6f4d00ee 2013-09-23 0intro return 1;
391 6f4d00ee 2013-09-23 0intro consInit(void)
393 6f4d00ee 2013-09-23 0intro console.iq = qAlloc();
394 6f4d00ee 2013-09-23 0intro console.oq = qAlloc();
395 6f4d00ee 2013-09-23 0intro console.nl = 0;
397 6f4d00ee 2013-09-23 0intro consPrompt(nil);
399 4b576658 2013-09-23 0intro if(proccreate(consProc, nil, STACK) < 0){
400 4b576658 2013-09-23 0intro sysfatal("can't start console proc");
401 6f4d00ee 2013-09-23 0intro return 0;
404 6f4d00ee 2013-09-23 0intro return 1;