2.9BSD/usr/src/cmd/at.c
/*
* at time mon day
* at time wday
* at time wday 'week'
*
*/
/*! Modified by PLWard, USGS 10/30/80 for csh and mail options */
#include <stdio.h>
#include <ctype.h>
#include <time.h>
#include <signal.h>
#include <pwd.h>
#define MAIL "Mail"
#define CD "cd"
#define RM "/bin/rm"
#define PWD "/bin/pwd"
#define PAST "/usr/spool/at/past"
#define THISDAY "/usr/spool/at"
#define HOUR 100
#define HALFDAY (12*HOUR)
#define DAY (24*HOUR)
int sendmail = 1; /* default to send mail = 1, else = 0. */
char *days[] = {
"sunday",
"monday",
"tuesday",
"wednesday",
"thursday",
"friday",
"saturday",
};
struct monstr {
char *mname;
int mlen;
} months[] = {
{ "january", 31 },
{ "february", 28 },
{ "march", 31 },
{ "april", 30 },
{ "may", 31 },
{ "june", 30 },
{ "july", 31 },
{ "august", 31 },
{ "september", 30 },
{ "october", 31 },
{ "november", 30 },
{ "december", 31 },
{ 0, 0 },
};
char fname[100];
char fileout[100];
char temp[15];
int utime; /* requested time in grains */
int now; /* when is it */
int uday; /* day of year to be done */
int uyear; /* year */
int today; /* day of year today */
FILE *file;
FILE *ifile;
char **environ;
char *prefix();
FILE *popen();
struct passwd *user;
main(argc, argv)
char **argv;
{
extern onintr();
register c;
char pwbuf[100];
FILE *pwfil;
int larg;
int csh;
int i,eq;
char *tmp;
char **ep;
char *cp;
if(argv[1][0] == '-'){
sendmail=sendmail ? 0 : 1;
for(i=2;i<=argc;i++)argv[i-1]=argv[i];
argc--;
}
/* argv[1] is the user's time: e.g., 3AM */
/* argv[2] is a month name or day of week */
/* argv[3] is day of month or 'week' */
/* another argument might be an input file */
if (argc < 2) {
fprintf(stderr, "at: arg count\n");
exit(1);
}
makeutime(argv[1]);
larg = makeuday(argc,argv)+1;
if (uday==today && larg<=2 && utime<=now)
uday++;
c = uyear%4==0? 366: 365;
if (uday >= c) {
uday -= c;
uyear++;
}
filename(THISDAY, uyear, uday, utime);
/* Create file and then change uids */
close(creat(fname,0755));
chown(fname,getuid(),getgid());
/* fix so that root can use at */
setuid(getuid());
user=getpwuid(getuid());
ifile = stdin;
if (argc > larg)
ifile = fopen(argv[larg], "r");
else argv[larg]="from stdin";
if (ifile == NULL) {
fprintf(stderr, "at: cannot open input: %s\n", argv[larg]);
exit(1);
}
if (signal(SIGINT, SIG_IGN) != SIG_IGN)
signal(SIGINT, onintr);
file = fopen(fname, "w");
if (file == NULL) {
fprintf(stderr, "at: cannot open file in %s.\n",THISDAY);
exit(1);
}
/* Decide if this is a cshell file */
if(ifile == stdin){
tmp=getenv("SHELL");
csh = (strcmp(tmp+strlen(tmp)-3,"csh")==0) ? 1 : 0;
}
else {
csh = ((c=getc(ifile)) == '#') ? 1 : 0;
fseek(ifile,0,0);
}
if (csh) fprintf(file,"#at:cshell\n");
if ((pwfil = popen(PWD, "r")) == NULL) {
fprintf(stderr, "at: can't execute pwd\n");
exit(1);
}
fgets(pwbuf, 100, pwfil);
pclose(pwfil);
fprintf(file, "%s %s",CD, pwbuf);
c=umask(0);
umask(c);
fprintf(file,"umask %.1o\n",c);
if (environ) {
ep = environ;
while (*ep){
for (tmp = *ep, cp = "TERMCAP"; *tmp==*cp; tmp++,cp++);
if (*cp == 0 && *tmp== '=') {
ep++;
continue;
}
if (csh) fprintf(file,"setenv ");
for(tmp = *ep ; *tmp != '=' ; tmp++) putc(*tmp,file);
if (csh) putc(' ', file);
else putc(*tmp,file);
putc('\'', file);
for (tmp++; *tmp; tmp++) {
if (*tmp == '\'')
putc('\\', file);
putc(*tmp, file);
}
putc('\'', file);
putc('\n',file);
ep++;
}
if (csh) fprintf(file,"source %s/.cshrc\n",getenv("HOME"));
}
fprintf(file,"echo 'Began executing at' `date`\n");
while((c = getc(ifile)) != EOF) {
putc(c, file);
}
fprintf(file,"%s %s\n",CD,PAST);
fprintf(file,"echo 'Done executing at' `date`\n");
if(sendmail)fprintf(file,"%s -s 'at: ran program %s.' %s < %s\n",
MAIL,argv[larg],user->pw_name,fileout);
fprintf(file,"%s %s\n",RM,fileout);
exit(0);
}
makeutime(pp)
char *pp;
{
register val;
register char *p;
/* p points to a user time */
p = pp;
val = 0;
while(isdigit(*p)) {
val = val*10+(*p++ -'0');
}
if (p-pp < 3)
val *= HOUR;
for (;;) {
switch(*p) {
case ':':
++p;
if (isdigit(*p)) {
if (isdigit(p[1])) {
val +=(10* *p + p[1] - 11*'0');
p += 2;
continue;
}
}
fprintf(stderr, "at: bad time format:\n");
exit(1);
case 'A':
case 'a':
if (val >= HALFDAY+HOUR)
val = DAY+1; /* illegal */
if (val >= HALFDAY && val <(HALFDAY+HOUR))
val -= HALFDAY;
break;
case 'P':
case 'p':
if (val >= HALFDAY+HOUR)
val = DAY+1; /* illegal */
if (val < HALFDAY)
val += HALFDAY;
break;
case 'n':
case 'N':
val = HALFDAY;
break;
case 'M':
case 'm':
val = 0;
break;
case '\0':
case ' ':
/* 24 hour time */
if (val == DAY)
val -= DAY;
break;
default:
fprintf(stderr, "at: bad time format\n");
exit(1);
}
break;
}
if (val < 0 || val >= DAY) {
fprintf(stderr, "at: time out of range\n");
exit(1);
}
if (val%HOUR >= 60) {
fprintf(stderr, "at: illegal minute field\n");
exit(1);
}
utime = val;
}
makeuday(argc,argv)
char **argv;
{
/* the presumption is that argv[2], argv[3] are either
month day OR weekday [week]. Returns either 2 or 3 as last
argument used */
/* first of all, what's today */
long tm;
int found = -1;
char **ps;
struct tm *detail, *localtime();
struct monstr *pt;
time(&tm);
detail = localtime(&tm);
uday = today = detail->tm_yday;
uyear = detail->tm_year;
now = detail->tm_hour*100+detail->tm_min;
if (argc<=2)
return(1);
/* is the next argument a month name ? */
for (pt=months; pt->mname; pt++) {
if (prefix(argv[2], pt->mname)) {
if (found<0)
found = pt-months;
else {
fprintf(stderr, "at: ambiguous month\n");
exit(1);
}
}
}
if (found>=0) {
if (argc<=3)
return(2);
uday = atoi(argv[3]) - 1;
if (uday<0) {
fprintf(stderr, "at: illegal day\n");
exit(1);
}
while(--found>=0)
uday += months[found].mlen;
if (detail->tm_year%4==0 && uday>59)
uday += 1;
return(3);
}
/* not a month, try day of week */
found = -1;
for (ps=days; ps<days+7; ps++) {
if (prefix(argv[2], *ps)) {
if (found<0)
found = ps-days;
else {
fprintf(stderr, "at: ambiguous day of week\n");
exit(1);
}
}
}
if (found<0)
return(1);
/* find next day of this sort */
uday = found - detail->tm_wday;
if (uday<=0)
uday += 7;
uday += today;
if (argc>3 && strcmp("week", argv[3])==0) {
uday += 7;
return(3);
}
return(2);
}
char *
prefix(begin, full)
char *begin, *full;
{
int c;
while (c = *begin++) {
if (isupper(c))
c = tolower(c);
if (*full != c)
return(0);
else
full++;
}
return(full);
}
filename(dir, y, d, t)
char *dir;
{
register i;
for (i=0; ; i += 53) {
sprintf(temp,"%02d.%03d.%04d.%02d",y,d,t,(getpid()+i)%100);
sprintf(fname,"%s/%s",dir,temp);
temp[0]='M';
sprintf(fileout,"%s/%s",PAST,temp);
if (access(fname, 0) == -1)
return;
}
}
onintr()
{
unlink(fname);
exit(1);
}