mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
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 commit is contained in:
Matthew Chapman
parent
52b4fd5b55
commit
cf8c476d2e
327
source/rpcclient/cmd_atsvc.c
Normal file
327
source/rpcclient/cmd_atsvc.c
Normal file
@ -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);
|
||||||
|
}
|
||||||
@ -1662,6 +1662,172 @@ void display_svc_info(FILE *out_hnd, enum action_type action, ENUM_SRVC_STATUS *
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static char *get_at_time_str(uint32 time)
|
||||||
|
{
|
||||||
|
static fstring timestr;
|
||||||
|
unsigned int hours, minutes, seconds;
|
||||||
|
|
||||||
|
hours = time / 1000;
|
||||||
|
seconds = hours % 60;
|
||||||
|
hours /= 60;
|
||||||
|
minutes = hours % 60;
|
||||||
|
hours /= 60;
|
||||||
|
|
||||||
|
slprintf(timestr, sizeof(timestr)-1, "%2d:%02d:%02d",
|
||||||
|
hours, minutes, seconds);
|
||||||
|
|
||||||
|
return timestr;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern char *daynames_short[];
|
||||||
|
|
||||||
|
static char *get_at_days_str(uint32 monthdays, uint8 weekdays, uint8 flags)
|
||||||
|
{
|
||||||
|
static fstring days;
|
||||||
|
fstring numstr;
|
||||||
|
int day, bit;
|
||||||
|
BOOL first = True;
|
||||||
|
|
||||||
|
if (monthdays == 0 && weekdays == 0)
|
||||||
|
return "Once";
|
||||||
|
|
||||||
|
if (flags & JOB_PERIODIC)
|
||||||
|
{
|
||||||
|
if (IS_BITS_SET_ALL(weekdays, 0x7F))
|
||||||
|
return "Every Day";
|
||||||
|
|
||||||
|
fstrcpy(days, "Every ");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
fstrcpy(days, "Next ");
|
||||||
|
}
|
||||||
|
|
||||||
|
for (day = 1, bit = 1; day < 32; day++, bit <<= 1)
|
||||||
|
{
|
||||||
|
if (monthdays & bit)
|
||||||
|
{
|
||||||
|
if (first)
|
||||||
|
first = False;
|
||||||
|
else
|
||||||
|
fstrcat(days, ",");
|
||||||
|
|
||||||
|
slprintf(numstr, sizeof(numstr)-1, "%d", day);
|
||||||
|
fstrcat(days, numstr);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (day = 0, bit = 1; day < 7; day++, bit <<= 1)
|
||||||
|
{
|
||||||
|
if (weekdays & bit)
|
||||||
|
{
|
||||||
|
if (first)
|
||||||
|
first = False;
|
||||||
|
else
|
||||||
|
fstrcat(days, ",");
|
||||||
|
|
||||||
|
fstrcat(days, daynames_short[day]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return days;
|
||||||
|
}
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
display scheduled jobs
|
||||||
|
****************************************************************************/
|
||||||
|
void display_at_enum_info(FILE *out_hnd, enum action_type action,
|
||||||
|
uint32 num_jobs, AT_ENUM_INFO *jobs, fstring *commands)
|
||||||
|
{
|
||||||
|
switch (action)
|
||||||
|
{
|
||||||
|
case ACTION_HEADER:
|
||||||
|
{
|
||||||
|
if (num_jobs == 0)
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\tNo Jobs.\n");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\tJobs:\n");
|
||||||
|
fprintf(out_hnd, "\t-----\n");
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ACTION_ENUMERATE:
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < num_jobs; i++)
|
||||||
|
{
|
||||||
|
AT_JOB_INFO *job = &jobs[i].info;
|
||||||
|
|
||||||
|
fprintf(out_hnd, "\t%d\t%s\t%s\t%s\n",
|
||||||
|
jobs[i].jobid,
|
||||||
|
get_at_time_str(job->time),
|
||||||
|
get_at_days_str(job->monthdays,
|
||||||
|
job->weekdays,
|
||||||
|
job->flags),
|
||||||
|
commands[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ACTION_FOOTER:
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
display information about a scheduled job
|
||||||
|
****************************************************************************/
|
||||||
|
void display_at_job_info(FILE *out_hnd, enum action_type action,
|
||||||
|
AT_JOB_INFO *job, fstring command)
|
||||||
|
{
|
||||||
|
switch (action)
|
||||||
|
{
|
||||||
|
case ACTION_HEADER:
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\tJob Information:\n");
|
||||||
|
fprintf(out_hnd, "\t----------------\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ACTION_ENUMERATE:
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\tTime: %s\n",
|
||||||
|
get_at_time_str(job->time));
|
||||||
|
|
||||||
|
fprintf(out_hnd, "\tSchedule: %s\n",
|
||||||
|
get_at_days_str(job->monthdays, job->weekdays,
|
||||||
|
job->flags));
|
||||||
|
|
||||||
|
fprintf(out_hnd, "\tStatus: %s",
|
||||||
|
(job->flags & JOB_EXEC_ERR) ? "Failed" : "OK");
|
||||||
|
|
||||||
|
if (job->flags & JOB_RUNS_TODAY)
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, ", Runs Today");
|
||||||
|
}
|
||||||
|
|
||||||
|
fprintf(out_hnd, "\n\tInteractive: %s\n",
|
||||||
|
(job->flags & JOB_NONINTERACTIVE) ? "No"
|
||||||
|
: "Yes");
|
||||||
|
|
||||||
|
fprintf(out_hnd, "\tCommand: %s\n", command);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ACTION_FOOTER:
|
||||||
|
{
|
||||||
|
fprintf(out_hnd, "\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
#if COPY_THIS_TEMPLATE
|
#if COPY_THIS_TEMPLATE
|
||||||
/****************************************************************************
|
/****************************************************************************
|
||||||
display structure
|
display structure
|
||||||
|
|||||||
@ -106,6 +106,7 @@ struct
|
|||||||
} commands[] =
|
} commands[] =
|
||||||
{
|
{
|
||||||
{"svcenum", cmd_svc_enum, "[-i] Lists Services Manager"},
|
{"svcenum", cmd_svc_enum, "[-i] Lists Services Manager"},
|
||||||
|
{"at", cmd_at, "Scheduler control (at /? for syntax)"},
|
||||||
{"regenum", cmd_reg_enum, "<keyname> Registry Enumeration (keys, values)"},
|
{"regenum", cmd_reg_enum, "<keyname> Registry Enumeration (keys, values)"},
|
||||||
{"regdeletekey",cmd_reg_delete_key, "<keyname> Registry Key Delete"},
|
{"regdeletekey",cmd_reg_delete_key, "<keyname> Registry Key Delete"},
|
||||||
{"regcreatekey",cmd_reg_create_key, "<keyname> [keyclass] Registry Key Create"},
|
{"regcreatekey",cmd_reg_create_key, "<keyname> [keyclass] Registry Key Create"},
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user