#pragma prototyped

static const char usage[] =

;

#include "at.h"

#include <fs3d.h>

#define ACCESS (1<<0)

#define ADMIN (1<<1)

#define EXEC (1<<2)

#define INFO (1<<3)

#define JOB (1<<4)

#define LIST (1<<5)

#define LOG (1<<6)

#define MAIL (1<<7)

#define QUIT (1<<8)

#define REMOVE (1<<9)

#define STATUS (1<<10)

#define TRACE (1<<11)

#define UPDATE (1<<12)

#define USER (1<<13)

#define VERSION (1<<14)

main(int argc, char** argv)

{

register int c;

register char* s;

register Sfio_t* sp;

register Sfio_t* tp;

char* e;

char* t;

int n;

int skip;

time_t now;

char buf[PATH_MAX];

int flags = 0;

int interval = 1;

int op = EXEC;

char* file = 0;

char* label = 0;

char* queue = 0;

char* service = AT_SERVICE;

char* date = 0;

char* every = 0;

char* trace = 0;

char* update = 0;

time_t start = 0;

static char que[2];

if (!(s = strrchr(*argv, '/')))

s = *argv;

else if (t = strchr(++s, '_'))

s = t + 1;

if (*s == 'b')

{

error_info.id = "batch";

*(queue = que) = *s;

op |= MAIL;

date = "now";

}

else

error_info.id = "at";

for (;;)

{

switch (optget(argv, usage))

{

case 'a':

op |= ACCESS;

continue;

case 'f':

file = opt_info.arg;

continue;

case 'h':

label = opt_info.arg;

continue;

case 'i':

op |= INFO;

continue;

case 'l':

op |= LIST;

continue;

case 'm':

op |= MAIL;

continue;

case 'n':

op &= ~EXEC;

continue;

case 'p':

op |= STATUS;

continue;

case 'q':

queue = opt_info.arg;

continue;

case 'r':

op |= REMOVE;

continue;

case 's':

service = opt_info.arg;

continue;

case 't':

date = opt_info.arg;

continue;

case 'u':

op |= USER;

continue;

case 'v':

op |= VERSION;

continue;

case 'A':

op |= ADMIN;

continue;

case 'D':

op |= TRACE;

trace = opt_info.arg;

continue;

case 'L':

op |= LOG;

continue;

case 'Q':

op |= QUIT;

continue;

case 'U':

op |= UPDATE;

update = opt_info.arg;

continue;

case ':':

error(2, "%s", opt_info.arg);

continue;

case '?':

error(ERROR_USAGE|4, "%s", opt_info.arg);

continue;

}

break;

}

argv += opt_info.index;

if (error_info.errors)

error(ERROR_USAGE|4, "%s", optusage(NiL));

if (!(sp = sfstropen()))

error(ERROR_SYSTEM|3, "out of space [message]");

sfprintf(sp, "#00000\n#");

if (op & ADMIN)

sfprintf(sp, "%c", AT_ADMIN);

if (queue)

sfprintf(sp, "%c%s ", AT_QUEUE, queue);

switch (op & (ACCESS|INFO|LIST|LOG|QUIT|REMOVE|STATUS|TRACE|UPDATE|USER|VERSION))

{

case ACCESS:

sfputc(sp, AT_ACCESS);

break;

case INFO:

sfputc(sp, AT_INFO);

break;

case LIST:

sfputc(sp, AT_LIST);

break;

case LOG:

sfputc(sp, AT_LOG);

break;

case QUIT:

sfputc(sp, AT_QUIT);

flags |= CS_OPEN_TEST;

break;

case REMOVE:

sfputc(sp, AT_REMOVE);

break;

case STATUS:

sfputc(sp, AT_STATUS);

break;

case TRACE:

sfprintf(sp, "%c%s", AT_DEBUG, trace);

break;

case UPDATE:

sfprintf(sp, "%c%s", AT_UPDATE, update ? update : "");

break;

case USER:

sfputc(sp, AT_USER);

break;

case VERSION:

sfputc(sp, AT_VERSION);

break;

default:

if (op & ~(EXEC|MAIL))

error(3, "only one of -[aglrsuvLQU] may be specified");

op |= JOB;

if (file && !sfopen(sfstdin, file, "r"))

error(ERROR_SYSTEM|3, "%s: cannot read", file);

if (!fs3d(FS3D_TEST) || !(s = pathpath("3d", NiL, PATH_ABSOLUTE|PATH_EXECUTE, buf, sizeof(buf))))

s = pathshell();

if (op & MAIL)

sfputc(sp, AT_MAIL);

skip = sfstrtell(sp) + 1;

sfprintf(sp, "%c00000 %s ", AT_JOB, s);

if (date || *argv)

{

if (!(tp = sfstropen()))

error(ERROR_SYSTEM|3, "out of space [time]");

sfputr(tp, "+0 ", -1);

if (date)

sfputr(tp, date, -1);

else

for (s = *argv;;)

{

for (n = ' '; c = *s++; n = c)

if (!isspace(c) || n != (c = ' '))

sfputc(tp, c);

if (!(s = *++argv))

break;

if (n != ' ')

sfputc(tp, ' ');

}

if (!(s = sfstruse(tp)))

error(ERROR_SYSTEM|3, "out of space");

s += 3;

if (strneq(s, "cron ", n = 5) || strneq(s, "each ", n = 5) || strneq(s, "every ", n = 6) || strneq(s, "repeat ", n = 7))

{

every = s + n;

t = s;

*t++ = '+';

*t++ = '0';

while (*t != ' ')

*t++ = ' ';

if (*++t == '+' && isdigit(*(t + 1)))

{

interval = strtol(t, NiL, 0);

while (*t && *t != ' ')

*t++ = ' ';

}

else if (isdigit(*t))

{

interval = strtol(t, &e, 0);

if (!isalpha(*e))

interval = 1;

else

while (*t && *t != ' ')

*t++ = ' ';

}

while (*every == ' ')

every++;

}

now = time(NiL);

start = tmdate(s, &t, &now);

if (!every && (unsigned long)start <= (unsigned long)now)

start = tmdate(s - 3, &t, &now);

if (*t)

error(3, "%s: invalid time specification", s);

if (s = label)

{

sfputc(sp, AT_LABEL);

while ((c = *s++) && !isspace(c))

if (isalnum(c))

sfputc(sp, c);

sfputc(sp, ' ');

}

sfprintf(sp, "%c%lu", AT_TIME, start);

if (s = every)

sfprintf(sp, " +%d %s", interval, s);

}

break;

}

if (op & JOB)

{

if (!(s = coinit(CO_EXPORT|CO_SERVER)))

error(ERROR_SYSTEM|3, "cannot generate initialization script");

sfprintf(sp, "\n%s\n", s);

s = sfstrbase(sp);

sfsprintf(s + skip, 6, "%05lu ", sfstrtell(sp) - 7);

s[skip + 5] = ' ';

if (s = *argv++)

{

sfprintf(sp, "%s", s);

while (s = *argv++)

sfprintf(sp, " %s", s);

}

else

sfmove(sfstdin, sp, SF_UNBOUND, -1);

sfputc(sp, '\n');

}

else

{

if (s = *argv++)

{

if (streq(s, "-") && !*argv)

s = "*";

sfprintf(sp, " %s", s);

while (s = *argv++)

sfprintf(sp, "|%s", s);

}

sfputc(sp, '\n');

}

n = sfstrtell(sp);

if (!(s = sfstruse(sp)))

error(ERROR_SYSTEM|3, "out of space");

sfsprintf(s, 7, "#%05d\n", n - 7);

s[6] = '\n';

if (op & EXEC)

{

if ((op = csopen(&cs, service, 0)) < 0)

error(ERROR_SYSTEM|3, "%s: cannot open service", service);

if (cswrite(&cs, op, s, n) != n)

error(ERROR_SYSTEM|3, "%s: service write error", service);

while ((n = read(op, buf, sizeof(buf))) > 0)

if (write(1, buf, n) != n)

error(ERROR_SYSTEM|3, "write error");

if (n < 0)

error(ERROR_SYSTEM|3, "%s: service read error", service);

}

else

{

if (op & JOB)

{

sfprintf(sfstdout, "submit job to run on %s", fmttime(NiL, start));

if (every)

sfprintf(sfstdout, " and repeat every +%d %s", interval, every);

sfprintf(sfstdout, "\n");

}

sfputr(sfstdout, s, -1);

}

return 0;

}