![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to script.c CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [CSRG BSD Unix] / 43BSD / ucb|
Return to script.c CVS log | Up to [CSRG BSD Unix] / 43BSD / ucb |
1.1 root 1: /*
2: * Copyright (c) 1980 Regents of the University of California.
3: * All rights reserved. The Berkeley software License Agreement
4: * specifies the terms and conditions for redistribution.
5: */
6:
7: #ifndef lint
8: char copyright[] =
9: "@(#) Copyright (c) 1980 Regents of the University of California.\n\
10: All rights reserved.\n";
11: #endif not lint
12:
13: #ifndef lint
14: static char sccsid[] = "@(#)script.c 5.4 (Berkeley) 11/13/85";
15: #endif not lint
16:
17: /*
18: * script
19: */
20: #include <stdio.h>
21: #include <signal.h>
22: #include <sys/types.h>
23: #include <sys/stat.h>
24: #include <sys/ioctl.h>
25: #include <sgtty.h>
26: #include <sys/time.h>
27: #include <sys/file.h>
28:
29: char *getenv();
30: char *ctime();
31: char *shell;
32: FILE *fscript;
33: int master;
34: int slave;
35: int child;
36: int subchild;
37: char *fname = "typescript";
38: int finish();
39:
40: struct sgttyb b;
41: struct tchars tc;
42: struct ltchars lc;
43: struct winsize win;
44: int lb;
45: int l;
46: char *line = "/dev/ptyXX";
47: int aflg;
48:
49: main(argc, argv)
50: int argc;
51: char *argv[];
52: {
53:
54: shell = getenv("SHELL");
55: if (shell == 0)
56: shell = "/bin/sh";
57: argc--, argv++;
58: while (argc > 0 && argv[0][0] == '-') {
59: switch (argv[0][1]) {
60:
61: case 'a':
62: aflg++;
63: break;
64:
65: default:
66: fprintf(stderr,
67: "usage: script [ -a ] [ typescript ]\n");
68: exit(1);
69: }
70: argc--, argv++;
71: }
72: if (argc > 0)
73: fname = argv[0];
74: if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL) {
75: perror(fname);
76: fail();
77: }
78: getmaster();
79: printf("Script started, file is %s\n", fname);
80: fixtty();
81:
82: (void) signal(SIGCHLD, finish);
83: child = fork();
84: if (child < 0) {
85: perror("fork");
86: fail();
87: }
88: if (child == 0) {
89: subchild = child = fork();
90: if (child < 0) {
91: perror("fork");
92: fail();
93: }
94: if (child)
95: dooutput();
96: else
97: doshell();
98: }
99: doinput();
100: }
101:
102: doinput()
103: {
104: char ibuf[BUFSIZ];
105: int cc;
106:
107: (void) fclose(fscript);
108: while ((cc = read(0, ibuf, BUFSIZ)) > 0)
109: (void) write(master, ibuf, cc);
110: done();
111: }
112:
113: #include <sys/wait.h>
114:
115: finish()
116: {
117: union wait status;
118: register int pid;
119: register int die = 0;
120:
121: while ((pid = wait3(&status, WNOHANG, 0)) > 0)
122: if (pid == child)
123: die = 1;
124:
125: if (die)
126: done();
127: }
128:
129: dooutput()
130: {
131: time_t tvec;
132: char obuf[BUFSIZ];
133: int cc;
134:
135: (void) close(0);
136: tvec = time((time_t *)0);
137: fprintf(fscript, "Script started on %s", ctime(&tvec));
138: for (;;) {
139: cc = read(master, obuf, sizeof (obuf));
140: if (cc <= 0)
141: break;
142: (void) write(1, obuf, cc);
143: (void) fwrite(obuf, 1, cc, fscript);
144: }
145: done();
146: }
147:
148: doshell()
149: {
150: int t;
151:
152: t = open("/dev/tty", O_RDWR);
153: if (t >= 0) {
154: (void) ioctl(t, TIOCNOTTY, (char *)0);
155: (void) close(t);
156: }
157: getslave();
158: (void) close(master);
159: (void) fclose(fscript);
160: (void) dup2(slave, 0);
161: (void) dup2(slave, 1);
162: (void) dup2(slave, 2);
163: (void) close(slave);
164: execl(shell, "sh", "-i", 0);
165: perror(shell);
166: fail();
167: }
168:
169: fixtty()
170: {
171: struct sgttyb sbuf;
172:
173: sbuf = b;
174: sbuf.sg_flags |= RAW;
175: sbuf.sg_flags &= ~ECHO;
176: (void) ioctl(0, TIOCSETP, (char *)&sbuf);
177: }
178:
179: fail()
180: {
181:
182: (void) kill(0, SIGTERM);
183: done();
184: }
185:
186: done()
187: {
188: time_t tvec;
189:
190: if (subchild) {
191: tvec = time((time_t *)0);
192: fprintf(fscript,"\nscript done on %s", ctime(&tvec));
193: (void) fclose(fscript);
194: (void) close(master);
195: } else {
196: (void) ioctl(0, TIOCSETP, (char *)&b);
197: printf("Script done, file is %s\n", fname);
198: }
199: exit(0);
200: }
201:
202: getmaster()
203: {
204: char *pty, *bank, *cp;
205: struct stat stb;
206:
207: pty = &line[strlen("/dev/ptyp")];
208: for (bank = "pqrs"; *bank; bank++) {
209: line[strlen("/dev/pty")] = *bank;
210: *pty = '0';
211: if (stat(line, &stb) < 0)
212: break;
213: for (cp = "0123456789abcdef"; *cp; cp++) {
214: *pty = *cp;
215: master = open(line, O_RDWR);
216: if (master >= 0) {
217: char *tp = &line[strlen("/dev/")];
218: int ok;
219:
220: /* verify slave side is usable */
221: *tp = 't';
222: ok = access(line, R_OK|W_OK) == 0;
223: *tp = 'p';
224: if (ok) {
225: (void) ioctl(0, TIOCGETP, (char *)&b);
226: (void) ioctl(0, TIOCGETC, (char *)&tc);
227: (void) ioctl(0, TIOCGETD, (char *)&l);
228: (void) ioctl(0, TIOCGLTC, (char *)&lc);
229: (void) ioctl(0, TIOCLGET, (char *)&lb);
230: (void) ioctl(0, TIOCGWINSZ, (char *)&win);
231: return;
232: }
233: (void) close(master);
234: }
235: }
236: }
237: fprintf(stderr, "Out of pty's\n");
238: fail();
239: }
240:
241: getslave()
242: {
243:
244: line[strlen("/dev/")] = 't';
245: slave = open(line, O_RDWR);
246: if (slave < 0) {
247: perror(line);
248: fail();
249: }
250: (void) ioctl(slave, TIOCSETP, (char *)&b);
251: (void) ioctl(slave, TIOCSETC, (char *)&tc);
252: (void) ioctl(slave, TIOCSLTC, (char *)&lc);
253: (void) ioctl(slave, TIOCLSET, (char *)&lb);
254: (void) ioctl(slave, TIOCSETD, (char *)&l);
255: (void) ioctl(slave, TIOCSWINSZ, (char *)&win);
256: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.