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