Annotation of researchv10no/cmd/at/at.c, revision 1.1.1.1

1.1       root        1: /*
                      2:  * at time mon day
                      3:  * at time wday
                      4:  * at time wday 'week'
                      5:  *
                      6:  */
                      7: #include <stdio.h>
                      8: #include <ctype.h>
                      9: #include <time.h>
                     10: #include <signal.h>
                     11: #include <sys/types.h>
                     12: #include <sys/stat.h>
                     13: #include <dir.h>
                     14: 
                     15: #define HOUR 100
                     16: #define HALFDAY        (12*HOUR)
                     17: #define DAY    (24*HOUR)
                     18: #define THISDAY "/usr/spool/at"
                     19: 
                     20: char *days[] = {
                     21:        "sunday",
                     22:        "monday",
                     23:        "tuesday",
                     24:        "wednesday",
                     25:        "thursday",
                     26:        "friday",
                     27:        "saturday",
                     28: };
                     29: 
                     30: struct monstr {
                     31:        char *mname; 
                     32:        int mlen;
                     33: } months[] = {
                     34:        { "january", 31 },
                     35:        { "february", 28 },
                     36:        { "march", 31 },
                     37:        { "april", 30 },
                     38:        { "may", 31 },
                     39:        { "june", 30 },
                     40:        { "july", 31 },
                     41:        { "august", 31 },
                     42:        { "september", 30 },
                     43:        { "october", 31 },
                     44:        { "november", 30 },
                     45:        { "december", 31 },
                     46:        { 0, 0 },
                     47: };
                     48: 
                     49: char   fname[100];
                     50: int    utime;  /* requested time in grains */
                     51: int    now;    /* when is it */
                     52: int    uday; /* day of year to be done */
                     53: int    uyear; /* year */
                     54: int    today; /* day of year today */
                     55: int    thisyear; /* this year */
                     56: FILE   *file;
                     57: FILE   *ifile;
                     58: char   **environ;
                     59: char   *prefix();
                     60: char   *idend();
                     61: char   *strchr();
                     62: FILE   *popen();
                     63: 
                     64: extern int optind;
                     65: char   lflg;
                     66: char   rflg;
                     67: char   errflg;
                     68: 
                     69: void   prdir(), prfile(), delfile(), vprint();
                     70: int    readdir();
                     71: char *pathdate();
                     72: 
                     73: #define EXP 10
                     74: 
                     75: main(argc, argv)
                     76: char **argv;
                     77: {
                     78:        extern onintr();
                     79:        register c;
                     80:        char pwbuf[100];
                     81:        FILE *pwfil;
                     82:        int larg;
                     83:        static char stdbuf[BUFSIZ];
                     84: 
                     85:        setbuf (stdout, stdbuf);
                     86: 
                     87:        while ((c = getopt (argc, argv, "rl")) != EOF) {
                     88:                switch (c) {
                     89: 
                     90:                case 'l':
                     91:                        lflg = 1;
                     92:                        break;
                     93:                
                     94:                case 'r':
                     95:                        rflg = 1;
                     96:                        break;
                     97:                
                     98:                default:
                     99:                        errflg = 1;
                    100:                        break;
                    101:                }
                    102:        }
                    103: 
                    104:        if (errflg) {
                    105:                fprintf (stderr, "usage: at [lr] time [filename]\n");
                    106:                exit (1);
                    107:        }
                    108: 
                    109:        /* argv[optind] is the user's time: e.g.,  3AM */
                    110:        /* argv[optind+1] is a month name or day of week */
                    111:        /* argv[optind+2] is day of month or 'week' */
                    112:        /* another argument might be an input file */
                    113:        if (argc < optind+1) {
                    114:                if (lflg && !rflg) {
                    115:                        (void) readdir("", prdir);
                    116:                        exit (0);
                    117:                }
                    118:                fprintf(stderr, "at: arg count\n");
                    119:                exit(1);
                    120:        }
                    121:        makeutime(argv[optind]);
                    122:        larg = makeuday(argc-optind+1,argv+optind-1)+optind;
                    123:        if (uday==today && larg<=optind+1 && utime<=now)
                    124:                uday++;
                    125:        c = uyear%4==0? 366: 365;
                    126:        /*
                    127:        The way makeuday is written, the following "if" is rarely
                    128:        satisfied. This makes a problem, which is fixed below.
                    129:        */
                    130:        if (uday >= c) {
                    131:                uday -= c;
                    132:                uyear++;
                    133:        }
                    134:        /* 
                    135:        Due to problem described above, "at" schedules a Jan call in
                    136:        Dec for LAST Jan.  Line below fixes that.
                    137:        */
                    138:        if(uday<today && uyear==thisyear)
                    139:                uyear++;
                    140: 
                    141:        if (lflg || rflg) {
                    142:                char prefix[50];
                    143:                sprintf (prefix, "%.2d.%.3d.%.4d", uyear, uday, utime);
                    144:                if (lflg) {
                    145:                        if (readdir (prefix, prfile) == 0)
                    146:                                exit (0);
                    147:                        if (rflg)
                    148:                                putchar ('\n');
                    149:                }
                    150: 
                    151:                if (rflg) {
                    152:                        (void) readdir (prefix, delfile);
                    153:                }
                    154: 
                    155:                exit (0);
                    156:        }
                    157: 
                    158:        filename(THISDAY, uyear, uday, utime);
                    159:        /* Create file, then change UIDS */
                    160:        close(creat(fname, 0644));
                    161:        file = fopen(fname, "w");
                    162:        if (file == NULL) {
                    163:                fprintf(stderr, "at: cannot open memo file\n");
                    164:                exit(1);
                    165:        }
                    166:        chown(fname, getuid(), getgid());
                    167:        setuid(getuid());
                    168:        ifile = stdin;
                    169:        if (argc > larg)
                    170:                ifile = fopen(argv[larg], "r");
                    171:        if (ifile == NULL) {
                    172:                fprintf(stderr, "at: cannot open input: %s\n", argv[larg]);
                    173:                exit(1);
                    174:        }
                    175:        if (signal(SIGINT, SIG_IGN) != SIG_IGN)
                    176:                signal(SIGINT, onintr);
                    177:        if ((pwfil = popen("pwd", "r")) == NULL) {
                    178:                fprintf(stderr, "at: can't execute pwd\n");
                    179:                exit(1);
                    180:        }
                    181:        fgets(pwbuf, 100, pwfil);
                    182:        pclose(pwfil);
                    183:        fprintf(file, "cd %s", pwbuf);
                    184:        c = umask(0);
                    185:        umask(c);
                    186:        fprintf(file, "umask %#o\n", c);
                    187:        if (environ) {
                    188:                register int pass;
                    189:                char *explist[EXP];
                    190:                int explen[EXP];
                    191:                register int exp = 0;
                    192:                register int i;
                    193:                for (pass = 1; pass <= 2; pass++) {
                    194:                        register char **ep;
                    195:                        for (ep = environ; *ep; ep++) {
                    196:                                if (pass == 1) {
                    197:                                        vprint(file, *ep);
                    198:                                        putc('\n', file);
                    199:                                } else {
                    200:                                        char *ix;
                    201:                                        ix = idend(*ep);
                    202:                                        if (*ix) {
                    203:                                                explen[exp] = ix-*ep;
                    204:                                                explist[exp++] = *ep;
                    205:                                                if (exp >= EXP) {
                    206:                                                        fprintf(file,"export");
                    207:                                                        for (i=0; i<EXP; i++) {
                    208:                                                                putc(' ',file);
                    209:                                                                fprintf(file, "%.*s",explen[i],explist[i]);
                    210:                                                        }
                    211:                                                        putc ('\n', file);
                    212:                                                        exp = 0;
                    213:                                                }
                    214:                                        }
                    215:                                }
                    216:                        }
                    217: 
                    218:                        if (exp > 0) {
                    219:                                fprintf(file,"export");
                    220:                                for (i = 0; i < exp; i++) {
                    221:                                        putc(' ',file);
                    222:                                        fprintf(file, "%.*s",explen[i],explist[i]);                             }
                    223:                                putc ('\n', file);
                    224:                        }
                    225:                }
                    226:        }
                    227:        while((c = getc(ifile)) != EOF) {
                    228:                putc(c, file);
                    229:        }
                    230:        exit(0);
                    231: }
                    232: 
                    233: makeutime(pp)
                    234: char *pp; 
                    235: {
                    236:        register val;
                    237:        register char *p;
                    238: 
                    239:        /* p points to a user time */
                    240:        p = pp;
                    241:        val = 0;
                    242:        while(isdigit(*p)) {
                    243:                val = val*10+(*p++ -'0');
                    244:        }
                    245:        if (p-pp < 3)
                    246:                val *= HOUR;
                    247: 
                    248:        for (;;) {
                    249:                switch(*p) {
                    250: 
                    251:                case ':':
                    252:                        ++p;
                    253:                        if (isdigit(*p)) {
                    254:                                if (isdigit(p[1])) {
                    255:                                        val +=(10* *p + p[1] - 11*'0');
                    256:                                        p += 2;
                    257:                                        continue;
                    258:                                }
                    259:                        }
                    260:                        fprintf(stderr, "at: bad time format:\n");
                    261:                        exit(1);
                    262: 
                    263:                case 'A':
                    264:                case 'a':
                    265:                        if (val >= HALFDAY+HOUR)
                    266:                                val = DAY+1;  /* illegal */
                    267:                        if (val >= HALFDAY && val <(HALFDAY+HOUR))
                    268:                                val -= HALFDAY;
                    269:                        break;
                    270: 
                    271:                case 'P':
                    272:                case 'p':
                    273:                        if (val >= HALFDAY+HOUR)
                    274:                                val = DAY+1;  /* illegal */
                    275:                        if (val < HALFDAY)
                    276:                                val += HALFDAY;
                    277:                        break;
                    278: 
                    279:                case 'n':
                    280:                case 'N':
                    281:                        val = HALFDAY;
                    282:                        break;
                    283: 
                    284:                case 'M':
                    285:                case 'm':
                    286:                        val = 0;
                    287:                        break;
                    288: 
                    289: 
                    290:                case '\0':
                    291:                case ' ':
                    292:                        /* 24 hour time */
                    293:                        if (val == DAY)
                    294:                                val -= DAY;
                    295:                        break;
                    296: 
                    297:                default:
                    298:                        fprintf(stderr, "at: bad time format\n");
                    299:                        exit(1);
                    300: 
                    301:                }
                    302:                break;
                    303:        }
                    304:        if (val < 0 || val >= DAY) {
                    305:                fprintf(stderr, "at: time out of range\n");
                    306:                exit(1);
                    307:        }
                    308:        if (val%HOUR >= 60) {
                    309:                fprintf(stderr, "at: illegal minute field\n");
                    310:                exit(1);
                    311:        }
                    312:        utime = val;
                    313: }
                    314: 
                    315: 
                    316: makeuday(argc,argv)
                    317: char **argv;
                    318: {
                    319:        /* the presumption is that argv[2], argv[3] are either
                    320:           month day OR weekday [week].  Returns 2, 3, or 4 as last
                    321:           argument used */
                    322:        /* first of all, what's today */
                    323:        long tm;
                    324:        int found = -1;
                    325:        char **ps;
                    326:        struct tm *detail, *localtime();
                    327:        struct monstr *pt;
                    328: 
                    329:        time(&tm);
                    330:        detail = localtime(&tm);
                    331:        uday = today = detail->tm_yday;
                    332:        thisyear = uyear = detail->tm_year;
                    333:        now = detail->tm_hour*100+detail->tm_min;
                    334:        if (argc<=2)
                    335:                return(1);
                    336:        /* is the next argument a month name ? */
                    337:        for (pt=months; pt->mname; pt++) {
                    338:                if (prefix(argv[2], pt->mname)) {
                    339:                        if (found<0)
                    340:                                found = pt-months;
                    341:                        else {
                    342:                                fprintf(stderr, "at: ambiguous month\n");
                    343:                                exit(1);
                    344:                        }
                    345:                }
                    346:        }
                    347:        if (found>=0) {
                    348:                if (argc<=3)
                    349:                        return(2);
                    350:                uday = atoi(argv[3]) - 1;
                    351:                if (uday<0) {
                    352:                        fprintf(stderr, "at: illegal day\n");
                    353:                        exit(1);
                    354:                }
                    355:                if (found > 1 && detail->tm_year%4==0)
                    356:                        uday++; /* because next loop will include Feb */
                    357:                while(--found>=0)
                    358:                        uday += months[found].mlen;
                    359:                /* this is bogus...
                    360:                if (detail->tm_year%4==0 && uday>59)
                    361:                        uday += 1;
                    362:                */
                    363:                if(argc>4) {
                    364:                        int t = atoi(argv[4])-1900;
                    365:                        if(t<100 && (t>uyear || t==uyear && (
                    366:                           uday>today || uday==today &&
                    367:                           utime>now))) {
                    368:                                uyear = t;
                    369:                                return(4);
                    370:                        }
                    371:                }
                    372:                if (uday == today && utime < now)
                    373:                {
                    374:                        fprintf(stderr, "at: Warning: time specified is past; assuming next year....\n");
                    375:                        ++uyear;
                    376:                }
                    377:                return(3);
                    378:        }
                    379:        /* not a month, try day of week */
                    380:        found = -1;
                    381:        for (ps=days; ps<days+7; ps++) {
                    382:                if (prefix(argv[2], *ps)) {
                    383:                        if (found<0)
                    384:                                found = ps-days;
                    385:                        else {
                    386:                                fprintf(stderr, "at: ambiguous day of week\n");
                    387:                                exit(1);
                    388:                        }
                    389:                }
                    390:        }
                    391:        if (found<0)
                    392:                return(1);
                    393:        /* find next day of this sort */
                    394:        uday = found - detail->tm_wday;
                    395:        if (uday<=0)
                    396:                uday += 7;
                    397:        uday += today;
                    398:        /* Bogus...
                    399:        if (detail->tm_year%4==0 && uday>59)
                    400:                uday += 1;
                    401:        */
                    402:        if (argc>3 && strcmp("week", argv[3])==0) {
                    403:                uday += 7;
                    404:                return(3);
                    405:        }
                    406:        return(2);
                    407: }
                    408: 
                    409: char *
                    410: prefix(begin, full)
                    411: char *begin, *full;
                    412: {
                    413:        int c;
                    414:        while (c = *begin++) {
                    415:                if (isupper(c))
                    416:                        c = tolower(c);
                    417:                if (*full != c)
                    418:                        return(0);
                    419:                else
                    420:                        full++;
                    421:        }
                    422:        return(full);
                    423: }
                    424: 
                    425: filename(dir, y, d, t)
                    426: char *dir;
                    427: {
                    428:        register i;
                    429: 
                    430:        for (i=0; ; i += 53) {
                    431:                sprintf(fname, "%s/%.2d.%.3d.%.4d.%.2d", dir, y, d, t,
                    432:                   (getpid()+i)%100);
                    433:                if (access(fname, 0) == -1)
                    434:                        return;
                    435:        }
                    436: }
                    437: 
                    438: onintr()
                    439: {
                    440:        unlink(fname);
                    441:        exit(1);
                    442: }
                    443: 
                    444: /* hand "fn" all path names in spool directory starting with "prefix" */
                    445: int
                    446: readdir(prefix, fn)
                    447:        char *prefix;
                    448:        void (*fn)();
                    449: {
                    450:        struct dir dbuf;
                    451:        register FILE *f;
                    452:        register int count = 0;
                    453: 
                    454:        f = fopen (THISDAY, "r");
                    455:        if (f == NULL) {
                    456:                fprintf (stderr, "cannot open %s\n", THISDAY);
                    457:                return;
                    458:        }
                    459: 
                    460:        while (fread (&dbuf, sizeof dbuf, 1, f) == 1) {
                    461:                if (dbuf.d_ino != 0 && dbuf.d_name[0] != '.'
                    462:                  && strncmp (dbuf.d_name, prefix, strlen(prefix)) == 0) {
                    463:                        char file[sizeof (THISDAY) + DIRSIZ + 1];
                    464:                        struct stat sb;
                    465:                        strcpy (file, THISDAY);
                    466:                        strcat (file, "/");
                    467:                        strncat (file, dbuf.d_name, DIRSIZ);
                    468:                        if (stat (file, &sb) >= 0 && sb.st_uid == getuid()) {
                    469:                                count++;
                    470:                                (*fn) (file);
                    471:                        }
                    472:                }
                    473:        }
                    474: 
                    475:        fclose (f);
                    476: 
                    477:        if (count == 0)
                    478:                fprintf (stderr, "no items\n");
                    479: }
                    480: 
                    481: /*
                    482:  *     return pointer to (static) area containing date and time
                    483:  *     for activation of file, or argument if invalid format.
                    484:  */
                    485: char *
                    486: pathdate (path)
                    487:        char *path;
                    488: {
                    489:        int month, mday, yday, year, tm, junk;
                    490:        register int i;
                    491:        register char *p, *q;
                    492:        static char res[50];
                    493: 
                    494:        /* point p at last component of path name */
                    495:        p = q = path;
                    496:        while (*q)
                    497:                if (*q++ == '/')
                    498:                        p = q;
                    499:        
                    500:        /* break out the fields */
                    501:        if (sscanf(p, "%2d.%3d.%4d.%2d", &year, &yday, &tm, &junk) != 4)
                    502:                return path;
                    503:        
                    504:        /* determine the month and day of month */
                    505:        month = 0;
                    506:        mday = yday + 1;
                    507:        while (months[month].mlen && mday > months[month].mlen) {
                    508:                mday -= months[month].mlen;
                    509:                month++;
                    510:        }
                    511: 
                    512:        /* check for impossible date */
                    513:        if (++month > 12)
                    514:                return path;
                    515: 
                    516:        /* check for leap year */
                    517:        if (month > 2 && year % 4 == 0 && --mday <= 0) {
                    518:                mday = months[--month - 1].mlen;
                    519:                if (month == 2)
                    520:                        mday++;
                    521:        }
                    522: 
                    523:        /* build the result */
                    524:        sprintf (res, "%.4d %s %d 19%d",
                    525:                tm, months[month-1].mname, mday, year);
                    526:        return res;
                    527: }
                    528: 
                    529: void
                    530: prdir (path)
                    531:        register char *path;
                    532: {
                    533:        printf ("%s\n", pathdate (path));
                    534: }
                    535: 
                    536: void
                    537: prfile (path)
                    538:        register char *path;
                    539: {
                    540:        register FILE *f;
                    541: 
                    542:        printf ("\n%s:", pathdate (path));
                    543:        f = fopen (path, "r");
                    544:        if (f == NULL) {
                    545:                fprintf (stderr, " cannot open %s\n", path);
                    546:        } else {
                    547:                register int c;
                    548:                printf ("\n\n");
                    549:                while ((c = getc(f)) != EOF)
                    550:                        putchar (c);
                    551:        }
                    552:        fclose (f);
                    553: }
                    554: 
                    555: void
                    556: delfile(path)
                    557:        register char *path;
                    558: {
                    559:        if (unlink(path) < 0)
                    560:                fprintf (stderr, "cannot remove %s\n", path);
                    561:        else
                    562:                fprintf (stderr, "remove %s\n", pathdate(path));
                    563: }
                    564: char *
                    565: idend(s)
                    566:        char *s;
                    567: {
                    568:        while(*s && *s!='=' && *s!='(')
                    569:                s++;
                    570:        return s;
                    571: }
                    572: /* print a shell variable with quoting */
                    573: void
                    574: vprint (f, str)
                    575:        register FILE *f;
                    576:        register char *str;
                    577: {
                    578:        if(*idend(str)=='(')    /* function; the shell's already quoted it right */
                    579:                fputs(str, f);
                    580:        else
                    581:                while (*str) {
                    582:                        if (strchr (" \t\b\n\f$\\[]*?'\"`&|#()^<>;", *str))
                    583:                                putc ('\\', f);
                    584:                        putc (*str++, f);
                    585:                }
                    586: }

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.