passwd.c revision 158115
1132451Sroberto/*- 2132451Sroberto * Copyright (c) 2005 Michael Bushkov <bushman@rsu.ru> 3290001Sglebius * All rights reserved. 4290001Sglebius * 5290001Sglebius * Redistribution and use in source and binary forms, with or without 6290001Sglebius * modification, are permitted provided that the following conditions 7290001Sglebius * are met: 8290001Sglebius * 1. Redistributions of source code must retain the above copyright 9290001Sglebius * notice, this list of conditions and the following disclaimer. 10290001Sglebius * 2. Redistributions in binary form must reproduce the above copyright 11290001Sglebius * notice, this list of conditions and the following disclaimer in the 12290001Sglebius * documentation and/or other materials provided with the distribution. 13290001Sglebius * 14290001Sglebius * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15290001Sglebius * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16290001Sglebius * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17290001Sglebius * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18290001Sglebius * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19290001Sglebius * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20290001Sglebius * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21290001Sglebius * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22290001Sglebius * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23290001Sglebius * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24290001Sglebius * SUCH DAMAGE. 25290001Sglebius * 26290001Sglebius */ 27290001Sglebius 28290001Sglebius#include <sys/cdefs.h> 29290001Sglebius__FBSDID("$FreeBSD: head/usr.sbin/nscd/agents/passwd.c 158115 2006-04-28 12:03:38Z ume $"); 30290001Sglebius 31290001Sglebius#include <sys/types.h> 32290001Sglebius#include <assert.h> 33290001Sglebius#include <nsswitch.h> 34290001Sglebius#include <pwd.h> 35290001Sglebius#include <string.h> 36290001Sglebius#include <stdlib.h> 37290001Sglebius#include "../debug.h" 38290001Sglebius#include "passwd.h" 39290001Sglebius 40290001Sglebiusstatic int passwd_marshal_func(struct passwd *, char *, size_t *); 41290001Sglebiusstatic int passwd_lookup_func(const char *, size_t, char **, size_t *); 42290001Sglebiusstatic void *passwd_mp_init_func(); 43290001Sglebiusstatic int passwd_mp_lookup_func(char **, size_t *, void *); 44290001Sglebiusstatic void passwd_mp_destroy_func(void *mdata); 45290001Sglebius 46290001Sglebiusstatic int 47290001Sglebiuspasswd_marshal_func(struct passwd *pwd, char *buffer, size_t *buffer_size) 48290001Sglebius{ 49290001Sglebius char *p; 50290001Sglebius struct passwd new_pwd; 51290001Sglebius size_t desired_size, size; 52290001Sglebius 53290001Sglebius TRACE_IN(passwd_marshal_func); 54290001Sglebius desired_size = sizeof(struct passwd) + sizeof(char *) + 55290001Sglebius strlen(pwd->pw_name) + 1; 56290001Sglebius if (pwd->pw_passwd != NULL) 57290001Sglebius desired_size += strlen(pwd->pw_passwd) + 1; 58290001Sglebius if (pwd->pw_class != NULL) 59290001Sglebius desired_size += strlen(pwd->pw_class) + 1; 60290001Sglebius if (pwd->pw_gecos != NULL) 61290001Sglebius desired_size += strlen(pwd->pw_gecos) + 1; 62290001Sglebius if (pwd->pw_dir != NULL) 63290001Sglebius desired_size += strlen(pwd->pw_dir) + 1; 64290001Sglebius if (pwd->pw_shell != NULL) 65290001Sglebius desired_size += strlen(pwd->pw_shell) + 1; 66290001Sglebius 67290001Sglebius if ((*buffer_size < desired_size) || (buffer == NULL)) { 68290001Sglebius *buffer_size = desired_size; 69290001Sglebius TRACE_OUT(passwd_marshal_func); 70290001Sglebius return (NS_RETURN); 71290001Sglebius } 72290001Sglebius 73290001Sglebius memcpy(&new_pwd, pwd, sizeof(struct passwd)); 74290001Sglebius memset(buffer, 0, desired_size); 75290001Sglebius 76290001Sglebius *buffer_size = desired_size; 77290001Sglebius p = buffer + sizeof(struct passwd) + sizeof(char *); 78290001Sglebius memcpy(buffer + sizeof(struct passwd), &p, sizeof(char *)); 79290001Sglebius 80290001Sglebius if (new_pwd.pw_name != NULL) { 81290001Sglebius size = strlen(new_pwd.pw_name); 82290001Sglebius memcpy(p, new_pwd.pw_name, size); 83290001Sglebius new_pwd.pw_name = p; 84290001Sglebius p += size + 1; 85290001Sglebius } 86290001Sglebius 87290001Sglebius if (new_pwd.pw_passwd != NULL) { 88290001Sglebius size = strlen(new_pwd.pw_passwd); 89290001Sglebius memcpy(p, new_pwd.pw_passwd, size); 90290001Sglebius new_pwd.pw_passwd = p; 91290001Sglebius p += size + 1; 92290001Sglebius } 93290001Sglebius 94290001Sglebius if (new_pwd.pw_class != NULL) { 95290001Sglebius size = strlen(new_pwd.pw_class); 96290001Sglebius memcpy(p, new_pwd.pw_class, size); 97290001Sglebius new_pwd.pw_class = p; 98290001Sglebius p += size + 1; 99290001Sglebius } 100290001Sglebius 101290001Sglebius if (new_pwd.pw_gecos != NULL) { 102290001Sglebius size = strlen(new_pwd.pw_gecos); 103290001Sglebius memcpy(p, new_pwd.pw_gecos, size); 104290001Sglebius new_pwd.pw_gecos = p; 105290001Sglebius p += size + 1; 106290001Sglebius } 107290001Sglebius 108290001Sglebius if (new_pwd.pw_dir != NULL) { 109290001Sglebius size = strlen(new_pwd.pw_dir); 110290001Sglebius memcpy(p, new_pwd.pw_dir, size); 111290001Sglebius new_pwd.pw_dir = p; 112290001Sglebius p += size + 1; 113290001Sglebius } 114290001Sglebius 115290001Sglebius if (new_pwd.pw_shell != NULL) { 116290001Sglebius size = strlen(new_pwd.pw_shell); 117290001Sglebius memcpy(p, new_pwd.pw_shell, size); 118290001Sglebius new_pwd.pw_shell = p; 119290001Sglebius p += size + 1; 120290001Sglebius } 121290001Sglebius 122290001Sglebius memcpy(buffer, &new_pwd, sizeof(struct passwd)); 123290001Sglebius TRACE_OUT(passwd_marshal_func); 124290001Sglebius return (NS_SUCCESS); 125290001Sglebius} 126290001Sglebius 127290001Sglebiusstatic int 128290001Sglebiuspasswd_lookup_func(const char *key, size_t key_size, char **buffer, 129290001Sglebius size_t *buffer_size) 130290001Sglebius{ 131290001Sglebius enum nss_lookup_type lookup_type; 132290001Sglebius char *login; 133290001Sglebius size_t size; 134290001Sglebius uid_t uid; 135290001Sglebius 136290001Sglebius struct passwd *result; 137290001Sglebius 138290001Sglebius TRACE_IN(passwd_lookup_func); 139290001Sglebius assert(buffer != NULL); 140290001Sglebius assert(buffer_size != NULL); 141290001Sglebius 142290001Sglebius if (key_size < sizeof(enum nss_lookup_type)) { 143290001Sglebius TRACE_OUT(passwd_lookup_func); 144290001Sglebius return (NS_UNAVAIL); 145290001Sglebius } 146290001Sglebius memcpy(&lookup_type, key, sizeof(enum nss_lookup_type)); 147290001Sglebius 148290001Sglebius switch (lookup_type) { 149290001Sglebius case nss_lt_name: 150290001Sglebius size = key_size - sizeof(enum nss_lookup_type) + 1; 151290001Sglebius login = (char *)malloc(size); 152290001Sglebius assert(login != NULL); 153290001Sglebius memset(login, 0, size); 154290001Sglebius memcpy(login, key + sizeof(enum nss_lookup_type), size - 1); 155290001Sglebius break; 156290001Sglebius case nss_lt_id: 157290001Sglebius if (key_size < sizeof(enum nss_lookup_type) + 158290001Sglebius sizeof(uid_t)) { 159290001Sglebius TRACE_OUT(passwd_lookup_func); 160290001Sglebius return (NS_UNAVAIL); 161290001Sglebius } 162290001Sglebius 163290001Sglebius memcpy(&uid, key + sizeof(enum nss_lookup_type), sizeof(uid_t)); 164290001Sglebius break; 165290001Sglebius default: 166290001Sglebius TRACE_OUT(passwd_lookup_func); 167290001Sglebius return (NS_UNAVAIL); 168290001Sglebius } 169290001Sglebius 170290001Sglebius switch (lookup_type) { 171290001Sglebius case nss_lt_name: 172290001Sglebius result = getpwnam(login); 173290001Sglebius free(login); 174290001Sglebius break; 175290001Sglebius case nss_lt_id: 176290001Sglebius result = getpwuid(uid); 177290001Sglebius break; 178290001Sglebius default: 179290001Sglebius /* SHOULD NOT BE REACHED */ 180290001Sglebius break; 181290001Sglebius } 182182007Sroberto 183 if (result != NULL) { 184 passwd_marshal_func(result, NULL, buffer_size); 185 *buffer = (char *)malloc(*buffer_size); 186 assert(*buffer != NULL); 187 passwd_marshal_func(result, *buffer, buffer_size); 188 } 189 190 TRACE_OUT(passwd_lookup_func); 191 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS); 192} 193 194static void * 195passwd_mp_init_func() 196{ 197 TRACE_IN(passwd_mp_init_func); 198 setpwent(); 199 TRACE_OUT(passwd_mp_init_func); 200 201 return (NULL); 202} 203 204static int 205passwd_mp_lookup_func(char **buffer, size_t *buffer_size, void *mdata) 206{ 207 struct passwd *result; 208 209 TRACE_IN(passwd_mp_lookup_func); 210 result = getpwent(); 211 if (result != NULL) { 212 passwd_marshal_func(result, NULL, buffer_size); 213 *buffer = (char *)malloc(*buffer_size); 214 assert(*buffer != NULL); 215 passwd_marshal_func(result, *buffer, buffer_size); 216 } 217 218 TRACE_OUT(passwd_mp_lookup_func); 219 return (result == NULL ? NS_NOTFOUND : NS_SUCCESS); 220} 221 222static void 223passwd_mp_destroy_func(void *mdata) 224{ 225 TRACE_IN(passwd_mp_destroy_func); 226 TRACE_OUT(passwd_mp_destroy_func); 227} 228 229struct agent * 230init_passwd_agent() 231{ 232 struct common_agent *retval; 233 234 TRACE_IN(init_passwd_agent); 235 retval = (struct common_agent *)malloc(sizeof(struct common_agent)); 236 assert(retval != NULL); 237 memset(retval, 0, sizeof(struct common_agent)); 238 239 retval->parent.name = strdup("passwd"); 240 assert(retval->parent.name != NULL); 241 242 retval->parent.type = COMMON_AGENT; 243 retval->lookup_func = passwd_lookup_func; 244 245 TRACE_OUT(init_passwd_agent); 246 return ((struct agent *)retval); 247} 248 249struct agent * 250init_passwd_mp_agent() 251{ 252 struct multipart_agent *retval; 253 254 TRACE_IN(init_passwd_mp_agent); 255 retval = (struct multipart_agent *)malloc( 256 sizeof(struct multipart_agent)); 257 assert(retval != NULL); 258 memset(retval, 0, sizeof(struct multipart_agent)); 259 260 retval->parent.name = strdup("passwd"); 261 retval->parent.type = MULTIPART_AGENT; 262 retval->mp_init_func = passwd_mp_init_func; 263 retval->mp_lookup_func = passwd_mp_lookup_func; 264 retval->mp_destroy_func = passwd_mp_destroy_func; 265 assert(retval->parent.name != NULL); 266 267 TRACE_OUT(init_passwd_mp_agent); 268 return ((struct agent *)retval); 269} 270