Add netsessdel command to rpcclient.

[sfrench/samba-autobuild/.git] / source3 / rpcclient / cmd_wkssvc.c

diff --git a/source3/rpcclient/cmd_wkssvc.c b/source3/rpcclient/cmd_wkssvc.c
index d32a4c591f7d8cd7787e1fe03fd024cc9317a5b3..68f408cf4815126fcdbe5c323417085505935fc9 100644 (file)
--- a/source3/rpcclient/cmd_wkssvc.c
+++ b/source3/rpcclient/cmd_wkssvc.c
@@ -64,20 +64,25 @@ static WERROR cmd_wkssvc_getjoininformation(struct rpc_pipe_client *cli,
        const char *name_buffer;
        enum wkssvc_NetJoinStatus name_type;
        NTSTATUS status;
+       WERROR werr;
 
        server_name = cli->cli->desthost;
        name_buffer = "";
 
        status = rpccli_wkssvc_NetrGetJoinInformation(cli, mem_ctx,
-                                                     server_name, &name_buffer,
-                                                     &name_type);
+                                                     server_name,
+                                                     &name_buffer,
+                                                     &name_type,
+                                                     &werr);
        if (!NT_STATUS_IS_OK(status)) {
                return ntstatus_to_werror(status);
        }
 
-       printf("%s (%d)\n", name_buffer, name_type);
+       if (W_ERROR_IS_OK(werr)) {
+               printf("%s (%d)\n", name_buffer, name_type);
+       }
 
-       return WERR_OK;
+       return werr;
 }
 
 static WERROR cmd_wkssvc_messagebuffersend(struct rpc_pipe_client *cli,
@@ -90,12 +95,17 @@ static WERROR cmd_wkssvc_messagebuffersend(struct rpc_pipe_client *cli,
        const char *message_sender_name = cli->cli->desthost;
        smb_ucs2_t *message_buffer = NULL;
        size_t message_size = 0;
+       const char *message = "my message";
        NTSTATUS status;
        WERROR werr;
 
+       if (argc > 1) {
+               message = argv[1];
+       }
+
        message_size = push_ucs2_talloc(mem_ctx,
                                        &message_buffer,
-                                       "my message");
+                                       message);
        if (message_size == -1) {
                return WERR_NOMEM;
        }
@@ -114,11 +124,48 @@ static WERROR cmd_wkssvc_messagebuffersend(struct rpc_pipe_client *cli,
        return werr;
 }
 
+static WERROR cmd_wkssvc_enumeratecomputernames(struct rpc_pipe_client *cli,
+                                               TALLOC_CTX *mem_ctx,
+                                               int argc,
+                                               const char **argv)
+{
+       const char *server_name;
+       enum wkssvc_ComputerNameType name_type = NetAllComputerNames;
+       NTSTATUS status;
+       struct wkssvc_ComputerNamesCtr *ctr = NULL;
+       WERROR werr;
+
+       server_name = cli->cli->desthost;
+
+       if (argc >= 2) {
+               name_type = atoi(argv[1]);
+       }
+
+       status = rpccli_wkssvc_NetrEnumerateComputerNames(cli, mem_ctx,
+                                                         server_name,
+                                                         name_type, 0,
+                                                         &ctr,
+                                                         &werr);
+       if (!NT_STATUS_IS_OK(status)) {
+               return ntstatus_to_werror(status);
+       }
+
+       if (W_ERROR_IS_OK(werr)) {
+               int i=0;
+               for (i = 0; i < ctr->count; i++) {
+                       printf("name: %d %s\n", i, ctr->computer_name->string);
+               }
+       }
+
+       return werr;
+}
+
 struct cmd_set wkssvc_commands[] = {
 
        { "WKSSVC" },
        { "wkssvc_wkstagetinfo", RPC_RTYPE_WERROR, NULL, cmd_wkssvc_wkstagetinfo, PI_WKSSVC, NULL, "Query WKSSVC Workstation Information", "" },
        { "wkssvc_getjoininformation", RPC_RTYPE_WERROR, NULL, cmd_wkssvc_getjoininformation, PI_WKSSVC, NULL, "Query WKSSVC Join Information", "" },
        { "wkssvc_messagebuffersend", RPC_RTYPE_WERROR, NULL, cmd_wkssvc_messagebuffersend, PI_WKSSVC, NULL, "Send WKSSVC message", "" },
+       { "wkssvc_enumeratecomputernames", RPC_RTYPE_WERROR, NULL, cmd_wkssvc_enumeratecomputernames, PI_WKSSVC, NULL, "Enumerate WKSSVC computer names", "" },
        { NULL }
 };