source3/winbindd/wb_gettoken.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    async gettoken
   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_wbint_c.h"
  23 #include "../libcli/security/security.h"
  24 #include "passdb/machine_sid.h"
  25
  26 struct wb_gettoken_state {
  27         struct tevent_context *ev;
  28         struct dom_sid usersid;
  29         int num_sids;
  30         struct dom_sid *sids;
  31 };
  32
  33 static bool wb_add_rids_to_sids(TALLOC_CTX *mem_ctx,
  34                                 int *pnum_sids, struct dom_sid **psids,
  35                                 const struct dom_sid *domain_sid,
  36                                 int num_rids, uint32_t *rids);
  37
  38 static void wb_gettoken_gotgroups(struct tevent_req *subreq);
  39 static void wb_gettoken_gotlocalgroups(struct tevent_req *subreq);
  40 static void wb_gettoken_gotbuiltins(struct tevent_req *subreq);
  41
  42 struct tevent_req *wb_gettoken_send(TALLOC_CTX *mem_ctx,
  43                                     struct tevent_context *ev,
  44                                     const struct dom_sid *sid)
  45 {
  46         struct tevent_req *req, *subreq;
  47         struct wb_gettoken_state *state;
  48         struct winbindd_domain *domain;
  49
  50         req = tevent_req_create(mem_ctx, &state, struct wb_gettoken_state);
  51         if (req == NULL) {
  52                 return NULL;
  53         }
  54         sid_copy(&state->usersid, sid);
  55         state->ev = ev;
  56
  57         domain = find_domain_from_sid_noinit(sid);
  58         if (domain == NULL) {
  59                 DEBUG(5, ("Could not find domain from SID %s\n",
  60                           sid_string_dbg(sid)));
  61                 tevent_req_nterror(req, NT_STATUS_NO_SUCH_USER);
  62                 return tevent_req_post(req, ev);
  63         }
  64
  65         if (lp_winbind_trusted_domains_only() && domain->primary) {
  66                 DEBUG(7, ("wb_gettoken: My domain -- rejecting getgroups() "
  67                           "for %s.\n", sid_string_tos(sid)));
  68                 tevent_req_nterror(req, NT_STATUS_NO_SUCH_USER);
  69                 return tevent_req_post(req, ev);
  70         }
  71
  72         subreq = wb_lookupusergroups_send(state, ev, domain, &state->usersid);
  73         if (tevent_req_nomem(subreq, req)) {
  74                 return tevent_req_post(req, ev);
  75         }
  76         tevent_req_set_callback(subreq, wb_gettoken_gotgroups, req);
  77         return req;
  78 }
  79
  80 static void wb_gettoken_gotgroups(struct tevent_req *subreq)
  81 {
  82         struct tevent_req *req = tevent_req_callback_data(
  83                 subreq, struct tevent_req);
  84         struct wb_gettoken_state *state = tevent_req_data(
  85                 req, struct wb_gettoken_state);
  86         struct dom_sid *sids;
  87         struct winbindd_domain *domain;
  88         NTSTATUS status;
  89
  90         status = wb_lookupusergroups_recv(subreq, state, &state->num_sids,
  91                                           &state->sids);
  92         TALLOC_FREE(subreq);
  93         if (!NT_STATUS_IS_OK(status)) {
  94                 tevent_req_nterror(req, status);
  95                 return;
  96         }
  97
  98         sids = talloc_realloc(state, state->sids, struct dom_sid,
  99                               state->num_sids + 1);
 100         if (tevent_req_nomem(sids, req)) {
 101                 return;
 102         }
 103         memmove(&sids[1], &sids[0], state->num_sids * sizeof(sids[0]));
 104         sid_copy(&sids[0], &state->usersid);
 105         state->num_sids += 1;
 106         state->sids = sids;
 107
 108         /*
 109          * Expand our domain's aliases
 110          */
 111         domain = find_domain_from_sid_noinit(get_global_sam_sid());
 112         if (domain == NULL) {
 113                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
 114                 return;
 115         }
 116
 117         subreq = wb_lookupuseraliases_send(state, state->ev, domain,
 118                                            state->num_sids, state->sids);
 119         if (tevent_req_nomem(subreq, req)) {
 120                 return;
 121         }
 122         tevent_req_set_callback(subreq, wb_gettoken_gotlocalgroups, req);
 123 }
 124
 125 static void wb_gettoken_gotlocalgroups(struct tevent_req *subreq)
 126 {
 127         struct tevent_req *req = tevent_req_callback_data(
 128                 subreq, struct tevent_req);
 129         struct wb_gettoken_state *state = tevent_req_data(
 130                 req, struct wb_gettoken_state);
 131         uint32_t num_rids;
 132         uint32_t *rids;
 133         struct winbindd_domain *domain;
 134         NTSTATUS status;
 135
 136         status = wb_lookupuseraliases_recv(subreq, state, &num_rids, &rids);
 137         TALLOC_FREE(subreq);
 138         if (!NT_STATUS_IS_OK(status)) {
 139                 tevent_req_nterror(req, status);
 140                 return;
 141         }
 142         domain = find_domain_from_sid_noinit(get_global_sam_sid());
 143         if (domain == NULL) {
 144                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
 145                 return;
 146         }
 147         if (!wb_add_rids_to_sids(state, &state->num_sids, &state->sids,
 148                                  &domain->sid, num_rids, rids)) {
 149                 tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
 150                 return;
 151         }
 152         TALLOC_FREE(rids);
 153
 154         /*
 155          * Now expand the builtin groups
 156          */
 157
 158         domain = find_builtin_domain();
 159         if (domain == NULL) {
 160                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
 161                 return;
 162         }
 163
 164         subreq = wb_lookupuseraliases_send(state, state->ev, domain,
 165                                            state->num_sids, state->sids);
 166         if (tevent_req_nomem(subreq, req)) {
 167                 return;
 168         }
 169         tevent_req_set_callback(subreq, wb_gettoken_gotbuiltins, req);
 170 }
 171
 172 static void wb_gettoken_gotbuiltins(struct tevent_req *subreq)
 173 {
 174         struct tevent_req *req = tevent_req_callback_data(
 175                 subreq, struct tevent_req);
 176         struct wb_gettoken_state *state = tevent_req_data(
 177                 req, struct wb_gettoken_state);
 178         uint32_t num_rids;
 179         uint32_t *rids;
 180         NTSTATUS status;
 181
 182         status = wb_lookupuseraliases_recv(subreq, state, &num_rids, &rids);
 183         TALLOC_FREE(subreq);
 184         if (!NT_STATUS_IS_OK(status)) {
 185                 tevent_req_nterror(req, status);
 186                 return;
 187         }
 188         if (!wb_add_rids_to_sids(state, &state->num_sids, &state->sids,
 189                                  &global_sid_Builtin, num_rids, rids)) {
 190                 tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
 191                 return;
 192         }
 193         tevent_req_done(req);
 194 }
 195
 196 NTSTATUS wb_gettoken_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
 197                           int *num_sids, struct dom_sid **sids)
 198 {
 199         struct wb_gettoken_state *state = tevent_req_data(
 200                 req, struct wb_gettoken_state);
 201         NTSTATUS status;
 202
 203         if (tevent_req_is_nterror(req, &status)) {
 204                 return status;
 205         }
 206         *num_sids = state->num_sids;
 207         *sids = talloc_move(mem_ctx, &state->sids);
 208         return NT_STATUS_OK;
 209 }
 210
 211 static bool wb_add_rids_to_sids(TALLOC_CTX *mem_ctx,
 212                                 int *pnum_sids, struct dom_sid **psids,
 213                                 const struct dom_sid *domain_sid,
 214                                 int num_rids, uint32_t *rids)
 215 {
 216         struct dom_sid *sids;
 217         int i;
 218
 219         sids = talloc_realloc(mem_ctx, *psids, struct dom_sid,
 220                               *pnum_sids + num_rids);
 221         if (sids == NULL) {
 222                 return false;
 223         }
 224         for (i=0; i<num_rids; i++) {
 225                 sid_compose(&sids[i+*pnum_sids], domain_sid, rids[i]);
 226         }
 227
 228         *pnum_sids += num_rids;
 229         *psids = sids;
 230         return true;
 231 }