s4:dsdb/common: add samdb_domain_guid() helper function
[samba.git] / nsswitch / winbind_nss_freebsd.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    AIX loadable authentication module, providing identification
5    routines against Samba winbind/Windows NT Domain
6
7    Copyright (C) Aaron Collins 2003
8    Copyright (C) Timur I. Bakeyev 2013
9
10    This library is free software; you can redistribute it and/or
11    modify it under the terms of the GNU Lesser General Public
12    License as published by the Free Software Foundation; either
13    version 3 of the License, or (at your option) any later version.
14
15    This library is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18    Library General Public License for more details.
19
20    You should have received a copy of the GNU Lesser General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "winbind_client.h"
25
26 /* Make sure that the module gets registered needed by freebsd 5.1 */
27 ns_mtab *nss_module_register(const char *, unsigned int *, nss_module_unregister_fn *);
28
29 NSS_METHOD_PROTOTYPE(__nss_compat_getgrnam_r);
30 NSS_METHOD_PROTOTYPE(__nss_compat_getgrgid_r);
31 NSS_METHOD_PROTOTYPE(__nss_compat_getgrent_r);
32 NSS_METHOD_PROTOTYPE(__nss_compat_setgrent);
33 NSS_METHOD_PROTOTYPE(__nss_compat_endgrent);
34
35 NSS_METHOD_PROTOTYPE(__nss_compat_getpwnam_r);
36 NSS_METHOD_PROTOTYPE(__nss_compat_getpwuid_r);
37 NSS_METHOD_PROTOTYPE(__nss_compat_getpwent_r);
38 NSS_METHOD_PROTOTYPE(__nss_compat_setpwent);
39 NSS_METHOD_PROTOTYPE(__nss_compat_endpwent);
40 NSS_METHOD_PROTOTYPE(__nss_compat_endpwent);
41
42 NSS_METHOD_PROTOTYPE(__freebsd_getgroupmembership);
43
44 static ns_mtab methods[] = {
45 { NSDB_GROUP, "getgrnam_r", __nss_compat_getgrnam_r, _nss_winbind_getgrnam_r },
46 { NSDB_GROUP, "getgrgid_r", __nss_compat_getgrgid_r, _nss_winbind_getgrgid_r },
47 { NSDB_GROUP, "getgrent_r", __nss_compat_getgrent_r, _nss_winbind_getgrent_r },
48 { NSDB_GROUP, "setgrent",   __nss_compat_setgrent,   _nss_winbind_setgrent },
49 { NSDB_GROUP, "endgrent",   __nss_compat_endgrent,   _nss_winbind_endgrent },
50 { NSDB_GROUP, "getgroupmembership", __freebsd_getgroupmembership, NULL },
51
52 { NSDB_PASSWD, "getpwnam_r", __nss_compat_getpwnam_r, _nss_winbind_getpwnam_r },
53 { NSDB_PASSWD, "getpwuid_r", __nss_compat_getpwuid_r, _nss_winbind_getpwuid_r },
54 { NSDB_PASSWD, "getpwent_r", __nss_compat_getpwent_r, _nss_winbind_getpwent_r },
55 { NSDB_PASSWD, "setpwent",   __nss_compat_setpwent,   _nss_winbind_setpwent },
56 { NSDB_PASSWD, "endpwent",   __nss_compat_endpwent,   _nss_winbind_endpwent },
57
58 };
59
60 /* Taken from libc */
61 static int
62 gr_addgid(gid_t gid, gid_t *groups, int maxgrp, int *grpcnt)
63 {
64         int     ret, dupc;
65
66         /* skip duplicates */
67         for (dupc = 0; dupc < MIN(maxgrp, *grpcnt); dupc++) {
68                 if (groups[dupc] == gid)
69                         return 1;
70         }
71
72         ret = 1;
73         if (*grpcnt < maxgrp)                   /* add this gid */
74                 groups[*grpcnt] = gid;
75         else
76                 ret = 0;
77
78         (*grpcnt)++;
79
80         return ret;
81 }
82
83 /*
84     rv = _nsdispatch(NULL, dtab, NSDB_GROUP, "getgroupmembership",
85                     defaultsrc, uname, agroup, groups, maxgrp, grpcnt);
86 */
87
88 int
89 __freebsd_getgroupmembership(void *retval, void *mdata, va_list ap)
90 {
91         const char      *uname  = va_arg(ap, const char *);
92         gid_t            group  = va_arg(ap, gid_t);
93         gid_t           *groups = va_arg(ap, gid_t *);
94         int              maxgrp = va_arg(ap, int);
95         int             *groupc = va_arg(ap, int *);
96
97         NSS_STATUS ret;
98         long int lcount, lsize;
99         int i, errnop;
100         gid_t *tmpgroups;
101
102         /* Can be realloc() inside _nss_winbind_initgroups_dyn() */
103         if ((tmpgroups=calloc(maxgrp, sizeof(gid_t))) == NULL) {
104                 errno = ENOMEM;
105                 return NS_TRYAGAIN;
106         }
107
108         lcount = 0;
109         lsize = maxgrp;
110         /* insert primary membership(possibly already there) */
111         gr_addgid(group, groups, maxgrp, groupc);
112         /* Don't limit number of groups, we want to know total size */
113         ret = _nss_winbind_initgroups_dyn(discard_const(uname),
114                                           group,
115                                           &lcount,
116                                           &lsize,
117                                           &tmpgroups,
118                                           0,
119                                           &errnop);
120         if (ret == NSS_STATUS_SUCCESS) {
121                 /* lcount potentially can be bigger than maxgrp, so would groupc */
122                 for (i = 0; i < lcount; i++)
123                          gr_addgid(tmpgroups[i], groups, maxgrp, groupc);
124         }
125         free(tmpgroups);
126         /* Let following nsswitch backend(s) add more groups(?) */
127         return NSS_STATUS_NOTFOUND;
128 }
129
130 ns_mtab *
131 nss_module_register(const char *source, unsigned int *mtabsize,
132     nss_module_unregister_fn *unreg)
133 {
134         *mtabsize = sizeof(methods)/sizeof(methods[0]);
135         *unreg = NULL;
136         return (methods);
137 }