2 Unix SMB/CIFS mplementation.
3 LDAP protocol helper functions for SAMBA
5 Copyright (C) Andrew Tridgell 2004
6 Copyright (C) Volker Lendecke 2004
7 Copyright (C) Stefan Metzmacher 2004
8 Copyright (C) Simo Sorce 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 2 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, write to the Free Software
22 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
27 #include "libcli/util/asn_1.h"
28 #include "dlinklist.h"
29 #include "lib/events/events.h"
30 #include "lib/socket/socket.h"
31 #include "libcli/ldap/ldap.h"
32 #include "libcli/ldap/ldap_client.h"
33 #include "libcli/composite/composite.h"
34 #include "lib/stream/packet.h"
35 #include "auth/gensec/gensec.h"
36 #include "system/time.h"
40 create a new ldap_connection stucture. The event context is optional
42 struct ldap_connection *ldap_new_connection(TALLOC_CTX *mem_ctx,
43 struct event_context *ev)
45 struct ldap_connection *conn;
47 conn = talloc_zero(mem_ctx, struct ldap_connection);
53 ev = event_context_init(conn);
60 conn->next_messageid = 1;
61 conn->event.event_ctx = ev;
63 /* set a reasonable request timeout */
66 /* explicitly avoid reconnections by default */
67 conn->reconnect.max_retries = 0;
73 the connection is dead
75 static void ldap_connection_dead(struct ldap_connection *conn)
77 struct ldap_request *req;
79 /* return an error for any pending request ... */
80 while (conn->pending) {
82 DLIST_REMOVE(req->conn->pending, req);
83 req->state = LDAP_REQUEST_DONE;
84 req->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR;
90 talloc_free(conn->tls);
91 talloc_free(conn->sock); /* this will also free event.fde */
92 talloc_free(conn->packet);
95 conn->event.fde = NULL;
99 static void ldap_reconnect(struct ldap_connection *conn);
104 static void ldap_error_handler(void *private_data, NTSTATUS status)
106 struct ldap_connection *conn = talloc_get_type(private_data,
107 struct ldap_connection);
108 ldap_connection_dead(conn);
110 /* but try to reconnect so that the ldb client can go on */
111 ldap_reconnect(conn);
116 match up with a pending message, adding to the replies list
118 static void ldap_match_message(struct ldap_connection *conn, struct ldap_message *msg)
120 struct ldap_request *req;
122 for (req=conn->pending; req; req=req->next) {
123 if (req->messageid == msg->messageid) break;
125 /* match a zero message id to the last request sent.
126 It seems that servers send 0 if unable to parse */
127 if (req == NULL && msg->messageid == 0) {
131 DEBUG(0,("ldap: no matching message id for %u\n",
137 /* add to the list of replies received */
138 talloc_steal(req, msg);
139 req->replies = talloc_realloc(req, req->replies,
140 struct ldap_message *, req->num_replies+1);
141 if (req->replies == NULL) {
142 req->status = NT_STATUS_NO_MEMORY;
143 req->state = LDAP_REQUEST_DONE;
144 DLIST_REMOVE(conn->pending, req);
151 req->replies[req->num_replies] = talloc_steal(req->replies, msg);
154 if (msg->type != LDAP_TAG_SearchResultEntry &&
155 msg->type != LDAP_TAG_SearchResultReference) {
156 /* currently only search results expect multiple
158 req->state = LDAP_REQUEST_DONE;
159 DLIST_REMOVE(conn->pending, req);
169 check if a blob is a complete ldap packet
170 handle wrapper or unwrapped connections
172 NTSTATUS ldap_complete_packet(void *private_data, DATA_BLOB blob, size_t *size)
174 struct ldap_connection *conn = talloc_get_type(private_data,
175 struct ldap_connection);
176 if (conn->enable_wrap) {
177 return packet_full_request_u32(private_data, blob, size);
179 return ldap_full_packet(private_data, blob, size);
183 decode/process plain data
185 static NTSTATUS ldap_decode_plain(struct ldap_connection *conn, DATA_BLOB blob)
187 struct asn1_data asn1;
188 struct ldap_message *msg = talloc(conn, struct ldap_message);
191 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
194 if (!asn1_load(&asn1, blob)) {
195 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
198 if (!ldap_decode(&asn1, msg)) {
199 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
202 ldap_match_message(conn, msg);
204 data_blob_free(&blob);
210 decode/process wrapped data
212 static NTSTATUS ldap_decode_wrapped(struct ldap_connection *conn, DATA_BLOB blob)
214 DATA_BLOB wrapped, unwrapped;
215 struct asn1_data asn1;
216 struct ldap_message *msg = talloc(conn, struct ldap_message);
220 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
223 wrapped = data_blob_const(blob.data+4, blob.length-4);
225 status = gensec_unwrap(conn->gensec, msg, &wrapped, &unwrapped);
226 if (!NT_STATUS_IS_OK(status)) {
227 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
230 data_blob_free(&blob);
232 if (!asn1_load(&asn1, unwrapped)) {
233 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
236 while (ldap_decode(&asn1, msg)) {
237 ldap_match_message(conn, msg);
238 msg = talloc(conn, struct ldap_message);
249 handle ldap recv events
251 static NTSTATUS ldap_recv_handler(void *private_data, DATA_BLOB blob)
253 struct ldap_connection *conn = talloc_get_type(private_data,
254 struct ldap_connection);
255 if (conn->enable_wrap) {
256 return ldap_decode_wrapped(conn, blob);
259 return ldap_decode_plain(conn, blob);
264 handle ldap socket events
266 static void ldap_io_handler(struct event_context *ev, struct fd_event *fde,
267 uint16_t flags, void *private_data)
269 struct ldap_connection *conn = talloc_get_type(private_data,
270 struct ldap_connection);
271 if (flags & EVENT_FD_WRITE) {
272 packet_queue_run(conn->packet);
273 if (conn->tls == NULL) return;
275 if (flags & EVENT_FD_READ) {
276 packet_recv(conn->packet);
283 static NTSTATUS ldap_parse_basic_url(TALLOC_CTX *mem_ctx, const char *url,
284 char **host, uint16_t *port, BOOL *ldaps)
292 /* skip leading "URL:" (if any) */
293 if (strncasecmp(p, "URL:", 4) == 0) {
298 SMB_ASSERT(sizeof(protocol)>10 && sizeof(tmp_host)>254);
300 ret = sscanf(p, "%10[^:]://%254[^:/]:%d", protocol, tmp_host, &tmp_port);
302 return NT_STATUS_INVALID_PARAMETER;
305 if (strequal(protocol, "ldap")) {
308 } else if (strequal(protocol, "ldaps")) {
312 DEBUG(0, ("unrecognised ldap protocol (%s)!\n", protocol));
313 return NT_STATUS_PROTOCOL_UNREACHABLE;
319 *host = talloc_strdup(mem_ctx, tmp_host);
320 NT_STATUS_HAVE_NO_MEMORY(*host);
326 connect to a ldap server
329 struct ldap_connect_state {
330 struct composite_context *ctx;
331 struct ldap_connection *conn;
334 static void ldap_connect_recv_conn(struct composite_context *ctx);
336 struct composite_context *ldap_connect_send(struct ldap_connection *conn,
339 struct composite_context *result, *ctx;
340 struct ldap_connect_state *state;
342 if (conn->reconnect.url == NULL) {
343 conn->reconnect.url = talloc_strdup(conn, url);
344 if (conn->reconnect.url == NULL) goto failed;
347 result = talloc_zero(NULL, struct composite_context);
348 if (result == NULL) goto failed;
349 result->state = COMPOSITE_STATE_IN_PROGRESS;
350 result->async.fn = NULL;
351 result->event_ctx = conn->event.event_ctx;
353 state = talloc(result, struct ldap_connect_state);
354 if (state == NULL) goto failed;
356 result->private_data = state;
360 state->ctx->status = ldap_parse_basic_url(conn, url, &conn->host,
361 &conn->port, &conn->ldaps);
362 if (!NT_STATUS_IS_OK(state->ctx->status)) {
363 composite_error(state->ctx, state->ctx->status);
367 ctx = socket_connect_multi_send(state, conn->host, 1, &conn->port,
368 conn->event.event_ctx);
369 if (ctx == NULL) goto failed;
371 ctx->async.fn = ldap_connect_recv_conn;
372 ctx->async.private_data = state;
380 static void ldap_connect_recv_conn(struct composite_context *ctx)
382 struct ldap_connect_state *state =
383 talloc_get_type(ctx->async.private_data,
384 struct ldap_connect_state);
385 struct ldap_connection *conn = state->conn;
388 state->ctx->status = socket_connect_multi_recv(ctx, state, &conn->sock,
390 if (!composite_is_ok(state->ctx)) return;
392 /* setup a handler for events on this socket */
393 conn->event.fde = event_add_fd(conn->event.event_ctx, conn->sock,
394 socket_get_fd(conn->sock),
395 EVENT_FD_READ, ldap_io_handler, conn);
396 if (conn->event.fde == NULL) {
397 composite_error(state->ctx, NT_STATUS_INTERNAL_ERROR);
401 conn->tls = tls_init_client(conn->sock, conn->event.fde, conn->ldaps);
402 if (conn->tls == NULL) {
403 talloc_free(conn->sock);
406 talloc_steal(conn, conn->tls);
407 talloc_steal(conn->tls, conn->sock);
409 conn->packet = packet_init(conn);
410 if (conn->packet == NULL) {
411 talloc_free(conn->sock);
414 packet_set_private(conn->packet, conn);
415 packet_set_tls(conn->packet, conn->tls);
416 packet_set_callback(conn->packet, ldap_recv_handler);
417 packet_set_full_request(conn->packet, ldap_complete_packet);
418 packet_set_error_handler(conn->packet, ldap_error_handler);
419 packet_set_event_context(conn->packet, conn->event.event_ctx);
420 packet_set_fde(conn->packet, conn->event.fde);
421 packet_set_serialise(conn->packet);
423 composite_done(state->ctx);
428 NTSTATUS ldap_connect_recv(struct composite_context *ctx)
430 NTSTATUS status = composite_wait(ctx);
435 NTSTATUS ldap_connect(struct ldap_connection *conn, const char *url)
437 struct composite_context *ctx = ldap_connect_send(conn, url);
438 return ldap_connect_recv(ctx);
441 /* Actually this function is NOT ASYNC safe, FIXME? */
442 static void ldap_reconnect(struct ldap_connection *conn)
445 time_t now = time(NULL);
447 /* do we have set up reconnect ? */
448 if (conn->reconnect.max_retries == 0) return;
450 /* is the retry time expired ? */
451 if (now > conn->reconnect.previous + 30) {
452 conn->reconnect.retries = 0;
453 conn->reconnect.previous = now;
456 /* are we reconnectind too often and too fast? */
457 if (conn->reconnect.retries > conn->reconnect.max_retries) return;
459 /* keep track of the number of reconnections */
460 conn->reconnect.retries++;
463 status = ldap_connect(conn, conn->reconnect.url);
464 if ( ! NT_STATUS_IS_OK(status)) {
469 status = ldap_rebind(conn);
470 if ( ! NT_STATUS_IS_OK(status)) {
471 ldap_connection_dead(conn);
475 /* destroy an open ldap request */
476 static int ldap_request_destructor(void *ptr)
478 struct ldap_request *req = talloc_get_type(ptr, struct ldap_request);
479 if (req->state == LDAP_REQUEST_PENDING) {
480 DLIST_REMOVE(req->conn->pending, req);
486 called on timeout of a ldap request
488 static void ldap_request_timeout(struct event_context *ev, struct timed_event *te,
489 struct timeval t, void *private_data)
491 struct ldap_request *req = talloc_get_type(private_data, struct ldap_request);
492 req->status = NT_STATUS_IO_TIMEOUT;
493 if (req->state == LDAP_REQUEST_PENDING) {
494 DLIST_REMOVE(req->conn->pending, req);
496 req->state = LDAP_REQUEST_DONE;
504 called on completion of a one-way ldap request
506 static void ldap_request_complete(struct event_context *ev, struct timed_event *te,
507 struct timeval t, void *private_data)
509 struct ldap_request *req = talloc_get_type(private_data, struct ldap_request);
516 send a ldap message - async interface
518 struct ldap_request *ldap_request_send(struct ldap_connection *conn,
519 struct ldap_message *msg)
521 struct ldap_request *req;
522 NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
524 req = talloc_zero(conn, struct ldap_request);
525 if (req == NULL) return NULL;
527 if (conn->tls == NULL) {
528 status = NT_STATUS_INVALID_CONNECTION;
532 req->state = LDAP_REQUEST_SEND;
534 req->messageid = conn->next_messageid++;
535 if (conn->next_messageid == 0) {
536 conn->next_messageid = 1;
538 req->type = msg->type;
539 if (req->messageid == -1) {
543 talloc_set_destructor(req, ldap_request_destructor);
545 msg->messageid = req->messageid;
547 if (!ldap_encode(msg, &req->data, req)) {
551 /* possibly encrypt/sign the request */
552 if (conn->enable_wrap) {
555 status = gensec_wrap(conn->gensec, req, &req->data, &wrapped);
556 if (!NT_STATUS_IS_OK(status)) {
559 data_blob_free(&req->data);
560 req->data = data_blob_talloc(req, NULL, wrapped.length + 4);
561 if (req->data.data == NULL) {
564 RSIVAL(req->data.data, 0, wrapped.length);
565 memcpy(req->data.data+4, wrapped.data, wrapped.length);
566 data_blob_free(&wrapped);
569 status = packet_send(conn->packet, req->data);
570 if (!NT_STATUS_IS_OK(status)) {
574 /* some requests don't expect a reply, so don't add those to the
576 if (req->type == LDAP_TAG_AbandonRequest ||
577 req->type == LDAP_TAG_UnbindRequest) {
578 req->status = NT_STATUS_OK;
579 req->state = LDAP_REQUEST_DONE;
580 /* we can't call the async callback now, as it isn't setup, so
581 call it as next event */
582 event_add_timed(conn->event.event_ctx, req, timeval_zero(),
583 ldap_request_complete, req);
587 req->state = LDAP_REQUEST_PENDING;
588 DLIST_ADD(conn->pending, req);
590 /* put a timeout on the request */
591 req->time_event = event_add_timed(conn->event.event_ctx, req,
592 timeval_current_ofs(conn->timeout, 0),
593 ldap_request_timeout, req);
598 req->status = status;
599 req->state = LDAP_REQUEST_ERROR;
600 event_add_timed(conn->event.event_ctx, req, timeval_zero(),
601 ldap_request_complete, req);
608 wait for a request to complete
609 note that this does not destroy the request
611 NTSTATUS ldap_request_wait(struct ldap_request *req)
613 while (req->state <= LDAP_REQUEST_DONE) {
614 if (event_loop_once(req->conn->event.event_ctx) != 0) {
615 req->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR;
624 a mapping of ldap response code to strings
626 static const struct {
627 enum ldap_result_code code;
629 } ldap_code_map[] = {
630 #define _LDAP_MAP_CODE(c) { c, #c }
631 _LDAP_MAP_CODE(LDAP_SUCCESS),
632 _LDAP_MAP_CODE(LDAP_OPERATIONS_ERROR),
633 _LDAP_MAP_CODE(LDAP_PROTOCOL_ERROR),
634 _LDAP_MAP_CODE(LDAP_TIME_LIMIT_EXCEEDED),
635 _LDAP_MAP_CODE(LDAP_SIZE_LIMIT_EXCEEDED),
636 _LDAP_MAP_CODE(LDAP_COMPARE_FALSE),
637 _LDAP_MAP_CODE(LDAP_COMPARE_TRUE),
638 _LDAP_MAP_CODE(LDAP_AUTH_METHOD_NOT_SUPPORTED),
639 _LDAP_MAP_CODE(LDAP_STRONG_AUTH_REQUIRED),
640 _LDAP_MAP_CODE(LDAP_REFERRAL),
641 _LDAP_MAP_CODE(LDAP_ADMIN_LIMIT_EXCEEDED),
642 _LDAP_MAP_CODE(LDAP_UNAVAILABLE_CRITICAL_EXTENSION),
643 _LDAP_MAP_CODE(LDAP_CONFIDENTIALITY_REQUIRED),
644 _LDAP_MAP_CODE(LDAP_SASL_BIND_IN_PROGRESS),
645 _LDAP_MAP_CODE(LDAP_NO_SUCH_ATTRIBUTE),
646 _LDAP_MAP_CODE(LDAP_UNDEFINED_ATTRIBUTE_TYPE),
647 _LDAP_MAP_CODE(LDAP_INAPPROPRIATE_MATCHING),
648 _LDAP_MAP_CODE(LDAP_CONSTRAINT_VIOLATION),
649 _LDAP_MAP_CODE(LDAP_ATTRIBUTE_OR_VALUE_EXISTS),
650 _LDAP_MAP_CODE(LDAP_INVALID_ATTRIBUTE_SYNTAX),
651 _LDAP_MAP_CODE(LDAP_NO_SUCH_OBJECT),
652 _LDAP_MAP_CODE(LDAP_ALIAS_PROBLEM),
653 _LDAP_MAP_CODE(LDAP_INVALID_DN_SYNTAX),
654 _LDAP_MAP_CODE(LDAP_ALIAS_DEREFERENCING_PROBLEM),
655 _LDAP_MAP_CODE(LDAP_INAPPROPRIATE_AUTHENTICATION),
656 _LDAP_MAP_CODE(LDAP_INVALID_CREDENTIALS),
657 _LDAP_MAP_CODE(LDAP_INSUFFICIENT_ACCESS_RIGHTs),
658 _LDAP_MAP_CODE(LDAP_BUSY),
659 _LDAP_MAP_CODE(LDAP_UNAVAILABLE),
660 _LDAP_MAP_CODE(LDAP_UNWILLING_TO_PERFORM),
661 _LDAP_MAP_CODE(LDAP_LOOP_DETECT),
662 _LDAP_MAP_CODE(LDAP_NAMING_VIOLATION),
663 _LDAP_MAP_CODE(LDAP_OBJECT_CLASS_VIOLATION),
664 _LDAP_MAP_CODE(LDAP_NOT_ALLOWED_ON_NON_LEAF),
665 _LDAP_MAP_CODE(LDAP_NOT_ALLOWED_ON_RDN),
666 _LDAP_MAP_CODE(LDAP_ENTRY_ALREADY_EXISTS),
667 _LDAP_MAP_CODE(LDAP_OBJECT_CLASS_MODS_PROHIBITED),
668 _LDAP_MAP_CODE(LDAP_AFFECTS_MULTIPLE_DSAS),
669 _LDAP_MAP_CODE(LDAP_OTHER)
673 used to setup the status code from a ldap response
675 NTSTATUS ldap_check_response(struct ldap_connection *conn, struct ldap_Result *r)
678 const char *codename = "unknown";
680 if (r->resultcode == LDAP_SUCCESS) {
684 if (conn->last_error) {
685 talloc_free(conn->last_error);
688 for (i=0;i<ARRAY_SIZE(ldap_code_map);i++) {
689 if (r->resultcode == ldap_code_map[i].code) {
690 codename = ldap_code_map[i].str;
695 conn->last_error = talloc_asprintf(conn, "LDAP error %u %s - %s <%s> <%s>",
698 r->dn?r->dn:"(NULL)",
699 r->errormessage?r->errormessage:"",
700 r->referral?r->referral:"");
702 return NT_STATUS_LDAP(r->resultcode);
706 return error string representing the last error
708 const char *ldap_errstr(struct ldap_connection *conn, NTSTATUS status)
710 if (NT_STATUS_IS_LDAP(status) && conn->last_error != NULL) {
711 return conn->last_error;
713 return nt_errstr(status);
718 return the Nth result message, waiting if necessary
720 NTSTATUS ldap_result_n(struct ldap_request *req, int n, struct ldap_message **msg)
724 NT_STATUS_HAVE_NO_MEMORY(req);
726 while (req->state <= LDAP_REQUEST_DONE && n >= req->num_replies) {
727 if (event_loop_once(req->conn->event.event_ctx) != 0) {
728 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
732 if (n < req->num_replies) {
733 *msg = req->replies[n];
737 if (!NT_STATUS_IS_OK(req->status)) {
741 return NT_STATUS_NO_MORE_ENTRIES;
746 return a single result message, checking if it is of the expected LDAP type
748 NTSTATUS ldap_result_one(struct ldap_request *req, struct ldap_message **msg, int type)
751 status = ldap_result_n(req, 0, msg);
752 if (!NT_STATUS_IS_OK(status)) {
755 if ((*msg)->type != type) {
757 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
763 a simple ldap transaction, for single result requests that only need a status code
764 this relies on single valued requests having the response type == request type + 1
766 NTSTATUS ldap_transaction(struct ldap_connection *conn, struct ldap_message *msg)
768 struct ldap_request *req = ldap_request_send(conn, msg);
769 struct ldap_message *res;
771 status = ldap_result_n(req, 0, &res);
772 if (!NT_STATUS_IS_OK(status)) {
776 if (res->type != msg->type + 1) {
778 return NT_STATUS_LDAP(LDAP_PROTOCOL_ERROR);
780 status = ldap_check_response(conn, &res->r.GeneralResult);