smbd: allow POSIX opens for file_set_dosmode() in rename_internals_fsp()
[Samba.git] / source4 / ldap_server / ldap_server.c
blob9c34c3e5712239e2944f8d0f3f1047892838834e
1 /*
2 Unix SMB/CIFS implementation.
4 LDAP server
6 Copyright (C) Andrew Tridgell 2005
7 Copyright (C) Volker Lendecke 2004
8 Copyright (C) Stefan Metzmacher 2004
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 3 of the License, or
13 (at your option) any later version.
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.
20 You should have received a copy of the GNU General Public License
21 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "includes.h"
25 #include "system/network.h"
26 #include "lib/events/events.h"
27 #include "auth/auth.h"
28 #include "auth/credentials/credentials.h"
29 #include "librpc/gen_ndr/ndr_samr.h"
30 #include "../lib/util/dlinklist.h"
31 #include "../lib/util/asn1.h"
32 #include "ldap_server/ldap_server.h"
33 #include "samba/service_task.h"
34 #include "samba/service_stream.h"
35 #include "samba/service.h"
36 #include "samba/process_model.h"
37 #include "lib/tls/tls.h"
38 #include "lib/messaging/irpc.h"
39 #include <ldb.h>
40 #include <ldb_errors.h>
41 #include "libcli/ldap/ldap_proto.h"
42 #include "system/network.h"
43 #include "lib/socket/netif.h"
44 #include "dsdb/samdb/samdb.h"
45 #include "param/param.h"
46 #include "../lib/tsocket/tsocket.h"
47 #include "../lib/util/tevent_ntstatus.h"
48 #include "../libcli/util/tstream.h"
49 #include "libds/common/roles.h"
50 #include "lib/util/time.h"
51 #include "lib/util/server_id.h"
52 #include "lib/util/server_id_db.h"
53 #include "lib/messaging/messaging_internal.h"
55 #undef strcasecmp
57 static void ldapsrv_terminate_connection_done(struct tevent_req *subreq);
60 close the socket and shutdown a server_context
62 static void ldapsrv_terminate_connection(struct ldapsrv_connection *conn,
63 const char *reason)
65 struct tevent_req *subreq;
67 if (conn->limits.reason) {
68 return;
71 DLIST_REMOVE(conn->service->connections, conn);
73 conn->limits.endtime = timeval_current_ofs(0, 500);
75 tevent_queue_stop(conn->sockets.send_queue);
76 TALLOC_FREE(conn->sockets.read_req);
77 TALLOC_FREE(conn->deferred_expire_disconnect);
78 if (conn->active_call) {
79 tevent_req_cancel(conn->active_call);
80 conn->active_call = NULL;
83 conn->limits.reason = talloc_strdup(conn, reason);
84 if (conn->limits.reason == NULL) {
85 TALLOC_FREE(conn->sockets.tls);
86 TALLOC_FREE(conn->sockets.sasl);
87 TALLOC_FREE(conn->sockets.raw);
88 stream_terminate_connection(conn->connection, reason);
89 return;
92 subreq = tstream_disconnect_send(conn,
93 conn->connection->event.ctx,
94 conn->sockets.active);
95 if (subreq == NULL) {
96 TALLOC_FREE(conn->sockets.tls);
97 TALLOC_FREE(conn->sockets.sasl);
98 TALLOC_FREE(conn->sockets.raw);
99 stream_terminate_connection(conn->connection, reason);
100 return;
102 tevent_req_set_endtime(subreq,
103 conn->connection->event.ctx,
104 conn->limits.endtime);
105 tevent_req_set_callback(subreq, ldapsrv_terminate_connection_done, conn);
108 static void ldapsrv_terminate_connection_done(struct tevent_req *subreq)
110 struct ldapsrv_connection *conn =
111 tevent_req_callback_data(subreq,
112 struct ldapsrv_connection);
113 int sys_errno;
114 bool ok;
116 tstream_disconnect_recv(subreq, &sys_errno);
117 TALLOC_FREE(subreq);
119 if (conn->sockets.active == conn->sockets.raw) {
120 TALLOC_FREE(conn->sockets.tls);
121 TALLOC_FREE(conn->sockets.sasl);
122 TALLOC_FREE(conn->sockets.raw);
123 stream_terminate_connection(conn->connection,
124 conn->limits.reason);
125 return;
128 TALLOC_FREE(conn->sockets.tls);
129 TALLOC_FREE(conn->sockets.sasl);
130 conn->sockets.active = conn->sockets.raw;
132 subreq = tstream_disconnect_send(conn,
133 conn->connection->event.ctx,
134 conn->sockets.active);
135 if (subreq == NULL) {
136 TALLOC_FREE(conn->sockets.raw);
137 stream_terminate_connection(conn->connection,
138 conn->limits.reason);
139 return;
141 ok = tevent_req_set_endtime(subreq,
142 conn->connection->event.ctx,
143 conn->limits.endtime);
144 if (!ok) {
145 TALLOC_FREE(conn->sockets.raw);
146 stream_terminate_connection(conn->connection,
147 conn->limits.reason);
148 return;
150 tevent_req_set_callback(subreq, ldapsrv_terminate_connection_done, conn);
154 called when a LDAP socket becomes readable
156 void ldapsrv_recv(struct stream_connection *c, uint16_t flags)
158 smb_panic(__location__);
162 called when a LDAP socket becomes writable
164 static void ldapsrv_send(struct stream_connection *c, uint16_t flags)
166 smb_panic(__location__);
169 static int ldapsrv_load_limits(struct ldapsrv_connection *conn)
171 TALLOC_CTX *tmp_ctx;
172 const char *attrs[] = { "configurationNamingContext", NULL };
173 const char *attrs2[] = { "lDAPAdminLimits", NULL };
174 struct ldb_message_element *el;
175 struct ldb_result *res = NULL;
176 struct ldb_dn *basedn;
177 struct ldb_dn *conf_dn;
178 struct ldb_dn *policy_dn;
179 unsigned int i;
180 int ret;
182 /* set defaults limits in case of failure */
183 conn->limits.initial_timeout = 120;
184 conn->limits.conn_idle_time = 900;
185 conn->limits.max_page_size = 1000;
186 conn->limits.max_notifications = 5;
187 conn->limits.search_timeout = 120;
188 conn->limits.expire_time = (struct timeval) {
189 .tv_sec = get_time_t_max(),
193 tmp_ctx = talloc_new(conn);
194 if (tmp_ctx == NULL) {
195 return -1;
198 basedn = ldb_dn_new(tmp_ctx, conn->ldb, NULL);
199 if (basedn == NULL) {
200 goto failed;
203 ret = ldb_search(conn->ldb, tmp_ctx, &res, basedn, LDB_SCOPE_BASE, attrs, NULL);
204 if (ret != LDB_SUCCESS) {
205 goto failed;
208 if (res->count != 1) {
209 goto failed;
212 conf_dn = ldb_msg_find_attr_as_dn(conn->ldb, tmp_ctx, res->msgs[0], "configurationNamingContext");
213 if (conf_dn == NULL) {
214 goto failed;
217 policy_dn = ldb_dn_copy(tmp_ctx, conf_dn);
218 ldb_dn_add_child_fmt(policy_dn, "CN=Default Query Policy,CN=Query-Policies,CN=Directory Service,CN=Windows NT,CN=Services");
219 if (policy_dn == NULL) {
220 goto failed;
223 ret = ldb_search(conn->ldb, tmp_ctx, &res, policy_dn, LDB_SCOPE_BASE, attrs2, NULL);
224 if (ret != LDB_SUCCESS) {
225 goto failed;
228 if (res->count != 1) {
229 goto failed;
232 el = ldb_msg_find_element(res->msgs[0], "lDAPAdminLimits");
233 if (el == NULL) {
234 goto failed;
237 for (i = 0; i < el->num_values; i++) {
238 char policy_name[256];
239 int policy_value, s;
241 s = sscanf((const char *)el->values[i].data, "%255[^=]=%d", policy_name, &policy_value);
242 if (s != 2 || policy_value == 0)
243 continue;
244 if (strcasecmp("InitRecvTimeout", policy_name) == 0) {
245 conn->limits.initial_timeout = policy_value;
246 continue;
248 if (strcasecmp("MaxConnIdleTime", policy_name) == 0) {
249 conn->limits.conn_idle_time = policy_value;
250 continue;
252 if (strcasecmp("MaxPageSize", policy_name) == 0) {
253 conn->limits.max_page_size = policy_value;
254 continue;
256 if (strcasecmp("MaxNotificationPerConn", policy_name) == 0) {
257 conn->limits.max_notifications = policy_value;
258 continue;
260 if (strcasecmp("MaxQueryDuration", policy_name) == 0) {
261 if (policy_value > 0) {
262 conn->limits.search_timeout = policy_value;
264 continue;
268 return 0;
270 failed:
271 DBG_ERR("Failed to load ldap server query policies\n");
272 talloc_free(tmp_ctx);
273 return -1;
276 static int ldapsrv_call_destructor(struct ldapsrv_call *call)
278 if (call->conn == NULL) {
279 return 0;
282 DLIST_REMOVE(call->conn->pending_calls, call);
284 call->conn = NULL;
285 return 0;
288 static struct tevent_req *ldapsrv_process_call_send(TALLOC_CTX *mem_ctx,
289 struct tevent_context *ev,
290 struct tevent_queue *call_queue,
291 struct ldapsrv_call *call);
292 static NTSTATUS ldapsrv_process_call_recv(struct tevent_req *req);
294 static bool ldapsrv_call_read_next(struct ldapsrv_connection *conn);
295 static void ldapsrv_accept_tls_done(struct tevent_req *subreq);
298 initialise a server_context from a open socket and register a event handler
299 for reading from that socket
301 static void ldapsrv_accept(struct stream_connection *c,
302 struct auth_session_info *session_info,
303 bool is_privileged)
305 struct ldapsrv_service *ldapsrv_service =
306 talloc_get_type(c->private_data, struct ldapsrv_service);
307 struct ldapsrv_connection *conn;
308 struct cli_credentials *server_credentials;
309 struct socket_address *socket_address;
310 int port;
311 int ret;
312 struct tevent_req *subreq;
313 struct timeval endtime;
314 char *errstring = NULL;
316 conn = talloc_zero(c, struct ldapsrv_connection);
317 if (!conn) {
318 stream_terminate_connection(c, "ldapsrv_accept: out of memory");
319 return;
321 conn->is_privileged = is_privileged;
323 conn->sockets.send_queue = tevent_queue_create(conn, "ldapsev send queue");
324 if (conn->sockets.send_queue == NULL) {
325 stream_terminate_connection(c,
326 "ldapsrv_accept: tevent_queue_create failed");
327 return;
330 TALLOC_FREE(c->event.fde);
332 ret = tstream_bsd_existing_socket(conn,
333 socket_get_fd(c->socket),
334 &conn->sockets.raw);
335 if (ret == -1) {
336 stream_terminate_connection(c,
337 "ldapsrv_accept: out of memory");
338 return;
340 socket_set_flags(c->socket, SOCKET_FLAG_NOCLOSE);
341 /* as server we want to fail early */
342 tstream_bsd_fail_readv_first_error(conn->sockets.raw, true);
344 conn->connection = c;
345 conn->service = ldapsrv_service;
346 conn->lp_ctx = ldapsrv_service->lp_ctx;
348 c->private_data = conn;
350 socket_address = socket_get_my_addr(c->socket, conn);
351 if (!socket_address) {
352 ldapsrv_terminate_connection(conn, "ldapsrv_accept: failed to obtain local socket address!");
353 return;
355 port = socket_address->port;
356 talloc_free(socket_address);
357 if (port == 3268 || port == 3269) /* Global catalog */ {
358 conn->global_catalog = true;
361 server_credentials = cli_credentials_init_server(conn, conn->lp_ctx);
362 if (!server_credentials) {
363 stream_terminate_connection(c, "Failed to init server credentials\n");
364 return;
367 conn->server_credentials = server_credentials;
369 conn->session_info = session_info;
371 conn->sockets.active = conn->sockets.raw;
373 if (conn->is_privileged) {
374 conn->require_strong_auth = LDAP_SERVER_REQUIRE_STRONG_AUTH_NO;
375 } else {
376 conn->require_strong_auth = lpcfg_ldap_server_require_strong_auth(conn->lp_ctx);
379 ret = ldapsrv_backend_Init(conn, &errstring);
380 if (ret != LDB_SUCCESS) {
381 char *reason = talloc_asprintf(conn,
382 "LDB backend for LDAP Init "
383 "failed: %s: %s",
384 errstring, ldb_strerror(ret));
385 ldapsrv_terminate_connection(conn, reason);
386 return;
389 /* load limits from the conf partition */
390 ldapsrv_load_limits(conn); /* should we fail on error ? */
392 /* register the server */
393 irpc_add_name(c->msg_ctx, "ldap_server");
395 DLIST_ADD_END(ldapsrv_service->connections, conn);
397 if (port != 636 && port != 3269) {
398 ldapsrv_call_read_next(conn);
399 return;
402 endtime = timeval_current_ofs(conn->limits.conn_idle_time, 0);
404 subreq = tstream_tls_accept_send(conn,
405 conn->connection->event.ctx,
406 conn->sockets.raw,
407 conn->service->tls_params);
408 if (subreq == NULL) {
409 ldapsrv_terminate_connection(conn, "ldapsrv_accept: "
410 "no memory for tstream_tls_accept_send");
411 return;
413 tevent_req_set_endtime(subreq,
414 conn->connection->event.ctx,
415 endtime);
416 tevent_req_set_callback(subreq, ldapsrv_accept_tls_done, conn);
419 static void ldapsrv_accept_tls_done(struct tevent_req *subreq)
421 struct ldapsrv_connection *conn =
422 tevent_req_callback_data(subreq,
423 struct ldapsrv_connection);
424 int ret;
425 int sys_errno;
427 ret = tstream_tls_accept_recv(subreq, &sys_errno,
428 conn, &conn->sockets.tls);
429 TALLOC_FREE(subreq);
430 if (ret == -1) {
431 const char *reason;
433 reason = talloc_asprintf(conn, "ldapsrv_accept_tls_loop: "
434 "tstream_tls_accept_recv() - %d:%s",
435 sys_errno, strerror(sys_errno));
436 if (!reason) {
437 reason = "ldapsrv_accept_tls_loop: "
438 "tstream_tls_accept_recv() - failed";
441 ldapsrv_terminate_connection(conn, reason);
442 return;
445 conn->sockets.active = conn->sockets.tls;
446 conn->referral_scheme = LDAP_REFERRAL_SCHEME_LDAPS;
447 ldapsrv_call_read_next(conn);
450 static void ldapsrv_call_read_done(struct tevent_req *subreq);
451 static NTSTATUS ldapsrv_packet_check(
452 struct tstream_context *stream,
453 void *private_data,
454 DATA_BLOB blob,
455 size_t *packet_size);
457 static bool ldapsrv_call_read_next(struct ldapsrv_connection *conn)
459 struct tevent_req *subreq;
461 if (conn->pending_calls != NULL) {
462 conn->limits.endtime = timeval_zero();
464 ldapsrv_notification_retry_setup(conn->service, false);
465 } else if (timeval_is_zero(&conn->limits.endtime)) {
466 conn->limits.endtime =
467 timeval_current_ofs(conn->limits.initial_timeout, 0);
468 } else {
469 conn->limits.endtime =
470 timeval_current_ofs(conn->limits.conn_idle_time, 0);
473 if (conn->sockets.read_req != NULL) {
474 return true;
478 * The minimum size of a LDAP pdu is 7 bytes
480 * dumpasn1 -hh ldap-unbind-min.dat
482 * <30 05 02 01 09 42 00>
483 * 0 5: SEQUENCE {
484 * <02 01 09>
485 * 2 1: INTEGER 9
486 * <42 00>
487 * 5 0: [APPLICATION 2]
488 * : Error: Object has zero length.
489 * : }
491 * dumpasn1 -hh ldap-unbind-windows.dat
493 * <30 84 00 00 00 05 02 01 09 42 00>
494 * 0 5: SEQUENCE {
495 * <02 01 09>
496 * 6 1: INTEGER 9
497 * <42 00>
498 * 9 0: [APPLICATION 2]
499 * : Error: Object has zero length.
500 * : }
502 * This means using an initial read size
503 * of 7 is ok.
505 subreq = tstream_read_pdu_blob_send(conn,
506 conn->connection->event.ctx,
507 conn->sockets.active,
508 7, /* initial_read_size */
509 ldapsrv_packet_check,
510 conn);
511 if (subreq == NULL) {
512 ldapsrv_terminate_connection(conn, "ldapsrv_call_read_next: "
513 "no memory for tstream_read_pdu_blob_send");
514 return false;
516 if (!timeval_is_zero(&conn->limits.endtime)) {
517 bool ok;
518 ok = tevent_req_set_endtime(subreq,
519 conn->connection->event.ctx,
520 conn->limits.endtime);
521 if (!ok) {
522 ldapsrv_terminate_connection(
523 conn,
524 "ldapsrv_call_read_next: "
525 "no memory for tevent_req_set_endtime");
526 return false;
529 tevent_req_set_callback(subreq, ldapsrv_call_read_done, conn);
530 conn->sockets.read_req = subreq;
531 return true;
534 static void ldapsrv_call_process_done(struct tevent_req *subreq);
535 static int ldapsrv_check_packet_size(
536 struct ldapsrv_connection *conn,
537 size_t size);
539 static void ldapsrv_call_read_done(struct tevent_req *subreq)
541 struct ldapsrv_connection *conn =
542 tevent_req_callback_data(subreq,
543 struct ldapsrv_connection);
544 NTSTATUS status;
545 struct ldapsrv_call *call;
546 struct asn1_data *asn1;
547 DATA_BLOB blob;
548 int ret = LDAP_SUCCESS;
549 struct ldap_request_limits limits = {0};
551 conn->sockets.read_req = NULL;
553 call = talloc_zero(conn, struct ldapsrv_call);
554 if (!call) {
555 ldapsrv_terminate_connection(conn, "no memory");
556 return;
558 talloc_set_destructor(call, ldapsrv_call_destructor);
560 call->conn = conn;
562 status = tstream_read_pdu_blob_recv(subreq,
563 call,
564 &blob);
565 TALLOC_FREE(subreq);
566 if (!NT_STATUS_IS_OK(status)) {
567 const char *reason;
569 reason = talloc_asprintf(call, "ldapsrv_call_loop: "
570 "tstream_read_pdu_blob_recv() - %s",
571 nt_errstr(status));
572 if (!reason) {
573 reason = nt_errstr(status);
576 ldapsrv_terminate_connection(conn, reason);
577 return;
580 ret = ldapsrv_check_packet_size(conn, blob.length);
581 if (ret != LDAP_SUCCESS) {
582 ldapsrv_terminate_connection(
583 conn,
584 "Request packet too large");
585 return;
588 asn1 = asn1_init(call, ASN1_MAX_TREE_DEPTH);
589 if (asn1 == NULL) {
590 ldapsrv_terminate_connection(conn, "no memory");
591 return;
594 call->request = talloc(call, struct ldap_message);
595 if (call->request == NULL) {
596 ldapsrv_terminate_connection(conn, "no memory");
597 return;
600 asn1_load_nocopy(asn1, blob.data, blob.length);
602 limits.max_search_size =
603 lpcfg_ldap_max_search_request_size(conn->lp_ctx);
604 status = ldap_decode(
605 asn1,
606 &limits,
607 samba_ldap_control_handlers(),
608 call->request);
609 if (!NT_STATUS_IS_OK(status)) {
610 ldapsrv_terminate_connection(conn, nt_errstr(status));
611 return;
614 data_blob_free(&blob);
615 TALLOC_FREE(asn1);
618 /* queue the call in the global queue */
619 subreq = ldapsrv_process_call_send(call,
620 conn->connection->event.ctx,
621 conn->service->call_queue,
622 call);
623 if (subreq == NULL) {
624 ldapsrv_terminate_connection(conn, "ldapsrv_process_call_send failed");
625 return;
627 tevent_req_set_callback(subreq, ldapsrv_call_process_done, call);
628 conn->active_call = subreq;
631 static void ldapsrv_call_wait_done(struct tevent_req *subreq);
632 static void ldapsrv_call_writev_start(struct ldapsrv_call *call);
633 static void ldapsrv_call_writev_done(struct tevent_req *subreq);
635 static void ldapsrv_call_process_done(struct tevent_req *subreq)
637 struct ldapsrv_call *call =
638 tevent_req_callback_data(subreq,
639 struct ldapsrv_call);
640 struct ldapsrv_connection *conn = call->conn;
641 NTSTATUS status;
643 conn->active_call = NULL;
645 status = ldapsrv_process_call_recv(subreq);
646 TALLOC_FREE(subreq);
647 if (!NT_STATUS_IS_OK(status)) {
648 ldapsrv_terminate_connection(conn, nt_errstr(status));
649 return;
652 if (call->wait_send != NULL) {
653 subreq = call->wait_send(call,
654 conn->connection->event.ctx,
655 call->wait_private);
656 if (subreq == NULL) {
657 ldapsrv_terminate_connection(conn,
658 "ldapsrv_call_process_done: "
659 "call->wait_send - no memory");
660 return;
662 tevent_req_set_callback(subreq,
663 ldapsrv_call_wait_done,
664 call);
665 conn->active_call = subreq;
666 return;
669 ldapsrv_call_writev_start(call);
672 static void ldapsrv_call_wait_done(struct tevent_req *subreq)
674 struct ldapsrv_call *call =
675 tevent_req_callback_data(subreq,
676 struct ldapsrv_call);
677 struct ldapsrv_connection *conn = call->conn;
678 NTSTATUS status;
680 conn->active_call = NULL;
682 status = call->wait_recv(subreq);
683 TALLOC_FREE(subreq);
684 if (!NT_STATUS_IS_OK(status)) {
685 const char *reason;
687 reason = talloc_asprintf(call, "ldapsrv_call_wait_done: "
688 "call->wait_recv() - %s",
689 nt_errstr(status));
690 if (reason == NULL) {
691 reason = nt_errstr(status);
694 ldapsrv_terminate_connection(conn, reason);
695 return;
698 ldapsrv_call_writev_start(call);
701 static void ldapsrv_call_writev_start(struct ldapsrv_call *call)
703 struct ldapsrv_connection *conn = call->conn;
704 struct ldapsrv_reply *reply = NULL;
705 struct tevent_req *subreq = NULL;
706 struct timeval endtime;
707 size_t length = 0;
708 size_t i;
710 call->iov_count = 0;
712 /* build all the replies into an IOV (no copy) */
713 for (reply = call->replies;
714 reply != NULL;
715 reply = reply->next) {
717 /* Cap output at 25MB per writev() */
718 if (length > length + reply->blob.length
719 || length + reply->blob.length > LDAP_SERVER_MAX_CHUNK_SIZE) {
720 break;
724 * Overflow is harmless here, just used below to
725 * decide if to read or write, but checked above anyway
727 length += reply->blob.length;
730 * At worst an overflow would mean we send less
731 * replies
733 call->iov_count++;
736 if (length == 0) {
737 if (!call->notification.busy) {
738 TALLOC_FREE(call);
741 ldapsrv_call_read_next(conn);
742 return;
745 /* Cap call->iov_count at IOV_MAX */
746 call->iov_count = MIN(call->iov_count, IOV_MAX);
748 call->out_iov = talloc_array(call,
749 struct iovec,
750 call->iov_count);
751 if (!call->out_iov) {
752 /* This is not ideal */
753 ldapsrv_terminate_connection(conn,
754 "failed to allocate "
755 "iovec array");
756 return;
759 /* We may have had to cap the number of replies at IOV_MAX */
760 for (i = 0;
761 i < call->iov_count && call->replies != NULL;
762 i++) {
763 reply = call->replies;
764 call->out_iov[i].iov_base = reply->blob.data;
765 call->out_iov[i].iov_len = reply->blob.length;
767 /* Keep only the ASN.1 encoded data */
768 talloc_steal(call->out_iov, reply->blob.data);
770 DLIST_REMOVE(call->replies, reply);
771 TALLOC_FREE(reply);
774 if (i > call->iov_count) {
775 /* This is not ideal, but also (essentially) impossible */
776 ldapsrv_terminate_connection(conn,
777 "call list ended"
778 "before iov_count");
779 return;
782 subreq = tstream_writev_queue_send(call,
783 conn->connection->event.ctx,
784 conn->sockets.active,
785 conn->sockets.send_queue,
786 call->out_iov, call->iov_count);
787 if (subreq == NULL) {
788 ldapsrv_terminate_connection(conn, "stream_writev_queue_send failed");
789 return;
791 endtime = timeval_current_ofs(conn->limits.conn_idle_time, 0);
792 tevent_req_set_endtime(subreq,
793 conn->connection->event.ctx,
794 endtime);
795 tevent_req_set_callback(subreq, ldapsrv_call_writev_done, call);
798 static void ldapsrv_call_postprocess_done(struct tevent_req *subreq);
800 static void ldapsrv_call_writev_done(struct tevent_req *subreq)
802 struct ldapsrv_call *call =
803 tevent_req_callback_data(subreq,
804 struct ldapsrv_call);
805 struct ldapsrv_connection *conn = call->conn;
806 int sys_errno;
807 int rc;
809 rc = tstream_writev_queue_recv(subreq, &sys_errno);
810 TALLOC_FREE(subreq);
812 /* This releases the ASN.1 encoded packets from memory */
813 TALLOC_FREE(call->out_iov);
814 if (rc == -1) {
815 const char *reason;
817 reason = talloc_asprintf(call, "ldapsrv_call_writev_done: "
818 "tstream_writev_queue_recv() - %d:%s",
819 sys_errno, strerror(sys_errno));
820 if (reason == NULL) {
821 reason = "ldapsrv_call_writev_done: "
822 "tstream_writev_queue_recv() failed";
825 ldapsrv_terminate_connection(conn, reason);
826 return;
829 if (call->postprocess_send) {
830 subreq = call->postprocess_send(call,
831 conn->connection->event.ctx,
832 call->postprocess_private);
833 if (subreq == NULL) {
834 ldapsrv_terminate_connection(conn, "ldapsrv_call_writev_done: "
835 "call->postprocess_send - no memory");
836 return;
838 tevent_req_set_callback(subreq,
839 ldapsrv_call_postprocess_done,
840 call);
841 return;
844 /* Perhaps still some more to send */
845 if (call->replies != NULL) {
846 ldapsrv_call_writev_start(call);
847 return;
850 if (!call->notification.busy) {
851 TALLOC_FREE(call);
854 ldapsrv_call_read_next(conn);
857 static void ldapsrv_call_postprocess_done(struct tevent_req *subreq)
859 struct ldapsrv_call *call =
860 tevent_req_callback_data(subreq,
861 struct ldapsrv_call);
862 struct ldapsrv_connection *conn = call->conn;
863 NTSTATUS status;
865 status = call->postprocess_recv(subreq);
866 TALLOC_FREE(subreq);
867 if (!NT_STATUS_IS_OK(status)) {
868 const char *reason;
870 reason = talloc_asprintf(call, "ldapsrv_call_postprocess_done: "
871 "call->postprocess_recv() - %s",
872 nt_errstr(status));
873 if (reason == NULL) {
874 reason = nt_errstr(status);
877 ldapsrv_terminate_connection(conn, reason);
878 return;
881 TALLOC_FREE(call);
883 ldapsrv_call_read_next(conn);
886 static void ldapsrv_notification_retry_done(struct tevent_req *subreq);
888 void ldapsrv_notification_retry_setup(struct ldapsrv_service *service, bool force)
890 struct ldapsrv_connection *conn = NULL;
891 struct timeval retry;
892 size_t num_pending = 0;
893 size_t num_active = 0;
895 if (force) {
896 TALLOC_FREE(service->notification.retry);
897 service->notification.generation += 1;
900 if (service->notification.retry != NULL) {
901 return;
904 for (conn = service->connections; conn != NULL; conn = conn->next) {
905 if (conn->pending_calls == NULL) {
906 continue;
909 num_pending += 1;
911 if (conn->pending_calls->notification.generation !=
912 service->notification.generation)
914 num_active += 1;
918 if (num_pending == 0) {
919 return;
922 if (num_active != 0) {
923 retry = timeval_current_ofs(0, 100);
924 } else {
925 retry = timeval_current_ofs(5, 0);
928 service->notification.retry = tevent_wakeup_send(service,
929 service->current_ev,
930 retry);
931 if (service->notification.retry == NULL) {
932 /* retry later */
933 return;
936 tevent_req_set_callback(service->notification.retry,
937 ldapsrv_notification_retry_done,
938 service);
941 static void ldapsrv_notification_retry_done(struct tevent_req *subreq)
943 struct ldapsrv_service *service =
944 tevent_req_callback_data(subreq,
945 struct ldapsrv_service);
946 struct ldapsrv_connection *conn = NULL;
947 struct ldapsrv_connection *conn_next = NULL;
948 bool ok;
950 service->notification.retry = NULL;
952 ok = tevent_wakeup_recv(subreq);
953 TALLOC_FREE(subreq);
954 if (!ok) {
955 /* ignore */
958 for (conn = service->connections; conn != NULL; conn = conn_next) {
959 struct ldapsrv_call *call = conn->pending_calls;
961 conn_next = conn->next;
963 if (conn->pending_calls == NULL) {
964 continue;
967 if (conn->active_call != NULL) {
968 continue;
971 DLIST_DEMOTE(conn->pending_calls, call);
972 call->notification.generation =
973 service->notification.generation;
975 /* queue the call in the global queue */
976 subreq = ldapsrv_process_call_send(call,
977 conn->connection->event.ctx,
978 conn->service->call_queue,
979 call);
980 if (subreq == NULL) {
981 ldapsrv_terminate_connection(conn,
982 "ldapsrv_process_call_send failed");
983 continue;
985 tevent_req_set_callback(subreq, ldapsrv_call_process_done, call);
986 conn->active_call = subreq;
989 ldapsrv_notification_retry_setup(service, false);
992 struct ldapsrv_process_call_state {
993 struct ldapsrv_call *call;
996 static void ldapsrv_process_call_trigger(struct tevent_req *req,
997 void *private_data);
999 static struct tevent_req *ldapsrv_process_call_send(TALLOC_CTX *mem_ctx,
1000 struct tevent_context *ev,
1001 struct tevent_queue *call_queue,
1002 struct ldapsrv_call *call)
1004 struct tevent_req *req;
1005 struct ldapsrv_process_call_state *state;
1006 bool ok;
1008 req = tevent_req_create(mem_ctx, &state,
1009 struct ldapsrv_process_call_state);
1010 if (req == NULL) {
1011 return req;
1014 state->call = call;
1016 ok = tevent_queue_add(call_queue, ev, req,
1017 ldapsrv_process_call_trigger, NULL);
1018 if (!ok) {
1019 tevent_req_oom(req);
1020 return tevent_req_post(req, ev);
1023 return req;
1026 static void ldapsrv_disconnect_ticket_expired(struct tevent_req *subreq);
1028 static void ldapsrv_process_call_trigger(struct tevent_req *req,
1029 void *private_data)
1031 struct ldapsrv_process_call_state *state =
1032 tevent_req_data(req,
1033 struct ldapsrv_process_call_state);
1034 struct ldapsrv_connection *conn = state->call->conn;
1035 NTSTATUS status;
1037 if (conn->deferred_expire_disconnect != NULL) {
1039 * Just drop this on the floor
1041 tevent_req_done(req);
1042 return;
1045 /* make the call */
1046 status = ldapsrv_do_call(state->call);
1048 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
1050 * For testing purposes, defer the TCP disconnect
1051 * after having sent the msgid 0
1052 * 1.3.6.1.4.1.1466.20036 exop response. LDAP clients
1053 * should not wait for the TCP connection to close but
1054 * handle this packet equivalent to a TCP
1055 * disconnect. This delay enables testing both cases
1056 * in LDAP client libraries.
1059 int defer_msec = lpcfg_parm_int(
1060 conn->lp_ctx,
1061 NULL,
1062 "ldap_server",
1063 "delay_expire_disconnect",
1066 conn->deferred_expire_disconnect = tevent_wakeup_send(
1067 conn,
1068 conn->connection->event.ctx,
1069 timeval_current_ofs_msec(defer_msec));
1070 if (tevent_req_nomem(conn->deferred_expire_disconnect, req)) {
1071 return;
1073 tevent_req_set_callback(
1074 conn->deferred_expire_disconnect,
1075 ldapsrv_disconnect_ticket_expired,
1076 conn);
1078 tevent_req_done(req);
1079 return;
1082 if (!NT_STATUS_IS_OK(status)) {
1083 tevent_req_nterror(req, status);
1084 return;
1087 tevent_req_done(req);
1090 static void ldapsrv_disconnect_ticket_expired(struct tevent_req *subreq)
1092 struct ldapsrv_connection *conn = tevent_req_callback_data(
1093 subreq, struct ldapsrv_connection);
1094 bool ok;
1096 ok = tevent_wakeup_recv(subreq);
1097 TALLOC_FREE(subreq);
1098 if (!ok) {
1099 DBG_WARNING("tevent_wakeup_recv failed\n");
1101 conn->deferred_expire_disconnect = NULL;
1102 ldapsrv_terminate_connection(conn, "network session expired");
1105 static NTSTATUS ldapsrv_process_call_recv(struct tevent_req *req)
1107 NTSTATUS status;
1109 if (tevent_req_is_nterror(req, &status)) {
1110 tevent_req_received(req);
1111 return status;
1114 tevent_req_received(req);
1115 return NT_STATUS_OK;
1118 static void ldapsrv_accept_nonpriv(struct stream_connection *c)
1120 struct ldapsrv_service *ldapsrv_service = talloc_get_type_abort(
1121 c->private_data, struct ldapsrv_service);
1122 struct auth_session_info *session_info;
1123 NTSTATUS status;
1125 status = auth_anonymous_session_info(
1126 c, ldapsrv_service->lp_ctx, &session_info);
1127 if (!NT_STATUS_IS_OK(status)) {
1128 stream_terminate_connection(c, "failed to setup anonymous "
1129 "session info");
1130 return;
1132 ldapsrv_accept(c, session_info, false);
1135 static const struct stream_server_ops ldap_stream_nonpriv_ops = {
1136 .name = "ldap",
1137 .accept_connection = ldapsrv_accept_nonpriv,
1138 .recv_handler = ldapsrv_recv,
1139 .send_handler = ldapsrv_send,
1142 /* The feature removed behind an #ifdef until we can do it properly
1143 * with an EXTERNAL bind. */
1145 #define WITH_LDAPI_PRIV_SOCKET
1147 #ifdef WITH_LDAPI_PRIV_SOCKET
1148 static void ldapsrv_accept_priv(struct stream_connection *c)
1150 struct ldapsrv_service *ldapsrv_service = talloc_get_type_abort(
1151 c->private_data, struct ldapsrv_service);
1152 struct auth_session_info *session_info;
1154 session_info = system_session(ldapsrv_service->lp_ctx);
1155 if (!session_info) {
1156 stream_terminate_connection(c, "failed to setup system "
1157 "session info");
1158 return;
1160 ldapsrv_accept(c, session_info, true);
1163 static const struct stream_server_ops ldap_stream_priv_ops = {
1164 .name = "ldap",
1165 .accept_connection = ldapsrv_accept_priv,
1166 .recv_handler = ldapsrv_recv,
1167 .send_handler = ldapsrv_send,
1170 #endif
1174 add a socket address to the list of events, one event per port
1176 static NTSTATUS add_socket(struct task_server *task,
1177 struct loadparm_context *lp_ctx,
1178 const struct model_ops *model_ops,
1179 const char *address, struct ldapsrv_service *ldap_service)
1181 uint16_t port = 389;
1182 NTSTATUS status;
1183 struct ldb_context *ldb;
1185 status = stream_setup_socket(task, task->event_ctx, lp_ctx,
1186 model_ops, &ldap_stream_nonpriv_ops,
1187 "ip", address, &port,
1188 lpcfg_socket_options(lp_ctx),
1189 ldap_service, task->process_context);
1190 if (!NT_STATUS_IS_OK(status)) {
1191 DBG_ERR("ldapsrv failed to bind to %s:%u - %s\n",
1192 address, port, nt_errstr(status));
1193 return status;
1196 if (tstream_tls_params_enabled(ldap_service->tls_params)) {
1197 /* add ldaps server */
1198 port = 636;
1199 status = stream_setup_socket(task, task->event_ctx, lp_ctx,
1200 model_ops,
1201 &ldap_stream_nonpriv_ops,
1202 "ip", address, &port,
1203 lpcfg_socket_options(lp_ctx),
1204 ldap_service,
1205 task->process_context);
1206 if (!NT_STATUS_IS_OK(status)) {
1207 DBG_ERR("ldapsrv failed to bind to %s:%u - %s\n",
1208 address, port, nt_errstr(status));
1209 return status;
1213 /* Load LDAP database, but only to read our settings */
1214 ldb = samdb_connect(ldap_service,
1215 ldap_service->current_ev,
1216 lp_ctx,
1217 system_session(lp_ctx),
1218 NULL,
1220 if (!ldb) {
1221 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1224 if (samdb_is_gc(ldb)) {
1225 port = 3268;
1226 status = stream_setup_socket(task, task->event_ctx, lp_ctx,
1227 model_ops,
1228 &ldap_stream_nonpriv_ops,
1229 "ip", address, &port,
1230 lpcfg_socket_options(lp_ctx),
1231 ldap_service,
1232 task->process_context);
1233 if (!NT_STATUS_IS_OK(status)) {
1234 DBG_ERR("ldapsrv failed to bind to %s:%u - %s\n",
1235 address, port, nt_errstr(status));
1236 return status;
1238 if (tstream_tls_params_enabled(ldap_service->tls_params)) {
1239 /* add ldaps server for the global catalog */
1240 port = 3269;
1241 status = stream_setup_socket(task, task->event_ctx, lp_ctx,
1242 model_ops,
1243 &ldap_stream_nonpriv_ops,
1244 "ip", address, &port,
1245 lpcfg_socket_options(lp_ctx),
1246 ldap_service,
1247 task->process_context);
1248 if (!NT_STATUS_IS_OK(status)) {
1249 DBG_ERR("ldapsrv failed to bind to %s:%u - %s\n",
1250 address, port, nt_errstr(status));
1251 return status;
1256 /* And once we are bound, free the temporary ldb, it will
1257 * connect again on each incoming LDAP connection */
1258 talloc_unlink(ldap_service, ldb);
1260 return NT_STATUS_OK;
1263 static void ldap_reload_certs(struct imessaging_context *msg_ctx,
1264 void *private_data,
1265 uint32_t msg_type,
1266 struct server_id server_id,
1267 size_t num_fds,
1268 int *fds,
1269 DATA_BLOB *data)
1271 TALLOC_CTX *frame = talloc_stackframe();
1272 struct ldapsrv_service *ldap_service =
1273 talloc_get_type_abort(private_data,
1274 struct ldapsrv_service);
1275 int default_children;
1276 int num_children;
1277 int i;
1278 bool ok;
1279 struct server_id ldap_master_id;
1280 NTSTATUS status;
1281 struct tstream_tls_params *new_tls_params = NULL;
1283 SMB_ASSERT(msg_ctx == ldap_service->current_msg);
1285 /* reload certificates */
1286 status = tstream_tls_params_server(ldap_service,
1287 ldap_service->dns_host_name,
1288 lpcfg_tls_enabled(ldap_service->lp_ctx),
1289 lpcfg_tls_keyfile(frame, ldap_service->lp_ctx),
1290 lpcfg_tls_certfile(frame, ldap_service->lp_ctx),
1291 lpcfg_tls_cafile(frame, ldap_service->lp_ctx),
1292 lpcfg_tls_crlfile(frame, ldap_service->lp_ctx),
1293 lpcfg_tls_dhpfile(frame, ldap_service->lp_ctx),
1294 lpcfg_tls_priority(ldap_service->lp_ctx),
1295 &new_tls_params);
1296 if (!NT_STATUS_IS_OK(status)) {
1297 DBG_ERR("ldapsrv failed tstream_tls_params_server - %s\n",
1298 nt_errstr(status));
1299 TALLOC_FREE(frame);
1300 return;
1303 TALLOC_FREE(ldap_service->tls_params);
1304 ldap_service->tls_params = new_tls_params;
1306 if (getpid() != ldap_service->parent_pid) {
1308 * If we are not the master process we are done
1310 TALLOC_FREE(frame);
1311 return;
1315 * Check we're running under the prefork model,
1316 * by checking if the prefork-master-ldap name
1317 * was registered
1319 ok = server_id_db_lookup_one(msg_ctx->names, "prefork-master-ldap", &ldap_master_id);
1320 if (!ok) {
1322 * We are done if another process model is in use.
1324 TALLOC_FREE(frame);
1325 return;
1329 * Now we loop over all possible prefork workers
1330 * in order to notify them about the reload
1332 default_children = lpcfg_prefork_children(ldap_service->lp_ctx);
1333 num_children = lpcfg_parm_int(ldap_service->lp_ctx,
1334 NULL, "prefork children", "ldap",
1335 default_children);
1336 for (i = 0; i < num_children; i++) {
1337 char child_name[64] = { 0, };
1338 struct server_id ldap_worker_id;
1340 snprintf(child_name, sizeof(child_name), "prefork-worker-ldap-%d", i);
1341 ok = server_id_db_lookup_one(msg_ctx->names, child_name, &ldap_worker_id);
1342 if (!ok) {
1343 DBG_ERR("server_id_db_lookup_one(%s) - failed\n",
1344 child_name);
1345 continue;
1348 status = imessaging_send(msg_ctx, ldap_worker_id,
1349 MSG_RELOAD_TLS_CERTIFICATES, NULL);
1350 if (!NT_STATUS_IS_OK(status)) {
1351 struct server_id_buf id_buf;
1352 DBG_ERR("ldapsrv failed imessaging_send(%s, %s) - %s\n",
1353 child_name,
1354 server_id_str_buf(ldap_worker_id, &id_buf),
1355 nt_errstr(status));
1356 continue;
1360 TALLOC_FREE(frame);
1364 open the ldap server sockets
1366 static NTSTATUS ldapsrv_task_init(struct task_server *task)
1368 char *ldapi_path;
1369 #ifdef WITH_LDAPI_PRIV_SOCKET
1370 char *priv_dir;
1371 #endif
1372 struct ldapsrv_service *ldap_service;
1373 NTSTATUS status;
1375 switch (lpcfg_server_role(task->lp_ctx)) {
1376 case ROLE_STANDALONE:
1377 task_server_terminate(task, "ldap_server: no LDAP server required in standalone configuration",
1378 false);
1379 return NT_STATUS_INVALID_DOMAIN_ROLE;
1380 case ROLE_DOMAIN_MEMBER:
1381 task_server_terminate(task, "ldap_server: no LDAP server required in member server configuration",
1382 false);
1383 return NT_STATUS_INVALID_DOMAIN_ROLE;
1384 case ROLE_ACTIVE_DIRECTORY_DC:
1385 /* Yes, we want an LDAP server */
1386 break;
1389 task_server_set_title(task, "task[ldapsrv]");
1391 ldap_service = talloc_zero(task, struct ldapsrv_service);
1392 if (ldap_service == NULL) {
1393 status = NT_STATUS_NO_MEMORY;
1394 goto failed;
1397 ldap_service->lp_ctx = task->lp_ctx;
1398 ldap_service->current_ev = task->event_ctx;
1399 ldap_service->current_msg = task->msg_ctx;
1401 ldap_service->dns_host_name = talloc_asprintf(ldap_service, "%s.%s",
1402 lpcfg_netbios_name(task->lp_ctx),
1403 lpcfg_dnsdomain(task->lp_ctx));
1404 if (ldap_service->dns_host_name == NULL) {
1405 status = NT_STATUS_NO_MEMORY;
1406 goto failed;
1409 ldap_service->parent_pid = getpid();
1411 status = tstream_tls_params_server(ldap_service,
1412 ldap_service->dns_host_name,
1413 lpcfg_tls_enabled(task->lp_ctx),
1414 lpcfg_tls_keyfile(ldap_service, task->lp_ctx),
1415 lpcfg_tls_certfile(ldap_service, task->lp_ctx),
1416 lpcfg_tls_cafile(ldap_service, task->lp_ctx),
1417 lpcfg_tls_crlfile(ldap_service, task->lp_ctx),
1418 lpcfg_tls_dhpfile(ldap_service, task->lp_ctx),
1419 lpcfg_tls_priority(task->lp_ctx),
1420 &ldap_service->tls_params);
1421 if (!NT_STATUS_IS_OK(status)) {
1422 DBG_ERR("ldapsrv failed tstream_tls_params_server - %s\n",
1423 nt_errstr(status));
1424 goto failed;
1427 ldap_service->call_queue = tevent_queue_create(ldap_service, "ldapsrv_call_queue");
1428 if (ldap_service->call_queue == NULL) {
1429 status = NT_STATUS_NO_MEMORY;
1430 goto failed;
1433 if (lpcfg_interfaces(task->lp_ctx) && lpcfg_bind_interfaces_only(task->lp_ctx)) {
1434 struct interface *ifaces;
1435 int num_interfaces;
1436 int i;
1438 load_interface_list(task, task->lp_ctx, &ifaces);
1439 num_interfaces = iface_list_count(ifaces);
1441 /* We have been given an interfaces line, and been
1442 told to only bind to those interfaces. Create a
1443 socket per interface and bind to only these.
1445 for(i = 0; i < num_interfaces; i++) {
1446 const char *address = iface_list_n_ip(ifaces, i);
1447 status = add_socket(task, task->lp_ctx, task->model_ops,
1448 address, ldap_service);
1449 if (!NT_STATUS_IS_OK(status)) goto failed;
1451 } else {
1452 char **wcard;
1453 size_t i;
1454 size_t num_binds = 0;
1455 wcard = iface_list_wildcard(task);
1456 if (wcard == NULL) {
1457 DBG_ERR("No wildcard addresses available\n");
1458 status = NT_STATUS_UNSUCCESSFUL;
1459 goto failed;
1461 for (i=0; wcard[i]; i++) {
1462 status = add_socket(task, task->lp_ctx, task->model_ops,
1463 wcard[i], ldap_service);
1464 if (NT_STATUS_IS_OK(status)) {
1465 num_binds++;
1468 talloc_free(wcard);
1469 if (num_binds == 0) {
1470 status = NT_STATUS_UNSUCCESSFUL;
1471 goto failed;
1475 ldapi_path = lpcfg_private_path(ldap_service, task->lp_ctx, "ldapi");
1476 if (!ldapi_path) {
1477 status = NT_STATUS_UNSUCCESSFUL;
1478 goto failed;
1481 status = stream_setup_socket(task, task->event_ctx, task->lp_ctx,
1482 task->model_ops, &ldap_stream_nonpriv_ops,
1483 "unix", ldapi_path, NULL,
1484 lpcfg_socket_options(task->lp_ctx),
1485 ldap_service, task->process_context);
1486 talloc_free(ldapi_path);
1487 if (!NT_STATUS_IS_OK(status)) {
1488 DBG_ERR("ldapsrv failed to bind to %s - %s\n",
1489 ldapi_path, nt_errstr(status));
1492 #ifdef WITH_LDAPI_PRIV_SOCKET
1493 priv_dir = lpcfg_private_path(ldap_service, task->lp_ctx, "ldap_priv");
1494 if (priv_dir == NULL) {
1495 status = NT_STATUS_UNSUCCESSFUL;
1496 goto failed;
1499 * Make sure the directory for the privileged ldapi socket exists, and
1500 * is of the correct permissions
1502 if (!directory_create_or_exist(priv_dir, 0750)) {
1503 task_server_terminate(task, "Cannot create ldap "
1504 "privileged ldapi directory", true);
1505 return NT_STATUS_UNSUCCESSFUL;
1507 ldapi_path = talloc_asprintf(ldap_service, "%s/ldapi", priv_dir);
1508 talloc_free(priv_dir);
1509 if (ldapi_path == NULL) {
1510 status = NT_STATUS_NO_MEMORY;
1511 goto failed;
1514 status = stream_setup_socket(task, task->event_ctx, task->lp_ctx,
1515 task->model_ops, &ldap_stream_priv_ops,
1516 "unix", ldapi_path, NULL,
1517 lpcfg_socket_options(task->lp_ctx),
1518 ldap_service,
1519 task->process_context);
1520 talloc_free(ldapi_path);
1521 if (!NT_STATUS_IS_OK(status)) {
1522 DBG_ERR("ldapsrv failed to bind to %s - %s\n",
1523 ldapi_path, nt_errstr(status));
1526 #endif
1528 /* register the server */
1529 irpc_add_name(task->msg_ctx, "ldap_server");
1531 task->private_data = ldap_service;
1533 return NT_STATUS_OK;
1535 failed:
1536 task_server_terminate(task, "Failed to startup ldap server task", true);
1537 return status;
1541 * Open a database to be later used by LDB wrap code (although it should be
1542 * plumbed through correctly eventually).
1544 static void ldapsrv_post_fork(struct task_server *task, struct process_details *pd)
1546 struct ldapsrv_service *ldap_service =
1547 talloc_get_type_abort(task->private_data, struct ldapsrv_service);
1550 * As ldapsrv_before_loop() may changed the values for the parent loop
1551 * we need to adjust the pointers to the correct value in the child
1553 ldap_service->lp_ctx = task->lp_ctx;
1554 ldap_service->current_ev = task->event_ctx;
1555 ldap_service->current_msg = task->msg_ctx;
1557 ldap_service->sam_ctx = samdb_connect(ldap_service,
1558 ldap_service->current_ev,
1559 ldap_service->lp_ctx,
1560 system_session(ldap_service->lp_ctx),
1561 NULL,
1563 if (ldap_service->sam_ctx == NULL) {
1564 task_server_terminate(task, "Cannot open system session LDB",
1565 true);
1566 return;
1570 static void ldapsrv_before_loop(struct task_server *task)
1572 struct ldapsrv_service *ldap_service =
1573 talloc_get_type_abort(task->private_data, struct ldapsrv_service);
1574 NTSTATUS status;
1576 if (ldap_service->sam_ctx != NULL) {
1578 * Make sure the values are still the same
1579 * as set in ldapsrv_post_fork()
1581 SMB_ASSERT(task->lp_ctx == ldap_service->lp_ctx);
1582 SMB_ASSERT(task->event_ctx == ldap_service->current_ev);
1583 SMB_ASSERT(task->msg_ctx == ldap_service->current_msg);
1584 } else {
1586 * We need to adjust the pointers to the correct value
1587 * in the parent loop.
1589 ldap_service->lp_ctx = task->lp_ctx;
1590 ldap_service->current_ev = task->event_ctx;
1591 ldap_service->current_msg = task->msg_ctx;
1594 status = imessaging_register(ldap_service->current_msg,
1595 ldap_service,
1596 MSG_RELOAD_TLS_CERTIFICATES,
1597 ldap_reload_certs);
1598 if (!NT_STATUS_IS_OK(status)) {
1599 task_server_terminate(task, "Cannot register ldap_reload_certs",
1600 true);
1601 return;
1606 * Check the size of an ldap request packet.
1608 * For authenticated connections the maximum packet size is controlled by
1609 * the smb.conf parameter "ldap max authenticated request size"
1611 * For anonymous connections the maximum packet size is controlled by
1612 * the smb.conf parameter "ldap max anonymous request size"
1614 static int ldapsrv_check_packet_size(
1615 struct ldapsrv_connection *conn,
1616 size_t size)
1618 bool is_anonymous = false;
1619 size_t max_size = 0;
1621 max_size = lpcfg_ldap_max_anonymous_request_size(conn->lp_ctx);
1622 if (size <= max_size) {
1623 return LDAP_SUCCESS;
1627 * Request is larger than the maximum unauthenticated request size.
1628 * As this code is called frequently we avoid calling
1629 * security_token_is_anonymous if possible
1631 if (conn->session_info != NULL &&
1632 conn->session_info->security_token != NULL) {
1633 is_anonymous = security_token_is_anonymous(
1634 conn->session_info->security_token);
1637 if (is_anonymous) {
1638 DBG_WARNING(
1639 "LDAP request size (%zu) exceeds (%zu)\n",
1640 size,
1641 max_size);
1642 return LDAP_UNWILLING_TO_PERFORM;
1645 max_size = lpcfg_ldap_max_authenticated_request_size(conn->lp_ctx);
1646 if (size > max_size) {
1647 DBG_WARNING(
1648 "LDAP request size (%zu) exceeds (%zu)\n",
1649 size,
1650 max_size);
1651 return LDAP_UNWILLING_TO_PERFORM;
1653 return LDAP_SUCCESS;
1658 * Check that the blob contains enough data to be a valid packet
1659 * If there is a packet header check the size to ensure that it does not
1660 * exceed the maximum sizes.
1663 static NTSTATUS ldapsrv_packet_check(
1664 struct tstream_context *stream,
1665 void *private_data,
1666 DATA_BLOB blob,
1667 size_t *packet_size)
1669 NTSTATUS ret;
1670 struct ldapsrv_connection *conn = private_data;
1671 int result = LDB_SUCCESS;
1673 ret = ldap_full_packet(stream, private_data, blob, packet_size);
1674 if (!NT_STATUS_IS_OK(ret)) {
1675 return ret;
1677 result = ldapsrv_check_packet_size(conn, *packet_size);
1678 if (result != LDAP_SUCCESS) {
1679 return NT_STATUS_LDAP(result);
1681 return NT_STATUS_OK;
1684 NTSTATUS server_service_ldap_init(TALLOC_CTX *ctx)
1686 static const struct service_details details = {
1687 .inhibit_fork_on_accept = false,
1688 .inhibit_pre_fork = false,
1689 .task_init = ldapsrv_task_init,
1690 .post_fork = ldapsrv_post_fork,
1691 .before_loop = ldapsrv_before_loop,
1693 return register_server_service(ctx, "ldap", &details);