improving syntax / useability of rpcclient "shutdown" command:

rpcclient [-m messsage] [-t timeout] [-r or --reboot].
(This used to be commit 8b859797aaa1b88cb1a932fed97b4c3ea2dbdde2)

1 files changed, 24 insertions, 8 deletions

diff --git a/source3/rpcclient/cmd_reg.c b/source3/rpcclient/cmd_reg.c
index 95a439c462..ed337a59f5 100644
--- a/source3/rpcclient/cmd_reg.c
+++ b/source3/rpcclient/cmd_reg.c
@@ -921,24 +921,40 @@ void cmd_reg_shutdown(struct client_info *info)
 	fstring msg;
 	fstring tmp;
 	uint32 timeout = 20;
+	uint16 flgs = 0;
 
 	DEBUG(5, ("cmd_reg_shutdown: smb_cli->fd:%d\n", smb_cli->fd));
 
-	if (!next_token(NULL, msg, NULL, sizeof(msg)))
+	while (next_token(NULL, tmp, NULL, sizeof(tmp)))
 	{
-		msg[0] = 0;
-	}
-	else if (next_token(NULL, tmp, NULL, sizeof(tmp)))
-	{
-		timeout = atoi(tmp);
+		if (strequal(tmp, "-m"))
+		{
+			if (next_token(NULL, msg, NULL, sizeof(msg)))
+			{
+				continue;
+			}
+		}
+		else if (strequal(tmp, "-t"))
+		{
+			if (next_token(NULL, tmp, NULL, sizeof(tmp)))
+			{
+				timeout = atoi(tmp);
+				continue;
+			}
+		}
+		else if (strequal(tmp, "-r") || strequal(tmp, "--reboot"))
+		{
+			flgs = 0x100;
+			continue;
+		}
+		fprintf(out_hnd,"shutdown [-m msg] [-t timeout] [-r or --reboot]\n");
 	}
 
-
 	/* 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;
+	res = res ? do_reg_shutdown(smb_cli, fnum, msg, timeout, flgs) : False;
 
 	/* close the session */
 	cli_nt_session_close(smb_cli, fnum);