added a ASQ search API for sysdb

2 files changed, 230 insertions, 0 deletions

diff --git a/server/db/sysdb.h b/server/db/sysdb.h
index e1cff852..5c15d3a1 100644
--- a/server/db/sysdb.h
+++ b/server/db/sysdb.h
@@ -555,4 +555,16 @@ struct tevent_req *sysdb_delete_custom_send(TALLOC_CTX *mem_ctx,
                                              const char *object_name,
                                              const char *subtree_name);
 int sysdb_delete_custom_recv(struct tevent_req *req);
+
+struct tevent_req *sysdb_asq_search_send(TALLOC_CTX *mem_ctx,
+                                         struct tevent_context *ev,
+                                         struct sysdb_ctx *sysdb,
+                                         struct sysdb_handle *handle,
+                                         struct sss_domain_info *domain,
+                                         struct ldb_dn *base_dn,
+                                         const char *expression,
+                                         const char *asq_attribute,
+                                         const char **attrs);
+int sysdb_asq_search_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
+                          size_t *msgs_count, struct ldb_message ***msgs);
 #endif /* __SYS_DB_H__ */
diff --git a/server/db/sysdb_ops.c b/server/db/sysdb_ops.c
index 0bb77d17..18022341 100644
--- a/server/db/sysdb_ops.c
+++ b/server/db/sysdb_ops.c
@@ -3854,3 +3854,221 @@ int sysdb_delete_custom_recv(struct tevent_req *req)
 
     return EOK;
 }
+
+/* = ASQ search request ======================================== */
+struct sysdb_asq_search_state {
+    struct tevent_context *ev;
+    struct sysdb_ctx *sysdb;
+    struct sysdb_handle *handle;
+    struct sss_domain_info *domain;
+    struct ldb_dn *base_dn;
+    const char *asq_attribute;
+    const char **attrs;
+    const char *expression;
+
+    int msgs_count;
+    struct ldb_message **msgs;
+};
+
+void sysdb_asq_search_check_handle_done(struct tevent_req *subreq);
+static void sysdb_asq_search_done(struct tevent_req *subreq);
+
+struct tevent_req *sysdb_asq_search_send(TALLOC_CTX *mem_ctx,
+                                         struct tevent_context *ev,
+                                         struct sysdb_ctx *sysdb,
+                                         struct sysdb_handle *handle,
+                                         struct sss_domain_info *domain,
+                                         struct ldb_dn *base_dn,
+                                         const char *expression,
+                                         const char *asq_attribute,
+                                         const char **attrs)
+{
+    struct tevent_req *req;
+    struct tevent_req *subreq;
+    struct sysdb_asq_search_state *state;
+    int ret;
+
+    if (sysdb == NULL && handle == NULL) {
+        DEBUG(1, ("Sysdb context not available.\n"));
+        return NULL;
+    }
+
+    req = tevent_req_create(mem_ctx, &state, struct sysdb_asq_search_state);
+    if (req == NULL) {
+        DEBUG(1, ("tevent_req_create failed.\n"));
+        return NULL;
+    }
+
+    state->ev = ev;
+    state->sysdb = (sysdb == NULL) ? handle->ctx : sysdb;
+    state->handle = handle;
+    state->domain = domain;
+    state->base_dn = base_dn;
+    state->expression = expression;
+    state->asq_attribute = asq_attribute;
+    state->attrs = attrs;
+
+    state->msgs_count = 0;
+    state->msgs = NULL;
+
+    subreq = sysdb_check_handle_send(state, state->ev, state->sysdb,
+                                     state->handle);
+    if (!subreq) {
+        DEBUG(1, ("sysdb_check_handle_send failed.\n"));
+        ret = ENOMEM;
+        goto fail;
+    }
+    tevent_req_set_callback(subreq, sysdb_asq_search_check_handle_done, req);
+
+    return req;
+
+fail:
+    tevent_req_error(req, ret);
+    tevent_req_post(req, ev);
+    return req;
+}
+
+void sysdb_asq_search_check_handle_done(struct tevent_req *subreq)
+{
+    struct tevent_req *req = tevent_req_callback_data(subreq,
+                                                      struct tevent_req);
+    struct sysdb_asq_search_state *state = tevent_req_data(req,
+                                            struct sysdb_asq_search_state);
+    struct ldb_request *ldb_req;
+    struct ldb_control **ctrl;
+    struct ldb_asq_control *asq_control;
+    int ret;
+
+    ret = sysdb_check_handle_recv(subreq, state, &state->handle);
+    talloc_zfree(subreq);
+    if (ret != EOK) {
+        tevent_req_error(req, ret);
+        return;
+    }
+
+    ctrl = talloc_array(state, struct ldb_control *, 2);
+    if (ctrl == NULL) {
+        ret = ENOMEM;
+        goto fail;
+    }
+
+    ctrl[0] = talloc(ctrl, struct ldb_control);
+    if (ctrl[0] == NULL) {
+        ret = ENOMEM;
+        goto fail;
+    }
+    ctrl[1] = NULL;
+
+    ctrl[0]->oid = LDB_CONTROL_ASQ_OID;
+    ctrl[0]->critical = 1;
+
+    asq_control = talloc(ctrl[0], struct ldb_asq_control);
+    if (asq_control == NULL) {
+        ret = ENOMEM;
+        goto fail;
+    }
+
+    asq_control->request = 1;
+    asq_control->source_attribute = talloc_strdup(asq_control,
+                                                  state->asq_attribute);
+    if (asq_control->source_attribute == NULL) {
+        ret = ENOMEM;
+        goto fail;
+    }
+    asq_control->src_attr_len = strlen(asq_control->source_attribute);
+    ctrl[0]->data = asq_control;
+
+    ret = ldb_build_search_req(&ldb_req, state->handle->ctx->ldb, state,
+            state->base_dn, LDB_SCOPE_BASE,
+            state->expression, state->attrs, ctrl,
+            NULL, NULL, NULL);
+    if (ret != LDB_SUCCESS) {
+        ret = sysdb_error_to_errno(ret);
+        goto fail;
+    }
+
+    subreq = sldb_request_send(state, state->ev, state->handle->ctx->ldb,
+                               ldb_req);
+    if (!subreq) {
+        ret = ENOMEM;
+        goto fail;
+    }
+
+    tevent_req_set_callback(subreq, sysdb_asq_search_done, req);
+    return;
+
+fail:
+    tevent_req_error(req, ret);
+    return;
+}
+
+static void sysdb_asq_search_done(struct tevent_req *subreq)
+{
+    struct tevent_req *req = tevent_req_callback_data(subreq,
+                                                  struct tevent_req);
+    struct sysdb_asq_search_state *state = tevent_req_data(req,
+                                                 struct sysdb_asq_search_state);
+    struct ldb_reply *ldbreply;
+    int ret;
+
+    ret = sldb_request_recv(subreq, state, &ldbreply);
+    if (ret != EOK) {
+        talloc_free(subreq);
+        DEBUG(6, ("Error: %d (%s)\n", ret, strerror(ret)));
+        tevent_req_error(req, ret);
+        return;
+    }
+
+    switch (ldbreply->type) {
+        case LDB_REPLY_ENTRY:
+            state->msgs = talloc_realloc(state, state->msgs,
+                                         struct ldb_message *,
+                                         state->msgs_count + 2);
+            if (state->msgs == NULL) {
+                tevent_req_error(req, ENOMEM);
+                return;
+            }
+
+            state->msgs[state->msgs_count + 1] = NULL;
+
+            state->msgs[state->msgs_count] = talloc_steal(state->msgs,
+                                                          ldbreply->message);
+            state->msgs_count++;
+
+            talloc_zfree(ldbreply);
+            return;
+
+        case LDB_REPLY_DONE:
+            break;
+
+        default:
+            DEBUG(1, ("Unknown ldb reply type [%d].\n", ldbreply->type));
+            tevent_req_error(req, EINVAL);
+            return;
+    }
+
+    talloc_zfree(subreq);
+    tevent_req_done(req);
+}
+
+int sysdb_asq_search_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
+                          size_t *msgs_count, struct ldb_message ***msgs)
+{
+    struct sysdb_asq_search_state *state = tevent_req_data(req,
+                                              struct sysdb_asq_search_state);
+    enum tevent_req_state tstate;
+    uint64_t err;
+    int i;
+
+    if (tevent_req_is_error(req, &tstate, &err)) {
+        return err;
+    }
+
+    *msgs_count = state->msgs_count;
+    for (i = 0; i < state->msgs_count; i++) {
+        talloc_steal(mem_ctx, state->msgs[i]);
+    }
+    *msgs = talloc_move(mem_ctx, &state->msgs);
+
+    return EOK;
+}