?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
kcc: Add corresponding methods for repsTo
[Samba.git] / source3 / lib / tldap.c
blob5fcb43c29f4e75dbb62d812f374c558b75eef6ff
1 /*
2 Unix SMB/CIFS implementation.
3 Infrastructure for async ldap client requests
4 Copyright (C) Volker Lendecke 2009
5
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.
10
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.
15
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/>.
18 */
19
20 #include "replace.h"
21 #include "tldap.h"
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"
30
31 static TLDAPRC tldap_simple_recv(struct tevent_req *req);
32
33 #define TEVENT_TLDAP_RC_MAGIC (0x87bcd26e)
34
35 bool tevent_req_ldap_error(struct tevent_req *req, TLDAPRC rc)
36 {
37 uint64_t err;
38
39 if (TLDAP_RC_IS_SUCCESS(rc)) {
40 return false;
41 }
42
43 err = TEVENT_TLDAP_RC_MAGIC;
44 err <<= 32;
45 err |= TLDAP_RC_V(rc);
46
47 return tevent_req_error(req, err);
48 }
49
50 bool tevent_req_is_ldap_error(struct tevent_req *req, TLDAPRC *perr)
51 {
52 enum tevent_req_state state;
53 uint64_t err;
54
55 if (!tevent_req_is_error(req, &state, &err)) {
56 return false;
57 }
58 switch (state) {
59 case TEVENT_REQ_TIMED_OUT:
60 *perr = TLDAP_TIMEOUT;
61 break;
62 case TEVENT_REQ_NO_MEMORY:
63 *perr = TLDAP_NO_MEMORY;
64 break;
65 case TEVENT_REQ_USER_ERROR:
66 if ((err >> 32) != TEVENT_TLDAP_RC_MAGIC) {
67 abort();
68 }
69 *perr = TLDAP_RC(err & 0xffffffff);
70 break;
71 default:
72 *perr = TLDAP_OPERATIONS_ERROR;
73 break;
74 }
75 return true;
76 }
77
78 struct tldap_ctx_attribute {
79 char *name;
80 void *ptr;
81 };
82
83 struct tldap_context {
84 int ld_version;
85 struct tstream_context *conn;
86 bool server_down;
87 int msgid;
88 struct tevent_queue *outgoing;
89 struct tevent_req **pending;
90
91 /* For the sync wrappers we need something like get_last_error... */
92 struct tldap_message *last_msg;
93
94 /* debug */
95 void (*log_fn)(void *context, enum tldap_debug_level level,
96 const char *fmt, va_list ap);
97 void *log_private;
98
99 struct tldap_ctx_attribute *ctx_attrs;
100 };
101
102 struct tldap_message {
103 struct asn1_data *data;
104 uint8_t *inbuf;
105 int type;
106 int id;
107
108 /* RESULT_ENTRY */
109 char *dn;
110 struct tldap_attribute *attribs;
111
112 /* Error data sent by the server */
113 TLDAPRC lderr;
114 char *res_matcheddn;
115 char *res_diagnosticmessage;
116 char *res_referral;
117 DATA_BLOB res_serverSaslCreds;
118 struct tldap_control *res_sctrls;
119
120 /* Controls sent by the server */
121 struct tldap_control *ctrls;
122 };
123
124 void tldap_set_debug(struct tldap_context *ld,
125 void (*log_fn)(void *log_private,
126 enum tldap_debug_level level,
127 const char *fmt,
128 va_list ap) PRINTF_ATTRIBUTE(3,0),
129 void *log_private)
130 {
131 ld->log_fn = log_fn;
132 ld->log_private = log_private;
133 }
134
135 static void tldap_debug(struct tldap_context *ld,
136 enum tldap_debug_level level,
137 const char *fmt, ...)
138 {
139 va_list ap;
140 if (!ld) {
141 return;
142 }
143 if (ld->log_fn == NULL) {
144 return;
145 }
146 va_start(ap, fmt);
147 ld->log_fn(ld->log_private, level, fmt, ap);
148 va_end(ap);
149 }
150
151 static int tldap_next_msgid(struct tldap_context *ld)
152 {
153 int result;
154
155 result = ld->msgid++;
156 if (ld->msgid == 2147483647) {
157 ld->msgid = 1;
158 }
159 return result;
160 }
161
162 struct tldap_context *tldap_context_create(TALLOC_CTX *mem_ctx, int fd)
163 {
164 struct tldap_context *ctx;
165 int ret;
166
167 ctx = talloc_zero(mem_ctx, struct tldap_context);
168 if (ctx == NULL) {
169 return NULL;
170 }
171 ret = tstream_bsd_existing_socket(ctx, fd, &ctx->conn);
172 if (ret == -1) {
173 TALLOC_FREE(ctx);
174 return NULL;
175 }
176 ctx->msgid = 1;
177 ctx->ld_version = 3;
178 ctx->outgoing = tevent_queue_create(ctx, "tldap_outgoing");
179 if (ctx->outgoing == NULL) {
180 TALLOC_FREE(ctx);
181 return NULL;
182 }
183 return ctx;
184 }
185
186 bool tldap_connection_ok(struct tldap_context *ld)
187 {
188 if (ld == NULL) {
189 return false;
190 }
191 return !ld->server_down;
192 }
193
194 static size_t tldap_pending_reqs(struct tldap_context *ld)
195 {
196 return talloc_array_length(ld->pending);
197 }
198
199 struct tstream_context *tldap_get_tstream(struct tldap_context *ld)
200 {
201 return ld->conn;
202 }
203
204 void tldap_set_tstream(struct tldap_context *ld,
205 struct tstream_context *stream)
206 {
207 ld->conn = stream;
208 }
209
210 static struct tldap_ctx_attribute *tldap_context_findattr(
211 struct tldap_context *ld, const char *name)
212 {
213 size_t i, num_attrs;
214
215 num_attrs = talloc_array_length(ld->ctx_attrs);
216
217 for (i=0; i<num_attrs; i++) {
218 if (strcmp(ld->ctx_attrs[i].name, name) == 0) {
219 return &ld->ctx_attrs[i];
220 }
221 }
222 return NULL;
223 }
224
225 bool tldap_context_setattr(struct tldap_context *ld,
226 const char *name, const void *_pptr)
227 {
228 struct tldap_ctx_attribute *tmp, *attr;
229 char *tmpname;
230 int num_attrs;
231 void **pptr = (void **)discard_const_p(void,_pptr);
232
233 attr = tldap_context_findattr(ld, name);
234 if (attr != NULL) {
235 /*
236 * We don't actually delete attrs, we don't expect tons of
237 * attributes being shuffled around.
238 */
239 TALLOC_FREE(attr->ptr);
240 if (*pptr != NULL) {
241 attr->ptr = talloc_move(ld->ctx_attrs, pptr);
242 *pptr = NULL;
243 }
244 return true;
245 }
246
247 tmpname = talloc_strdup(ld, name);
248 if (tmpname == NULL) {
249 return false;
250 }
251
252 num_attrs = talloc_array_length(ld->ctx_attrs);
253
254 tmp = talloc_realloc(ld, ld->ctx_attrs, struct tldap_ctx_attribute,
255 num_attrs+1);
256 if (tmp == NULL) {
257 TALLOC_FREE(tmpname);
258 return false;
259 }
260 tmp[num_attrs].name = talloc_move(tmp, &tmpname);
261 if (*pptr != NULL) {
262 tmp[num_attrs].ptr = talloc_move(tmp, pptr);
263 } else {
264 tmp[num_attrs].ptr = NULL;
265 }
266 *pptr = NULL;
267 ld->ctx_attrs = tmp;
268 return true;
269 }
270
271 void *tldap_context_getattr(struct tldap_context *ld, const char *name)
272 {
273 struct tldap_ctx_attribute *attr = tldap_context_findattr(ld, name);
274
275 if (attr == NULL) {
276 return NULL;
277 }
278 return attr->ptr;
279 }
280
281 struct read_ldap_state {
282 uint8_t *buf;
283 bool done;
284 };
285
286 static ssize_t read_ldap_more(uint8_t *buf, size_t buflen, void *private_data);
287 static void read_ldap_done(struct tevent_req *subreq);
288
289 static struct tevent_req *read_ldap_send(TALLOC_CTX *mem_ctx,
290 struct tevent_context *ev,
291 struct tstream_context *conn)
292 {
293 struct tevent_req *req, *subreq;
294 struct read_ldap_state *state;
295
296 req = tevent_req_create(mem_ctx, &state, struct read_ldap_state);
297 if (req == NULL) {
298 return NULL;
299 }
300 state->done = false;
301
302 subreq = tstream_read_packet_send(state, ev, conn, 2, read_ldap_more,
303 state);
304 if (tevent_req_nomem(subreq, req)) {
305 return tevent_req_post(req, ev);
306 }
307 tevent_req_set_callback(subreq, read_ldap_done, req);
308 return req;
309 }
310
311 static ssize_t read_ldap_more(uint8_t *buf, size_t buflen, void *private_data)
312 {
313 struct read_ldap_state *state = talloc_get_type_abort(
314 private_data, struct read_ldap_state);
315 size_t len;
316 int i, lensize;
317
318 if (state->done) {
319 /* We've been here, we're done */
320 return 0;
321 }
322
323 /*
324 * From ldap.h: LDAP_TAG_MESSAGE is 0x30
325 */
326 if (buf[0] != 0x30) {
327 return -1;
328 }
329
330 len = buf[1];
331 if ((len & 0x80) == 0) {
332 state->done = true;
333 return len;
334 }
335
336 lensize = (len & 0x7f);
337 len = 0;
338
339 if (buflen == 2) {
340 /* Please get us the full length */
341 return lensize;
342 }
343 if (buflen > 2 + lensize) {
344 state->done = true;
345 return 0;
346 }
347 if (buflen != 2 + lensize) {
348 return -1;
349 }
350
351 for (i=0; i<lensize; i++) {
352 len = (len << 8) | buf[2+i];
353 }
354 return len;
355 }
356
357 static void read_ldap_done(struct tevent_req *subreq)
358 {
359 struct tevent_req *req = tevent_req_callback_data(
360 subreq, struct tevent_req);
361 struct read_ldap_state *state = tevent_req_data(
362 req, struct read_ldap_state);
363 ssize_t nread;
364 int err;
365
366 nread = tstream_read_packet_recv(subreq, state, &state->buf, &err);
367 TALLOC_FREE(subreq);
368 if (nread == -1) {
369 tevent_req_error(req, err);
370 return;
371 }
372 tevent_req_done(req);
373 }
374
375 static ssize_t read_ldap_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
376 uint8_t **pbuf, int *perrno)
377 {
378 struct read_ldap_state *state = tevent_req_data(
379 req, struct read_ldap_state);
380
381 if (tevent_req_is_unix_error(req, perrno)) {
382 return -1;
383 }
384 *pbuf = talloc_move(mem_ctx, &state->buf);
385 return talloc_get_size(*pbuf);
386 }
387
388 struct tldap_msg_state {
389 struct tldap_context *ld;
390 struct tevent_context *ev;
391 int id;
392 struct iovec iov;
393
394 struct asn1_data *data;
395 uint8_t *inbuf;
396 };
397
398 static bool tldap_push_controls(struct asn1_data *data,
399 struct tldap_control *sctrls,
400 int num_sctrls)
401 {
402 int i;
403
404 if ((sctrls == NULL) || (num_sctrls == 0)) {
405 return true;
406 }
407
408 if (!asn1_push_tag(data, ASN1_CONTEXT(0))) return false;
409
410 for (i=0; i<num_sctrls; i++) {
411 struct tldap_control *c = &sctrls[i];
412 if (!asn1_push_tag(data, ASN1_SEQUENCE(0))) return false;
413 if (!asn1_write_OctetString(data, c->oid, strlen(c->oid))) return false;
414 if (c->critical) {
415 if (!asn1_write_BOOLEAN(data, true)) return false;
416 }
417 if (c->value.data != NULL) {
418 if (!asn1_write_OctetString(data, c->value.data,
419 c->value.length)) return false;
420 }
421 if (!asn1_pop_tag(data)) return false; /* ASN1_SEQUENCE(0) */
422 }
423
424 return asn1_pop_tag(data); /* ASN1_CONTEXT(0) */
425 }
426
427 static void tldap_msg_sent(struct tevent_req *subreq);
428 static void tldap_msg_received(struct tevent_req *subreq);
429
430 static struct tevent_req *tldap_msg_send(TALLOC_CTX *mem_ctx,
431 struct tevent_context *ev,
432 struct tldap_context *ld,
433 int id, struct asn1_data *data,
434 struct tldap_control *sctrls,
435 int num_sctrls)
436 {
437 struct tevent_req *req, *subreq;
438 struct tldap_msg_state *state;
439 DATA_BLOB blob;
440
441 tldap_debug(ld, TLDAP_DEBUG_TRACE, "tldap_msg_send: sending msg %d\n",
442 id);
443
444 req = tevent_req_create(mem_ctx, &state, struct tldap_msg_state);
445 if (req == NULL) {
446 return NULL;
447 }
448 state->ld = ld;
449 state->ev = ev;
450 state->id = id;
451
452 if (state->ld->server_down) {
453 tevent_req_ldap_error(req, TLDAP_SERVER_DOWN);
454 return tevent_req_post(req, ev);
455 }
456
457 if (!tldap_push_controls(data, sctrls, num_sctrls)) {
458 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
459 return tevent_req_post(req, ev);
460 }
461
462
463 if (!asn1_pop_tag(data)) {
464 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
465 return tevent_req_post(req, ev);
466 }
467
468 if (!asn1_blob(data, &blob)) {
469 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
470 return tevent_req_post(req, ev);
471 }
472
473 state->iov.iov_base = (void *)blob.data;
474 state->iov.iov_len = blob.length;
475
476 subreq = tstream_writev_queue_send(state, ev, ld->conn, ld->outgoing,
477 &state->iov, 1);
478 if (tevent_req_nomem(subreq, req)) {
479 return tevent_req_post(req, ev);
480 }
481 tevent_req_set_callback(subreq, tldap_msg_sent, req);
482 return req;
483 }
484
485 static void tldap_msg_unset_pending(struct tevent_req *req)
486 {
487 struct tldap_msg_state *state = tevent_req_data(
488 req, struct tldap_msg_state);
489 struct tldap_context *ld = state->ld;
490 int num_pending = tldap_pending_reqs(ld);
491 int i;
492
493 tevent_req_set_cleanup_fn(req, NULL);
494
495 if (num_pending == 1) {
496 TALLOC_FREE(ld->pending);
497 return;
498 }
499
500 for (i=0; i<num_pending; i++) {
501 if (req == ld->pending[i]) {
502 break;
503 }
504 }
505 if (i == num_pending) {
506 /*
507 * Something's seriously broken. Just returning here is the
508 * right thing nevertheless, the point of this routine is to
509 * remove ourselves from cli->pending.
510 */
511 return;
512 }
513
514 /*
515 * Remove ourselves from the cli->pending array
516 */
517 if (num_pending > 1) {
518 ld->pending[i] = ld->pending[num_pending-1];
519 }
520
521 /*
522 * No NULL check here, we're shrinking by sizeof(void *), and
523 * talloc_realloc just adjusts the size for this.
524 */
525 ld->pending = talloc_realloc(NULL, ld->pending, struct tevent_req *,
526 num_pending - 1);
527 }
528
529 static void tldap_msg_cleanup(struct tevent_req *req,
530 enum tevent_req_state req_state)
531 {
532 switch (req_state) {
533 case TEVENT_REQ_USER_ERROR:
534 case TEVENT_REQ_RECEIVED:
535 tldap_msg_unset_pending(req);
536 return;
537 default:
538 return;
539 }
540 }
541
542 static bool tldap_msg_set_pending(struct tevent_req *req)
543 {
544 struct tldap_msg_state *state = tevent_req_data(
545 req, struct tldap_msg_state);
546 struct tldap_context *ld;
547 struct tevent_req **pending;
548 int num_pending;
549 struct tevent_req *subreq;
550
551 ld = state->ld;
552 num_pending = tldap_pending_reqs(ld);
553
554 pending = talloc_realloc(ld, ld->pending, struct tevent_req *,
555 num_pending+1);
556 if (pending == NULL) {
557 return false;
558 }
559 pending[num_pending] = req;
560 ld->pending = pending;
561 tevent_req_set_cleanup_fn(req, tldap_msg_cleanup);
562
563 if (num_pending > 0) {
564 return true;
565 }
566
567 /*
568 * We're the first one, add the read_ldap request that waits for the
569 * answer from the server
570 */
571 subreq = read_ldap_send(ld->pending, state->ev, ld->conn);
572 if (subreq == NULL) {
573 tldap_msg_unset_pending(req);
574 return false;
575 }
576 tevent_req_set_callback(subreq, tldap_msg_received, ld);
577 return true;
578 }
579
580 static void tldap_msg_sent(struct tevent_req *subreq)
581 {
582 struct tevent_req *req = tevent_req_callback_data(
583 subreq, struct tevent_req);
584 struct tldap_msg_state *state = tevent_req_data(
585 req, struct tldap_msg_state);
586 ssize_t nwritten;
587 int err;
588
589 nwritten = tstream_writev_queue_recv(subreq, &err);
590 TALLOC_FREE(subreq);
591 if (nwritten == -1) {
592 state->ld->server_down = true;
593 tevent_req_ldap_error(req, TLDAP_SERVER_DOWN);
594 return;
595 }
596
597 if (!tldap_msg_set_pending(req)) {
598 tevent_req_oom(req);
599 return;
600 }
601 }
602
603 static int tldap_msg_msgid(struct tevent_req *req)
604 {
605 struct tldap_msg_state *state = tevent_req_data(
606 req, struct tldap_msg_state);
607
608 return state->id;
609 }
610
611 static void tldap_msg_received(struct tevent_req *subreq)
612 {
613 struct tldap_context *ld = tevent_req_callback_data(
614 subreq, struct tldap_context);
615 struct tevent_req *req;
616 struct tldap_msg_state *state;
617 struct asn1_data *data;
618 uint8_t *inbuf;
619 ssize_t received;
620 size_t num_pending;
621 int i, err;
622 TLDAPRC status;
623 int id;
624 uint8_t type;
625 bool ok;
626
627 received = read_ldap_recv(subreq, talloc_tos(), &inbuf, &err);
628 TALLOC_FREE(subreq);
629 if (received == -1) {
630 ld->server_down = true;
631 status = TLDAP_SERVER_DOWN;
632 goto fail;
633 }
634
635 data = asn1_init(talloc_tos());
636 if (data == NULL) {
637 status = TLDAP_NO_MEMORY;
638 goto fail;
639 }
640 asn1_load_nocopy(data, inbuf, received);
641
642 ok = true;
643 ok &= asn1_start_tag(data, ASN1_SEQUENCE(0));
644 ok &= asn1_read_Integer(data, &id);
645 ok &= asn1_peek_uint8(data, &type);
646
647 if (!ok) {
648 status = TLDAP_PROTOCOL_ERROR;
649 goto fail;
650 }
651
652 tldap_debug(ld, TLDAP_DEBUG_TRACE, "tldap_msg_received: got msg %d "
653 "type %d\n", id, (int)type);
654
655 num_pending = talloc_array_length(ld->pending);
656
657 for (i=0; i<num_pending; i++) {
658 if (id == tldap_msg_msgid(ld->pending[i])) {
659 break;
660 }
661 }
662 if (i == num_pending) {
663 /* Dump unexpected reply */
664 tldap_debug(ld, TLDAP_DEBUG_WARNING, "tldap_msg_received: "
665 "No request pending for msg %d\n", id);
666 TALLOC_FREE(data);
667 TALLOC_FREE(inbuf);
668 goto done;
669 }
670
671 req = ld->pending[i];
672 state = tevent_req_data(req, struct tldap_msg_state);
673
674 state->inbuf = talloc_move(state, &inbuf);
675 state->data = talloc_move(state, &data);
676
677 tldap_msg_unset_pending(req);
678 num_pending = talloc_array_length(ld->pending);
679
680 tevent_req_done(req);
681
682 done:
683 if (num_pending == 0) {
684 return;
685 }
686 if (talloc_array_length(ld->pending) > num_pending) {
687 /*
688 * The callback functions called from tevent_req_done() above
689 * have put something on the pending queue. We don't have to
690 * trigger the read_ldap_send(), tldap_msg_set_pending() has
691 * done it for us already.
692 */
693 return;
694 }
695
696 state = tevent_req_data(ld->pending[0], struct tldap_msg_state);
697 subreq = read_ldap_send(ld->pending, state->ev, ld->conn);
698 if (subreq == NULL) {
699 status = TLDAP_NO_MEMORY;
700 goto fail;
701 }
702 tevent_req_set_callback(subreq, tldap_msg_received, ld);
703 return;
704
705 fail:
706 while (talloc_array_length(ld->pending) > 0) {
707 req = ld->pending[0];
708 state = tevent_req_data(req, struct tldap_msg_state);
709 tevent_req_defer_callback(req, state->ev);
710 tevent_req_ldap_error(req, status);
711 }
712 }
713
714 static TLDAPRC tldap_msg_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
715 struct tldap_message **pmsg)
716 {
717 struct tldap_msg_state *state = tevent_req_data(
718 req, struct tldap_msg_state);
719 struct tldap_message *msg;
720 TLDAPRC err;
721 uint8_t msgtype;
722
723 if (tevent_req_is_ldap_error(req, &err)) {
724 return err;
725 }
726
727 if (!asn1_peek_uint8(state->data, &msgtype)) {
728 return TLDAP_PROTOCOL_ERROR;
729 }
730
731 if (pmsg == NULL) {
732 return TLDAP_SUCCESS;
733 }
734
735 msg = talloc_zero(mem_ctx, struct tldap_message);
736 if (msg == NULL) {
737 return TLDAP_NO_MEMORY;
738 }
739 msg->id = state->id;
740
741 msg->inbuf = talloc_move(msg, &state->inbuf);
742 msg->data = talloc_move(msg, &state->data);
743 msg->type = msgtype;
744
745 *pmsg = msg;
746 return TLDAP_SUCCESS;
747 }
748
749 struct tldap_req_state {
750 int id;
751 struct asn1_data *out;
752 struct tldap_message *result;
753 };
754
755 static struct tevent_req *tldap_req_create(TALLOC_CTX *mem_ctx,
756 struct tldap_context *ld,
757 struct tldap_req_state **pstate)
758 {
759 struct tevent_req *req;
760 struct tldap_req_state *state;
761
762 req = tevent_req_create(mem_ctx, &state, struct tldap_req_state);
763 if (req == NULL) {
764 return NULL;
765 }
766 state->out = asn1_init(state);
767 if (state->out == NULL) {
768 goto err;
769 }
770 state->id = tldap_next_msgid(ld);
771
772 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
773 if (!asn1_write_Integer(state->out, state->id)) goto err;
774
775 *pstate = state;
776 return req;
777
778 err:
779
780 TALLOC_FREE(req);
781 return NULL;
782 }
783
784 static void tldap_save_msg(struct tldap_context *ld, struct tevent_req *req)
785 {
786 struct tldap_req_state *state = tevent_req_data(
787 req, struct tldap_req_state);
788
789 TALLOC_FREE(ld->last_msg);
790 ld->last_msg = talloc_move(ld, &state->result);
791 }
792
793 static char *blob2string_talloc(TALLOC_CTX *mem_ctx, DATA_BLOB blob)
794 {
795 char *result = talloc_array(mem_ctx, char, blob.length+1);
796
797 if (result == NULL) {
798 return NULL;
799 }
800
801 memcpy(result, blob.data, blob.length);
802 result[blob.length] = '\0';
803 return result;
804 }
805
806 static bool asn1_read_OctetString_talloc(TALLOC_CTX *mem_ctx,
807 struct asn1_data *data,
808 char **presult)
809 {
810 DATA_BLOB string;
811 char *result;
812 if (!asn1_read_OctetString(data, mem_ctx, &string))
813 return false;
814
815 result = blob2string_talloc(mem_ctx, string);
816
817 data_blob_free(&string);
818
819 if (result == NULL) {
820 return false;
821 }
822 *presult = result;
823 return true;
824 }
825
826 static bool tldap_decode_controls(struct tldap_req_state *state);
827
828 static bool tldap_decode_response(struct tldap_req_state *state)
829 {
830 struct asn1_data *data = state->result->data;
831 struct tldap_message *msg = state->result;
832 int rc;
833 bool ok = true;
834
835 ok &= asn1_read_enumerated(data, &rc);
836 if (ok) {
837 msg->lderr = TLDAP_RC(rc);
838 }
839
840 ok &= asn1_read_OctetString_talloc(msg, data, &msg->res_matcheddn);
841 ok &= asn1_read_OctetString_talloc(msg, data,
842 &msg->res_diagnosticmessage);
843 if (!ok) return ok;
844 if (asn1_peek_tag(data, ASN1_CONTEXT(3))) {
845 ok &= asn1_start_tag(data, ASN1_CONTEXT(3));
846 ok &= asn1_read_OctetString_talloc(msg, data,
847 &msg->res_referral);
848 ok &= asn1_end_tag(data);
849 } else {
850 msg->res_referral = NULL;
851 }
852
853 return ok;
854 }
855
856 static void tldap_sasl_bind_done(struct tevent_req *subreq);
857
858 struct tevent_req *tldap_sasl_bind_send(TALLOC_CTX *mem_ctx,
859 struct tevent_context *ev,
860 struct tldap_context *ld,
861 const char *dn,
862 const char *mechanism,
863 DATA_BLOB *creds,
864 struct tldap_control *sctrls,
865 int num_sctrls,
866 struct tldap_control *cctrls,
867 int num_cctrls)
868 {
869 struct tevent_req *req, *subreq;
870 struct tldap_req_state *state;
871
872 req = tldap_req_create(mem_ctx, ld, &state);
873 if (req == NULL) {
874 return NULL;
875 }
876
877 if (dn == NULL) {
878 dn = "";
879 }
880
881 if (!asn1_push_tag(state->out, TLDAP_REQ_BIND)) goto err;
882 if (!asn1_write_Integer(state->out, ld->ld_version)) goto err;
883 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
884
885 if (mechanism == NULL) {
886 if (!asn1_push_tag(state->out, ASN1_CONTEXT_SIMPLE(0))) goto err;
887 if (!asn1_write(state->out, creds->data, creds->length)) goto err;
888 if (!asn1_pop_tag(state->out)) goto err;
889 } else {
890 if (!asn1_push_tag(state->out, ASN1_CONTEXT(3))) goto err;
891 if (!asn1_write_OctetString(state->out, mechanism,
892 strlen(mechanism))) goto err;
893 if ((creds != NULL) && (creds->data != NULL)) {
894 if (!asn1_write_OctetString(state->out, creds->data,
895 creds->length)) goto err;
896 }
897 if (!asn1_pop_tag(state->out)) goto err;
898 }
899
900 if (!asn1_pop_tag(state->out)) goto err;
901
902 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
903 sctrls, num_sctrls);
904 if (tevent_req_nomem(subreq, req)) {
905 return tevent_req_post(req, ev);
906 }
907 tevent_req_set_callback(subreq, tldap_sasl_bind_done, req);
908 return req;
909
910 err:
911
912 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
913 return tevent_req_post(req, ev);
914 }
915
916 static void tldap_sasl_bind_done(struct tevent_req *subreq)
917 {
918 struct tevent_req *req = tevent_req_callback_data(
919 subreq, struct tevent_req);
920 struct tldap_req_state *state = tevent_req_data(
921 req, struct tldap_req_state);
922 TLDAPRC rc;
923 bool ok;
924
925 rc = tldap_msg_recv(subreq, state, &state->result);
926 TALLOC_FREE(subreq);
927 if (tevent_req_ldap_error(req, rc)) {
928 return;
929 }
930 if (state->result->type != TLDAP_RES_BIND) {
931 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
932 return;
933 }
934
935 ok = asn1_start_tag(state->result->data, TLDAP_RES_BIND);
936 ok &= tldap_decode_response(state);
937
938 if (asn1_peek_tag(state->result->data, ASN1_CONTEXT_SIMPLE(7))) {
939 int len;
940
941 ok &= asn1_start_tag(state->result->data,
942 ASN1_CONTEXT_SIMPLE(7));
943 if (!ok) {
944 goto decode_error;
945 }
946
947 len = asn1_tag_remaining(state->result->data);
948 if (len == -1) {
949 goto decode_error;
950 }
951
952 state->result->res_serverSaslCreds =
953 data_blob_talloc(state->result, NULL, len);
954 if (state->result->res_serverSaslCreds.data == NULL) {
955 goto decode_error;
956 }
957
958 ok = asn1_read(state->result->data,
959 state->result->res_serverSaslCreds.data,
960 state->result->res_serverSaslCreds.length);
961
962 ok &= asn1_end_tag(state->result->data);
963 }
964
965 ok &= asn1_end_tag(state->result->data);
966
967 if (!ok) {
968 goto decode_error;
969 }
970
971 if (!TLDAP_RC_IS_SUCCESS(state->result->lderr) &&
972 !TLDAP_RC_EQUAL(state->result->lderr,
973 TLDAP_SASL_BIND_IN_PROGRESS)) {
974 tevent_req_ldap_error(req, state->result->lderr);
975 return;
976 }
977 tevent_req_done(req);
978 return;
979
980 decode_error:
981 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
982 return;
983 }
984
985 TLDAPRC tldap_sasl_bind_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
986 DATA_BLOB *serverSaslCreds)
987 {
988 struct tldap_req_state *state = tevent_req_data(
989 req, struct tldap_req_state);
990 TLDAPRC rc;
991
992 if (tevent_req_is_ldap_error(req, &rc)) {
993 return rc;
994 }
995
996 if (serverSaslCreds != NULL) {
997 serverSaslCreds->data = talloc_move(
998 mem_ctx, &state->result->res_serverSaslCreds.data);
999 serverSaslCreds->length =
1000 state->result->res_serverSaslCreds.length;
1001 }
1002
1003 return state->result->lderr;
1004 }
1005
1006 TLDAPRC tldap_sasl_bind(struct tldap_context *ld,
1007 const char *dn,
1008 const char *mechanism,
1009 DATA_BLOB *creds,
1010 struct tldap_control *sctrls,
1011 int num_sctrls,
1012 struct tldap_control *cctrls,
1013 int num_cctrls,
1014 TALLOC_CTX *mem_ctx,
1015 DATA_BLOB *serverSaslCreds)
1016 {
1017 TALLOC_CTX *frame = talloc_stackframe();
1018 struct tevent_context *ev;
1019 struct tevent_req *req;
1020 TLDAPRC rc = TLDAP_NO_MEMORY;
1021
1022 ev = samba_tevent_context_init(frame);
1023 if (ev == NULL) {
1024 goto fail;
1025 }
1026 req = tldap_sasl_bind_send(frame, ev, ld, dn, mechanism, creds,
1027 sctrls, num_sctrls, cctrls, num_cctrls);
1028 if (req == NULL) {
1029 goto fail;
1030 }
1031 if (!tevent_req_poll(req, ev)) {
1032 rc = TLDAP_OPERATIONS_ERROR;
1033 goto fail;
1034 }
1035 rc = tldap_sasl_bind_recv(req, mem_ctx, serverSaslCreds);
1036 tldap_save_msg(ld, req);
1037 fail:
1038 TALLOC_FREE(frame);
1039 return rc;
1040 }
1041
1042 struct tevent_req *tldap_simple_bind_send(TALLOC_CTX *mem_ctx,
1043 struct tevent_context *ev,
1044 struct tldap_context *ld,
1045 const char *dn,
1046 const char *passwd)
1047 {
1048 DATA_BLOB cred;
1049
1050 if (passwd != NULL) {
1051 cred.data = discard_const_p(uint8_t, passwd);
1052 cred.length = strlen(passwd);
1053 } else {
1054 cred.data = discard_const_p(uint8_t, "");
1055 cred.length = 0;
1056 }
1057 return tldap_sasl_bind_send(mem_ctx, ev, ld, dn, NULL, &cred, NULL, 0,
1058 NULL, 0);
1059 }
1060
1061 TLDAPRC tldap_simple_bind_recv(struct tevent_req *req)
1062 {
1063 return tldap_sasl_bind_recv(req, NULL, NULL);
1064 }
1065
1066 TLDAPRC tldap_simple_bind(struct tldap_context *ld, const char *dn,
1067 const char *passwd)
1068 {
1069 DATA_BLOB cred;
1070
1071 if (passwd != NULL) {
1072 cred.data = discard_const_p(uint8_t, passwd);
1073 cred.length = strlen(passwd);
1074 } else {
1075 cred.data = discard_const_p(uint8_t, "");
1076 cred.length = 0;
1077 }
1078 return tldap_sasl_bind(ld, dn, NULL, &cred, NULL, 0, NULL, 0,
1079 NULL, NULL);
1080 }
1081
1082 /*****************************************************************************/
1083
1084 /* can't use isalpha() as only a strict set is valid for LDAP */
1085
1086 static bool tldap_is_alpha(char c)
1087 {
1088 return (((c >= 'a') && (c <= 'z')) || \
1089 ((c >= 'A') && (c <= 'Z')));
1090 }
1091
1092 static bool tldap_is_adh(char c)
1093 {
1094 return tldap_is_alpha(c) || isdigit(c) || (c == '-');
1095 }
1096
1097 #define TLDAP_FILTER_AND ASN1_CONTEXT(0)
1098 #define TLDAP_FILTER_OR ASN1_CONTEXT(1)
1099 #define TLDAP_FILTER_NOT ASN1_CONTEXT(2)
1100 #define TLDAP_FILTER_EQ ASN1_CONTEXT(3)
1101 #define TLDAP_FILTER_SUB ASN1_CONTEXT(4)
1102 #define TLDAP_FILTER_LE ASN1_CONTEXT(5)
1103 #define TLDAP_FILTER_GE ASN1_CONTEXT(6)
1104 #define TLDAP_FILTER_PRES ASN1_CONTEXT_SIMPLE(7)
1105 #define TLDAP_FILTER_APX ASN1_CONTEXT(8)
1106 #define TLDAP_FILTER_EXT ASN1_CONTEXT(9)
1107
1108 #define TLDAP_SUB_INI ASN1_CONTEXT_SIMPLE(0)
1109 #define TLDAP_SUB_ANY ASN1_CONTEXT_SIMPLE(1)
1110 #define TLDAP_SUB_FIN ASN1_CONTEXT_SIMPLE(2)
1111
1112
1113 /* oid's should be numerical only in theory,
1114 * but apparently some broken servers may have alphanum aliases instead.
1115 * Do like openldap libraries and allow alphanum aliases for oids, but
1116 * do not allow Tagging options in that case.
1117 */
1118 static bool tldap_is_attrdesc(const char *s, int len, bool no_tagopts)
1119 {
1120 bool is_oid = false;
1121 bool dot = false;
1122 int i;
1123
1124 /* first char has stricter rules */
1125 if (isdigit(*s)) {
1126 is_oid = true;
1127 } else if (!tldap_is_alpha(*s)) {
1128 /* bad first char */
1129 return false;
1130 }
1131
1132 for (i = 1; i < len; i++) {
1133
1134 if (is_oid) {
1135 if (isdigit(s[i])) {
1136 dot = false;
1137 continue;
1138 }
1139 if (s[i] == '.') {
1140 if (dot) {
1141 /* malformed */
1142 return false;
1143 }
1144 dot = true;
1145 continue;
1146 }
1147 } else {
1148 if (tldap_is_adh(s[i])) {
1149 continue;
1150 }
1151 }
1152
1153 if (s[i] == ';') {
1154 if (no_tagopts) {
1155 /* no tagging options */
1156 return false;
1157 }
1158 if (dot) {
1159 /* malformed */
1160 return false;
1161 }
1162 if ((i + 1) == len) {
1163 /* malformed */
1164 return false;
1165 }
1166
1167 is_oid = false;
1168 continue;
1169 }
1170 }
1171
1172 if (dot) {
1173 /* malformed */
1174 return false;
1175 }
1176
1177 return true;
1178 }
1179
1180 /* this function copies the value until the closing parenthesis is found. */
1181 static char *tldap_get_val(TALLOC_CTX *memctx,
1182 const char *value, const char **_s)
1183 {
1184 const char *s = value;
1185
1186 /* find terminator */
1187 while (*s) {
1188 s = strchr(s, ')');
1189 if (s && (*(s - 1) == '\\')) {
1190 continue;
1191 }
1192 break;
1193 }
1194 if (!s || !(*s == ')')) {
1195 /* malformed filter */
1196 return NULL;
1197 }
1198
1199 *_s = s;
1200
1201 return talloc_strndup(memctx, value, s - value);
1202 }
1203
1204 static int tldap_hex2char(const char *x)
1205 {
1206 if (isxdigit(x[0]) && isxdigit(x[1])) {
1207 const char h1 = x[0], h2 = x[1];
1208 int c = 0;
1209
1210 if (h1 >= 'a') c = h1 - (int)'a' + 10;
1211 else if (h1 >= 'A') c = h1 - (int)'A' + 10;
1212 else if (h1 >= '0') c = h1 - (int)'0';
1213 c = c << 4;
1214 if (h2 >= 'a') c += h2 - (int)'a' + 10;
1215 else if (h2 >= 'A') c += h2 - (int)'A' + 10;
1216 else if (h2 >= '0') c += h2 - (int)'0';
1217
1218 return c;
1219 }
1220
1221 return -1;
1222 }
1223
1224 static bool tldap_find_first_star(const char *val, const char **star)
1225 {
1226 const char *s;
1227
1228 for (s = val; *s; s++) {
1229 switch (*s) {
1230 case '\\':
1231 if (isxdigit(s[1]) && isxdigit(s[2])) {
1232 s += 2;
1233 break;
1234 }
1235 /* not hex based escape, check older syntax */
1236 switch (s[1]) {
1237 case '(':
1238 case ')':
1239 case '*':
1240 case '\\':
1241 s++;
1242 break;
1243 default:
1244 /* invalid escape sequence */
1245 return false;
1246 }
1247 break;
1248 case ')':
1249 /* end of val, nothing found */
1250 *star = s;
1251 return true;
1252
1253 case '*':
1254 *star = s;
1255 return true;
1256 }
1257 }
1258
1259 /* string ended without closing parenthesis, filter is malformed */
1260 return false;
1261 }
1262
1263 static bool tldap_unescape_inplace(char *value, size_t *val_len)
1264 {
1265 int c, i, p;
1266
1267 for (i = 0,p = 0; i < *val_len; i++) {
1268
1269 switch (value[i]) {
1270 case '(':
1271 case ')':
1272 case '*':
1273 /* these must be escaped */
1274 return false;
1275
1276 case '\\':
1277 if (!value[i + 1]) {
1278 /* invalid EOL */
1279 return false;
1280 }
1281 i++;
1282
1283 c = tldap_hex2char(&value[i]);
1284 if (c >= 0 && c < 256) {
1285 value[p] = c;
1286 i++;
1287 p++;
1288 break;
1289 }
1290
1291 switch (value[i]) {
1292 case '(':
1293 case ')':
1294 case '*':
1295 case '\\':
1296 value[p] = value[i];
1297 p++;
1298 default:
1299 /* invalid */
1300 return false;
1301 }
1302 break;
1303
1304 default:
1305 value[p] = value[i];
1306 p++;
1307 }
1308 }
1309 value[p] = '\0';
1310 *val_len = p;
1311 return true;
1312 }
1313
1314 static bool tldap_push_filter_basic(struct tldap_context *ld,
1315 struct asn1_data *data,
1316 const char **_s);
1317 static bool tldap_push_filter_substring(struct tldap_context *ld,
1318 struct asn1_data *data,
1319 const char *val,
1320 const char **_s);
1321 static bool tldap_push_filter_int(struct tldap_context *ld,
1322 struct asn1_data *data,
1323 const char **_s)
1324 {
1325 const char *s = *_s;
1326 bool ret;
1327
1328 if (*s != '(') {
1329 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1330 "Incomplete or malformed filter\n");
1331 return false;
1332 }
1333 s++;
1334
1335 /* we are right after a parenthesis,
1336 * find out what op we have at hand */
1337 switch (*s) {
1338 case '&':
1339 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: AND\n");
1340 if (!asn1_push_tag(data, TLDAP_FILTER_AND)) return false;
1341 s++;
1342 break;
1343
1344 case '|':
1345 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: OR\n");
1346 if (!asn1_push_tag(data, TLDAP_FILTER_OR)) return false;
1347 s++;
1348 break;
1349
1350 case '!':
1351 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: NOT\n");
1352 if (!asn1_push_tag(data, TLDAP_FILTER_NOT)) return false;
1353 s++;
1354 ret = tldap_push_filter_int(ld, data, &s);
1355 if (!ret) {
1356 return false;
1357 }
1358 if (!asn1_pop_tag(data)) return false;
1359 goto done;
1360
1361 case '(':
1362 case ')':
1363 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1364 "Invalid parenthesis '%c'\n", *s);
1365 return false;
1366
1367 case '\0':
1368 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1369 "Invalid filter termination\n");
1370 return false;
1371
1372 default:
1373 ret = tldap_push_filter_basic(ld, data, &s);
1374 if (!ret) {
1375 return false;
1376 }
1377 goto done;
1378 }
1379
1380 /* only and/or filters get here.
1381 * go through the list of filters */
1382
1383 if (*s == ')') {
1384 /* RFC 4526: empty and/or */
1385 if (!asn1_pop_tag(data)) return false;
1386 goto done;
1387 }
1388
1389 while (*s) {
1390 ret = tldap_push_filter_int(ld, data, &s);
1391 if (!ret) {
1392 return false;
1393 }
1394
1395 if (*s == ')') {
1396 /* end of list, return */
1397 if (!asn1_pop_tag(data)) return false;
1398 break;
1399 }
1400 }
1401
1402 done:
1403 if (*s != ')') {
1404 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1405 "Incomplete or malformed filter\n");
1406 return false;
1407 }
1408 s++;
1409
1410 if (asn1_has_error(data)) {
1411 return false;
1412 }
1413
1414 *_s = s;
1415 return true;
1416 }
1417
1418
1419 static bool tldap_push_filter_basic(struct tldap_context *ld,
1420 struct asn1_data *data,
1421 const char **_s)
1422 {
1423 TALLOC_CTX *tmpctx = talloc_tos();
1424 const char *s = *_s;
1425 const char *e;
1426 const char *eq;
1427 const char *val;
1428 const char *type;
1429 const char *dn;
1430 const char *rule;
1431 const char *star;
1432 size_t type_len = 0;
1433 char *uval;
1434 size_t uval_len;
1435 bool write_octect = true;
1436 bool ret;
1437
1438 eq = strchr(s, '=');
1439 if (!eq) {
1440 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1441 "Invalid filter, missing equal sign\n");
1442 return false;
1443 }
1444
1445 val = eq + 1;
1446 e = eq - 1;
1447
1448 switch (*e) {
1449 case '<':
1450 if (!asn1_push_tag(data, TLDAP_FILTER_LE)) return false;
1451 break;
1452
1453 case '>':
1454 if (!asn1_push_tag(data, TLDAP_FILTER_GE)) return false;
1455 break;
1456
1457 case '~':
1458 if (!asn1_push_tag(data, TLDAP_FILTER_APX)) return false;
1459 break;
1460
1461 case ':':
1462 if (!asn1_push_tag(data, TLDAP_FILTER_EXT)) return false;
1463 write_octect = false;
1464
1465 type = NULL;
1466 dn = NULL;
1467 rule = NULL;
1468
1469 if (*s == ':') { /* [:dn]:rule:= value */
1470 if (s == e) {
1471 /* malformed filter */
1472 return false;
1473 }
1474 dn = s;
1475 } else { /* type[:dn][:rule]:= value */
1476 type = s;
1477 dn = strchr(s, ':');
1478 type_len = dn - type;
1479 if (dn == e) { /* type:= value */
1480 dn = NULL;
1481 }
1482 }
1483 if (dn) {
1484 dn++;
1485
1486 rule = strchr(dn, ':');
1487 if (rule == NULL) {
1488 return false;
1489 }
1490 if ((rule == dn + 1) || rule + 1 == e) {
1491 /* malformed filter, contains "::" */
1492 return false;
1493 }
1494
1495 if (strncasecmp_m(dn, "dn:", 3) != 0) {
1496 if (rule == e) {
1497 rule = dn;
1498 dn = NULL;
1499 } else {
1500 /* malformed filter. With two
1501 * optionals, the first must be "dn"
1502 */
1503 return false;
1504 }
1505 } else {
1506 if (rule == e) {
1507 rule = NULL;
1508 } else {
1509 rule++;
1510 }
1511 }
1512 }
1513
1514 if (!type && !dn && !rule) {
1515 /* malformed filter, there must be at least one */
1516 return false;
1517 }
1518
1519 /*
1520 MatchingRuleAssertion ::= SEQUENCE {
1521 matchingRule [1] MatchingRuleID OPTIONAL,
1522 type [2] AttributeDescription OPTIONAL,
1523 matchValue [3] AssertionValue,
1524 dnAttributes [4] BOOLEAN DEFAULT FALSE
1525 }
1526 */
1527
1528 /* check and add rule */
1529 if (rule) {
1530 ret = tldap_is_attrdesc(rule, e - rule, true);
1531 if (!ret) {
1532 return false;
1533 }
1534 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(1))) return false;
1535 if (!asn1_write(data, rule, e - rule)) return false;
1536 if (!asn1_pop_tag(data)) return false;
1537 }
1538
1539 /* check and add type */
1540 if (type) {
1541 ret = tldap_is_attrdesc(type, type_len, false);
1542 if (!ret) {
1543 return false;
1544 }
1545 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(2))) return false;
1546 if (!asn1_write(data, type, type_len)) return false;
1547 if (!asn1_pop_tag(data)) return false;
1548 }
1549
1550 uval = tldap_get_val(tmpctx, val, _s);
1551 if (!uval) {
1552 return false;
1553 }
1554 uval_len = *_s - val;
1555 ret = tldap_unescape_inplace(uval, &uval_len);
1556 if (!ret) {
1557 return false;
1558 }
1559
1560 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(3))) return false;
1561 if (!asn1_write(data, uval, uval_len)) return false;
1562 if (!asn1_pop_tag(data)) return false;
1563
1564 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(4))) return false;
1565 if (!asn1_write_uint8(data, dn?1:0)) return false;
1566 if (!asn1_pop_tag(data)) return false;
1567 break;
1568
1569 default:
1570 e = eq;
1571
1572 ret = tldap_is_attrdesc(s, e - s, false);
1573 if (!ret) {
1574 return false;
1575 }
1576
1577 if (strncmp(val, "*)", 2) == 0) {
1578 /* presence */
1579 if (!asn1_push_tag(data, TLDAP_FILTER_PRES)) return false;
1580 if (!asn1_write(data, s, e - s)) return false;
1581 *_s = val + 1;
1582 write_octect = false;
1583 break;
1584 }
1585
1586 ret = tldap_find_first_star(val, &star);
1587 if (!ret) {
1588 return false;
1589 }
1590 if (*star == '*') {
1591 /* substring */
1592 if (!asn1_push_tag(data, TLDAP_FILTER_SUB)) return false;
1593 if (!asn1_write_OctetString(data, s, e - s)) return false;
1594 ret = tldap_push_filter_substring(ld, data, val, &s);
1595 if (!ret) {
1596 return false;
1597 }
1598 *_s = s;
1599 write_octect = false;
1600 break;
1601 }
1602
1603 /* if nothing else, then it is just equality */
1604 if (!asn1_push_tag(data, TLDAP_FILTER_EQ)) return false;
1605 write_octect = true;
1606 break;
1607 }
1608
1609 if (write_octect) {
1610 uval = tldap_get_val(tmpctx, val, _s);
1611 if (!uval) {
1612 return false;
1613 }
1614 uval_len = *_s - val;
1615 ret = tldap_unescape_inplace(uval, &uval_len);
1616 if (!ret) {
1617 return false;
1618 }
1619
1620 if (!asn1_write_OctetString(data, s, e - s)) return false;
1621 if (!asn1_write_OctetString(data, uval, uval_len)) return false;
1622 }
1623
1624 if (asn1_has_error(data)) {
1625 return false;
1626 }
1627 return asn1_pop_tag(data);
1628 }
1629
1630 static bool tldap_push_filter_substring(struct tldap_context *ld,
1631 struct asn1_data *data,
1632 const char *val,
1633 const char **_s)
1634 {
1635 TALLOC_CTX *tmpctx = talloc_tos();
1636 bool initial = true;
1637 const char *star;
1638 char *chunk;
1639 size_t chunk_len;
1640 bool ret;
1641
1642 /*
1643 SubstringFilter ::= SEQUENCE {
1644 type AttributeDescription,
1645 -- at least one must be present
1646 substrings SEQUENCE OF CHOICE {
1647 initial [0] LDAPString,
1648 any [1] LDAPString,
1649 final [2] LDAPString } }
1650 */
1651 if (!asn1_push_tag(data, ASN1_SEQUENCE(0))) return false;
1652
1653 do {
1654 ret = tldap_find_first_star(val, &star);
1655 if (!ret) {
1656 return false;
1657 }
1658 chunk_len = star - val;
1659
1660 switch (*star) {
1661 case '*':
1662 if (!initial && chunk_len == 0) {
1663 /* found '**', which is illegal */
1664 return false;
1665 }
1666 break;
1667 case ')':
1668 if (initial) {
1669 /* no stars ?? */
1670 return false;
1671 }
1672 /* we are done */
1673 break;
1674 default:
1675 /* ?? */
1676 return false;
1677 }
1678
1679 if (initial && chunk_len == 0) {
1680 val = star + 1;
1681 initial = false;
1682 continue;
1683 }
1684
1685 chunk = talloc_strndup(tmpctx, val, chunk_len);
1686 if (!chunk) {
1687 return false;
1688 }
1689 ret = tldap_unescape_inplace(chunk, &chunk_len);
1690 if (!ret) {
1691 return false;
1692 }
1693 switch (*star) {
1694 case '*':
1695 if (initial) {
1696 if (!asn1_push_tag(data, TLDAP_SUB_INI)) return false;
1697 initial = false;
1698 } else {
1699 if (!asn1_push_tag(data, TLDAP_SUB_ANY)) return false;
1700 }
1701 break;
1702 case ')':
1703 if (!asn1_push_tag(data, TLDAP_SUB_FIN)) return false;
1704 break;
1705 default:
1706 /* ?? */
1707 return false;
1708 }
1709 if (!asn1_write(data, chunk, chunk_len)) return false;
1710 if (!asn1_pop_tag(data)) return false;
1711
1712 val = star + 1;
1713
1714 } while (*star == '*');
1715
1716 *_s = star;
1717
1718 /* end of sequence */
1719 return asn1_pop_tag(data);
1720 }
1721
1722 /* NOTE: although openldap libraries allow for spaces in some places, mosly
1723 * around parenthesis, we do not allow any spaces (except in values of
1724 * course) as I couldn't fine any place in RFC 4512 or RFC 4515 where
1725 * leading or trailing spaces where allowed.
1726 */
1727 static bool tldap_push_filter(struct tldap_context *ld,
1728 struct asn1_data *data,
1729 const char *filter)
1730 {
1731 const char *s = filter;
1732 bool ret;
1733
1734 ret = tldap_push_filter_int(ld, data, &s);
1735 if (ret && *s) {
1736 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1737 "Incomplete or malformed filter\n");
1738 return false;
1739 }
1740 return ret;
1741 }
1742
1743 /*****************************************************************************/
1744
1745 static void tldap_search_done(struct tevent_req *subreq);
1746
1747 struct tevent_req *tldap_search_send(TALLOC_CTX *mem_ctx,
1748 struct tevent_context *ev,
1749 struct tldap_context *ld,
1750 const char *base, int scope,
1751 const char *filter,
1752 const char **attrs,
1753 int num_attrs,
1754 int attrsonly,
1755 struct tldap_control *sctrls,
1756 int num_sctrls,
1757 struct tldap_control *cctrls,
1758 int num_cctrls,
1759 int timelimit,
1760 int sizelimit,
1761 int deref)
1762 {
1763 struct tevent_req *req, *subreq;
1764 struct tldap_req_state *state;
1765 int i;
1766
1767 req = tldap_req_create(mem_ctx, ld, &state);
1768 if (req == NULL) {
1769 return NULL;
1770 }
1771
1772 if (!asn1_push_tag(state->out, TLDAP_REQ_SEARCH)) goto encoding_error;
1773 if (!asn1_write_OctetString(state->out, base, strlen(base))) goto encoding_error;
1774 if (!asn1_write_enumerated(state->out, scope)) goto encoding_error;
1775 if (!asn1_write_enumerated(state->out, deref)) goto encoding_error;
1776 if (!asn1_write_Integer(state->out, sizelimit)) goto encoding_error;
1777 if (!asn1_write_Integer(state->out, timelimit)) goto encoding_error;
1778 if (!asn1_write_BOOLEAN(state->out, attrsonly)) goto encoding_error;
1779
1780 if (!tldap_push_filter(ld, state->out, filter)) {
1781 goto encoding_error;
1782 }
1783
1784 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto encoding_error;
1785 for (i=0; i<num_attrs; i++) {
1786 if (!asn1_write_OctetString(state->out, attrs[i], strlen(attrs[i]))) goto encoding_error;
1787 }
1788 if (!asn1_pop_tag(state->out)) goto encoding_error;
1789 if (!asn1_pop_tag(state->out)) goto encoding_error;
1790
1791 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
1792 sctrls, num_sctrls);
1793 if (tevent_req_nomem(subreq, req)) {
1794 return tevent_req_post(req, ev);
1795 }
1796 tevent_req_set_callback(subreq, tldap_search_done, req);
1797 return req;
1798
1799 encoding_error:
1800 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
1801 return tevent_req_post(req, ev);
1802 }
1803
1804 static void tldap_search_done(struct tevent_req *subreq)
1805 {
1806 struct tevent_req *req = tevent_req_callback_data(
1807 subreq, struct tevent_req);
1808 struct tldap_req_state *state = tevent_req_data(
1809 req, struct tldap_req_state);
1810 TLDAPRC rc;
1811
1812 rc = tldap_msg_recv(subreq, state, &state->result);
1813 if (tevent_req_ldap_error(req, rc)) {
1814 return;
1815 }
1816 switch (state->result->type) {
1817 case TLDAP_RES_SEARCH_ENTRY:
1818 case TLDAP_RES_SEARCH_REFERENCE:
1819 if (!tldap_msg_set_pending(subreq)) {
1820 tevent_req_oom(req);
1821 return;
1822 }
1823 tevent_req_notify_callback(req);
1824 break;
1825 case TLDAP_RES_SEARCH_RESULT:
1826 TALLOC_FREE(subreq);
1827 if (!asn1_start_tag(state->result->data,
1828 state->result->type) ||
1829 !tldap_decode_response(state) ||
1830 !asn1_end_tag(state->result->data) ||
1831 !tldap_decode_controls(state)) {
1832 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
1833 return;
1834 }
1835 tevent_req_done(req);
1836 break;
1837 default:
1838 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
1839 return;
1840 }
1841 }
1842
1843 TLDAPRC tldap_search_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1844 struct tldap_message **pmsg)
1845 {
1846 struct tldap_req_state *state = tevent_req_data(
1847 req, struct tldap_req_state);
1848 TLDAPRC rc;
1849
1850 if (!tevent_req_is_in_progress(req)
1851 && tevent_req_is_ldap_error(req, &rc)) {
1852 return rc;
1853 }
1854
1855 if (tevent_req_is_in_progress(req)) {
1856 switch (state->result->type) {
1857 case TLDAP_RES_SEARCH_ENTRY:
1858 case TLDAP_RES_SEARCH_REFERENCE:
1859 break;
1860 default:
1861 return TLDAP_OPERATIONS_ERROR;
1862 }
1863 }
1864
1865 *pmsg = talloc_move(mem_ctx, &state->result);
1866 return TLDAP_SUCCESS;
1867 }
1868
1869 struct tldap_search_all_state {
1870 struct tldap_message **msgs;
1871 struct tldap_message *result;
1872 };
1873
1874 static void tldap_search_all_done(struct tevent_req *subreq);
1875
1876 struct tevent_req *tldap_search_all_send(
1877 TALLOC_CTX *mem_ctx, struct tevent_context *ev,
1878 struct tldap_context *ld, const char *base, int scope,
1879 const char *filter, const char **attrs, int num_attrs, int attrsonly,
1880 struct tldap_control *sctrls, int num_sctrls,
1881 struct tldap_control *cctrls, int num_cctrls,
1882 int timelimit, int sizelimit, int deref)
1883 {
1884 struct tevent_req *req, *subreq;
1885 struct tldap_search_all_state *state;
1886
1887 req = tevent_req_create(mem_ctx, &state,
1888 struct tldap_search_all_state);
1889 if (req == NULL) {
1890 return NULL;
1891 }
1892
1893 subreq = tldap_search_send(state, ev, ld, base, scope, filter,
1894 attrs, num_attrs, attrsonly,
1895 sctrls, num_sctrls, cctrls, num_cctrls,
1896 timelimit, sizelimit, deref);
1897 if (tevent_req_nomem(subreq, req)) {
1898 return tevent_req_post(req, ev);
1899 }
1900 tevent_req_set_callback(subreq, tldap_search_all_done, req);
1901 return req;
1902 }
1903
1904 static void tldap_search_all_done(struct tevent_req *subreq)
1905 {
1906 struct tevent_req *req = tevent_req_callback_data(
1907 subreq, struct tevent_req);
1908 struct tldap_search_all_state *state = tevent_req_data(
1909 req, struct tldap_search_all_state);
1910 struct tldap_message *msg, **tmp;
1911 size_t num_msgs;
1912 TLDAPRC rc;
1913 int msgtype;
1914
1915 rc = tldap_search_recv(subreq, state, &msg);
1916 /* No TALLOC_FREE(subreq), this is multi-step */
1917 if (tevent_req_ldap_error(req, rc)) {
1918 TALLOC_FREE(subreq);
1919 return;
1920 }
1921
1922 msgtype = tldap_msg_type(msg);
1923 if (msgtype == TLDAP_RES_SEARCH_RESULT) {
1924 state->result = msg;
1925 tevent_req_done(req);
1926 return;
1927 }
1928
1929 num_msgs = talloc_array_length(state->msgs);
1930
1931 tmp = talloc_realloc(state, state->msgs, struct tldap_message *,
1932 num_msgs + 1);
1933 if (tevent_req_nomem(tmp, req)) {
1934 return;
1935 }
1936 state->msgs = tmp;
1937 state->msgs[num_msgs] = talloc_move(state->msgs, &msg);
1938 }
1939
1940 TLDAPRC tldap_search_all_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1941 struct tldap_message ***msgs,
1942 struct tldap_message **result)
1943 {
1944 struct tldap_search_all_state *state = tevent_req_data(
1945 req, struct tldap_search_all_state);
1946 TLDAPRC rc;
1947
1948 if (tevent_req_is_ldap_error(req, &rc)) {
1949 return rc;
1950 }
1951
1952 if (msgs != NULL) {
1953 *msgs = talloc_move(mem_ctx, &state->msgs);
1954 }
1955 if (result != NULL) {
1956 *result = talloc_move(mem_ctx, &state->result);
1957 }
1958
1959 return TLDAP_SUCCESS;
1960 }
1961
1962 TLDAPRC tldap_search(struct tldap_context *ld,
1963 const char *base, int scope, const char *filter,
1964 const char **attrs, int num_attrs, int attrsonly,
1965 struct tldap_control *sctrls, int num_sctrls,
1966 struct tldap_control *cctrls, int num_cctrls,
1967 int timelimit, int sizelimit, int deref,
1968 TALLOC_CTX *mem_ctx, struct tldap_message ***pmsgs)
1969 {
1970 TALLOC_CTX *frame;
1971 struct tevent_context *ev;
1972 struct tevent_req *req;
1973 TLDAPRC rc = TLDAP_NO_MEMORY;
1974 struct tldap_message **msgs;
1975 struct tldap_message *result;
1976
1977 if (tldap_pending_reqs(ld)) {
1978 return TLDAP_BUSY;
1979 }
1980
1981 frame = talloc_stackframe();
1982
1983 ev = samba_tevent_context_init(frame);
1984 if (ev == NULL) {
1985 goto fail;
1986 }
1987 req = tldap_search_all_send(frame, ev, ld, base, scope, filter,
1988 attrs, num_attrs, attrsonly,
1989 sctrls, num_sctrls, cctrls, num_cctrls,
1990 timelimit, sizelimit, deref);
1991 if (req == NULL) {
1992 goto fail;
1993 }
1994 if (!tevent_req_poll(req, ev)) {
1995 rc = TLDAP_OPERATIONS_ERROR;
1996 goto fail;
1997 }
1998 rc = tldap_search_all_recv(req, mem_ctx, &msgs, &result);
1999 TALLOC_FREE(req);
2000 if (!TLDAP_RC_IS_SUCCESS(rc)) {
2001 goto fail;
2002 }
2003
2004 TALLOC_FREE(ld->last_msg);
2005 ld->last_msg = talloc_move(ld, &result);
2006
2007 *pmsgs = msgs;
2008 fail:
2009 TALLOC_FREE(frame);
2010 return rc;
2011 }
2012
2013 static bool tldap_parse_search_entry(struct tldap_message *msg)
2014 {
2015 int num_attribs = 0;
2016
2017 if (msg->type != TLDAP_RES_SEARCH_ENTRY) {
2018 return false;
2019 }
2020 if (!asn1_start_tag(msg->data, TLDAP_RES_SEARCH_ENTRY)) {
2021 return false;
2022 }
2023
2024 /* dn */
2025
2026 if (!asn1_read_OctetString_talloc(msg, msg->data, &msg->dn)) return false;
2027
2028 if (msg->dn == NULL) {
2029 return false;
2030 }
2031
2032 /*
2033 * Attributes: We overallocate msg->attribs by one, so that while
2034 * looping over the attributes we can directly parse into the last
2035 * array element. Same for the values in the inner loop.
2036 */
2037
2038 msg->attribs = talloc_array(msg, struct tldap_attribute, 1);
2039 if (msg->attribs == NULL) {
2040 return false;
2041 }
2042
2043 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2044 while (asn1_peek_tag(msg->data, ASN1_SEQUENCE(0))) {
2045 struct tldap_attribute *attrib;
2046 int num_values = 0;
2047
2048 attrib = &msg->attribs[num_attribs];
2049 attrib->values = talloc_array(msg->attribs, DATA_BLOB, 1);
2050 if (attrib->values == NULL) {
2051 return false;
2052 }
2053 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2054 if (!asn1_read_OctetString_talloc(msg->attribs, msg->data,
2055 &attrib->name)) return false;
2056 if (!asn1_start_tag(msg->data, ASN1_SET)) return false;
2057
2058 while (asn1_peek_tag(msg->data, ASN1_OCTET_STRING)) {
2059 if (!asn1_read_OctetString(msg->data, msg,
2060 &attrib->values[num_values])) return false;
2061
2062 attrib->values = talloc_realloc(
2063 msg->attribs, attrib->values, DATA_BLOB,
2064 num_values + 2);
2065 if (attrib->values == NULL) {
2066 return false;
2067 }
2068 num_values += 1;
2069 }
2070 attrib->values = talloc_realloc(msg->attribs, attrib->values,
2071 DATA_BLOB, num_values);
2072 attrib->num_values = num_values;
2073
2074 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SET */
2075 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SEQUENCE(0) */
2076 msg->attribs = talloc_realloc(
2077 msg, msg->attribs, struct tldap_attribute,
2078 num_attribs + 2);
2079 if (msg->attribs == NULL) {
2080 return false;
2081 }
2082 num_attribs += 1;
2083 }
2084 msg->attribs = talloc_realloc(
2085 msg, msg->attribs, struct tldap_attribute, num_attribs);
2086 return asn1_end_tag(msg->data);
2087 }
2088
2089 bool tldap_entry_dn(struct tldap_message *msg, char **dn)
2090 {
2091 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2092 return false;
2093 }
2094 *dn = msg->dn;
2095 return true;
2096 }
2097
2098 bool tldap_entry_attributes(struct tldap_message *msg,
2099 struct tldap_attribute **attributes,
2100 int *num_attributes)
2101 {
2102 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2103 return false;
2104 }
2105 *attributes = msg->attribs;
2106 *num_attributes = talloc_array_length(msg->attribs);
2107 return true;
2108 }
2109
2110 static bool tldap_decode_controls(struct tldap_req_state *state)
2111 {
2112 struct tldap_message *msg = state->result;
2113 struct asn1_data *data = msg->data;
2114 struct tldap_control *sctrls = NULL;
2115 int num_controls = 0;
2116 bool ret = false;
2117
2118 msg->res_sctrls = NULL;
2119
2120 if (!asn1_peek_tag(data, ASN1_CONTEXT(0))) {
2121 return true;
2122 }
2123
2124 if (!asn1_start_tag(data, ASN1_CONTEXT(0))) goto out;
2125
2126 while (asn1_peek_tag(data, ASN1_SEQUENCE(0))) {
2127 struct tldap_control *c;
2128 char *oid = NULL;
2129
2130 sctrls = talloc_realloc(msg, sctrls, struct tldap_control,
2131 num_controls + 1);
2132 if (sctrls == NULL) {
2133 goto out;
2134 }
2135 c = &sctrls[num_controls];
2136
2137 if (!asn1_start_tag(data, ASN1_SEQUENCE(0))) goto out;
2138 if (!asn1_read_OctetString_talloc(msg, data, &oid)) goto out;
2139 if (asn1_has_error(data) || (oid == NULL)) {
2140 goto out;
2141 }
2142 c->oid = oid;
2143 if (asn1_peek_tag(data, ASN1_BOOLEAN)) {
2144 if (!asn1_read_BOOLEAN(data, &c->critical)) goto out;
2145 } else {
2146 c->critical = false;
2147 }
2148 c->value = data_blob_null;
2149 if (asn1_peek_tag(data, ASN1_OCTET_STRING) &&
2150 !asn1_read_OctetString(data, msg, &c->value)) {
2151 goto out;
2152 }
2153 if (!asn1_end_tag(data)) goto out; /* ASN1_SEQUENCE(0) */
2154
2155 num_controls += 1;
2156 }
2157
2158 if (!asn1_end_tag(data)) goto out; /* ASN1_CONTEXT(0) */
2159
2160 ret = true;
2161
2162 out:
2163
2164 if (ret) {
2165 msg->res_sctrls = sctrls;
2166 } else {
2167 TALLOC_FREE(sctrls);
2168 }
2169 return ret;
2170 }
2171
2172 static void tldap_simple_done(struct tevent_req *subreq, int type)
2173 {
2174 struct tevent_req *req = tevent_req_callback_data(
2175 subreq, struct tevent_req);
2176 struct tldap_req_state *state = tevent_req_data(
2177 req, struct tldap_req_state);
2178 TLDAPRC rc;
2179
2180 rc = tldap_msg_recv(subreq, state, &state->result);
2181 TALLOC_FREE(subreq);
2182 if (tevent_req_ldap_error(req, rc)) {
2183 return;
2184 }
2185 if (state->result->type != type) {
2186 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
2187 return;
2188 }
2189 if (!asn1_start_tag(state->result->data, state->result->type) ||
2190 !tldap_decode_response(state) ||
2191 !asn1_end_tag(state->result->data) ||
2192 !tldap_decode_controls(state)) {
2193 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
2194 return;
2195 }
2196 if (!TLDAP_RC_IS_SUCCESS(state->result->lderr)) {
2197 tevent_req_ldap_error(req, state->result->lderr);
2198 return;
2199 }
2200 tevent_req_done(req);
2201 }
2202
2203 static TLDAPRC tldap_simple_recv(struct tevent_req *req)
2204 {
2205 TLDAPRC rc;
2206 if (tevent_req_is_ldap_error(req, &rc)) {
2207 return rc;
2208 }
2209 return TLDAP_SUCCESS;
2210 }
2211
2212 static void tldap_add_done(struct tevent_req *subreq);
2213
2214 struct tevent_req *tldap_add_send(TALLOC_CTX *mem_ctx,
2215 struct tevent_context *ev,
2216 struct tldap_context *ld,
2217 const char *dn,
2218 struct tldap_mod *attributes,
2219 int num_attributes,
2220 struct tldap_control *sctrls,
2221 int num_sctrls,
2222 struct tldap_control *cctrls,
2223 int num_cctrls)
2224 {
2225 struct tevent_req *req, *subreq;
2226 struct tldap_req_state *state;
2227 int i, j;
2228
2229 req = tldap_req_create(mem_ctx, ld, &state);
2230 if (req == NULL) {
2231 return NULL;
2232 }
2233
2234 if (!asn1_push_tag(state->out, TLDAP_REQ_ADD)) goto err;
2235 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2236 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2237
2238 for (i=0; i<num_attributes; i++) {
2239 struct tldap_mod *attrib = &attributes[i];
2240 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2241 if (!asn1_write_OctetString(state->out, attrib->attribute,
2242 strlen(attrib->attribute))) goto err;
2243 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2244 for (j=0; j<attrib->num_values; j++) {
2245 if (!asn1_write_OctetString(state->out,
2246 attrib->values[j].data,
2247 attrib->values[j].length)) goto err;
2248 }
2249 if (!asn1_pop_tag(state->out)) goto err;
2250 if (!asn1_pop_tag(state->out)) goto err;
2251 }
2252
2253 if (!asn1_pop_tag(state->out)) goto err;
2254 if (!asn1_pop_tag(state->out)) goto err;
2255
2256 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2257 sctrls, num_sctrls);
2258 if (tevent_req_nomem(subreq, req)) {
2259 return tevent_req_post(req, ev);
2260 }
2261 tevent_req_set_callback(subreq, tldap_add_done, req);
2262 return req;
2263
2264 err:
2265
2266 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2267 return tevent_req_post(req, ev);
2268 }
2269
2270 static void tldap_add_done(struct tevent_req *subreq)
2271 {
2272 tldap_simple_done(subreq, TLDAP_RES_ADD);
2273 }
2274
2275 TLDAPRC tldap_add_recv(struct tevent_req *req)
2276 {
2277 return tldap_simple_recv(req);
2278 }
2279
2280 TLDAPRC tldap_add(struct tldap_context *ld, const char *dn,
2281 struct tldap_mod *attributes, int num_attributes,
2282 struct tldap_control *sctrls, int num_sctrls,
2283 struct tldap_control *cctrls, int num_cctrls)
2284 {
2285 TALLOC_CTX *frame = talloc_stackframe();
2286 struct tevent_context *ev;
2287 struct tevent_req *req;
2288 TLDAPRC rc = TLDAP_NO_MEMORY;
2289
2290 ev = samba_tevent_context_init(frame);
2291 if (ev == NULL) {
2292 goto fail;
2293 }
2294 req = tldap_add_send(frame, ev, ld, dn, attributes, num_attributes,
2295 sctrls, num_sctrls, cctrls, num_cctrls);
2296 if (req == NULL) {
2297 goto fail;
2298 }
2299 if (!tevent_req_poll(req, ev)) {
2300 rc = TLDAP_OPERATIONS_ERROR;
2301 goto fail;
2302 }
2303 rc = tldap_add_recv(req);
2304 tldap_save_msg(ld, req);
2305 fail:
2306 TALLOC_FREE(frame);
2307 return rc;
2308 }
2309
2310 static void tldap_modify_done(struct tevent_req *subreq);
2311
2312 struct tevent_req *tldap_modify_send(TALLOC_CTX *mem_ctx,
2313 struct tevent_context *ev,
2314 struct tldap_context *ld,
2315 const char *dn,
2316 struct tldap_mod *mods, int num_mods,
2317 struct tldap_control *sctrls,
2318 int num_sctrls,
2319 struct tldap_control *cctrls,
2320 int num_cctrls)
2321 {
2322 struct tevent_req *req, *subreq;
2323 struct tldap_req_state *state;
2324 int i, j;
2325
2326 req = tldap_req_create(mem_ctx, ld, &state);
2327 if (req == NULL) {
2328 return NULL;
2329 }
2330
2331 if (!asn1_push_tag(state->out, TLDAP_REQ_MODIFY)) goto err;
2332 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2333 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2334
2335 for (i=0; i<num_mods; i++) {
2336 struct tldap_mod *mod = &mods[i];
2337 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2338 if (!asn1_write_enumerated(state->out, mod->mod_op)) goto err;
2339 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2340 if (!asn1_write_OctetString(state->out, mod->attribute,
2341 strlen(mod->attribute))) goto err;
2342 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2343 for (j=0; j<mod->num_values; j++) {
2344 if (!asn1_write_OctetString(state->out,
2345 mod->values[j].data,
2346 mod->values[j].length)) goto err;
2347 }
2348 if (!asn1_pop_tag(state->out)) goto err;
2349 if (!asn1_pop_tag(state->out)) goto err;
2350 if (!asn1_pop_tag(state->out)) goto err;
2351 }
2352
2353 if (!asn1_pop_tag(state->out)) goto err;
2354 if (!asn1_pop_tag(state->out)) goto err;
2355
2356 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2357 sctrls, num_sctrls);
2358 if (tevent_req_nomem(subreq, req)) {
2359 return tevent_req_post(req, ev);
2360 }
2361 tevent_req_set_callback(subreq, tldap_modify_done, req);
2362 return req;
2363
2364 err:
2365
2366 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2367 return tevent_req_post(req, ev);
2368 }
2369
2370 static void tldap_modify_done(struct tevent_req *subreq)
2371 {
2372 tldap_simple_done(subreq, TLDAP_RES_MODIFY);
2373 }
2374
2375 TLDAPRC tldap_modify_recv(struct tevent_req *req)
2376 {
2377 return tldap_simple_recv(req);
2378 }
2379
2380 TLDAPRC tldap_modify(struct tldap_context *ld, const char *dn,
2381 struct tldap_mod *mods, int num_mods,
2382 struct tldap_control *sctrls, int num_sctrls,
2383 struct tldap_control *cctrls, int num_cctrls)
2384 {
2385 TALLOC_CTX *frame = talloc_stackframe();
2386 struct tevent_context *ev;
2387 struct tevent_req *req;
2388 TLDAPRC rc = TLDAP_NO_MEMORY;
2389
2390 ev = samba_tevent_context_init(frame);
2391 if (ev == NULL) {
2392 goto fail;
2393 }
2394 req = tldap_modify_send(frame, ev, ld, dn, mods, num_mods,
2395 sctrls, num_sctrls, cctrls, num_cctrls);
2396 if (req == NULL) {
2397 goto fail;
2398 }
2399 if (!tevent_req_poll(req, ev)) {
2400 rc = TLDAP_OPERATIONS_ERROR;
2401 goto fail;
2402 }
2403 rc = tldap_modify_recv(req);
2404 tldap_save_msg(ld, req);
2405 fail:
2406 TALLOC_FREE(frame);
2407 return rc;
2408 }
2409
2410 static void tldap_delete_done(struct tevent_req *subreq);
2411
2412 struct tevent_req *tldap_delete_send(TALLOC_CTX *mem_ctx,
2413 struct tevent_context *ev,
2414 struct tldap_context *ld,
2415 const char *dn,
2416 struct tldap_control *sctrls,
2417 int num_sctrls,
2418 struct tldap_control *cctrls,
2419 int num_cctrls)
2420 {
2421 struct tevent_req *req, *subreq;
2422 struct tldap_req_state *state;
2423
2424 req = tldap_req_create(mem_ctx, ld, &state);
2425 if (req == NULL) {
2426 return NULL;
2427 }
2428
2429 if (!asn1_push_tag(state->out, TLDAP_REQ_DELETE)) goto err;
2430 if (!asn1_write(state->out, dn, strlen(dn))) goto err;
2431 if (!asn1_pop_tag(state->out)) goto err;
2432
2433 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2434 sctrls, num_sctrls);
2435 if (tevent_req_nomem(subreq, req)) {
2436 return tevent_req_post(req, ev);
2437 }
2438 tevent_req_set_callback(subreq, tldap_delete_done, req);
2439 return req;
2440
2441 err:
2442
2443 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2444 return tevent_req_post(req, ev);
2445 }
2446
2447 static void tldap_delete_done(struct tevent_req *subreq)
2448 {
2449 tldap_simple_done(subreq, TLDAP_RES_DELETE);
2450 }
2451
2452 TLDAPRC tldap_delete_recv(struct tevent_req *req)
2453 {
2454 return tldap_simple_recv(req);
2455 }
2456
2457 TLDAPRC tldap_delete(struct tldap_context *ld, const char *dn,
2458 struct tldap_control *sctrls, int num_sctrls,
2459 struct tldap_control *cctrls, int num_cctrls)
2460 {
2461 TALLOC_CTX *frame = talloc_stackframe();
2462 struct tevent_context *ev;
2463 struct tevent_req *req;
2464 TLDAPRC rc = TLDAP_NO_MEMORY;
2465
2466 ev = samba_tevent_context_init(frame);
2467 if (ev == NULL) {
2468 goto fail;
2469 }
2470 req = tldap_delete_send(frame, ev, ld, dn, sctrls, num_sctrls,
2471 cctrls, num_cctrls);
2472 if (req == NULL) {
2473 goto fail;
2474 }
2475 if (!tevent_req_poll(req, ev)) {
2476 rc = TLDAP_OPERATIONS_ERROR;
2477 goto fail;
2478 }
2479 rc = tldap_delete_recv(req);
2480 tldap_save_msg(ld, req);
2481 fail:
2482 TALLOC_FREE(frame);
2483 return rc;
2484 }
2485
2486 int tldap_msg_id(const struct tldap_message *msg)
2487 {
2488 return msg->id;
2489 }
2490
2491 int tldap_msg_type(const struct tldap_message *msg)
2492 {
2493 return msg->type;
2494 }
2495
2496 const char *tldap_msg_matcheddn(struct tldap_message *msg)
2497 {
2498 if (msg == NULL) {
2499 return NULL;
2500 }
2501 return msg->res_matcheddn;
2502 }
2503
2504 const char *tldap_msg_diagnosticmessage(struct tldap_message *msg)
2505 {
2506 if (msg == NULL) {
2507 return NULL;
2508 }
2509 return msg->res_diagnosticmessage;
2510 }
2511
2512 const char *tldap_msg_referral(struct tldap_message *msg)
2513 {
2514 if (msg == NULL) {
2515 return NULL;
2516 }
2517 return msg->res_referral;
2518 }
2519
2520 void tldap_msg_sctrls(struct tldap_message *msg, int *num_sctrls,
2521 struct tldap_control **sctrls)
2522 {
2523 if (msg == NULL) {
2524 *sctrls = NULL;
2525 *num_sctrls = 0;
2526 return;
2527 }
2528 *sctrls = msg->res_sctrls;
2529 *num_sctrls = talloc_array_length(msg->res_sctrls);
2530 }
2531
2532 struct tldap_message *tldap_ctx_lastmsg(struct tldap_context *ld)
2533 {
2534 return ld->last_msg;
2535 }
2536
2537 static const struct { TLDAPRC rc; const char *string; } tldaprc_errmap[] =
2538 {
2539 { TLDAP_SUCCESS,
2540 "TLDAP_SUCCESS" },
2541 { TLDAP_OPERATIONS_ERROR,
2542 "TLDAP_OPERATIONS_ERROR" },
2543 { TLDAP_PROTOCOL_ERROR,
2544 "TLDAP_PROTOCOL_ERROR" },
2545 { TLDAP_TIMELIMIT_EXCEEDED,
2546 "TLDAP_TIMELIMIT_EXCEEDED" },
2547 { TLDAP_SIZELIMIT_EXCEEDED,
2548 "TLDAP_SIZELIMIT_EXCEEDED" },
2549 { TLDAP_COMPARE_FALSE,
2550 "TLDAP_COMPARE_FALSE" },
2551 { TLDAP_COMPARE_TRUE,
2552 "TLDAP_COMPARE_TRUE" },
2553 { TLDAP_STRONG_AUTH_NOT_SUPPORTED,
2554 "TLDAP_STRONG_AUTH_NOT_SUPPORTED" },
2555 { TLDAP_STRONG_AUTH_REQUIRED,
2556 "TLDAP_STRONG_AUTH_REQUIRED" },
2557 { TLDAP_REFERRAL,
2558 "TLDAP_REFERRAL" },
2559 { TLDAP_ADMINLIMIT_EXCEEDED,
2560 "TLDAP_ADMINLIMIT_EXCEEDED" },
2561 { TLDAP_UNAVAILABLE_CRITICAL_EXTENSION,
2562 "TLDAP_UNAVAILABLE_CRITICAL_EXTENSION" },
2563 { TLDAP_CONFIDENTIALITY_REQUIRED,
2564 "TLDAP_CONFIDENTIALITY_REQUIRED" },
2565 { TLDAP_SASL_BIND_IN_PROGRESS,
2566 "TLDAP_SASL_BIND_IN_PROGRESS" },
2567 { TLDAP_NO_SUCH_ATTRIBUTE,
2568 "TLDAP_NO_SUCH_ATTRIBUTE" },
2569 { TLDAP_UNDEFINED_TYPE,
2570 "TLDAP_UNDEFINED_TYPE" },
2571 { TLDAP_INAPPROPRIATE_MATCHING,
2572 "TLDAP_INAPPROPRIATE_MATCHING" },
2573 { TLDAP_CONSTRAINT_VIOLATION,
2574 "TLDAP_CONSTRAINT_VIOLATION" },
2575 { TLDAP_TYPE_OR_VALUE_EXISTS,
2576 "TLDAP_TYPE_OR_VALUE_EXISTS" },
2577 { TLDAP_INVALID_SYNTAX,
2578 "TLDAP_INVALID_SYNTAX" },
2579 { TLDAP_NO_SUCH_OBJECT,
2580 "TLDAP_NO_SUCH_OBJECT" },
2581 { TLDAP_ALIAS_PROBLEM,
2582 "TLDAP_ALIAS_PROBLEM" },
2583 { TLDAP_INVALID_DN_SYNTAX,
2584 "TLDAP_INVALID_DN_SYNTAX" },
2585 { TLDAP_IS_LEAF,
2586 "TLDAP_IS_LEAF" },
2587 { TLDAP_ALIAS_DEREF_PROBLEM,
2588 "TLDAP_ALIAS_DEREF_PROBLEM" },
2589 { TLDAP_INAPPROPRIATE_AUTH,
2590 "TLDAP_INAPPROPRIATE_AUTH" },
2591 { TLDAP_INVALID_CREDENTIALS,
2592 "TLDAP_INVALID_CREDENTIALS" },
2593 { TLDAP_INSUFFICIENT_ACCESS,
2594 "TLDAP_INSUFFICIENT_ACCESS" },
2595 { TLDAP_BUSY,
2596 "TLDAP_BUSY" },
2597 { TLDAP_UNAVAILABLE,
2598 "TLDAP_UNAVAILABLE" },
2599 { TLDAP_UNWILLING_TO_PERFORM,
2600 "TLDAP_UNWILLING_TO_PERFORM" },
2601 { TLDAP_LOOP_DETECT,
2602 "TLDAP_LOOP_DETECT" },
2603 { TLDAP_NAMING_VIOLATION,
2604 "TLDAP_NAMING_VIOLATION" },
2605 { TLDAP_OBJECT_CLASS_VIOLATION,
2606 "TLDAP_OBJECT_CLASS_VIOLATION" },
2607 { TLDAP_NOT_ALLOWED_ON_NONLEAF,
2608 "TLDAP_NOT_ALLOWED_ON_NONLEAF" },
2609 { TLDAP_NOT_ALLOWED_ON_RDN,
2610 "TLDAP_NOT_ALLOWED_ON_RDN" },
2611 { TLDAP_ALREADY_EXISTS,
2612 "TLDAP_ALREADY_EXISTS" },
2613 { TLDAP_NO_OBJECT_CLASS_MODS,
2614 "TLDAP_NO_OBJECT_CLASS_MODS" },
2615 { TLDAP_RESULTS_TOO_LARGE,
2616 "TLDAP_RESULTS_TOO_LARGE" },
2617 { TLDAP_AFFECTS_MULTIPLE_DSAS,
2618 "TLDAP_AFFECTS_MULTIPLE_DSAS" },
2619 { TLDAP_OTHER,
2620 "TLDAP_OTHER" },
2621 { TLDAP_SERVER_DOWN,
2622 "TLDAP_SERVER_DOWN" },
2623 { TLDAP_LOCAL_ERROR,
2624 "TLDAP_LOCAL_ERROR" },
2625 { TLDAP_ENCODING_ERROR,
2626 "TLDAP_ENCODING_ERROR" },
2627 { TLDAP_DECODING_ERROR,
2628 "TLDAP_DECODING_ERROR" },
2629 { TLDAP_TIMEOUT,
2630 "TLDAP_TIMEOUT" },
2631 { TLDAP_AUTH_UNKNOWN,
2632 "TLDAP_AUTH_UNKNOWN" },
2633 { TLDAP_FILTER_ERROR,
2634 "TLDAP_FILTER_ERROR" },
2635 { TLDAP_USER_CANCELLED,
2636 "TLDAP_USER_CANCELLED" },
2637 { TLDAP_PARAM_ERROR,
2638 "TLDAP_PARAM_ERROR" },
2639 { TLDAP_NO_MEMORY,
2640 "TLDAP_NO_MEMORY" },
2641 { TLDAP_CONNECT_ERROR,
2642 "TLDAP_CONNECT_ERROR" },
2643 { TLDAP_NOT_SUPPORTED,
2644 "TLDAP_NOT_SUPPORTED" },
2645 { TLDAP_CONTROL_NOT_FOUND,
2646 "TLDAP_CONTROL_NOT_FOUND" },
2647 { TLDAP_NO_RESULTS_RETURNED,
2648 "TLDAP_NO_RESULTS_RETURNED" },
2649 { TLDAP_MORE_RESULTS_TO_RETURN,
2650 "TLDAP_MORE_RESULTS_TO_RETURN" },
2651 { TLDAP_CLIENT_LOOP,
2652 "TLDAP_CLIENT_LOOP" },
2653 { TLDAP_REFERRAL_LIMIT_EXCEEDED,
2654 "TLDAP_REFERRAL_LIMIT_EXCEEDED" },
2655 };
2656
2657 const char *tldap_rc2string(TLDAPRC rc)
2658 {
2659 size_t i;
2660
2661 for (i=0; i<ARRAY_SIZE(tldaprc_errmap); i++) {
2662 if (TLDAP_RC_EQUAL(rc, tldaprc_errmap[i].rc)) {
2663 return tldaprc_errmap[i].string;
2664 }
2665 }
2666
2667 return "Unknown LDAP Error";
2668 }
Samba GIT mirror