1 /*-
2 * Copyright (c) 2005 Michael Bushkov <bushman@rsu.ru>
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 __FBSDID("$FreeBSD$");
30
31 #include <sys/param.h>
32
33 #include <assert.h>
34 #include <grp.h>
35 #include <nsswitch.h>
36 #include <stdlib.h>
37 #include <string.h>
38
39 #include "../debug.h"
40 #include "group.h"
41
42 static int group_marshal_func(struct group *, char *, size_t *);
43 static int group_lookup_func(const char *, size_t, char **, size_t *);
44 static void *group_mp_init_func(void);
45 static int group_mp_lookup_func(char **, size_t *, void *);
46 static void group_mp_destroy_func(void *);
47
48 static int
group_marshal_func(struct group * grp,char * buffer,size_t * buffer_size)49 group_marshal_func(struct group *grp, char *buffer, size_t *buffer_size)
50 {
51 struct group new_grp;
52 size_t desired_size, size, mem_size;
53 char *p, **mem;
54
55 TRACE_IN(group_marshal_func);
56 desired_size = ALIGNBYTES + sizeof(struct group) + sizeof(char *);
57
58 if (grp->gr_name != NULL)
59 desired_size += strlen(grp->gr_name) + 1;
60 if (grp->gr_passwd != NULL)
61 desired_size += strlen(grp->gr_passwd) + 1;
62
63 if (grp->gr_mem != NULL) {
64 mem_size = 0;
65 for (mem = grp->gr_mem; *mem; ++mem) {
66 desired_size += strlen(*mem) + 1;
67 ++mem_size;
68 }
69
70 desired_size += ALIGNBYTES + (mem_size + 1) * sizeof(char *);
71 }
72
73 if ((desired_size > *buffer_size) || (buffer == NULL)) {
74 *buffer_size = desired_size;
75 TRACE_OUT(group_marshal_func);
76 return (NS_RETURN);
77 }
78
79 memcpy(&new_grp, grp, sizeof(struct group));
80 memset(buffer, 0, desired_size);
81
82 *buffer_size = desired_size;
83 p = buffer + sizeof(struct group) + sizeof(char *);
84 memcpy(buffer + sizeof(struct group), &p, sizeof(char *));
85 p = (char *)ALIGN(p);
86
87 if (new_grp.gr_name != NULL) {
88 size = strlen(new_grp.gr_name);
89 memcpy(p, new_grp.gr_name, size);
90 new_grp.gr_name = p;
91 p += size + 1;
92 }
93
94 if (new_grp.gr_passwd != NULL) {
95 size = strlen(new_grp.gr_passwd);
96 memcpy(p, new_grp.gr_passwd, size);
97 new_grp.gr_passwd = p;
98 p += size + 1;
99 }
100
101 if (new_grp.gr_mem != NULL) {
102 p = (char *)ALIGN(p);
103 memcpy(p, new_grp.gr_mem, sizeof(char *) * mem_size);
104 new_grp.gr_mem = (char **)p;
105 p += sizeof(char *) * (mem_size + 1);
106
107 for (mem = new_grp.gr_mem; *mem; ++mem) {
108 size = strlen(*mem);
109 memcpy(p, *mem, size);
110 *mem = p;
111 p += size + 1;
112 }
113 }
114
115 memcpy(buffer, &new_grp, sizeof(struct group));
116 TRACE_OUT(group_marshal_func);
117 return (NS_SUCCESS);
118 }
119
120 static int
group_lookup_func(const char * key,size_t key_size,char ** buffer,size_t * buffer_size)121 group_lookup_func(const char *key, size_t key_size, char **buffer,
122 size_t *buffer_size)
123 {
124 enum nss_lookup_type lookup_type;
125 char *name;
126 size_t size;
127 gid_t gid;
128
129 struct group *result;
130
131 TRACE_IN(group_lookup_func);
132 assert(buffer != NULL);
133 assert(buffer_size != NULL);
134
135 if (key_size < sizeof(enum nss_lookup_type)) {
136 TRACE_OUT(group_lookup_func);
137 return (NS_UNAVAIL);
138 }
139 memcpy(&lookup_type, key, sizeof(enum nss_lookup_type));
140
141 switch (lookup_type) {
142 case nss_lt_name:
143 size = key_size - sizeof(enum nss_lookup_type) + 1;
144 name = calloc(1, size);
145 assert(name != NULL);
146 memcpy(name, key + sizeof(enum nss_lookup_type), size - 1);
147 break;
148 case nss_lt_id:
149 if (key_size < sizeof(enum nss_lookup_type) +
150 sizeof(gid_t)) {
151 TRACE_OUT(passwd_lookup_func);
152 return (NS_UNAVAIL);
153 }
154
155 memcpy(&gid, key + sizeof(enum nss_lookup_type), sizeof(gid_t));
156 break;
157 default:
158 TRACE_OUT(group_lookup_func);
159 return (NS_UNAVAIL);
160 }
161
162 switch (lookup_type) {
163 case nss_lt_name:
164 TRACE_STR(name);
165 result = getgrnam(name);
166 free(name);
167 break;
168 case nss_lt_id:
169 result = getgrgid(gid);
170 break;
171 default:
172 /* SHOULD NOT BE REACHED */
173 break;
174 }
175
176 if (result != NULL) {
177 group_marshal_func(result, NULL, buffer_size);
178 *buffer = malloc(*buffer_size);
179 assert(*buffer != NULL);
180 group_marshal_func(result, *buffer, buffer_size);
181 }
182
183 TRACE_OUT(group_lookup_func);
184 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS);
185 }
186
187 static void *
group_mp_init_func(void)188 group_mp_init_func(void)
189 {
190 TRACE_IN(group_mp_init_func);
191 setgrent();
192 TRACE_OUT(group_mp_init_func);
193
194 return (NULL);
195 }
196
197 static int
group_mp_lookup_func(char ** buffer,size_t * buffer_size,void * mdata)198 group_mp_lookup_func(char **buffer, size_t *buffer_size, void *mdata)
199 {
200 struct group *result;
201
202 TRACE_IN(group_mp_lookup_func);
203 result = getgrent();
204 if (result != NULL) {
205 group_marshal_func(result, NULL, buffer_size);
206 *buffer = malloc(*buffer_size);
207 assert(*buffer != NULL);
208 group_marshal_func(result, *buffer, buffer_size);
209 }
210
211 TRACE_OUT(group_mp_lookup_func);
212 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS);
213 }
214
215 static void
group_mp_destroy_func(void * mdata)216 group_mp_destroy_func(void *mdata)
217 {
218 TRACE_IN(group_mp_destroy_func);
219 TRACE_OUT(group_mp_destroy_func);
220 }
221
222 struct agent *
init_group_agent(void)223 init_group_agent(void)
224 {
225 struct common_agent *retval;
226
227 TRACE_IN(init_group_agent);
228 retval = calloc(1, sizeof(*retval));
229 assert(retval != NULL);
230
231 retval->parent.name = strdup("group");
232 assert(retval->parent.name != NULL);
233
234 retval->parent.type = COMMON_AGENT;
235 retval->lookup_func = group_lookup_func;
236
237 TRACE_OUT(init_group_agent);
238 return ((struct agent *)retval);
239 }
240
241 struct agent *
init_group_mp_agent(void)242 init_group_mp_agent(void)
243 {
244 struct multipart_agent *retval;
245
246 TRACE_IN(init_group_mp_agent);
247 retval = calloc(1,
248 sizeof(*retval));
249 assert(retval != NULL);
250
251 retval->parent.name = strdup("group");
252 retval->parent.type = MULTIPART_AGENT;
253 retval->mp_init_func = group_mp_init_func;
254 retval->mp_lookup_func = group_mp_lookup_func;
255 retval->mp_destroy_func = group_mp_destroy_func;
256 assert(retval->parent.name != NULL);
257
258 TRACE_OUT(init_group_mp_agent);
259 return ((struct agent *)retval);
260 }
261