source4/rpc_server/dcerpc_server.h

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    server side dcerpc defines
   5
   6    Copyright (C) Andrew Tridgell 2003-2005
   7    Copyright (C) Stefan (metze) Metzmacher 2004-2005
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #ifndef SAMBA_DCERPC_SERVER_H
  24 #define SAMBA_DCERPC_SERVER_H
  25
  26 #include "librpc/gen_ndr/server_id.h"
  27 #include "librpc/rpc/dcerpc.h"
  28 #include "librpc/ndr/libndr.h"
  29
  30 /* modules can use the following to determine if the interface has changed
  31  * please increment the version number after each interface change
  32  * with a comment and maybe update struct dcesrv_critical_sizes.
  33  */
  34 /* version 1 - initial version - metze */
  35 #define DCERPC_MODULE_VERSION 1
  36
  37 struct dcesrv_connection;
  38 struct dcesrv_call_state;
  39 struct dcesrv_auth;
  40 struct dcesrv_connection_context;
  41
  42 struct dcesrv_interface {
  43         const char *name;
  44         struct ndr_syntax_id syntax_id;
  45
  46         /* this function is called when the client binds to this interface  */
  47         NTSTATUS (*bind)(struct dcesrv_call_state *, const struct dcesrv_interface *, uint32_t if_version);
  48
  49         /* this function is called when the client disconnects the endpoint */
  50         void (*unbind)(struct dcesrv_connection_context *, const struct dcesrv_interface *);
  51
  52         /* the ndr_pull function for the chosen interface.
  53          */
  54         NTSTATUS (*ndr_pull)(struct dcesrv_call_state *, TALLOC_CTX *, struct ndr_pull *, void **);
  55
  56         /* the dispatch function for the chosen interface.
  57          */
  58         NTSTATUS (*dispatch)(struct dcesrv_call_state *, TALLOC_CTX *, void *);
  59
  60         /* the reply function for the chosen interface.
  61          */
  62         NTSTATUS (*reply)(struct dcesrv_call_state *, TALLOC_CTX *, void *);
  63
  64         /* the ndr_push function for the chosen interface.
  65          */
  66         NTSTATUS (*ndr_push)(struct dcesrv_call_state *, TALLOC_CTX *, struct ndr_push *, const void *);
  67
  68         /* for any private use by the interface code */
  69         const void *private_data;
  70 };
  71
  72 enum dcesrv_call_list {
  73         DCESRV_LIST_NONE,
  74         DCESRV_LIST_CALL_LIST,
  75         DCESRV_LIST_FRAGMENTED_CALL_LIST,
  76         DCESRV_LIST_PENDING_CALL_LIST
  77 };
  78
  79 /* the state of an ongoing dcerpc call */
  80 struct dcesrv_call_state {
  81         struct dcesrv_call_state *next, *prev;
  82         struct dcesrv_connection *conn;
  83         struct dcesrv_connection_context *context;
  84         struct ncacn_packet pkt;
  85
  86         /*
  87           which list this request is in, if any
  88          */
  89         enum dcesrv_call_list list;
  90
  91         /* the backend can mark the call
  92          * with DCESRV_CALL_STATE_FLAG_ASYNC
  93          * that will cause the frontend to not touch r->out
  94          * and skip the reply
  95          *
  96          * this is only allowed to the backend when DCESRV_CALL_STATE_FLAG_MAY_ASYNC
  97          * is alerady set by the frontend
  98          *
  99          * the backend then needs to call dcesrv_reply() when it's
 100          * ready to send the reply
 101          */
 102 #define DCESRV_CALL_STATE_FLAG_ASYNC (1<<0)
 103 #define DCESRV_CALL_STATE_FLAG_MAY_ASYNC (1<<1)
 104 #define DCESRV_CALL_STATE_FLAG_HEADER_SIGNING (1<<2)
 105         uint32_t state_flags;
 106
 107         /* the time the request arrived in the server */
 108         struct timeval time;
 109
 110         /* the backend can use this event context for async replies */
 111         struct tevent_context *event_ctx;
 112
 113         /* the message_context that will be used for async replies */
 114         struct imessaging_context *msg_ctx;
 115
 116         /* this is the pointer to the allocated function struct */
 117         void *r;
 118
 119         /*
 120          * that's the ndr pull context used in dcesrv_request()
 121          * needed by dcesrv_reply() to carry over information
 122          * for full pointer support.
 123          */
 124         struct ndr_pull *ndr_pull;
 125
 126         DATA_BLOB input;
 127
 128         struct data_blob_list_item *replies;
 129
 130         /* this is used by the boilerplate code to generate DCERPC faults */
 131         uint32_t fault_code;
 132 };
 133
 134 #define DCESRV_HANDLE_ANY 255
 135
 136 /* a dcerpc handle in internal format */
 137 struct dcesrv_handle {
 138         struct dcesrv_handle *next, *prev;
 139         struct dcesrv_assoc_group *assoc_group;
 140         struct policy_handle wire_handle;
 141         struct dom_sid *sid;
 142         const struct dcesrv_interface *iface;
 143         void *data;
 144 };
 145
 146 /* hold the authentication state information */
 147 struct dcesrv_auth {
 148         struct dcerpc_auth *auth_info;
 149         struct gensec_security *gensec_security;
 150         struct auth_session_info *session_info;
 151         NTSTATUS (*session_key)(struct dcesrv_connection *, DATA_BLOB *session_key);
 152 };
 153
 154 struct dcesrv_connection_context {
 155         struct dcesrv_connection_context *next, *prev;
 156         uint32_t context_id;
 157
 158         struct dcesrv_assoc_group *assoc_group;
 159
 160         /* the connection this is on */
 161         struct dcesrv_connection *conn;
 162
 163         /* the ndr function table for the chosen interface */
 164         const struct dcesrv_interface *iface;
 165
 166         /* private data for the interface implementation */
 167         void *private_data;
 168 };
 169
 170
 171 /* the state associated with a dcerpc server connection */
 172 struct dcesrv_connection {
 173         /* for the broken_connections DLIST */
 174         struct dcesrv_connection *prev, *next;
 175
 176         /* the top level context for this server */
 177         struct dcesrv_context *dce_ctx;
 178
 179         /* the endpoint that was opened */
 180         const struct dcesrv_endpoint *endpoint;
 181
 182         /* a list of established context_ids */
 183         struct dcesrv_connection_context *contexts;
 184
 185         /* the state of the current incoming call fragments */
 186         struct dcesrv_call_state *incoming_fragmented_call_list;
 187
 188         /* the state of the async pending calls */
 189         struct dcesrv_call_state *pending_call_list;
 190
 191         /* the state of the current outgoing calls */
 192         struct dcesrv_call_state *call_list;
 193
 194         /* the maximum size the client wants to receive */
 195         uint32_t cli_max_recv_frag;
 196
 197         DATA_BLOB partial_input;
 198
 199         /* the current authentication state */
 200         struct dcesrv_auth auth_state;
 201
 202         /* the event_context that will be used for this connection */
 203         struct tevent_context *event_ctx;
 204
 205         /* the message_context that will be used for this connection */
 206         struct imessaging_context *msg_ctx;
 207
 208         /* the server_id that will be used for this connection */
 209         struct server_id server_id;
 210
 211         /* the transport level session key */
 212         DATA_BLOB transport_session_key;
 213
 214         /* is this connection pending termination?  If so, why? */
 215         const char *terminate;
 216
 217         const char *packet_log_dir;
 218
 219         /* this is the default state_flags for dcesrv_call_state structs */
 220         uint32_t state_flags;
 221
 222         struct {
 223                 void *private_data;
 224                 void (*report_output_data)(struct dcesrv_connection *);
 225         } transport;
 226
 227         struct tstream_context *stream;
 228         struct tevent_queue *send_queue;
 229
 230         const struct tsocket_address *local_address;
 231         const struct tsocket_address *remote_address;
 232 };
 233
 234
 235 struct dcesrv_endpoint_server {
 236         /* this is the name of the endpoint server */
 237         const char *name;
 238
 239         /* this function should register endpoints and some other setup stuff,
 240          * it is called when the dcesrv_context gets initialized.
 241          */
 242         NTSTATUS (*init_server)(struct dcesrv_context *, const struct dcesrv_endpoint_server *);
 243
 244         /* this function can be used by other endpoint servers to
 245          * ask for a dcesrv_interface implementation
 246          * - iface must be reference to an already existing struct !
 247          */
 248         bool (*interface_by_uuid)(struct dcesrv_interface *iface, const struct GUID *, uint32_t);
 249
 250         /* this function can be used by other endpoint servers to
 251          * ask for a dcesrv_interface implementation
 252          * - iface must be reference to an already existeng struct !
 253          */
 254         bool (*interface_by_name)(struct dcesrv_interface *iface, const char *);
 255 };
 256
 257
 258 /* one association groups */
 259 struct dcesrv_assoc_group {
 260         /* the wire id */
 261         uint32_t id;
 262
 263         /* list of handles in this association group */
 264         struct dcesrv_handle *handles;
 265
 266         /* parent context */
 267         struct dcesrv_context *dce_ctx;
 268
 269         /* Remote association group ID (if proxied) */
 270         uint32_t proxied_id;
 271 };
 272
 273 /* server-wide context information for the dcerpc server */
 274 struct dcesrv_context {
 275         /* the list of endpoints that have registered
 276          * by the configured endpoint servers
 277          */
 278         struct dcesrv_endpoint {
 279                 struct dcesrv_endpoint *next, *prev;
 280                 /* the type and location of the endpoint */
 281                 struct dcerpc_binding *ep_description;
 282                 /* the security descriptor for smb named pipes */
 283                 struct security_descriptor *sd;
 284                 /* the list of interfaces available on this endpoint */
 285                 struct dcesrv_if_list {
 286                         struct dcesrv_if_list *next, *prev;
 287                         struct dcesrv_interface iface;
 288                 } *interface_list;
 289         } *endpoint_list;
 290
 291         /* loadparm context to use for this connection */
 292         struct loadparm_context *lp_ctx;
 293
 294         struct idr_context *assoc_groups_idr;
 295
 296         struct dcesrv_connection *broken_connections;
 297 };
 298
 299 /* this structure is used by modules to determine the size of some critical types */
 300 struct dcesrv_critical_sizes {
 301         int interface_version;
 302         int sizeof_dcesrv_context;
 303         int sizeof_dcesrv_endpoint;
 304         int sizeof_dcesrv_endpoint_server;
 305         int sizeof_dcesrv_interface;
 306         int sizeof_dcesrv_if_list;
 307         int sizeof_dcesrv_connection;
 308         int sizeof_dcesrv_call_state;
 309         int sizeof_dcesrv_auth;
 310         int sizeof_dcesrv_handle;
 311 };
 312
 313 struct model_ops;
 314
 315 NTSTATUS dcesrv_interface_register(struct dcesrv_context *dce_ctx,
 316                                    const char *ep_name,
 317                                    const struct dcesrv_interface *iface,
 318                                    const struct security_descriptor *sd);
 319 NTSTATUS dcerpc_register_ep_server(const void *_ep_server);
 320 NTSTATUS dcesrv_init_context(TALLOC_CTX *mem_ctx,
 321                                       struct loadparm_context *lp_ctx,
 322                                       const char **endpoint_servers, struct dcesrv_context **_dce_ctx);
 323 NTSTATUS dcesrv_endpoint_connect(struct dcesrv_context *dce_ctx,
 324                                  TALLOC_CTX *mem_ctx,
 325                                  const struct dcesrv_endpoint *ep,
 326                                  struct auth_session_info *session_info,
 327                                  struct tevent_context *event_ctx,
 328                                  struct imessaging_context *msg_ctx,
 329                                  struct server_id server_id,
 330                                  uint32_t state_flags,
 331                                  struct dcesrv_connection **_p);
 332
 333 NTSTATUS dcesrv_reply(struct dcesrv_call_state *call);
 334 struct dcesrv_handle *dcesrv_handle_new(struct dcesrv_connection_context *context,
 335                                         uint8_t handle_type);
 336
 337 struct dcesrv_handle *dcesrv_handle_fetch(
 338                                           struct dcesrv_connection_context *context,
 339                                           struct policy_handle *p,
 340                                           uint8_t handle_type);
 341 struct socket_address *dcesrv_connection_get_my_addr(struct dcesrv_connection *conn, TALLOC_CTX *mem_ctx);
 342
 343 struct socket_address *dcesrv_connection_get_peer_addr(struct dcesrv_connection *conn, TALLOC_CTX *mem_ctx);
 344 const struct tsocket_address *dcesrv_connection_get_local_address(struct dcesrv_connection *conn);
 345 const struct tsocket_address *dcesrv_connection_get_remote_address(struct dcesrv_connection *conn);
 346
 347 NTSTATUS dcesrv_fetch_session_key(struct dcesrv_connection *p, DATA_BLOB *session_key);
 348
 349 /* a useful macro for generating a RPC fault in the backend code */
 350 #define DCESRV_FAULT(code) do { \
 351         dce_call->fault_code = code; \
 352         return r->out.result; \
 353 } while(0)
 354
 355 /* a useful macro for generating a RPC fault in the backend code */
 356 #define DCESRV_FAULT_VOID(code) do { \
 357         dce_call->fault_code = code; \
 358         return; \
 359 } while(0)
 360
 361 /* a useful macro for checking the validity of a dcerpc policy handle
 362    and giving the right fault code if invalid */
 363 #define DCESRV_CHECK_HANDLE(h) do {if (!(h)) DCESRV_FAULT(DCERPC_FAULT_CONTEXT_MISMATCH); } while (0)
 364
 365 /* this checks for a valid policy handle, and gives a fault if an
 366    invalid handle or retval if the handle is of the
 367    wrong type */
 368 #define DCESRV_PULL_HANDLE_RETVAL(h, inhandle, t, retval) do { \
 369         (h) = dcesrv_handle_fetch(dce_call->context, (inhandle), DCESRV_HANDLE_ANY); \
 370         DCESRV_CHECK_HANDLE(h); \
 371         if ((t) != DCESRV_HANDLE_ANY && (h)->wire_handle.handle_type != (t)) { \
 372                 return retval; \
 373         } \
 374 } while (0)
 375
 376 /* this checks for a valid policy handle and gives a dcerpc fault
 377    if its the wrong type of handle */
 378 #define DCESRV_PULL_HANDLE_FAULT(h, inhandle, t) do { \
 379         (h) = dcesrv_handle_fetch(dce_call->context, (inhandle), t); \
 380         DCESRV_CHECK_HANDLE(h); \
 381 } while (0)
 382
 383 #define DCESRV_PULL_HANDLE(h, inhandle, t) DCESRV_PULL_HANDLE_RETVAL(h, inhandle, t, NT_STATUS_INVALID_HANDLE)
 384 #define DCESRV_PULL_HANDLE_WERR(h, inhandle, t) DCESRV_PULL_HANDLE_RETVAL(h, inhandle, t, WERR_BADFID)
 385
 386 NTSTATUS dcesrv_add_ep(struct dcesrv_context *dce_ctx,
 387                        struct loadparm_context *lp_ctx,
 388                        struct dcesrv_endpoint *e,
 389                        struct tevent_context *event_ctx,
 390                        const struct model_ops *model_ops);
 391
 392 /**
 393  * retrieve credentials from a dce_call
 394  */
 395 _PUBLIC_ struct cli_credentials *dcesrv_call_credentials(struct dcesrv_call_state *dce_call);
 396
 397 /**
 398  * returns true if this is an authenticated call
 399  */
 400 _PUBLIC_ bool dcesrv_call_authenticated(struct dcesrv_call_state *dce_call);
 401
 402 /**
 403  * retrieve account_name for a dce_call
 404  */
 405 _PUBLIC_ const char *dcesrv_call_account_name(struct dcesrv_call_state *dce_call);
 406
 407
 408 #endif /* SAMBA_DCERPC_SERVER_H */