2 Unix SMB/CIFS implementation.
3 Infrastructure for async ldap client requests
4 Copyright (C) Volker Lendecke 2009
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "system/network.h"
23 #include "system/locale.h"
24 #include "lib/util/talloc_stack.h"
25 #include "lib/util/samba_util.h"
26 #include "lib/util_tsock.h"
27 #include "../lib/util/asn1.h"
28 #include "../lib/tsocket/tsocket.h"
29 #include "../lib/util/tevent_unix.h"
31 static TLDAPRC tldap_simple_recv(struct tevent_req *req);
32 static bool tldap_msg_set_pending(struct tevent_req *req);
34 #define TEVENT_TLDAP_RC_MAGIC (0x87bcd26e)
36 bool tevent_req_ldap_error(struct tevent_req *req, TLDAPRC rc)
40 if (TLDAP_RC_IS_SUCCESS(rc)) {
44 err = TEVENT_TLDAP_RC_MAGIC;
46 err |= TLDAP_RC_V(rc);
48 return tevent_req_error(req, err);
51 bool tevent_req_is_ldap_error(struct tevent_req *req, TLDAPRC *perr)
53 enum tevent_req_state state;
56 if (!tevent_req_is_error(req, &state, &err)) {
60 case TEVENT_REQ_TIMED_OUT:
61 *perr = TLDAP_TIMEOUT;
63 case TEVENT_REQ_NO_MEMORY:
64 *perr = TLDAP_NO_MEMORY;
66 case TEVENT_REQ_USER_ERROR:
67 if ((err >> 32) != TEVENT_TLDAP_RC_MAGIC) {
70 *perr = TLDAP_RC(err & 0xffffffff);
73 *perr = TLDAP_OPERATIONS_ERROR;
79 struct tldap_ctx_attribute {
84 struct tldap_context {
86 struct tstream_context *conn;
88 struct tevent_queue *outgoing;
89 struct tevent_req **pending;
90 struct tevent_req *read_req;
92 /* For the sync wrappers we need something like get_last_error... */
93 struct tldap_message *last_msg;
96 void (*log_fn)(void *context, enum tldap_debug_level level,
97 const char *fmt, va_list ap);
100 struct tldap_ctx_attribute *ctx_attrs;
103 struct tldap_message {
104 struct asn1_data *data;
111 struct tldap_attribute *attribs;
113 /* Error data sent by the server */
116 char *res_diagnosticmessage;
118 DATA_BLOB res_serverSaslCreds;
119 struct tldap_control *res_sctrls;
121 /* Controls sent by the server */
122 struct tldap_control *ctrls;
125 void tldap_set_debug(struct tldap_context *ld,
126 void (*log_fn)(void *log_private,
127 enum tldap_debug_level level,
129 va_list ap) PRINTF_ATTRIBUTE(3,0),
133 ld->log_private = log_private;
136 static void tldap_debug(
137 struct tldap_context *ld,
138 enum tldap_debug_level level,
139 const char *fmt, ...) PRINTF_ATTRIBUTE(3,4);
141 static void tldap_debug(struct tldap_context *ld,
142 enum tldap_debug_level level,
143 const char *fmt, ...)
149 if (ld->log_fn == NULL) {
153 ld->log_fn(ld->log_private, level, fmt, ap);
157 static int tldap_next_msgid(struct tldap_context *ld)
161 result = ld->msgid++;
162 if (ld->msgid == INT_MAX) {
168 struct tldap_context *tldap_context_create(TALLOC_CTX *mem_ctx, int fd)
170 struct tldap_context *ctx;
173 ctx = talloc_zero(mem_ctx, struct tldap_context);
177 ret = tstream_bsd_existing_socket(ctx, fd, &ctx->conn);
184 ctx->outgoing = tevent_queue_create(ctx, "tldap_outgoing");
185 if (ctx->outgoing == NULL) {
192 bool tldap_connection_ok(struct tldap_context *ld)
200 if (ld->conn == NULL) {
204 ret = tstream_pending_bytes(ld->conn);
212 static size_t tldap_pending_reqs(struct tldap_context *ld)
214 return talloc_array_length(ld->pending);
217 struct tstream_context *tldap_get_tstream(struct tldap_context *ld)
222 void tldap_set_tstream(struct tldap_context *ld,
223 struct tstream_context *stream)
228 static struct tldap_ctx_attribute *tldap_context_findattr(
229 struct tldap_context *ld, const char *name)
233 num_attrs = talloc_array_length(ld->ctx_attrs);
235 for (i=0; i<num_attrs; i++) {
236 if (strcmp(ld->ctx_attrs[i].name, name) == 0) {
237 return &ld->ctx_attrs[i];
243 bool tldap_context_setattr(struct tldap_context *ld,
244 const char *name, const void *_pptr)
246 struct tldap_ctx_attribute *tmp, *attr;
249 void **pptr = (void **)discard_const_p(void,_pptr);
251 attr = tldap_context_findattr(ld, name);
254 * We don't actually delete attrs, we don't expect tons of
255 * attributes being shuffled around.
257 TALLOC_FREE(attr->ptr);
259 attr->ptr = talloc_move(ld->ctx_attrs, pptr);
265 tmpname = talloc_strdup(ld, name);
266 if (tmpname == NULL) {
270 num_attrs = talloc_array_length(ld->ctx_attrs);
272 tmp = talloc_realloc(ld, ld->ctx_attrs, struct tldap_ctx_attribute,
275 TALLOC_FREE(tmpname);
278 tmp[num_attrs].name = talloc_move(tmp, &tmpname);
280 tmp[num_attrs].ptr = talloc_move(tmp, pptr);
282 tmp[num_attrs].ptr = NULL;
289 void *tldap_context_getattr(struct tldap_context *ld, const char *name)
291 struct tldap_ctx_attribute *attr = tldap_context_findattr(ld, name);
299 struct read_ldap_state {
304 static ssize_t read_ldap_more(uint8_t *buf, size_t buflen, void *private_data);
305 static void read_ldap_done(struct tevent_req *subreq);
307 static struct tevent_req *read_ldap_send(TALLOC_CTX *mem_ctx,
308 struct tevent_context *ev,
309 struct tstream_context *conn)
311 struct tevent_req *req, *subreq;
312 struct read_ldap_state *state;
314 req = tevent_req_create(mem_ctx, &state, struct read_ldap_state);
320 subreq = tstream_read_packet_send(state, ev, conn, 2, read_ldap_more,
322 if (tevent_req_nomem(subreq, req)) {
323 return tevent_req_post(req, ev);
325 tevent_req_set_callback(subreq, read_ldap_done, req);
329 static ssize_t read_ldap_more(uint8_t *buf, size_t buflen, void *private_data)
331 struct read_ldap_state *state = talloc_get_type_abort(
332 private_data, struct read_ldap_state);
337 /* We've been here, we're done */
342 * From ldap.h: LDAP_TAG_MESSAGE is 0x30
344 if (buf[0] != 0x30) {
349 if ((len & 0x80) == 0) {
354 lensize = (len & 0x7f);
358 /* Please get us the full length */
361 if (buflen > 2 + lensize) {
365 if (buflen != 2 + lensize) {
369 for (i=0; i<lensize; i++) {
370 len = (len << 8) | buf[2+i];
375 static void read_ldap_done(struct tevent_req *subreq)
377 struct tevent_req *req = tevent_req_callback_data(
378 subreq, struct tevent_req);
379 struct read_ldap_state *state = tevent_req_data(
380 req, struct read_ldap_state);
384 nread = tstream_read_packet_recv(subreq, state, &state->buf, &err);
387 tevent_req_error(req, err);
390 tevent_req_done(req);
393 static ssize_t read_ldap_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
394 uint8_t **pbuf, int *perrno)
396 struct read_ldap_state *state = tevent_req_data(
397 req, struct read_ldap_state);
399 if (tevent_req_is_unix_error(req, perrno)) {
402 *pbuf = talloc_move(mem_ctx, &state->buf);
403 return talloc_get_size(*pbuf);
406 struct tldap_msg_state {
407 struct tldap_context *ld;
408 struct tevent_context *ev;
412 struct asn1_data *data;
416 static bool tldap_push_controls(struct asn1_data *data,
417 struct tldap_control *sctrls,
422 if ((sctrls == NULL) || (num_sctrls == 0)) {
426 if (!asn1_push_tag(data, ASN1_CONTEXT(0))) return false;
428 for (i=0; i<num_sctrls; i++) {
429 struct tldap_control *c = &sctrls[i];
430 if (!asn1_push_tag(data, ASN1_SEQUENCE(0))) return false;
431 if (!asn1_write_OctetString(data, c->oid, strlen(c->oid))) return false;
433 if (!asn1_write_BOOLEAN(data, true)) return false;
435 if (c->value.data != NULL) {
436 if (!asn1_write_OctetString(data, c->value.data,
437 c->value.length)) return false;
439 if (!asn1_pop_tag(data)) return false; /* ASN1_SEQUENCE(0) */
442 return asn1_pop_tag(data); /* ASN1_CONTEXT(0) */
445 #define tldap_context_disconnect(ld, status) \
446 _tldap_context_disconnect(ld, status, __location__)
448 static void _tldap_context_disconnect(struct tldap_context *ld,
450 const char *location)
452 if (ld->conn == NULL) {
454 * We don't need to tldap_debug() on
455 * a potential 2nd run.
457 * The rest of the function would just
458 * be a noop for the 2nd run anyway.
463 tldap_debug(ld, TLDAP_DEBUG_WARNING,
464 "tldap_context_disconnect: %s at %s\n",
465 tldap_rc2string(status),
467 tevent_queue_stop(ld->outgoing);
468 TALLOC_FREE(ld->read_req);
469 TALLOC_FREE(ld->conn);
471 while (talloc_array_length(ld->pending) > 0) {
472 struct tevent_req *req = NULL;
473 struct tldap_msg_state *state = NULL;
475 req = ld->pending[0];
476 state = tevent_req_data(req, struct tldap_msg_state);
477 tevent_req_defer_callback(req, state->ev);
478 tevent_req_ldap_error(req, status);
482 static void tldap_msg_sent(struct tevent_req *subreq);
483 static void tldap_msg_received(struct tevent_req *subreq);
485 static struct tevent_req *tldap_msg_send(TALLOC_CTX *mem_ctx,
486 struct tevent_context *ev,
487 struct tldap_context *ld,
488 int id, struct asn1_data *data,
489 struct tldap_control *sctrls,
492 struct tevent_req *req, *subreq;
493 struct tldap_msg_state *state;
497 tldap_debug(ld, TLDAP_DEBUG_TRACE, "tldap_msg_send: sending msg %d\n",
500 req = tevent_req_create(mem_ctx, &state, struct tldap_msg_state);
508 ok = tldap_connection_ok(ld);
510 tevent_req_ldap_error(req, TLDAP_SERVER_DOWN);
511 return tevent_req_post(req, ev);
514 if (!tldap_push_controls(data, sctrls, num_sctrls)) {
515 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
516 return tevent_req_post(req, ev);
520 if (!asn1_pop_tag(data)) {
521 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
522 return tevent_req_post(req, ev);
525 if (!asn1_blob(data, &blob)) {
526 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
527 return tevent_req_post(req, ev);
530 if (!tldap_msg_set_pending(req)) {
532 return tevent_req_post(req, ev);;
535 state->iov.iov_base = (void *)blob.data;
536 state->iov.iov_len = blob.length;
538 subreq = tstream_writev_queue_send(state, ev, ld->conn, ld->outgoing,
540 if (tevent_req_nomem(subreq, req)) {
541 return tevent_req_post(req, ev);
543 tevent_req_set_callback(subreq, tldap_msg_sent, req);
547 static void tldap_msg_unset_pending(struct tevent_req *req)
549 struct tldap_msg_state *state = tevent_req_data(
550 req, struct tldap_msg_state);
551 struct tldap_context *ld = state->ld;
552 int num_pending = tldap_pending_reqs(ld);
555 tevent_req_set_cleanup_fn(req, NULL);
557 for (i=0; i<num_pending; i++) {
558 if (req == ld->pending[i]) {
562 if (i == num_pending) {
564 * Something's seriously broken. Just returning here is the
565 * right thing nevertheless, the point of this routine is to
566 * remove ourselves from cli->pending.
571 if (num_pending == 1) {
572 TALLOC_FREE(ld->pending);
577 * Remove ourselves from the cli->pending array
579 if (num_pending > 1) {
580 ld->pending[i] = ld->pending[num_pending-1];
584 * No NULL check here, we're shrinking by sizeof(void *), and
585 * talloc_realloc just adjusts the size for this.
587 ld->pending = talloc_realloc(NULL, ld->pending, struct tevent_req *,
591 static void tldap_msg_cleanup(struct tevent_req *req,
592 enum tevent_req_state req_state)
594 tldap_msg_unset_pending(req);
597 static bool tldap_msg_set_pending(struct tevent_req *req)
599 struct tldap_msg_state *state = tevent_req_data(
600 req, struct tldap_msg_state);
601 struct tldap_context *ld;
602 struct tevent_req **pending;
606 num_pending = tldap_pending_reqs(ld);
608 pending = talloc_realloc(ld, ld->pending, struct tevent_req *,
610 if (pending == NULL) {
613 pending[num_pending] = req;
614 ld->pending = pending;
615 tevent_req_set_cleanup_fn(req, tldap_msg_cleanup);
617 if (ld->read_req != NULL) {
622 * We're the first one, add the read_ldap request that waits for the
623 * answer from the server
625 ld->read_req = read_ldap_send(ld->pending, state->ev, ld->conn);
626 if (ld->read_req == NULL) {
627 tldap_msg_unset_pending(req);
630 tevent_req_set_callback(ld->read_req, tldap_msg_received, ld);
634 static void tldap_msg_sent(struct tevent_req *subreq)
636 struct tevent_req *req = tevent_req_callback_data(
637 subreq, struct tevent_req);
638 struct tldap_msg_state *state = tevent_req_data(
639 req, struct tldap_msg_state);
643 nwritten = tstream_writev_queue_recv(subreq, &err);
645 if (nwritten == -1) {
646 tldap_context_disconnect(state->ld, TLDAP_SERVER_DOWN);
651 static int tldap_msg_msgid(struct tevent_req *req)
653 struct tldap_msg_state *state = tevent_req_data(
654 req, struct tldap_msg_state);
659 static void tldap_msg_received(struct tevent_req *subreq)
661 struct tldap_context *ld = tevent_req_callback_data(
662 subreq, struct tldap_context);
663 struct tevent_req *req;
664 struct tldap_msg_state *state;
665 struct asn1_data *data;
671 TLDAPRC status = TLDAP_PROTOCOL_ERROR;
676 received = read_ldap_recv(subreq, talloc_tos(), &inbuf, &err);
679 if (received == -1) {
680 status = TLDAP_SERVER_DOWN;
684 data = asn1_init(talloc_tos(), ASN1_MAX_TREE_DEPTH);
687 * We have to disconnect all, we can't tell which of
688 * the requests this reply is for.
690 status = TLDAP_NO_MEMORY;
693 asn1_load_nocopy(data, inbuf, received);
696 ok &= asn1_start_tag(data, ASN1_SEQUENCE(0));
697 ok &= asn1_read_Integer(data, &id);
698 ok &= asn1_peek_uint8(data, &type);
701 status = TLDAP_PROTOCOL_ERROR;
705 tldap_debug(ld, TLDAP_DEBUG_TRACE, "tldap_msg_received: got msg %d "
706 "type %d\n", id, (int)type);
712 "tldap_msg_received: got msgid 0 of "
713 "type %"PRIu8", disconnecting\n",
715 tldap_context_disconnect(ld, TLDAP_SERVER_DOWN);
719 num_pending = talloc_array_length(ld->pending);
721 for (i=0; i<num_pending; i++) {
722 if (id == tldap_msg_msgid(ld->pending[i])) {
726 if (i == num_pending) {
727 /* Dump unexpected reply */
728 tldap_debug(ld, TLDAP_DEBUG_WARNING, "tldap_msg_received: "
729 "No request pending for msg %d\n", id);
735 req = ld->pending[i];
736 state = tevent_req_data(req, struct tldap_msg_state);
738 state->inbuf = talloc_move(state, &inbuf);
739 state->data = talloc_move(state, &data);
741 tldap_msg_unset_pending(req);
742 num_pending = talloc_array_length(ld->pending);
744 tevent_req_defer_callback(req, state->ev);
745 tevent_req_done(req);
748 if (num_pending == 0) {
752 state = tevent_req_data(ld->pending[0], struct tldap_msg_state);
753 ld->read_req = read_ldap_send(ld->pending, state->ev, ld->conn);
754 if (ld->read_req == NULL) {
755 status = TLDAP_NO_MEMORY;
758 tevent_req_set_callback(ld->read_req, tldap_msg_received, ld);
762 tldap_context_disconnect(ld, status);
765 static TLDAPRC tldap_msg_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
766 struct tldap_message **pmsg)
768 struct tldap_msg_state *state = tevent_req_data(
769 req, struct tldap_msg_state);
770 struct tldap_message *msg;
774 if (tevent_req_is_ldap_error(req, &err)) {
778 if (!asn1_peek_uint8(state->data, &msgtype)) {
779 return TLDAP_PROTOCOL_ERROR;
783 return TLDAP_SUCCESS;
786 msg = talloc_zero(mem_ctx, struct tldap_message);
788 return TLDAP_NO_MEMORY;
792 msg->inbuf = talloc_move(msg, &state->inbuf);
793 msg->data = talloc_move(msg, &state->data);
797 return TLDAP_SUCCESS;
800 struct tldap_req_state {
802 struct asn1_data *out;
803 struct tldap_message *result;
806 static struct tevent_req *tldap_req_create(TALLOC_CTX *mem_ctx,
807 struct tldap_context *ld,
808 struct tldap_req_state **pstate)
810 struct tevent_req *req;
811 struct tldap_req_state *state;
813 req = tevent_req_create(mem_ctx, &state, struct tldap_req_state);
817 state->out = asn1_init(state, ASN1_MAX_TREE_DEPTH);
818 if (state->out == NULL) {
821 state->id = tldap_next_msgid(ld);
823 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
824 if (!asn1_write_Integer(state->out, state->id)) goto err;
835 static void tldap_save_msg(struct tldap_context *ld, struct tevent_req *req)
837 struct tldap_req_state *state = tevent_req_data(
838 req, struct tldap_req_state);
840 TALLOC_FREE(ld->last_msg);
841 ld->last_msg = talloc_move(ld, &state->result);
844 static char *blob2string_talloc(TALLOC_CTX *mem_ctx, DATA_BLOB blob)
846 char *result = talloc_array(mem_ctx, char, blob.length+1);
848 if (result == NULL) {
852 memcpy(result, blob.data, blob.length);
853 result[blob.length] = '\0';
857 static bool asn1_read_OctetString_talloc(TALLOC_CTX *mem_ctx,
858 struct asn1_data *data,
863 if (!asn1_read_OctetString(data, mem_ctx, &string))
866 result = blob2string_talloc(mem_ctx, string);
868 data_blob_free(&string);
870 if (result == NULL) {
877 static bool tldap_decode_controls(struct tldap_req_state *state);
879 static bool tldap_decode_response(struct tldap_req_state *state)
881 struct asn1_data *data = state->result->data;
882 struct tldap_message *msg = state->result;
886 ok &= asn1_read_enumerated(data, &rc);
888 msg->lderr = TLDAP_RC(rc);
891 ok &= asn1_read_OctetString_talloc(msg, data, &msg->res_matcheddn);
892 ok &= asn1_read_OctetString_talloc(msg, data,
893 &msg->res_diagnosticmessage);
895 if (asn1_peek_tag(data, ASN1_CONTEXT(3))) {
896 ok &= asn1_start_tag(data, ASN1_CONTEXT(3));
897 ok &= asn1_read_OctetString_talloc(msg, data,
899 ok &= asn1_end_tag(data);
901 msg->res_referral = NULL;
907 static void tldap_sasl_bind_done(struct tevent_req *subreq);
909 struct tevent_req *tldap_sasl_bind_send(TALLOC_CTX *mem_ctx,
910 struct tevent_context *ev,
911 struct tldap_context *ld,
913 const char *mechanism,
915 struct tldap_control *sctrls,
917 struct tldap_control *cctrls,
920 struct tevent_req *req, *subreq;
921 struct tldap_req_state *state;
923 req = tldap_req_create(mem_ctx, ld, &state);
932 if (!asn1_push_tag(state->out, TLDAP_REQ_BIND)) goto err;
933 if (!asn1_write_Integer(state->out, ld->ld_version)) goto err;
934 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
936 if (mechanism == NULL) {
937 if (!asn1_push_tag(state->out, ASN1_CONTEXT_SIMPLE(0))) goto err;
938 if (!asn1_write(state->out, creds->data, creds->length)) goto err;
939 if (!asn1_pop_tag(state->out)) goto err;
941 if (!asn1_push_tag(state->out, ASN1_CONTEXT(3))) goto err;
942 if (!asn1_write_OctetString(state->out, mechanism,
943 strlen(mechanism))) goto err;
944 if ((creds != NULL) && (creds->data != NULL)) {
945 if (!asn1_write_OctetString(state->out, creds->data,
946 creds->length)) goto err;
948 if (!asn1_pop_tag(state->out)) goto err;
951 if (!asn1_pop_tag(state->out)) goto err;
953 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
955 if (tevent_req_nomem(subreq, req)) {
956 return tevent_req_post(req, ev);
958 tevent_req_set_callback(subreq, tldap_sasl_bind_done, req);
963 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
964 return tevent_req_post(req, ev);
967 static void tldap_sasl_bind_done(struct tevent_req *subreq)
969 struct tevent_req *req = tevent_req_callback_data(
970 subreq, struct tevent_req);
971 struct tldap_req_state *state = tevent_req_data(
972 req, struct tldap_req_state);
976 rc = tldap_msg_recv(subreq, state, &state->result);
978 if (tevent_req_ldap_error(req, rc)) {
981 if (state->result->type != TLDAP_RES_BIND) {
982 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
986 ok = asn1_start_tag(state->result->data, TLDAP_RES_BIND);
987 ok &= tldap_decode_response(state);
989 if (asn1_peek_tag(state->result->data, ASN1_CONTEXT_SIMPLE(7))) {
992 ok &= asn1_start_tag(state->result->data,
993 ASN1_CONTEXT_SIMPLE(7));
998 len = asn1_tag_remaining(state->result->data);
1003 state->result->res_serverSaslCreds =
1004 data_blob_talloc(state->result, NULL, len);
1005 if (state->result->res_serverSaslCreds.data == NULL) {
1009 ok = asn1_read(state->result->data,
1010 state->result->res_serverSaslCreds.data,
1011 state->result->res_serverSaslCreds.length);
1013 ok &= asn1_end_tag(state->result->data);
1016 ok &= asn1_end_tag(state->result->data);
1022 if (!TLDAP_RC_IS_SUCCESS(state->result->lderr) &&
1023 !TLDAP_RC_EQUAL(state->result->lderr,
1024 TLDAP_SASL_BIND_IN_PROGRESS)) {
1025 tevent_req_ldap_error(req, state->result->lderr);
1028 tevent_req_done(req);
1032 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
1036 TLDAPRC tldap_sasl_bind_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1037 DATA_BLOB *serverSaslCreds)
1039 struct tldap_req_state *state = tevent_req_data(
1040 req, struct tldap_req_state);
1043 if (tevent_req_is_ldap_error(req, &rc)) {
1047 if (serverSaslCreds != NULL) {
1048 serverSaslCreds->data = talloc_move(
1049 mem_ctx, &state->result->res_serverSaslCreds.data);
1050 serverSaslCreds->length =
1051 state->result->res_serverSaslCreds.length;
1054 return state->result->lderr;
1057 TLDAPRC tldap_sasl_bind(struct tldap_context *ld,
1059 const char *mechanism,
1061 struct tldap_control *sctrls,
1063 struct tldap_control *cctrls,
1065 TALLOC_CTX *mem_ctx,
1066 DATA_BLOB *serverSaslCreds)
1068 TALLOC_CTX *frame = talloc_stackframe();
1069 struct tevent_context *ev;
1070 struct tevent_req *req;
1071 TLDAPRC rc = TLDAP_NO_MEMORY;
1073 ev = samba_tevent_context_init(frame);
1077 req = tldap_sasl_bind_send(frame, ev, ld, dn, mechanism, creds,
1078 sctrls, num_sctrls, cctrls, num_cctrls);
1082 if (!tevent_req_poll(req, ev)) {
1083 rc = TLDAP_OPERATIONS_ERROR;
1086 rc = tldap_sasl_bind_recv(req, mem_ctx, serverSaslCreds);
1087 tldap_save_msg(ld, req);
1093 struct tevent_req *tldap_simple_bind_send(TALLOC_CTX *mem_ctx,
1094 struct tevent_context *ev,
1095 struct tldap_context *ld,
1101 if (passwd != NULL) {
1102 cred.data = discard_const_p(uint8_t, passwd);
1103 cred.length = strlen(passwd);
1105 cred.data = discard_const_p(uint8_t, "");
1108 return tldap_sasl_bind_send(mem_ctx, ev, ld, dn, NULL, &cred, NULL, 0,
1112 TLDAPRC tldap_simple_bind_recv(struct tevent_req *req)
1114 return tldap_sasl_bind_recv(req, NULL, NULL);
1117 TLDAPRC tldap_simple_bind(struct tldap_context *ld, const char *dn,
1122 if (passwd != NULL) {
1123 cred.data = discard_const_p(uint8_t, passwd);
1124 cred.length = strlen(passwd);
1126 cred.data = discard_const_p(uint8_t, "");
1129 return tldap_sasl_bind(ld, dn, NULL, &cred, NULL, 0, NULL, 0,
1133 /*****************************************************************************/
1135 /* can't use isalpha() as only a strict set is valid for LDAP */
1137 static bool tldap_is_alpha(char c)
1139 return (((c >= 'a') && (c <= 'z')) || \
1140 ((c >= 'A') && (c <= 'Z')));
1143 static bool tldap_is_adh(char c)
1145 return tldap_is_alpha(c) || isdigit(c) || (c == '-');
1148 #define TLDAP_FILTER_AND ASN1_CONTEXT(0)
1149 #define TLDAP_FILTER_OR ASN1_CONTEXT(1)
1150 #define TLDAP_FILTER_NOT ASN1_CONTEXT(2)
1151 #define TLDAP_FILTER_EQ ASN1_CONTEXT(3)
1152 #define TLDAP_FILTER_SUB ASN1_CONTEXT(4)
1153 #define TLDAP_FILTER_LE ASN1_CONTEXT(5)
1154 #define TLDAP_FILTER_GE ASN1_CONTEXT(6)
1155 #define TLDAP_FILTER_PRES ASN1_CONTEXT_SIMPLE(7)
1156 #define TLDAP_FILTER_APX ASN1_CONTEXT(8)
1157 #define TLDAP_FILTER_EXT ASN1_CONTEXT(9)
1159 #define TLDAP_SUB_INI ASN1_CONTEXT_SIMPLE(0)
1160 #define TLDAP_SUB_ANY ASN1_CONTEXT_SIMPLE(1)
1161 #define TLDAP_SUB_FIN ASN1_CONTEXT_SIMPLE(2)
1164 /* oid's should be numerical only in theory,
1165 * but apparently some broken servers may have alphanum aliases instead.
1166 * Do like openldap libraries and allow alphanum aliases for oids, but
1167 * do not allow Tagging options in that case.
1169 static bool tldap_is_attrdesc(const char *s, int len, bool no_tagopts)
1171 bool is_oid = false;
1175 /* first char has stricter rules */
1178 } else if (!tldap_is_alpha(*s)) {
1179 /* bad first char */
1183 for (i = 1; i < len; i++) {
1186 if (isdigit(s[i])) {
1199 if (tldap_is_adh(s[i])) {
1206 /* no tagging options */
1213 if ((i + 1) == len) {
1231 /* this function copies the value until the closing parenthesis is found. */
1232 static char *tldap_get_val(TALLOC_CTX *memctx,
1233 const char *value, const char **_s)
1235 const char *s = value;
1237 /* find terminator */
1240 if (s && (*(s - 1) == '\\')) {
1245 if (!s || !(*s == ')')) {
1246 /* malformed filter */
1252 return talloc_strndup(memctx, value, s - value);
1255 static int tldap_hex2char(const char *x)
1257 if (isxdigit(x[0]) && isxdigit(x[1])) {
1258 const char h1 = x[0], h2 = x[1];
1261 if (h1 >= 'a') c = h1 - (int)'a' + 10;
1262 else if (h1 >= 'A') c = h1 - (int)'A' + 10;
1263 else if (h1 >= '0') c = h1 - (int)'0';
1265 if (h2 >= 'a') c += h2 - (int)'a' + 10;
1266 else if (h2 >= 'A') c += h2 - (int)'A' + 10;
1267 else if (h2 >= '0') c += h2 - (int)'0';
1275 static bool tldap_find_first_star(const char *val, const char **star)
1279 for (s = val; *s; s++) {
1282 if (isxdigit(s[1]) && isxdigit(s[2])) {
1286 /* not hex based escape, check older syntax */
1295 /* invalid escape sequence */
1300 /* end of val, nothing found */
1310 /* string ended without closing parenthesis, filter is malformed */
1314 static bool tldap_unescape_inplace(char *value, size_t *val_len)
1319 for (i = 0,p = 0; i < *val_len; i++) {
1325 /* these must be escaped */
1329 if (!value[i + 1]) {
1335 /* LDAPv3 escaped */
1336 c = tldap_hex2char(&value[i]);
1337 if (c >= 0 && c < 256) {
1344 /* LDAPv2 escaped */
1350 value[p] = value[i];
1361 value[p] = value[i];
1370 static bool tldap_push_filter_basic(struct tldap_context *ld,
1371 struct asn1_data *data,
1373 static bool tldap_push_filter_substring(struct tldap_context *ld,
1374 struct asn1_data *data,
1377 static bool tldap_push_filter_int(struct tldap_context *ld,
1378 struct asn1_data *data,
1381 const char *s = *_s;
1385 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1386 "Incomplete or malformed filter\n");
1391 /* we are right after a parenthesis,
1392 * find out what op we have at hand */
1395 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: AND\n");
1396 if (!asn1_push_tag(data, TLDAP_FILTER_AND)) return false;
1401 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: OR\n");
1402 if (!asn1_push_tag(data, TLDAP_FILTER_OR)) return false;
1407 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: NOT\n");
1408 if (!asn1_push_tag(data, TLDAP_FILTER_NOT)) return false;
1410 ret = tldap_push_filter_int(ld, data, &s);
1414 if (!asn1_pop_tag(data)) return false;
1419 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1420 "Invalid parenthesis '%c'\n", *s);
1424 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1425 "Invalid filter termination\n");
1429 ret = tldap_push_filter_basic(ld, data, &s);
1436 /* only and/or filters get here.
1437 * go through the list of filters */
1440 /* RFC 4526: empty and/or */
1441 if (!asn1_pop_tag(data)) return false;
1446 ret = tldap_push_filter_int(ld, data, &s);
1452 /* end of list, return */
1453 if (!asn1_pop_tag(data)) return false;
1460 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1461 "Incomplete or malformed filter\n");
1466 if (asn1_has_error(data)) {
1475 static bool tldap_push_filter_basic(struct tldap_context *ld,
1476 struct asn1_data *data,
1479 TALLOC_CTX *tmpctx = talloc_tos();
1480 const char *s = *_s;
1488 size_t type_len = 0;
1491 bool write_octect = true;
1494 eq = strchr(s, '=');
1496 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1497 "Invalid filter, missing equal sign\n");
1506 if (!asn1_push_tag(data, TLDAP_FILTER_LE)) return false;
1510 if (!asn1_push_tag(data, TLDAP_FILTER_GE)) return false;
1514 if (!asn1_push_tag(data, TLDAP_FILTER_APX)) return false;
1518 if (!asn1_push_tag(data, TLDAP_FILTER_EXT)) return false;
1519 write_octect = false;
1525 if (*s == ':') { /* [:dn]:rule:= value */
1527 /* malformed filter */
1531 } else { /* type[:dn][:rule]:= value */
1533 dn = strchr(s, ':');
1534 type_len = dn - type;
1535 if (dn == e) { /* type:= value */
1542 rule = strchr(dn, ':');
1546 if ((rule == dn + 1) || rule + 1 == e) {
1547 /* malformed filter, contains "::" */
1551 if (strncasecmp_m(dn, "dn:", 3) != 0) {
1556 /* malformed filter. With two
1557 * optionals, the first must be "dn"
1570 if (!type && !dn && !rule) {
1571 /* malformed filter, there must be at least one */
1576 MatchingRuleAssertion ::= SEQUENCE {
1577 matchingRule [1] MatchingRuleID OPTIONAL,
1578 type [2] AttributeDescription OPTIONAL,
1579 matchValue [3] AssertionValue,
1580 dnAttributes [4] BOOLEAN DEFAULT FALSE
1584 /* check and add rule */
1586 ret = tldap_is_attrdesc(rule, e - rule, true);
1590 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(1))) return false;
1591 if (!asn1_write(data, rule, e - rule)) return false;
1592 if (!asn1_pop_tag(data)) return false;
1595 /* check and add type */
1597 ret = tldap_is_attrdesc(type, type_len, false);
1601 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(2))) return false;
1602 if (!asn1_write(data, type, type_len)) return false;
1603 if (!asn1_pop_tag(data)) return false;
1606 uval = tldap_get_val(tmpctx, val, _s);
1610 uval_len = *_s - val;
1611 ret = tldap_unescape_inplace(uval, &uval_len);
1616 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(3))) return false;
1617 if (!asn1_write(data, uval, uval_len)) return false;
1618 if (!asn1_pop_tag(data)) return false;
1620 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(4))) return false;
1621 if (!asn1_write_uint8(data, dn?1:0)) return false;
1622 if (!asn1_pop_tag(data)) return false;
1628 ret = tldap_is_attrdesc(s, e - s, false);
1633 if (strncmp(val, "*)", 2) == 0) {
1635 if (!asn1_push_tag(data, TLDAP_FILTER_PRES)) return false;
1636 if (!asn1_write(data, s, e - s)) return false;
1638 write_octect = false;
1642 ret = tldap_find_first_star(val, &star);
1648 if (!asn1_push_tag(data, TLDAP_FILTER_SUB)) return false;
1649 if (!asn1_write_OctetString(data, s, e - s)) return false;
1650 ret = tldap_push_filter_substring(ld, data, val, &s);
1655 write_octect = false;
1659 /* if nothing else, then it is just equality */
1660 if (!asn1_push_tag(data, TLDAP_FILTER_EQ)) return false;
1661 write_octect = true;
1666 uval = tldap_get_val(tmpctx, val, _s);
1670 uval_len = *_s - val;
1671 ret = tldap_unescape_inplace(uval, &uval_len);
1676 if (!asn1_write_OctetString(data, s, e - s)) return false;
1677 if (!asn1_write_OctetString(data, uval, uval_len)) return false;
1680 if (asn1_has_error(data)) {
1683 return asn1_pop_tag(data);
1686 static bool tldap_push_filter_substring(struct tldap_context *ld,
1687 struct asn1_data *data,
1691 TALLOC_CTX *tmpctx = talloc_tos();
1692 bool initial = true;
1699 SubstringFilter ::= SEQUENCE {
1700 type AttributeDescription,
1701 -- at least one must be present
1702 substrings SEQUENCE OF CHOICE {
1703 initial [0] LDAPString,
1705 final [2] LDAPString } }
1707 if (!asn1_push_tag(data, ASN1_SEQUENCE(0))) return false;
1710 ret = tldap_find_first_star(val, &star);
1714 chunk_len = star - val;
1718 if (!initial && chunk_len == 0) {
1719 /* found '**', which is illegal */
1735 if (initial && chunk_len == 0) {
1741 chunk = talloc_strndup(tmpctx, val, chunk_len);
1745 ret = tldap_unescape_inplace(chunk, &chunk_len);
1752 if (!asn1_push_tag(data, TLDAP_SUB_INI)) return false;
1755 if (!asn1_push_tag(data, TLDAP_SUB_ANY)) return false;
1759 if (!asn1_push_tag(data, TLDAP_SUB_FIN)) return false;
1765 if (!asn1_write(data, chunk, chunk_len)) return false;
1766 if (!asn1_pop_tag(data)) return false;
1770 } while (*star == '*');
1774 /* end of sequence */
1775 return asn1_pop_tag(data);
1778 /* NOTE: although openldap libraries allow for spaces in some places, mosly
1779 * around parenthesis, we do not allow any spaces (except in values of
1780 * course) as I couldn't fine any place in RFC 4512 or RFC 4515 where
1781 * leading or trailing spaces where allowed.
1783 static bool tldap_push_filter(struct tldap_context *ld,
1784 struct asn1_data *data,
1787 const char *s = filter;
1790 ret = tldap_push_filter_int(ld, data, &s);
1792 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1793 "Incomplete or malformed filter\n");
1799 /*****************************************************************************/
1801 static void tldap_search_done(struct tevent_req *subreq);
1803 struct tevent_req *tldap_search_send(TALLOC_CTX *mem_ctx,
1804 struct tevent_context *ev,
1805 struct tldap_context *ld,
1806 const char *base, int scope,
1811 struct tldap_control *sctrls,
1813 struct tldap_control *cctrls,
1819 struct tevent_req *req, *subreq;
1820 struct tldap_req_state *state;
1823 req = tldap_req_create(mem_ctx, ld, &state);
1828 if (!asn1_push_tag(state->out, TLDAP_REQ_SEARCH)) goto encoding_error;
1829 if (!asn1_write_OctetString(state->out, base, strlen(base))) goto encoding_error;
1830 if (!asn1_write_enumerated(state->out, scope)) goto encoding_error;
1831 if (!asn1_write_enumerated(state->out, deref)) goto encoding_error;
1832 if (!asn1_write_Integer(state->out, sizelimit)) goto encoding_error;
1833 if (!asn1_write_Integer(state->out, timelimit)) goto encoding_error;
1834 if (!asn1_write_BOOLEAN(state->out, attrsonly)) goto encoding_error;
1836 if (!tldap_push_filter(ld, state->out, filter)) {
1837 goto encoding_error;
1840 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto encoding_error;
1841 for (i=0; i<num_attrs; i++) {
1842 if (!asn1_write_OctetString(state->out, attrs[i], strlen(attrs[i]))) goto encoding_error;
1844 if (!asn1_pop_tag(state->out)) goto encoding_error;
1845 if (!asn1_pop_tag(state->out)) goto encoding_error;
1847 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
1848 sctrls, num_sctrls);
1849 if (tevent_req_nomem(subreq, req)) {
1850 return tevent_req_post(req, ev);
1852 tevent_req_set_callback(subreq, tldap_search_done, req);
1856 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
1857 return tevent_req_post(req, ev);
1860 static void tldap_search_done(struct tevent_req *subreq)
1862 struct tevent_req *req = tevent_req_callback_data(
1863 subreq, struct tevent_req);
1864 struct tldap_req_state *state = tevent_req_data(
1865 req, struct tldap_req_state);
1868 rc = tldap_msg_recv(subreq, state, &state->result);
1869 if (tevent_req_ldap_error(req, rc)) {
1872 switch (state->result->type) {
1873 case TLDAP_RES_SEARCH_ENTRY:
1874 case TLDAP_RES_SEARCH_REFERENCE:
1875 if (!tldap_msg_set_pending(subreq)) {
1876 tevent_req_oom(req);
1879 tevent_req_notify_callback(req);
1881 case TLDAP_RES_SEARCH_RESULT:
1882 TALLOC_FREE(subreq);
1883 if (!asn1_start_tag(state->result->data,
1884 state->result->type) ||
1885 !tldap_decode_response(state) ||
1886 !asn1_end_tag(state->result->data) ||
1887 !tldap_decode_controls(state)) {
1888 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
1891 tevent_req_done(req);
1894 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
1899 TLDAPRC tldap_search_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1900 struct tldap_message **pmsg)
1902 struct tldap_req_state *state = tevent_req_data(
1903 req, struct tldap_req_state);
1906 if (!tevent_req_is_in_progress(req)
1907 && tevent_req_is_ldap_error(req, &rc)) {
1911 if (tevent_req_is_in_progress(req)) {
1912 switch (state->result->type) {
1913 case TLDAP_RES_SEARCH_ENTRY:
1914 case TLDAP_RES_SEARCH_REFERENCE:
1917 return TLDAP_OPERATIONS_ERROR;
1921 *pmsg = talloc_move(mem_ctx, &state->result);
1922 return TLDAP_SUCCESS;
1925 struct tldap_search_all_state {
1926 struct tldap_message **msgs;
1927 struct tldap_message *result;
1930 static void tldap_search_all_done(struct tevent_req *subreq);
1932 struct tevent_req *tldap_search_all_send(
1933 TALLOC_CTX *mem_ctx, struct tevent_context *ev,
1934 struct tldap_context *ld, const char *base, int scope,
1935 const char *filter, const char **attrs, int num_attrs, int attrsonly,
1936 struct tldap_control *sctrls, int num_sctrls,
1937 struct tldap_control *cctrls, int num_cctrls,
1938 int timelimit, int sizelimit, int deref)
1940 struct tevent_req *req, *subreq;
1941 struct tldap_search_all_state *state;
1943 req = tevent_req_create(mem_ctx, &state,
1944 struct tldap_search_all_state);
1949 subreq = tldap_search_send(state, ev, ld, base, scope, filter,
1950 attrs, num_attrs, attrsonly,
1951 sctrls, num_sctrls, cctrls, num_cctrls,
1952 timelimit, sizelimit, deref);
1953 if (tevent_req_nomem(subreq, req)) {
1954 return tevent_req_post(req, ev);
1956 tevent_req_set_callback(subreq, tldap_search_all_done, req);
1960 static void tldap_search_all_done(struct tevent_req *subreq)
1962 struct tevent_req *req = tevent_req_callback_data(
1963 subreq, struct tevent_req);
1964 struct tldap_search_all_state *state = tevent_req_data(
1965 req, struct tldap_search_all_state);
1966 struct tldap_message *msg, **tmp;
1971 rc = tldap_search_recv(subreq, state, &msg);
1972 /* No TALLOC_FREE(subreq), this is multi-step */
1973 if (tevent_req_ldap_error(req, rc)) {
1977 msgtype = tldap_msg_type(msg);
1978 if (msgtype == TLDAP_RES_SEARCH_RESULT) {
1979 state->result = msg;
1980 tevent_req_done(req);
1984 num_msgs = talloc_array_length(state->msgs);
1986 tmp = talloc_realloc(state, state->msgs, struct tldap_message *,
1988 if (tevent_req_nomem(tmp, req)) {
1992 state->msgs[num_msgs] = talloc_move(state->msgs, &msg);
1995 TLDAPRC tldap_search_all_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1996 struct tldap_message ***msgs,
1997 struct tldap_message **result)
1999 struct tldap_search_all_state *state = tevent_req_data(
2000 req, struct tldap_search_all_state);
2003 if (tevent_req_is_ldap_error(req, &rc)) {
2008 *msgs = talloc_move(mem_ctx, &state->msgs);
2010 if (result != NULL) {
2011 *result = talloc_move(mem_ctx, &state->result);
2014 return TLDAP_SUCCESS;
2017 TLDAPRC tldap_search(struct tldap_context *ld,
2018 const char *base, int scope, const char *filter,
2019 const char **attrs, int num_attrs, int attrsonly,
2020 struct tldap_control *sctrls, int num_sctrls,
2021 struct tldap_control *cctrls, int num_cctrls,
2022 int timelimit, int sizelimit, int deref,
2023 TALLOC_CTX *mem_ctx, struct tldap_message ***pmsgs)
2026 struct tevent_context *ev;
2027 struct tevent_req *req;
2028 TLDAPRC rc = TLDAP_NO_MEMORY;
2029 struct tldap_message **msgs;
2030 struct tldap_message *result;
2032 if (tldap_pending_reqs(ld)) {
2036 frame = talloc_stackframe();
2038 ev = samba_tevent_context_init(frame);
2042 req = tldap_search_all_send(frame, ev, ld, base, scope, filter,
2043 attrs, num_attrs, attrsonly,
2044 sctrls, num_sctrls, cctrls, num_cctrls,
2045 timelimit, sizelimit, deref);
2049 if (!tevent_req_poll(req, ev)) {
2050 rc = TLDAP_OPERATIONS_ERROR;
2053 rc = tldap_search_all_recv(req, frame, &msgs, &result);
2055 if (!TLDAP_RC_IS_SUCCESS(rc)) {
2059 TALLOC_FREE(ld->last_msg);
2060 ld->last_msg = talloc_move(ld, &result);
2062 if (pmsgs != NULL) {
2063 *pmsgs = talloc_move(mem_ctx, &msgs);
2070 static bool tldap_parse_search_entry(struct tldap_message *msg)
2072 int num_attribs = 0;
2074 if (msg->type != TLDAP_RES_SEARCH_ENTRY) {
2077 if (!asn1_start_tag(msg->data, TLDAP_RES_SEARCH_ENTRY)) {
2083 if (!asn1_read_OctetString_talloc(msg, msg->data, &msg->dn)) return false;
2085 if (msg->dn == NULL) {
2090 * Attributes: We overallocate msg->attribs by one, so that while
2091 * looping over the attributes we can directly parse into the last
2092 * array element. Same for the values in the inner loop.
2095 msg->attribs = talloc_array(msg, struct tldap_attribute, 1);
2096 if (msg->attribs == NULL) {
2100 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2101 while (asn1_peek_tag(msg->data, ASN1_SEQUENCE(0))) {
2102 struct tldap_attribute *attrib;
2105 attrib = &msg->attribs[num_attribs];
2106 attrib->values = talloc_array(msg->attribs, DATA_BLOB, 1);
2107 if (attrib->values == NULL) {
2110 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2111 if (!asn1_read_OctetString_talloc(msg->attribs, msg->data,
2112 &attrib->name)) return false;
2113 if (!asn1_start_tag(msg->data, ASN1_SET)) return false;
2115 while (asn1_peek_tag(msg->data, ASN1_OCTET_STRING)) {
2116 if (!asn1_read_OctetString(msg->data, msg,
2117 &attrib->values[num_values])) return false;
2119 attrib->values = talloc_realloc(
2120 msg->attribs, attrib->values, DATA_BLOB,
2122 if (attrib->values == NULL) {
2127 attrib->values = talloc_realloc(msg->attribs, attrib->values,
2128 DATA_BLOB, num_values);
2129 attrib->num_values = num_values;
2131 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SET */
2132 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SEQUENCE(0) */
2133 msg->attribs = talloc_realloc(
2134 msg, msg->attribs, struct tldap_attribute,
2136 if (msg->attribs == NULL) {
2141 msg->attribs = talloc_realloc(
2142 msg, msg->attribs, struct tldap_attribute, num_attribs);
2143 return asn1_end_tag(msg->data);
2146 bool tldap_entry_dn(struct tldap_message *msg, char **dn)
2148 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2155 bool tldap_entry_attributes(struct tldap_message *msg,
2156 struct tldap_attribute **attributes,
2157 int *num_attributes)
2159 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2162 *attributes = msg->attribs;
2163 *num_attributes = talloc_array_length(msg->attribs);
2167 static bool tldap_decode_controls(struct tldap_req_state *state)
2169 struct tldap_message *msg = state->result;
2170 struct asn1_data *data = msg->data;
2171 struct tldap_control *sctrls = NULL;
2172 int num_controls = 0;
2175 msg->res_sctrls = NULL;
2177 if (!asn1_peek_tag(data, ASN1_CONTEXT(0))) {
2181 if (!asn1_start_tag(data, ASN1_CONTEXT(0))) goto out;
2183 while (asn1_peek_tag(data, ASN1_SEQUENCE(0))) {
2184 struct tldap_control *c;
2187 sctrls = talloc_realloc(msg, sctrls, struct tldap_control,
2189 if (sctrls == NULL) {
2192 c = &sctrls[num_controls];
2194 if (!asn1_start_tag(data, ASN1_SEQUENCE(0))) goto out;
2195 if (!asn1_read_OctetString_talloc(msg, data, &oid)) goto out;
2196 if (asn1_has_error(data) || (oid == NULL)) {
2200 if (asn1_peek_tag(data, ASN1_BOOLEAN)) {
2201 if (!asn1_read_BOOLEAN(data, &c->critical)) goto out;
2203 c->critical = false;
2205 c->value = data_blob_null;
2206 if (asn1_peek_tag(data, ASN1_OCTET_STRING) &&
2207 !asn1_read_OctetString(data, msg, &c->value)) {
2210 if (!asn1_end_tag(data)) goto out; /* ASN1_SEQUENCE(0) */
2215 if (!asn1_end_tag(data)) goto out; /* ASN1_CONTEXT(0) */
2222 msg->res_sctrls = sctrls;
2224 TALLOC_FREE(sctrls);
2229 static void tldap_simple_done(struct tevent_req *subreq, int type)
2231 struct tevent_req *req = tevent_req_callback_data(
2232 subreq, struct tevent_req);
2233 struct tldap_req_state *state = tevent_req_data(
2234 req, struct tldap_req_state);
2237 rc = tldap_msg_recv(subreq, state, &state->result);
2238 TALLOC_FREE(subreq);
2239 if (tevent_req_ldap_error(req, rc)) {
2242 if (state->result->type != type) {
2243 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
2246 if (!asn1_start_tag(state->result->data, state->result->type) ||
2247 !tldap_decode_response(state) ||
2248 !asn1_end_tag(state->result->data) ||
2249 !tldap_decode_controls(state)) {
2250 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
2253 if (!TLDAP_RC_IS_SUCCESS(state->result->lderr)) {
2254 tevent_req_ldap_error(req, state->result->lderr);
2257 tevent_req_done(req);
2260 static TLDAPRC tldap_simple_recv(struct tevent_req *req)
2263 if (tevent_req_is_ldap_error(req, &rc)) {
2266 return TLDAP_SUCCESS;
2269 static void tldap_add_done(struct tevent_req *subreq);
2271 struct tevent_req *tldap_add_send(TALLOC_CTX *mem_ctx,
2272 struct tevent_context *ev,
2273 struct tldap_context *ld,
2275 struct tldap_mod *attributes,
2277 struct tldap_control *sctrls,
2279 struct tldap_control *cctrls,
2282 struct tevent_req *req, *subreq;
2283 struct tldap_req_state *state;
2286 req = tldap_req_create(mem_ctx, ld, &state);
2291 if (!asn1_push_tag(state->out, TLDAP_REQ_ADD)) goto err;
2292 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2293 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2295 for (i=0; i<num_attributes; i++) {
2296 struct tldap_mod *attrib = &attributes[i];
2297 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2298 if (!asn1_write_OctetString(state->out, attrib->attribute,
2299 strlen(attrib->attribute))) goto err;
2300 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2301 for (j=0; j<attrib->num_values; j++) {
2302 if (!asn1_write_OctetString(state->out,
2303 attrib->values[j].data,
2304 attrib->values[j].length)) goto err;
2306 if (!asn1_pop_tag(state->out)) goto err;
2307 if (!asn1_pop_tag(state->out)) goto err;
2310 if (!asn1_pop_tag(state->out)) goto err;
2311 if (!asn1_pop_tag(state->out)) goto err;
2313 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2314 sctrls, num_sctrls);
2315 if (tevent_req_nomem(subreq, req)) {
2316 return tevent_req_post(req, ev);
2318 tevent_req_set_callback(subreq, tldap_add_done, req);
2323 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2324 return tevent_req_post(req, ev);
2327 static void tldap_add_done(struct tevent_req *subreq)
2329 tldap_simple_done(subreq, TLDAP_RES_ADD);
2332 TLDAPRC tldap_add_recv(struct tevent_req *req)
2334 return tldap_simple_recv(req);
2337 TLDAPRC tldap_add(struct tldap_context *ld, const char *dn,
2338 struct tldap_mod *attributes, int num_attributes,
2339 struct tldap_control *sctrls, int num_sctrls,
2340 struct tldap_control *cctrls, int num_cctrls)
2342 TALLOC_CTX *frame = talloc_stackframe();
2343 struct tevent_context *ev;
2344 struct tevent_req *req;
2345 TLDAPRC rc = TLDAP_NO_MEMORY;
2347 ev = samba_tevent_context_init(frame);
2351 req = tldap_add_send(frame, ev, ld, dn, attributes, num_attributes,
2352 sctrls, num_sctrls, cctrls, num_cctrls);
2356 if (!tevent_req_poll(req, ev)) {
2357 rc = TLDAP_OPERATIONS_ERROR;
2360 rc = tldap_add_recv(req);
2361 tldap_save_msg(ld, req);
2367 static void tldap_modify_done(struct tevent_req *subreq);
2369 struct tevent_req *tldap_modify_send(TALLOC_CTX *mem_ctx,
2370 struct tevent_context *ev,
2371 struct tldap_context *ld,
2373 struct tldap_mod *mods, int num_mods,
2374 struct tldap_control *sctrls,
2376 struct tldap_control *cctrls,
2379 struct tevent_req *req, *subreq;
2380 struct tldap_req_state *state;
2383 req = tldap_req_create(mem_ctx, ld, &state);
2388 if (!asn1_push_tag(state->out, TLDAP_REQ_MODIFY)) goto err;
2389 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2390 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2392 for (i=0; i<num_mods; i++) {
2393 struct tldap_mod *mod = &mods[i];
2394 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2395 if (!asn1_write_enumerated(state->out, mod->mod_op)) goto err;
2396 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2397 if (!asn1_write_OctetString(state->out, mod->attribute,
2398 strlen(mod->attribute))) goto err;
2399 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2400 for (j=0; j<mod->num_values; j++) {
2401 if (!asn1_write_OctetString(state->out,
2402 mod->values[j].data,
2403 mod->values[j].length)) goto err;
2405 if (!asn1_pop_tag(state->out)) goto err;
2406 if (!asn1_pop_tag(state->out)) goto err;
2407 if (!asn1_pop_tag(state->out)) goto err;
2410 if (!asn1_pop_tag(state->out)) goto err;
2411 if (!asn1_pop_tag(state->out)) goto err;
2413 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2414 sctrls, num_sctrls);
2415 if (tevent_req_nomem(subreq, req)) {
2416 return tevent_req_post(req, ev);
2418 tevent_req_set_callback(subreq, tldap_modify_done, req);
2423 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2424 return tevent_req_post(req, ev);
2427 static void tldap_modify_done(struct tevent_req *subreq)
2429 tldap_simple_done(subreq, TLDAP_RES_MODIFY);
2432 TLDAPRC tldap_modify_recv(struct tevent_req *req)
2434 return tldap_simple_recv(req);
2437 TLDAPRC tldap_modify(struct tldap_context *ld, const char *dn,
2438 struct tldap_mod *mods, int num_mods,
2439 struct tldap_control *sctrls, int num_sctrls,
2440 struct tldap_control *cctrls, int num_cctrls)
2442 TALLOC_CTX *frame = talloc_stackframe();
2443 struct tevent_context *ev;
2444 struct tevent_req *req;
2445 TLDAPRC rc = TLDAP_NO_MEMORY;
2447 ev = samba_tevent_context_init(frame);
2451 req = tldap_modify_send(frame, ev, ld, dn, mods, num_mods,
2452 sctrls, num_sctrls, cctrls, num_cctrls);
2456 if (!tevent_req_poll(req, ev)) {
2457 rc = TLDAP_OPERATIONS_ERROR;
2460 rc = tldap_modify_recv(req);
2461 tldap_save_msg(ld, req);
2467 static void tldap_delete_done(struct tevent_req *subreq);
2469 struct tevent_req *tldap_delete_send(TALLOC_CTX *mem_ctx,
2470 struct tevent_context *ev,
2471 struct tldap_context *ld,
2473 struct tldap_control *sctrls,
2475 struct tldap_control *cctrls,
2478 struct tevent_req *req, *subreq;
2479 struct tldap_req_state *state;
2481 req = tldap_req_create(mem_ctx, ld, &state);
2486 if (!asn1_push_tag(state->out, TLDAP_REQ_DELETE)) goto err;
2487 if (!asn1_write(state->out, dn, strlen(dn))) goto err;
2488 if (!asn1_pop_tag(state->out)) goto err;
2490 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2491 sctrls, num_sctrls);
2492 if (tevent_req_nomem(subreq, req)) {
2493 return tevent_req_post(req, ev);
2495 tevent_req_set_callback(subreq, tldap_delete_done, req);
2500 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2501 return tevent_req_post(req, ev);
2504 static void tldap_delete_done(struct tevent_req *subreq)
2506 tldap_simple_done(subreq, TLDAP_RES_DELETE);
2509 TLDAPRC tldap_delete_recv(struct tevent_req *req)
2511 return tldap_simple_recv(req);
2514 TLDAPRC tldap_delete(struct tldap_context *ld, const char *dn,
2515 struct tldap_control *sctrls, int num_sctrls,
2516 struct tldap_control *cctrls, int num_cctrls)
2518 TALLOC_CTX *frame = talloc_stackframe();
2519 struct tevent_context *ev;
2520 struct tevent_req *req;
2521 TLDAPRC rc = TLDAP_NO_MEMORY;
2523 ev = samba_tevent_context_init(frame);
2527 req = tldap_delete_send(frame, ev, ld, dn, sctrls, num_sctrls,
2528 cctrls, num_cctrls);
2532 if (!tevent_req_poll(req, ev)) {
2533 rc = TLDAP_OPERATIONS_ERROR;
2536 rc = tldap_delete_recv(req);
2537 tldap_save_msg(ld, req);
2543 int tldap_msg_id(const struct tldap_message *msg)
2548 int tldap_msg_type(const struct tldap_message *msg)
2553 const char *tldap_msg_matcheddn(struct tldap_message *msg)
2558 return msg->res_matcheddn;
2561 const char *tldap_msg_diagnosticmessage(struct tldap_message *msg)
2566 return msg->res_diagnosticmessage;
2569 const char *tldap_msg_referral(struct tldap_message *msg)
2574 return msg->res_referral;
2577 void tldap_msg_sctrls(struct tldap_message *msg, int *num_sctrls,
2578 struct tldap_control **sctrls)
2585 *sctrls = msg->res_sctrls;
2586 *num_sctrls = talloc_array_length(msg->res_sctrls);
2589 struct tldap_message *tldap_ctx_lastmsg(struct tldap_context *ld)
2591 return ld->last_msg;
2594 static const struct { TLDAPRC rc; const char *string; } tldaprc_errmap[] =
2598 { TLDAP_OPERATIONS_ERROR,
2599 "TLDAP_OPERATIONS_ERROR" },
2600 { TLDAP_PROTOCOL_ERROR,
2601 "TLDAP_PROTOCOL_ERROR" },
2602 { TLDAP_TIMELIMIT_EXCEEDED,
2603 "TLDAP_TIMELIMIT_EXCEEDED" },
2604 { TLDAP_SIZELIMIT_EXCEEDED,
2605 "TLDAP_SIZELIMIT_EXCEEDED" },
2606 { TLDAP_COMPARE_FALSE,
2607 "TLDAP_COMPARE_FALSE" },
2608 { TLDAP_COMPARE_TRUE,
2609 "TLDAP_COMPARE_TRUE" },
2610 { TLDAP_STRONG_AUTH_NOT_SUPPORTED,
2611 "TLDAP_STRONG_AUTH_NOT_SUPPORTED" },
2612 { TLDAP_STRONG_AUTH_REQUIRED,
2613 "TLDAP_STRONG_AUTH_REQUIRED" },
2616 { TLDAP_ADMINLIMIT_EXCEEDED,
2617 "TLDAP_ADMINLIMIT_EXCEEDED" },
2618 { TLDAP_UNAVAILABLE_CRITICAL_EXTENSION,
2619 "TLDAP_UNAVAILABLE_CRITICAL_EXTENSION" },
2620 { TLDAP_CONFIDENTIALITY_REQUIRED,
2621 "TLDAP_CONFIDENTIALITY_REQUIRED" },
2622 { TLDAP_SASL_BIND_IN_PROGRESS,
2623 "TLDAP_SASL_BIND_IN_PROGRESS" },
2624 { TLDAP_NO_SUCH_ATTRIBUTE,
2625 "TLDAP_NO_SUCH_ATTRIBUTE" },
2626 { TLDAP_UNDEFINED_TYPE,
2627 "TLDAP_UNDEFINED_TYPE" },
2628 { TLDAP_INAPPROPRIATE_MATCHING,
2629 "TLDAP_INAPPROPRIATE_MATCHING" },
2630 { TLDAP_CONSTRAINT_VIOLATION,
2631 "TLDAP_CONSTRAINT_VIOLATION" },
2632 { TLDAP_TYPE_OR_VALUE_EXISTS,
2633 "TLDAP_TYPE_OR_VALUE_EXISTS" },
2634 { TLDAP_INVALID_SYNTAX,
2635 "TLDAP_INVALID_SYNTAX" },
2636 { TLDAP_NO_SUCH_OBJECT,
2637 "TLDAP_NO_SUCH_OBJECT" },
2638 { TLDAP_ALIAS_PROBLEM,
2639 "TLDAP_ALIAS_PROBLEM" },
2640 { TLDAP_INVALID_DN_SYNTAX,
2641 "TLDAP_INVALID_DN_SYNTAX" },
2644 { TLDAP_ALIAS_DEREF_PROBLEM,
2645 "TLDAP_ALIAS_DEREF_PROBLEM" },
2646 { TLDAP_INAPPROPRIATE_AUTH,
2647 "TLDAP_INAPPROPRIATE_AUTH" },
2648 { TLDAP_INVALID_CREDENTIALS,
2649 "TLDAP_INVALID_CREDENTIALS" },
2650 { TLDAP_INSUFFICIENT_ACCESS,
2651 "TLDAP_INSUFFICIENT_ACCESS" },
2654 { TLDAP_UNAVAILABLE,
2655 "TLDAP_UNAVAILABLE" },
2656 { TLDAP_UNWILLING_TO_PERFORM,
2657 "TLDAP_UNWILLING_TO_PERFORM" },
2658 { TLDAP_LOOP_DETECT,
2659 "TLDAP_LOOP_DETECT" },
2660 { TLDAP_NAMING_VIOLATION,
2661 "TLDAP_NAMING_VIOLATION" },
2662 { TLDAP_OBJECT_CLASS_VIOLATION,
2663 "TLDAP_OBJECT_CLASS_VIOLATION" },
2664 { TLDAP_NOT_ALLOWED_ON_NONLEAF,
2665 "TLDAP_NOT_ALLOWED_ON_NONLEAF" },
2666 { TLDAP_NOT_ALLOWED_ON_RDN,
2667 "TLDAP_NOT_ALLOWED_ON_RDN" },
2668 { TLDAP_ALREADY_EXISTS,
2669 "TLDAP_ALREADY_EXISTS" },
2670 { TLDAP_NO_OBJECT_CLASS_MODS,
2671 "TLDAP_NO_OBJECT_CLASS_MODS" },
2672 { TLDAP_RESULTS_TOO_LARGE,
2673 "TLDAP_RESULTS_TOO_LARGE" },
2674 { TLDAP_AFFECTS_MULTIPLE_DSAS,
2675 "TLDAP_AFFECTS_MULTIPLE_DSAS" },
2678 { TLDAP_SERVER_DOWN,
2679 "TLDAP_SERVER_DOWN" },
2680 { TLDAP_LOCAL_ERROR,
2681 "TLDAP_LOCAL_ERROR" },
2682 { TLDAP_ENCODING_ERROR,
2683 "TLDAP_ENCODING_ERROR" },
2684 { TLDAP_DECODING_ERROR,
2685 "TLDAP_DECODING_ERROR" },
2688 { TLDAP_AUTH_UNKNOWN,
2689 "TLDAP_AUTH_UNKNOWN" },
2690 { TLDAP_FILTER_ERROR,
2691 "TLDAP_FILTER_ERROR" },
2692 { TLDAP_USER_CANCELLED,
2693 "TLDAP_USER_CANCELLED" },
2694 { TLDAP_PARAM_ERROR,
2695 "TLDAP_PARAM_ERROR" },
2697 "TLDAP_NO_MEMORY" },
2698 { TLDAP_CONNECT_ERROR,
2699 "TLDAP_CONNECT_ERROR" },
2700 { TLDAP_NOT_SUPPORTED,
2701 "TLDAP_NOT_SUPPORTED" },
2702 { TLDAP_CONTROL_NOT_FOUND,
2703 "TLDAP_CONTROL_NOT_FOUND" },
2704 { TLDAP_NO_RESULTS_RETURNED,
2705 "TLDAP_NO_RESULTS_RETURNED" },
2706 { TLDAP_MORE_RESULTS_TO_RETURN,
2707 "TLDAP_MORE_RESULTS_TO_RETURN" },
2708 { TLDAP_CLIENT_LOOP,
2709 "TLDAP_CLIENT_LOOP" },
2710 { TLDAP_REFERRAL_LIMIT_EXCEEDED,
2711 "TLDAP_REFERRAL_LIMIT_EXCEEDED" },
2714 const char *tldap_rc2string(TLDAPRC rc)
2718 for (i=0; i<ARRAY_SIZE(tldaprc_errmap); i++) {
2719 if (TLDAP_RC_EQUAL(rc, tldaprc_errmap[i].rc)) {
2720 return tldaprc_errmap[i].string;
2724 return "Unknown LDAP Error";