Fix "buffer to small" problems and memory leaks.
[kopensolaris-gnu/glibc.git] / nis / nss_nisplus / nisplus-spwd.c
1 /* Copyright (C) 1997 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Thorsten Kukuk <kukuk@vt.uni-paderborn.de>, 1997.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Library General Public License as
7    published by the Free Software Foundation; either version 2 of the
8    License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Library General Public License for more details.
14
15    You should have received a copy of the GNU Library General Public
16    License along with the GNU C Library; see the file COPYING.LIB.  If not,
17    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18    Boston, MA 02111-1307, USA.  */
19
20 #include <nss.h>
21 #include <errno.h>
22 #include <shadow.h>
23 #include <string.h>
24 #include <bits/libc-lock.h>
25 #include <rpcsvc/nis.h>
26 #include <rpcsvc/nislib.h>
27
28 #include "nss-nisplus.h"
29 #include "nisplus-parser.h"
30
31 __libc_lock_define_initialized (static, lock)
32
33 static nis_result *result = NULL;
34 static nis_name tablename_val = NULL;
35 static u_long tablename_len = 0;
36
37 static enum nss_status
38 _nss_create_tablename (void)
39 {
40   if (tablename_val == NULL)
41     {
42       char buf [40 + strlen (nis_local_directory ())];
43       char *p;
44
45       p = stpcpy (buf, "passwd.org_dir.");
46       p = stpcpy (p, nis_local_directory ());
47       tablename_val = strdup (buf);
48       if (tablename_val == NULL)
49         return NSS_STATUS_TRYAGAIN;
50       tablename_len = strlen (tablename_val);
51     }
52   return NSS_STATUS_SUCCESS;
53 }
54
55 enum nss_status
56 _nss_nisplus_setspent (void)
57 {
58   enum nss_status status = NSS_STATUS_SUCCESS;
59
60   __libc_lock_lock (lock);
61
62   if (result)
63     nis_freeresult (result);
64   result = NULL;
65
66   if (tablename_val == NULL)
67     status = _nss_create_tablename ();
68
69   __libc_lock_unlock (lock);
70
71   return NSS_STATUS_SUCCESS;
72 }
73
74 enum nss_status
75 _nss_nisplus_endspent (void)
76 {
77   __libc_lock_lock (lock);
78
79   if (result)
80     nis_freeresult (result);
81   result = NULL;
82
83   __libc_lock_unlock (lock);
84
85   return NSS_STATUS_SUCCESS;
86 }
87
88 static enum nss_status
89 internal_nisplus_getspent_r (struct spwd *sp, char *buffer, size_t buflen)
90 {
91   int parse_res;
92
93   /* Get the next entry until we found a correct one. */
94   do
95     {
96       nis_result *saved_res;
97
98       if (result == NULL)
99         {
100           saved_res = NULL;
101
102           if (tablename_val == NULL)
103             if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
104               return NSS_STATUS_UNAVAIL;
105
106           result = nis_first_entry (tablename_val);
107           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
108             return niserr2nss (result->status);
109         }
110       else
111         {
112           nis_result *res;
113
114           saved_res = result;
115           res = nis_next_entry (tablename_val, &result->cookie);
116           result = res;
117           if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
118             {
119               nis_freeresult (saved_res);
120               return niserr2nss (result->status);
121             }
122         }
123
124       if ((parse_res = _nss_nisplus_parse_spent (result, sp, buffer,
125                                                  buflen)) == -1)
126         {
127           nis_freeresult (result);
128           result = saved_res;
129           return NSS_STATUS_TRYAGAIN;
130         }
131       else
132         {
133           if (saved_res)
134             nis_freeresult (saved_res);
135         }
136     } while (!parse_res);
137
138   return NSS_STATUS_SUCCESS;
139 }
140
141 enum nss_status
142 _nss_nisplus_getspent_r (struct spwd *result, char *buffer, size_t buflen)
143 {
144   int status;
145
146   __libc_lock_lock (lock);
147
148   status = internal_nisplus_getspent_r (result, buffer, buflen);
149
150   __libc_lock_unlock (lock);
151
152   return status;
153 }
154
155 enum nss_status
156 _nss_nisplus_getspnam_r (const char *name, struct spwd *sp,
157                      char *buffer, size_t buflen)
158 {
159   int parse_res;
160
161   if (tablename_val == NULL)
162     if (_nss_create_tablename () != NSS_STATUS_SUCCESS)
163       return NSS_STATUS_UNAVAIL;
164
165   if (name == NULL || strlen (name) > 8)
166     return NSS_STATUS_NOTFOUND;
167   else
168     {
169       nis_result *result;
170       char buf[strlen (name) + 24 + tablename_len];
171
172       sprintf (buf, "[name=%s],%s", name, tablename_val);
173
174       result = nis_list (buf, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
175
176       if (niserr2nss (result->status) != NSS_STATUS_SUCCESS)
177         {
178           enum nss_status status = niserr2nss (result->status);
179
180           nis_freeresult (result);
181           return status;
182         }
183
184       parse_res = _nss_nisplus_parse_spent (result, sp, buffer, buflen);
185       nis_freeresult (result);
186
187       if (parse_res == -1)
188         return NSS_STATUS_TRYAGAIN;
189
190       if (parse_res)
191         return NSS_STATUS_SUCCESS;
192
193       return NSS_STATUS_NOTFOUND;
194     }
195 }