rpcclient shutdown command

7 files changed, 203 insertions, 2 deletions

diff --git a/source/include/proto.h b/source/include/proto.h
index 1d3b041e4d7..a1b8f2eeacb 100644
--- a/source/include/proto.h
+++ b/source/include/proto.h
@@ -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  */
 
diff --git a/source/include/rpc_reg.h b/source/include/rpc_reg.h
index 77660525888..cc163540d73 100644
--- a/source/include/rpc_reg.h
+++ b/source/include/rpc_reg.h
@@ -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 */
 
diff --git a/source/include/smb.h b/source/include/smb.h
index 662ba245b1e..daf617bd7f0 100644
--- a/source/include/smb.h
+++ b/source/include/smb.h
@@ -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 */
diff --git a/source/rpc_client/cli_reg.c b/source/rpc_client/cli_reg.c
index 166617ff374..fc7f6a5b534 100644
--- a/source/rpc_client/cli_reg.c
+++ b/source/rpc_client/cli_reg.c
@@ -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;
+}
+
 
diff --git a/source/rpc_parse/parse_reg.c b/source/rpc_parse/parse_reg.c
index 8072ed2a64d..ecec4985284 100644
--- a/source/rpc_parse/parse_reg.c
+++ b/source/rpc_parse/parse_reg.c
@@ -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));
+}
+
diff --git a/source/rpcclient/cmd_reg.c b/source/rpcclient/cmd_reg.c
index f6a8a0a8986..54e726847a4 100644
--- a/source/rpcclient/cmd_reg.c
+++ b/source/rpcclient/cmd_reg.c
@@ -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");
+	}
+}
+
diff --git a/source/rpcclient/rpcclient.c b/source/rpcclient/rpcclient.c
index 79ab1b68a1f..c0ff8526f1a 100644
--- a/source/rpcclient/rpcclient.c
+++ b/source/rpcclient/rpcclient.c
@@ -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"},