source3/winbindd/winbindd_getuserdomgroups.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    async implementation of WINBINDD_GETUSERDOMGROUPS
   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 "../libcli/security/security.h"
  23
  24 struct winbindd_getuserdomgroups_state {
  25         struct dom_sid sid;
  26         uint32_t num_sids;
  27         struct dom_sid *sids;
  28 };
  29
  30 static void winbindd_getuserdomgroups_done(struct tevent_req *subreq);
  31
  32 struct tevent_req *winbindd_getuserdomgroups_send(TALLOC_CTX *mem_ctx,
  33                                                   struct tevent_context *ev,
  34                                                   struct winbindd_cli_state *cli,
  35                                                   struct winbindd_request *request)
  36 {
  37         struct tevent_req *req, *subreq;
  38         struct winbindd_getuserdomgroups_state *state;
  39
  40         req = tevent_req_create(mem_ctx, &state,
  41                                 struct winbindd_getuserdomgroups_state);
  42         if (req == NULL) {
  43                 return NULL;
  44         }
  45
  46         /* Ensure null termination */
  47         request->data.sid[sizeof(request->data.sid)-1]='\0';
  48
  49         D_NOTICE("[%s (%u)] Winbind external command GETUSERDOMGROUPS start.\n"
  50                  "sid=%s\n",
  51                  cli->client_name,
  52                  (unsigned int)cli->pid,
  53                  request->data.sid);
  54
  55         if (!string_to_sid(&state->sid, request->data.sid)) {
  56                 D_WARNING("Could not get convert sid %s from string\n",
  57                           request->data.sid);
  58                 tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
  59                 return tevent_req_post(req, ev);
  60         }
  61
  62         subreq = wb_gettoken_send(state, ev, &state->sid, false);
  63         if (tevent_req_nomem(subreq, req)) {
  64                 return tevent_req_post(req, ev);
  65         }
  66         tevent_req_set_callback(subreq, winbindd_getuserdomgroups_done, req);
  67         return req;
  68 }
  69
  70 static void winbindd_getuserdomgroups_done(struct tevent_req *subreq)
  71 {
  72         struct tevent_req *req = tevent_req_callback_data(
  73                 subreq, struct tevent_req);
  74         struct winbindd_getuserdomgroups_state *state = tevent_req_data(
  75                 req, struct winbindd_getuserdomgroups_state);
  76         NTSTATUS status;
  77
  78         status = wb_gettoken_recv(subreq, state, &state->num_sids,
  79                                   &state->sids);
  80         TALLOC_FREE(subreq);
  81         if (tevent_req_nterror(req, status)) {
  82                 return;
  83         }
  84         tevent_req_done(req);
  85 }
  86
  87 NTSTATUS winbindd_getuserdomgroups_recv(struct tevent_req *req,
  88                                         struct winbindd_response *response)
  89 {
  90         struct winbindd_getuserdomgroups_state *state = tevent_req_data(
  91                 req, struct winbindd_getuserdomgroups_state);
  92         NTSTATUS status;
  93         uint32_t i;
  94         char *sidlist;
  95
  96         if (tevent_req_is_nterror(req, &status)) {
  97                 D_WARNING("Failed with %s.\n", nt_errstr(status));
  98                 return status;
  99         }
 100
 101         sidlist = talloc_strdup(response, "");
 102         if (sidlist == NULL) {
 103                 return NT_STATUS_NO_MEMORY;
 104         }
 105         D_NOTICE("Winbind external command GETUSERDOMGROUPS end.\n"
 106                  "Received %"PRIu32" entries.\n",
 107                  state->num_sids);
 108         for (i=0; i<state->num_sids; i++) {
 109                 struct dom_sid_buf tmp;
 110                 sidlist = talloc_asprintf_append_buffer(
 111                         sidlist, "%s\n",
 112                         dom_sid_str_buf(&state->sids[i], &tmp));
 113                 if (sidlist == NULL) {
 114                         return NT_STATUS_NO_MEMORY;
 115                 }
 116                 D_NOTICE("%"PRIu32": %s\n",
 117                          i, dom_sid_str_buf(&state->sids[i], &tmp));
 118         }
 119         response->extra_data.data = sidlist;
 120         response->length += talloc_get_size(sidlist);
 121         response->data.num_entries = state->num_sids;
 122         return NT_STATUS_OK;
 123 }