2 * at.c : Put file into atrun queue
3 * Copyright (C) 1993, 1994 Thomas Koenig
5 * Atrun & Atq modifications
6 * Copyright (C) 1993 David Parsons
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. The name of the author(s) may not be used to endorse or promote
14 * products derived from this software without specific prior written
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 * $FreeBSD: src/usr.bin/at/at.c,v 1.30 2007/09/21 01:55:11 kevlo Exp $
29 * $DragonFly: src/usr.bin/at/at.c,v 1.9 2007/09/22 20:22:51 pavalos Exp $
36 #include <sys/param.h>
60 #include "parsetime.h"
69 #define ATJOB_DIR "/usr/spool/atjobs/"
73 #define LFILE ATJOB_DIR ".lockfile"
80 #define ALARMC 10 /* Number of seconds to wait for timeout */
85 enum { ATQ
, ATRM
, AT
, BATCH
, CAT
}; /* what program we want to run */
87 /* File scope variables */
89 const char *no_export
[] =
91 "TERM", "TERMCAP", "DISPLAY", "_"
93 static int send_mail
= 0;
95 /* External variables */
96 uid_t real_uid
, effective_uid
;
97 gid_t real_gid
, effective_gid
;
99 extern char **environ
;
101 char atfile
[sizeof(ATJOB_DIR
) + 14] = ATJOB_DIR
;
103 char *atinput
= NULL
; /* where to get input from */
104 char atqueue
= 0; /* which queue to examine for jobs (atq) */
105 char atverify
= 0; /* verify time instead of queuing job */
108 /* Function declarations */
110 static void sigc(int signo
);
111 static void alarmc(int signo
);
112 static char *cwdname(void);
113 static void writefile(time_t runtimer
, char queue
);
114 static void list_jobs(long *, int);
115 static long nextjob(void);
116 static time_t ttime(const char *arg
);
117 static char * timer2str(time_t runtimer
);
118 static int in_job_list(long, long *, int);
119 static long *get_job_list(int, char *[], int *);
121 /* Signal catching functions */
124 sigc(int signo __unused
)
126 /* If the user presses ^C, remove the spool file and exit
139 alarmc(int signo __unused
)
143 /* Time out after some seconds. */
144 strlcpy(buf
, namep
, sizeof(buf
));
145 strlcat(buf
, ": file locking timed out\n", sizeof(buf
));
146 write(STDERR_FILENO
, buf
, strlen(buf
));
150 /* Local functions */
155 /* Read in the current directory; the name will be overwritten on
158 static char *ptr
= NULL
;
159 static size_t size
= SIZE
;
162 if ((ptr
= malloc(size
)) == NULL
)
163 errx(EXIT_FAILURE
, "virtual memory exhausted");
168 panic("out of memory");
170 if (getcwd(ptr
, size
-1) != NULL
)
174 perr("cannot get directory");
178 if ((ptr
= malloc(size
)) == NULL
)
179 errx(EXIT_FAILURE
, "virtual memory exhausted");
189 if ((fid
= fopen(ATJOB_DIR
".SEQ", "r+")) != NULL
) {
190 if (fscanf(fid
, "%5lx", &jobno
) == 1) {
192 jobno
= (1+jobno
) % 0xfffff; /* 2^20 jobs enough? */
193 fprintf(fid
, "%05lx\n", jobno
);
200 else if ((fid
= fopen(ATJOB_DIR
".SEQ", "w")) != NULL
) {
201 fprintf(fid
, "%05lx\n", jobno
= 1);
209 writefile(time_t runtimer
, char queue
)
211 /* This does most of the work if at or batch are invoked for writing a job.
214 char *ap
, *ppos
, *mailname
;
215 struct passwd
*pass_entry
;
217 int fdes
, lockdes
, fd2
;
219 struct sigaction act
;
225 setlocale(LC_TIME
, "");
227 /* Install the signal handler for SIGINT; terminate after removing the
228 * spool file if necessary
230 act
.sa_handler
= sigc
;
231 sigemptyset(&(act
.sa_mask
));
234 sigaction(SIGINT
, &act
, NULL
);
236 /* Loop over all possible file names for running something at this
237 * particular time, see if a file is there; the first empty slot at any
238 * particular time is used. Lock the file LFILE first to make sure
239 * we're alone when doing this.
244 if ((lockdes
= open(LFILE
, O_WRONLY
| O_CREAT
, S_IWUSR
| S_IRUSR
)) < 0)
245 perr("cannot open lockfile " LFILE
);
247 lock
.l_type
= F_WRLCK
; lock
.l_whence
= SEEK_SET
; lock
.l_start
= 0;
250 act
.sa_handler
= alarmc
;
251 sigemptyset(&(act
.sa_mask
));
254 /* Set an alarm so a timeout occurs after ALARMC seconds, in case
255 * something is seriously broken.
257 sigaction(SIGALRM
, &act
, NULL
);
259 fcntl(lockdes
, F_SETLKW
, &lock
);
262 if ((jobno
= nextjob()) == EOF
)
263 perr("cannot generate job number");
265 ppos
= atfile
+ strlen(atfile
);
266 snprintf(ppos
, sizeof(atfile
) - strlen(atfile
), "%c%5lx%8lx", queue
,
267 jobno
, (unsigned long) (runtimer
/60));
269 for(ap
=ppos
; *ap
!= '\0'; ap
++)
273 if (stat(atfile
, &statbuf
) != 0)
275 perr("cannot access " ATJOB_DIR
);
277 /* Create the file. The x bit is only going to be set after it has
278 * been completely written out, to make sure it is not executed in the
279 * meantime. To make sure they do not get deleted, turn off their r
280 * bit. Yes, this is a kluge.
282 cmask
= umask(S_IRUSR
| S_IWUSR
| S_IXUSR
);
283 if ((fdes
= creat(atfile
, O_WRONLY
)) == -1)
284 perr("cannot create atjob file");
286 if ((fd2
= dup(fdes
)) <0)
287 perr("error in dup() of job file");
289 if(fchown(fd2
, real_uid
, real_gid
) != 0)
290 perr("cannot give away file");
294 /* We no longer need suid root; now we just need to be able to write
295 * to the directory, if necessary.
298 REDUCE_PRIV(DAEMON_UID
, DAEMON_GID
)
300 /* We've successfully created the file; let's set the flag so it
301 * gets removed in case of an interrupt or error.
305 /* Now we can release the lock, so other people can access it
307 lock
.l_type
= F_UNLCK
; lock
.l_whence
= SEEK_SET
; lock
.l_start
= 0;
309 fcntl(lockdes
, F_SETLKW
, &lock
);
312 if((fp
= fdopen(fdes
, "w")) == NULL
)
313 panic("cannot reopen atjob file");
315 /* Get the userid to mail to, first by trying getlogin(),
316 * then from LOGNAME, finally from getpwuid().
318 mailname
= getlogin();
319 if (mailname
== NULL
)
320 mailname
= getenv("LOGNAME");
322 if ((mailname
== NULL
) || (mailname
[0] == '\0')
323 || (strlen(mailname
) >= MAXLOGNAME
) || (getpwnam(mailname
)==NULL
))
325 pass_entry
= getpwuid(real_uid
);
326 if (pass_entry
!= NULL
)
327 mailname
= pass_entry
->pw_name
;
332 fpin
= freopen(atinput
, "r", stdin
);
334 perr("cannot open input file");
336 fprintf(fp
, "#!/bin/sh\n# atrun uid=%ld gid=%ld\n# mail %*s %d\n",
337 (long) real_uid
, (long) real_gid
, MAXLOGNAME
- 1, mailname
,
340 /* Write out the umask at the time of invocation
342 fprintf(fp
, "umask %lo\n", (unsigned long) cmask
);
344 /* Write out the environment. Anything that may look like a
345 * special character to the shell is quoted, except for \n, which is
346 * done with a pair of "'s. Don't export the no_export list (such
347 * as TERM or DISPLAY) because we don't want these.
349 for (atenv
= environ
; *atenv
!= NULL
; atenv
++)
354 eqp
= strchr(*atenv
, '=');
360 for (i
=0; i
<sizeof(no_export
)/sizeof(no_export
[0]); i
++)
363 && (strncmp(*atenv
, no_export
[i
],
364 (size_t) (eqp
-*atenv
)) != 0);
371 fwrite(*atenv
, sizeof(char), eqp
-*atenv
, fp
);
372 for(ap
= eqp
;*ap
!= '\0'; ap
++)
375 fprintf(fp
, "\"\n\"");
380 case '%': case '/': case '{': case '[':
381 case ']': case '=': case '}': case '@':
382 case '+': case '#': case ',': case '.':
383 case ':': case '-': case '_':
393 fputs("; export ", fp
);
394 fwrite(*atenv
, sizeof(char), eqp
-*atenv
-1, fp
);
399 /* Cd to the directory at the time and write out all the
400 * commands the user supplies from stdin.
403 for (ap
= cwdname(); *ap
!= '\0'; ap
++)
406 fprintf(fp
, "\"\n\"");
409 if (*ap
!= '/' && !isalnum(*ap
))
415 /* Test cd's exit status: die if the original directory has been
416 * removed, become unreadable or whatever
418 fprintf(fp
, " || {\n\t echo 'Execution directory "
419 "inaccessible' >&2\n\t exit 1\n}\n");
421 while((ch
= getchar()) != EOF
)
426 panic("output error");
429 panic("input error");
433 /* Set the x bit so that we're ready to start executing
436 if (fchmod(fd2
, S_IRUSR
| S_IWUSR
| S_IXUSR
) < 0)
437 perr("cannot give away file");
442 char *timestr
= timer2str(runtimer
);
444 fprintf(stderr
, "Job %ld will be executed using /bin/sh\n", jobno
);
445 fprintf(stderr
, "Job %ld at %s\n", jobno
, timestr
);
450 in_job_list(long job
, long *joblist
, int len
)
454 for (i
= 0; i
< len
; i
++)
455 if (job
== joblist
[i
])
462 list_jobs(long *joblist
, int len
)
464 /* List all a user's jobs in the queue, by looping through ATJOB_DIR,
465 * or everybody's if we are root
469 struct dirent
*dirent
;
478 setlocale(LC_TIME
, "");
482 if (chdir(ATJOB_DIR
) != 0)
483 perr("cannot change to " ATJOB_DIR
);
485 if ((spool
= opendir(".")) == NULL
)
486 perr("cannot open " ATJOB_DIR
);
488 /* Loop over every file in the directory
490 while((dirent
= readdir(spool
)) != NULL
) {
491 if (stat(dirent
->d_name
, &buf
) != 0)
492 perr("cannot stat in " ATJOB_DIR
);
494 /* See it's a regular file and has its x bit turned on and
497 if (!S_ISREG(buf
.st_mode
)
498 || ((buf
.st_uid
!= real_uid
) && ! (real_uid
== 0))
499 || !(S_IXUSR
& buf
.st_mode
|| atverify
))
502 if(sscanf(dirent
->d_name
, "%c%5lx%8lx", &queue
, &jobno
, &ctm
)!=3)
505 /* If jobs are given, only list those jobs */
506 if (joblist
&& !in_job_list(jobno
, joblist
, len
))
509 if (atqueue
&& (queue
!= atqueue
))
513 runtimer
= 60*(time_t) ctm
;
514 timestr
= timer2str(runtimer
);
516 /* Look up owner's uid in password database file */
517 pw
= getpwuid(buf
.st_uid
);
520 printf("Date\t\t\t\tOwner\t\tQueue\tJob#\n");
523 printf("%s\t%-16s%c%s\t%ld\n",
525 pw
? pw
->pw_name
: "???",
527 (S_IXUSR
& buf
.st_mode
) ? "":"(done)",
537 process_jobs(int argc
, char **argv
, int what
)
539 /* Delete every argument (job - ID) given
544 struct dirent
*dirent
;
551 if (chdir(ATJOB_DIR
) != 0)
552 perr("cannot change to " ATJOB_DIR
);
554 if ((spool
= opendir(".")) == NULL
)
555 perr("cannot open " ATJOB_DIR
);
559 /* Loop over every file in the directory
561 while((dirent
= readdir(spool
)) != NULL
) {
564 if (stat(dirent
->d_name
, &buf
) != 0)
565 perr("cannot stat in " ATJOB_DIR
);
568 if(sscanf(dirent
->d_name
, "%c%5lx%8lx", &queue
, &jobno
, &ctm
)!=3)
571 for (i
=optind
; i
< argc
; i
++) {
572 if (atoi(argv
[i
]) == jobno
) {
573 if ((buf
.st_uid
!= real_uid
) && !(real_uid
== 0))
574 errx(EXIT_FAILURE
, "%s: not owner", argv
[i
]);
580 if (unlink(dirent
->d_name
) != 0)
581 perr(dirent
->d_name
);
594 fp
= fopen(dirent
->d_name
,"r");
599 perr("cannot open file");
601 while((ch
= getc(fp
)) != EOF
) {
609 errx(EXIT_FAILURE
, "internal error, process_jobs = %d",
618 #define ATOI2(ar) ((ar)[0] - '0') * 10 + ((ar)[1] - '0'); (ar) += 2;
621 ttime(const char *arg
)
624 * This is pretty much a copy of stime_arg1() from touch.c. I changed
625 * the return value and the argument list because it's more convenient
626 * (IMO) to do everything in one place. - Joe Halpin
628 struct timeval tv
[2];
634 if (gettimeofday(&tv
[0], NULL
))
635 panic("Cannot get current time");
637 /* Start with the current time. */
639 if ((t
= localtime(&now
)) == NULL
)
641 /* [[CC]YY]MMDDhhmm[.SS] */
642 if ((p
= strchr(arg
, '.')) == NULL
)
643 t
->tm_sec
= 0; /* Seconds defaults to 0. */
645 if (strlen(p
+ 1) != 2)
648 t
->tm_sec
= ATOI2(p
);
652 switch(strlen(arg
)) {
653 case 12: /* CCYYMMDDhhmm */
654 t
->tm_year
= ATOI2(arg
);
658 case 10: /* YYMMDDhhmm */
660 yearset
= ATOI2(arg
);
661 t
->tm_year
+= yearset
;
663 yearset
= ATOI2(arg
);
664 t
->tm_year
= yearset
+ 2000;
666 t
->tm_year
-= 1900; /* Convert to UNIX time. */
668 case 8: /* MMDDhhmm */
669 t
->tm_mon
= ATOI2(arg
);
670 --t
->tm_mon
; /* Convert from 01-12 to 00-11 */
671 t
->tm_mday
= ATOI2(arg
);
672 t
->tm_hour
= ATOI2(arg
);
673 t
->tm_min
= ATOI2(arg
);
679 t
->tm_isdst
= -1; /* Figure out DST. */
680 tv
[0].tv_sec
= tv
[1].tv_sec
= mktime(t
);
681 if (tv
[0].tv_sec
!= -1)
686 "out of range or illegal time specification: [[CC]YY]MMDDhhmm[.SS]");
689 /* The caller must free up the memory. */
691 timer2str(time_t runtimer
)
697 timestr
= malloc(TIMESIZE
);
699 panic("out of memory");
701 runtime
= *localtime(&runtimer
);
702 rv
= strftime(timestr
, TIMESIZE
, nl_langinfo(D_T_FMT
), &runtime
);
704 panic("TIMESIZE too low");
710 get_job_list(int argc
, char *argv
[], int *joblen
)
719 if ((joblist
= malloc(len
* sizeof(*joblist
))) == NULL
)
720 panic("out of memory");
722 for (i
= 0; i
< argc
; i
++) {
724 if ((joblist
[i
] = strtol(argv
[i
], &ep
, 10)) < 0 ||
725 ep
== argv
[i
] || *ep
!= '\0' || errno
)
726 panic("invalid job number");
735 main(int argc
, char **argv
)
738 char queue
= DEFAULT_AT_QUEUE
;
742 int program
= AT
; /* our default program */
743 const char *options
= "q:f:t:rmvldbc"; /* default options for at */
753 /* Eat any leading paths
755 if ((pgm
= strrchr(argv
[0], '/')) == NULL
)
762 /* find out what this program is supposed to do
764 if (strcmp(pgm
, "atq") == 0) {
768 else if (strcmp(pgm
, "atrm") == 0) {
772 else if (strcmp(pgm
, "batch") == 0) {
777 /* process whatever options we can process
780 while ((c
=getopt(argc
, argv
, options
)) != -1)
782 case 'v': /* verify time settings */
786 case 'm': /* send mail when job is complete */
794 case 'q': /* specify queue */
795 if (strlen(optarg
) > 1)
798 atqueue
= queue
= *optarg
;
799 if (!(islower(queue
)||isupper(queue
)))
806 warnx("-d is deprecated; use -r instead");
807 /* fall through to 'r' */
820 timer
= ttime(optarg
);
848 /* end of options eating
851 /* select our program
853 if(!check_permission())
854 errx(EXIT_FAILURE
, "you do not have permission to use this program");
858 REDUCE_PRIV(DAEMON_UID
, DAEMON_GID
)
861 joblist
= get_job_list(argc
- optind
, argv
+ optind
, &joblen
);
862 list_jobs(joblist
, joblen
);
867 REDUCE_PRIV(DAEMON_UID
, DAEMON_GID
)
869 process_jobs(argc
, argv
, ATRM
);
874 process_jobs(argc
, argv
, CAT
);
879 * If timer is > -1, then the user gave the time with -t. In that
880 * case, it's already been set. If not, set it now.
883 timer
= parsetime(argc
, argv
);
887 struct tm
*tm
= localtime(&timer
);
888 fprintf(stderr
, "%s\n", asctime(tm
));
890 writefile(timer
, queue
);
895 queue
= toupper(queue
);
897 queue
= DEFAULT_BATCH_QUEUE
;
900 timer
= parsetime(argc
, argv
);
906 struct tm
*tm
= localtime(&timer
);
907 fprintf(stderr
, "%s\n", asctime(tm
));
910 writefile(timer
, queue
);
914 panic("internal error");