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