?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s4:selftest: explicitly set NSS/RESOLV_WAPPER_* in wait_for_start
[Samba.git] / source3 / lib / tldap.c
blobbfb24ee8661097b39497fb600fcbf240b03c4210
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;
1266 size_t i, p;
1267
1268 for (i = 0,p = 0; i < *val_len; i++) {
1269
1270 switch (value[i]) {
1271 case '(':
1272 case ')':
1273 case '*':
1274 /* these must be escaped */
1275 return false;
1276
1277 case '\\':
1278 if (!value[i + 1]) {
1279 /* invalid EOL */
1280 return false;
1281 }
1282 i++;
1283
1284 /* LDAPv3 escaped */
1285 c = tldap_hex2char(&value[i]);
1286 if (c >= 0 && c < 256) {
1287 value[p] = c;
1288 i++;
1289 p++;
1290 break;
1291 }
1292
1293 /* LDAPv2 escaped */
1294 switch (value[i]) {
1295 case '(':
1296 case ')':
1297 case '*':
1298 case '\\':
1299 value[p] = value[i];
1300 p++;
1301
1302 break;
1303 default:
1304 /* invalid */
1305 return false;
1306 }
1307 break;
1308
1309 default:
1310 value[p] = value[i];
1311 p++;
1312 }
1313 }
1314 value[p] = '\0';
1315 *val_len = p;
1316 return true;
1317 }
1318
1319 static bool tldap_push_filter_basic(struct tldap_context *ld,
1320 struct asn1_data *data,
1321 const char **_s);
1322 static bool tldap_push_filter_substring(struct tldap_context *ld,
1323 struct asn1_data *data,
1324 const char *val,
1325 const char **_s);
1326 static bool tldap_push_filter_int(struct tldap_context *ld,
1327 struct asn1_data *data,
1328 const char **_s)
1329 {
1330 const char *s = *_s;
1331 bool ret;
1332
1333 if (*s != '(') {
1334 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1335 "Incomplete or malformed filter\n");
1336 return false;
1337 }
1338 s++;
1339
1340 /* we are right after a parenthesis,
1341 * find out what op we have at hand */
1342 switch (*s) {
1343 case '&':
1344 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: AND\n");
1345 if (!asn1_push_tag(data, TLDAP_FILTER_AND)) return false;
1346 s++;
1347 break;
1348
1349 case '|':
1350 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: OR\n");
1351 if (!asn1_push_tag(data, TLDAP_FILTER_OR)) return false;
1352 s++;
1353 break;
1354
1355 case '!':
1356 tldap_debug(ld, TLDAP_DEBUG_TRACE, "Filter op: NOT\n");
1357 if (!asn1_push_tag(data, TLDAP_FILTER_NOT)) return false;
1358 s++;
1359 ret = tldap_push_filter_int(ld, data, &s);
1360 if (!ret) {
1361 return false;
1362 }
1363 if (!asn1_pop_tag(data)) return false;
1364 goto done;
1365
1366 case '(':
1367 case ')':
1368 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1369 "Invalid parenthesis '%c'\n", *s);
1370 return false;
1371
1372 case '\0':
1373 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1374 "Invalid filter termination\n");
1375 return false;
1376
1377 default:
1378 ret = tldap_push_filter_basic(ld, data, &s);
1379 if (!ret) {
1380 return false;
1381 }
1382 goto done;
1383 }
1384
1385 /* only and/or filters get here.
1386 * go through the list of filters */
1387
1388 if (*s == ')') {
1389 /* RFC 4526: empty and/or */
1390 if (!asn1_pop_tag(data)) return false;
1391 goto done;
1392 }
1393
1394 while (*s) {
1395 ret = tldap_push_filter_int(ld, data, &s);
1396 if (!ret) {
1397 return false;
1398 }
1399
1400 if (*s == ')') {
1401 /* end of list, return */
1402 if (!asn1_pop_tag(data)) return false;
1403 break;
1404 }
1405 }
1406
1407 done:
1408 if (*s != ')') {
1409 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1410 "Incomplete or malformed filter\n");
1411 return false;
1412 }
1413 s++;
1414
1415 if (asn1_has_error(data)) {
1416 return false;
1417 }
1418
1419 *_s = s;
1420 return true;
1421 }
1422
1423
1424 static bool tldap_push_filter_basic(struct tldap_context *ld,
1425 struct asn1_data *data,
1426 const char **_s)
1427 {
1428 TALLOC_CTX *tmpctx = talloc_tos();
1429 const char *s = *_s;
1430 const char *e;
1431 const char *eq;
1432 const char *val;
1433 const char *type;
1434 const char *dn;
1435 const char *rule;
1436 const char *star;
1437 size_t type_len = 0;
1438 char *uval;
1439 size_t uval_len;
1440 bool write_octect = true;
1441 bool ret;
1442
1443 eq = strchr(s, '=');
1444 if (!eq) {
1445 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1446 "Invalid filter, missing equal sign\n");
1447 return false;
1448 }
1449
1450 val = eq + 1;
1451 e = eq - 1;
1452
1453 switch (*e) {
1454 case '<':
1455 if (!asn1_push_tag(data, TLDAP_FILTER_LE)) return false;
1456 break;
1457
1458 case '>':
1459 if (!asn1_push_tag(data, TLDAP_FILTER_GE)) return false;
1460 break;
1461
1462 case '~':
1463 if (!asn1_push_tag(data, TLDAP_FILTER_APX)) return false;
1464 break;
1465
1466 case ':':
1467 if (!asn1_push_tag(data, TLDAP_FILTER_EXT)) return false;
1468 write_octect = false;
1469
1470 type = NULL;
1471 dn = NULL;
1472 rule = NULL;
1473
1474 if (*s == ':') { /* [:dn]:rule:= value */
1475 if (s == e) {
1476 /* malformed filter */
1477 return false;
1478 }
1479 dn = s;
1480 } else { /* type[:dn][:rule]:= value */
1481 type = s;
1482 dn = strchr(s, ':');
1483 type_len = dn - type;
1484 if (dn == e) { /* type:= value */
1485 dn = NULL;
1486 }
1487 }
1488 if (dn) {
1489 dn++;
1490
1491 rule = strchr(dn, ':');
1492 if (rule == NULL) {
1493 return false;
1494 }
1495 if ((rule == dn + 1) || rule + 1 == e) {
1496 /* malformed filter, contains "::" */
1497 return false;
1498 }
1499
1500 if (strncasecmp_m(dn, "dn:", 3) != 0) {
1501 if (rule == e) {
1502 rule = dn;
1503 dn = NULL;
1504 } else {
1505 /* malformed filter. With two
1506 * optionals, the first must be "dn"
1507 */
1508 return false;
1509 }
1510 } else {
1511 if (rule == e) {
1512 rule = NULL;
1513 } else {
1514 rule++;
1515 }
1516 }
1517 }
1518
1519 if (!type && !dn && !rule) {
1520 /* malformed filter, there must be at least one */
1521 return false;
1522 }
1523
1524 /*
1525 MatchingRuleAssertion ::= SEQUENCE {
1526 matchingRule [1] MatchingRuleID OPTIONAL,
1527 type [2] AttributeDescription OPTIONAL,
1528 matchValue [3] AssertionValue,
1529 dnAttributes [4] BOOLEAN DEFAULT FALSE
1530 }
1531 */
1532
1533 /* check and add rule */
1534 if (rule) {
1535 ret = tldap_is_attrdesc(rule, e - rule, true);
1536 if (!ret) {
1537 return false;
1538 }
1539 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(1))) return false;
1540 if (!asn1_write(data, rule, e - rule)) return false;
1541 if (!asn1_pop_tag(data)) return false;
1542 }
1543
1544 /* check and add type */
1545 if (type) {
1546 ret = tldap_is_attrdesc(type, type_len, false);
1547 if (!ret) {
1548 return false;
1549 }
1550 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(2))) return false;
1551 if (!asn1_write(data, type, type_len)) return false;
1552 if (!asn1_pop_tag(data)) return false;
1553 }
1554
1555 uval = tldap_get_val(tmpctx, val, _s);
1556 if (!uval) {
1557 return false;
1558 }
1559 uval_len = *_s - val;
1560 ret = tldap_unescape_inplace(uval, &uval_len);
1561 if (!ret) {
1562 return false;
1563 }
1564
1565 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(3))) return false;
1566 if (!asn1_write(data, uval, uval_len)) return false;
1567 if (!asn1_pop_tag(data)) return false;
1568
1569 if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(4))) return false;
1570 if (!asn1_write_uint8(data, dn?1:0)) return false;
1571 if (!asn1_pop_tag(data)) return false;
1572 break;
1573
1574 default:
1575 e = eq;
1576
1577 ret = tldap_is_attrdesc(s, e - s, false);
1578 if (!ret) {
1579 return false;
1580 }
1581
1582 if (strncmp(val, "*)", 2) == 0) {
1583 /* presence */
1584 if (!asn1_push_tag(data, TLDAP_FILTER_PRES)) return false;
1585 if (!asn1_write(data, s, e - s)) return false;
1586 *_s = val + 1;
1587 write_octect = false;
1588 break;
1589 }
1590
1591 ret = tldap_find_first_star(val, &star);
1592 if (!ret) {
1593 return false;
1594 }
1595 if (*star == '*') {
1596 /* substring */
1597 if (!asn1_push_tag(data, TLDAP_FILTER_SUB)) return false;
1598 if (!asn1_write_OctetString(data, s, e - s)) return false;
1599 ret = tldap_push_filter_substring(ld, data, val, &s);
1600 if (!ret) {
1601 return false;
1602 }
1603 *_s = s;
1604 write_octect = false;
1605 break;
1606 }
1607
1608 /* if nothing else, then it is just equality */
1609 if (!asn1_push_tag(data, TLDAP_FILTER_EQ)) return false;
1610 write_octect = true;
1611 break;
1612 }
1613
1614 if (write_octect) {
1615 uval = tldap_get_val(tmpctx, val, _s);
1616 if (!uval) {
1617 return false;
1618 }
1619 uval_len = *_s - val;
1620 ret = tldap_unescape_inplace(uval, &uval_len);
1621 if (!ret) {
1622 return false;
1623 }
1624
1625 if (!asn1_write_OctetString(data, s, e - s)) return false;
1626 if (!asn1_write_OctetString(data, uval, uval_len)) return false;
1627 }
1628
1629 if (asn1_has_error(data)) {
1630 return false;
1631 }
1632 return asn1_pop_tag(data);
1633 }
1634
1635 static bool tldap_push_filter_substring(struct tldap_context *ld,
1636 struct asn1_data *data,
1637 const char *val,
1638 const char **_s)
1639 {
1640 TALLOC_CTX *tmpctx = talloc_tos();
1641 bool initial = true;
1642 const char *star;
1643 char *chunk;
1644 size_t chunk_len;
1645 bool ret;
1646
1647 /*
1648 SubstringFilter ::= SEQUENCE {
1649 type AttributeDescription,
1650 -- at least one must be present
1651 substrings SEQUENCE OF CHOICE {
1652 initial [0] LDAPString,
1653 any [1] LDAPString,
1654 final [2] LDAPString } }
1655 */
1656 if (!asn1_push_tag(data, ASN1_SEQUENCE(0))) return false;
1657
1658 do {
1659 ret = tldap_find_first_star(val, &star);
1660 if (!ret) {
1661 return false;
1662 }
1663 chunk_len = star - val;
1664
1665 switch (*star) {
1666 case '*':
1667 if (!initial && chunk_len == 0) {
1668 /* found '**', which is illegal */
1669 return false;
1670 }
1671 break;
1672 case ')':
1673 if (initial) {
1674 /* no stars ?? */
1675 return false;
1676 }
1677 /* we are done */
1678 break;
1679 default:
1680 /* ?? */
1681 return false;
1682 }
1683
1684 if (initial && chunk_len == 0) {
1685 val = star + 1;
1686 initial = false;
1687 continue;
1688 }
1689
1690 chunk = talloc_strndup(tmpctx, val, chunk_len);
1691 if (!chunk) {
1692 return false;
1693 }
1694 ret = tldap_unescape_inplace(chunk, &chunk_len);
1695 if (!ret) {
1696 return false;
1697 }
1698 switch (*star) {
1699 case '*':
1700 if (initial) {
1701 if (!asn1_push_tag(data, TLDAP_SUB_INI)) return false;
1702 initial = false;
1703 } else {
1704 if (!asn1_push_tag(data, TLDAP_SUB_ANY)) return false;
1705 }
1706 break;
1707 case ')':
1708 if (!asn1_push_tag(data, TLDAP_SUB_FIN)) return false;
1709 break;
1710 default:
1711 /* ?? */
1712 return false;
1713 }
1714 if (!asn1_write(data, chunk, chunk_len)) return false;
1715 if (!asn1_pop_tag(data)) return false;
1716
1717 val = star + 1;
1718
1719 } while (*star == '*');
1720
1721 *_s = star;
1722
1723 /* end of sequence */
1724 return asn1_pop_tag(data);
1725 }
1726
1727 /* NOTE: although openldap libraries allow for spaces in some places, mosly
1728 * around parenthesis, we do not allow any spaces (except in values of
1729 * course) as I couldn't fine any place in RFC 4512 or RFC 4515 where
1730 * leading or trailing spaces where allowed.
1731 */
1732 static bool tldap_push_filter(struct tldap_context *ld,
1733 struct asn1_data *data,
1734 const char *filter)
1735 {
1736 const char *s = filter;
1737 bool ret;
1738
1739 ret = tldap_push_filter_int(ld, data, &s);
1740 if (ret && *s) {
1741 tldap_debug(ld, TLDAP_DEBUG_ERROR,
1742 "Incomplete or malformed filter\n");
1743 return false;
1744 }
1745 return ret;
1746 }
1747
1748 /*****************************************************************************/
1749
1750 static void tldap_search_done(struct tevent_req *subreq);
1751
1752 struct tevent_req *tldap_search_send(TALLOC_CTX *mem_ctx,
1753 struct tevent_context *ev,
1754 struct tldap_context *ld,
1755 const char *base, int scope,
1756 const char *filter,
1757 const char **attrs,
1758 int num_attrs,
1759 int attrsonly,
1760 struct tldap_control *sctrls,
1761 int num_sctrls,
1762 struct tldap_control *cctrls,
1763 int num_cctrls,
1764 int timelimit,
1765 int sizelimit,
1766 int deref)
1767 {
1768 struct tevent_req *req, *subreq;
1769 struct tldap_req_state *state;
1770 int i;
1771
1772 req = tldap_req_create(mem_ctx, ld, &state);
1773 if (req == NULL) {
1774 return NULL;
1775 }
1776
1777 if (!asn1_push_tag(state->out, TLDAP_REQ_SEARCH)) goto encoding_error;
1778 if (!asn1_write_OctetString(state->out, base, strlen(base))) goto encoding_error;
1779 if (!asn1_write_enumerated(state->out, scope)) goto encoding_error;
1780 if (!asn1_write_enumerated(state->out, deref)) goto encoding_error;
1781 if (!asn1_write_Integer(state->out, sizelimit)) goto encoding_error;
1782 if (!asn1_write_Integer(state->out, timelimit)) goto encoding_error;
1783 if (!asn1_write_BOOLEAN(state->out, attrsonly)) goto encoding_error;
1784
1785 if (!tldap_push_filter(ld, state->out, filter)) {
1786 goto encoding_error;
1787 }
1788
1789 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto encoding_error;
1790 for (i=0; i<num_attrs; i++) {
1791 if (!asn1_write_OctetString(state->out, attrs[i], strlen(attrs[i]))) goto encoding_error;
1792 }
1793 if (!asn1_pop_tag(state->out)) goto encoding_error;
1794 if (!asn1_pop_tag(state->out)) goto encoding_error;
1795
1796 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
1797 sctrls, num_sctrls);
1798 if (tevent_req_nomem(subreq, req)) {
1799 return tevent_req_post(req, ev);
1800 }
1801 tevent_req_set_callback(subreq, tldap_search_done, req);
1802 return req;
1803
1804 encoding_error:
1805 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
1806 return tevent_req_post(req, ev);
1807 }
1808
1809 static void tldap_search_done(struct tevent_req *subreq)
1810 {
1811 struct tevent_req *req = tevent_req_callback_data(
1812 subreq, struct tevent_req);
1813 struct tldap_req_state *state = tevent_req_data(
1814 req, struct tldap_req_state);
1815 TLDAPRC rc;
1816
1817 rc = tldap_msg_recv(subreq, state, &state->result);
1818 if (tevent_req_ldap_error(req, rc)) {
1819 return;
1820 }
1821 switch (state->result->type) {
1822 case TLDAP_RES_SEARCH_ENTRY:
1823 case TLDAP_RES_SEARCH_REFERENCE:
1824 if (!tldap_msg_set_pending(subreq)) {
1825 tevent_req_oom(req);
1826 return;
1827 }
1828 tevent_req_notify_callback(req);
1829 break;
1830 case TLDAP_RES_SEARCH_RESULT:
1831 TALLOC_FREE(subreq);
1832 if (!asn1_start_tag(state->result->data,
1833 state->result->type) ||
1834 !tldap_decode_response(state) ||
1835 !asn1_end_tag(state->result->data) ||
1836 !tldap_decode_controls(state)) {
1837 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
1838 return;
1839 }
1840 tevent_req_done(req);
1841 break;
1842 default:
1843 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
1844 return;
1845 }
1846 }
1847
1848 TLDAPRC tldap_search_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1849 struct tldap_message **pmsg)
1850 {
1851 struct tldap_req_state *state = tevent_req_data(
1852 req, struct tldap_req_state);
1853 TLDAPRC rc;
1854
1855 if (!tevent_req_is_in_progress(req)
1856 && tevent_req_is_ldap_error(req, &rc)) {
1857 return rc;
1858 }
1859
1860 if (tevent_req_is_in_progress(req)) {
1861 switch (state->result->type) {
1862 case TLDAP_RES_SEARCH_ENTRY:
1863 case TLDAP_RES_SEARCH_REFERENCE:
1864 break;
1865 default:
1866 return TLDAP_OPERATIONS_ERROR;
1867 }
1868 }
1869
1870 *pmsg = talloc_move(mem_ctx, &state->result);
1871 return TLDAP_SUCCESS;
1872 }
1873
1874 struct tldap_search_all_state {
1875 struct tldap_message **msgs;
1876 struct tldap_message *result;
1877 };
1878
1879 static void tldap_search_all_done(struct tevent_req *subreq);
1880
1881 struct tevent_req *tldap_search_all_send(
1882 TALLOC_CTX *mem_ctx, struct tevent_context *ev,
1883 struct tldap_context *ld, const char *base, int scope,
1884 const char *filter, const char **attrs, int num_attrs, int attrsonly,
1885 struct tldap_control *sctrls, int num_sctrls,
1886 struct tldap_control *cctrls, int num_cctrls,
1887 int timelimit, int sizelimit, int deref)
1888 {
1889 struct tevent_req *req, *subreq;
1890 struct tldap_search_all_state *state;
1891
1892 req = tevent_req_create(mem_ctx, &state,
1893 struct tldap_search_all_state);
1894 if (req == NULL) {
1895 return NULL;
1896 }
1897
1898 subreq = tldap_search_send(state, ev, ld, base, scope, filter,
1899 attrs, num_attrs, attrsonly,
1900 sctrls, num_sctrls, cctrls, num_cctrls,
1901 timelimit, sizelimit, deref);
1902 if (tevent_req_nomem(subreq, req)) {
1903 return tevent_req_post(req, ev);
1904 }
1905 tevent_req_set_callback(subreq, tldap_search_all_done, req);
1906 return req;
1907 }
1908
1909 static void tldap_search_all_done(struct tevent_req *subreq)
1910 {
1911 struct tevent_req *req = tevent_req_callback_data(
1912 subreq, struct tevent_req);
1913 struct tldap_search_all_state *state = tevent_req_data(
1914 req, struct tldap_search_all_state);
1915 struct tldap_message *msg, **tmp;
1916 size_t num_msgs;
1917 TLDAPRC rc;
1918 int msgtype;
1919
1920 rc = tldap_search_recv(subreq, state, &msg);
1921 /* No TALLOC_FREE(subreq), this is multi-step */
1922 if (tevent_req_ldap_error(req, rc)) {
1923 TALLOC_FREE(subreq);
1924 return;
1925 }
1926
1927 msgtype = tldap_msg_type(msg);
1928 if (msgtype == TLDAP_RES_SEARCH_RESULT) {
1929 state->result = msg;
1930 tevent_req_done(req);
1931 return;
1932 }
1933
1934 num_msgs = talloc_array_length(state->msgs);
1935
1936 tmp = talloc_realloc(state, state->msgs, struct tldap_message *,
1937 num_msgs + 1);
1938 if (tevent_req_nomem(tmp, req)) {
1939 return;
1940 }
1941 state->msgs = tmp;
1942 state->msgs[num_msgs] = talloc_move(state->msgs, &msg);
1943 }
1944
1945 TLDAPRC tldap_search_all_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1946 struct tldap_message ***msgs,
1947 struct tldap_message **result)
1948 {
1949 struct tldap_search_all_state *state = tevent_req_data(
1950 req, struct tldap_search_all_state);
1951 TLDAPRC rc;
1952
1953 if (tevent_req_is_ldap_error(req, &rc)) {
1954 return rc;
1955 }
1956
1957 if (msgs != NULL) {
1958 *msgs = talloc_move(mem_ctx, &state->msgs);
1959 }
1960 if (result != NULL) {
1961 *result = talloc_move(mem_ctx, &state->result);
1962 }
1963
1964 return TLDAP_SUCCESS;
1965 }
1966
1967 TLDAPRC tldap_search(struct tldap_context *ld,
1968 const char *base, int scope, const char *filter,
1969 const char **attrs, int num_attrs, int attrsonly,
1970 struct tldap_control *sctrls, int num_sctrls,
1971 struct tldap_control *cctrls, int num_cctrls,
1972 int timelimit, int sizelimit, int deref,
1973 TALLOC_CTX *mem_ctx, struct tldap_message ***pmsgs)
1974 {
1975 TALLOC_CTX *frame;
1976 struct tevent_context *ev;
1977 struct tevent_req *req;
1978 TLDAPRC rc = TLDAP_NO_MEMORY;
1979 struct tldap_message **msgs;
1980 struct tldap_message *result;
1981
1982 if (tldap_pending_reqs(ld)) {
1983 return TLDAP_BUSY;
1984 }
1985
1986 frame = talloc_stackframe();
1987
1988 ev = samba_tevent_context_init(frame);
1989 if (ev == NULL) {
1990 goto fail;
1991 }
1992 req = tldap_search_all_send(frame, ev, ld, base, scope, filter,
1993 attrs, num_attrs, attrsonly,
1994 sctrls, num_sctrls, cctrls, num_cctrls,
1995 timelimit, sizelimit, deref);
1996 if (req == NULL) {
1997 goto fail;
1998 }
1999 if (!tevent_req_poll(req, ev)) {
2000 rc = TLDAP_OPERATIONS_ERROR;
2001 goto fail;
2002 }
2003 rc = tldap_search_all_recv(req, frame, &msgs, &result);
2004 TALLOC_FREE(req);
2005 if (!TLDAP_RC_IS_SUCCESS(rc)) {
2006 goto fail;
2007 }
2008
2009 TALLOC_FREE(ld->last_msg);
2010 ld->last_msg = talloc_move(ld, &result);
2011
2012 if (pmsgs != NULL) {
2013 *pmsgs = talloc_move(mem_ctx, &msgs);
2014 }
2015 fail:
2016 TALLOC_FREE(frame);
2017 return rc;
2018 }
2019
2020 static bool tldap_parse_search_entry(struct tldap_message *msg)
2021 {
2022 int num_attribs = 0;
2023
2024 if (msg->type != TLDAP_RES_SEARCH_ENTRY) {
2025 return false;
2026 }
2027 if (!asn1_start_tag(msg->data, TLDAP_RES_SEARCH_ENTRY)) {
2028 return false;
2029 }
2030
2031 /* dn */
2032
2033 if (!asn1_read_OctetString_talloc(msg, msg->data, &msg->dn)) return false;
2034
2035 if (msg->dn == NULL) {
2036 return false;
2037 }
2038
2039 /*
2040 * Attributes: We overallocate msg->attribs by one, so that while
2041 * looping over the attributes we can directly parse into the last
2042 * array element. Same for the values in the inner loop.
2043 */
2044
2045 msg->attribs = talloc_array(msg, struct tldap_attribute, 1);
2046 if (msg->attribs == NULL) {
2047 return false;
2048 }
2049
2050 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2051 while (asn1_peek_tag(msg->data, ASN1_SEQUENCE(0))) {
2052 struct tldap_attribute *attrib;
2053 int num_values = 0;
2054
2055 attrib = &msg->attribs[num_attribs];
2056 attrib->values = talloc_array(msg->attribs, DATA_BLOB, 1);
2057 if (attrib->values == NULL) {
2058 return false;
2059 }
2060 if (!asn1_start_tag(msg->data, ASN1_SEQUENCE(0))) return false;
2061 if (!asn1_read_OctetString_talloc(msg->attribs, msg->data,
2062 &attrib->name)) return false;
2063 if (!asn1_start_tag(msg->data, ASN1_SET)) return false;
2064
2065 while (asn1_peek_tag(msg->data, ASN1_OCTET_STRING)) {
2066 if (!asn1_read_OctetString(msg->data, msg,
2067 &attrib->values[num_values])) return false;
2068
2069 attrib->values = talloc_realloc(
2070 msg->attribs, attrib->values, DATA_BLOB,
2071 num_values + 2);
2072 if (attrib->values == NULL) {
2073 return false;
2074 }
2075 num_values += 1;
2076 }
2077 attrib->values = talloc_realloc(msg->attribs, attrib->values,
2078 DATA_BLOB, num_values);
2079 attrib->num_values = num_values;
2080
2081 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SET */
2082 if (!asn1_end_tag(msg->data)) return false; /* ASN1_SEQUENCE(0) */
2083 msg->attribs = talloc_realloc(
2084 msg, msg->attribs, struct tldap_attribute,
2085 num_attribs + 2);
2086 if (msg->attribs == NULL) {
2087 return false;
2088 }
2089 num_attribs += 1;
2090 }
2091 msg->attribs = talloc_realloc(
2092 msg, msg->attribs, struct tldap_attribute, num_attribs);
2093 return asn1_end_tag(msg->data);
2094 }
2095
2096 bool tldap_entry_dn(struct tldap_message *msg, char **dn)
2097 {
2098 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2099 return false;
2100 }
2101 *dn = msg->dn;
2102 return true;
2103 }
2104
2105 bool tldap_entry_attributes(struct tldap_message *msg,
2106 struct tldap_attribute **attributes,
2107 int *num_attributes)
2108 {
2109 if ((msg->dn == NULL) && (!tldap_parse_search_entry(msg))) {
2110 return false;
2111 }
2112 *attributes = msg->attribs;
2113 *num_attributes = talloc_array_length(msg->attribs);
2114 return true;
2115 }
2116
2117 static bool tldap_decode_controls(struct tldap_req_state *state)
2118 {
2119 struct tldap_message *msg = state->result;
2120 struct asn1_data *data = msg->data;
2121 struct tldap_control *sctrls = NULL;
2122 int num_controls = 0;
2123 bool ret = false;
2124
2125 msg->res_sctrls = NULL;
2126
2127 if (!asn1_peek_tag(data, ASN1_CONTEXT(0))) {
2128 return true;
2129 }
2130
2131 if (!asn1_start_tag(data, ASN1_CONTEXT(0))) goto out;
2132
2133 while (asn1_peek_tag(data, ASN1_SEQUENCE(0))) {
2134 struct tldap_control *c;
2135 char *oid = NULL;
2136
2137 sctrls = talloc_realloc(msg, sctrls, struct tldap_control,
2138 num_controls + 1);
2139 if (sctrls == NULL) {
2140 goto out;
2141 }
2142 c = &sctrls[num_controls];
2143
2144 if (!asn1_start_tag(data, ASN1_SEQUENCE(0))) goto out;
2145 if (!asn1_read_OctetString_talloc(msg, data, &oid)) goto out;
2146 if (asn1_has_error(data) || (oid == NULL)) {
2147 goto out;
2148 }
2149 c->oid = oid;
2150 if (asn1_peek_tag(data, ASN1_BOOLEAN)) {
2151 if (!asn1_read_BOOLEAN(data, &c->critical)) goto out;
2152 } else {
2153 c->critical = false;
2154 }
2155 c->value = data_blob_null;
2156 if (asn1_peek_tag(data, ASN1_OCTET_STRING) &&
2157 !asn1_read_OctetString(data, msg, &c->value)) {
2158 goto out;
2159 }
2160 if (!asn1_end_tag(data)) goto out; /* ASN1_SEQUENCE(0) */
2161
2162 num_controls += 1;
2163 }
2164
2165 if (!asn1_end_tag(data)) goto out; /* ASN1_CONTEXT(0) */
2166
2167 ret = true;
2168
2169 out:
2170
2171 if (ret) {
2172 msg->res_sctrls = sctrls;
2173 } else {
2174 TALLOC_FREE(sctrls);
2175 }
2176 return ret;
2177 }
2178
2179 static void tldap_simple_done(struct tevent_req *subreq, int type)
2180 {
2181 struct tevent_req *req = tevent_req_callback_data(
2182 subreq, struct tevent_req);
2183 struct tldap_req_state *state = tevent_req_data(
2184 req, struct tldap_req_state);
2185 TLDAPRC rc;
2186
2187 rc = tldap_msg_recv(subreq, state, &state->result);
2188 TALLOC_FREE(subreq);
2189 if (tevent_req_ldap_error(req, rc)) {
2190 return;
2191 }
2192 if (state->result->type != type) {
2193 tevent_req_ldap_error(req, TLDAP_PROTOCOL_ERROR);
2194 return;
2195 }
2196 if (!asn1_start_tag(state->result->data, state->result->type) ||
2197 !tldap_decode_response(state) ||
2198 !asn1_end_tag(state->result->data) ||
2199 !tldap_decode_controls(state)) {
2200 tevent_req_ldap_error(req, TLDAP_DECODING_ERROR);
2201 return;
2202 }
2203 if (!TLDAP_RC_IS_SUCCESS(state->result->lderr)) {
2204 tevent_req_ldap_error(req, state->result->lderr);
2205 return;
2206 }
2207 tevent_req_done(req);
2208 }
2209
2210 static TLDAPRC tldap_simple_recv(struct tevent_req *req)
2211 {
2212 TLDAPRC rc;
2213 if (tevent_req_is_ldap_error(req, &rc)) {
2214 return rc;
2215 }
2216 return TLDAP_SUCCESS;
2217 }
2218
2219 static void tldap_add_done(struct tevent_req *subreq);
2220
2221 struct tevent_req *tldap_add_send(TALLOC_CTX *mem_ctx,
2222 struct tevent_context *ev,
2223 struct tldap_context *ld,
2224 const char *dn,
2225 struct tldap_mod *attributes,
2226 int num_attributes,
2227 struct tldap_control *sctrls,
2228 int num_sctrls,
2229 struct tldap_control *cctrls,
2230 int num_cctrls)
2231 {
2232 struct tevent_req *req, *subreq;
2233 struct tldap_req_state *state;
2234 int i, j;
2235
2236 req = tldap_req_create(mem_ctx, ld, &state);
2237 if (req == NULL) {
2238 return NULL;
2239 }
2240
2241 if (!asn1_push_tag(state->out, TLDAP_REQ_ADD)) goto err;
2242 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2243 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2244
2245 for (i=0; i<num_attributes; i++) {
2246 struct tldap_mod *attrib = &attributes[i];
2247 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2248 if (!asn1_write_OctetString(state->out, attrib->attribute,
2249 strlen(attrib->attribute))) goto err;
2250 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2251 for (j=0; j<attrib->num_values; j++) {
2252 if (!asn1_write_OctetString(state->out,
2253 attrib->values[j].data,
2254 attrib->values[j].length)) goto err;
2255 }
2256 if (!asn1_pop_tag(state->out)) goto err;
2257 if (!asn1_pop_tag(state->out)) goto err;
2258 }
2259
2260 if (!asn1_pop_tag(state->out)) goto err;
2261 if (!asn1_pop_tag(state->out)) goto err;
2262
2263 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2264 sctrls, num_sctrls);
2265 if (tevent_req_nomem(subreq, req)) {
2266 return tevent_req_post(req, ev);
2267 }
2268 tevent_req_set_callback(subreq, tldap_add_done, req);
2269 return req;
2270
2271 err:
2272
2273 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2274 return tevent_req_post(req, ev);
2275 }
2276
2277 static void tldap_add_done(struct tevent_req *subreq)
2278 {
2279 tldap_simple_done(subreq, TLDAP_RES_ADD);
2280 }
2281
2282 TLDAPRC tldap_add_recv(struct tevent_req *req)
2283 {
2284 return tldap_simple_recv(req);
2285 }
2286
2287 TLDAPRC tldap_add(struct tldap_context *ld, const char *dn,
2288 struct tldap_mod *attributes, int num_attributes,
2289 struct tldap_control *sctrls, int num_sctrls,
2290 struct tldap_control *cctrls, int num_cctrls)
2291 {
2292 TALLOC_CTX *frame = talloc_stackframe();
2293 struct tevent_context *ev;
2294 struct tevent_req *req;
2295 TLDAPRC rc = TLDAP_NO_MEMORY;
2296
2297 ev = samba_tevent_context_init(frame);
2298 if (ev == NULL) {
2299 goto fail;
2300 }
2301 req = tldap_add_send(frame, ev, ld, dn, attributes, num_attributes,
2302 sctrls, num_sctrls, cctrls, num_cctrls);
2303 if (req == NULL) {
2304 goto fail;
2305 }
2306 if (!tevent_req_poll(req, ev)) {
2307 rc = TLDAP_OPERATIONS_ERROR;
2308 goto fail;
2309 }
2310 rc = tldap_add_recv(req);
2311 tldap_save_msg(ld, req);
2312 fail:
2313 TALLOC_FREE(frame);
2314 return rc;
2315 }
2316
2317 static void tldap_modify_done(struct tevent_req *subreq);
2318
2319 struct tevent_req *tldap_modify_send(TALLOC_CTX *mem_ctx,
2320 struct tevent_context *ev,
2321 struct tldap_context *ld,
2322 const char *dn,
2323 struct tldap_mod *mods, int num_mods,
2324 struct tldap_control *sctrls,
2325 int num_sctrls,
2326 struct tldap_control *cctrls,
2327 int num_cctrls)
2328 {
2329 struct tevent_req *req, *subreq;
2330 struct tldap_req_state *state;
2331 int i, j;
2332
2333 req = tldap_req_create(mem_ctx, ld, &state);
2334 if (req == NULL) {
2335 return NULL;
2336 }
2337
2338 if (!asn1_push_tag(state->out, TLDAP_REQ_MODIFY)) goto err;
2339 if (!asn1_write_OctetString(state->out, dn, strlen(dn))) goto err;
2340 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2341
2342 for (i=0; i<num_mods; i++) {
2343 struct tldap_mod *mod = &mods[i];
2344 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2345 if (!asn1_write_enumerated(state->out, mod->mod_op)) goto err;
2346 if (!asn1_push_tag(state->out, ASN1_SEQUENCE(0))) goto err;
2347 if (!asn1_write_OctetString(state->out, mod->attribute,
2348 strlen(mod->attribute))) goto err;
2349 if (!asn1_push_tag(state->out, ASN1_SET)) goto err;
2350 for (j=0; j<mod->num_values; j++) {
2351 if (!asn1_write_OctetString(state->out,
2352 mod->values[j].data,
2353 mod->values[j].length)) goto err;
2354 }
2355 if (!asn1_pop_tag(state->out)) goto err;
2356 if (!asn1_pop_tag(state->out)) goto err;
2357 if (!asn1_pop_tag(state->out)) goto err;
2358 }
2359
2360 if (!asn1_pop_tag(state->out)) goto err;
2361 if (!asn1_pop_tag(state->out)) goto err;
2362
2363 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2364 sctrls, num_sctrls);
2365 if (tevent_req_nomem(subreq, req)) {
2366 return tevent_req_post(req, ev);
2367 }
2368 tevent_req_set_callback(subreq, tldap_modify_done, req);
2369 return req;
2370
2371 err:
2372
2373 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2374 return tevent_req_post(req, ev);
2375 }
2376
2377 static void tldap_modify_done(struct tevent_req *subreq)
2378 {
2379 tldap_simple_done(subreq, TLDAP_RES_MODIFY);
2380 }
2381
2382 TLDAPRC tldap_modify_recv(struct tevent_req *req)
2383 {
2384 return tldap_simple_recv(req);
2385 }
2386
2387 TLDAPRC tldap_modify(struct tldap_context *ld, const char *dn,
2388 struct tldap_mod *mods, int num_mods,
2389 struct tldap_control *sctrls, int num_sctrls,
2390 struct tldap_control *cctrls, int num_cctrls)
2391 {
2392 TALLOC_CTX *frame = talloc_stackframe();
2393 struct tevent_context *ev;
2394 struct tevent_req *req;
2395 TLDAPRC rc = TLDAP_NO_MEMORY;
2396
2397 ev = samba_tevent_context_init(frame);
2398 if (ev == NULL) {
2399 goto fail;
2400 }
2401 req = tldap_modify_send(frame, ev, ld, dn, mods, num_mods,
2402 sctrls, num_sctrls, cctrls, num_cctrls);
2403 if (req == NULL) {
2404 goto fail;
2405 }
2406 if (!tevent_req_poll(req, ev)) {
2407 rc = TLDAP_OPERATIONS_ERROR;
2408 goto fail;
2409 }
2410 rc = tldap_modify_recv(req);
2411 tldap_save_msg(ld, req);
2412 fail:
2413 TALLOC_FREE(frame);
2414 return rc;
2415 }
2416
2417 static void tldap_delete_done(struct tevent_req *subreq);
2418
2419 struct tevent_req *tldap_delete_send(TALLOC_CTX *mem_ctx,
2420 struct tevent_context *ev,
2421 struct tldap_context *ld,
2422 const char *dn,
2423 struct tldap_control *sctrls,
2424 int num_sctrls,
2425 struct tldap_control *cctrls,
2426 int num_cctrls)
2427 {
2428 struct tevent_req *req, *subreq;
2429 struct tldap_req_state *state;
2430
2431 req = tldap_req_create(mem_ctx, ld, &state);
2432 if (req == NULL) {
2433 return NULL;
2434 }
2435
2436 if (!asn1_push_tag(state->out, TLDAP_REQ_DELETE)) goto err;
2437 if (!asn1_write(state->out, dn, strlen(dn))) goto err;
2438 if (!asn1_pop_tag(state->out)) goto err;
2439
2440 subreq = tldap_msg_send(state, ev, ld, state->id, state->out,
2441 sctrls, num_sctrls);
2442 if (tevent_req_nomem(subreq, req)) {
2443 return tevent_req_post(req, ev);
2444 }
2445 tevent_req_set_callback(subreq, tldap_delete_done, req);
2446 return req;
2447
2448 err:
2449
2450 tevent_req_ldap_error(req, TLDAP_ENCODING_ERROR);
2451 return tevent_req_post(req, ev);
2452 }
2453
2454 static void tldap_delete_done(struct tevent_req *subreq)
2455 {
2456 tldap_simple_done(subreq, TLDAP_RES_DELETE);
2457 }
2458
2459 TLDAPRC tldap_delete_recv(struct tevent_req *req)
2460 {
2461 return tldap_simple_recv(req);
2462 }
2463
2464 TLDAPRC tldap_delete(struct tldap_context *ld, const char *dn,
2465 struct tldap_control *sctrls, int num_sctrls,
2466 struct tldap_control *cctrls, int num_cctrls)
2467 {
2468 TALLOC_CTX *frame = talloc_stackframe();
2469 struct tevent_context *ev;
2470 struct tevent_req *req;
2471 TLDAPRC rc = TLDAP_NO_MEMORY;
2472
2473 ev = samba_tevent_context_init(frame);
2474 if (ev == NULL) {
2475 goto fail;
2476 }
2477 req = tldap_delete_send(frame, ev, ld, dn, sctrls, num_sctrls,
2478 cctrls, num_cctrls);
2479 if (req == NULL) {
2480 goto fail;
2481 }
2482 if (!tevent_req_poll(req, ev)) {
2483 rc = TLDAP_OPERATIONS_ERROR;
2484 goto fail;
2485 }
2486 rc = tldap_delete_recv(req);
2487 tldap_save_msg(ld, req);
2488 fail:
2489 TALLOC_FREE(frame);
2490 return rc;
2491 }
2492
2493 int tldap_msg_id(const struct tldap_message *msg)
2494 {
2495 return msg->id;
2496 }
2497
2498 int tldap_msg_type(const struct tldap_message *msg)
2499 {
2500 return msg->type;
2501 }
2502
2503 const char *tldap_msg_matcheddn(struct tldap_message *msg)
2504 {
2505 if (msg == NULL) {
2506 return NULL;
2507 }
2508 return msg->res_matcheddn;
2509 }
2510
2511 const char *tldap_msg_diagnosticmessage(struct tldap_message *msg)
2512 {
2513 if (msg == NULL) {
2514 return NULL;
2515 }
2516 return msg->res_diagnosticmessage;
2517 }
2518
2519 const char *tldap_msg_referral(struct tldap_message *msg)
2520 {
2521 if (msg == NULL) {
2522 return NULL;
2523 }
2524 return msg->res_referral;
2525 }
2526
2527 void tldap_msg_sctrls(struct tldap_message *msg, int *num_sctrls,
2528 struct tldap_control **sctrls)
2529 {
2530 if (msg == NULL) {
2531 *sctrls = NULL;
2532 *num_sctrls = 0;
2533 return;
2534 }
2535 *sctrls = msg->res_sctrls;
2536 *num_sctrls = talloc_array_length(msg->res_sctrls);
2537 }
2538
2539 struct tldap_message *tldap_ctx_lastmsg(struct tldap_context *ld)
2540 {
2541 return ld->last_msg;
2542 }
2543
2544 static const struct { TLDAPRC rc; const char *string; } tldaprc_errmap[] =
2545 {
2546 { TLDAP_SUCCESS,
2547 "TLDAP_SUCCESS" },
2548 { TLDAP_OPERATIONS_ERROR,
2549 "TLDAP_OPERATIONS_ERROR" },
2550 { TLDAP_PROTOCOL_ERROR,
2551 "TLDAP_PROTOCOL_ERROR" },
2552 { TLDAP_TIMELIMIT_EXCEEDED,
2553 "TLDAP_TIMELIMIT_EXCEEDED" },
2554 { TLDAP_SIZELIMIT_EXCEEDED,
2555 "TLDAP_SIZELIMIT_EXCEEDED" },
2556 { TLDAP_COMPARE_FALSE,
2557 "TLDAP_COMPARE_FALSE" },
2558 { TLDAP_COMPARE_TRUE,
2559 "TLDAP_COMPARE_TRUE" },
2560 { TLDAP_STRONG_AUTH_NOT_SUPPORTED,
2561 "TLDAP_STRONG_AUTH_NOT_SUPPORTED" },
2562 { TLDAP_STRONG_AUTH_REQUIRED,
2563 "TLDAP_STRONG_AUTH_REQUIRED" },
2564 { TLDAP_REFERRAL,
2565 "TLDAP_REFERRAL" },
2566 { TLDAP_ADMINLIMIT_EXCEEDED,
2567 "TLDAP_ADMINLIMIT_EXCEEDED" },
2568 { TLDAP_UNAVAILABLE_CRITICAL_EXTENSION,
2569 "TLDAP_UNAVAILABLE_CRITICAL_EXTENSION" },
2570 { TLDAP_CONFIDENTIALITY_REQUIRED,
2571 "TLDAP_CONFIDENTIALITY_REQUIRED" },
2572 { TLDAP_SASL_BIND_IN_PROGRESS,
2573 "TLDAP_SASL_BIND_IN_PROGRESS" },
2574 { TLDAP_NO_SUCH_ATTRIBUTE,
2575 "TLDAP_NO_SUCH_ATTRIBUTE" },
2576 { TLDAP_UNDEFINED_TYPE,
2577 "TLDAP_UNDEFINED_TYPE" },
2578 { TLDAP_INAPPROPRIATE_MATCHING,
2579 "TLDAP_INAPPROPRIATE_MATCHING" },
2580 { TLDAP_CONSTRAINT_VIOLATION,
2581 "TLDAP_CONSTRAINT_VIOLATION" },
2582 { TLDAP_TYPE_OR_VALUE_EXISTS,
2583 "TLDAP_TYPE_OR_VALUE_EXISTS" },
2584 { TLDAP_INVALID_SYNTAX,
2585 "TLDAP_INVALID_SYNTAX" },
2586 { TLDAP_NO_SUCH_OBJECT,
2587 "TLDAP_NO_SUCH_OBJECT" },
2588 { TLDAP_ALIAS_PROBLEM,
2589 "TLDAP_ALIAS_PROBLEM" },
2590 { TLDAP_INVALID_DN_SYNTAX,
2591 "TLDAP_INVALID_DN_SYNTAX" },
2592 { TLDAP_IS_LEAF,
2593 "TLDAP_IS_LEAF" },
2594 { TLDAP_ALIAS_DEREF_PROBLEM,
2595 "TLDAP_ALIAS_DEREF_PROBLEM" },
2596 { TLDAP_INAPPROPRIATE_AUTH,
2597 "TLDAP_INAPPROPRIATE_AUTH" },
2598 { TLDAP_INVALID_CREDENTIALS,
2599 "TLDAP_INVALID_CREDENTIALS" },
2600 { TLDAP_INSUFFICIENT_ACCESS,
2601 "TLDAP_INSUFFICIENT_ACCESS" },
2602 { TLDAP_BUSY,
2603 "TLDAP_BUSY" },
2604 { TLDAP_UNAVAILABLE,
2605 "TLDAP_UNAVAILABLE" },
2606 { TLDAP_UNWILLING_TO_PERFORM,
2607 "TLDAP_UNWILLING_TO_PERFORM" },
2608 { TLDAP_LOOP_DETECT,
2609 "TLDAP_LOOP_DETECT" },
2610 { TLDAP_NAMING_VIOLATION,
2611 "TLDAP_NAMING_VIOLATION" },
2612 { TLDAP_OBJECT_CLASS_VIOLATION,
2613 "TLDAP_OBJECT_CLASS_VIOLATION" },
2614 { TLDAP_NOT_ALLOWED_ON_NONLEAF,
2615 "TLDAP_NOT_ALLOWED_ON_NONLEAF" },
2616 { TLDAP_NOT_ALLOWED_ON_RDN,
2617 "TLDAP_NOT_ALLOWED_ON_RDN" },
2618 { TLDAP_ALREADY_EXISTS,
2619 "TLDAP_ALREADY_EXISTS" },
2620 { TLDAP_NO_OBJECT_CLASS_MODS,
2621 "TLDAP_NO_OBJECT_CLASS_MODS" },
2622 { TLDAP_RESULTS_TOO_LARGE,
2623 "TLDAP_RESULTS_TOO_LARGE" },
2624 { TLDAP_AFFECTS_MULTIPLE_DSAS,
2625 "TLDAP_AFFECTS_MULTIPLE_DSAS" },
2626 { TLDAP_OTHER,
2627 "TLDAP_OTHER" },
2628 { TLDAP_SERVER_DOWN,
2629 "TLDAP_SERVER_DOWN" },
2630 { TLDAP_LOCAL_ERROR,
2631 "TLDAP_LOCAL_ERROR" },
2632 { TLDAP_ENCODING_ERROR,
2633 "TLDAP_ENCODING_ERROR" },
2634 { TLDAP_DECODING_ERROR,
2635 "TLDAP_DECODING_ERROR" },
2636 { TLDAP_TIMEOUT,
2637 "TLDAP_TIMEOUT" },
2638 { TLDAP_AUTH_UNKNOWN,
2639 "TLDAP_AUTH_UNKNOWN" },
2640 { TLDAP_FILTER_ERROR,
2641 "TLDAP_FILTER_ERROR" },
2642 { TLDAP_USER_CANCELLED,
2643 "TLDAP_USER_CANCELLED" },
2644 { TLDAP_PARAM_ERROR,
2645 "TLDAP_PARAM_ERROR" },
2646 { TLDAP_NO_MEMORY,
2647 "TLDAP_NO_MEMORY" },
2648 { TLDAP_CONNECT_ERROR,
2649 "TLDAP_CONNECT_ERROR" },
2650 { TLDAP_NOT_SUPPORTED,
2651 "TLDAP_NOT_SUPPORTED" },
2652 { TLDAP_CONTROL_NOT_FOUND,
2653 "TLDAP_CONTROL_NOT_FOUND" },
2654 { TLDAP_NO_RESULTS_RETURNED,
2655 "TLDAP_NO_RESULTS_RETURNED" },
2656 { TLDAP_MORE_RESULTS_TO_RETURN,
2657 "TLDAP_MORE_RESULTS_TO_RETURN" },
2658 { TLDAP_CLIENT_LOOP,
2659 "TLDAP_CLIENT_LOOP" },
2660 { TLDAP_REFERRAL_LIMIT_EXCEEDED,
2661 "TLDAP_REFERRAL_LIMIT_EXCEEDED" },
2662 };
2663
2664 const char *tldap_rc2string(TLDAPRC rc)
2665 {
2666 size_t i;
2667
2668 for (i=0; i<ARRAY_SIZE(tldaprc_errmap); i++) {
2669 if (TLDAP_RC_EQUAL(rc, tldaprc_errmap[i].rc)) {
2670 return tldaprc_errmap[i].string;
2671 }
2672 }
2673
2674 return "Unknown LDAP Error";
2675 }
Samba GIT mirror