1 /*-
2 * Copyright (c) 2005 Michael Bushkov <[email protected]>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 */
27
28 #include <sys/cdefs.h>
29 #include <sys/param.h>
30
31 #include <assert.h>
32 #include <grp.h>
33 #include <nsswitch.h>
34 #include <stdlib.h>
35 #include <string.h>
36
37 #include "../debug.h"
38 #include "group.h"
39
40 static int group_marshal_func(struct group *, char *, size_t *);
41 static int group_lookup_func(const char *, size_t, char **, size_t *);
42 static void *group_mp_init_func(void);
43 static int group_mp_lookup_func(char **, size_t *, void *);
44 static void group_mp_destroy_func(void *);
45
46 static int
group_marshal_func(struct group * grp,char * buffer,size_t * buffer_size)47 group_marshal_func(struct group *grp, char *buffer, size_t *buffer_size)
48 {
49 struct group new_grp;
50 size_t desired_size, size, mem_size;
51 char *p, **mem;
52
53 TRACE_IN(group_marshal_func);
54 desired_size = ALIGNBYTES + sizeof(struct group) + sizeof(char *);
55
56 if (grp->gr_name != NULL)
57 desired_size += strlen(grp->gr_name) + 1;
58 if (grp->gr_passwd != NULL)
59 desired_size += strlen(grp->gr_passwd) + 1;
60
61 if (grp->gr_mem != NULL) {
62 mem_size = 0;
63 for (mem = grp->gr_mem; *mem; ++mem) {
64 desired_size += strlen(*mem) + 1;
65 ++mem_size;
66 }
67
68 desired_size += ALIGNBYTES + (mem_size + 1) * sizeof(char *);
69 }
70
71 if ((desired_size > *buffer_size) || (buffer == NULL)) {
72 *buffer_size = desired_size;
73 TRACE_OUT(group_marshal_func);
74 return (NS_RETURN);
75 }
76
77 memcpy(&new_grp, grp, sizeof(struct group));
78 memset(buffer, 0, desired_size);
79
80 *buffer_size = desired_size;
81 p = buffer + sizeof(struct group) + sizeof(char *);
82 memcpy(buffer + sizeof(struct group), &p, sizeof(char *));
83 p = (char *)ALIGN(p);
84
85 if (new_grp.gr_name != NULL) {
86 size = strlen(new_grp.gr_name);
87 memcpy(p, new_grp.gr_name, size);
88 new_grp.gr_name = p;
89 p += size + 1;
90 }
91
92 if (new_grp.gr_passwd != NULL) {
93 size = strlen(new_grp.gr_passwd);
94 memcpy(p, new_grp.gr_passwd, size);
95 new_grp.gr_passwd = p;
96 p += size + 1;
97 }
98
99 if (new_grp.gr_mem != NULL) {
100 p = (char *)ALIGN(p);
101 memcpy(p, new_grp.gr_mem, sizeof(char *) * mem_size);
102 new_grp.gr_mem = (char **)p;
103 p += sizeof(char *) * (mem_size + 1);
104
105 for (mem = new_grp.gr_mem; *mem; ++mem) {
106 size = strlen(*mem);
107 memcpy(p, *mem, size);
108 *mem = p;
109 p += size + 1;
110 }
111 }
112
113 memcpy(buffer, &new_grp, sizeof(struct group));
114 TRACE_OUT(group_marshal_func);
115 return (NS_SUCCESS);
116 }
117
118 static int
group_lookup_func(const char * key,size_t key_size,char ** buffer,size_t * buffer_size)119 group_lookup_func(const char *key, size_t key_size, char **buffer,
120 size_t *buffer_size)
121 {
122 enum nss_lookup_type lookup_type;
123 char *name;
124 size_t size;
125 gid_t gid;
126
127 struct group *result;
128
129 TRACE_IN(group_lookup_func);
130 assert(buffer != NULL);
131 assert(buffer_size != NULL);
132
133 if (key_size < sizeof(enum nss_lookup_type)) {
134 TRACE_OUT(group_lookup_func);
135 return (NS_UNAVAIL);
136 }
137 memcpy(&lookup_type, key, sizeof(enum nss_lookup_type));
138
139 switch (lookup_type) {
140 case nss_lt_name:
141 size = key_size - sizeof(enum nss_lookup_type) + 1;
142 name = calloc(1, size);
143 assert(name != NULL);
144 memcpy(name, key + sizeof(enum nss_lookup_type), size - 1);
145 break;
146 case nss_lt_id:
147 if (key_size < sizeof(enum nss_lookup_type) +
148 sizeof(gid_t)) {
149 TRACE_OUT(passwd_lookup_func);
150 return (NS_UNAVAIL);
151 }
152
153 memcpy(&gid, key + sizeof(enum nss_lookup_type), sizeof(gid_t));
154 break;
155 default:
156 TRACE_OUT(group_lookup_func);
157 return (NS_UNAVAIL);
158 }
159
160 switch (lookup_type) {
161 case nss_lt_name:
162 TRACE_STR(name);
163 result = getgrnam(name);
164 free(name);
165 break;
166 case nss_lt_id:
167 result = getgrgid(gid);
168 break;
169 default:
170 /* SHOULD NOT BE REACHED */
171 break;
172 }
173
174 if (result != NULL) {
175 group_marshal_func(result, NULL, buffer_size);
176 *buffer = malloc(*buffer_size);
177 assert(*buffer != NULL);
178 group_marshal_func(result, *buffer, buffer_size);
179 }
180
181 TRACE_OUT(group_lookup_func);
182 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS);
183 }
184
185 static void *
group_mp_init_func(void)186 group_mp_init_func(void)
187 {
188 TRACE_IN(group_mp_init_func);
189 setgrent();
190 TRACE_OUT(group_mp_init_func);
191
192 return (NULL);
193 }
194
195 static int
group_mp_lookup_func(char ** buffer,size_t * buffer_size,void * mdata)196 group_mp_lookup_func(char **buffer, size_t *buffer_size, void *mdata)
197 {
198 struct group *result;
199
200 TRACE_IN(group_mp_lookup_func);
201 result = getgrent();
202 if (result != NULL) {
203 group_marshal_func(result, NULL, buffer_size);
204 *buffer = malloc(*buffer_size);
205 assert(*buffer != NULL);
206 group_marshal_func(result, *buffer, buffer_size);
207 }
208
209 TRACE_OUT(group_mp_lookup_func);
210 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS);
211 }
212
213 static void
group_mp_destroy_func(void * mdata)214 group_mp_destroy_func(void *mdata)
215 {
216 TRACE_IN(group_mp_destroy_func);
217 TRACE_OUT(group_mp_destroy_func);
218 }
219
220 struct agent *
init_group_agent(void)221 init_group_agent(void)
222 {
223 struct common_agent *retval;
224
225 TRACE_IN(init_group_agent);
226 retval = calloc(1, sizeof(*retval));
227 assert(retval != NULL);
228
229 retval->parent.name = strdup("group");
230 assert(retval->parent.name != NULL);
231
232 retval->parent.type = COMMON_AGENT;
233 retval->lookup_func = group_lookup_func;
234
235 TRACE_OUT(init_group_agent);
236 return ((struct agent *)retval);
237 }
238
239 struct agent *
init_group_mp_agent(void)240 init_group_mp_agent(void)
241 {
242 struct multipart_agent *retval;
243
244 TRACE_IN(init_group_mp_agent);
245 retval = calloc(1,
246 sizeof(*retval));
247 assert(retval != NULL);
248
249 retval->parent.name = strdup("group");
250 retval->parent.type = MULTIPART_AGENT;
251 retval->mp_init_func = group_mp_init_func;
252 retval->mp_lookup_func = group_mp_lookup_func;
253 retval->mp_destroy_func = group_mp_destroy_func;
254 assert(retval->parent.name != NULL);
255
256 TRACE_OUT(init_group_mp_agent);
257 return ((struct agent *)retval);
258 }
259