Converted some more functions to create and dispose of a talloc context on a
[samba.git] / source3 / nsswitch / winbindd_group.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 2.0
4
5    Winbind daemon for ntdom nss module
6
7    Copyright (C) Tim Potter 2000
8    
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 2 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program; if not, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24 #include "winbindd.h"
25
26 /* Fill a grent structure from various other information */
27
28 static BOOL fill_grent(struct winbindd_gr *gr, char *gr_name,
29                        gid_t unix_gid)
30 {
31         /* Fill in uid/gid */
32
33         gr->gr_gid = unix_gid;
34     
35         /* Group name and password */
36     
37         safe_strcpy(gr->gr_name, gr_name, sizeof(gr->gr_name) - 1);
38         safe_strcpy(gr->gr_passwd, "x", sizeof(gr->gr_passwd) - 1);
39
40         return True;
41 }
42
43 /* Fill in the group membership field of a NT group given by group_rid */
44
45 static BOOL fill_grent_mem(struct winbindd_domain *domain,
46                            uint32 group_rid, 
47                            enum SID_NAME_USE group_name_type, 
48                            int *num_gr_mem, char **gr_mem, int *gr_mem_len)
49 {
50         uint32 *rid_mem = NULL, num_names = 0;
51         uint32 *name_types = NULL;
52         int buf_len, buf_ndx, i;
53         char **names = NULL, *buf;
54         BOOL result = False;
55         TALLOC_CTX *mem_ctx;
56
57         if (!(mem_ctx = talloc_init()))
58                 return False;
59
60         /* Initialise group membership information */
61         
62         DEBUG(10, ("fill_grent_mem(): group %s rid 0x%x\n",
63                    domain ? domain->name : "NULL", group_rid));
64
65         *num_gr_mem = 0;
66         
67         if (group_name_type != SID_NAME_DOM_GRP) {
68                 DEBUG(1, ("fill_grent_mem(): rid %d in domain %s isn't a "
69                           "domain group\n", group_rid, domain->name));
70                 goto done;
71         }
72
73         /* Lookup group members */
74
75         if (!winbindd_lookup_groupmem(domain, mem_ctx, group_rid, &num_names, 
76                                       &rid_mem, &names, &name_types)) {
77
78                 DEBUG(1, ("fill_grent_mem(): could not lookup membership "
79                           "for group rid %d in domain %s\n", 
80                           group_rid, domain->name));
81
82                 goto done;
83         }
84
85         DEBUG(10, ("fill_grent_mem(): looked up %d names\n", num_names));
86
87         if (DEBUGLEVEL >= 10) {
88                 for (i = 0; i < num_names; i++)
89                         DEBUG(10, ("\t%20s %x %d\n", names[i], rid_mem[i],
90                                    name_types[i]));
91         }
92
93         /* Add members to list */
94
95         buf = NULL;
96         buf_len = buf_ndx = 0;
97
98  again:
99
100         for (i = 0; i < num_names; i++) {
101                 char *the_name;
102                 fstring name;
103                 int len;
104                         
105                 the_name = names[i];
106
107                 DEBUG(10, ("fill_grent_mem(): processing name %s\n", 
108                            the_name));
109
110                 /* Only add domain users */
111
112                 if (name_types[i] != SID_NAME_USER) {
113                         DEBUG(3, ("fill_grent_mem(): name %s isn't a domain "
114                                   "user\n", the_name));
115                         continue;
116                 }
117
118                 /* Don't bother with machine accounts */
119                 
120                 if (the_name[strlen(the_name) - 1] == '$') {
121                         DEBUG(10, ("fill_grent_mem(): %s is machine account\n",
122                                    the_name));
123                         continue;
124                 }
125
126                 /* Append domain name */
127
128                 snprintf(name, sizeof(name), "%s%s%s", domain->name,
129                          lp_winbind_separator(), the_name);
130
131                 len = strlen(name);
132                 
133                 /* Add to list or calculate buffer length */
134
135                 if (!buf) {
136                         buf_len += len + 1; /* List is comma separated */
137                         (*num_gr_mem)++;
138                         DEBUG(10, ("fill_grent_mem(): buf_len + %d = %d\n", len + 1,
139                                    buf_len));
140                 } else {
141                         DEBUG(10, ("fill_grent_mem(): appending %s at index %d\n",
142                                    name, len));
143                         safe_strcpy(&buf[buf_ndx], name, len);
144                         buf_ndx += len;
145                         buf[buf_ndx] = ',';
146                         buf_ndx++;
147                 }
148         }
149
150         /* Allocate buffer */
151
152         if (!buf) {
153                 if (!(buf = malloc(buf_len))) {
154                         DEBUG(1, ("fill_grent_mem(): out of memory\n"));
155                         result = False;
156                         goto done;
157                 }
158                 memset(buf, 0, buf_len);
159                 goto again;
160         }
161
162         if (buf && buf_ndx > 0) {
163                 buf[buf_ndx - 1] = '\0';
164         }
165
166         *gr_mem = buf;
167         *gr_mem_len = buf_len;
168
169         DEBUG(10, ("fill_grent_mem(): num_mem = %d, len = %d, mem = %s\n", *num_gr_mem,
170                    buf_len, buf));
171
172         result = True;
173
174 done:
175         talloc_destroy(mem_ctx);
176         
177         DEBUG(10, ("fill_grent_mem(): returning %d\n", result));
178
179         return result;
180 }
181
182 /* Return a group structure from a group name */
183
184 enum winbindd_result winbindd_getgrnam_from_group(struct winbindd_cli_state 
185                                                   *state)
186 {
187         DOM_SID group_sid;
188         struct winbindd_domain *domain;
189         enum SID_NAME_USE name_type;
190         uint32 group_rid;
191         fstring name_domain, name_group, name;
192         char *tmp, *gr_mem;
193         gid_t gid;
194         int extra_data_len, gr_mem_len;
195         
196         DEBUG(3, ("[%5d]: getgrnam %s\n", state->pid,
197                   state->request.data.groupname));
198
199         /* Parse domain and groupname */
200         
201         memset(name_group, 0, sizeof(fstring));
202
203         tmp = state->request.data.groupname;
204         parse_domain_user(tmp, name_domain, name_group);
205
206         /* Reject names that don't have a domain - i.e name_domain contains 
207            the entire name. */
208
209         if (strequal(name_group, ""))
210                 return WINBINDD_ERROR;
211
212         /* Get info for the domain */
213
214         if ((domain = find_domain_from_name(name_domain)) == NULL) {
215                 DEBUG(0, ("getgrname_from_group(): could not get domain "
216                           "sid for domain %s\n", name_domain));
217                 return WINBINDD_ERROR;
218         }
219
220         /* Check for cached group entry */
221
222         if (winbindd_fetch_group_cache_entry(domain, name_group,
223                                              &state->response.data.gr,
224                                              &state->response.extra_data,
225                                              &extra_data_len)) {
226                 state->response.length += extra_data_len;
227                 return WINBINDD_OK;
228         }
229
230         snprintf(name, sizeof(name), "%s\\%s", name_domain, name_group);
231
232         /* Get rid and name type from name */
233         
234         if (!winbindd_lookup_sid_by_name(name, &group_sid, &name_type)) {
235                 DEBUG(1, ("group %s in domain %s does not exist\n", 
236                           name_group, name_domain));
237                 return WINBINDD_ERROR;
238         }
239
240         if ((name_type != SID_NAME_ALIAS) && (name_type != SID_NAME_DOM_GRP)) {
241                 DEBUG(1, ("from_group: name '%s' is not a local or domain "
242                           "group: %d\n", name_group, name_type));
243                 return WINBINDD_ERROR;
244         }
245
246         /* Fill in group structure */
247
248         sid_split_rid(&group_sid, &group_rid);
249
250         if (!winbindd_idmap_get_gid_from_rid(domain->name, group_rid, &gid)) {
251                 DEBUG(1, ("error sursing unix gid for sid\n"));
252                 return WINBINDD_ERROR;
253         }
254
255         if (!fill_grent(&state->response.data.gr, 
256                         state->request.data.groupname, gid) ||
257             !fill_grent_mem(domain, group_rid, name_type,
258                             &state->response.data.gr.num_gr_mem,
259                             &gr_mem, &gr_mem_len))
260                 return WINBINDD_ERROR;
261
262         /* Group membership lives at start of extra data */
263
264         state->response.data.gr.gr_mem_ofs = 0;
265
266         state->response.length += gr_mem_len;
267         state->response.extra_data = gr_mem;
268
269         /* Update cached group info */
270
271         winbindd_store_group_cache_entry(domain, name_group, 
272                                          &state->response.data.gr,
273                                          state->response.extra_data,
274                                          gr_mem_len);
275
276         return WINBINDD_OK;
277 }
278
279 /* Return a group structure from a gid number */
280
281 enum winbindd_result winbindd_getgrnam_from_gid(struct winbindd_cli_state 
282                                                 *state)
283 {
284         struct winbindd_domain *domain;
285         DOM_SID group_sid;
286         enum SID_NAME_USE name_type;
287         fstring group_name;
288         uint32 group_rid;
289         int extra_data_len, gr_mem_len;
290         char *gr_mem;
291
292         DEBUG(3, ("[%5d]: getgrgid %d\n", state->pid, 
293                   state->request.data.gid));
294
295         /* Bug out if the gid isn't in the winbind range */
296
297         if ((state->request.data.gid < server_state.gid_low) ||
298             (state->request.data.gid > server_state.gid_high))
299                 return WINBINDD_ERROR;
300
301         /* Get rid from gid */
302
303         if (!winbindd_idmap_get_rid_from_gid(state->request.data.gid, 
304                                              &group_rid, &domain)) {
305                 DEBUG(1, ("Could not convert gid %d to rid\n", 
306                           state->request.data.gid));
307                 return WINBINDD_ERROR;
308         }
309
310         /* Try a cached entry */
311
312         if (winbindd_fetch_gid_cache_entry(domain, 
313                                            state->request.data.gid,
314                                            &state->response.data.gr,
315                                            &state->response.extra_data,
316                                            &extra_data_len)) {
317                 state->response.length += extra_data_len;
318                 return WINBINDD_OK;
319         }
320
321         /* Get sid from gid */
322
323         sid_copy(&group_sid, &domain->sid);
324         sid_append_rid(&group_sid, group_rid);
325
326         if (!winbindd_lookup_name_by_sid(&group_sid, group_name, &name_type)) {
327                 DEBUG(1, ("Could not lookup sid\n"));
328                 return WINBINDD_ERROR;
329         }
330
331         if (strcmp(lp_winbind_separator(),"\\"))
332                 string_sub(group_name, "\\", lp_winbind_separator(), 
333                            sizeof(fstring));
334
335         if (!((name_type == SID_NAME_ALIAS) || 
336               (name_type == SID_NAME_DOM_GRP))) {
337                 DEBUG(1, ("from_gid: name '%s' is not a local or domain "
338                           "group: %d\n", group_name, name_type));
339                 return WINBINDD_ERROR;
340         }
341
342         /* Fill in group structure */
343
344         if (!fill_grent(&state->response.data.gr, group_name, 
345                         state->request.data.gid) ||
346             !fill_grent_mem(domain, group_rid, name_type,
347                             &state->response.data.gr.num_gr_mem,
348                             &gr_mem, &gr_mem_len))
349                 return WINBINDD_ERROR;
350
351         /* Group membership lives at start of extra data */
352
353         state->response.data.gr.gr_mem_ofs = 0;
354
355         state->response.length += gr_mem_len;
356         state->response.extra_data = gr_mem;
357
358         /* Update cached group info */
359
360         winbindd_store_gid_cache_entry(domain, state->request.data.gid,
361                                        &state->response.data.gr,
362                                        state->response.extra_data,
363                                        gr_mem_len);
364
365         return WINBINDD_OK;
366 }
367
368 /*
369  * set/get/endgrent functions
370  */
371
372 /* "Rewind" file pointer for group database enumeration */
373
374 enum winbindd_result winbindd_setgrent(struct winbindd_cli_state *state)
375 {
376         struct winbindd_domain *tmp;
377
378         DEBUG(3, ("[%5d]: setgrent\n", state->pid));
379
380         /* Check user has enabled this */
381
382         if (!lp_winbind_enum_groups())
383                 return WINBINDD_ERROR;
384
385         /* Free old static data if it exists */
386         
387         if (state->getgrent_state != NULL) {
388                 free_getent_state(state->getgrent_state);
389                 state->getgrent_state = NULL;
390         }
391         
392         /* Create sam pipes for each domain we know about */
393         
394         for (tmp = domain_list; tmp != NULL; tmp = tmp->next) {
395                 struct getent_state *domain_state;
396                 
397                 /* Skip domains other than WINBINDD_DOMAIN environment 
398                    variable */
399                 
400                 if ((strcmp(state->request.domain, "") != 0) &&
401                     !check_domain_env(state->request.domain, tmp->name))
402                         continue;
403                 
404                 /* Create a state record for this domain */
405                 
406                 if ((domain_state = (struct getent_state *)
407                      malloc(sizeof(struct getent_state))) == NULL)
408                         return WINBINDD_ERROR;
409                 
410                 ZERO_STRUCTP(domain_state);
411                 
412                 domain_state->domain = tmp;
413                 domain_state->mem_ctx = talloc_init();
414
415                 /* Add to list of open domains */
416                 
417                 DLIST_ADD(state->getgrent_state, domain_state);
418         }
419         
420         return WINBINDD_OK;
421 }
422
423 /* Close file pointer to ntdom group database */
424
425 enum winbindd_result winbindd_endgrent(struct winbindd_cli_state *state)
426 {
427         DEBUG(3, ("[%5d]: endgrent\n", state->pid));
428
429         free_getent_state(state->getgrent_state);
430         state->getgrent_state = NULL;
431         
432         return WINBINDD_OK;
433 }
434
435 /* Get the list of domain groups and domain aliases for a domain.  We fill in
436    the sam_entries and num_sam_entries fields with domain group information.  
437    The dispinfo_ndx field is incremented to the index of the next group to 
438    fetch. Return True if some groups were returned, False otherwise. */
439
440 #define MAX_FETCH_SAM_ENTRIES 100
441
442 static BOOL get_sam_group_entries(struct getent_state *ent)
443 {
444         NTSTATUS status;
445         uint32 num_entries;
446         struct acct_info *name_list = NULL, *tnl;
447         
448         if (ent->got_all_sam_entries)
449                 return False;
450
451 #if 0
452         if (winbindd_fetch_group_cache(ent->domain, 
453                                        &ent->sam_entries,
454                                        &ent->num_sam_entries))
455                 return True;
456 #endif
457                 
458         /* Free any existing group info */
459
460         SAFE_FREE(ent->sam_entries);
461         ent->num_sam_entries = 0;
462                 
463         /* Enumerate domain groups */
464                 
465         do {
466                 struct acct_info *sam_grp_entries = NULL;
467                 CLI_POLICY_HND *hnd;
468
469                 num_entries = 0;
470
471                 if (!(hnd = cm_get_sam_dom_handle(ent->domain->name,
472                                                   &ent->domain->sid)))
473                         break;
474
475                 status = cli_samr_enum_dom_groups(
476                         hnd->cli, ent->mem_ctx, &hnd->pol,
477                         &ent->grp_query_start_ndx,
478                         0x8000, /* buffer size? */
479                         (struct acct_info **) &sam_grp_entries, &num_entries);
480
481                 /* Copy entries into return buffer */
482
483                 if (num_entries) {
484
485                         tnl = Realloc(name_list,
486                                     sizeof(struct acct_info) *
487                                     (ent->num_sam_entries +
488                                      num_entries));
489
490                         if (tnl == NULL) {
491                                 DEBUG(0,("get_sam_group_entries: unable to "
492                                          "realloc a structure!\n"));
493                                 SAFE_FREE(name_list);
494
495                                 return False; 
496                         } else 
497                                 name_list = tnl;
498
499                         memcpy(&name_list[ent->num_sam_entries],
500                                sam_grp_entries, 
501                                num_entries * sizeof(struct acct_info));
502
503                         SAFE_FREE(sam_grp_entries);
504                 }
505
506                 ent->num_sam_entries += num_entries;
507                 
508                 if (NT_STATUS_V(status) != NT_STATUS_V(STATUS_MORE_ENTRIES))
509                         break;
510
511         } while (ent->num_sam_entries < MAX_FETCH_SAM_ENTRIES);
512                 
513 #if 0
514         /* Fill cache with received entries */
515
516         winbindd_store_group_cache(ent->domain, ent->sam_entries,
517                                    ent->num_sam_entries);
518 #endif
519
520         /* Fill in remaining fields */
521
522         ent->sam_entries = name_list;
523         ent->sam_entry_index = 0;
524         ent->got_all_sam_entries = (NT_STATUS_V(status) != 
525                                     NT_STATUS_V(STATUS_MORE_ENTRIES));
526
527         return ent->num_sam_entries > 0;
528 }
529
530 /* Fetch next group entry from ntdom database */
531
532 #define MAX_GETGRENT_GROUPS 500
533
534 enum winbindd_result winbindd_getgrent(struct winbindd_cli_state *state)
535 {
536         struct getent_state *ent;
537         struct winbindd_gr *group_list = NULL;
538         int num_groups, group_list_ndx = 0, i, gr_mem_list_len = 0;
539         char *sep, *new_extra_data, *gr_mem_list = NULL;
540
541         DEBUG(3, ("[%5d]: getgrent\n", state->pid));
542
543         /* Check user has enabled this */
544
545         if (!lp_winbind_enum_groups())
546                 return WINBINDD_ERROR;
547
548         num_groups = MIN(MAX_GETGRENT_GROUPS, state->request.data.num_entries);
549
550         if ((state->response.extra_data = 
551              malloc(num_groups * sizeof(struct winbindd_gr))) == NULL)
552                 return WINBINDD_ERROR;
553
554         state->response.data.num_entries = 0;
555
556         group_list = (struct winbindd_gr *)state->response.extra_data;
557         sep = lp_winbind_separator();
558
559         if (!(ent = state->getgrent_state))
560                 return WINBINDD_ERROR;
561
562         /* Start sending back groups */
563
564         for (i = 0; i < num_groups; i++) {
565                 struct acct_info *name_list = NULL;
566                 fstring domain_group_name;
567                 uint32 result;
568                 gid_t group_gid;
569                 
570                 /* Do we need to fetch another chunk of groups? */
571
572         tryagain:
573
574                 DEBUG(10, ("getgrent(): entry_index = %d, num_entries = %d\n",
575                            ent->sam_entry_index, ent->num_sam_entries));
576
577                 if (ent->num_sam_entries == ent->sam_entry_index) {
578
579                         while(ent && !get_sam_group_entries(ent)) {
580                                 struct getent_state *next_ent;
581
582                                 DEBUG(10, ("getgrent(): freeing state info for "
583                                            "domain %s\n", ent->domain->name)); 
584
585                                 /* Free state information for this domain */
586
587                                 SAFE_FREE(ent->sam_entries);
588                                 ent->sam_entries = NULL;
589
590                                 next_ent = ent->next;
591                                 DLIST_REMOVE(state->getgrent_state, ent);
592                                 
593                                 SAFE_FREE(ent);
594                                 ent = next_ent;
595                         }
596
597                         /* No more domains */
598
599                         if (!ent) 
600                                 break;
601                 }
602                 
603                 name_list = ent->sam_entries;
604                 
605                 /* Lookup group info */
606                 
607                 if (!winbindd_idmap_get_gid_from_rid(
608                         ent->domain->name,
609                         name_list[ent->sam_entry_index].rid,
610                         &group_gid)) {
611                         
612                         DEBUG(1, ("getgrent(): could not look up gid for group %s\n",
613                                   name_list[ent->sam_entry_index].acct_name));
614
615                         ent->sam_entry_index++;
616                         goto tryagain;
617                 }
618
619                 DEBUG(10, ("getgrent(): got gid %d for group %x\n", group_gid,
620                            name_list[ent->sam_entry_index].rid));
621                 
622                 /* Fill in group entry */
623
624                 slprintf(domain_group_name, sizeof(domain_group_name) - 1,
625                          "%s%s%s", ent->domain->name, lp_winbind_separator(), 
626                          name_list[ent->sam_entry_index].acct_name);
627    
628                 result = fill_grent(&group_list[group_list_ndx], 
629                                     domain_group_name, group_gid);
630
631                 /* Fill in group membership entry */
632
633                 if (result) {
634                         int gr_mem_len;
635                         char *gr_mem, *new_gr_mem_list;
636
637                         /* Get group membership */
638
639                         result = fill_grent_mem(
640                                 ent->domain,
641                                 name_list[ent->sam_entry_index].rid,
642                                 SID_NAME_DOM_GRP,
643                                 &group_list[group_list_ndx].num_gr_mem, 
644                                 &gr_mem, &gr_mem_len);
645
646                         /* Append to group membership list */
647
648                         new_gr_mem_list = Realloc(
649                                 gr_mem_list,
650                                 gr_mem_list_len + gr_mem_len);
651
652                         if (!new_gr_mem_list && (group_list[group_list_ndx].num_gr_mem != 0)) {
653                                 DEBUG(0, ("getgrent(): out of memory\n"));
654                                 SAFE_FREE(gr_mem_list);
655                                 gr_mem_list_len = 0;
656                                 break;
657                         }
658
659                         DEBUG(10, ("getgrent(): list_len = %d, mem_len = %d\n",
660                                    gr_mem_list_len, gr_mem_len));
661
662                         gr_mem_list = new_gr_mem_list;
663
664                         memcpy(&gr_mem_list[gr_mem_list_len], gr_mem,
665                                gr_mem_len);
666
667                         SAFE_FREE(gr_mem);
668
669                         group_list[group_list_ndx].gr_mem_ofs = 
670                                 gr_mem_list_len;
671
672                         gr_mem_list_len += gr_mem_len;
673                 }
674
675                 ent->sam_entry_index++;
676                 
677                 /* Add group to return list */
678                 
679                 if (result) {
680
681                         DEBUG(10, ("getgrent(): adding group num_entries = %d\n",
682                                    state->response.data.num_entries));
683
684                         group_list_ndx++;
685                         state->response.data.num_entries++;
686                         
687                         state->response.length +=
688                                 sizeof(struct winbindd_gr);
689                         
690                 } else {
691                         DEBUG(0, ("could not lookup domain group %s\n", 
692                                   domain_group_name));
693                 }
694         }
695
696         /* Copy the list of group memberships to the end of the extra data */
697
698         if (group_list_ndx == 0)
699                 goto done;
700
701         new_extra_data = Realloc(
702                 state->response.extra_data,
703                 group_list_ndx * sizeof(struct winbindd_gr) + gr_mem_list_len);
704
705         if (!new_extra_data) {
706                 DEBUG(0, ("out of memory\n"));
707                 group_list_ndx = 0;
708                 SAFE_FREE(state->response.extra_data);
709                 SAFE_FREE(gr_mem_list);
710
711                 return WINBINDD_ERROR;
712         }
713
714         state->response.extra_data = new_extra_data;
715
716         memcpy(&((char *)state->response.extra_data)
717                [group_list_ndx * sizeof(struct winbindd_gr)], 
718                gr_mem_list, gr_mem_list_len);
719
720         SAFE_FREE(gr_mem_list);
721
722         state->response.length += gr_mem_list_len;
723
724         DEBUG(10, ("getgrent(): returning %d groups, length = %d\n",
725                    group_list_ndx, gr_mem_list_len));
726
727         /* Out of domains */
728
729  done:
730         return (group_list_ndx > 0) ? WINBINDD_OK : WINBINDD_ERROR;
731 }
732
733 /* List domain groups without mapping to unix ids */
734
735 enum winbindd_result winbindd_list_groups(struct winbindd_cli_state *state)
736 {
737         uint32 total_entries = 0;
738         uint32 num_domain_entries;
739         struct winbindd_domain *domain;
740         struct getent_state groups;
741         char *extra_data = NULL;
742         char *ted = NULL;
743         int extra_data_len = 0, i;
744         void *sam_entries = NULL;
745
746         DEBUG(3, ("[%5d]: list groups\n", state->pid));
747
748         /* Enumerate over trusted domains */
749
750         ZERO_STRUCT(groups);
751
752         for (domain = domain_list; domain; domain = domain->next) {
753
754                 /* Skip domains other than WINBINDD_DOMAIN environment
755                    variable */ 
756
757                 if ((strcmp(state->request.domain, "") != 0) &&
758                     !check_domain_env(state->request.domain, domain->name))
759                         continue;
760
761                 /* Get list of sam groups */
762
763                 ZERO_STRUCT(groups);
764                 groups.domain = domain;
765                 groups.mem_ctx = talloc_init();
766
767                 /* 
768                  * iterate through all groups 
769                  * total_entries:  maintains a total count over **all domains**
770                  * num_domain_entries:  is the running count for this domain
771                  */
772                  
773                 num_domain_entries = 0;
774
775                 while (get_sam_group_entries(&groups)) {
776                         int new_size;
777                         int offset;
778                         
779                         offset = sizeof(struct acct_info) * num_domain_entries;
780                         new_size = sizeof(struct acct_info) 
781                                 * (groups.num_sam_entries + num_domain_entries);
782                         sam_entries = Realloc(sam_entries, new_size);
783                         
784                         if (!sam_entries)
785                                 return WINBINDD_ERROR;
786                         
787                         num_domain_entries += groups.num_sam_entries;
788                         memcpy (((char *)sam_entries)+offset, 
789                                 groups.sam_entries, 
790                                 sizeof(struct acct_info) * 
791                                 groups.num_sam_entries);
792                         
793                         groups.sam_entries = NULL;
794                         groups.num_sam_entries = 0;
795                 }
796
797                 /* skip remainder of loop if we idn;t retrieve any groups */
798                 
799                 if (num_domain_entries == 0) 
800                         goto next_group;
801
802                 /* setup the groups struct to contain all the groups 
803                    retrieved for this domain */
804                   
805                 groups.num_sam_entries = num_domain_entries;
806                 groups.sam_entries = sam_entries;
807
808                 /* keep track the of the total number of groups seen so 
809                    far over all domains */
810
811                 total_entries += groups.num_sam_entries;
812                 
813                 /* Allocate some memory for extra data.  Note that we limit
814                    account names to sizeof(fstring) = 128 characters.  */
815                 
816                 ted = Realloc(extra_data, sizeof(fstring) * total_entries);
817  
818                 if (!ted) {
819                         DEBUG(0,("winbindd_list_groups: failed to enlarge "
820                                  "buffer!\n"));
821                         SAFE_FREE(extra_data);
822                         talloc_destroy(groups.mem_ctx);
823                         return WINBINDD_ERROR;
824                 } else
825                         extra_data = ted;
826
827                 /* Pack group list into extra data fields */
828
829                 for (i = 0; i < groups.num_sam_entries; i++) {
830                         char *group_name = ((struct acct_info *)
831                                             groups.sam_entries)[i].acct_name; 
832                         fstring name;
833
834                         /* Convert unistring to ascii */
835
836                         snprintf(name, sizeof(name), "%s%s%s", domain->name, 
837                                 lp_winbind_separator(), group_name);
838
839                         /* Append to extra data */
840                         
841                         memcpy(&extra_data[extra_data_len], name, 
842                                strlen(name));
843
844                         extra_data_len += strlen(name);
845
846                         extra_data[extra_data_len++] = ',';
847                 }
848
849         next_group:
850                 talloc_destroy(groups.mem_ctx);
851         }
852
853         /* Assign extra_data fields in response structure */
854
855         if (extra_data) {
856                 extra_data[extra_data_len - 1] = '\0';
857                 state->response.extra_data = extra_data;
858                 state->response.length += extra_data_len;
859         }
860
861         /* No domains may have responded but that's still OK so don't
862            return an error. */
863
864         return WINBINDD_OK;
865 }
866
867 /* Get user supplementary groups.  This is much quicker than trying to
868    invert the groups database. */
869
870 enum winbindd_result winbindd_getgroups(struct winbindd_cli_state *state)
871 {
872         fstring name_domain, name_user, name;
873         DOM_SID user_sid;
874         enum SID_NAME_USE name_type;
875         uint32 user_rid, num_groups, num_gids;
876         DOM_GID *user_groups = NULL;
877         struct winbindd_domain *domain;
878         enum winbindd_result result;
879         gid_t *gid_list;
880         int i;
881         
882         DEBUG(3, ("[%5d]: getgroups %s\n", state->pid,
883                   state->request.data.username));
884
885         /* Parse domain and username */
886
887         parse_domain_user(state->request.data.username, name_domain, 
888                           name_user);
889
890         /* Reject names that don't have a domain - i.e name_domain contains 
891            the entire name. */
892  
893         if (strequal(name_domain, ""))
894                 return WINBINDD_ERROR;
895
896         /* Get info for the domain */
897         
898         if ((domain = find_domain_from_name(name_domain)) == NULL) {
899                 DEBUG(0, ("could not find domain entry for domain %s\n", 
900                           name_domain));
901                 return WINBINDD_ERROR;
902         }
903
904         slprintf(name, sizeof(name) - 1, "%s\\%s", name_domain, name_user);
905         
906         /* Get rid and name type from name.  The following costs 1 packet */
907
908         if (!winbindd_lookup_sid_by_name(name, &user_sid, &name_type)) {
909                 DEBUG(1, ("user '%s' does not exist\n", name_user));
910                 return WINBINDD_ERROR;
911         }
912
913         if (name_type != SID_NAME_USER) {
914                 DEBUG(1, ("name '%s' is not a user name: %d\n", name_user, 
915                           name_type));
916                 return WINBINDD_ERROR;
917         }
918
919         sid_split_rid(&user_sid, &user_rid);
920
921         if (!winbindd_lookup_usergroups(domain, user_rid, &num_groups,
922                                         &user_groups))
923                 return WINBINDD_ERROR;
924
925         /* Copy data back to client */
926
927         num_gids = 0;
928         gid_list = malloc(sizeof(gid_t) * num_groups);
929
930         if (state->response.extra_data) {
931                 result = WINBINDD_ERROR;
932                 goto done;
933         }
934
935         for (i = 0; i < num_groups; i++) {
936                 if (!winbindd_idmap_get_gid_from_rid(
937                         domain->name, user_groups[i].g_rid, 
938                         &gid_list[num_gids])) {
939
940                         DEBUG(1, ("unable to convert group rid %d to gid\n", 
941                                   user_groups[i].g_rid));
942                         continue;
943                 }
944                         
945                 num_gids++;
946         }
947
948         state->response.data.num_entries = num_gids;
949         state->response.extra_data = gid_list;
950         state->response.length += num_gids * sizeof(gid_t);
951
952         result = WINBINDD_OK;
953
954  done:
955         SAFE_FREE(user_groups);
956
957         return result;
958 }