sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-19 10:03:58 +03:00
Code

added yet another rpcclient command: svcstart <service name> [arg0] [arg1] ...

Browse Source 
works with command-line completion on the service name (ohh yesss, this
is becoming my favourite bit of functionality-on-the-side hee hee :)

had to fix the svc_io_q_start_service() code which was missing the
ptr_argv[] array in between the array-size and the UNISTR2-array.
i.e it's actually an array of _pointers_ to unicode strings...
This commit is contained in:
Luke Leighton -
parent 5ca07721ba
commit 2903f22e7e
6 changed files with 184 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
source/include/proto.h
View File

@ -2051,6 +2051,10 @@ BOOL svc_enum_svcs(struct cli_state *cli, uint16 fnum,
uint32 *buf_size, uint32 *resume_hnd,
uint32 *dos_error,
ENUM_SRVC_STATUS **svcs, uint32 *num_svcs);
BOOL svc_start_service(struct cli_state *cli, uint16 fnum,
POLICY_HND *hnd,
uint32 argc,
char **argv);
BOOL svc_query_svc_cfg(struct cli_state *cli, uint16 fnum,
POLICY_HND *hnd,
QUERY_SERVICE_CONFIG *cfg,
@ -3055,6 +3059,9 @@ BOOL svc_io_q_open_service(char *desc, SVC_Q_OPEN_SERVICE *q_u, prs_struct *ps,
BOOL make_svc_r_open_service(SVC_R_OPEN_SERVICE *r_u, POLICY_HND *hnd,
uint32 status) ;
BOOL svc_io_r_open_service(char *desc, SVC_R_OPEN_SERVICE *r_u, prs_struct *ps, int depth);
BOOL make_svc_q_start_service(SVC_Q_START_SERVICE *q_c, POLICY_HND *hnd,
uint32 argc,
char **argv);
BOOL svc_io_q_start_service(char *desc, SVC_Q_START_SERVICE *q_s, prs_struct *ps, int depth);
BOOL svc_io_r_start_service(char *desc, SVC_R_START_SERVICE *r_s, prs_struct *ps, int depth);
BOOL make_svc_query_svc_cfg(QUERY_SERVICE_CONFIG *q_u,
@ -3315,6 +3322,7 @@ BOOL msrpc_svc_enum(struct client_info *info,
SVC_INFO_FN(info_fn),
SVC_QUERY_FN(query_fn));
void cmd_svc_enum(struct client_info *info);
void cmd_svc_start(struct client_info *info);
/*The following definitions come from rpcclient/cmd_wkssvc.c */

5
source/include/rpc_svcctl.h
View File

@ -73,7 +73,7 @@ typedef struct r_svc_open_service_info
} SVC_R_OPEN_SERVICE;
#define MAX_SVC_ARGS 4
#define MAX_SVC_ARGS 10
/* SVC_Q_START_SERVICE */
typedef struct q_svc_start_service_info
@ -81,8 +81,9 @@ typedef struct q_svc_start_service_info
POLICY_HND pol;
uint32 argc;
uint32 ptr_argv;
uint32 ptr_args;
uint32 argc2;
uint32 ptr_argv[MAX_SVC_ARGS];
UNISTR2 argv[MAX_SVC_ARGS];
} SVC_Q_START_SERVICE;

59
source/rpc_client/cli_svcctl.c
View File

@ -229,6 +229,65 @@ BOOL svc_enum_svcs(struct cli_state *cli, uint16 fnum,
}
/****************************************************************************
do a SVC Start Service
****************************************************************************/
BOOL svc_start_service(struct cli_state *cli, uint16 fnum,
POLICY_HND *hnd,
uint32 argc,
char **argv)
{
prs_struct rbuf;
prs_struct buf;
SVC_Q_START_SERVICE q_c;
BOOL valid_cfg = False;
if (hnd == NULL) return False;
/* create and send a MSRPC command with api SVC_START_SERVICE */
prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
prs_init(&rbuf, 0 , 4, SAFETY_MARGIN, True );
DEBUG(4,("SVC Start Service\n"));
/* store the parameters */
make_svc_q_start_service(&q_c, hnd, argc, argv);
/* turn parameters into data stream */
svc_io_q_start_service("", &q_c, &buf, 0);
/* send the data on \PIPE\ */
if (rpc_api_pipe_req(cli, fnum, SVC_START_SERVICE, &buf, &rbuf))
{
SVC_R_START_SERVICE r_c;
BOOL p;
ZERO_STRUCT (r_c);
svc_io_r_start_service("", &r_c, &rbuf, 0);
p = rbuf.offset != 0;
if (p && r_c.status != 0)
{
/* report error code */
DEBUG(1,("SVC_START_SERVICE: %s\n", get_nt_error_msg(r_c.status)));
p = False;
}
if (p)
{
valid_cfg = True;
}
}
prs_mem_free(&rbuf);
prs_mem_free(&buf );
return valid_cfg;
}
/****************************************************************************
do a SVC Query Service Config
****************************************************************************/

40
source/rpc_parse/parse_svc.c
View File

@ -178,6 +178,34 @@ BOOL svc_io_r_open_service(char *desc, SVC_R_OPEN_SERVICE *r_u, prs_struct *ps,
return True;
}
/*******************************************************************
makes an SVC_Q_START_SERVICE structure.
********************************************************************/
BOOL make_svc_q_start_service(SVC_Q_START_SERVICE *q_c, POLICY_HND *hnd,
uint32 argc,
char **argv)
{
uint32 i;
if (q_c == NULL || hnd == NULL) return False;
DEBUG(5,("make_svc_q_query_svc_config\n"));
memcpy(&(q_c->pol), hnd, sizeof(q_c->pol));
q_c->argc = argc;
q_c->ptr_args = 1;
q_c->argc2 = argc;
for (i = 0; i < argc; i++)
{
size_t len_argv = argv[i] != NULL ? strlen(argv[i])+1 : 0;
q_c->ptr_argv[i] = argv[i] != NULL ? 1 : 0;
make_unistr2(&(q_c->argv[i]), argv[i], len_argv);
}
return True;
}
/*******************************************************************
reads or writes a SVC_Q_START_SERVICE structure.
********************************************************************/
@ -193,9 +221,9 @@ BOOL svc_io_q_start_service(char *desc, SVC_Q_START_SERVICE *q_s, prs_struct *ps
prs_align(ps);
prs_uint32("argc ", ps, depth, &(q_s->argc ));
prs_uint32("ptr_argv", ps, depth, &(q_s->ptr_argv));
prs_uint32("ptr_args", ps, depth, &(q_s->ptr_args));
if (q_s->ptr_argv != 0)
if (q_s->ptr_args != 0)
{
uint32 i;
@ -203,12 +231,16 @@ BOOL svc_io_q_start_service(char *desc, SVC_Q_START_SERVICE *q_s, prs_struct *ps
if (q_s->argc2 > MAX_SVC_ARGS)
{
q_s->argc = q_s->argc2 = MAX_SVC_ARGS;
return False;
}
for (i = 0; i < q_s->argc2; i++)
{
smb_io_unistr2("", &(q_s->argv[i]), 1, ps, depth);
prs_uint32("", ps, depth, &(q_s->ptr_argv[i]));
}
for (i = 0; i < q_s->argc2; i++)
{
smb_io_unistr2("", &(q_s->argv[i]), q_s->ptr_argv[i], ps, depth);
prs_align(ps);
}
}

71
source/rpcclient/cmd_svcctl.c
View File

@ -241,3 +241,74 @@ void cmd_svc_enum(struct client_info *info)
}
}
/****************************************************************************
nt start service
****************************************************************************/
void cmd_svc_start(struct client_info *info)
{
uint16 fnum;
BOOL res = True;
BOOL res1 = True;
fstring svc_name;
fstring tmp;
BOOL res2 = True;
POLICY_HND pol_svc;
POLICY_HND pol_scm;
uint32 argc = 0;
char **argv = NULL;
fstring srv_name;
fstrcpy(srv_name, "\\\\");
fstrcat(srv_name, info->myhostname);
strupper(srv_name);
DEBUG(4,("cmd_svc_info: server:%s\n", srv_name));
if (!next_token(NULL, svc_name, NULL, sizeof(svc_name)))
{
report(out_hnd,"svcstart <service name> [arg 0] [arg 1]...]\n");
return;
}
while (next_token(NULL, tmp, NULL, sizeof(tmp)))
{
add_chars_to_array(&argc, &argv, tmp);
}
/* open SVCCTL session. */
res = res ? cli_nt_session_open(smb_cli, PIPE_SVCCTL, &fnum) : False;
/* open service control manager receive a policy handle */
res = res ? svc_open_sc_man(smb_cli, fnum,
srv_name, NULL, 0x80000000,
&pol_scm) : False;
res1 = res ? svc_open_service(smb_cli, fnum,
&pol_scm,
svc_name, 0x80000010,
&pol_svc) : False;
res2 = res1 ? svc_start_service(smb_cli, fnum,
&pol_svc, argc, argv) : False;
res1 = res1 ? svc_close(smb_cli, fnum, &pol_svc) : False;
res = res ? svc_close(smb_cli, fnum, &pol_scm) : False;
/* close the session */
cli_nt_session_close(smb_cli, fnum);
if (res2)
{
report(out_hnd,"Started Service %s\n", svc_name);
DEBUG(5,("cmd_svc_start: succeeded\n"));
}
else
report(out_hnd,"Failed Service Startup (%s)\n", svc_name);
{
DEBUG(5,("cmd_svc_start: failed\n"));
}
free_char_array(argc, argv);
}

7
source/rpcclient/rpcclient.c
View File

@ -146,6 +146,13 @@ commands[] =
{COMPL_SVCLST, COMPL_NONE}
},
{
"svcstart",
cmd_svc_start,
"<service> [arg 0] [arg 1] ... Start Service",
{COMPL_SVCLST, COMPL_NONE}
},
/*
* scheduler
*/