source3/winbindd/wb_lookupusergroups.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    async lookupusergroups
   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 "winbindd.h"
  22 #include "librpc/gen_ndr/cli_wbint.h"
  23
  24 struct wb_lookupusergroups_state {
  25         struct tevent_context *ev;
  26         struct dom_sid sid;
  27         struct wbint_SidArray sids;
  28 };
  29
  30 static void wb_lookupusergroups_done(struct tevent_req *subreq);
  31
  32 struct tevent_req *wb_lookupusergroups_send(TALLOC_CTX *mem_ctx,
  33                                             struct tevent_context *ev,
  34                                             struct winbindd_domain *domain,
  35                                             const struct dom_sid *sid)
  36 {
  37         struct tevent_req *req, *subreq;
  38         struct wb_lookupusergroups_state *state;
  39         NTSTATUS status;
  40
  41         req = tevent_req_create(mem_ctx, &state,
  42                                 struct wb_lookupusergroups_state);
  43         if (req == NULL) {
  44                 return NULL;
  45         }
  46         sid_copy(&state->sid, sid);
  47
  48         status = wcache_lookup_usergroups(domain, state, sid,
  49                                           &state->sids.num_sids,
  50                                           &state->sids.sids);
  51         if (NT_STATUS_IS_OK(status)) {
  52                 tevent_req_done(req);
  53                 return tevent_req_post(req, ev);
  54         }
  55
  56         subreq = rpccli_wbint_LookupUserGroups_send(
  57                 state, ev, domain->child.rpccli, &state->sid, &state->sids);
  58         if (tevent_req_nomem(subreq, req)) {
  59                 return tevent_req_post(req, ev);
  60         }
  61         tevent_req_set_callback(subreq, wb_lookupusergroups_done, req);
  62         return req;
  63 }
  64
  65 static void wb_lookupusergroups_done(struct tevent_req *subreq)
  66 {
  67         struct tevent_req *req = tevent_req_callback_data(
  68                 subreq, struct tevent_req);
  69         struct wb_lookupusergroups_state *state = tevent_req_data(
  70                 req, struct wb_lookupusergroups_state);
  71         NTSTATUS status, result;
  72
  73         status = rpccli_wbint_LookupUserGroups_recv(subreq, state, &result);
  74         TALLOC_FREE(subreq);
  75         if (!NT_STATUS_IS_OK(status)) {
  76                 tevent_req_nterror(req, status);
  77                 return;
  78         }
  79         if (!NT_STATUS_IS_OK(result)) {
  80                 tevent_req_nterror(req, result);
  81                 return;
  82         }
  83         tevent_req_done(req);
  84 }
  85
  86 NTSTATUS wb_lookupusergroups_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
  87                                   int *num_sids, struct dom_sid **sids)
  88 {
  89         struct wb_lookupusergroups_state *state = tevent_req_data(
  90                 req, struct wb_lookupusergroups_state);
  91         NTSTATUS status;
  92
  93         if (tevent_req_is_nterror(req, &status)) {
  94                 return status;
  95         }
  96         *num_sids = state->sids.num_sids;
  97         *sids = talloc_move(mem_ctx, &state->sids.sids);
  98         return NT_STATUS_OK;
  99 }