rpcclient shutdown command

[samba.git] / source3 / rpc_client / cli_reg.c

diff --git a/source3/rpc_client/cli_reg.c b/source3/rpc_client/cli_reg.c
index 166617ff374f0fc5f95eb83e0d178c2e4ac8022e..fc7f6a5b534c26f4a6c8cdc04ee9c33e2b884835 100644 (file)
--- a/source3/rpc_client/cli_reg.c
+++ b/source3/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;
+}
+