BSD 4_1_snap release

[unix-history] / usr / src / cmd / lpr / lpr.c

static  char *sccsid = "@(#)lpr.c       4.4 (Berkeley) 81/05/21";
#include <signal.h>
#include <pagsiz.h>

#define BUFSIZ  BSIZE
/*
* lpr -- on-line printer spooler
*/

char    tfname[]        = "/usr/spool/lpd/tfaXXXXX";
char    cfname[]        = "/usr/spool/lpd/cfaXXXXX";
char    lfname[]        = "/usr/spool/lpd/lfaXXXXX";
char    dfname[]        = "/usr/spool/lpd/dfaXXXXX";
int     nact;
int     tff;
int     mailflg;
char    person[10];
int     inchar;
int     maxrec  = 400;
char    *prnt_ban;

main(argc, argv)
       int argc;
       char *argv[];
{
       register char *arg, *remote;
       int c, f, flag;
       int out();

       pidfn();
       if (signal(SIGINT, SIG_IGN) == SIG_DFL)
               signal(SIGINT, out);
       if (signal(SIGQUIT, SIG_IGN) == SIG_DFL)
               signal(SIGQUIT, out);
       if (signal(SIGHUP, SIG_IGN) == SIG_DFL)
               signal(SIGHUP, out);
       if (signal(SIGTERM, SIG_IGN) == SIG_DFL)
               signal(SIGTERM, out);
       remote = "$     remote  **,onl";
       flag = 0;
       tff = nfile(tfname);
       while (argc>1 && (arg = argv[1])[0]=='-') {
               if (arg[1] && arg[2]) {
                       remote[12] = arg[1];
                       remote[13] = arg[2];
                       remote[14] = 0;
               } else switch (arg[1]) {

               case '-':
                       remote[12] = 'r';
                       remote[13] = '1';
                       remote[14] = '\0';
                       break;

               case 'b':
                       prnt_ban = &argv[2][0];
                       argv++;
                       argc--;
                       break;
               case 'c':
                       flag = '+';
                       break;

               case 'r':
                       flag = '-';
                       break;

               case 'm':
                       mailflg = 1;
                       break;
               }
               argc--;
               argv++;
       }
       ident();
       if(argc == 1)
               copy(0);
       while(--argc) {
               arg = *++argv;
               if(flag == '+')
                       goto cf;
/*
* This may not work because the daemon runs as someone else, so don't bother
               if(*arg == '/' && flag != '-') {
                       card('F', arg);
                       nact++;
                       continue;
               }
*/
               if(link(arg, lfname) < 0)
                       goto cf;
               card('F', lfname);
               card('U', lfname);
               lfname[inchar]++;
               nact++;
               goto df;

       cf:
               f = open(arg, 0);
               if(f < 0) {
                       printf("Cannot open %s\n", arg);
                       continue;
               }
               copy(f);
               close(f);

       df:
               if(flag == '-') {
                       f = unlink(arg);
                       if(f < 0)
                               printf("Cannot remove %s\n", arg);
               }
       }

       if(nact) {
               tfname[inchar]--;
               f = link(tfname, dfname);
               if(f < 0) {
                       printf("Cannot rename %s\n", dfname);
                       tfname[inchar]++;
                       out();
               }
               unlink(tfname);
               execl("/usr/lib/lpd", "lpd", 0);
               dfname[inchar]++;
               printf("Daemon doesn't exist\n");
               exit(0);
       }
       out();
}

copy(f)
int f;
{
       int ff, i, j, nr, nc;
       static char buf[BUFSIZ];

       card('F', cfname);
       card('U', cfname);
       ff = nfile(cfname);
       nc = 0;
       nr = 0;
       while((i = read(f, buf, BUFSIZ)) > 0) {
               /* Avoid premature trunction of files */
               for (j=0; j<i; j++) {
                       buf[j] &= 0177;
                       if (buf[j] == 0)
                               buf[j] = 1;
               }
               write(ff, buf, i);
               nc += i;
               if(nc >= BUFSIZ) {
                       nc -= BUFSIZ;
                       nr++;
                       if(nr > maxrec) {
                               printf("Copy file is too large\n");
                               break;
                       }
               }
       }
       close(ff);
       nact++;
}

card(c, s)
int c;
char s[];
{
       char *p1, *p2;
       static char buf[BUFSIZ];
       int col;

       p1 = buf;
       p2 = s;
       col = 0;
       *p1++ = c;
       while((c = *p2++) != '\0') {
               *p1++ = c;
               col++;
       }
       *p1++ = '\n';
       write(tff, buf, col+2);
}

ident()
{
       int c, n;
       register char *b1p, *pp, *b2p;
       static char b1[100], b2[100];

       b1p = b1;
       if(getpw(getuid(), b1p)) {
               b1p = "pdp::::m0000,m000:";
       }
       n = 0;
       b2p = b2;
       while(*b2p++ = "$       ident   "[n++]);
       b2p--;
       n = 5;
       while(--n) while(*b1p++ != ':');
       while((*b2p++ = *b1p++) != ':');
       b2p[-1] = ',';
       b1p = b1;
       pp = person;
       while((c = *b1p++) != ':') {
               *b2p++ = c;
               *pp++ = c;
       }
       *b2p++ = 0;
       *pp++ = 0;
       card('L', person);
       if (prnt_ban != 0 )
               card('B',prnt_ban);
       if (mailflg)
               card('M', person);
}

pidfn()
{
       register i, j, c;
       int s;
       int p;

       s = p = getpid();
       p &= 077777;
       i = 0;
       while(tfname[i] != 'X')
               i++;
       i += 4;
       for(j=0; j<5; j++) {
               c = (p%10) + '0';
               if(s<0 && j==4)
                       c += 4;
               p /= 10;
               tfname[i] = c;
               cfname[i] = c;
               lfname[i] = c;
               dfname[i] = c;
               i--;
       }
       inchar = i;
}

nfile(name)
char *name;
{
       register f;

       f = creat(name, 0644);
       if(f < 0) {
               printf("Cannot create %s\n", name);
               out();
       }
       name[inchar]++;
       return(f);
}

out()
{
       register i;

       signal(SIGINT, SIG_IGN);
       signal(SIGQUIT, SIG_IGN);
       signal(SIGHUP, SIG_IGN);
       signal(SIGTERM, SIG_IGN);
       i = inchar;
       while(tfname[i] != 'a') {
               tfname[i]--;
               unlink(tfname);
       }
       while(cfname[i] != 'a') {
               cfname[i]--;
               unlink(cfname);
       }
       while(lfname[i] != 'a') {
               lfname[i]--;
               unlink(lfname);
       }
       while(dfname[i] != 'a') {
               dfname[i]--;
               unlink(dfname);
       }
       exit(0);
}