s4:rpc_server: only pass context to op_bind() hooks
[gd/samba-autobuild/.git] / source4 / rpc_server / remote / dcesrv_remote.c
index 9c4174be96bfe53d9ef29fdfea5e9f6469f19bb7..0d3b123bcd243e3e3b93954dc4b514a4f76aa931 100644 (file)
 */
 
 #include "includes.h"
+#include <tevent.h>
 #include "rpc_server/dcerpc_server.h"
 #include "auth/auth.h"
 #include "auth/credentials/credentials.h"
 #include "librpc/ndr/ndr_table.h"
 #include "param/param.h"
 
+NTSTATUS dcerpc_server_remote_init(TALLOC_CTX *ctx);
 
+#define DCESRV_REMOTE_ASSOC_MAGIC 0x782f50c4
+struct dcesrv_remote_assoc {
+       uint32_t assoc_group_id;
+};
+
+#define DCESRV_REMOTE_PRIVATE_MAGIC 0x7eceafa6
 struct dcesrv_remote_private {
        struct dcerpc_pipe *c_pipe;
 };
@@ -37,43 +45,78 @@ static NTSTATUS remote_op_reply(struct dcesrv_call_state *dce_call, TALLOC_CTX *
        return NT_STATUS_OK;
 }
 
-static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct dcesrv_interface *iface)
+static NTSTATUS remote_op_bind(struct dcesrv_connection_context *context,
+                              const struct dcesrv_interface *iface)
 {
-        NTSTATUS status;
-       const struct ndr_interface_table *table;
-       struct dcesrv_remote_private *priv;
-       const char *binding = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "binding");
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS remote_get_private(struct dcesrv_call_state *dce_call,
+                                  struct dcesrv_remote_private **_priv)
+{
+       const struct ndr_interface_table *table =
+               (const struct ndr_interface_table *)dce_call->context->iface->private_data;
+       struct dcesrv_remote_private *priv = NULL;
+       struct dcesrv_remote_assoc *assoc = NULL;
+       const char *binding = NULL;
        const char *user, *pass, *domain;
        struct cli_credentials *credentials;
-       bool must_free_credentials = true;
+       bool must_free_credentials = false;
        bool machine_account;
+       bool allow_anonymous;
        struct dcerpc_binding           *b;
        struct composite_context        *pipe_conn_req;
+       uint32_t flags = 0;
+       NTSTATUS status;
+
+       priv = dcesrv_iface_state_find_conn(dce_call,
+                                           DCESRV_REMOTE_PRIVATE_MAGIC,
+                                           struct dcesrv_remote_private);
+       if (priv != NULL) {
+               *_priv = priv;
+               return NT_STATUS_OK;
+       }
 
-       machine_account = lp_parm_bool(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "use_machine_account", false);
+       priv = talloc_zero(dce_call, struct dcesrv_remote_private);
+       if (priv == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-       priv = talloc(dce_call->conn, struct dcesrv_remote_private);
-       if (!priv) {
-               return NT_STATUS_NO_MEMORY;     
+       assoc = dcesrv_iface_state_find_assoc(dce_call,
+                                       DCESRV_REMOTE_ASSOC_MAGIC,
+                                       struct dcesrv_remote_assoc);
+       if (assoc == NULL) {
+               assoc = talloc_zero(dce_call, struct dcesrv_remote_assoc);
+               if (assoc == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
-       
-       priv->c_pipe = NULL;
-       dce_call->context->private_data = priv;
 
-       if (!binding) {
+       binding = lpcfg_parm_string(dce_call->conn->dce_ctx->lp_ctx,
+                                   NULL,
+                                   "dcerpc_remote",
+                                   "binding");
+       if (binding == NULL) {
                DEBUG(0,("You must specify a DCE/RPC binding string\n"));
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       user = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "user");
-       pass = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "password");
-       domain = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dceprc_remote", "domain");
+       user = lpcfg_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "user");
+       pass = lpcfg_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dcerpc_remote", "password");
+       domain = lpcfg_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "dceprc_remote", "domain");
 
-       table = ndr_table_by_uuid(&iface->syntax_id.uuid); /* FIXME: What about if_version ? */
-       if (!table) {
-               dce_call->fault_code = DCERPC_FAULT_UNK_IF;
-               return NT_STATUS_NET_WRITE_FAULT;
-       }
+       machine_account = lpcfg_parm_bool(dce_call->conn->dce_ctx->lp_ctx,
+                                         NULL,
+                                         "dcerpc_remote",
+                                         "use_machine_account",
+                                         false);
+       allow_anonymous = lpcfg_parm_bool(dce_call->conn->dce_ctx->lp_ctx,
+                                         NULL,
+                                         "dcerpc_remote",
+                                         "allow_anonymous_fallback",
+                                         false);
+
+       credentials = dcesrv_call_credentials(dce_call);
 
        if (user && pass) {
                DEBUG(5, ("dcerpc_remote: RPC Proxy: Using specified account\n"));
@@ -81,6 +124,7 @@ static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct
                if (!credentials) {
                        return NT_STATUS_NO_MEMORY;
                }
+               must_free_credentials = true;
                cli_credentials_set_conf(credentials, dce_call->conn->dce_ctx->lp_ctx);
                cli_credentials_set_username(credentials, user, CRED_SPECIFIED);
                if (domain) {
@@ -90,6 +134,10 @@ static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct
        } else if (machine_account) {
                DEBUG(5, ("dcerpc_remote: RPC Proxy: Using machine account\n"));
                credentials = cli_credentials_init(priv);
+               if (!credentials) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               must_free_credentials = true;
                cli_credentials_set_conf(credentials, dce_call->conn->dce_ctx->lp_ctx);
                if (domain) {
                        cli_credentials_set_domain(credentials, domain, CRED_SPECIFIED);
@@ -98,32 +146,59 @@ static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
-       } else if (dce_call->conn->auth_state.session_info->credentials) {
+       } else if (credentials != NULL) {
                DEBUG(5, ("dcerpc_remote: RPC Proxy: Using delegated credentials\n"));
-               credentials = dce_call->conn->auth_state.session_info->credentials;
-               must_free_credentials = false;
+       } else if (allow_anonymous) {
+               DEBUG(5, ("dcerpc_remote: RPC Proxy: Using anonymous\n"));
+               credentials = cli_credentials_init_anon(priv);
+               if (!credentials) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               must_free_credentials = true;
        } else {
                DEBUG(1,("dcerpc_remote: RPC Proxy: You must supply binding, user and password or have delegated credentials\n"));
                return NT_STATUS_INVALID_PARAMETER;
        }
 
        /* parse binding string to the structure */
-       status = dcerpc_parse_binding(dce_call->context, binding, &b);
+       status = dcerpc_parse_binding(priv, binding, &b);
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0, ("Failed to parse dcerpc binding '%s'\n", binding));
                return status;
        }
-       
-       DEBUG(3, ("Using binding %s\n", dcerpc_binding_string(dce_call->context, b)));
-       
+
        /* If we already have a remote association group ID, then use that */
-       if (dce_call->context->assoc_group->proxied_id != 0) {
-               b->assoc_group_id = dce_call->context->assoc_group->proxied_id;
+       if (assoc->assoc_group_id != 0) {
+               status = dcerpc_binding_set_assoc_group_id(b,
+                       assoc->assoc_group_id);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(0, ("dcerpc_binding_set_assoc_group_id() - %s'\n",
+                                 nt_errstr(status)));
+                       return status;
+               }
+       }
+
+       status = dcerpc_binding_set_abstract_syntax(b, &table->syntax_id);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0, ("dcerpc_binding_set_abstract_syntax() - %s'\n",
+                         nt_errstr(status)));
+               return status;
+       }
+
+       if (dce_call->conn->state_flags & DCESRV_CALL_STATE_FLAG_MULTIPLEXED) {
+               status = dcerpc_binding_set_flags(b, DCERPC_CONCURRENT_MULTIPLEX, 0);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(0, ("dcerpc_binding_set_flags(CONC_MPX) - %s'\n",
+                                 nt_errstr(status)));
+                       return status;
+               }
        }
 
-       pipe_conn_req = dcerpc_pipe_connect_b_send(dce_call->context, b, table,
+       DEBUG(3, ("Using binding %s\n", dcerpc_binding_string(dce_call->context, b)));
+
+       pipe_conn_req = dcerpc_pipe_connect_b_send(priv, b, table,
                                                   credentials, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx);
-       status = dcerpc_pipe_connect_b_recv(pipe_conn_req, dce_call->context, &(priv->c_pipe));
+       status = dcerpc_pipe_connect_b_recv(pipe_conn_req, priv, &(priv->c_pipe));
        
        if (must_free_credentials) {
                talloc_free(credentials);
@@ -133,24 +208,39 @@ static NTSTATUS remote_op_bind(struct dcesrv_call_state *dce_call, const struct
                return status;
        }
 
-       if (dce_call->context->assoc_group->proxied_id == 0) {
-               dce_call->context->assoc_group->proxied_id = priv->c_pipe->assoc_group_id;
+       if (dce_call->conn->state_flags & DCESRV_CALL_STATE_FLAG_MULTIPLEXED) {
+               flags = dcerpc_binding_get_flags(priv->c_pipe->binding);
+               if (!(flags & DCERPC_CONCURRENT_MULTIPLEX)) {
+                       DEBUG(1,("dcerpc_remote: RPC Proxy: "
+                                "Remote server doesn't support MPX\n"));
+                       return NT_STATUS_INVALID_NETWORK_RESPONSE;
+               }
        }
 
+       if (assoc->assoc_group_id == 0) {
+               assoc->assoc_group_id =
+                       dcerpc_binding_get_assoc_group_id(priv->c_pipe->binding);
+               if (assoc->assoc_group_id == 0) {
+                       return NT_STATUS_INVALID_NETWORK_RESPONSE;
+               }
+
+               status = dcesrv_iface_state_store_assoc(dce_call,
+                                               DCESRV_REMOTE_ASSOC_MAGIC,
+                                               assoc);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+       }
+
+       status = dcesrv_iface_state_store_conn(dce_call,
+                                       DCESRV_REMOTE_PRIVATE_MAGIC,
+                                       priv);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
 
-       return NT_STATUS_OK;    
-}
-
-static void remote_op_unbind(struct dcesrv_connection_context *context, const struct dcesrv_interface *iface)
-{
-       struct dcesrv_remote_private *priv = (struct dcesrv_remote_private *)context->private_data;
-
-       talloc_free(priv->c_pipe);
-
-       return; 
+       *_priv = priv;
+       return NT_STATUS_OK;
 }
 
 static NTSTATUS remote_op_ndr_pull(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct ndr_pull *pull, void **r)
@@ -166,6 +256,18 @@ static NTSTATUS remote_op_ndr_pull(struct dcesrv_call_state *dce_call, TALLOC_CT
                return NT_STATUS_NET_WRITE_FAULT;
        }
 
+       /*
+        * We don't have support for calls with pipes.
+        */
+       if (table->calls[opnum].in_pipes.num_pipes != 0) {
+               dce_call->fault_code = DCERPC_FAULT_OP_RNG_ERROR;
+               return NT_STATUS_NET_WRITE_FAULT;
+       }
+       if (table->calls[opnum].out_pipes.num_pipes != 0) {
+               dce_call->fault_code = DCERPC_FAULT_OP_RNG_ERROR;
+               return NT_STATUS_NET_WRITE_FAULT;
+       }
+
        *r = talloc_size(mem_ctx, table->calls[opnum].struct_size);
        if (!*r) {
                return NT_STATUS_NO_MEMORY;
@@ -184,17 +286,40 @@ static NTSTATUS remote_op_ndr_pull(struct dcesrv_call_state *dce_call, TALLOC_CT
        return NT_STATUS_OK;
 }
 
+static void remote_op_dispatch_done(struct tevent_req *subreq);
+
+struct dcesrv_remote_call {
+       struct dcesrv_call_state *dce_call;
+       struct dcesrv_remote_private *priv;
+};
+
 static NTSTATUS remote_op_dispatch(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, void *r)
 {
-       struct dcesrv_remote_private *priv = dce_call->context->private_data;
+       struct dcesrv_remote_call *rcall = NULL;
+       struct dcesrv_remote_private *priv = NULL;
        uint16_t opnum = dce_call->pkt.u.request.opnum;
        const struct ndr_interface_table *table = dce_call->context->iface->private_data;
        const struct ndr_interface_call *call;
        const char *name;
+       struct tevent_req *subreq;
+       NTSTATUS status;
 
        name = table->calls[opnum].name;
        call = &table->calls[opnum];
 
+       status = remote_get_private(dce_call, &priv);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("dcesrv_remote: call[%s] %s\n", name, nt_errstr(status)));
+               return status;
+       }
+
+       rcall = talloc_zero(dce_call, struct dcesrv_remote_call);
+       if (rcall == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       rcall->dce_call = dce_call;
+       rcall->priv = priv;
+
        if (priv->c_pipe->conn->flags & DCERPC_DEBUG_PRINT_IN) {
                ndr_print_function_debug(call->ndr_print, name, NDR_IN | NDR_SET_VALUES, r);            
        }
@@ -202,20 +327,58 @@ static NTSTATUS remote_op_dispatch(struct dcesrv_call_state *dce_call, TALLOC_CT
        priv->c_pipe->conn->flags |= DCERPC_NDR_REF_ALLOC;
 
        /* we didn't use the return code of this function as we only check the last_fault_code */
-       dcerpc_ndr_request(priv->c_pipe, NULL, table, opnum, mem_ctx,r);
+       subreq = dcerpc_binding_handle_call_send(rcall, dce_call->event_ctx,
+                                                priv->c_pipe->binding_handle,
+                                                NULL, table,
+                                                opnum, mem_ctx, r);
+       if (subreq == NULL) {
+               DEBUG(0,("dcesrv_remote: call[%s] dcerpc_binding_handle_call_send() failed!\n", name));
+               return NT_STATUS_NO_MEMORY;
+       }
+       tevent_req_set_callback(subreq, remote_op_dispatch_done, rcall);
+
+       dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
+       return NT_STATUS_OK;
+}
+
+static void remote_op_dispatch_done(struct tevent_req *subreq)
+{
+       struct dcesrv_remote_call *rcall =
+               tevent_req_callback_data(subreq,
+               struct dcesrv_remote_call);
+       struct dcesrv_call_state *dce_call = rcall->dce_call;
+       struct dcesrv_remote_private *priv = rcall->priv;
+       uint16_t opnum = dce_call->pkt.u.request.opnum;
+       const struct ndr_interface_table *table = dce_call->context->iface->private_data;
+       const struct ndr_interface_call *call;
+       const char *name;
+       NTSTATUS status;
+
+       name = table->calls[opnum].name;
+       call = &table->calls[opnum];
+
+       /* we didn't use the return code of this function as we only check the last_fault_code */
+       status = dcerpc_binding_handle_call_recv(subreq);
+       TALLOC_FREE(subreq);
 
        dce_call->fault_code = priv->c_pipe->last_fault_code;
        if (dce_call->fault_code != 0) {
-               DEBUG(0,("dcesrv_remote: call[%s] failed with: %s!\n",name, dcerpc_errstr(mem_ctx, dce_call->fault_code)));
-               return NT_STATUS_NET_WRITE_FAULT;
+               DEBUG(0,("dcesrv_remote: call[%s] failed with: %s!\n",
+                       name, dcerpc_errstr(dce_call, dce_call->fault_code)));
+               goto reply;
        }
 
-       if ((dce_call->fault_code == 0) && 
+       if (NT_STATUS_IS_OK(status) &&
            (priv->c_pipe->conn->flags & DCERPC_DEBUG_PRINT_OUT)) {
-               ndr_print_function_debug(call->ndr_print, name, NDR_OUT, r);            
+               ndr_print_function_debug(call->ndr_print, name, NDR_OUT, dce_call->r);
        }
 
-       return NT_STATUS_OK;
+reply:
+       status = dcesrv_reply(dce_call);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("dcesrv_remote: call[%s]: dcesrv_reply() failed - %s\n",
+                       name, nt_errstr(status)));
+       }
 }
 
 static NTSTATUS remote_op_ndr_push(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct ndr_push *push, const void *r)
@@ -236,7 +399,7 @@ static NTSTATUS remote_op_ndr_push(struct dcesrv_call_state *dce_call, TALLOC_CT
 
 static NTSTATUS remote_register_one_iface(struct dcesrv_context *dce_ctx, const struct dcesrv_interface *iface)
 {
-       int i;
+       unsigned int i;
        const struct ndr_interface_table *table = iface->private_data;
 
        for (i=0;i<table->endpoints->count;i++) {
@@ -255,8 +418,8 @@ static NTSTATUS remote_register_one_iface(struct dcesrv_context *dce_ctx, const
 
 static NTSTATUS remote_op_init_server(struct dcesrv_context *dce_ctx, const struct dcesrv_endpoint_server *ep_server)
 {
-       int i;
-       const char **ifaces = (const char **)str_list_make(dce_ctx, lp_parm_string(dce_ctx->lp_ctx, NULL, "dcerpc_remote", "interfaces"),NULL);
+       unsigned int i;
+       char **ifaces = str_list_make(dce_ctx, lpcfg_parm_string(dce_ctx->lp_ctx, NULL, "dcerpc_remote", "interfaces"),NULL);
 
        if (!ifaces) {
                DEBUG(3,("remote_op_init_server: no interfaces configured\n"));
@@ -291,7 +454,7 @@ static bool remote_fill_interface(struct dcesrv_interface *iface, const struct n
        iface->syntax_id = if_tabl->syntax_id;
        
        iface->bind = remote_op_bind;
-       iface->unbind = remote_op_unbind;
+       iface->unbind = NULL;
 
        iface->ndr_pull = remote_op_ndr_pull;
        iface->dispatch = remote_op_dispatch;
@@ -299,6 +462,7 @@ static bool remote_fill_interface(struct dcesrv_interface *iface, const struct n
        iface->ndr_push = remote_op_ndr_push;
 
        iface->private_data = if_tabl;
+       iface->flags = 0;
 
        return true;
 }
@@ -327,21 +491,19 @@ static bool remote_op_interface_by_name(struct dcesrv_interface *iface, const ch
        return false;   
 }
 
-NTSTATUS dcerpc_server_remote_init(void)
+NTSTATUS dcerpc_server_remote_init(TALLOC_CTX *ctx)
 {
        NTSTATUS ret;
-       struct dcesrv_endpoint_server ep_server;
-
-       ZERO_STRUCT(ep_server);
-
-       /* fill in our name */
-       ep_server.name = "remote";
+       static const struct dcesrv_endpoint_server ep_server = {
+               /* fill in our name */
+               .name = "remote",
 
-       /* fill in all the operations */
-       ep_server.init_server = remote_op_init_server;
+               /* fill in all the operations */
+               .init_server = remote_op_init_server,
 
-       ep_server.interface_by_uuid = remote_op_interface_by_uuid;
-       ep_server.interface_by_name = remote_op_interface_by_name;
+               .interface_by_uuid = remote_op_interface_by_uuid,
+               .interface_by_name = remote_op_interface_by_name
+       };
 
        /* register ourselves with the DCERPC subsystem. */
        ret = dcerpc_register_ep_server(&ep_server);