V7M/src/cmd/lpr/spool.c

Compare this file to the similar file:
Show the results in this format:

#include	<signal.h>
#include	<sys/types.h>
#include	<sys/stat.h>

#define	ONL	0
#define	TOSS	1
int	INCHAR	= 0;		/*index of incremented character in
					temporary file names. */

char	version[] = "Version 2/6/79";

char	grade;
char	remote[]= "$	remote	**,onl";
char	toss[]	= "$	sysout	toss";
int	remotsw;		/*toss-output flag*/
char	*mailfile = 0;
char	wantmail = 0;
char	*pp	= 0;		/*recipient of mail*/
char	*identf = 0;		/*ident card info*/
int	uidf	= 0;
char	gcosid[13];		/*gcos userid*/
char	cpflag = 'l';		/*copy/link flag*/
int	rmflag	= 0;		/*remove flag*/
int	debug	= 0;
int	gcdebug	= 0;		/*GCOS debug switch*/
int	archsw = 0;		/*archive switch*/

int	argc;
char	**argv;
char	*arg;
char	buf[80];		/*used by card */
int	nact = 0;		/*number of non-null files to process.*/
int	gsize	= 20;		/*size of current file in GCOS blocks.*/
long	usize	= 20*1200;	/*size of current file in bytes.*/
FILE	*tff;		/*temporary control card file*/
FILE	*nfile();
char	*getarg();
char	*sprintf();


comopt(o)		/*routine to test for common options.*/
char o;
{
	switch (o){

	case 'c':
		cpflag = 'c';
		break;

	case 'i':
		identf = getarg('i');
		break;

	case 'm':
		wantmail++;
		if(arg[2])
			pp = &arg[2];
		break;
	
	case 'n':		/*new option to suppress mail. MRW*/
		wantmail = 0;
		break;

	case 'o':
		remotsw = ONL;
		break;

	case 'r':
		rmflag++;
		break;

	case 's':
		if(arg[2] < '1' || arg[2] > '3')
			goto unknown;
		grade = arg[2];
		break;

	case 't':
		if(arg[2])
			goto unknown;
		remotsw = TOSS;
		break;

	case '#':
		debug = 1;
		break;

	case 'Z':			/*GCOS debugging switch*/
		gcdebug = 1;
		break;

unknown:
	default:
		return(0);
	}
	return(1);
}


#if LPR == 0

spool1()		/*set up common initial GCOS control cards.*/
{
	if(debug)
		tff = stdout;
	else
		if((tff = nfile(tfname)) == NULL){
			fprintf(stderr, "%s: Can't create %s.\n", NAME, tfname);
			out();
		}
	card('S', "");
	card('L', sprintf(buf, "$	sgrade	%c   %s", grade, version ) );
	if(ident())
		out();
	card('L', remote);
	if(remotsw == TOSS)
		card('L', toss);
}


spool2()			/*add final control cards, and spool job.*/
{
	if(wantmail)
		card('N', mailfile);
	card('L', "$	endjob");
	if(debug)
		out();
	fclose(tff);
	if(nact) {
		dfname[INCHAR]++;
		if(link(tfname, dfname) < 0){
			fprintf(stderr, "%s: Cannot rename %s\n", NAME, tfname);
			out();
		}
		unlink(tfname);
		execl("/usr/lib/dpd", "dpd", 0);
		execl("/etc/dpd", "dpd", 0);
		fprintf(stderr, "%s: Can't find dpd.\nFiles left in spooling dir.\n", NAME);
		exit(1);
	}
}

#endif


#if FGET == 0

filargs()		/*process file arguments for dpr, gcat, fsend, lpr.*/
{
	int i;
	FILE *f;

	if(argc == 1){
		if(mailfile == 0)
			mailfile = "pipe.end";
		if(copy(stdin, mailfile, GCAT) == -1)
			out();
		if(archsw)
			archive();
	}
	while(--argc) {
		arg = *++argv;
		switch(cpflag){

		case 'l':
			if(lfname[INCHAR]++ >= 'z')
				cpflag = rmflag ? 'c' : 'n';
			else if(link(arg, lfname) == 0){
				if(size(arg,arg) <= 0)
					continue;
				nuact(arg);
				card(BF, lfname);
				card('U', lfname);
				break;
			}

		case 'n':
			if(*arg == '/' && !rmflag){
				if(size(arg,arg) <= 0)
					continue;
				nuact(arg);
				card(BF, arg);
				break;
			}

		case 'c':
			f = fopen(arg, "r");
			if(f == NULL){
				fprintf(stderr, "%s: Cannot open %s\n", NAME, arg);
				continue;
			}
			i = copy(f, arg, GCAT);
			fclose(f);
			if(i == -1)
				continue;
			break;
		}
		if(archsw)
			archive();
		if(rmflag){
			if(unlink(arg) < 0)
				fprintf(stderr, "%s: Cannot remove %s\n", NAME, arg);
		}
		if(mailfile == 0)
			mailfile = arg;
	}
}

#endif


FILE *nfile(name)		/*generate a new file name, and open file.*/
char *name;
{
	FILE *f;

	if(name[INCHAR] >= 'z')
		return(NULL);
	name[INCHAR]++;
	if(!access(name, 0) || (f = fopen(name, "w")) == NULL)
		return(NULL);
	return(f);
}

#if FGET == 0
 
copy(f, gname, gcatsw)
FILE	*f;
char	*gname;
int	gcatsw;
{
	int c;
	FILE *ff;
	long cnt;

	if((ff = nfile(cfname)) == NULL){
		fprintf(stderr, "%s: Too many copy files; %s not copied\n", NAME, gname);
		return(-1);
	}
	cnt = 0;
	while((c = getc(f)) != EOF){
		if(gcatsw)
		    if(c != 0){
			fprintf(stderr, "%s: Bad input from %s.\n", NAME, gname);
			out();
		    }else  gcatsw = 0;
		if((putc(c, ff) == EOF) && ferror(ff)){
			fprintf(stderr, "%s: Write error on copy of %s.\n", NAME, gname);
			break;
		}
		cnt++;
		if(cnt > MAXCOPY){
			fprintf(stderr, "%s: Copy file %s is too large\n", NAME, gname);
			break;
		}
	}
	fclose(ff);
	if(size(cfname,gname) <= 0)
		return(-1);
	nuact(gname);
	card(BF, cfname);
	card('U', cfname);
	return(0);
}

#endif

card(c, s)
int c;
char	*s;
{
	putc( c, tff );

	while( (c = *s++) != '\0') putc( c, tff );

	c = putc( '\n', tff );

	if(c == EOF){
		fprintf(stderr, "%s: Error writing control file.\n", NAME);
		out();
		}
}

size(file, name)
char	*file, *name;
{
	struct stat stbuf;

	if(stat(file,&stbuf) < 0){
		fprintf(stderr, "%s: Cannot open %s\n", NAME, file);
		return(-1);
	}
	if(!stbuf.st_size){
		fprintf(stderr, "%s: File %s is empty.\n", NAME, name);
		return(0);
	}
	usize = stbuf.st_size;
	gsize = usize / 1200;
	gsize++;
	nact++;
	return(gsize);
}


char *
getarg(c)		/*get modifier for complex options --
			    from either same or next argument. MRW
			    e.g. either "-ffile" or "-f file"*/
char	c;
{

	if(arg[2])
		return(&arg[2]);
	else if(--argc>1)
		return(arg = (++argv)[1]);
	fprintf(stderr, "%s: Incomplete -%c option\n", NAME,c);
	out();
}

#include	<pwd.h>
struct passwd *getpwuid();

ident()
{
	int c, i, j, n, test, jsave;
	struct passwd *b1;
	static char b2[100];

	if((b1 = getpwuid(getuid())) == NULL) {
		fprintf(stderr, "%s: Invalid user id\n", NAME);
		return(1);
	}
	j = 0;
#if LPR == 0
	while(c = "$	ident	"[j])
		b2[j++] = c;

	i = 0;
	if(identf) 
		while(c = identf[i++])
			b2[j++] = c;
	else{
		jsave = j;		/*use either usg or pwb-style passwd. MRW*/
		while((c = b1->pw_gecos[i++]) && c != ':')
			if(c == ')')
				j = jsave;
			else
				b2[j++] = c;
	}
	b2[j++] = ',';
#endif

	i = 0;
	if(!pp)
		pp = &b2[j];
	while(c = b1->pw_name[i++])
		b2[j++] = c;
	b2[j] = '\0';

#if LPR == 0
	i = 0;
	n = 3;
	while(--n) {
		test = 0;
		while((c=b2[i++]) && c != ',') {
			if('0' <= c && c <= '9') test += c -'0';
			else test = 0;
		}
		if(test == 0) {
			b2[j] = '\0';
			fprintf(stderr, "%s: Invalid IDENT information - %s\n", NAME, b2);
			return (1);
		}
	}

	if(!uidf) {
		n = 0;
		while((c = b2[i++]) && c != ',') {
			if(n >= 12) break;
			gcosid[n++] = c;
		}
		gcosid[n++] = '\0';
	}
#endif
	card('L', b2);
	if(wantmail){
		card('M',pp);
		if(identf)
			card('Q', b2);	/*mail back $IDENT card.*/
	}
	return (0);
}

pidfn()			/*rewrite using mktemp. MRW*/
{
	int out();

	while(tfname[INCHAR] != 'X')
		INCHAR++;
	INCHAR--;
	mktemp(cfname);
	mktemp(dfname);
	mktemp(lfname);
	mktemp(tfname);
	mktemp(zfname);
	if(signal(SIGHUP, SIG_IGN) != SIG_IGN)
		signal(SIGHUP, out);
	if(signal(SIGINT, SIG_IGN) != SIG_IGN)
		signal(SIGINT, out);
	if(signal(SIGQUIT, SIG_IGN) != SIG_IGN)
		signal(SIGQUIT, out);
	if(signal(SIGTERM, SIG_IGN) != SIG_IGN)
		signal(SIGTERM, out);
}

out()
{
	register i;

	signal(SIGHUP, SIG_IGN);
	signal(SIGINT, SIG_IGN);
	signal(SIGQUIT, SIG_IGN);
	signal(SIGTERM, SIG_IGN);
	i = INCHAR;
	for(; cfname[i] != FIRSTCHAR; cfname[i]--) 
		unlink(cfname);
	if(dfname[i] != FIRSTCHAR)
		unlink(dfname);
	for(; lfname[i] != FIRSTCHAR; lfname[i]--) 
		unlink(lfname);
	if(tfname[i] != FIRSTCHAR)
		unlink(tfname);
	for(; zfname[i] != FIRSTCHAR; zfname[i]--) 
		unlink(zfname);
	exit(1);
}