2 Unix SMB/CIFS implementation.
4 Copyright (C) Volker Lendecke 2004
5 Copyright (C) Stefan Metzmacher 2004
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 close the socket and shutdown a server_context
27 static void ldapsrv_terminate_connection(struct ldapsrv_connection *ldap_conn, const char *reason)
29 server_terminate_connection(ldap_conn->connection, reason);
33 add a socket address to the list of events, one event per port
35 static void add_socket(struct server_service *service,
36 const struct model_ops *model_ops,
39 struct server_socket *srv_sock;
41 char *ip_str = talloc_strdup(service, inet_ntoa(*ifip));
43 srv_sock = service_setup_socket(service, model_ops, ip_str, &port);
48 /****************************************************************************
49 Open the socket communication.
50 ****************************************************************************/
51 static void ldapsrv_init(struct server_service *service,
52 const struct model_ops *model_ops)
54 struct ldapsrv_service *ldap_service;
55 struct ldapsrv_partition *part;
57 DEBUG(10,("ldapsrv_init\n"));
59 ldap_service = talloc_p(service, struct ldapsrv_service);
61 DEBUG(0,("talloc_p(service, struct ldapsrv_service) failed\n"));
64 ZERO_STRUCTP(ldap_service);
66 part = talloc_p(ldap_service, struct ldapsrv_partition);
68 DEBUG(0,("talloc_p(ldap_service, struct ldapsrv_partition) failed\n"));
71 part->base_dn = ""; /* RootDSE */
72 part->ops = ldapsrv_get_rootdse_partition_ops();
74 ldap_service->rootDSE = part;
75 DLIST_ADD_END(ldap_service->partitions, part, struct ldapsrv_partition *);
77 part = talloc_p(ldap_service, struct ldapsrv_partition);
79 DEBUG(0,("talloc_p(ldap_service, struct ldapsrv_partition) failed\n"));
82 part->base_dn = "*"; /* default partition */
83 part->ops = ldapsrv_get_sldb_partition_ops();
85 ldap_service->default_partition = part;
86 DLIST_ADD_END(ldap_service->partitions, part, struct ldapsrv_partition *);
88 service->private_data = ldap_service;
90 if (lp_interfaces() && lp_bind_interfaces_only()) {
91 int num_interfaces = iface_count();
94 /* We have been given an interfaces line, and been
95 told to only bind to those interfaces. Create a
96 socket per interface and bind to only these.
98 for(i = 0; i < num_interfaces; i++) {
99 struct in_addr *ifip = iface_n_ip(i);
102 DEBUG(0,("ldapsrv_init: interface %d has NULL "
103 "IP address !\n", i));
107 add_socket(service, model_ops, ifip);
110 struct in_addr *ifip;
112 /* Just bind to lp_socket_address() (usually 0.0.0.0) */
113 ifip = interpret_addr2(service, lp_socket_address());
114 add_socket(service, model_ops, ifip);
116 talloc_destroy(ifip);
120 /* This rw-buf api is made to avoid memcpy. For now do that like mad... The
121 idea is to write into a circular list of buffers where the ideal case is
122 that a read(2) holds a complete request that is then thrown away
125 static void consumed_from_buf(struct rw_buffer *buf,
128 memcpy(buf->data, buf->data+length, buf->length-length);
129 buf->length -= length;
132 static BOOL append_to_buf(struct rw_buffer *buf, uint8_t *data, size_t length)
134 buf->data = realloc(buf->data, buf->length+length);
136 if (buf->data == NULL)
139 memcpy(buf->data+buf->length, data, length);
141 buf->length += length;
145 static BOOL read_into_buf(struct socket_context *sock, struct rw_buffer *buf)
151 status = socket_recv(sock, sock, &tmp_blob, 1024, 0);
152 if (!NT_STATUS_IS_OK(status)) {
153 DEBUG(10,("socket_recv: %s\n",nt_errstr(status)));
157 ret = append_to_buf(buf, tmp_blob.data, tmp_blob.length);
159 talloc_free(tmp_blob.data);
164 static BOOL write_from_buf(struct socket_context *sock, struct rw_buffer *buf)
170 tmp_blob.data = buf->data;
171 tmp_blob.length = buf->length;
173 status = socket_send(sock, sock, &tmp_blob, &sendlen, 0);
174 if (!NT_STATUS_IS_OK(status)) {
175 DEBUG(10,("socket_send() %s\n",nt_errstr(status)));
179 consumed_from_buf(buf, sendlen);
184 static void peek_into_read_buf(struct rw_buffer *buf, uint8_t **out,
188 *out_length = buf->length;
191 static BOOL ldap_append_to_buf(struct ldap_message *msg, struct rw_buffer *buf)
196 if (!ldap_encode(msg, &blob))
199 res = append_to_buf(buf, blob.data, blob.length);
201 data_blob_free(&blob);
205 struct ldapsrv_reply *ldapsrv_init_reply(struct ldapsrv_call *call, enum ldap_request_tag type)
207 struct ldapsrv_reply *reply;
209 reply = talloc_p(call, struct ldapsrv_reply);
214 reply->prev = reply->next = NULL;
215 reply->state = LDAPSRV_REPLY_STATE_NEW;
216 reply->msg.messageid = call->request.messageid;
217 reply->msg.type = type;
218 reply->msg.mem_ctx = reply;
223 NTSTATUS ldapsrv_queue_reply(struct ldapsrv_call *call, struct ldapsrv_reply *reply)
225 DLIST_ADD_END(call->replies, reply, struct ldapsrv_reply *);
229 struct ldapsrv_partition *ldapsrv_get_partition(struct ldapsrv_connection *conn, const char *dn)
231 if (strcasecmp("", dn) == 0) {
232 return conn->service->rootDSE;
235 return conn->service->default_partition;
238 NTSTATUS ldapsrv_unwilling(struct ldapsrv_call *call, int error)
240 struct ldapsrv_reply *reply;
241 struct ldap_ExtendedResponse *r;
243 DEBUG(10,("Unwilling type[%d] id[%d]\n", call->request.type, call->request.messageid));
245 reply = ldapsrv_init_reply(call, LDAP_TAG_ExtendedResponse);
247 return NT_STATUS_NO_MEMORY;
250 r = &reply->msg.r.ExtendedResponse;
251 r->response.resultcode = error;
252 r->response.dn = NULL;
253 r->response.errormessage = NULL;
254 r->response.referral = NULL;
256 r->value.data = NULL;
259 return ldapsrv_queue_reply(call, reply);
262 static NTSTATUS ldapsrv_BindRequest(struct ldapsrv_call *call)
264 struct ldap_BindRequest *req = &call->request.r.BindRequest;
265 struct ldapsrv_reply *reply;
266 struct ldap_BindResponse *resp;
268 DEBUG(10, ("BindRequest dn: %s\n",req->dn));
270 reply = ldapsrv_init_reply(call, LDAP_TAG_BindResponse);
272 return NT_STATUS_NO_MEMORY;
275 resp = &reply->msg.r.BindResponse;
276 resp->response.resultcode = 0;
277 resp->response.dn = NULL;
278 resp->response.errormessage = NULL;
279 resp->response.referral = NULL;
280 resp->SASL.secblob = data_blob(NULL, 0);
282 return ldapsrv_queue_reply(call, reply);
285 static NTSTATUS ldapsrv_UnbindRequest(struct ldapsrv_call *call)
287 /* struct ldap_UnbindRequest *req = &call->request->r.UnbindRequest;*/
288 DEBUG(10, ("UnbindRequest\n"));
292 static NTSTATUS ldapsrv_SearchRequest(struct ldapsrv_call *call)
294 struct ldap_SearchRequest *req = &call->request.r.SearchRequest;
295 struct ldapsrv_partition *part;
297 DEBUG(10, ("SearchRequest"));
298 DEBUGADD(10, (" basedn: %s", req->basedn));
299 DEBUGADD(10, (" filter: %s\n", req->filter));
301 part = ldapsrv_get_partition(call->conn, req->basedn);
303 if (!part->ops->Search) {
304 struct ldap_Result *done;
305 struct ldapsrv_reply *done_r;
307 done_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultDone);
309 return NT_STATUS_NO_MEMORY;
312 done = &done_r->msg.r.SearchResultDone;
313 done->resultcode = 53;
315 done->errormessage = NULL;
316 done->referral = NULL;
318 return ldapsrv_queue_reply(call, done_r);
321 return part->ops->Search(part, call, req);
324 static NTSTATUS ldapsrv_ModifyRequest(struct ldapsrv_call *call)
326 struct ldap_ModifyRequest *req = &call->request.r.ModifyRequest;
327 struct ldapsrv_partition *part;
329 DEBUG(10, ("ModifyRequest"));
330 DEBUGADD(10, (" dn: %s", req->dn));
332 part = ldapsrv_get_partition(call->conn, req->dn);
334 if (!part->ops->Modify) {
335 return ldapsrv_unwilling(call, 53);
338 return part->ops->Modify(part, call, req);
341 static NTSTATUS ldapsrv_AddRequest(struct ldapsrv_call *call)
343 struct ldap_AddRequest *req = &call->request.r.AddRequest;
344 struct ldapsrv_partition *part;
346 DEBUG(10, ("AddRequest"));
347 DEBUGADD(10, (" dn: %s", req->dn));
349 part = ldapsrv_get_partition(call->conn, req->dn);
351 if (!part->ops->Add) {
352 return ldapsrv_unwilling(call, 53);
355 return part->ops->Add(part, call, req);
358 static NTSTATUS ldapsrv_DelRequest(struct ldapsrv_call *call)
360 struct ldap_DelRequest *req = &call->request.r.DelRequest;
361 struct ldapsrv_partition *part;
363 DEBUG(10, ("DelRequest"));
364 DEBUGADD(10, (" dn: %s", req->dn));
366 part = ldapsrv_get_partition(call->conn, req->dn);
368 if (!part->ops->Del) {
369 return ldapsrv_unwilling(call, 53);
372 return part->ops->Del(part, call, req);
375 static NTSTATUS ldapsrv_ModifyDNRequest(struct ldapsrv_call *call)
377 struct ldap_ModifyDNRequest *req = &call->request.r.ModifyDNRequest;
378 struct ldapsrv_partition *part;
380 DEBUG(10, ("ModifyDNRequrest"));
381 DEBUGADD(10, (" dn: %s", req->dn));
382 DEBUGADD(10, (" newrdn: %s", req->newrdn));
384 part = ldapsrv_get_partition(call->conn, req->dn);
386 if (!part->ops->ModifyDN) {
387 return ldapsrv_unwilling(call, 53);
390 return part->ops->ModifyDN(part, call, req);
393 static NTSTATUS ldapsrv_CompareRequest(struct ldapsrv_call *call)
395 struct ldap_CompareRequest *req = &call->request.r.CompareRequest;
396 struct ldapsrv_partition *part;
398 DEBUG(10, ("CompareRequest"));
399 DEBUGADD(10, (" dn: %s", req->dn));
401 part = ldapsrv_get_partition(call->conn, req->dn);
403 if (!part->ops->Compare) {
404 return ldapsrv_unwilling(call, 53);
407 return part->ops->Compare(part, call, req);
410 static NTSTATUS ldapsrv_AbandonRequest(struct ldapsrv_call *call)
412 /* struct ldap_AbandonRequest *req = &call->request.r.AbandonRequest;*/
413 DEBUG(10, ("AbandonRequest\n"));
417 static NTSTATUS ldapsrv_ExtendedRequest(struct ldapsrv_call *call)
419 /* struct ldap_ExtendedRequest *req = &call->request.r.ExtendedRequest;*/
420 struct ldapsrv_reply *reply;
422 DEBUG(10, ("Extended\n"));
424 reply = ldapsrv_init_reply(call, LDAP_TAG_ExtendedResponse);
426 return NT_STATUS_NO_MEMORY;
429 ZERO_STRUCT(reply->msg.r);
431 return ldapsrv_queue_reply(call, reply);
434 static NTSTATUS ldapsrv_do_call(struct ldapsrv_call *call)
436 switch(call->request.type) {
437 case LDAP_TAG_BindRequest:
438 return ldapsrv_BindRequest(call);
439 case LDAP_TAG_UnbindRequest:
440 return ldapsrv_UnbindRequest(call);
441 case LDAP_TAG_SearchRequest:
442 return ldapsrv_SearchRequest(call);
443 case LDAP_TAG_ModifyRequest:
444 return ldapsrv_ModifyRequest(call);
445 case LDAP_TAG_AddRequest:
446 return ldapsrv_AddRequest(call);
447 case LDAP_TAG_DelRequest:
448 return ldapsrv_DelRequest(call);
449 case LDAP_TAG_ModifyDNRequest:
450 return ldapsrv_ModifyDNRequest(call);
451 case LDAP_TAG_CompareRequest:
452 return ldapsrv_CompareRequest(call);
453 case LDAP_TAG_AbandonRequest:
454 return ldapsrv_AbandonRequest(call);
455 case LDAP_TAG_ExtendedRequest:
456 return ldapsrv_ExtendedRequest(call);
458 return ldapsrv_unwilling(call, 2);
462 static NTSTATUS ldapsrv_do_responses(struct ldapsrv_connection *conn)
464 struct ldapsrv_call *call, *next_call = NULL;
465 struct ldapsrv_reply *reply, *next_reply = NULL;
467 for (call=conn->calls; call; call=next_call) {
468 for (reply=call->replies; reply; reply=next_reply) {
469 if (!ldap_append_to_buf(&reply->msg, &conn->out_buffer)) {
470 return NT_STATUS_FOOBAR;
472 next_reply = reply->next;
473 DLIST_REMOVE(call->replies, reply);
474 reply->state = LDAPSRV_REPLY_STATE_SEND;
477 next_call = call->next;
478 DLIST_REMOVE(conn->calls, call);
479 call->state = LDAPSRV_CALL_STATE_COMPLETE;
487 called when a LDAP socket becomes readable
489 static void ldapsrv_recv(struct server_connection *conn, time_t t,
492 struct ldapsrv_connection *ldap_conn = conn->private_data;
494 int buf_length, msg_length;
497 struct ldapsrv_call *call;
500 DEBUG(10,("ldapsrv_recv\n"));
502 if (!read_into_buf(conn->socket, &ldap_conn->in_buffer)) {
503 ldapsrv_terminate_connection(ldap_conn, "read_into_buf() failed");
507 peek_into_read_buf(&ldap_conn->in_buffer, &buf, &buf_length);
509 while (buf_length > 0) {
511 peek_into_read_buf(&ldap_conn->in_buffer, &buf, &buf_length);
512 /* LDAP Messages are always SEQUENCES */
514 if (!asn1_object_length(buf, buf_length, ASN1_SEQUENCE(0),
516 ldapsrv_terminate_connection(ldap_conn, "asn1_object_length() failed");
520 if (buf_length < msg_length) {
525 /* We've got a complete LDAP request in the in-buffer, convert
526 * that to a ldap_message and put it into the incoming
530 blob.length = msg_length;
532 if (!asn1_load(&data, blob)) {
533 ldapsrv_terminate_connection(ldap_conn, "asn1_load() failed");
537 call = talloc_p(ldap_conn, struct ldapsrv_call);
539 ldapsrv_terminate_connection(ldap_conn, "no memory");
544 call->state = LDAPSRV_CALL_STATE_NEW;
545 call->conn = ldap_conn;
546 /* TODO: we should use talloc_reference() here */
547 call->session_info = ldap_conn->session_info;
548 call->request.mem_ctx = call;
550 if (!ldap_decode(&data, &call->request)) {
551 dump_data(0,buf, msg_length);
553 ldapsrv_terminate_connection(ldap_conn, "ldap_decode() failed");
559 DLIST_ADD_END(ldap_conn->calls, call,
560 struct ldapsrv_call *);
562 consumed_from_buf(&ldap_conn->in_buffer, msg_length);
564 status = ldapsrv_do_call(call);
565 if (!NT_STATUS_IS_OK(status)) {
566 ldapsrv_terminate_connection(ldap_conn, "ldapsrv_do_call() failed");
570 peek_into_read_buf(&ldap_conn->in_buffer, &buf, &buf_length);
573 status = ldapsrv_do_responses(ldap_conn);
574 if (!NT_STATUS_IS_OK(status)) {
575 ldapsrv_terminate_connection(ldap_conn, "ldapsrv_do_responses() failed");
579 if (ldap_conn->out_buffer.length > 0) {
580 conn->event.fde->flags |= EVENT_FD_WRITE;
587 called when a LDAP socket becomes writable
589 static void ldapsrv_send(struct server_connection *conn, time_t t,
592 struct ldapsrv_connection *ldap_conn = conn->private_data;
594 DEBUG(10,("ldapsrv_send\n"));
596 if (!write_from_buf(conn->socket, &ldap_conn->out_buffer)) {
597 ldapsrv_terminate_connection(ldap_conn, "write_from_buf() failed");
601 if (ldap_conn->out_buffer.length == 0) {
602 conn->event.fde->flags &= ~EVENT_FD_WRITE;
609 called when connection is idle
611 static void ldapsrv_idle(struct server_connection *conn, time_t t)
613 DEBUG(10,("ldapsrv_idle: not implemented!\n"));
617 static void ldapsrv_close(struct server_connection *conn, const char *reason)
623 initialise a server_context from a open socket and register a event handler
624 for reading from that socket
626 static void ldapsrv_accept(struct server_connection *conn)
628 struct ldapsrv_connection *ldap_conn;
630 DEBUG(10, ("ldapsrv_accept\n"));
632 ldap_conn = talloc_p(conn, struct ldapsrv_connection);
634 if (ldap_conn == NULL)
637 ZERO_STRUCTP(ldap_conn);
638 ldap_conn->connection = conn;
639 ldap_conn->service = talloc_reference(ldap_conn, conn->service->private_data);
641 conn->private_data = ldap_conn;
647 called on a fatal error that should cause this server to terminate
649 static void ldapsrv_exit(struct server_service *service, const char *reason)
651 DEBUG(10,("ldapsrv_exit\n"));
655 static const struct server_service_ops ldap_server_ops = {
657 .service_init = ldapsrv_init,
658 .accept_connection = ldapsrv_accept,
659 .recv_handler = ldapsrv_recv,
660 .send_handler = ldapsrv_send,
661 .idle_handler = ldapsrv_idle,
662 .close_connection = ldapsrv_close,
663 .service_exit = ldapsrv_exit,
666 const struct server_service_ops *ldapsrv_get_ops(void)
668 return &ldap_server_ops;
671 NTSTATUS server_service_ldap_init(void)