2 * Unix SMB/CIFS implementation.
3 * Shell around net rpc subcommands
4 * Copyright (C) Volker Lendecke 2006
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "utils/net.h"
24 static NTSTATUS
rpc_sh_info(struct net_context
*c
,
25 TALLOC_CTX
*mem_ctx
, struct rpc_sh_ctx
*ctx
,
26 struct rpc_pipe_client
*pipe_hnd
,
27 int argc
, const char **argv
)
29 return rpc_info_internals(c
, ctx
->domain_sid
, ctx
->domain_name
,
30 ctx
->cli
, pipe_hnd
, mem_ctx
,
34 static struct rpc_sh_ctx
*this_ctx
;
36 static char **completion_fn(const char *text
, int start
, int end
)
46 ADD_TO_ARRAY(NULL
, char *, SMB_STRDUP(text
), &cmds
, &n_cmds
);
48 for (c
= this_ctx
->cmds
; c
->name
!= NULL
; c
++) {
49 bool match
= (strncmp(text
, c
->name
, strlen(text
)) == 0);
52 ADD_TO_ARRAY(NULL
, char *, SMB_STRDUP(c
->name
),
63 ADD_TO_ARRAY(NULL
, char *, NULL
, &cmds
, &n_cmds
);
67 static NTSTATUS
net_sh_run(struct net_context
*c
,
68 struct rpc_sh_ctx
*ctx
, struct rpc_sh_cmd
*cmd
,
69 int argc
, const char **argv
)
72 struct rpc_pipe_client
*pipe_hnd
;
75 mem_ctx
= talloc_new(ctx
);
76 if (mem_ctx
== NULL
) {
77 d_fprintf(stderr
, "talloc_new failed\n");
78 return NT_STATUS_NO_MEMORY
;
81 status
= cli_rpc_pipe_open_noauth(ctx
->cli
,
82 cli_get_iface(cmd
->pipe_idx
),
84 if (!NT_STATUS_IS_OK(status
)) {
85 d_fprintf(stderr
, "Could not open pipe: %s\n",
90 status
= cmd
->fn(c
, mem_ctx
, ctx
, pipe_hnd
, argc
, argv
);
92 TALLOC_FREE(pipe_hnd
);
94 talloc_destroy(mem_ctx
);
99 static bool net_sh_process(struct net_context
*c
,
100 struct rpc_sh_ctx
*ctx
,
101 int argc
, const char **argv
)
103 struct rpc_sh_cmd
*cmd
;
104 struct rpc_sh_ctx
*new_ctx
;
111 if (ctx
== this_ctx
) {
113 /* We've been called from the cmd line */
114 if (strequal(argv
[0], "..") &&
115 (this_ctx
->parent
!= NULL
)) {
116 new_ctx
= this_ctx
->parent
;
117 TALLOC_FREE(this_ctx
);
123 if (strequal(argv
[0], "exit") || strequal(argv
[0], "quit")) {
127 if (strequal(argv
[0], "help") || strequal(argv
[0], "?")) {
128 for (cmd
= ctx
->cmds
; cmd
->name
!= NULL
; cmd
++) {
129 if (ctx
!= this_ctx
) {
130 d_printf("%s ", ctx
->whoami
);
132 d_printf("%-15s %s\n", cmd
->name
, cmd
->help
);
137 for (cmd
= ctx
->cmds
; cmd
->name
!= NULL
; cmd
++) {
138 if (strequal(cmd
->name
, argv
[0])) {
143 if (cmd
->name
== NULL
) {
145 d_fprintf(stderr
, "%s: unknown cmd\n", argv
[0]);
149 new_ctx
= TALLOC_P(ctx
, struct rpc_sh_ctx
);
150 if (new_ctx
== NULL
) {
151 d_fprintf(stderr
, "talloc failed\n");
154 new_ctx
->cli
= ctx
->cli
;
155 new_ctx
->whoami
= talloc_asprintf(new_ctx
, "%s %s",
156 ctx
->whoami
, cmd
->name
);
157 new_ctx
->thiscmd
= talloc_strdup(new_ctx
, cmd
->name
);
159 if (cmd
->sub
!= NULL
) {
160 new_ctx
->cmds
= cmd
->sub(c
, new_ctx
, ctx
);
162 new_ctx
->cmds
= NULL
;
165 new_ctx
->parent
= ctx
;
166 new_ctx
->domain_name
= ctx
->domain_name
;
167 new_ctx
->domain_sid
= ctx
->domain_sid
;
172 if (cmd
->sub
!= NULL
) {
177 return net_sh_process(c
, new_ctx
, argc
, argv
);
180 status
= net_sh_run(c
, new_ctx
, cmd
, argc
, argv
);
182 if (!NT_STATUS_IS_OK(status
)) {
183 d_fprintf(stderr
, "%s failed: %s\n", new_ctx
->whoami
,
190 static struct rpc_sh_cmd sh_cmds
[6] = {
192 { "info", NULL
, PI_SAMR
, rpc_sh_info
,
193 "Print information about the domain connected to" },
195 { "rights", net_rpc_rights_cmds
, 0, NULL
,
196 "List/Grant/Revoke user rights" },
198 { "share", net_rpc_share_cmds
, 0, NULL
,
199 "List/Add/Remove etc shares" },
201 { "user", net_rpc_user_cmds
, 0, NULL
,
202 "List/Add/Remove user info" },
204 { "account", net_rpc_acct_cmds
, 0, NULL
,
205 "Show/Change account policy settings" },
207 { NULL
, NULL
, 0, NULL
, NULL
}
210 int net_rpc_shell(struct net_context
*c
, int argc
, const char **argv
)
213 struct rpc_sh_ctx
*ctx
;
215 if (argc
!= 0 || c
->display_usage
) {
221 ctx
= TALLOC_P(NULL
, struct rpc_sh_ctx
);
223 d_fprintf(stderr
, "talloc failed\n");
227 status
= net_make_ipc_connection(c
, 0, &(ctx
->cli
));
228 if (!NT_STATUS_IS_OK(status
)) {
229 d_fprintf(stderr
, "Could not open connection: %s\n",
235 ctx
->whoami
= "net rpc";
238 status
= net_get_remote_domain_sid(ctx
->cli
, ctx
, &ctx
->domain_sid
,
240 if (!NT_STATUS_IS_OK(status
)) {
244 d_printf("Talking to domain %s (%s)\n", ctx
->domain_name
,
245 sid_string_tos(ctx
->domain_sid
));
254 if (asprintf(&prompt
, "%s> ", this_ctx
->whoami
) < 0) {
258 line
= smb_readline(prompt
, NULL
, completion_fn
);
265 ret
= poptParseArgvString(line
, &argc
, &argv
);
266 if (ret
== POPT_ERROR_NOARG
) {
271 d_fprintf(stderr
, "cmdline invalid: %s\n",
277 if ((line
[0] != '\n') &&
278 (!net_sh_process(c
, this_ctx
, argc
, argv
))) {
285 cli_shutdown(ctx
->cli
);