1d78e98d2SNate Williams /* 2d78e98d2SNate Williams * at.c : Put file into atrun queue 3b89321a5SAndrey A. Chernov * Copyright (C) 1993, 1994 Thomas Koenig 4d78e98d2SNate Williams * 5d78e98d2SNate Williams * Atrun & Atq modifications 6d78e98d2SNate Williams * Copyright (C) 1993 David Parsons 7d78e98d2SNate Williams * 8d78e98d2SNate Williams * Redistribution and use in source and binary forms, with or without 9d78e98d2SNate Williams * modification, are permitted provided that the following conditions 10d78e98d2SNate Williams * are met: 11d78e98d2SNate Williams * 1. Redistributions of source code must retain the above copyright 12d78e98d2SNate Williams * notice, this list of conditions and the following disclaimer. 13d78e98d2SNate Williams * 2. The name of the author(s) may not be used to endorse or promote 14d78e98d2SNate Williams * products derived from this software without specific prior written 15d78e98d2SNate Williams * permission. 16d78e98d2SNate Williams * 17d78e98d2SNate Williams * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 18d78e98d2SNate Williams * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19d78e98d2SNate Williams * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20ddcf8022SAndrey A. Chernov * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 21d78e98d2SNate Williams * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22d78e98d2SNate Williams * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23d78e98d2SNate Williams * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24d78e98d2SNate Williams * THEORY OF LIABILITY, WETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25d78e98d2SNate Williams * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26d78e98d2SNate Williams * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27d78e98d2SNate Williams */ 28d78e98d2SNate Williams 29d78e98d2SNate Williams #define _USE_BSD 1 30d78e98d2SNate Williams 31d78e98d2SNate Williams /* System Headers */ 32b89321a5SAndrey A. Chernov 33d78e98d2SNate Williams #include <sys/types.h> 34d78e98d2SNate Williams #include <sys/stat.h> 35d78e98d2SNate Williams #include <sys/wait.h> 36a88368cfSDavid Nugent #include <sys/param.h> 37d78e98d2SNate Williams #include <ctype.h> 38d78e98d2SNate Williams #include <dirent.h> 39d78e98d2SNate Williams #include <errno.h> 40d78e98d2SNate Williams #include <fcntl.h> 41d78e98d2SNate Williams #include <pwd.h> 42d78e98d2SNate Williams #include <signal.h> 43d78e98d2SNate Williams #include <stddef.h> 44d78e98d2SNate Williams #include <stdio.h> 45d78e98d2SNate Williams #include <stdlib.h> 46d78e98d2SNate Williams #include <string.h> 47d78e98d2SNate Williams #include <time.h> 48d78e98d2SNate Williams #include <unistd.h> 49a88368cfSDavid Nugent #include <utmp.h> 50b89321a5SAndrey A. Chernov #ifndef __FreeBSD__ 51b89321a5SAndrey A. Chernov #include <getopt.h> 52238d3ffdSAndrey A. Chernov #else 53238d3ffdSAndrey A. Chernov #include <locale.h> 54b89321a5SAndrey A. Chernov #endif 55d78e98d2SNate Williams 5608a77c42SAndrey A. Chernov #if (MAXLOGNAME-1) > UT_NAMESIZE 57a88368cfSDavid Nugent #define LOGNAMESIZE UT_NAMESIZE 58a88368cfSDavid Nugent #else 5908a77c42SAndrey A. Chernov #define LOGNAMESIZE (MAXLOGNAME-1) 60a88368cfSDavid Nugent #endif 61a88368cfSDavid Nugent 62d78e98d2SNate Williams /* Local headers */ 63b89321a5SAndrey A. Chernov 64d78e98d2SNate Williams #include "at.h" 65d78e98d2SNate Williams #include "panic.h" 66d78e98d2SNate Williams #include "parsetime.h" 67b89321a5SAndrey A. Chernov #include "perm.h" 68b89321a5SAndrey A. Chernov 69d78e98d2SNate Williams #define MAIN 70d78e98d2SNate Williams #include "privs.h" 71d78e98d2SNate Williams 72d78e98d2SNate Williams /* Macros */ 73b89321a5SAndrey A. Chernov 74b89321a5SAndrey A. Chernov #ifndef ATJOB_DIR 75b89321a5SAndrey A. Chernov #define ATJOB_DIR "/usr/spool/atjobs/" 76b89321a5SAndrey A. Chernov #endif 77b89321a5SAndrey A. Chernov 78b89321a5SAndrey A. Chernov #ifndef LFILE 79b89321a5SAndrey A. Chernov #define LFILE ATJOB_DIR ".lockfile" 80b89321a5SAndrey A. Chernov #endif 81b89321a5SAndrey A. Chernov 82b89321a5SAndrey A. Chernov #ifndef ATJOB_MX 83b89321a5SAndrey A. Chernov #define ATJOB_MX 255 84b89321a5SAndrey A. Chernov #endif 85b89321a5SAndrey A. Chernov 86d78e98d2SNate Williams #define ALARMC 10 /* Number of seconds to wait for timeout */ 87d78e98d2SNate Williams 88d78e98d2SNate Williams #define SIZE 255 89d78e98d2SNate Williams #define TIMESIZE 50 90d78e98d2SNate Williams 91ddcf8022SAndrey A. Chernov enum { ATQ, ATRM, AT, BATCH, CAT }; /* what program we want to run */ 92ddcf8022SAndrey A. Chernov 93d78e98d2SNate Williams /* File scope variables */ 94b89321a5SAndrey A. Chernov 9596846ff6SWarner Losh static const char rcsid[] = "$Id: at.c,v 1.15 1998/12/06 07:42:09 archie Exp $"; 96d78e98d2SNate Williams char *no_export[] = 97d78e98d2SNate Williams { 98d78e98d2SNate Williams "TERM", "TERMCAP", "DISPLAY", "_" 99d78e98d2SNate Williams } ; 10096846ff6SWarner Losh static int send_mail = 0; 101d78e98d2SNate Williams 102d78e98d2SNate Williams /* External variables */ 103b89321a5SAndrey A. Chernov 104d78e98d2SNate Williams extern char **environ; 105d78e98d2SNate Williams int fcreated; 106d78e98d2SNate Williams char *namep; 107b89321a5SAndrey A. Chernov char atfile[] = ATJOB_DIR "12345678901234"; 108d78e98d2SNate Williams 109d78e98d2SNate Williams char *atinput = (char*)0; /* where to get input from */ 110d78e98d2SNate Williams char atqueue = 0; /* which queue to examine for jobs (atq) */ 111d78e98d2SNate Williams char atverify = 0; /* verify time instead of queuing job */ 112d78e98d2SNate Williams 113d78e98d2SNate Williams /* Function declarations */ 114b89321a5SAndrey A. Chernov 115b89321a5SAndrey A. Chernov static void sigc(int signo); 116b89321a5SAndrey A. Chernov static void alarmc(int signo); 117b89321a5SAndrey A. Chernov static char *cwdname(void); 118b89321a5SAndrey A. Chernov static void writefile(time_t runtimer, char queue); 119b89321a5SAndrey A. Chernov static void list_jobs(void); 120d78e98d2SNate Williams 121d78e98d2SNate Williams /* Signal catching functions */ 122d78e98d2SNate Williams 123b89321a5SAndrey A. Chernov static void sigc(int signo) 124d78e98d2SNate Williams { 125d78e98d2SNate Williams /* If the user presses ^C, remove the spool file and exit 126d78e98d2SNate Williams */ 127b89321a5SAndrey A. Chernov if (fcreated) 128b89321a5SAndrey A. Chernov { 129d78e98d2SNate Williams PRIV_START 130d78e98d2SNate Williams unlink(atfile); 131d78e98d2SNate Williams PRIV_END 132d78e98d2SNate Williams } 133d78e98d2SNate Williams 134d78e98d2SNate Williams exit(EXIT_FAILURE); 135d78e98d2SNate Williams } 136d78e98d2SNate Williams 137b89321a5SAndrey A. Chernov static void alarmc(int signo) 138d78e98d2SNate Williams { 139d78e98d2SNate Williams /* Time out after some seconds 140d78e98d2SNate Williams */ 141d78e98d2SNate Williams panic("File locking timed out"); 142d78e98d2SNate Williams } 143d78e98d2SNate Williams 144d78e98d2SNate Williams /* Local functions */ 145d78e98d2SNate Williams 146b89321a5SAndrey A. Chernov static char *cwdname(void) 147d78e98d2SNate Williams { 148d78e98d2SNate Williams /* Read in the current directory; the name will be overwritten on 149d78e98d2SNate Williams * subsequent calls. 150d78e98d2SNate Williams */ 151d78e98d2SNate Williams static char *ptr = NULL; 152d78e98d2SNate Williams static size_t size = SIZE; 153d78e98d2SNate Williams 154d78e98d2SNate Williams if (ptr == NULL) 155b89321a5SAndrey A. Chernov ptr = (char *) mymalloc(size); 156d78e98d2SNate Williams 157b89321a5SAndrey A. Chernov while (1) 158b89321a5SAndrey A. Chernov { 159d78e98d2SNate Williams if (ptr == NULL) 160d78e98d2SNate Williams panic("Out of memory"); 161d78e98d2SNate Williams 162d78e98d2SNate Williams if (getcwd(ptr, size-1) != NULL) 163d78e98d2SNate Williams return ptr; 164d78e98d2SNate Williams 165d78e98d2SNate Williams if (errno != ERANGE) 166d78e98d2SNate Williams perr("Cannot get directory"); 167d78e98d2SNate Williams 168d78e98d2SNate Williams free (ptr); 169d78e98d2SNate Williams size += SIZE; 170b89321a5SAndrey A. Chernov ptr = (char *) mymalloc(size); 171d78e98d2SNate Williams } 172d78e98d2SNate Williams } 173d78e98d2SNate Williams 174ddcf8022SAndrey A. Chernov static long 175ddcf8022SAndrey A. Chernov nextjob() 176ddcf8022SAndrey A. Chernov { 177ddcf8022SAndrey A. Chernov long jobno; 178ddcf8022SAndrey A. Chernov FILE *fid; 179ddcf8022SAndrey A. Chernov 180ddcf8022SAndrey A. Chernov if ((fid = fopen(ATJOB_DIR ".SEQ", "r+")) != (FILE*)0) { 181ddcf8022SAndrey A. Chernov if (fscanf(fid, "%5lx", &jobno) == 1) { 182ddcf8022SAndrey A. Chernov rewind(fid); 183ddcf8022SAndrey A. Chernov jobno = (1+jobno) % 0xfffff; /* 2^20 jobs enough? */ 184ddcf8022SAndrey A. Chernov fprintf(fid, "%05lx\n", jobno); 185ddcf8022SAndrey A. Chernov } 186ddcf8022SAndrey A. Chernov else 187ddcf8022SAndrey A. Chernov jobno = EOF; 188ddcf8022SAndrey A. Chernov fclose(fid); 189ddcf8022SAndrey A. Chernov return jobno; 190ddcf8022SAndrey A. Chernov } 191ddcf8022SAndrey A. Chernov else if ((fid = fopen(ATJOB_DIR ".SEQ", "w")) != (FILE*)0) { 192ddcf8022SAndrey A. Chernov fprintf(fid, "%05lx\n", jobno = 1); 193ddcf8022SAndrey A. Chernov fclose(fid); 194ddcf8022SAndrey A. Chernov return 1; 195ddcf8022SAndrey A. Chernov } 196ddcf8022SAndrey A. Chernov return EOF; 197ddcf8022SAndrey A. Chernov } 198ddcf8022SAndrey A. Chernov 199d78e98d2SNate Williams static void 200b89321a5SAndrey A. Chernov writefile(time_t runtimer, char queue) 201d78e98d2SNate Williams { 202b89321a5SAndrey A. Chernov /* This does most of the work if at or batch are invoked for writing a job. 203d78e98d2SNate Williams */ 204ddcf8022SAndrey A. Chernov long jobno; 205d78e98d2SNate Williams char *ap, *ppos, *mailname; 206d78e98d2SNate Williams struct passwd *pass_entry; 207d78e98d2SNate Williams struct stat statbuf; 208d78e98d2SNate Williams int fdes, lockdes, fd2; 209d78e98d2SNate Williams FILE *fp, *fpin; 210d78e98d2SNate Williams struct sigaction act; 211d78e98d2SNate Williams char **atenv; 212d78e98d2SNate Williams int ch; 213d78e98d2SNate Williams mode_t cmask; 214d78e98d2SNate Williams struct flock lock; 215d78e98d2SNate Williams 216238d3ffdSAndrey A. Chernov #ifdef __FreeBSD__ 217238d3ffdSAndrey A. Chernov (void) setlocale(LC_TIME, ""); 218238d3ffdSAndrey A. Chernov #endif 219238d3ffdSAndrey A. Chernov 220b89321a5SAndrey A. Chernov /* Install the signal handler for SIGINT; terminate after removing the 221d78e98d2SNate Williams * spool file if necessary 222d78e98d2SNate Williams */ 223d78e98d2SNate Williams act.sa_handler = sigc; 224d78e98d2SNate Williams sigemptyset(&(act.sa_mask)); 225d78e98d2SNate Williams act.sa_flags = 0; 226d78e98d2SNate Williams 227d78e98d2SNate Williams sigaction(SIGINT, &act, NULL); 228d78e98d2SNate Williams 229b89321a5SAndrey A. Chernov ppos = atfile + strlen(ATJOB_DIR); 230d78e98d2SNate Williams 231b89321a5SAndrey A. Chernov /* Loop over all possible file names for running something at this 232b89321a5SAndrey A. Chernov * particular time, see if a file is there; the first empty slot at any 233b89321a5SAndrey A. Chernov * particular time is used. Lock the file LFILE first to make sure 234b89321a5SAndrey A. Chernov * we're alone when doing this. 235d78e98d2SNate Williams */ 236d78e98d2SNate Williams 237d78e98d2SNate Williams PRIV_START 238d78e98d2SNate Williams 239b89321a5SAndrey A. Chernov if ((lockdes = open(LFILE, O_WRONLY | O_CREAT, S_IWUSR | S_IRUSR)) < 0) 240b89321a5SAndrey A. Chernov perr("Cannot open lockfile " LFILE); 241d78e98d2SNate Williams 242b89321a5SAndrey A. Chernov lock.l_type = F_WRLCK; lock.l_whence = SEEK_SET; lock.l_start = 0; 243d78e98d2SNate Williams lock.l_len = 0; 244d78e98d2SNate Williams 245d78e98d2SNate Williams act.sa_handler = alarmc; 246d78e98d2SNate Williams sigemptyset(&(act.sa_mask)); 247d78e98d2SNate Williams act.sa_flags = 0; 248d78e98d2SNate Williams 249b89321a5SAndrey A. Chernov /* Set an alarm so a timeout occurs after ALARMC seconds, in case 250d78e98d2SNate Williams * something is seriously broken. 251d78e98d2SNate Williams */ 252d78e98d2SNate Williams sigaction(SIGALRM, &act, NULL); 253d78e98d2SNate Williams alarm(ALARMC); 254d78e98d2SNate Williams fcntl(lockdes, F_SETLKW, &lock); 255d78e98d2SNate Williams alarm(0); 256d78e98d2SNate Williams 257ddcf8022SAndrey A. Chernov if ((jobno = nextjob()) == EOF) 258ddcf8022SAndrey A. Chernov perr("Cannot generate job number"); 259ddcf8022SAndrey A. Chernov 260ddcf8022SAndrey A. Chernov sprintf(ppos, "%c%5lx%8lx", queue, 261ddcf8022SAndrey A. Chernov jobno, (unsigned long) (runtimer/60)); 262ddcf8022SAndrey A. Chernov 263d78e98d2SNate Williams for(ap=ppos; *ap != '\0'; ap ++) 264d78e98d2SNate Williams if (*ap == ' ') 265d78e98d2SNate Williams *ap = '0'; 266d78e98d2SNate Williams 267b89321a5SAndrey A. Chernov if (stat(atfile, &statbuf) != 0) 268ddcf8022SAndrey A. Chernov if (errno != ENOENT) 269b89321a5SAndrey A. Chernov perr("Cannot access " ATJOB_DIR); 270d78e98d2SNate Williams 271b89321a5SAndrey A. Chernov /* Create the file. The x bit is only going to be set after it has 272b89321a5SAndrey A. Chernov * been completely written out, to make sure it is not executed in the 273b89321a5SAndrey A. Chernov * meantime. To make sure they do not get deleted, turn off their r 274b89321a5SAndrey A. Chernov * bit. Yes, this is a kluge. 275d78e98d2SNate Williams */ 276d78e98d2SNate Williams cmask = umask(S_IRUSR | S_IWUSR | S_IXUSR); 277d78e98d2SNate Williams if ((fdes = creat(atfile, O_WRONLY)) == -1) 278d78e98d2SNate Williams perr("Cannot create atjob file"); 279d78e98d2SNate Williams 280d78e98d2SNate Williams if ((fd2 = dup(fdes)) <0) 281d78e98d2SNate Williams perr("Error in dup() of job file"); 282d78e98d2SNate Williams 283b89321a5SAndrey A. Chernov if(fchown(fd2, real_uid, real_gid) != 0) 284d78e98d2SNate Williams perr("Cannot give away file"); 285d78e98d2SNate Williams 286d78e98d2SNate Williams PRIV_END 287d78e98d2SNate Williams 288b5c3f5e7SAndrey A. Chernov /* We no longer need suid root; now we just need to be able to write 289b5c3f5e7SAndrey A. Chernov * to the directory, if necessary. 290b5c3f5e7SAndrey A. Chernov */ 291b5c3f5e7SAndrey A. Chernov 292b5c3f5e7SAndrey A. Chernov REDUCE_PRIV(DAEMON_UID, DAEMON_GID) 293b5c3f5e7SAndrey A. Chernov 294b89321a5SAndrey A. Chernov /* We've successfully created the file; let's set the flag so it 295d78e98d2SNate Williams * gets removed in case of an interrupt or error. 296d78e98d2SNate Williams */ 297d78e98d2SNate Williams fcreated = 1; 298d78e98d2SNate Williams 299b89321a5SAndrey A. Chernov /* Now we can release the lock, so other people can access it 300b89321a5SAndrey A. Chernov */ 301b89321a5SAndrey A. Chernov lock.l_type = F_UNLCK; lock.l_whence = SEEK_SET; lock.l_start = 0; 302d78e98d2SNate Williams lock.l_len = 0; 303d78e98d2SNate Williams fcntl(lockdes, F_SETLKW, &lock); 304d78e98d2SNate Williams close(lockdes); 305d78e98d2SNate Williams 306d78e98d2SNate Williams if((fp = fdopen(fdes, "w")) == NULL) 307d78e98d2SNate Williams panic("Cannot reopen atjob file"); 308d78e98d2SNate Williams 309b89321a5SAndrey A. Chernov /* Get the userid to mail to, first by trying getlogin(), which reads 310b89321a5SAndrey A. Chernov * /etc/utmp, then from LOGNAME, finally from getpwuid(). 311d78e98d2SNate Williams */ 312d78e98d2SNate Williams mailname = getlogin(); 313d78e98d2SNate Williams if (mailname == NULL) 314d78e98d2SNate Williams mailname = getenv("LOGNAME"); 315d78e98d2SNate Williams 316d78e98d2SNate Williams if ((mailname == NULL) || (mailname[0] == '\0') 317a88368cfSDavid Nugent || (strlen(mailname) > LOGNAMESIZE) || (getpwnam(mailname)==NULL)) 318b89321a5SAndrey A. Chernov { 319ddcf8022SAndrey A. Chernov pass_entry = getpwuid(real_uid); 320d78e98d2SNate Williams if (pass_entry != NULL) 321d78e98d2SNate Williams mailname = pass_entry->pw_name; 322d78e98d2SNate Williams } 323d78e98d2SNate Williams 324b89321a5SAndrey A. Chernov if (atinput != (char *) NULL) 325b89321a5SAndrey A. Chernov { 326d78e98d2SNate Williams fpin = freopen(atinput, "r", stdin); 327d78e98d2SNate Williams if (fpin == NULL) 328d78e98d2SNate Williams perr("Cannot open input file"); 329d78e98d2SNate Williams } 330a88368cfSDavid Nugent fprintf(fp, "#!/bin/sh\n# atrun uid=%ld gid=%ld\n# mail %*s %d\n", 331a88368cfSDavid Nugent (long) real_uid, (long) real_gid, LOGNAMESIZE, mailname, send_mail); 332d78e98d2SNate Williams 333b89321a5SAndrey A. Chernov /* Write out the umask at the time of invocation 334b89321a5SAndrey A. Chernov */ 335d78e98d2SNate Williams fprintf(fp, "umask %lo\n", (unsigned long) cmask); 336d78e98d2SNate Williams 337b89321a5SAndrey A. Chernov /* Write out the environment. Anything that may look like a 338b89321a5SAndrey A. Chernov * special character to the shell is quoted, except for \n, which is 339b89321a5SAndrey A. Chernov * done with a pair of "'s. Dont't export the no_export list (such 340b89321a5SAndrey A. Chernov * as TERM or DISPLAY) because we don't want these. 341d78e98d2SNate Williams */ 342b89321a5SAndrey A. Chernov for (atenv= environ; *atenv != NULL; atenv++) 343b89321a5SAndrey A. Chernov { 344d78e98d2SNate Williams int export = 1; 345d78e98d2SNate Williams char *eqp; 346d78e98d2SNate Williams 347d78e98d2SNate Williams eqp = strchr(*atenv, '='); 348d78e98d2SNate Williams if (ap == NULL) 349d78e98d2SNate Williams eqp = *atenv; 350b89321a5SAndrey A. Chernov else 351b89321a5SAndrey A. Chernov { 352d78e98d2SNate Williams int i; 353b89321a5SAndrey A. Chernov for (i=0; i<sizeof(no_export)/sizeof(no_export[0]); i++) 354b89321a5SAndrey A. Chernov { 355d78e98d2SNate Williams export = export 356d78e98d2SNate Williams && (strncmp(*atenv, no_export[i], 357d78e98d2SNate Williams (size_t) (eqp-*atenv)) != 0); 358d78e98d2SNate Williams } 359d78e98d2SNate Williams eqp++; 360d78e98d2SNate Williams } 361d78e98d2SNate Williams 362b89321a5SAndrey A. Chernov if (export) 363b89321a5SAndrey A. Chernov { 364d78e98d2SNate Williams fwrite(*atenv, sizeof(char), eqp-*atenv, fp); 365b89321a5SAndrey A. Chernov for(ap = eqp;*ap != '\0'; ap++) 366b89321a5SAndrey A. Chernov { 367d78e98d2SNate Williams if (*ap == '\n') 368d78e98d2SNate Williams fprintf(fp, "\"\n\""); 369b89321a5SAndrey A. Chernov else 370b89321a5SAndrey A. Chernov { 371ddcf8022SAndrey A. Chernov if (!isalnum(*ap)) { 372ddcf8022SAndrey A. Chernov switch (*ap) { 373ddcf8022SAndrey A. Chernov case '%': case '/': case '{': case '[': 374ddcf8022SAndrey A. Chernov case ']': case '=': case '}': case '@': 375ddcf8022SAndrey A. Chernov case '+': case '#': case ',': case '.': 376ddcf8022SAndrey A. Chernov case ':': case '-': case '_': 377ddcf8022SAndrey A. Chernov break; 378ddcf8022SAndrey A. Chernov default: 379d78e98d2SNate Williams fputc('\\', fp); 380ddcf8022SAndrey A. Chernov break; 381ddcf8022SAndrey A. Chernov } 382ddcf8022SAndrey A. Chernov } 383d78e98d2SNate Williams fputc(*ap, fp); 384d78e98d2SNate Williams } 385d78e98d2SNate Williams } 386d78e98d2SNate Williams fputs("; export ", fp); 387d78e98d2SNate Williams fwrite(*atenv, sizeof(char), eqp-*atenv -1, fp); 388d78e98d2SNate Williams fputc('\n', fp); 389d78e98d2SNate Williams 390d78e98d2SNate Williams } 391d78e98d2SNate Williams } 392b89321a5SAndrey A. Chernov /* Cd to the directory at the time and write out all the 393b89321a5SAndrey A. Chernov * commands the user supplies from stdin. 394d78e98d2SNate Williams */ 395b89321a5SAndrey A. Chernov fprintf(fp, "cd "); 396b89321a5SAndrey A. Chernov for (ap = cwdname(); *ap != '\0'; ap++) 397b89321a5SAndrey A. Chernov { 398b89321a5SAndrey A. Chernov if (*ap == '\n') 399b89321a5SAndrey A. Chernov fprintf(fp, "\"\n\""); 400b89321a5SAndrey A. Chernov else 401b89321a5SAndrey A. Chernov { 402b89321a5SAndrey A. Chernov if (*ap != '/' && !isalnum(*ap)) 403b89321a5SAndrey A. Chernov fputc('\\', fp); 404b89321a5SAndrey A. Chernov 405b89321a5SAndrey A. Chernov fputc(*ap, fp); 406b89321a5SAndrey A. Chernov } 407b89321a5SAndrey A. Chernov } 408b89321a5SAndrey A. Chernov /* Test cd's exit status: die if the original directory has been 409b89321a5SAndrey A. Chernov * removed, become unreadable or whatever 410b89321a5SAndrey A. Chernov */ 411b89321a5SAndrey A. Chernov fprintf(fp, " || {\n\t echo 'Execution directory " 412b89321a5SAndrey A. Chernov "inaccessible' >&2\n\t exit 1\n}\n"); 413d78e98d2SNate Williams 414d78e98d2SNate Williams while((ch = getchar()) != EOF) 415d78e98d2SNate Williams fputc(ch, fp); 416d78e98d2SNate Williams 417d78e98d2SNate Williams fprintf(fp, "\n"); 418d78e98d2SNate Williams if (ferror(fp)) 419d78e98d2SNate Williams panic("Output error"); 420d78e98d2SNate Williams 421d78e98d2SNate Williams if (ferror(stdin)) 422d78e98d2SNate Williams panic("Input error"); 423d78e98d2SNate Williams 424d78e98d2SNate Williams fclose(fp); 425d78e98d2SNate Williams 426b89321a5SAndrey A. Chernov /* Set the x bit so that we're ready to start executing 427d78e98d2SNate Williams */ 428b89321a5SAndrey A. Chernov 429d78e98d2SNate Williams if (fchmod(fd2, S_IRUSR | S_IWUSR | S_IXUSR) < 0) 430d78e98d2SNate Williams perr("Cannot give away file"); 431d78e98d2SNate Williams 432d78e98d2SNate Williams close(fd2); 433ddcf8022SAndrey A. Chernov fprintf(stderr, "Job %ld will be executed using /bin/sh\n", jobno); 434d78e98d2SNate Williams } 435d78e98d2SNate Williams 436d78e98d2SNate Williams static void 437d78e98d2SNate Williams list_jobs() 438d78e98d2SNate Williams { 439b89321a5SAndrey A. Chernov /* List all a user's jobs in the queue, by looping through ATJOB_DIR, 440b89321a5SAndrey A. Chernov * or everybody's if we are root 441d78e98d2SNate Williams */ 442d78e98d2SNate Williams struct passwd *pw; 443d78e98d2SNate Williams DIR *spool; 444d78e98d2SNate Williams struct dirent *dirent; 445d78e98d2SNate Williams struct stat buf; 446d78e98d2SNate Williams struct tm runtime; 447d78e98d2SNate Williams unsigned long ctm; 448d78e98d2SNate Williams char queue; 449ddcf8022SAndrey A. Chernov long jobno; 450d78e98d2SNate Williams time_t runtimer; 451d78e98d2SNate Williams char timestr[TIMESIZE]; 452d78e98d2SNate Williams int first=1; 453d78e98d2SNate Williams 454f4a747a1SStephen McKay #ifdef __FreeBSD__ 455f4a747a1SStephen McKay (void) setlocale(LC_TIME, ""); 456f4a747a1SStephen McKay #endif 457f4a747a1SStephen McKay 458d78e98d2SNate Williams PRIV_START 459d78e98d2SNate Williams 460b89321a5SAndrey A. Chernov if (chdir(ATJOB_DIR) != 0) 461b89321a5SAndrey A. Chernov perr("Cannot change to " ATJOB_DIR); 462d78e98d2SNate Williams 463d78e98d2SNate Williams if ((spool = opendir(".")) == NULL) 464b89321a5SAndrey A. Chernov perr("Cannot open " ATJOB_DIR); 465d78e98d2SNate Williams 466b89321a5SAndrey A. Chernov /* Loop over every file in the directory 467b89321a5SAndrey A. Chernov */ 468d78e98d2SNate Williams while((dirent = readdir(spool)) != NULL) { 469d78e98d2SNate Williams if (stat(dirent->d_name, &buf) != 0) 470b89321a5SAndrey A. Chernov perr("Cannot stat in " ATJOB_DIR); 471d78e98d2SNate Williams 472b89321a5SAndrey A. Chernov /* See it's a regular file and has its x bit turned on and 473d78e98d2SNate Williams * is the user's 474d78e98d2SNate Williams */ 475d78e98d2SNate Williams if (!S_ISREG(buf.st_mode) 476d78e98d2SNate Williams || ((buf.st_uid != real_uid) && ! (real_uid == 0)) 477d78e98d2SNate Williams || !(S_IXUSR & buf.st_mode || atverify)) 478d78e98d2SNate Williams continue; 479d78e98d2SNate Williams 480ddcf8022SAndrey A. Chernov if(sscanf(dirent->d_name, "%c%5lx%8lx", &queue, &jobno, &ctm)!=3) 481d78e98d2SNate Williams continue; 482d78e98d2SNate Williams 483d78e98d2SNate Williams if (atqueue && (queue != atqueue)) 484d78e98d2SNate Williams continue; 485d78e98d2SNate Williams 486d78e98d2SNate Williams runtimer = 60*(time_t) ctm; 487d78e98d2SNate Williams runtime = *localtime(&runtimer); 488d78e98d2SNate Williams strftime(timestr, TIMESIZE, "%X %x", &runtime); 489d78e98d2SNate Williams if (first) { 490d78e98d2SNate Williams printf("Date\t\t\tOwner\tQueue\tJob#\n"); 491d78e98d2SNate Williams first=0; 492d78e98d2SNate Williams } 493d78e98d2SNate Williams pw = getpwuid(buf.st_uid); 494d78e98d2SNate Williams 495ddcf8022SAndrey A. Chernov printf("%s\t%s\t%c%s\t%ld\n", 496d78e98d2SNate Williams timestr, 497d78e98d2SNate Williams pw ? pw->pw_name : "???", 498d78e98d2SNate Williams queue, 499d78e98d2SNate Williams (S_IXUSR & buf.st_mode) ? "":"(done)", 500ddcf8022SAndrey A. Chernov jobno); 501d78e98d2SNate Williams } 502d78e98d2SNate Williams PRIV_END 503d78e98d2SNate Williams } 504d78e98d2SNate Williams 505d78e98d2SNate Williams static void 506ddcf8022SAndrey A. Chernov process_jobs(int argc, char **argv, int what) 507d78e98d2SNate Williams { 508b89321a5SAndrey A. Chernov /* Delete every argument (job - ID) given 509b89321a5SAndrey A. Chernov */ 510d78e98d2SNate Williams int i; 511d78e98d2SNate Williams struct stat buf; 512ddcf8022SAndrey A. Chernov DIR *spool; 513ddcf8022SAndrey A. Chernov struct dirent *dirent; 514ddcf8022SAndrey A. Chernov unsigned long ctm; 515ddcf8022SAndrey A. Chernov char queue; 516ddcf8022SAndrey A. Chernov long jobno; 517d78e98d2SNate Williams 518d78e98d2SNate Williams PRIV_START 519d78e98d2SNate Williams 520b89321a5SAndrey A. Chernov if (chdir(ATJOB_DIR) != 0) 521b89321a5SAndrey A. Chernov perr("Cannot change to " ATJOB_DIR); 522d78e98d2SNate Williams 523ddcf8022SAndrey A. Chernov if ((spool = opendir(".")) == NULL) 524ddcf8022SAndrey A. Chernov perr("Cannot open " ATJOB_DIR); 525ddcf8022SAndrey A. Chernov 526ddcf8022SAndrey A. Chernov PRIV_END 527ddcf8022SAndrey A. Chernov 528ddcf8022SAndrey A. Chernov /* Loop over every file in the directory 529ddcf8022SAndrey A. Chernov */ 530ddcf8022SAndrey A. Chernov while((dirent = readdir(spool)) != NULL) { 531ddcf8022SAndrey A. Chernov 532ddcf8022SAndrey A. Chernov PRIV_START 533ddcf8022SAndrey A. Chernov if (stat(dirent->d_name, &buf) != 0) 534ddcf8022SAndrey A. Chernov perr("Cannot stat in " ATJOB_DIR); 535ddcf8022SAndrey A. Chernov PRIV_END 536ddcf8022SAndrey A. Chernov 537ddcf8022SAndrey A. Chernov if(sscanf(dirent->d_name, "%c%5lx%8lx", &queue, &jobno, &ctm)!=3) 5389cb939b4SAndrey A. Chernov continue; 539ddcf8022SAndrey A. Chernov 540ddcf8022SAndrey A. Chernov for (i=optind; i < argc; i++) { 541ddcf8022SAndrey A. Chernov if (atoi(argv[i]) == jobno) { 542d78e98d2SNate Williams if ((buf.st_uid != real_uid) && !(real_uid == 0)) { 543d78e98d2SNate Williams fprintf(stderr, "%s: Not owner\n", argv[i]); 544ddcf8022SAndrey A. Chernov exit(EXIT_FAILURE); 545d78e98d2SNate Williams } 546ddcf8022SAndrey A. Chernov switch (what) { 547ddcf8022SAndrey A. Chernov case ATRM: 548ddcf8022SAndrey A. Chernov 549ddcf8022SAndrey A. Chernov PRIV_START 550ddcf8022SAndrey A. Chernov 551ddcf8022SAndrey A. Chernov if (unlink(dirent->d_name) != 0) 552ddcf8022SAndrey A. Chernov perr(dirent->d_name); 553ddcf8022SAndrey A. Chernov 554d78e98d2SNate Williams PRIV_END 555ddcf8022SAndrey A. Chernov 556ddcf8022SAndrey A. Chernov break; 557ddcf8022SAndrey A. Chernov 558ddcf8022SAndrey A. Chernov case CAT: 559ddcf8022SAndrey A. Chernov { 560ddcf8022SAndrey A. Chernov FILE *fp; 561ddcf8022SAndrey A. Chernov int ch; 562ddcf8022SAndrey A. Chernov 563ddcf8022SAndrey A. Chernov PRIV_START 564ddcf8022SAndrey A. Chernov 565ddcf8022SAndrey A. Chernov fp = fopen(dirent->d_name,"r"); 566ddcf8022SAndrey A. Chernov 567ddcf8022SAndrey A. Chernov PRIV_END 568ddcf8022SAndrey A. Chernov 569ddcf8022SAndrey A. Chernov if (!fp) { 570ddcf8022SAndrey A. Chernov perr("Cannot open file"); 571ddcf8022SAndrey A. Chernov } 572ddcf8022SAndrey A. Chernov while((ch = getc(fp)) != EOF) { 573ddcf8022SAndrey A. Chernov putchar(ch); 574ddcf8022SAndrey A. Chernov } 575ddcf8022SAndrey A. Chernov } 576ddcf8022SAndrey A. Chernov break; 577ddcf8022SAndrey A. Chernov 578ddcf8022SAndrey A. Chernov default: 579ddcf8022SAndrey A. Chernov fprintf(stderr, 580ddcf8022SAndrey A. Chernov "Internal error, process_jobs = %d\n",what); 581ddcf8022SAndrey A. Chernov exit(EXIT_FAILURE); 582ddcf8022SAndrey A. Chernov break; 583ddcf8022SAndrey A. Chernov } 584ddcf8022SAndrey A. Chernov } 585ddcf8022SAndrey A. Chernov } 586ddcf8022SAndrey A. Chernov } 587d78e98d2SNate Williams } /* delete_jobs */ 588d78e98d2SNate Williams 589d78e98d2SNate Williams /* Global functions */ 590d78e98d2SNate Williams 591b89321a5SAndrey A. Chernov void * 592b89321a5SAndrey A. Chernov mymalloc(size_t n) 593b89321a5SAndrey A. Chernov { 594b89321a5SAndrey A. Chernov void *p; 595b89321a5SAndrey A. Chernov if ((p=malloc(n))==(void *)0) 596b89321a5SAndrey A. Chernov { 597b89321a5SAndrey A. Chernov fprintf(stderr,"Virtual memory exhausted\n"); 598b89321a5SAndrey A. Chernov exit(EXIT_FAILURE); 599b89321a5SAndrey A. Chernov } 600b89321a5SAndrey A. Chernov return p; 601b89321a5SAndrey A. Chernov } 602b89321a5SAndrey A. Chernov 603d78e98d2SNate Williams int 604b89321a5SAndrey A. Chernov main(int argc, char **argv) 605d78e98d2SNate Williams { 606d78e98d2SNate Williams int c; 607b89321a5SAndrey A. Chernov char queue = DEFAULT_AT_QUEUE; 608b89321a5SAndrey A. Chernov char queue_set = 0; 609d78e98d2SNate Williams char *pgm; 610d78e98d2SNate Williams 611ddcf8022SAndrey A. Chernov enum { ATQ, ATRM, AT, BATCH, CAT }; /* what program we want to run */ 612d78e98d2SNate Williams int program = AT; /* our default program */ 613ddcf8022SAndrey A. Chernov char *options = "q:f:mvldbVc"; /* default options for at */ 614b89321a5SAndrey A. Chernov int disp_version = 0; 615d78e98d2SNate Williams time_t timer; 616d78e98d2SNate Williams 617d78e98d2SNate Williams RELINQUISH_PRIVS 618d78e98d2SNate Williams 619b89321a5SAndrey A. Chernov /* Eat any leading paths 620b89321a5SAndrey A. Chernov */ 621d78e98d2SNate Williams if ((pgm = strrchr(argv[0], '/')) == NULL) 622d78e98d2SNate Williams pgm = argv[0]; 623d78e98d2SNate Williams else 624d78e98d2SNate Williams pgm++; 625d78e98d2SNate Williams 626d78e98d2SNate Williams namep = pgm; 627d78e98d2SNate Williams 628b89321a5SAndrey A. Chernov /* find out what this program is supposed to do 629b89321a5SAndrey A. Chernov */ 630d78e98d2SNate Williams if (strcmp(pgm, "atq") == 0) { 631d78e98d2SNate Williams program = ATQ; 632b89321a5SAndrey A. Chernov options = "q:vV"; 633b89321a5SAndrey A. Chernov } 634b89321a5SAndrey A. Chernov else if (strcmp(pgm, "atrm") == 0) { 635d78e98d2SNate Williams program = ATRM; 636b89321a5SAndrey A. Chernov options = "V"; 637b89321a5SAndrey A. Chernov } 638b89321a5SAndrey A. Chernov else if (strcmp(pgm, "batch") == 0) { 639d78e98d2SNate Williams program = BATCH; 640b89321a5SAndrey A. Chernov options = "f:q:mvV"; 641d78e98d2SNate Williams } 642d78e98d2SNate Williams 643b89321a5SAndrey A. Chernov /* process whatever options we can process 644b89321a5SAndrey A. Chernov */ 645d78e98d2SNate Williams opterr=1; 6461c8af878SWarner Losh while ((c=getopt(argc, argv, options)) != -1) 647d78e98d2SNate Williams switch (c) { 648d78e98d2SNate Williams case 'v': /* verify time settings */ 649d78e98d2SNate Williams atverify = 1; 650d78e98d2SNate Williams break; 651d78e98d2SNate Williams 652d78e98d2SNate Williams case 'm': /* send mail when job is complete */ 653d78e98d2SNate Williams send_mail = 1; 654d78e98d2SNate Williams break; 655d78e98d2SNate Williams 656d78e98d2SNate Williams case 'f': 657d78e98d2SNate Williams atinput = optarg; 658d78e98d2SNate Williams break; 659d78e98d2SNate Williams 660d78e98d2SNate Williams case 'q': /* specify queue */ 661d78e98d2SNate Williams if (strlen(optarg) > 1) 662d78e98d2SNate Williams usage(); 663d78e98d2SNate Williams 664d78e98d2SNate Williams atqueue = queue = *optarg; 665b89321a5SAndrey A. Chernov if (!(islower(queue)||isupper(queue))) 666d78e98d2SNate Williams usage(); 667b89321a5SAndrey A. Chernov 668b89321a5SAndrey A. Chernov queue_set = 1; 669b89321a5SAndrey A. Chernov break; 670b89321a5SAndrey A. Chernov 671b89321a5SAndrey A. Chernov case 'd': 672b89321a5SAndrey A. Chernov if (program != AT) 673b89321a5SAndrey A. Chernov usage(); 674b89321a5SAndrey A. Chernov 675b89321a5SAndrey A. Chernov program = ATRM; 676b89321a5SAndrey A. Chernov options = "V"; 677b89321a5SAndrey A. Chernov break; 678b89321a5SAndrey A. Chernov 679b89321a5SAndrey A. Chernov case 'l': 680b89321a5SAndrey A. Chernov if (program != AT) 681b89321a5SAndrey A. Chernov usage(); 682b89321a5SAndrey A. Chernov 683b89321a5SAndrey A. Chernov program = ATQ; 684b89321a5SAndrey A. Chernov options = "q:vV"; 685b89321a5SAndrey A. Chernov break; 686b89321a5SAndrey A. Chernov 687b89321a5SAndrey A. Chernov case 'b': 688b89321a5SAndrey A. Chernov if (program != AT) 689b89321a5SAndrey A. Chernov usage(); 690b89321a5SAndrey A. Chernov 691b89321a5SAndrey A. Chernov program = BATCH; 692b89321a5SAndrey A. Chernov options = "f:q:mvV"; 693b89321a5SAndrey A. Chernov break; 694b89321a5SAndrey A. Chernov 695b89321a5SAndrey A. Chernov case 'V': 696b89321a5SAndrey A. Chernov disp_version = 1; 697d78e98d2SNate Williams break; 698d78e98d2SNate Williams 699ddcf8022SAndrey A. Chernov case 'c': 700ddcf8022SAndrey A. Chernov program = CAT; 701ddcf8022SAndrey A. Chernov options = ""; 702ddcf8022SAndrey A. Chernov break; 703ddcf8022SAndrey A. Chernov 704d78e98d2SNate Williams default: 705d78e98d2SNate Williams usage(); 706d78e98d2SNate Williams break; 707d78e98d2SNate Williams } 708b89321a5SAndrey A. Chernov /* end of options eating 709b89321a5SAndrey A. Chernov */ 710d78e98d2SNate Williams 711b89321a5SAndrey A. Chernov if (disp_version) 712b89321a5SAndrey A. Chernov fprintf(stderr, "at version " VERSION "\n" 713b89321a5SAndrey A. Chernov "Bug reports to: ig25@rz.uni-karlsruhe.de (Thomas Koenig)\n"); 714b89321a5SAndrey A. Chernov 715b89321a5SAndrey A. Chernov /* select our program 716b89321a5SAndrey A. Chernov */ 717b89321a5SAndrey A. Chernov if(!check_permission()) 718b89321a5SAndrey A. Chernov { 719b89321a5SAndrey A. Chernov fprintf(stderr, "You do not have permission to use %s.\n",namep); 720b89321a5SAndrey A. Chernov exit(EXIT_FAILURE); 721b89321a5SAndrey A. Chernov } 722d78e98d2SNate Williams switch (program) { 723d78e98d2SNate Williams case ATQ: 724d78e98d2SNate Williams 725b89321a5SAndrey A. Chernov REDUCE_PRIV(DAEMON_UID, DAEMON_GID) 726d78e98d2SNate Williams 727d78e98d2SNate Williams list_jobs(); 728d78e98d2SNate Williams break; 729d78e98d2SNate Williams 730d78e98d2SNate Williams case ATRM: 731d78e98d2SNate Williams 732b89321a5SAndrey A. Chernov REDUCE_PRIV(DAEMON_UID, DAEMON_GID) 733d78e98d2SNate Williams 734ddcf8022SAndrey A. Chernov process_jobs(argc, argv, ATRM); 735ddcf8022SAndrey A. Chernov break; 736ddcf8022SAndrey A. Chernov 737ddcf8022SAndrey A. Chernov case CAT: 738ddcf8022SAndrey A. Chernov 739ddcf8022SAndrey A. Chernov process_jobs(argc, argv, CAT); 740d78e98d2SNate Williams break; 741d78e98d2SNate Williams 742d78e98d2SNate Williams case AT: 743d78e98d2SNate Williams timer = parsetime(argc, argv); 744b89321a5SAndrey A. Chernov if (atverify) 745b89321a5SAndrey A. Chernov { 746d78e98d2SNate Williams struct tm *tm = localtime(&timer); 747d78e98d2SNate Williams fprintf(stderr, "%s\n", asctime(tm)); 748d78e98d2SNate Williams } 749d78e98d2SNate Williams writefile(timer, queue); 750d78e98d2SNate Williams break; 751d78e98d2SNate Williams 752d78e98d2SNate Williams case BATCH: 753b89321a5SAndrey A. Chernov if (queue_set) 754b89321a5SAndrey A. Chernov queue = toupper(queue); 755b89321a5SAndrey A. Chernov else 756b89321a5SAndrey A. Chernov queue = DEFAULT_BATCH_QUEUE; 757b89321a5SAndrey A. Chernov 758b89321a5SAndrey A. Chernov if (argc > optind) 759b89321a5SAndrey A. Chernov timer = parsetime(argc, argv); 760b89321a5SAndrey A. Chernov else 761b89321a5SAndrey A. Chernov timer = time(NULL); 762b89321a5SAndrey A. Chernov 763b89321a5SAndrey A. Chernov if (atverify) 764b89321a5SAndrey A. Chernov { 765b89321a5SAndrey A. Chernov struct tm *tm = localtime(&timer); 766b89321a5SAndrey A. Chernov fprintf(stderr, "%s\n", asctime(tm)); 767b89321a5SAndrey A. Chernov } 768b89321a5SAndrey A. Chernov 769b89321a5SAndrey A. Chernov writefile(timer, queue); 770d78e98d2SNate Williams break; 771d78e98d2SNate Williams 772d78e98d2SNate Williams default: 773d78e98d2SNate Williams panic("Internal error"); 774d78e98d2SNate Williams break; 775d78e98d2SNate Williams } 776d78e98d2SNate Williams exit(EXIT_SUCCESS); 777d78e98d2SNate Williams } 778