source4/ldap_server/ldap_server.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    LDAP server
   5
   6    Copyright (C) Andrew Tridgell 2005
   7    Copyright (C) Volker Lendecke 2004
   8    Copyright (C) Stefan Metzmacher 2004
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 2 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program; if not, write to the Free Software
  22    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  23 */
  24
  25 #include "includes.h"
  26 #include "lib/events/events.h"
  27 #include "auth/auth.h"
  28 #include "dlinklist.h"
  29 #include "libcli/util/asn_1.h"
  30 #include "ldap_server/ldap_server.h"
  31 #include "smbd/service_task.h"
  32 #include "smbd/service_stream.h"
  33 #include "lib/socket/socket.h"
  34 #include "lib/tls/tls.h"
  35 #include "lib/messaging/irpc.h"
  36 #include "lib/stream/packet.h"
  37 #include "lib/ldb/include/ldb.h"
  38 #include "lib/ldb/include/ldb_errors.h"
  39
  40 /*
  41   close the socket and shutdown a server_context
  42 */
  43 static void ldapsrv_terminate_connection(struct ldapsrv_connection *conn,
  44                                          const char *reason)
  45 {
  46         if (conn->tls) {
  47                 talloc_free(conn->tls);
  48                 conn->tls = NULL;
  49         }
  50         stream_terminate_connection(conn->connection, reason);
  51 }
  52
  53 /*
  54   handle packet errors
  55 */
  56 static void ldapsrv_error_handler(void *private, NTSTATUS status)
  57 {
  58         struct ldapsrv_connection *conn = talloc_get_type(private,
  59                                                           struct ldapsrv_connection);
  60         ldapsrv_terminate_connection(conn, nt_errstr(status));
  61 }
  62
  63 /*
  64   process a decoded ldap message
  65 */
  66 static void ldapsrv_process_message(struct ldapsrv_connection *conn,
  67                                     struct ldap_message *msg)
  68 {
  69         struct ldapsrv_call *call;
  70         NTSTATUS status;
  71         DATA_BLOB blob;
  72         BOOL enable_wrap = conn->enable_wrap;
  73
  74         call = talloc(conn, struct ldapsrv_call);
  75         if (!call) {
  76                 ldapsrv_terminate_connection(conn, "no memory");
  77                 return;
  78         }
  79
  80         call->request = talloc_steal(call, msg);
  81         call->conn = conn;
  82         call->replies = NULL;
  83
  84         /* make the call */
  85         status = ldapsrv_do_call(call);
  86         if (!NT_STATUS_IS_OK(status)) {
  87                 goto failed;
  88         }
  89
  90         blob = data_blob(NULL, 0);
  91
  92         if (call->replies == NULL) {
  93                 talloc_free(call);
  94                 return;
  95         }
  96
  97         /* build all the replies into a single blob */
  98         while (call->replies) {
  99                 DATA_BLOB b;
 100
 101                 msg = call->replies->msg;
 102                 if (!ldap_encode(msg, &b, call)) {
 103                         DEBUG(0,("Failed to encode ldap reply of type %d\n", msg->type));
 104                         goto failed;
 105                 }
 106
 107                 status = data_blob_append(call, &blob, b.data, b.length);
 108                 data_blob_free(&b);
 109
 110                 if (!NT_STATUS_IS_OK(status)) goto failed;
 111
 112                 DLIST_REMOVE(call->replies, call->replies);
 113         }
 114
 115         /* possibly encrypt/sign the reply */
 116         if (enable_wrap) {
 117                 DATA_BLOB wrapped;
 118
 119                 status = gensec_wrap(conn->gensec, call, &blob, &wrapped);
 120                 if (!NT_STATUS_IS_OK(status)) {
 121                         goto failed;
 122                 }
 123                 data_blob_free(&blob);
 124                 blob = data_blob_talloc(call, NULL, wrapped.length + 4);
 125                 if (blob.data == NULL) {
 126                         goto failed;
 127                 }
 128                 RSIVAL(blob.data, 0, wrapped.length);
 129                 memcpy(blob.data+4, wrapped.data, wrapped.length);
 130                 data_blob_free(&wrapped);
 131         }
 132
 133         packet_send(conn->packet, blob);
 134         talloc_free(call);
 135         return;
 136
 137 failed:
 138         talloc_free(call);
 139 }
 140
 141
 142 /*
 143   decode the input buffer
 144 */
 145 static NTSTATUS ldapsrv_decode_plain(struct ldapsrv_connection *conn, DATA_BLOB blob)
 146 {
 147         struct asn1_data asn1;
 148         struct ldap_message *msg = talloc(conn, struct ldap_message);
 149
 150         if (msg == NULL) {
 151                 return NT_STATUS_NO_MEMORY;
 152         }
 153
 154         if (!asn1_load(&asn1, blob)) {
 155                 return NT_STATUS_NO_MEMORY;
 156         }
 157
 158         if (!ldap_decode(&asn1, msg)) {
 159                 asn1_free(&asn1);
 160                 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
 161         }
 162
 163         data_blob_free(&blob);
 164         ldapsrv_process_message(conn, msg);
 165         asn1_free(&asn1);
 166         return NT_STATUS_OK;
 167 }
 168
 169
 170 /*
 171   decode/process wrapped data
 172 */
 173 static NTSTATUS ldapsrv_decode_wrapped(struct ldapsrv_connection *conn,
 174                                        DATA_BLOB blob)
 175 {
 176         DATA_BLOB wrapped, unwrapped;
 177         struct asn1_data asn1;
 178         struct ldap_message *msg = talloc(conn, struct ldap_message);
 179         NTSTATUS status;
 180
 181         if (msg == NULL) {
 182                 return NT_STATUS_NO_MEMORY;
 183         }
 184
 185         wrapped = data_blob_const(blob.data+4, blob.length-4);
 186
 187         status = gensec_unwrap(conn->gensec, msg, &wrapped, &unwrapped);
 188         if (!NT_STATUS_IS_OK(status)) {
 189                 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
 190         }
 191
 192         data_blob_free(&blob);
 193
 194         if (!asn1_load(&asn1, unwrapped)) {
 195                 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
 196         }
 197
 198         while (ldap_decode(&asn1, msg)) {
 199                 ldapsrv_process_message(conn, msg);
 200                 msg = talloc(conn, struct ldap_message);
 201         }
 202
 203         if (asn1.ofs < asn1.length) {
 204                 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
 205         }
 206
 207         talloc_free(msg);
 208         asn1_free(&asn1);
 209
 210         return NT_STATUS_OK;
 211 }
 212
 213 /*
 214   decode/process data
 215 */
 216 static NTSTATUS ldapsrv_decode(void *private, DATA_BLOB blob)
 217 {
 218         struct ldapsrv_connection *conn = talloc_get_type(private,
 219                                                           struct ldapsrv_connection);
 220         if (conn->enable_wrap) {
 221                 return ldapsrv_decode_wrapped(conn, blob);
 222         }
 223         return ldapsrv_decode_plain(conn, blob);
 224 }
 225
 226 /*
 227   called when a LDAP socket becomes readable
 228 */
 229 static void ldapsrv_recv(struct stream_connection *c, uint16_t flags)
 230 {
 231         struct ldapsrv_connection *conn =
 232                 talloc_get_type(c->private, struct ldapsrv_connection);
 233
 234         if (conn->limits.ite) {
 235                 talloc_free(conn->limits.ite);
 236                 conn->limits.ite = NULL;
 237         }
 238
 239         packet_recv(conn->packet);
 240 }
 241
 242 /*
 243   check if a blob is a complete ldap packet
 244   handle wrapper or unwrapped connections
 245 */
 246 NTSTATUS ldapsrv_complete_packet(void *private, DATA_BLOB blob, size_t *size)
 247 {
 248         struct ldapsrv_connection *conn = talloc_get_type(private,
 249                                                           struct ldapsrv_connection);
 250         if (conn->enable_wrap) {
 251                 return packet_full_request_u32(private, blob, size);
 252         }
 253         return ldap_full_packet(private, blob, size);
 254 }
 255
 256 /*
 257   called when a LDAP socket becomes writable
 258 */
 259 static void ldapsrv_send(struct stream_connection *c, uint16_t flags)
 260 {
 261         struct ldapsrv_connection *conn =
 262                 talloc_get_type(c->private, struct ldapsrv_connection);
 263
 264         packet_queue_run(conn->packet);
 265 }
 266
 267 static void ldapsrv_conn_init_timeout(struct event_context *ev,
 268                                       struct timed_event *te,
 269                                       struct timeval t,
 270                                       void *private)
 271 {
 272         struct ldapsrv_connection *conn = talloc_get_type(private, struct ldapsrv_connection);
 273
 274         ldapsrv_terminate_connection(conn, "Timeout. No requests after initial connection");
 275 }
 276
 277 static int ldapsrv_load_limits(struct ldapsrv_connection *conn)
 278 {
 279         TALLOC_CTX *tmp_ctx;
 280         const char *attrs[] = { "configurationNamingContext", NULL };
 281         const char *attrs2[] = { "lDAPAdminLimits", NULL };
 282         const char *conf_dn_s;
 283         struct ldb_message_element *el;
 284         struct ldb_result *res = NULL;
 285         struct ldb_dn *basedn;
 286         struct ldb_dn *conf_dn;
 287         struct ldb_dn *policy_dn;
 288         int i,ret;
 289
 290         /* set defaults limits in case of failure */
 291         conn->limits.initial_timeout = 120;
 292         conn->limits.conn_idle_time = 900;
 293         conn->limits.max_page_size = 1000;
 294         conn->limits.search_timeout = 120;
 295
 296
 297         tmp_ctx = talloc_new(conn);
 298         if (tmp_ctx == NULL) {
 299                 return -1;
 300         }
 301
 302         basedn = ldb_dn_explode(tmp_ctx, "");
 303         if (basedn == NULL) {
 304                 goto failed;
 305         }
 306
 307         ret = ldb_search(conn->ldb, basedn, LDB_SCOPE_BASE, NULL, attrs, &res);
 308         talloc_steal(tmp_ctx, res);
 309         if (ret != LDB_SUCCESS || res->count != 1) {
 310                 goto failed;
 311         }
 312
 313         conf_dn_s = ldb_msg_find_string(res->msgs[0], "configurationNamingContext", NULL);
 314         if (conf_dn_s == NULL) {
 315                 goto failed;
 316         }
 317         conf_dn = ldb_dn_explode(tmp_ctx, conf_dn_s);
 318         if (conf_dn == NULL) {
 319                 goto failed;
 320         }
 321
 322         policy_dn = ldb_dn_string_compose(tmp_ctx, conf_dn, "CN=Default Query Policy,CN=Query-Policies,CN=Directory Service,CN=Windows NT,CN=Services");
 323         if (policy_dn == NULL) {
 324                 goto failed;
 325         }
 326
 327         ret = ldb_search(conn->ldb, policy_dn, LDB_SCOPE_BASE, NULL, attrs2, &res);
 328         talloc_steal(tmp_ctx, res);
 329         if (ret != LDB_SUCCESS || res->count != 1) {
 330                 goto failed;
 331         }
 332
 333         el = ldb_msg_find_element(res->msgs[0], "lDAPAdminLimits");
 334         if (el == NULL) {
 335                 goto failed;
 336         }
 337
 338         for (i = 0; i < el->num_values; i++) {
 339                 char policy_name[256];
 340                 int policy_value, s;
 341
 342                 s = sscanf(el->values[i].data, "%255[^=]=%d", policy_name, &policy_value);
 343                 if (ret != 2 || policy_value == 0)
 344                         continue;
 345
 346                 if (strcasecmp("InitRecvTimeout", policy_name) == 0) {
 347                         conn->limits.initial_timeout = policy_value;
 348                         continue;
 349                 }
 350                 if (strcasecmp("MaxConnIdleTime", policy_name) == 0) {
 351                         conn->limits.conn_idle_time = policy_value;
 352                         continue;
 353                 }
 354                 if (strcasecmp("MaxPageSize", policy_name) == 0) {
 355                         conn->limits.max_page_size = policy_value;
 356                         continue;
 357                 }
 358                 if (strcasecmp("MaxQueryDuration", policy_name) == 0) {
 359                         conn->limits.search_timeout = policy_value;
 360                         continue;
 361                 }
 362         }
 363
 364         return 0;
 365
 366 failed:
 367         DEBUG(0, ("Failed to load ldap server query policies\n"));
 368         talloc_free(tmp_ctx);
 369         return -1;
 370 }
 371
 372 /*
 373   initialise a server_context from a open socket and register a event handler
 374   for reading from that socket
 375 */
 376 static void ldapsrv_accept(struct stream_connection *c)
 377 {
 378         struct ldapsrv_service *ldapsrv_service =
 379                 talloc_get_type(c->private, struct ldapsrv_service);
 380         struct ldapsrv_connection *conn;
 381         struct cli_credentials *server_credentials;
 382         struct socket_address *socket_address;
 383         NTSTATUS status;
 384         int port;
 385
 386         conn = talloc_zero(c, struct ldapsrv_connection);
 387         if (!conn) {
 388                 stream_terminate_connection(c, "ldapsrv_accept: out of memory");
 389                 return;
 390         }
 391
 392         conn->enable_wrap = False;
 393         conn->packet      = NULL;
 394         conn->connection  = c;
 395         conn->service     = ldapsrv_service;
 396
 397         server_credentials
 398                 = cli_credentials_init(conn);
 399         if (!server_credentials) {
 400                 stream_terminate_connection(c, "Failed to init server credentials\n");
 401                 talloc_free(conn);
 402                 return;
 403         }
 404
 405         cli_credentials_set_conf(server_credentials);
 406         status = cli_credentials_set_machine_account(server_credentials);
 407         if (!NT_STATUS_IS_OK(status)) {
 408                 stream_terminate_connection(c, talloc_asprintf(conn, "Failed to obtain server credentials, perhaps a standalone server?: %s\n", nt_errstr(status)));
 409                 talloc_free(conn);
 410                 return;
 411         }
 412         conn->server_credentials = server_credentials;
 413
 414         c->private        = conn;
 415
 416         socket_address = socket_get_my_addr(c->socket, conn);
 417         if (!socket_address) {
 418                 ldapsrv_terminate_connection(conn, "ldapsrv_accept: failed to obtain local socket address!");
 419                 return;
 420         }
 421         port = socket_address->port;
 422         talloc_free(socket_address);
 423
 424         conn->tls = tls_init_server(ldapsrv_service->tls_params, c->socket,
 425                                     c->event.fde, NULL, port != 389);
 426         if (!conn->tls) {
 427                 ldapsrv_terminate_connection(conn, "ldapsrv_accept: tls_init_server() failed");
 428                 return;
 429         }
 430
 431         conn->packet = packet_init(conn);
 432         if (conn->packet == NULL) {
 433                 ldapsrv_terminate_connection(conn, "out of memory");
 434                 return;
 435         }
 436         packet_set_private(conn->packet, conn);
 437         packet_set_tls(conn->packet, conn->tls);
 438         packet_set_callback(conn->packet, ldapsrv_decode);
 439         packet_set_full_request(conn->packet, ldapsrv_complete_packet);
 440         packet_set_error_handler(conn->packet, ldapsrv_error_handler);
 441         packet_set_event_context(conn->packet, c->event.ctx);
 442         packet_set_fde(conn->packet, c->event.fde);
 443         packet_set_serialise(conn->packet);
 444
 445         /* Connections start out anonymous */
 446         if (!NT_STATUS_IS_OK(auth_anonymous_session_info(conn, &conn->session_info))) {
 447                 ldapsrv_terminate_connection(conn, "failed to setup anonymous session info");
 448                 return;
 449         }
 450
 451         if (!NT_STATUS_IS_OK(ldapsrv_backend_Init(conn))) {
 452                 ldapsrv_terminate_connection(conn, "backend Init failed");
 453                 return;
 454         }
 455
 456         /* load limits from the conf partition */
 457         ldapsrv_load_limits(conn); /* should we fail on error ? */
 458
 459         /* register the server */
 460         irpc_add_name(c->msg_ctx, "ldap_server");
 461
 462         /* set connections limits */
 463         conn->limits.ite = event_add_timed(c->event.ctx, conn,
 464                                            timeval_current_ofs(conn->limits.initial_timeout, 0),
 465                                            ldapsrv_conn_init_timeout, conn);
 466 }
 467
 468 static const struct stream_server_ops ldap_stream_ops = {
 469         .name                   = "ldap",
 470         .accept_connection      = ldapsrv_accept,
 471         .recv_handler           = ldapsrv_recv,
 472         .send_handler           = ldapsrv_send,
 473 };
 474
 475 /*
 476   add a socket address to the list of events, one event per port
 477 */
 478 static NTSTATUS add_socket(struct event_context *event_context,
 479                            const struct model_ops *model_ops,
 480                            const char *address, struct ldapsrv_service *ldap_service)
 481 {
 482         uint16_t port = 389;
 483         NTSTATUS status;
 484
 485         status = stream_setup_socket(event_context, model_ops, &ldap_stream_ops,
 486                                      "ipv4", address, &port, ldap_service);
 487         if (!NT_STATUS_IS_OK(status)) {
 488                 DEBUG(0,("ldapsrv failed to bind to %s:%u - %s\n",
 489                          address, port, nt_errstr(status)));
 490         }
 491
 492         if (tls_support(ldap_service->tls_params)) {
 493                 /* add ldaps server */
 494                 port = 636;
 495                 status = stream_setup_socket(event_context, model_ops, &ldap_stream_ops,
 496                                              "ipv4", address, &port, ldap_service);
 497                 if (!NT_STATUS_IS_OK(status)) {
 498                         DEBUG(0,("ldapsrv failed to bind to %s:%u - %s\n",
 499                                  address, port, nt_errstr(status)));
 500                 }
 501         }
 502
 503         /* if we are a PDC, then also enable the global catalog server port, 3268 */
 504         if (lp_server_role() == ROLE_DOMAIN_PDC) {
 505                 port = 3268;
 506                 status = stream_setup_socket(event_context, model_ops, &ldap_stream_ops,
 507                                              "ipv4", address, &port, ldap_service);
 508                 if (!NT_STATUS_IS_OK(status)) {
 509                         DEBUG(0,("ldapsrv failed to bind to %s:%u - %s\n",
 510                                  address, port, nt_errstr(status)));
 511                 }
 512         }
 513
 514         return status;
 515 }
 516
 517 /*
 518   open the ldap server sockets
 519 */
 520 static void ldapsrv_task_init(struct task_server *task)
 521 {
 522         struct ldapsrv_service *ldap_service;
 523         NTSTATUS status;
 524
 525         ldap_service = talloc_zero(task, struct ldapsrv_service);
 526         if (ldap_service == NULL) goto failed;
 527
 528         ldap_service->tls_params = tls_initialise(ldap_service);
 529         if (ldap_service->tls_params == NULL) goto failed;
 530
 531         if (lp_interfaces() && lp_bind_interfaces_only()) {
 532                 int num_interfaces = iface_count();
 533                 int i;
 534
 535                 /* We have been given an interfaces line, and been
 536                    told to only bind to those interfaces. Create a
 537                    socket per interface and bind to only these.
 538                 */
 539                 for(i = 0; i < num_interfaces; i++) {
 540                         const char *address = iface_n_ip(i);
 541                         status = add_socket(task->event_ctx, task->model_ops, address, ldap_service);
 542                         if (!NT_STATUS_IS_OK(status)) goto failed;
 543                 }
 544         } else {
 545                 status = add_socket(task->event_ctx, task->model_ops, lp_socket_address(), ldap_service);
 546                 if (!NT_STATUS_IS_OK(status)) goto failed;
 547         }
 548
 549         return;
 550
 551 failed:
 552         task_server_terminate(task, "Failed to startup ldap server task");
 553 }
 554
 555 /*
 556   called on startup of the web server service It's job is to start
 557   listening on all configured sockets
 558 */
 559 static NTSTATUS ldapsrv_init(struct event_context *event_context,
 560                              const struct model_ops *model_ops)
 561 {
 562         return task_server_startup(event_context, model_ops, ldapsrv_task_init);
 563 }
 564
 565
 566 NTSTATUS server_service_ldap_init(void)
 567 {
 568         return register_server_service("ldap", ldapsrv_init);
 569 }