source4/rpc_server/epmapper/rpc_epmapper.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    endpoint server for the epmapper pipe
   5
   6    Copyright (C) Andrew Tridgell 2003
   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 2 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, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23 #include "includes.h"
  24 #include "rpc_server/common/common.h"
  25
  26 typedef uint32_t error_status_t;
  27
  28 /* handle types for this module */
  29 enum handle_types {HTYPE_LOOKUP};
  30
  31 /* a endpoint combined with an interface description */
  32 struct dcesrv_ep_iface {
  33         const char *name;
  34         struct dcesrv_ep_description ep_description;
  35         const char *uuid;
  36         uint32_t if_version;
  37 };
  38
  39 /*
  40   simple routine to compare a GUID string to a GUID structure
  41 */
  42 static int guid_cmp(TALLOC_CTX *mem_ctx, const struct GUID *guid, const char *uuid_str)
  43 {
  44         const char *s = GUID_string(mem_ctx, guid);
  45         if (!s || strcasecmp(s, uuid_str)) {
  46                 return -1;
  47         }
  48         return 0;
  49 }
  50
  51 /*
  52   fill a protocol tower
  53 */
  54 static BOOL fill_protocol_tower(TALLOC_CTX *mem_ctx, struct epm_towers *twr,
  55                                 struct dcesrv_ep_iface *e)
  56 {
  57         twr->num_floors = 5;
  58         twr->floors = talloc_array_p(mem_ctx, struct epm_floor, 5);
  59         if (!twr->floors) {
  60                 return False;
  61         }
  62
  63         twr->floors[0].lhs.protocol = EPM_PROTOCOL_UUID;
  64         GUID_from_string(e->uuid, &twr->floors[0].lhs.info.uuid.uuid);
  65         twr->floors[0].lhs.info.uuid.version = e->if_version;
  66         twr->floors[0].rhs.rhs_data = data_blob_talloc_zero(mem_ctx, 2);
  67
  68         /* encoded with NDR ... */
  69         twr->floors[1].lhs.protocol = EPM_PROTOCOL_UUID;
  70         GUID_from_string(NDR_GUID, &twr->floors[1].lhs.info.uuid.uuid);
  71         twr->floors[1].lhs.info.uuid.version = NDR_GUID_VERSION;
  72         twr->floors[1].rhs.rhs_data = data_blob_talloc_zero(mem_ctx, 2);
  73
  74         /* on an RPC connection ... */
  75         twr->floors[2].lhs.protocol = EPM_PROTOCOL_NCACN;
  76         twr->floors[2].lhs.info.lhs_data = data_blob(NULL, 0);
  77         twr->floors[2].rhs.rhs_data = data_blob_talloc_zero(mem_ctx, 2);
  78
  79         switch (e->ep_description.type) {
  80         case ENDPOINT_SMB:
  81                 /* on a SMB pipe ... */
  82                 twr->floors[3].lhs.protocol = EPM_PROTOCOL_SMB;
  83                 twr->floors[3].lhs.info.lhs_data = data_blob(NULL, 0);
  84                 twr->floors[3].rhs.rhs_data.data = talloc_asprintf(mem_ctx, "\\PIPE\\%s",
  85                                                                    e->ep_description.info.smb_pipe);
  86                 twr->floors[3].rhs.rhs_data.length = strlen(twr->floors[3].rhs.rhs_data.data)+1;
  87
  88                 /* on an NetBIOS link ... */
  89                 twr->floors[4].lhs.protocol = EPM_PROTOCOL_NETBIOS;
  90                 twr->floors[4].lhs.info.lhs_data = data_blob(NULL, 0);
  91                 twr->floors[4].rhs.rhs_data.data = talloc_asprintf(mem_ctx, "\\\\%s",
  92                                                                    lp_netbios_name());
  93                 twr->floors[4].rhs.rhs_data.length = strlen(twr->floors[4].rhs.rhs_data.data)+1;
  94                 break;
  95
  96         case ENDPOINT_TCP:
  97                 /* on a TCP connection ... */
  98                 twr->floors[3].lhs.protocol = EPM_PROTOCOL_TCP;
  99                 twr->floors[3].lhs.info.lhs_data = data_blob(NULL, 0);
 100                 twr->floors[3].rhs.rhs_data = data_blob_talloc(mem_ctx, NULL, 2);
 101                 RSSVAL(twr->floors[3].rhs.rhs_data.data, 0, e->ep_description.info.tcp_port);
 102
 103                 /* on an IP link ... */
 104                 twr->floors[4].lhs.protocol = EPM_PROTOCOL_IP;
 105                 twr->floors[4].lhs.info.lhs_data = data_blob(NULL, 0);
 106                 twr->floors[4].rhs.rhs_data = data_blob_talloc_zero(mem_ctx, 4);
 107                 /* TODO: we should fill in our IP address here as a hint to the
 108                    client */
 109                 break;
 110         }
 111
 112         return True;
 113 }
 114
 115
 116 /*
 117   build a list of all interfaces handled by all endpoint servers
 118 */
 119 static uint32_t build_ep_list(TALLOC_CTX *mem_ctx,
 120                               struct dcesrv_endpoint *endpoint_list,
 121                               struct dcesrv_ep_iface **eps)
 122 {
 123         struct dcesrv_endpoint *d;
 124         uint32_t total = 0;
 125
 126         *eps = NULL;
 127
 128         for (d=endpoint_list; d; d=d->next) {
 129                 struct dcesrv_if_list *iface;
 130
 131                 for (iface=d->interface_list;iface;iface=iface->next) {
 132                         (*eps) = talloc_realloc_p(mem_ctx,
 133                                                   *eps,
 134                                                   struct dcesrv_ep_iface,
 135                                                   total + 1);
 136                         if (!*eps) {
 137                                 return 0;
 138                         }
 139                         (*eps)[total].name = iface->iface.ndr->name;
 140                         (*eps)[total].uuid = iface->iface.ndr->uuid;
 141                         (*eps)[total].if_version = iface->iface.ndr->if_version;
 142                         (*eps)[total].ep_description = d->ep_description;
 143                         total++;
 144                 }
 145         }
 146
 147         return total;
 148 }
 149
 150
 151 static error_status_t epm_Insert(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 152                                  struct epm_Insert *r)
 153 {
 154         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 155 }
 156
 157 static error_status_t epm_Delete(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 158                                  struct epm_Delete *r)
 159 {
 160         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 161 }
 162
 163
 164 /*
 165   implement epm_Lookup. This call is used to enumerate the interfaces
 166   available on a rpc server
 167 */
 168 static error_status_t epm_Lookup(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 169                                  struct epm_Lookup *r)
 170 {
 171         struct dcesrv_handle *h;
 172         struct rpc_eps {
 173                 uint32_t count;
 174                 struct dcesrv_ep_iface *e;
 175         } *eps;
 176         uint32_t num_ents;
 177         int i;
 178
 179         h = dcesrv_handle_fetch(dce_call->conn, r->in.entry_handle, HTYPE_LOOKUP);
 180         DCESRV_CHECK_HANDLE(h);
 181
 182         eps = h->data;
 183
 184         if (!eps) {
 185                 /* this is the first call - fill the list. Subsequent calls
 186                    will feed from this list, stored in the handle */
 187                 eps = talloc_p(h, struct rpc_eps);
 188                 if (!eps) {
 189                         return EPMAPPER_STATUS_NO_MEMORY;
 190                 }
 191                 h->data = eps;
 192
 193                 eps->count = build_ep_list(h, dce_call->conn->dce_ctx->endpoint_list, &eps->e);
 194         }
 195
 196         /* return the next N elements */
 197         num_ents = r->in.max_ents;
 198         if (num_ents > eps->count) {
 199                 num_ents = eps->count;
 200         }
 201
 202         *r->out.entry_handle = h->wire_handle;
 203         r->out.num_ents = num_ents;
 204
 205         if (num_ents == 0) {
 206                 r->out.entries = NULL;
 207                 ZERO_STRUCTP(r->out.entry_handle);
 208                 dcesrv_handle_destroy(dce_call->conn, h);
 209                 return EPMAPPER_STATUS_NO_MORE_ENTRIES;
 210         }
 211
 212         r->out.entries = talloc_array_p(mem_ctx, struct epm_entry_t, num_ents);
 213         if (!r->out.entries) {
 214                 return EPMAPPER_STATUS_NO_MEMORY;
 215         }
 216
 217         for (i=0;i<num_ents;i++) {
 218                 ZERO_STRUCT(r->out.entries[i].object);
 219                 r->out.entries[i].annotation = eps->e[i].name;
 220                 r->out.entries[i].tower = talloc_p(mem_ctx, struct epm_twr_t);
 221                 if (!r->out.entries[i].tower) {
 222                         return EPMAPPER_STATUS_NO_MEMORY;
 223                 }
 224
 225                 if (!fill_protocol_tower(mem_ctx, &r->out.entries[i].tower->towers, &eps->e[i])) {
 226                         return EPMAPPER_STATUS_NO_MEMORY;
 227                 }
 228         }
 229
 230         eps->count -= num_ents;
 231         eps->e += num_ents;
 232
 233         return EPMAPPER_STATUS_OK;
 234 }
 235
 236
 237 /*
 238   implement epm_Map. This is used to find the specific endpoint to talk to given
 239   a generic protocol tower
 240 */
 241 static error_status_t epm_Map(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 242                               struct epm_Map *r)
 243 {
 244         uint32_t count;
 245         int i;
 246         struct dcesrv_ep_iface *eps;
 247         struct epm_floor *floors;
 248
 249         count = build_ep_list(mem_ctx, dce_call->conn->dce_ctx->endpoint_list, &eps);
 250
 251         ZERO_STRUCT(*r->out.entry_handle);
 252         r->out.num_towers = 1;
 253         r->out.towers = talloc_p(mem_ctx, struct epm_twr_p_t);
 254         if (!r->out.towers) {
 255                 return EPMAPPER_STATUS_NO_MEMORY;
 256         }
 257         r->out.towers->twr = talloc_p(mem_ctx, struct epm_twr_t);
 258         if (!r->out.towers->twr) {
 259                 return EPMAPPER_STATUS_NO_MEMORY;
 260         }
 261
 262         if (!r->in.map_tower || r->in.max_towers == 0 ||
 263             r->in.map_tower->towers.num_floors != 5) {
 264                 goto failed;
 265         }
 266
 267         floors = r->in.map_tower->towers.floors;
 268
 269         if (floors[0].lhs.protocol != EPM_PROTOCOL_UUID ||
 270             floors[1].lhs.protocol != EPM_PROTOCOL_UUID ||
 271             guid_cmp(mem_ctx, &floors[1].lhs.info.uuid.uuid, NDR_GUID) != 0 ||
 272             floors[1].lhs.info.uuid.version != NDR_GUID_VERSION ||
 273             floors[2].lhs.protocol != EPM_PROTOCOL_NCACN) {
 274                 goto failed;
 275         }
 276
 277         for (i=0;i<count;i++) {
 278                 if (guid_cmp(mem_ctx, &floors[0].lhs.info.uuid.uuid, eps[i].uuid) != 0 ||
 279                     floors[0].lhs.info.uuid.version != eps[i].if_version) {
 280                         continue;
 281                 }
 282                 switch (eps[i].ep_description.type) {
 283                 case ENDPOINT_SMB:
 284                         if (floors[3].lhs.protocol != EPM_PROTOCOL_SMB ||
 285                             floors[4].lhs.protocol != EPM_PROTOCOL_NETBIOS) {
 286                                 continue;
 287                         }
 288                         break;
 289                 case ENDPOINT_TCP:
 290                         if (floors[3].lhs.protocol != EPM_PROTOCOL_TCP ||
 291                             floors[4].lhs.protocol != EPM_PROTOCOL_IP) {
 292                                 continue;
 293                         }
 294                         break;
 295                 }
 296                 fill_protocol_tower(mem_ctx, &r->out.towers->twr->towers, &eps[i]);
 297                 r->out.towers->twr->tower_length = 0;
 298                 return EPMAPPER_STATUS_OK;
 299         }
 300
 301
 302 failed:
 303         r->out.num_towers = 0;
 304         r->out.towers->twr = NULL;
 305
 306         return EPMAPPER_STATUS_NO_MORE_ENTRIES;
 307 }
 308
 309 static error_status_t epm_LookupHandleFree(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 310                                            struct epm_LookupHandleFree *r)
 311 {
 312         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 313 }
 314
 315 static error_status_t epm_InqObject(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 316                                     struct epm_InqObject *r)
 317 {
 318         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 319 }
 320
 321 static error_status_t epm_MgmtDelete(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 322                                struct epm_MgmtDelete *r)
 323 {
 324         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 325 }
 326
 327 static error_status_t epm_MapAuth(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 328                             struct epm_MapAuth *r)
 329 {
 330         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 331 }
 332
 333 /* include the generated boilerplate */
 334 #include "librpc/gen_ndr/ndr_epmapper_s.c"