New rpcclient command "at" for NT scheduler control, a slightly improved

version of the NT command.

at { time [/INTERACTIVE] [{/EVERY|/NEXT}:5,Sun,...] command | [/DEL] [jobid] }

Examples (options used in abbreviated form):
at                                         ; Shows all jobs
at 1                                       ; Detail on job 1
at /D                                      ; Deletes all jobs
at /D 1                                    ; Deletes job 1
at 11:11:11AM /I /N:1 d:\humour\silly.exe  ; First of next month
at 9:00AM /E:M,T,W,Th,F net send MATTY Hi  ; Each weekday
at 11:00PM /E c:\winnt\backup.exe          ; Every day
(This used to be commit cf8c476d2eec24c150877b6bb8af7f7875cc1840)

1 files changed, 327 insertions, 0 deletions

diff --git a/source3/rpcclient/cmd_atsvc.c b/source3/rpcclient/cmd_atsvc.c
new file mode 100644
index 0000000000..a7b771e48e
--- /dev/null
+++ b/source3/rpcclient/cmd_atsvc.c
@@ -0,0 +1,327 @@
+/* 
+   Unix SMB/Netbios implementation.
+   Version 2.1.
+   MSRPC client: scheduler service
+   Copyright (C) Matthew Chapman 1999
+   Copyright (C) Luke Kenneth Casson Leighton 1996-1999
+   Copyright (C) Andrew Tridgell 1994-1999
+   
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+
+#ifdef SYSLOG
+#undef SYSLOG
+#endif
+
+#include "includes.h"
+#include "nterr.h"
+
+extern int DEBUGLEVEL;
+
+#define DEBUG_TESTING
+
+extern struct cli_state *smb_cli;
+extern FILE* out_hnd;
+
+
+/****************************************************************************
+checks for a /OPTION:param style option
+****************************************************************************/
+static BOOL checkopt(char *input, char *optname, char **params)
+{
+	char *inend;
+	int i, len;
+
+	if (*input++ != '/')
+		return False;
+
+	for (inend = input; *inend != 0 && *inend != ':'; inend++);
+
+	if (params != NULL)
+	{
+		*inend = 0;
+		*params = inend;
+	}
+
+	while(input < inend)
+	{
+		if (toupper(*input++) != *optname++)
+			return False;
+	}
+
+	return True;
+}
+
+char *daynames[] = {"Mon","Tue","Wed","Thu","Fri","Sat","Sun"};
+char *daynames_short[] = {"M", "T", "W", "Th", "F", "S", "Su"};
+
+/****************************************************************************
+parses a list of days of the week and month
+****************************************************************************/
+static BOOL at_parse_days(char *str, uint32 *monthdays, uint8 *weekdays)
+{
+	char *tok;
+	char *nexttok = str;
+	int day;
+
+	do {
+		tok = nexttok;
+
+		if ((nexttok = strchr(tok, ',')) != NULL)
+		{
+			*nexttok++ = 0;
+		}
+
+		if (isdigit(*tok))
+		{
+			day = strtol(tok, NULL, 10);
+			if (day == 0 || day > 31)
+			{
+				printf("\tInvalid day of month.\n\n");
+				return False;
+			}
+
+			*monthdays |= (1 << (day-1));
+		}
+		else
+		{
+			if (strlen(tok) < 3)
+			{
+				for (day = 0; day < 7; day++)
+				{
+					if (!strcasecmp(tok, daynames_short[day]))
+						break;
+				}
+			}
+			else
+			{
+				for (day = 0; day < 7; day++)
+				{
+					if (!strncasecmp(tok, daynames[day], 3))
+						break;
+				}
+			}
+
+			if (day < 7)
+			{
+				*weekdays |= (1 << day);
+			}
+			else
+			{
+				printf("\tInvalid day of week\n\n");
+				return False;
+			}
+		}
+
+	} while (nexttok != NULL);
+
+	return True;
+}
+
+
+/****************************************************************************
+scheduler add job
+****************************************************************************/
+void cmd_at(struct client_info *info)
+{
+	uint16 nt_pipe_fnum;
+	fstring temp;
+	fstring dest_wks;
+	BOOL add = False;
+	BOOL del = False;
+	char *p;
+
+	uint32 jobid = -1;
+	unsigned int hours, minutes, seconds = 0;
+	uint32 monthdays = 0;
+	uint8 weekdays = 0;
+	uint8 flags = JOB_NONINTERACTIVE;
+	fstring command;
+
+        while (next_token(NULL, temp, NULL, sizeof(temp)))
+	{
+		if (checkopt(temp, "DELETE", NULL))
+		{
+			del = True;
+			continue;
+		}
+		else if (checkopt(temp, "YES", NULL))
+		{
+			/* Compatibility */
+			continue;
+		}
+
+		jobid = strtol(temp, &p, 10);
+		if (*p == 0)   /* Entirely numeric field */
+			continue;
+
+		if ((p == temp) || (sscanf(temp, "%d:%d:%d",
+				     &hours, &minutes, &seconds) < 2))
+		{
+			printf("at { time [/INTERACTIVE] [{/EVERY|/NEXT}:5,Sun,...] command | [/DEL] [jobid] }\n\n");
+			return;
+		}
+		
+		p = strchr(temp, 0);
+		
+		if (!strcasecmp(p-2, "AM"))
+		{
+			hours = (hours == 12) ? 0 : hours;
+		}
+
+		if (!strcasecmp(p-2, "PM"))
+		{
+			hours = (hours == 12) ? 12 : hours + 12;
+		}
+
+		if (hours > 23 || minutes > 59 || seconds > 59)
+		{
+			printf("\tInvalid time.\n\n");
+			return;
+		}
+
+		add = True;
+		command[0] = 0;
+		p = NULL;
+
+		if (!next_token(NULL, temp, NULL, sizeof(temp)))
+			break;
+
+		if (checkopt(temp, "INTERACTIVE", NULL))
+		{
+			flags &= ~JOB_NONINTERACTIVE;
+
+			if (!next_token(NULL, temp, NULL, sizeof(temp)))
+				break;
+		}
+
+		if (checkopt(temp, "EVERY", &p))
+		{
+			flags |= JOB_PERIODIC;
+		}
+		else
+		{
+			checkopt(temp, "NEXT", &p);
+		}
+
+		if (p != NULL)
+		{
+			if (*p == ':')
+			{
+				if (!at_parse_days(p, &monthdays, &weekdays))
+					return;
+			}
+			else
+			{
+				weekdays = 0x7F;
+			}
+
+			if (!next_token(NULL, temp, NULL, sizeof(temp)))
+				break;
+		}
+
+		while (True)
+		{
+			fstrcat(command, temp);
+
+			if (!next_token(NULL, temp, NULL, sizeof(temp)))
+				break;
+
+			fstrcat(command, " ");
+		}
+
+		break;
+	}
+
+	if (add && !command[0])
+	{
+		printf("\tNo command specified.\n\n");
+		return;
+	}
+
+	fstrcpy(dest_wks, "\\\\");
+	fstrcat(dest_wks, info->dest_host);
+	strupper(dest_wks);
+
+	/* open scheduler session. */
+	if (!cli_nt_session_open(smb_cli, PIPE_ATSVC, &nt_pipe_fnum))
+		return;
+
+	if (add) /* add job */
+	{
+		AT_JOB_INFO job;
+
+		job.time = ((((hours * 60) + minutes) * 60) + seconds) * 1000;
+		job.monthdays = monthdays;
+		job.weekdays = weekdays;
+		job.flags = flags;
+		job.ptr_command = 1;
+
+		display_at_job_info(out_hnd, ACTION_HEADER   , &job, command);
+		display_at_job_info(out_hnd, ACTION_ENUMERATE, &job, command);
+		display_at_job_info(out_hnd, ACTION_FOOTER   , &job, command);
+
+		if (at_add_job(smb_cli, nt_pipe_fnum, dest_wks, &job,
+				     command, &jobid))
+		{
+			fprintf(out_hnd, "\tJob ID:      %d\n\n", jobid);
+		}
+	}
+	else if (del) /* delete */
+	{
+		if (jobid == -1)
+		{
+			fprintf(out_hnd, "\tDeleting all jobs.\n\n");
+			at_del_job(smb_cli, nt_pipe_fnum, dest_wks,
+				   0, 0xffffffff);
+		}
+		else
+		{
+			fprintf(out_hnd, "\tDeleting job %d.\n\n", jobid);
+			at_del_job(smb_cli, nt_pipe_fnum, dest_wks,
+				   jobid, jobid);
+		}
+
+	}
+	else if (jobid == -1) /* enumerate */
+	{
+		AT_ENUM_INFO jobs[AT_MAX_JOBS];
+		fstring commands[AT_MAX_JOBS];
+		uint32 num_jobs;
+
+		if (at_enum_jobs(smb_cli, nt_pipe_fnum, dest_wks, &num_jobs,
+			     jobs, commands))
+		{
+			display_at_enum_info(out_hnd, ACTION_HEADER   , num_jobs, jobs, commands);
+			display_at_enum_info(out_hnd, ACTION_ENUMERATE, num_jobs, jobs, commands);
+			display_at_enum_info(out_hnd, ACTION_FOOTER   , num_jobs, jobs, commands);
+		}
+	}
+	else /* job info */
+	{
+		AT_JOB_INFO job;
+		
+		if (at_query_job(smb_cli, nt_pipe_fnum, dest_wks, jobid, &job, command))
+		{
+			display_at_job_info(out_hnd, ACTION_HEADER   , &job, command);
+			display_at_job_info(out_hnd, ACTION_ENUMERATE, &job, command);
+			display_at_job_info(out_hnd, ACTION_FOOTER   , &job, command);
+		}
+	}
+
+	/* close the session */
+	cli_nt_session_close(smb_cli, nt_pipe_fnum);
+}