Line data    Source code

       1             : /* 
       2             :    Unix SMB/CIFS implementation.
       3             : 
       4             :    forwarding of RPC calls to other tasks
       5             : 
       6             :    Copyright (C) Andrew Tridgell 2009
       7             :    
       8             :    This program is free software; you can redistribute it and/or modify
       9             :    it under the terms of the GNU General Public License as published by
      10             :    the Free Software Foundation; either version 3 of the License, or
      11             :    (at your option) any later version.
      12             :    
      13             :    This program is distributed in the hope that it will be useful,
      14             :    but WITHOUT ANY WARRANTY; without even the implied warranty of
      15             :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      16             :    GNU General Public License for more details.
      17             :    
      18             :    You should have received a copy of the GNU General Public License
      19             :    along with this program.  If not, see <http://www.gnu.org/licenses/>.
      20             : */
      21             : 
      22             : #include "includes.h"
      23             : #include <tevent.h>
      24             : #include "rpc_server/dcerpc_server.h"
      25             : #include "librpc/gen_ndr/dcerpc.h"
      26             : #include "rpc_server/common/common.h"
      27             : #include "messaging/irpc.h"
      28             : #include "auth/auth.h"
      29             : 
      30             : 
      31             : struct dcesrv_forward_state {
      32             :         const char *opname;
      33             :         struct dcesrv_call_state *dce_call;
      34             : };
      35             : 
      36             : /*
      37             :   called when the forwarded rpc request is finished
      38             :  */
      39        2428 : static void dcesrv_irpc_forward_callback(struct tevent_req *subreq)
      40             : {
      41        2428 :         struct dcesrv_forward_state *st =
      42        2428 :                 tevent_req_callback_data(subreq,
      43             :                 struct dcesrv_forward_state);
      44        2428 :         const char *opname = st->opname;
      45             :         NTSTATUS status;
      46             : 
      47        2428 :         status = dcerpc_binding_handle_call_recv(subreq);
      48        2428 :         TALLOC_FREE(subreq);
      49        2428 :         if (!NT_STATUS_IS_OK(status)) {
      50           0 :                 DEBUG(0,("IRPC callback failed for %s - %s\n",
      51             :                          opname, nt_errstr(status)));
      52           0 :                 st->dce_call->fault_code = DCERPC_FAULT_CANT_PERFORM;
      53             :         }
      54        2428 :         status = dcesrv_reply(st->dce_call);
      55        2428 :         if (!NT_STATUS_IS_OK(status)) {
      56           0 :                 DEBUG(0,("%s_handler: dcesrv_reply() failed - %s\n",
      57             :                          opname, nt_errstr(status)));
      58             :         }
      59        2428 : }
      60             : 
      61             : 
      62             : 
      63             : /**
      64             :  * Forward a RPC call using IRPC to another task
      65             :  */
      66        2428 : void dcesrv_irpc_forward_rpc_call(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
      67             :                                   void *r, uint32_t callid,
      68             :                                   const struct ndr_interface_table *ndr_table,
      69             :                                   const char *dest_task, const char *opname,
      70             :                                   uint32_t timeout)
      71             : {
      72             :         struct dcesrv_forward_state *st;
      73             :         struct dcerpc_binding_handle *binding_handle;
      74             :         struct tevent_req *subreq;
      75        2428 :         struct auth_session_info *session_info =
      76           0 :                 dcesrv_call_session_info(dce_call);
      77        2428 :         struct imessaging_context *imsg_ctx =
      78        2428 :                 dcesrv_imessaging_context(dce_call->conn);
      79             : 
      80        2428 :         st = talloc(mem_ctx, struct dcesrv_forward_state);
      81        2428 :         if (st == NULL) {
      82           0 :                 dce_call->fault_code = DCERPC_FAULT_CANT_PERFORM;
      83           0 :                 return;
      84             :         }
      85             : 
      86        2428 :         st->dce_call = dce_call;
      87        2428 :         st->opname   = opname;
      88             : 
      89             :         /* if the caller has said they can't support async calls
      90             :            then fail the call */
      91        2428 :         if (!(dce_call->state_flags & DCESRV_CALL_STATE_FLAG_MAY_ASYNC)) {
      92             :                 /* we're not allowed to reply async */
      93           0 :                 DEBUG(0,("%s: Not available synchronously\n", dest_task));
      94           0 :                 dce_call->fault_code = DCERPC_FAULT_CANT_PERFORM;
      95           0 :                 return;
      96             :         }
      97             : 
      98        2428 :         binding_handle = irpc_binding_handle_by_name(st,
      99             :                                                      imsg_ctx,
     100             :                                                      dest_task,
     101             :                                                      ndr_table);
     102        2428 :         if (binding_handle == NULL) {
     103           0 :                 DEBUG(0,("%s: Failed to forward request to %s task\n",
     104             :                          opname, dest_task));
     105           0 :                 dce_call->fault_code = DCERPC_FAULT_CANT_PERFORM;
     106           0 :                 return;
     107             :         }
     108             : 
     109             :         /* reset timeout for the handle */
     110        2428 :         dcerpc_binding_handle_set_timeout(binding_handle, timeout);
     111             : 
     112             :         /* add security token to the handle*/
     113        2428 :         irpc_binding_handle_add_security_token(binding_handle,
     114             :                                                session_info->security_token);
     115             : 
     116             :         /* forward the call */
     117        2428 :         subreq = dcerpc_binding_handle_call_send(st, dce_call->event_ctx,
     118             :                                                  binding_handle,
     119             :                                                  NULL, ndr_table,
     120             :                                                  callid,
     121             :                                                  dce_call, r);
     122        2428 :         if (subreq == NULL) {
     123           0 :                 DEBUG(0,("%s: Failed to forward request to %s task\n", 
     124             :                          opname, dest_task));
     125           0 :                 dce_call->fault_code = DCERPC_FAULT_CANT_PERFORM;
     126           0 :                 return;
     127             :         }
     128             : 
     129             :         /* mark the request as replied async */
     130        2428 :         dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
     131             : 
     132             :         /* setup the callback */
     133        2428 :         tevent_req_set_callback(subreq, dcesrv_irpc_forward_callback, st);
     134             : }