Fix PReg write support.
[Samba/eduardoll.git] / source4 / rpc_server / remote / dcesrv_remote.c
blobcd32160d88957d1c1f8d46325c587fac3a5ef223
1 /*
2 Unix SMB/CIFS implementation.
3 remote dcerpc operations
5 Copyright (C) Stefan (metze) Metzmacher 2004
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "includes.h"
22 #include "rpc_server/dcerpc_server.h"
23 #include "auth/auth.h"
24 #include "auth/credentials/credentials.h"
25 #include "librpc/ndr/ndr_table.h"
26 #include "param/param.h"
29 struct dcesrv_remote_private {
30 struct dcerpc_pipe *c_pipe;
33 static NTSTATUS remote_op_reply(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, void *r)
35 return NT_STATUS_OK;
38 static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct dcesrv_interface *iface)
40 NTSTATUS status;
41 const struct ndr_interface_table *table;
42 struct dcesrv_remote_private *private;
43 const char *binding = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "binding");
44 const char *user, *pass, *domain;
45 struct cli_credentials *credentials;
46 bool machine_account;
48 machine_account = lp_parm_bool(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "use_machine_account", false);
50 private = talloc(dce_call->conn, struct dcesrv_remote_private);
51 if (!private) {
52 return NT_STATUS_NO_MEMORY;
55 private->c_pipe = NULL;
56 dce_call->context->private = private;
58 if (!binding) {
59 DEBUG(0,("You must specify a DCE/RPC binding string\n"));
60 return NT_STATUS_INVALID_PARAMETER;
63 user = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "user");
64 pass = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "password");
65 domain = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dceprc_remote", "domain");
67 table = ndr_table_by_uuid(&iface->syntax_id.uuid); /* FIXME: What about if_version ? */
68 if (!table) {
69 dce_call->fault_code = DCERPC_FAULT_UNK_IF;
70 return NT_STATUS_NET_WRITE_FAULT;
73 if (user && pass) {
74 DEBUG(5, ("dcerpc_remote: RPC Proxy: Using specified account\n"));
75 credentials = cli_credentials_init(private);
76 if (!credentials) {
77 return NT_STATUS_NO_MEMORY;
79 cli_credentials_set_conf(credentials, dce_call->conn->dce_ctx->lp_ctx);
80 cli_credentials_set_username(credentials, user, CRED_SPECIFIED);
81 if (domain) {
82 cli_credentials_set_domain(credentials, domain, CRED_SPECIFIED);
84 cli_credentials_set_password(credentials, pass, CRED_SPECIFIED);
85 } else if (machine_account) {
86 DEBUG(5, ("dcerpc_remote: RPC Proxy: Using machine account\n"));
87 credentials = cli_credentials_init(private);
88 cli_credentials_set_conf(credentials, dce_call->conn->dce_ctx->lp_ctx);
89 if (domain) {
90 cli_credentials_set_domain(credentials, domain, CRED_SPECIFIED);
92 status = cli_credentials_set_machine_account(credentials, dce_call->conn->dce_ctx->lp_ctx);
93 if (!NT_STATUS_IS_OK(status)) {
94 return status;
96 } else if (dce_call->conn->auth_state.session_info->credentials) {
97 DEBUG(5, ("dcerpc_remote: RPC Proxy: Using delegated credentials\n"));
98 credentials = dce_call->conn->auth_state.session_info->credentials;
99 } else {
100 DEBUG(1,("dcerpc_remote: RPC Proxy: You must supply binding, user and password or have delegated credentials\n"));
101 return NT_STATUS_INVALID_PARAMETER;
104 status = dcerpc_pipe_connect(private,
105 &(private->c_pipe), binding, table,
106 credentials, dce_call->event_ctx,
107 dce_call->conn->dce_ctx->lp_ctx);
109 talloc_free(credentials);
110 if (!NT_STATUS_IS_OK(status)) {
111 return status;
114 return NT_STATUS_OK;
117 static void remote_op_unbind(struct dcesrv_connection_context *context, const struct dcesrv_interface *iface)
119 struct dcesrv_remote_private *private = (struct dcesrv_remote_private *)context->private;
121 talloc_free(private->c_pipe);
123 return;
126 static NTSTATUS remote_op_ndr_pull(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct ndr_pull *pull, void **r)
128 enum ndr_err_code ndr_err;
129 const struct ndr_interface_table *table = (const struct ndr_interface_table *)dce_call->context->iface->private;
130 uint16_t opnum = dce_call->pkt.u.request.opnum;
132 dce_call->fault_code = 0;
134 if (opnum >= table->num_calls) {
135 dce_call->fault_code = DCERPC_FAULT_OP_RNG_ERROR;
136 return NT_STATUS_NET_WRITE_FAULT;
139 *r = talloc_size(mem_ctx, table->calls[opnum].struct_size);
140 if (!*r) {
141 return NT_STATUS_NO_MEMORY;
144 /* unravel the NDR for the packet */
145 ndr_err = table->calls[opnum].ndr_pull(pull, NDR_IN, *r);
146 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
147 dcerpc_log_packet(table, opnum, NDR_IN,
148 &dce_call->pkt.u.request.stub_and_verifier);
149 dce_call->fault_code = DCERPC_FAULT_NDR;
150 return NT_STATUS_NET_WRITE_FAULT;
153 return NT_STATUS_OK;
156 static NTSTATUS remote_op_dispatch(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, void *r)
158 struct dcesrv_remote_private *private = dce_call->context->private;
159 uint16_t opnum = dce_call->pkt.u.request.opnum;
160 const struct ndr_interface_table *table = dce_call->context->iface->private;
161 const struct ndr_interface_call *call;
162 const char *name;
164 name = table->calls[opnum].name;
165 call = &table->calls[opnum];
167 if (private->c_pipe->conn->flags & DCERPC_DEBUG_PRINT_IN) {
168 ndr_print_function_debug(call->ndr_print, name, NDR_IN | NDR_SET_VALUES, r);
171 private->c_pipe->conn->flags |= DCERPC_NDR_REF_ALLOC;
173 /* we didn't use the return code of this function as we only check the last_fault_code */
174 dcerpc_ndr_request(private->c_pipe, NULL, table, opnum, mem_ctx,r);
176 dce_call->fault_code = private->c_pipe->last_fault_code;
177 if (dce_call->fault_code != 0) {
178 DEBUG(0,("dcesrv_remote: call[%s] failed with: %s!\n",name, dcerpc_errstr(mem_ctx, dce_call->fault_code)));
179 return NT_STATUS_NET_WRITE_FAULT;
182 if ((dce_call->fault_code == 0) &&
183 (private->c_pipe->conn->flags & DCERPC_DEBUG_PRINT_OUT)) {
184 ndr_print_function_debug(call->ndr_print, name, NDR_OUT, r);
187 return NT_STATUS_OK;
190 static NTSTATUS remote_op_ndr_push(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct ndr_push *push, const void *r)
192 enum ndr_err_code ndr_err;
193 const struct ndr_interface_table *table = dce_call->context->iface->private;
194 uint16_t opnum = dce_call->pkt.u.request.opnum;
196 /* unravel the NDR for the packet */
197 ndr_err = table->calls[opnum].ndr_push(push, NDR_OUT, r);
198 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
199 dce_call->fault_code = DCERPC_FAULT_NDR;
200 return NT_STATUS_NET_WRITE_FAULT;
203 return NT_STATUS_OK;
206 static NTSTATUS remote_register_one_iface(struct dcesrv_context *dce_ctx, const struct dcesrv_interface *iface)
208 int i;
209 const struct ndr_interface_table *table = iface->private;
211 for (i=0;i<table->endpoints->count;i++) {
212 NTSTATUS ret;
213 const char *name = table->endpoints->names[i];
215 ret = dcesrv_interface_register(dce_ctx, name, iface, NULL);
216 if (!NT_STATUS_IS_OK(ret)) {
217 DEBUG(1,("remote_op_init_server: failed to register endpoint '%s'\n",name));
218 return ret;
222 return NT_STATUS_OK;
225 static NTSTATUS remote_op_init_server(struct dcesrv_context *dce_ctx, const struct dcesrv_endpoint_server *ep_server)
227 int i;
228 const char **ifaces = str_list_make(dce_ctx, lp_parm_string(dce_ctx->lp_ctx, NULL, "dcerpc_remote", "interfaces"),NULL);
230 if (!ifaces) {
231 DEBUG(3,("remote_op_init_server: no interfaces configured\n"));
232 return NT_STATUS_OK;
235 for (i=0;ifaces[i];i++) {
236 NTSTATUS ret;
237 struct dcesrv_interface iface;
239 if (!ep_server->interface_by_name(&iface, ifaces[i])) {
240 DEBUG(0,("remote_op_init_server: failed to find interface = '%s'\n", ifaces[i]));
241 talloc_free(ifaces);
242 return NT_STATUS_UNSUCCESSFUL;
245 ret = remote_register_one_iface(dce_ctx, &iface);
246 if (!NT_STATUS_IS_OK(ret)) {
247 DEBUG(0,("remote_op_init_server: failed to register interface = '%s'\n", ifaces[i]));
248 talloc_free(ifaces);
249 return ret;
253 talloc_free(ifaces);
254 return NT_STATUS_OK;
257 static bool remote_fill_interface(struct dcesrv_interface *iface, const struct ndr_interface_table *if_tabl)
259 iface->name = if_tabl->name;
260 iface->syntax_id = if_tabl->syntax_id;
262 iface->bind = remote_op_bind;
263 iface->unbind = remote_op_unbind;
265 iface->ndr_pull = remote_op_ndr_pull;
266 iface->dispatch = remote_op_dispatch;
267 iface->reply = remote_op_reply;
268 iface->ndr_push = remote_op_ndr_push;
270 iface->private = if_tabl;
272 return true;
275 static bool remote_op_interface_by_uuid(struct dcesrv_interface *iface, const struct GUID *uuid, uint32_t if_version)
277 const struct ndr_interface_list *l;
279 for (l=ndr_table_list();l;l=l->next) {
280 if (l->table->syntax_id.if_version == if_version &&
281 GUID_equal(&l->table->syntax_id.uuid, uuid)==0) {
282 return remote_fill_interface(iface, l->table);
286 return false;
289 static bool remote_op_interface_by_name(struct dcesrv_interface *iface, const char *name)
291 const struct ndr_interface_table *tbl = ndr_table_by_name(name);
293 if (tbl)
294 return remote_fill_interface(iface, tbl);
296 return false;
299 NTSTATUS dcerpc_server_remote_init(void)
301 NTSTATUS ret;
302 struct dcesrv_endpoint_server ep_server;
304 ZERO_STRUCT(ep_server);
306 /* fill in our name */
307 ep_server.name = "remote";
309 /* fill in all the operations */
310 ep_server.init_server = remote_op_init_server;
312 ep_server.interface_by_uuid = remote_op_interface_by_uuid;
313 ep_server.interface_by_name = remote_op_interface_by_name;
315 /* register ourselves with the DCERPC subsystem. */
316 ret = dcerpc_register_ep_server(&ep_server);
317 if (!NT_STATUS_IS_OK(ret)) {
318 DEBUG(0,("Failed to register 'remote' endpoint server!\n"));
319 return ret;
322 /* We need the full DCE/RPC interface table */
323 ndr_table_init();
325 return ret;