On Mon, 2011-04-25 at 16:21 -0400, Stephen Gallagher wrote:
> On Mon, 2011-04-25 at 15:33 -0400, Stephen Gallagher wrote:
> > On Mon, 2011-04-25 at 14:26 -0400, Stephen Gallagher wrote:
> > > Patch 0001: Added a debug message to see which record type we're
> > > processing on each loop through sdap_process_message(). This is purely
> > > informational.
> > > 
> > > Patch 0002: Add support for paged LDAP results.
> > > I changed the internals of sdap_get_generic_send() somewhat here and
> > > added a new sdap_get_generic_internal() routine that can be used to
> > > handle multiple calls to LDAP for a single request. We should be able to
> > > use this in the future to handle Active Directory's non-standard
> > > attribute-level paging as well (though that's not addressed in this
> > > patch).
> > > 
> > > Patch 0003: Add ldap_page_size configuration option
> > > I made this a separate patch for simplicity of review. I set 1000
> > > records as the default, as this seemed to be the most-compatible value
> > > among 389, OpenLDAP and ActiveDirectory as best I could determine.
> > 
> > 
> > Simo made some comments on IRC that I have incorporated.
> > 
> > First, I renamed sdap_get_generic_send_internal() to
> > sdap_get_generic_step() and reordered it according to the tevent_req
> > style. Second, I added talloc_zfree(state->op) to the beginning of
> > sdap_get_generic_step() to ensure that we aren't carrying around extra
> > sdap_op memory for subsequent pages.
> 
> Simo had a few remaining comments. To follow tevent_req style, I've
> moved forward declarations to be above the sdap_get_generic_send().
> 
> Also, there was a memory issue (I was accessing memory that had been
> leaked). The cookie had gone out of scope, but I hadn't freed the memory
> (which is why I saw no issues in my tests).
> 
> I now make an appropriate copy of the cookie berval and free the
> original memory.

One more change: the cookie is now copied with talloc_memdup() instead
of talloc_strdup(), since it may not be a proper string (I'm looking at
YOU Active Directory).
From 8cde8b64e572b737a493750755e1c4b152b70414 Mon Sep 17 00:00:00 2001
From: Stephen Gallagher <sgall...@redhat.com>
Date: Mon, 25 Apr 2011 13:11:16 -0400
Subject: [PATCH 1/4] Log the LDAP message type we're processing

---
 src/providers/ldap/sdap_async.c |   57 +++++++++++++++++++++++++++++++++++++++
 1 files changed, 57 insertions(+), 0 deletions(-)

diff --git a/src/providers/ldap/sdap_async.c b/src/providers/ldap/sdap_async.c
index ed85a910818a38e5677ce7a2f409371a634b75a9..b96bbdea438cffb61988d8fccaf2c0cb375b201c 100644
--- a/src/providers/ldap/sdap_async.c
+++ b/src/providers/ldap/sdap_async.c
@@ -207,6 +207,61 @@ static void sdap_process_result(struct tevent_context *ev, void *pvt)
     sdap_process_message(ev, sh, msg);
 }
 
+static const char *sdap_ldap_result_str(int msgtype)
+{
+    switch (msgtype) {
+    case LDAP_RES_BIND:
+        return "LDAP_RES_BIND";
+
+    case LDAP_RES_SEARCH_ENTRY:
+        return "LDAP_RES_SEARCH_ENTRY";
+
+    case LDAP_RES_SEARCH_REFERENCE:
+        return "LDAP_RES_SEARCH_REFERENCE";
+
+    case LDAP_RES_SEARCH_RESULT:
+        return "LDAP_RES_SEARCH_RESULT";
+
+    case LDAP_RES_MODIFY:
+        return "LDAP_RES_MODIFY";
+
+    case LDAP_RES_ADD:
+        return "LDAP_RES_ADD";
+
+    case LDAP_RES_DELETE:
+        return "LDAP_RES_DELETE";
+
+    case LDAP_RES_MODDN:
+    /* These are the same result
+    case LDAP_RES_MODRDN:
+    case LDAP_RES_RENAME:
+    */
+        return "LDAP_RES_RENAME";
+
+    case LDAP_RES_COMPARE:
+        return "LDAP_RES_COMPARE";
+
+    case LDAP_RES_EXTENDED:
+        return "LDAP_RES_EXTENDED";
+
+    case LDAP_RES_INTERMEDIATE:
+        return "LDAP_RES_INTERMEDIATE";
+
+    case LDAP_RES_ANY:
+        return "LDAP_RES_ANY";
+
+    case LDAP_RES_UNSOLICITED:
+        return "LDAP_RES_UNSOLICITED";
+
+    default:
+        /* Unmatched, fall through */
+        break;
+    }
+
+    /* Unknown result type */
+    return "Unknown result type!";
+}
+
 /* process a messgae calling the right operation callback.
  * msg is completely taken care of (including freeeing it)
  * NOTE: this function may even end up freeing the sdap_handle
@@ -248,6 +303,8 @@ static void sdap_process_message(struct tevent_context *ev,
         return;
     }
 
+    DEBUG(9, ("Message type: [%s]\n", sdap_ldap_result_str(msgtype)));
+
     switch (msgtype) {
     case LDAP_RES_SEARCH_ENTRY:
         /* go and process entry */
-- 
1.7.4.4

From c7aebb7b4323f065e6e2c8bf55ea7f00bf431e08 Mon Sep 17 00:00:00 2001
From: Stephen Gallagher <sgall...@redhat.com>
Date: Mon, 25 Apr 2011 13:14:01 -0400
Subject: [PATCH 2/4] Enable paging support for LDAP

---
 src/providers/ldap/sdap_async.c |  141 ++++++++++++++++++++++++++++++++------
 1 files changed, 118 insertions(+), 23 deletions(-)

diff --git a/src/providers/ldap/sdap_async.c b/src/providers/ldap/sdap_async.c
index b96bbdea438cffb61988d8fccaf2c0cb375b201c..2d82c67740fa5adc78c0596261b3ddef6795937e 100644
--- a/src/providers/ldap/sdap_async.c
+++ b/src/providers/ldap/sdap_async.c
@@ -806,21 +806,24 @@ struct sdap_get_generic_state {
     const char **attrs;
     struct sdap_attr_map *map;
     int map_num_attrs;
+    int timeout;
 
     struct sdap_op *op;
 
+    struct berval cookie;
+
     size_t reply_max;
     size_t reply_count;
     struct sysdb_attrs **reply;
 };
 
+static errno_t sdap_get_generic_step(struct tevent_req *req);
+static void sdap_get_generic_done(struct sdap_op *op,
+                                 struct sdap_msg *reply,
+                                 int error, void *pvt);
 static errno_t add_to_reply(struct sdap_get_generic_state *state,
                             struct sysdb_attrs *msg);
 
-static void sdap_get_generic_done(struct sdap_op *op,
-                                  struct sdap_msg *reply,
-                                  int error, void *pvt);
-
 struct tevent_req *sdap_get_generic_send(TALLOC_CTX *memctx,
                                          struct tevent_context *ev,
                                          struct sdap_options *opts,
@@ -833,13 +836,9 @@ struct tevent_req *sdap_get_generic_send(TALLOC_CTX *memctx,
                                          int map_num_attrs,
                                          int timeout)
 {
-    struct tevent_req *req = NULL;
-    struct sdap_get_generic_state *state = NULL;
-    char *errmsg;
-    int lret;
-    int optret;
-    int ret;
-    int msgid;
+    errno_t ret;
+    struct sdap_get_generic_state *state;
+    struct tevent_req *req;
 
     req = tevent_req_create(memctx, &state, struct sdap_get_generic_state);
     if (!req) return NULL;
@@ -857,6 +856,40 @@ struct tevent_req *sdap_get_generic_send(TALLOC_CTX *memctx,
     state->reply_max = 0;
     state->reply_count = 0;
     state->reply = NULL;
+    state->timeout = timeout;
+    state->cookie.bv_len = 0;
+    state->cookie.bv_val = NULL;
+
+    ret = sdap_get_generic_step(req);
+    if (ret != EOK) {
+        tevent_req_error(req, ret);
+        tevent_req_post(req, ev);
+        return req;
+    }
+
+    return req;
+}
+
+static errno_t sdap_get_generic_step(struct tevent_req *req)
+{
+    struct sdap_get_generic_state *state =
+            tevent_req_data(req, struct sdap_get_generic_state);
+    char *errmsg;
+    int lret;
+    int optret;
+    errno_t ret;
+    int msgid;
+
+    ber_int_t page_size = 1000;
+    char paging_criticality = 'F';
+    LDAPControl *page_control = NULL;
+    LDAPControl *m_controls[2] = { NULL, NULL };
+
+    /* Make sure to free any previous operations so
+     * if we are handling a large number of pages we
+     * don't waste memory.
+     */
+    talloc_zfree(state->op);
 
     DEBUG(6, ("calling ldap_search_ext with [%s][%s].\n", state->filter,
                                                           state->search_base));
@@ -870,10 +903,26 @@ struct tevent_req *sdap_get_generic_send(TALLOC_CTX *memctx,
         }
     }
 
+    if (sdap_is_control_supported(state->sh,
+                                  LDAP_CONTROL_PAGEDRESULTS)) {
+        lret = ldap_create_page_control(state->sh->ldap,
+                                        page_size,
+                                        state->cookie.bv_val ?
+                                            &state->cookie :
+                                            NULL,
+                                        paging_criticality,
+                                        &page_control);
+        if (lret != LDAP_SUCCESS) {
+            ret = EIO;
+            goto done;
+        }
+        m_controls[0] = page_control;
+    }
+
     lret = ldap_search_ext(state->sh->ldap, state->search_base,
                            state->scope, state->filter,
                            discard_const(state->attrs),
-                           false, NULL, NULL, NULL, 0, &msgid);
+                           false, m_controls, NULL, NULL, 0, &msgid);
     if (lret != LDAP_SUCCESS) {
         DEBUG(3, ("ldap_search_ext failed: %s\n", ldap_err2string(lret)));
         if (lret == LDAP_SERVER_DOWN) {
@@ -893,27 +942,23 @@ struct tevent_req *sdap_get_generic_send(TALLOC_CTX *memctx,
         else {
             ret = EIO;
         }
-        goto fail;
+        goto done;
     }
     DEBUG(8, ("ldap_search_ext called, msgid = %d\n", msgid));
 
     ret = sdap_op_add(state, state->ev, state->sh, msgid,
-                      sdap_get_generic_done, req, timeout,
+                      sdap_get_generic_done, req,
+                      state->timeout,
                       &state->op);
     if (ret != EOK) {
         DEBUG(1, ("Failed to set up operation!\n"));
-        goto fail;
+        goto done;
     }
 
-    return req;
-
-fail:
-    tevent_req_error(req, ret);
-    tevent_req_post(req, ev);
-    return req;
+done:
+    return ret;
 }
 
-
 static void sdap_get_generic_done(struct sdap_op *op,
                                  struct sdap_msg *reply,
                                  int error, void *pvt)
@@ -925,6 +970,11 @@ static void sdap_get_generic_done(struct sdap_op *op,
     char *errmsg = NULL;
     int result;
     int ret;
+    int lret;
+    ber_int_t total_count;
+    struct berval cookie;
+    LDAPControl **returned_controls = NULL;
+    LDAPControl *page_control;
 
     if (error) {
         tevent_req_error(req, error);
@@ -962,7 +1012,8 @@ static void sdap_get_generic_done(struct sdap_op *op,
 
     case LDAP_RES_SEARCH_RESULT:
         ret = ldap_parse_result(state->sh->ldap, reply->msg,
-                                &result, NULL, &errmsg, NULL, NULL, 0);
+                                &result, NULL, &errmsg, NULL,
+                                &returned_controls, 0);
         if (ret != LDAP_SUCCESS) {
             DEBUG(2, ("ldap_parse_result failed (%d)\n", state->op->msgid));
             tevent_req_error(req, EIO);
@@ -978,6 +1029,50 @@ static void sdap_get_generic_done(struct sdap_op *op,
         }
         ldap_memfree(errmsg);
 
+        /* Determine if there are more pages to retrieve */
+        page_control = ldap_control_find(LDAP_CONTROL_PAGEDRESULTS,
+                                         returned_controls, NULL );
+        if (!page_control) {
+            /* No paging support. We are done */
+            tevent_req_done(req);
+            return;
+        }
+
+        lret = ldap_parse_pageresponse_control(state->sh->ldap, page_control,
+                                               &total_count, &cookie);
+        if (lret != LDAP_SUCCESS) {
+            DEBUG(1, ("Could not determine page control"));
+            tevent_req_error(req, EIO);
+            return;
+        }
+        DEBUG(7, ("Total count [%lu]\n", total_count));
+
+        if (cookie.bv_val != NULL && cookie.bv_len > 0) {
+            /* Cookie contains data, which means there are more requests
+             * to be processed.
+             */
+            talloc_zfree(state->cookie.bv_val);
+            state->cookie.bv_len = cookie.bv_len;
+            state->cookie.bv_val = talloc_memdup(state,
+                                                 cookie.bv_val,
+                                                 cookie.bv_len);
+            if (!state->cookie.bv_val) {
+                tevent_req_error(req, ENOMEM);
+                return;
+            }
+            ber_memfree(cookie.bv_val);
+
+            ret = sdap_get_generic_step(req);
+            if (ret != EOK) {
+                tevent_req_error(req, ENOMEM);
+                return;
+            }
+
+            return;
+        }
+
+        /* This was the last page. We're done */
+
         tevent_req_done(req);
         return;
 
-- 
1.7.4.4

From 5f988f1333e77f9768285e25f847620d1ef1382e Mon Sep 17 00:00:00 2001
From: Stephen Gallagher <sgall...@redhat.com>
Date: Mon, 25 Apr 2011 14:17:45 -0400
Subject: [PATCH 3/4] Add ldap_page_size configuration option

---
 src/config/SSSDConfig.py                   |    1 +
 src/config/etc/sssd.api.d/sssd-ldap.conf   |    1 +
 src/man/sssd-ldap.5.xml                    |   14 ++++++++++++++
 src/providers/ipa/ipa_common.c             |    3 ++-
 src/providers/ipa/ipa_common.h             |    2 +-
 src/providers/ldap/ldap_common.c           |    3 ++-
 src/providers/ldap/sdap.h                  |    2 ++
 src/providers/ldap/sdap_async.c            |    3 +--
 src/providers/ldap/sdap_async_connection.c |    4 ++++
 9 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/src/config/SSSDConfig.py b/src/config/SSSDConfig.py
index 02f76af28a0011c36faf4c99bd0abc6ae74bf03d..b613cfe4e435888906fac423df986be7b534faf1 100644
--- a/src/config/SSSDConfig.py
+++ b/src/config/SSSDConfig.py
@@ -140,6 +140,7 @@ option_strings = {
     'ldap_krb5_ticket_lifetime' : _('Lifetime of TGT for LDAP connection'),
     'ldap_deref' : _('How to dereference aliases'),
     'ldap_dns_service_name' : _('Service name for DNS service lookups'),
+    'ldap_page_size' : _('The number of records to retrieve in a single LDAP query'),
 
     'ldap_entry_usn' : _('entryUSN attribute'),
     'ldap_rootdse_last_usn' : _('lastUSN attribute'),
diff --git a/src/config/etc/sssd.api.d/sssd-ldap.conf b/src/config/etc/sssd.api.d/sssd-ldap.conf
index 8672f0b24fb698881f0d1d8d2c705c7acda6e198..e568c74d3d1763ada387fa11fe39876c7879a7bc 100644
--- a/src/config/etc/sssd.api.d/sssd-ldap.conf
+++ b/src/config/etc/sssd.api.d/sssd-ldap.conf
@@ -27,6 +27,7 @@ ldap_referrals = bool, None, false
 ldap_krb5_ticket_lifetime = int, None, false
 ldap_dns_service_name = str, None, false
 ldap_deref = str, None, false
+ldap_page_size = int, None, false
 
 [provider/ldap/id]
 ldap_search_timeout = int, None, false
diff --git a/src/man/sssd-ldap.5.xml b/src/man/sssd-ldap.5.xml
index 9d585e2ae87eab81444a4295ad246638673ae80a..49c9e49157f45449617028e75934e0f28a408278 100644
--- a/src/man/sssd-ldap.5.xml
+++ b/src/man/sssd-ldap.5.xml
@@ -855,6 +855,20 @@
                 </varlistentry>
 
                 <varlistentry>
+                    <term>ldap_page_size (integer)</term>
+                    <listitem>
+                        <para>
+                            Specify the number of records to retrieve from
+                            LDAP in a single request. Some LDAP servers
+                            enforce a maximum limit per-request.
+                        </para>
+                        <para>
+                            Default: 1000
+                        </para>
+                    </listitem>
+                </varlistentry>
+
+                <varlistentry>
                     <term>ldap_tls_reqcert (string)</term>
                     <listitem>
                         <para>
diff --git a/src/providers/ipa/ipa_common.c b/src/providers/ipa/ipa_common.c
index 7ba4fd5a43e54a6b2d8d24ce58d91f31a6ee8d9b..a0c728ef43a63e6d1c3fffd131362741a5e59d96 100644
--- a/src/providers/ipa/ipa_common.c
+++ b/src/providers/ipa/ipa_common.c
@@ -93,7 +93,8 @@ struct dp_option ipa_def_ldap_opts[] = {
     /* Do not include ldap_auth_disable_tls_never_use_in_production in the
      * manpages or SSSDConfig API
      */
-    { "ldap_auth_disable_tls_never_use_in_production", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE }
+    { "ldap_auth_disable_tls_never_use_in_production", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
+    { "ldap_page_size", DP_OPT_NUMBER, { .number = 1000 }, NULL_NUMBER }
 };
 
 struct sdap_attr_map ipa_attr_map[] = {
diff --git a/src/providers/ipa/ipa_common.h b/src/providers/ipa/ipa_common.h
index 12a4927072eaed47072b32e18eced342be5969c4..02c4276cab7382749ac113f75e1019636daca991 100644
--- a/src/providers/ipa/ipa_common.h
+++ b/src/providers/ipa/ipa_common.h
@@ -35,7 +35,7 @@ struct ipa_service {
 /* the following defines are used to keep track of the options in the ldap
  * module, so that if they change and ipa is not updated correspondingly
  * this will trigger a runtime abort error */
-#define IPA_OPTS_BASIC_TEST 49
+#define IPA_OPTS_BASIC_TEST 50
 
 /* the following define is used to keep track of the options in the krb5
  * module, so that if they change and ipa is not updated correspondingly
diff --git a/src/providers/ldap/ldap_common.c b/src/providers/ldap/ldap_common.c
index 11c4491f90990c512becdc06c58ddbfec02940c5..12028b0131f4f4cdaca2800bdf135ee98617d045 100644
--- a/src/providers/ldap/ldap_common.c
+++ b/src/providers/ldap/ldap_common.c
@@ -86,7 +86,8 @@ struct dp_option default_basic_opts[] = {
     /* Do not include ldap_auth_disable_tls_never_use_in_production in the
      * manpages or SSSDConfig API
      */
-    { "ldap_auth_disable_tls_never_use_in_production", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE }
+    { "ldap_auth_disable_tls_never_use_in_production", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
+    { "ldap_page_size", DP_OPT_NUMBER, { .number = 1000 }, NULL_NUMBER }
 };
 
 struct sdap_attr_map generic_attr_map[] = {
diff --git a/src/providers/ldap/sdap.h b/src/providers/ldap/sdap.h
index c06b8a3b76920b5f25250d54749ff841673e8d8a..0f6b755041681e1b59d2e07583d38ab51eea1d1e 100644
--- a/src/providers/ldap/sdap.h
+++ b/src/providers/ldap/sdap.h
@@ -77,6 +77,7 @@ struct sdap_handle {
     bool connected;
     /* Authentication ticket expiration time (if any) */
     time_t expire_time;
+    ber_int_t page_size;
 
     struct sdap_fd_events *sdap_fd_events;
 
@@ -192,6 +193,7 @@ enum sdap_basic_opt {
     SDAP_CHPASS_DNS_SERVICE_NAME,
     SDAP_ENUM_SEARCH_TIMEOUT,
     SDAP_DISABLE_AUTH_TLS,
+    SDAP_PAGE_SIZE,
 
     SDAP_OPTS_BASIC /* opts counter */
 };
diff --git a/src/providers/ldap/sdap_async.c b/src/providers/ldap/sdap_async.c
index 2d82c67740fa5adc78c0596261b3ddef6795937e..6953426171e9a8dc3872c83722b54f92176d8906 100644
--- a/src/providers/ldap/sdap_async.c
+++ b/src/providers/ldap/sdap_async.c
@@ -880,7 +880,6 @@ static errno_t sdap_get_generic_step(struct tevent_req *req)
     errno_t ret;
     int msgid;
 
-    ber_int_t page_size = 1000;
     char paging_criticality = 'F';
     LDAPControl *page_control = NULL;
     LDAPControl *m_controls[2] = { NULL, NULL };
@@ -906,7 +905,7 @@ static errno_t sdap_get_generic_step(struct tevent_req *req)
     if (sdap_is_control_supported(state->sh,
                                   LDAP_CONTROL_PAGEDRESULTS)) {
         lret = ldap_create_page_control(state->sh->ldap,
-                                        page_size,
+                                        state->sh->page_size,
                                         state->cookie.bv_val ?
                                             &state->cookie :
                                             NULL,
diff --git a/src/providers/ldap/sdap_async_connection.c b/src/providers/ldap/sdap_async_connection.c
index 500e5f886f0855f10171096fc9dc29a4a5233fea..40ed585c6bf555478da93b09e267292a878080a7 100644
--- a/src/providers/ldap/sdap_async_connection.c
+++ b/src/providers/ldap/sdap_async_connection.c
@@ -108,6 +108,10 @@ struct tevent_req *sdap_connect_send(TALLOC_CTX *memctx,
         talloc_zfree(req);
         return NULL;
     }
+
+    state->sh->page_size = dp_opt_get_int(state->opts->basic,
+                                          SDAP_PAGE_SIZE);
+
     /* Initialize LDAP handler */
     lret = ldap_initialize(&state->sh->ldap, uri);
     if (lret != LDAP_SUCCESS) {
-- 
1.7.4.4

Attachment: signature.asc
Description: This is a digitally signed message part

_______________________________________________
sssd-devel mailing list
sssd-devel@lists.fedorahosted.org
https://fedorahosted.org/mailman/listinfo/sssd-devel

Reply via email to