group.c revision 194093
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: head/usr.sbin/nscd/agents/group.c 194093 2009-06-13 00:43:56Z des $"); 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 "passwd.h" 41 42static int group_marshal_func(struct group *, char *, size_t *); 43static int group_lookup_func(const char *, size_t, char **, size_t *); 44static void *group_mp_init_func(void); 45static int group_mp_lookup_func(char **, size_t *, void *); 46static void group_mp_destroy_func(void *); 47 48static int 49group_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 120static int 121group_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 = (char *)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 = (char *)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 187static void * 188group_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 197static int 198group_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 = (char *)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 215static void 216group_mp_destroy_func(void *mdata) 217{ 218 TRACE_IN(group_mp_destroy_func); 219 TRACE_OUT(group_mp_destroy_func); 220} 221 222struct agent * 223init_group_agent(void) 224{ 225 struct common_agent *retval; 226 227 TRACE_IN(init_group_agent); 228 retval = (struct common_agent *)calloc(1, sizeof(struct common_agent)); 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 241struct agent * 242init_group_mp_agent(void) 243{ 244 struct multipart_agent *retval; 245 246 TRACE_IN(init_group_mp_agent); 247 retval = (struct multipart_agent *)calloc(1, 248 sizeof(struct multipart_agent)); 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