torture: Fix copy and paste error in debug message.
[kai/samba.git] / source4 / winbind / wb_async_helpers.c
index 25d52a16b5f900d828253cc8cacaf290beb56392..2af8567fd4cfe1358d74238f4baa007a5557b4cb 100644 (file)
 */
 
 #include "includes.h"
+#include <tevent.h>
 #include "libcli/composite/composite.h"
 #include "winbind/wb_async_helpers.h"
 
-#include "lib/messaging/irpc.h"
-#include "librpc/gen_ndr/irpc.h"
-#include "auth/credentials/credentials.h"
 #include "libcli/security/security.h"
-#include "libcli/auth/libcli_auth.h"
-#include "librpc/gen_ndr/ndr_netlogon_c.h"
 #include "librpc/gen_ndr/ndr_lsa_c.h"
 #include "librpc/gen_ndr/ndr_samr_c.h"
 
 #include "winbind/wb_helper.h"
 
+
 struct lsa_lookupsids_state {
        struct composite_context *ctx;
-       int num_sids;
+       uint32_t num_sids;
        struct lsa_LookupSids r;
        struct lsa_SidArray sids;
        struct lsa_TransNameArray names;
+       struct lsa_RefDomainList *domains;
        uint32_t count;
        struct wb_sid_object **result;
 };
 
-static void lsa_lookupsids_recv_names(struct rpc_request *req);
+static void lsa_lookupsids_recv_names(struct tevent_req *subreq);
 
 struct composite_context *wb_lsa_lookupsids_send(TALLOC_CTX *mem_ctx,
                                                 struct dcerpc_pipe *lsa_pipe,
                                                 struct policy_handle *handle,
-                                                int num_sids,
+                                                uint32_t num_sids,
                                                 const struct dom_sid **sids)
 {
        struct composite_context *result;
-       struct rpc_request *req;
        struct lsa_lookupsids_state *state;
-       int i;
+       uint32_t i;
+       struct tevent_req *subreq;
 
        result = composite_create(mem_ctx, lsa_pipe->conn->event_ctx);
        if (result == NULL) goto failed;
@@ -76,6 +74,9 @@ struct composite_context *wb_lsa_lookupsids_send(TALLOC_CTX *mem_ctx,
                if (state->sids.sids[i].sid == NULL) goto failed;
        }
 
+       state->domains = talloc(state, struct lsa_RefDomainList);
+       if (state->domains == NULL) goto failed;
+
        state->count = 0;
        state->num_sids = num_sids;
        state->names.count = 0;
@@ -88,12 +89,15 @@ struct composite_context *wb_lsa_lookupsids_send(TALLOC_CTX *mem_ctx,
        state->r.in.count = &state->count;
        state->r.out.names = &state->names;
        state->r.out.count = &state->count;
+       state->r.out.domains = &state->domains;
 
-       req = dcerpc_lsa_LookupSids_send(lsa_pipe, state, &state->r);
-       if (req == NULL) goto failed;
+       subreq = dcerpc_lsa_LookupSids_r_send(state,
+                                             result->event_ctx,
+                                             lsa_pipe->binding_handle,
+                                             &state->r);
+       if (subreq == NULL) goto failed;
+       tevent_req_set_callback(subreq, lsa_lookupsids_recv_names, state);
 
-       req->async.callback = lsa_lookupsids_recv_names;
-       req->async.private_data = state;
        return result;
 
  failed:
@@ -101,14 +105,15 @@ struct composite_context *wb_lsa_lookupsids_send(TALLOC_CTX *mem_ctx,
        return NULL;
 }
 
-static void lsa_lookupsids_recv_names(struct rpc_request *req)
+static void lsa_lookupsids_recv_names(struct tevent_req *subreq)
 {
        struct lsa_lookupsids_state *state =
-               talloc_get_type(req->async.private_data,
-                               struct lsa_lookupsids_state);
-       int i;
+               tevent_req_callback_data(subreq,
+               struct lsa_lookupsids_state);
+       uint32_t i;
 
-       state->ctx->status = dcerpc_ndr_request_recv(req);
+       state->ctx->status = dcerpc_lsa_LookupSids_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
        if (!composite_is_ok(state->ctx)) return;
        state->ctx->status = state->r.out.result;
        if (!NT_STATUS_IS_OK(state->ctx->status) &&
@@ -125,6 +130,8 @@ static void lsa_lookupsids_recv_names(struct rpc_request *req)
                struct lsa_TranslatedName *name =
                        &state->r.out.names->names[i];
                struct lsa_DomainInfo *dom;
+               struct lsa_RefDomainList *domains =
+                       state->domains;
 
                state->result[i] = talloc_zero(state->result,
                                               struct wb_sid_object);
@@ -135,13 +142,13 @@ static void lsa_lookupsids_recv_names(struct rpc_request *req)
                        continue;
                }
 
-               if (name->sid_index >= state->r.out.domains->count) {
+               if (name->sid_index >= domains->count) {
                        composite_error(state->ctx,
                                        NT_STATUS_INVALID_PARAMETER);
                        return;
                }
 
-               dom = &state->r.out.domains->domains[name->sid_index];
+               dom = &domains->domains[name->sid_index];
                state->result[i]->domain = talloc_reference(state->result[i],
                                                            dom->name.string);
                if ((name->sid_type == SID_NAME_DOMAIN) ||
@@ -183,24 +190,25 @@ struct lsa_lookupnames_state {
        uint32_t num_names;
        struct lsa_LookupNames r;
        struct lsa_TransSidArray sids;
+       struct lsa_RefDomainList *domains;
        uint32_t count;
        struct wb_sid_object **result;
 };
 
-static void lsa_lookupnames_recv_sids(struct rpc_request *req);
+static void lsa_lookupnames_recv_sids(struct tevent_req *subreq);
 
 struct composite_context *wb_lsa_lookupnames_send(TALLOC_CTX *mem_ctx,
                                                  struct dcerpc_pipe *lsa_pipe,
                                                  struct policy_handle *handle,
-                                                 int num_names,
+                                                 uint32_t num_names,
                                                  const char **names)
 {
        struct composite_context *result;
-       struct rpc_request *req;
        struct lsa_lookupnames_state *state;
+       struct tevent_req *subreq;
 
        struct lsa_String *lsa_names;
-       int i;
+       uint32_t i;
 
        result = composite_create(mem_ctx, lsa_pipe->conn->event_ctx);
        if (result == NULL) goto failed;
@@ -222,6 +230,9 @@ struct composite_context *wb_lsa_lookupnames_send(TALLOC_CTX *mem_ctx,
                lsa_names[i].string = names[i];
        }
 
+       state->domains = talloc(state, struct lsa_RefDomainList);
+       if (state->domains == NULL) goto failed;
+
        state->r.in.handle = handle;
        state->r.in.num_names = num_names;
        state->r.in.names = lsa_names;
@@ -230,12 +241,15 @@ struct composite_context *wb_lsa_lookupnames_send(TALLOC_CTX *mem_ctx,
        state->r.in.count = &state->count;
        state->r.out.count = &state->count;
        state->r.out.sids = &state->sids;
+       state->r.out.domains = &state->domains;
 
-       req = dcerpc_lsa_LookupNames_send(lsa_pipe, state, &state->r);
-       if (req == NULL) goto failed;
+       subreq = dcerpc_lsa_LookupNames_r_send(state,
+                                              result->event_ctx,
+                                              lsa_pipe->binding_handle,
+                                              &state->r);
+       if (subreq == NULL) goto failed;
+       tevent_req_set_callback(subreq, lsa_lookupnames_recv_sids, state);
 
-       req->async.callback = lsa_lookupnames_recv_sids;
-       req->async.private_data = state;
        return result;
 
  failed:
@@ -243,14 +257,15 @@ struct composite_context *wb_lsa_lookupnames_send(TALLOC_CTX *mem_ctx,
        return NULL;
 }
 
-static void lsa_lookupnames_recv_sids(struct rpc_request *req)
+static void lsa_lookupnames_recv_sids(struct tevent_req *subreq)
 {
        struct lsa_lookupnames_state *state =
-               talloc_get_type(req->async.private_data,
-                               struct lsa_lookupnames_state);
-       int i;
+               tevent_req_callback_data(subreq,
+               struct lsa_lookupnames_state);
+       uint32_t i;
 
-       state->ctx->status = dcerpc_ndr_request_recv(req);
+       state->ctx->status = dcerpc_lsa_LookupNames_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
        if (!composite_is_ok(state->ctx)) return;
        state->ctx->status = state->r.out.result;
        if (!NT_STATUS_IS_OK(state->ctx->status) &&
@@ -265,6 +280,7 @@ static void lsa_lookupnames_recv_sids(struct rpc_request *req)
 
        for (i=0; i<state->num_names; i++) {
                struct lsa_TranslatedSid *sid = &state->r.out.sids->sids[i];
+               struct lsa_RefDomainList *domains = state->domains;
                struct lsa_DomainInfo *dom;
 
                state->result[i] = talloc_zero(state->result,
@@ -276,13 +292,13 @@ static void lsa_lookupnames_recv_sids(struct rpc_request *req)
                        continue;
                }
 
-               if (sid->sid_index >= state->r.out.domains->count) {
+               if (sid->sid_index >= domains->count) {
                        composite_error(state->ctx,
                                        NT_STATUS_INVALID_PARAMETER);
                        return;
                }
 
-               dom = &state->r.out.domains->domains[sid->sid_index];
+               dom = &domains->domains[sid->sid_index];
 
                state->result[i]->sid = dom_sid_add_rid(state->result[i],
                                                        dom->sid, sid->rid);
@@ -309,18 +325,20 @@ struct samr_getuserdomgroups_state {
        struct composite_context *ctx;
        struct dcerpc_pipe *samr_pipe;
 
-       int num_rids;
+       uint32_t num_rids;
        uint32_t *rids;
 
+       struct samr_RidWithAttributeArray *rid_array;
+
        struct policy_handle *user_handle;
        struct samr_OpenUser o;
        struct samr_GetGroupsForUser g;
        struct samr_Close c;
 };
 
-static void samr_usergroups_recv_open(struct rpc_request *req);
-static void samr_usergroups_recv_groups(struct rpc_request *req);
-static void samr_usergroups_recv_close(struct rpc_request *req);
+static void samr_usergroups_recv_open(struct tevent_req *subreq);
+static void samr_usergroups_recv_groups(struct tevent_req *subreq);
+static void samr_usergroups_recv_close(struct tevent_req *subreq);
 
 struct composite_context *wb_samr_userdomgroups_send(TALLOC_CTX *mem_ctx,
                                                     struct dcerpc_pipe *samr_pipe,
@@ -328,8 +346,8 @@ struct composite_context *wb_samr_userdomgroups_send(TALLOC_CTX *mem_ctx,
                                                     uint32_t rid)
 {
        struct composite_context *result;
-       struct rpc_request *req;
        struct samr_getuserdomgroups_state *state;
+       struct tevent_req *subreq;
 
        result = composite_create(mem_ctx, samr_pipe->conn->event_ctx);
        if (result == NULL) goto failed;
@@ -349,11 +367,13 @@ struct composite_context *wb_samr_userdomgroups_send(TALLOC_CTX *mem_ctx,
        state->o.in.rid = rid;
        state->o.out.user_handle = state->user_handle;
 
-       req = dcerpc_samr_OpenUser_send(state->samr_pipe, state, &state->o);
-       if (req == NULL) goto failed;
+       subreq = dcerpc_samr_OpenUser_r_send(state,
+                                            result->event_ctx,
+                                            state->samr_pipe->binding_handle,
+                                            &state->o);
+       if (subreq == NULL) goto failed;
+       tevent_req_set_callback(subreq, samr_usergroups_recv_open, state);
 
-       req->async.callback = samr_usergroups_recv_open;
-       req->async.private_data = state;
        return result;
 
  failed:
@@ -361,32 +381,37 @@ struct composite_context *wb_samr_userdomgroups_send(TALLOC_CTX *mem_ctx,
        return NULL;
 }
                                              
-static void samr_usergroups_recv_open(struct rpc_request *req)
+static void samr_usergroups_recv_open(struct tevent_req *subreq)
 {
        struct samr_getuserdomgroups_state *state =
-               talloc_get_type(req->async.private_data,
-                               struct samr_getuserdomgroups_state);
+               tevent_req_callback_data(subreq,
+               struct samr_getuserdomgroups_state);
 
-       state->ctx->status = dcerpc_ndr_request_recv(req);
+       state->ctx->status = dcerpc_samr_OpenUser_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
        if (!composite_is_ok(state->ctx)) return;
        state->ctx->status = state->o.out.result;
        if (!composite_is_ok(state->ctx)) return;
 
        state->g.in.user_handle = state->user_handle;
-
-       req = dcerpc_samr_GetGroupsForUser_send(state->samr_pipe, state,
-                                               &state->g);
-       composite_continue_rpc(state->ctx, req, samr_usergroups_recv_groups,
-                              state);
+       state->g.out.rids = &state->rid_array;
+
+       subreq = dcerpc_samr_GetGroupsForUser_r_send(state,
+                                                    state->ctx->event_ctx,
+                                                    state->samr_pipe->binding_handle,
+                                                    &state->g);
+       if (composite_nomem(subreq, state->ctx)) return;
+       tevent_req_set_callback(subreq, samr_usergroups_recv_groups, state);
 }
 
-static void samr_usergroups_recv_groups(struct rpc_request *req)
+static void samr_usergroups_recv_groups(struct tevent_req *subreq)
 {
        struct samr_getuserdomgroups_state *state =
-               talloc_get_type(req->async.private_data,
-                               struct samr_getuserdomgroups_state);
+               tevent_req_callback_data(subreq,
+               struct samr_getuserdomgroups_state);
 
-       state->ctx->status = dcerpc_ndr_request_recv(req);
+       state->ctx->status = dcerpc_samr_GetGroupsForUser_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
        if (!composite_is_ok(state->ctx)) return;
        state->ctx->status = state->g.out.result;
        if (!composite_is_ok(state->ctx)) return;
@@ -394,18 +419,22 @@ static void samr_usergroups_recv_groups(struct rpc_request *req)
        state->c.in.handle = state->user_handle;
        state->c.out.handle = state->user_handle;
 
-       req = dcerpc_samr_Close_send(state->samr_pipe, state, &state->c);
-       composite_continue_rpc(state->ctx, req, samr_usergroups_recv_close,
-                              state);
+       subreq = dcerpc_samr_Close_r_send(state,
+                                         state->ctx->event_ctx,
+                                         state->samr_pipe->binding_handle,
+                                         &state->c);
+       if (composite_nomem(subreq, state->ctx)) return;
+       tevent_req_set_callback(subreq, samr_usergroups_recv_close, state);
 }
 
-static void samr_usergroups_recv_close(struct rpc_request *req)
+static void samr_usergroups_recv_close(struct tevent_req *subreq)
 {
         struct samr_getuserdomgroups_state *state =
-                talloc_get_type(req->async.private_data,
-                                struct samr_getuserdomgroups_state);
+               tevent_req_callback_data(subreq,
+               struct samr_getuserdomgroups_state);
 
-        state->ctx->status = dcerpc_ndr_request_recv(req);
+       state->ctx->status = dcerpc_samr_Close_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
         if (!composite_is_ok(state->ctx)) return;
         state->ctx->status = state->c.out.result;
         if (!composite_is_ok(state->ctx)) return;
@@ -415,17 +444,17 @@ static void samr_usergroups_recv_close(struct rpc_request *req)
 
 NTSTATUS wb_samr_userdomgroups_recv(struct composite_context *ctx,
                                    TALLOC_CTX *mem_ctx,
-                                   int *num_rids, uint32_t **rids)
+                                   uint32_t *num_rids, uint32_t **rids)
 {
         struct samr_getuserdomgroups_state *state =
                 talloc_get_type(ctx->private_data,
                                 struct samr_getuserdomgroups_state);
 
-       int i;
+       uint32_t i;
        NTSTATUS status = composite_wait(ctx);
        if (!NT_STATUS_IS_OK(status)) goto done;
 
-       *num_rids = state->g.out.rids->count;
+       *num_rids = state->rid_array->count;
        *rids = talloc_array(mem_ctx, uint32_t, *num_rids);
        if (*rids == NULL) {
                status = NT_STATUS_NO_MEMORY;
@@ -433,7 +462,7 @@ NTSTATUS wb_samr_userdomgroups_recv(struct composite_context *ctx,
        }
 
        for (i=0; i<*num_rids; i++) {
-               (*rids)[i] = state->g.out.rids->rids[i].rid;
+               (*rids)[i] = state->rid_array->rids[i].rid;
        }
 
  done: