Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.bin/script/script.c	Lines:	0	149	0.0 %
Date:	2017-11-13	Branches:	0	86	0.0 %


/*	$OpenBSD: script.c,v 1.33 2017/04/12 14:49:05 deraadt Exp $	*/
/*	$NetBSD: script.c,v 1.3 1994/12/21 08:55:43 jtc Exp $	*/

/*
 * Copyright (c) 2001 Theo de Raadt
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/*
 * Copyright (c) 1980, 1992, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/time.h>

#include <errno.h>
#include <fcntl.h>
#include <paths.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>

#include <util.h>
#include <err.h>

FILE	*fscript;
int	master, slave;
volatile sig_atomic_t child;
pid_t	subchild;
char	*fname;

volatile sig_atomic_t dead;
volatile sig_atomic_t sigdeadstatus;
volatile sig_atomic_t flush;

struct	termios tt;
int		istty;

__dead void done(int);
void dooutput(void);
void doshell(void);
void fail(void);
void finish(int);
void scriptflush(int);
void handlesigwinch(int);

int
main(int argc, char *argv[])
{
	extern char *__progname;
	struct sigaction sa;
	struct winsize win;
	char ibuf[BUFSIZ];
	ssize_t cc, off;
	int aflg, ch;

	aflg = 0;
	while ((ch = getopt(argc, argv, "a")) != -1)
		switch(ch) {
		case 'a':
			aflg = 1;
			break;
		default:
			fprintf(stderr, "usage: %s [-a] [file]\n", __progname);
			exit(1);
		}
	argc -= optind;
	argv += optind;

	if (argc > 0)
		fname = argv[0];
	else
		fname = "typescript";

	if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL)
		err(1, "%s", fname);

	if (isatty(0)) {
		if (tcgetattr(STDIN_FILENO, &tt) == 0 &&
		    ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == 0)
			istty = 1;
	}
	if (openpty(&master, &slave, NULL, &tt, &win) == -1)
		err(1, "openpty");

	(void)printf("Script started, output file is %s\n", fname);
	if (istty) {
		struct termios rtt = tt;

		cfmakeraw(&rtt);
		rtt.c_lflag &= ~ECHO;
		(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &rtt);
	}

	bzero(&sa, sizeof sa);
	sigemptyset(&sa.sa_mask);
	sa.sa_handler = handlesigwinch;
	sa.sa_flags = SA_RESTART;
	(void)sigaction(SIGWINCH, &sa, NULL);

	child = fork();
	if (child < 0) {
		warn("fork");
		fail();
	}
	if (child == 0) {
		subchild = child = fork();
		if (child < 0) {
			warn("fork");
			fail();
		}
		if (child)
			dooutput();
		else
			doshell();
	}

	bzero(&sa, sizeof sa);
	sigemptyset(&sa.sa_mask);
	sa.sa_handler = finish;
	(void)sigaction(SIGCHLD, &sa, NULL);

	if (pledge("stdio proc tty flock rpath cpath wpath", NULL) == -1)
		err(1, "pledge");

	(void)fclose(fscript);
	while (1) {
		if (dead)
			break;
		cc = read(STDIN_FILENO, ibuf, BUFSIZ);
		if (cc == -1 && errno == EINTR)
			continue;
		if (cc <= 0)
			break;
		for (off = 0; off < cc; ) {
			ssize_t n = write(master, ibuf + off, cc - off);
			if (n == -1 && errno != EAGAIN)
				break;
			if (n == 0)
				break;	/* skip writing */
			if (n > 0)
				off += n;
		}
	}
	done(sigdeadstatus);
}

/* ARGSUSED */
void
finish(int signo)
{
	int save_errno = errno;
	int status, e = 1;
	pid_t pid;

	while ((pid = wait3(&status, WNOHANG, 0)) > 0) {
		if (pid == (pid_t)child) {
			if (WIFEXITED(status))
				e = WEXITSTATUS(status);
		}
	}
	dead = 1;
	sigdeadstatus = e;
	errno = save_errno;
}

/* ARGSUSED */
void
handlesigwinch(int signo)
{
	int save_errno = errno;
	struct winsize win;
	pid_t pgrp;

	if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) != -1) {
		ioctl(slave, TIOCSWINSZ, &win);
		if (ioctl(slave, TIOCGPGRP, &pgrp) != -1)
			killpg(pgrp, SIGWINCH);
	}
	errno = save_errno;
}

void
dooutput(void)
{
	struct sigaction sa;
	struct itimerval value;
	sigset_t blkalrm;
	char obuf[BUFSIZ];
	time_t tvec;
	ssize_t outcc = 0, cc, off;

	(void)close(STDIN_FILENO);
	tvec = time(NULL);
	(void)fprintf(fscript, "Script started on %s", ctime(&tvec));

	sigemptyset(&blkalrm);
	sigaddset(&blkalrm, SIGALRM);
	bzero(&sa, sizeof sa);
	sigemptyset(&sa.sa_mask);
	sa.sa_handler = scriptflush;
	(void)sigaction(SIGALRM, &sa, NULL);

	bzero(&sa, sizeof sa);
	sigemptyset(&sa.sa_mask);
	sa.sa_handler = SIG_IGN;
	(void)sigaction(SIGCHLD, &sa, NULL);

	if (pledge("stdio proc flock rpath cpath wpath", NULL) == -1)
		err(1, "pledge");

	value.it_interval.tv_sec = 30;
	value.it_interval.tv_usec = 0;
	value.it_value = value.it_interval;
	(void)setitimer(ITIMER_REAL, &value, NULL);
	for (;;) {
		if (flush) {
			if (outcc) {
				(void)fflush(fscript);
				outcc = 0;
			}
			flush = 0;
		}
		cc = read(master, obuf, sizeof (obuf));
		if (cc == -1 && errno == EINTR)
			continue;
		if (cc <= 0)
			break;
		sigprocmask(SIG_BLOCK, &blkalrm, NULL);
		for (off = 0; off < cc; ) {
			ssize_t n = write(STDOUT_FILENO, obuf + off, cc - off);
			if (n == -1 && errno != EAGAIN)
				break;
			if (n == 0)
				break;	/* skip writing */
			if (n > 0)
				off += n;
		}
		(void)fwrite(obuf, 1, cc, fscript);
		outcc += cc;
		sigprocmask(SIG_UNBLOCK, &blkalrm, NULL);
	}
	done(0);
}

/* ARGSUSED */
void
scriptflush(int signo)
{
	flush = 1;
}

void
doshell(void)
{
	char *shell;

	shell = getenv("SHELL");
	if (shell == NULL)
		shell = _PATH_BSHELL;

	(void)close(master);
	(void)fclose(fscript);
	login_tty(slave);
	execl(shell, shell, "-i", (char *)NULL);
	warn("%s", shell);
	fail();
}

void
fail(void)
{

	(void)kill(0, SIGTERM);
	done(1);
}

void
done(int eval)
{
	time_t tvec;

	if (subchild) {
		tvec = time(NULL);
		(void)fprintf(fscript,"\nScript done on %s", ctime(&tvec));
		(void)fclose(fscript);
		(void)close(master);
	} else {
		if (istty)
			(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
		(void)printf("Script done, output file is %s\n", fname);
	}
	exit(eval);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: script.c,v 1.33 2017/04/12 14:49:05 deraadt Exp $ */
2			/* $NetBSD: script.c,v 1.3 1994/12/21 08:55:43 jtc Exp $ */
3
4			/*
5			* Copyright (c) 2001 Theo de Raadt
6			* All rights reserved.
7			*
8			* Redistribution and use in source and binary forms, with or without
9			* modification, are permitted provided that the following conditions
10			* are met:
11			* 1. Redistributions of source code must retain the above copyright
12			* notice, this list of conditions and the following disclaimer.
13			* 2. Redistributions in binary form must reproduce the above copyright
14			* notice, this list of conditions and the following disclaimer in the
15			* documentation and/or other materials provided with the distribution.
16			*
17			* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18			* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19			* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20			* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21			* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22			* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23			* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24			* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25			* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26			* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27			*/
28
29			/*
30			* Copyright (c) 1980, 1992, 1993
31			* The Regents of the University of California. All rights reserved.
32			*
33			* Redistribution and use in source and binary forms, with or without
34			* modification, are permitted provided that the following conditions
35			* are met:
36			* 1. Redistributions of source code must retain the above copyright
37			* notice, this list of conditions and the following disclaimer.
38			* 2. Redistributions in binary form must reproduce the above copyright
39			* notice, this list of conditions and the following disclaimer in the
40			* documentation and/or other materials provided with the distribution.
41			* 3. Neither the name of the University nor the names of its contributors
42			* may be used to endorse or promote products derived from this software
43			* without specific prior written permission.
44			*
45			* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
46			* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
47			* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
48			* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
49			* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
50			* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
51			* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
52			* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
53			* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
54			* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
55			* SUCH DAMAGE.
56			*/
57
58			#include <sys/types.h>
59			#include <sys/wait.h>
60			#include <sys/stat.h>
61			#include <sys/ioctl.h>
62			#include <sys/time.h>
63
64			#include <errno.h>
65			#include <fcntl.h>
66			#include <paths.h>
67			#include <signal.h>
68			#include <stdio.h>
69			#include <stdlib.h>
70			#include <string.h>
71			#include <termios.h>
72			#include <unistd.h>
73
74			#include <util.h>
75			#include <err.h>
76
77			FILE *fscript;
78			int master, slave;
79			volatile sig_atomic_t child;
80			pid_t subchild;
81			char *fname;
82
83			volatile sig_atomic_t dead;
84			volatile sig_atomic_t sigdeadstatus;
85			volatile sig_atomic_t flush;
86
87			struct termios tt;
88			int istty;
89
90			__dead void done(int);
91			void dooutput(void);
92			void doshell(void);
93			void fail(void);
94			void finish(int);
95			void scriptflush(int);
96			void handlesigwinch(int);
97
98			int
99			main(int argc, char *argv[])
100			{
101			extern char *__progname;
102			struct sigaction sa;
103			struct winsize win;
104			char ibuf[BUFSIZ];
105			ssize_t cc, off;
106			int aflg, ch;
107
108			aflg = 0;
109			while ((ch = getopt(argc, argv, "a")) != -1)
110			switch(ch) {
111			case 'a':
112			aflg = 1;
113			break;
114			default:
115			fprintf(stderr, "usage: %s [-a] [file]\n", __progname);
116			exit(1);
117			}
118			argc -= optind;
119			argv += optind;
120
121			if (argc > 0)
122			fname = argv[0];
123			else
124			fname = "typescript";
125
126			if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL)
127			err(1, "%s", fname);
128
129			if (isatty(0)) {
130			if (tcgetattr(STDIN_FILENO, &tt) == 0 &&
131			ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == 0)
132			istty = 1;
133			}
134			if (openpty(&master, &slave, NULL, &tt, &win) == -1)
135			err(1, "openpty");
136
137			(void)printf("Script started, output file is %s\n", fname);
138			if (istty) {
139			struct termios rtt = tt;
140
141			cfmakeraw(&rtt);
142			rtt.c_lflag &= ~ECHO;
143			(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &rtt);
144			}
145
146			bzero(&sa, sizeof sa);
147			sigemptyset(&sa.sa_mask);
148			sa.sa_handler = handlesigwinch;
149			sa.sa_flags = SA_RESTART;
150			(void)sigaction(SIGWINCH, &sa, NULL);
151
152			child = fork();
153			if (child < 0) {
154			warn("fork");
155			fail();
156			}
157			if (child == 0) {
158			subchild = child = fork();
159			if (child < 0) {
160			warn("fork");
161			fail();
162			}
163			if (child)
164			dooutput();
165			else
166			doshell();
167			}
168
169			bzero(&sa, sizeof sa);
170			sigemptyset(&sa.sa_mask);
171			sa.sa_handler = finish;
172			(void)sigaction(SIGCHLD, &sa, NULL);
173
174			if (pledge("stdio proc tty flock rpath cpath wpath", NULL) == -1)
175			err(1, "pledge");
176
177			(void)fclose(fscript);
178			while (1) {
179			if (dead)
180			break;
181			cc = read(STDIN_FILENO, ibuf, BUFSIZ);
182			if (cc == -1 && errno == EINTR)
183			continue;
184			if (cc <= 0)
185			break;
186			for (off = 0; off < cc; ) {
187			ssize_t n = write(master, ibuf + off, cc - off);
188			if (n == -1 && errno != EAGAIN)
189			break;
190			if (n == 0)
191			break; /* skip writing */
192			if (n > 0)
193			off += n;
194			}
195			}
196			done(sigdeadstatus);
197			}
198
199			/* ARGSUSED */
200			void
201			finish(int signo)
202			{
203			int save_errno = errno;
204			int status, e = 1;
205			pid_t pid;
206
207			while ((pid = wait3(&status, WNOHANG, 0)) > 0) {
208			if (pid == (pid_t)child) {
209			if (WIFEXITED(status))
210			e = WEXITSTATUS(status);
211			}
212			}
213			dead = 1;
214			sigdeadstatus = e;
215			errno = save_errno;
216			}
217
218			/* ARGSUSED */
219			void
220			handlesigwinch(int signo)
221			{
222			int save_errno = errno;
223			struct winsize win;
224			pid_t pgrp;
225
226			if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) != -1) {
227			ioctl(slave, TIOCSWINSZ, &win);
228			if (ioctl(slave, TIOCGPGRP, &pgrp) != -1)
229			killpg(pgrp, SIGWINCH);
230			}
231			errno = save_errno;
232			}
233
234			void
235			dooutput(void)
236			{
237			struct sigaction sa;
238			struct itimerval value;
239			sigset_t blkalrm;
240			char obuf[BUFSIZ];
241			time_t tvec;
242			ssize_t outcc = 0, cc, off;
243
244			(void)close(STDIN_FILENO);
245			tvec = time(NULL);
246			(void)fprintf(fscript, "Script started on %s", ctime(&tvec));
247
248			sigemptyset(&blkalrm);
249			sigaddset(&blkalrm, SIGALRM);
250			bzero(&sa, sizeof sa);
251			sigemptyset(&sa.sa_mask);
252			sa.sa_handler = scriptflush;
253			(void)sigaction(SIGALRM, &sa, NULL);
254
255			bzero(&sa, sizeof sa);
256			sigemptyset(&sa.sa_mask);
257			sa.sa_handler = SIG_IGN;
258			(void)sigaction(SIGCHLD, &sa, NULL);
259
260			if (pledge("stdio proc flock rpath cpath wpath", NULL) == -1)
261			err(1, "pledge");
262
263			value.it_interval.tv_sec = 30;
264			value.it_interval.tv_usec = 0;
265			value.it_value = value.it_interval;
266			(void)setitimer(ITIMER_REAL, &value, NULL);
267			for (;;) {
268			if (flush) {
269			if (outcc) {
270			(void)fflush(fscript);
271			outcc = 0;
272			}
273			flush = 0;
274			}
275			cc = read(master, obuf, sizeof (obuf));
276			if (cc == -1 && errno == EINTR)
277			continue;
278			if (cc <= 0)
279			break;
280			sigprocmask(SIG_BLOCK, &blkalrm, NULL);
281			for (off = 0; off < cc; ) {
282			ssize_t n = write(STDOUT_FILENO, obuf + off, cc - off);
283			if (n == -1 && errno != EAGAIN)
284			break;
285			if (n == 0)
286			break; /* skip writing */
287			if (n > 0)
288			off += n;
289			}
290			(void)fwrite(obuf, 1, cc, fscript);
291			outcc += cc;
292			sigprocmask(SIG_UNBLOCK, &blkalrm, NULL);
293			}
294			done(0);
295			}
296
297			/* ARGSUSED */
298			void
299			scriptflush(int signo)
300			{
301			flush = 1;
302			}
303
304			void
305			doshell(void)
306			{
307			char *shell;
308
309			shell = getenv("SHELL");
310			if (shell == NULL)
311			shell = _PATH_BSHELL;
312
313			(void)close(master);
314			(void)fclose(fscript);
315			login_tty(slave);
316			execl(shell, shell, "-i", (char *)NULL);
317			warn("%s", shell);
318			fail();
319			}
320
321			void
322			fail(void)
323			{
324
325			(void)kill(0, SIGTERM);
326			done(1);
327			}
328
329			void
330			done(int eval)
331			{
332			time_t tvec;
333
334			if (subchild) {
335			tvec = time(NULL);
336			(void)fprintf(fscript,"\nScript done on %s", ctime(&tvec));
337			(void)fclose(fscript);
338			(void)close(master);
339			} else {
340			if (istty)
341			(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
342			(void)printf("Script done, output file is %s\n", fname);
343			}
344			exit(eval);
345			}