/*  * This file is part of the UCB release of Plan 9. It is subject to the license * terms in the LICENSE file found in the top-level directory of this * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No * part of the UCB release of Plan 9, including this file, may be copied, * modified, propagated, or distributed except according to the terms contained * in the LICENSE file. */ /* * shell buffered IO and formatted output */ #include <ctype.h>#include "sh.h"#include "ksh_stat.h" static int initio_done; /* * formatted output functions */  /* A shell error occured (eg, syntax error, etc.) */void#ifdef HAVE_PROTOTYPESerrorf(const char *fmt, ...)#elseerrorf(fmt, va_alist)	const char *fmt;	va_dcl#endif{	va_list va; 	shl_stdout_ok = 0;	/* debugging: note that stdout not valid */	exstat = 1;	if (*fmt) {		error_prefix(TRUE);		SH_VA_START(va, fmt);		shf_vfprintf(shl_out, fmt, va);		va_end(va);		shf_putchar('\n', shl_out);	}	shf_flush(shl_out);	unwind(LERROR);} /* like errorf(), but no unwind is done */void#ifdef HAVE_PROTOTYPESwarningf(int fileline, const char *fmt, ...)#elsewarningf(fileline, fmt, va_alist)	int fileline;	const char *fmt;	va_dcl#endif{	va_list va; 	error_prefix(fileline);	SH_VA_START(va, fmt);	shf_vfprintf(shl_out, fmt, va);	va_end(va);	shf_putchar('\n', shl_out);	shf_flush(shl_out);} /* Used by built-in utilities to prefix shell and utility name to message * (also unwinds environments for special builtins). */void#ifdef HAVE_PROTOTYPESbi_errorf(const char *fmt, ...)#elsebi_errorf(fmt, va_alist)	const char *fmt;	va_dcl#endif{	va_list va; 	shl_stdout_ok = 0;	/* debugging: note that stdout not valid */	exstat = 1;	if (*fmt) {		error_prefix(TRUE);		/* not set when main() calls parse_args() */		if (builtin_argv0)			shf_fprintf(shl_out, "%s: ", builtin_argv0);		SH_VA_START(va, fmt);		shf_vfprintf(shl_out, fmt, va);		va_end(va);		shf_putchar('\n', shl_out);	}	shf_flush(shl_out);	/* POSIX special builtins and ksh special builtins cause	 * non-interactive shells to exit.	 * XXX odd use of KEEPASN; also may not want LERROR here	 */	if ((builtin_flag & SPEC_BI)	    || (Flag(FPOSIX) && (builtin_flag & KEEPASN)))	{		builtin_argv0 = (char *) 0;		unwind(LERROR);	}} /* Called when something that shouldn't happen does */void#ifdef HAVE_PROTOTYPESinternal_errorf(int jump, const char *fmt, ...)#elseinternal_errorf(jump, fmt, va_alist)	int jump;	const char *fmt;	va_dcl#endif{	va_list va; 	error_prefix(TRUE);	shf_fprintf(shl_out, "internal error: ");	SH_VA_START(va, fmt);	shf_vfprintf(shl_out, fmt, va);	va_end(va);	shf_putchar('\n', shl_out);	shf_flush(shl_out);	if (jump)		unwind(LERROR);} /* used by error reporting functions to print "ksh: .kshrc[25]: " */voiderror_prefix(fileline)	int fileline;{	/* Avoid foo: foo[2]: ... */	if (!fileline || !source || !source->file	    || strcmp(source->file, kshname) != 0)		shf_fprintf(shl_out, "%s: ", kshname + (*kshname == '-'));	if (fileline && source && source->file != NULL) {		shf_fprintf(shl_out, "%s[%d]: ", source->file,			source->errline > 0 ? source->errline : source->line);		source->errline = 0;	}} /* printf to shl_out (stderr) with flush */void#ifdef HAVE_PROTOTYPESshellf(const char *fmt, ...)#elseshellf(fmt, va_alist)	const char *fmt;	va_dcl#endif{	va_list va; 	if (!initio_done) /* shl_out may not be set up yet... */		return;	SH_VA_START(va, fmt);	shf_vfprintf(shl_out, fmt, va);	va_end(va);	shf_flush(shl_out);} /* printf to shl_stdout (stdout) */void#ifdef HAVE_PROTOTYPESshprintf(const char *fmt, ...)#elseshprintf(fmt, va_alist)	const char *fmt;	va_dcl#endif{	va_list va; 	if (!shl_stdout_ok)		internal_errorf(1, "shl_stdout not valid");	SH_VA_START(va, fmt);	shf_vfprintf(shl_stdout, fmt, va);	va_end(va);} #ifdef KSH_DEBUGstatic struct shf *kshdebug_shf; voidkshdebug_init_(){	if (kshdebug_shf)		shf_close(kshdebug_shf);	kshdebug_shf = shf_open("/tmp/ksh-debug.log",				O_WRONLY|O_APPEND|O_CREAT, 0600,				SHF_WR|SHF_MAPHI);	if (kshdebug_shf) {		shf_fprintf(kshdebug_shf, "\nNew shell[pid %d]\n", getpid());		shf_flush(kshdebug_shf);	}} /* print to debugging log */void# ifdef HAVE_PROTOTYPESkshdebug_printf_(const char *fmt, ...)# elsekshdebug_printf_(fmt, va_alist)	const char *fmt;	va_dcl# endif{	va_list va; 	if (!kshdebug_shf)		return;	SH_VA_START(va, fmt);	shf_fprintf(kshdebug_shf, "[%d] ", getpid());	shf_vfprintf(kshdebug_shf, fmt, va);	va_end(va);	shf_flush(kshdebug_shf);} voidkshdebug_dump_(str, mem, nbytes)	const char *str;	const void *mem;	int nbytes;{	int i, j;	int nprow = 16; 	if (!kshdebug_shf)		return;	shf_fprintf(kshdebug_shf, "[%d] %s:\n", getpid(), str);	for (i = 0; i < nbytes; i += nprow) {		char c = '\t';		for (j = 0; j < nprow && i + j < nbytes; j++) {			shf_fprintf(kshdebug_shf, "%c%02x",				c, ((const unsigned char *) mem)[i + j]);			c = ' ';		}		shf_fprintf(kshdebug_shf, "\n");	}	shf_flush(kshdebug_shf);}#endif /* KSH_DEBUG */ /* test if we can seek backwards fd (returns 0 or SHF_UNBUF) */intcan_seek(fd)	int fd;{	struct stat statb; 	return fstat(fd, &statb) == 0 && !S_ISREG(statb.st_mode) ?		SHF_UNBUF : 0;} struct shf	shf_iob[3]; voidinitio(){	shf_fdopen(1, SHF_WR, shl_stdout);	/* force buffer allocation */	shf_fdopen(2, SHF_WR, shl_out);	shf_fdopen(2, SHF_WR, shl_spare);	/* force buffer allocation */	initio_done = 1;	kshdebug_init();} /* A dup2() with error checking */intksh_dup2(ofd, nfd, errok)	int ofd;	int nfd;	int errok;{	int ret = dup2(ofd, nfd); 	if (ret < 0 && errno != EBADF && !errok)		errorf("too many files open in shell"); #ifdef DUP2_BROKEN	/* Ultrix systems like to preserve the close-on-exec flag */	if (ret >= 0)		(void) fcntl(nfd, F_SETFD, 0);#endif /* DUP2_BROKEN */ 	return ret;} /* * move fd from user space (0<=fd<10) to shell space (fd>=10), * set close-on-exec flag. */intsavefd(fd, noclose)	int fd;	int noclose;{	int nfd; 	if (fd < FDBASE) {		nfd = ksh_dupbase(fd, FDBASE);		if (nfd < 0)			if (errno == EBADF)				return -1;			else				errorf("too many files open in shell");		if (!noclose)			close(fd);	} else		nfd = fd;	fd_clexec(nfd);	return nfd;} voidrestfd(fd, ofd)	int fd, ofd;{	if (fd == 2)		shf_flush(&shf_iob[fd]);	if (ofd < 0)		/* original fd closed */		close(fd);	else {		ksh_dup2(ofd, fd, TRUE); /* XXX: what to do if this fails? */		close(ofd);	}} voidopenpipe(pv)	register int *pv;{	if (pipe(pv) < 0)		errorf("can't create pipe - try again");	pv[0] = savefd(pv[0], 0);	pv[1] = savefd(pv[1], 0);} voidclosepipe(pv)	register int *pv;{	close(pv[0]);	close(pv[1]);} /* Called by iosetup() (deals with 2>&4, etc.), c_read, c_print to turn * a string (the X in 2>&X, read -uX, print -uX) into a file descriptor. */intcheck_fd(name, mode, emsgp)	char *name;	int mode;	const char **emsgp;{	int fd, fl; 	if (isdigit(name[0]) && !name[1]) {		fd = name[0] - '0';		if ((fl = fcntl(fd = name[0] - '0', F_GETFL, 0)) < 0) {			if (emsgp)				*emsgp = "bad file descriptor";			return -1;		}		fl &= O_ACCMODE;#ifdef OS2		if (mode == W_OK ) { 		       if (setmode(fd, O_TEXT) == -1) {				if (emsgp)					*emsgp = "couldn't set write mode";				return -1;			}		 } else if (mode == R_OK)	      		if (setmode(fd, O_BINARY) == -1) {				if (emsgp)					*emsgp = "couldn't set read mode";				return -1; 			}#else /* OS2 */		/* X_OK is a kludge to disable this check for dups (x<&1):		 * historical shells never did this check (XXX don't know what		 * posix has to say).		 */		if (!(mode & X_OK) && fl != O_RDWR		    && (((mode & R_OK) && fl != O_RDONLY)			|| ((mode & W_OK) && fl != O_WRONLY)))		{			if (emsgp)				*emsgp = (fl == O_WRONLY) ?						"fd not open for reading"					      : "fd not open for writing";			return -1;		}#endif /* OS2 */		return fd;	}#ifdef KSH	else if (name[0] == 'p' && !name[1])		return coproc_getfd(mode, emsgp);#endif /* KSH */	if (emsgp)		*emsgp = "illegal file descriptor name";	return -1;} #ifdef KSH/* Called once from main */voidcoproc_init(){	coproc.read = coproc.readw = coproc.write = -1;	coproc.njobs = 0;	coproc.id = 0;} /* Called by c_read() when eof is read - close fd if it is the co-process fd */voidcoproc_read_close(fd)	int fd;{	if (coproc.read >= 0 && fd == coproc.read) {		coproc_readw_close(fd);		close(coproc.read);		coproc.read = -1;	}} /* Called by c_read() and by iosetup() to close the other side of the * read pipe, so reads will actually terminate. */voidcoproc_readw_close(fd)	int fd;{	if (coproc.readw >= 0 && coproc.read >= 0 && fd == coproc.read) {		close(coproc.readw);		coproc.readw = -1;	}} /* Called by c_print when a write to a fd fails with EPIPE and by iosetup * when co-process input is dup'd */voidcoproc_write_close(fd)	int fd;{	if (coproc.write >= 0 && fd == coproc.write) {		close(coproc.write);		coproc.write = -1;	}} /* Called to check for existance of/value of the co-process file descriptor. * (Used by check_fd() and by c_read/c_print to deal with -p option). */intcoproc_getfd(mode, emsgp)	int mode;	const char **emsgp;{	int fd = (mode & R_OK) ? coproc.read : coproc.write; 	if (fd >= 0)		return fd;	if (emsgp)		*emsgp = "no coprocess";	return -1;} /* called to close file descriptors related to the coprocess (if any) * Should be called with SIGCHLD blocked. */voidcoproc_cleanup(reuse)	int reuse;{	/* This to allow co-processes to share output pipe */	if (!reuse || coproc.readw < 0 || coproc.read < 0) {		if (coproc.read >= 0) {			close(coproc.read);			coproc.read = -1;		}		if (coproc.readw >= 0) {			close(coproc.readw);			coproc.readw = -1;		}	}	if (coproc.write >= 0) {		close(coproc.write);		coproc.write = -1;	}}

Lines 1–500 of 560. Load more