sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2024-12-25 23:21:54 +03:00
Code

rpcclient shutdown command

Browse Source
This commit is contained in:
Luke Leighton 0001-01-01 00:00:00 +00:00
parent ba5919bcae
commit 59f081069a
7 changed files with 203 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
source/include/proto.h
View File

@ -561,6 +561,8 @@ BOOL string_set(char **dest,const char *src);
void string_sub(char *s,const char *pattern,const char *insert);
void all_string_sub(char *s,const char *pattern,const char *insert);
void split_at_last_component(char *path, char *front, char sep, char *back);
char *bit_field_to_str(uint32 type, struct field_info *bs);
char *enum_field_to_str(uint32 type, struct field_info *bs, BOOL first_default);
/*The following definitions come from lib/util_unistr.c */
@ -568,11 +570,11 @@ char *ascii_to_unibuf(char *dest, const char *src, int maxlen);
void unibuf_to_ascii(char *dest, const char *src, int maxlen);
void ascii_to_unistr(uint16 *dest, const char *src, int maxlen);
void unistr_to_ascii(char *dest, const uint16 *src, int len);
void unistr2_to_ascii(char *dest, const UNISTR2 *str, int destlen);
void unistr2_to_ascii(char *dest, const UNISTR2 *str, int maxlen);
char *skip_unibuf(char *srcbuf, int len);
char *uni_strncpy(char *destbuf, const char *srcbuf, int len);
uint32 buffer2_to_uint32(const BUFFER2 *str);
void buffer2_to_multistr(char *dest, const BUFFER2 *str, int destlen);
void buffer2_to_multistr(char *dest, const BUFFER2 *str, int maxlen);
/*The following definitions come from libsmb/clientgen.c */
@ -1632,6 +1634,8 @@ BOOL do_reg_open_entry(struct cli_state *cli, uint16 fnum, POLICY_HND *hnd,
char *key_name, uint32 unk_0,
POLICY_HND *key_hnd);
BOOL do_reg_close(struct cli_state *cli, uint16 fnum, POLICY_HND *hnd);
BOOL do_reg_shutdown(struct cli_state *cli, uint16 fnum,
char *msg, uint32 timeout, uint16 flags);
/*The following definitions come from rpc_client/cli_samr.c */
@ -2092,6 +2096,10 @@ void reg_io_q_open_entry(char *desc, REG_Q_OPEN_ENTRY *r_q, prs_struct *ps, int
void make_reg_r_open_entry(REG_R_OPEN_ENTRY *r_r,
POLICY_HND *pol, uint32 status);
void reg_io_r_open_entry(char *desc, REG_R_OPEN_ENTRY *r_r, prs_struct *ps, int depth);
void make_reg_q_shutdown(REG_Q_SHUTDOWN *q_i,
char *msg, uint32 timeout, uint16 flags);
void reg_io_q_shutdown(char *desc, REG_Q_SHUTDOWN *q_q, prs_struct *ps, int depth);
void reg_io_r_shutdown(char *desc, REG_R_SHUTDOWN *r_q, prs_struct *ps, int depth);
/*The following definitions come from rpc_parse/parse_rpc.c */
@ -2677,6 +2685,7 @@ void cmd_reg_delete_key(struct client_info *info);
void cmd_reg_create_key(struct client_info *info);
void cmd_reg_test_key_sec(struct client_info *info);
void cmd_reg_get_key_sec(struct client_info *info);
void cmd_reg_shutdown(struct client_info *info);
/*The following definitions come from rpcclient/cmd_samr.c */

21
source/include/rpc_reg.h
View File

@ -42,6 +42,7 @@
#define REG_OPEN_ENTRY 0x0f
#define REG_INFO 0x11
#define REG_CLOSE 0x05
#define REG_SHUTDOWN 0x18
#define HKEY_LOCAL_MACHINE 0x80000000
#define HKEY_USERS 0x80000003
@ -461,6 +462,26 @@ typedef struct r_reg_open_entry_info
} REG_R_OPEN_ENTRY;
/* REG_Q_SHUTDOWN */
typedef struct q_reg_shutdown_info
{
uint32 ptr_0;
uint32 ptr_1;
uint32 ptr_2;
UNIHDR hdr_msg; /* shutdown message */
UNISTR2 uni_msg;
uint32 timeout; /* seconds */
uint16 flags;
} REG_Q_SHUTDOWN;
/* REG_R_SHUTDOWN */
typedef struct r_reg_shutdown_info
{
uint32 status; /* return status */
} REG_R_SHUTDOWN;
#endif /* _RPC_REG_H */

7
source/include/smb.h
View File

@ -1900,6 +1900,13 @@ struct nmb_name {
#define MAP_TO_GUEST_ON_BAD_USER 1
#define MAP_TO_GUEST_ON_BAD_PASSWORD 2
/* associate bit field or enumeration field with a string */
struct field_info
{
uint32 bits;
char *str;
};
#endif /* _SMB_H */
/* _SMB_H */

55
source/rpc_client/cli_reg.c
View File

@ -1041,4 +1041,59 @@ BOOL do_reg_close(struct cli_state *cli, uint16 fnum, POLICY_HND *hnd)
return valid_close;
}
/****************************************************************************
do a REG Shutdown Server
****************************************************************************/
BOOL do_reg_shutdown(struct cli_state *cli, uint16 fnum,
char *msg, uint32 timeout, uint16 flags)
{
prs_struct rbuf;
prs_struct buf;
REG_Q_SHUTDOWN q_o;
BOOL valid_shutdown = False;
if (msg == NULL) return False;
prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
prs_init(&rbuf, 0 , 4, SAFETY_MARGIN, True );
/* create and send a MSRPC command with api REG_SHUTDOWN */
DEBUG(4,("REG Shutdown: (timeout: %d secs) %s\n", timeout, msg));
make_reg_q_shutdown(&q_o, msg, timeout, flags);
/* turn parameters into data stream */
reg_io_q_shutdown("", &q_o, &buf, 0);
/* send the data on \PIPE\ */
if (rpc_api_pipe_req(cli, fnum, REG_SHUTDOWN, &buf, &rbuf))
{
REG_R_SHUTDOWN r_o;
BOOL p;
ZERO_STRUCT(r_o);
reg_io_r_shutdown("", &r_o, &rbuf, 0);
p = rbuf.offset != 0;
if (p && r_o.status != 0)
{
/* report error code */
DEBUG(0,("REG_SHUTDOWN: %s\n", get_nt_error_msg(r_o.status)));
p = False;
}
if (p)
{
valid_shutdown = True;
}
}
prs_mem_free(&rbuf);
prs_mem_free(&buf );
return valid_shutdown;
}

63
source/rpc_parse/parse_reg.c
View File

@ -1180,3 +1180,66 @@ void reg_io_r_open_entry(char *desc, REG_R_OPEN_ENTRY *r_r, prs_struct *ps, int
prs_uint32("status", ps, depth, &(r_r->status));
}
/*******************************************************************
makes a structure.
********************************************************************/
void make_reg_q_shutdown(REG_Q_SHUTDOWN *q_i,
char *msg, uint32 timeout, uint16 flags)
{
int len = strlen(msg) + 1;
if (q_i == NULL) return;
ZERO_STRUCTP(q_i);
q_i->ptr_0 = 1;
q_i->ptr_1 = 1;
q_i->ptr_2 = 1;
make_uni_hdr(&q_i->hdr_msg, len, len, 1);
make_unistr2(&(q_i->uni_msg), msg, len);
q_i->timeout = timeout;
q_i->flags = flags;
}
/*******************************************************************
reads or writes a structure.
********************************************************************/
void reg_io_q_shutdown(char *desc, REG_Q_SHUTDOWN *q_q, prs_struct *ps, int depth)
{
if (q_q == NULL) return;
prs_debug(ps, depth, desc, "reg_io_q_shutdown");
depth++;
prs_align(ps);
prs_uint32("ptr_0", ps, depth, &(q_q->ptr_0));
prs_uint32("ptr_1", ps, depth, &(q_q->ptr_1));
prs_uint32("ptr_2", ps, depth, &(q_q->ptr_2));
smb_io_unihdr ("hdr_msg", &(q_q->hdr_msg), ps, depth);
smb_io_unistr2("uni_msg", &(q_q->uni_msg), q_q->hdr_msg.buffer, ps, depth);
prs_align(ps);
prs_uint32("timeout", ps, depth, &(q_q->timeout));
prs_uint16("flags ", ps, depth, &(q_q->flags ));
prs_align(ps);
}
/*******************************************************************
reads or writes a structure.
********************************************************************/
void reg_io_r_shutdown(char *desc, REG_R_SHUTDOWN *r_q, prs_struct *ps, int depth)
{
if (r_q == NULL) return;
prs_debug(ps, depth, desc, "reg_io_r_shutdown");
depth++;
prs_align(ps);
prs_uint32("status", ps, depth, &(r_q->status));
}

45
source/rpcclient/cmd_reg.c
View File

@ -918,3 +918,48 @@ void cmd_reg_get_key_sec(struct client_info *info)
}
}
/****************************************************************************
nt registry shutdown
****************************************************************************/
void cmd_reg_shutdown(struct client_info *info)
{
uint16 fnum;
BOOL res = True;
fstring msg;
fstring tmp;
uint32 timeout = 20;
DEBUG(5, ("cmd_reg_shutdown: smb_cli->fd:%d\n", smb_cli->fd));
if (!next_token(NULL, msg, NULL, sizeof(msg)))
{
msg[0] = 0;
}
else if (next_token(NULL, tmp, NULL, sizeof(tmp)))
{
timeout = atoi(tmp);
}
/* open WINREG session. */
res = res ? cli_nt_session_open(smb_cli, PIPE_WINREG, &fnum) : False;
/* create an entry */
res = res ? do_reg_shutdown(smb_cli, fnum, msg, timeout, 1) : False;
/* close the session */
cli_nt_session_close(smb_cli, fnum);
if (res)
{
DEBUG(5,("cmd_reg_shutdown: query succeeded\n"));
fprintf(out_hnd,"OK\n");
}
else
{
DEBUG(5,("cmd_reg_shutdown: query failed\n"));
fprintf(out_hnd,"Failed\n");
}
}

1
source/rpcclient/rpcclient.c
View File

@ -109,6 +109,7 @@ struct
{"regenum", cmd_reg_enum, "<keyname> Registry Enumeration (keys, values)"},
{"regdeletekey",cmd_reg_delete_key, "<keyname> Registry Key Delete"},
{"regcreatekey",cmd_reg_create_key, "<keyname> [keyclass] Registry Key Create"},
{"shutdown",cmd_reg_shutdown, "[message] [timeout] Server Shutdown"},
{"regquerykey",cmd_reg_query_key, "<keyname> Registry Key Query"},
{"regdeleteval",cmd_reg_delete_val, "<valname> Registry Value Delete"},
{"regcreateval",cmd_reg_create_val, "<valname> <valtype> <value> Registry Key Create"},