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/types.h>
30
31 #include <sys/event.h>
32 #include <sys/socket.h>
33 #include <sys/uio.h>
34 #include <sys/un.h>
35
36 #include <assert.h>
37 #include <errno.h>
38 #include <fcntl.h>
39 #include <stdlib.h>
40 #include <string.h>
41 #include <unistd.h>
42
43 #include "debug.h"
44 #include "nscdcli.h"
45 #include "protocol.h"
46
47 #define DEFAULT_NSCD_IO_TIMEOUT 4
48
49 static int safe_write(struct nscd_connection_ *, const void *, size_t);
50 static int safe_read(struct nscd_connection_ *, void *, size_t);
51 static int send_credentials(struct nscd_connection_ *, int);
52
53 static int
safe_write(struct nscd_connection_ * connection,const void * data,size_t data_size)54 safe_write(struct nscd_connection_ *connection, const void *data,
55 size_t data_size)
56 {
57 struct kevent eventlist;
58 int nevents;
59 size_t result;
60 ssize_t s_result;
61 struct timespec timeout;
62
63 if (data_size == 0)
64 return (0);
65
66 timeout.tv_sec = DEFAULT_NSCD_IO_TIMEOUT;
67 timeout.tv_nsec = 0;
68 result = 0;
69 do {
70 nevents = kevent(connection->write_queue, NULL, 0, &eventlist,
71 1, &timeout);
72 if ((nevents == 1) && (eventlist.filter == EVFILT_WRITE)) {
73 s_result = write(connection->sockfd,
74 (char *)data + result,
75 (size_t)eventlist.data < data_size - result ?
76 (size_t)eventlist.data : data_size - result);
77 if (s_result == -1)
78 return (-1);
79 else
80 result += s_result;
81
82 if (eventlist.flags & EV_EOF)
83 return (result < data_size ? -1 : 0);
84 } else
85 return (-1);
86 } while (result < data_size);
87
88 return (0);
89 }
90
91 static int
safe_read(struct nscd_connection_ * connection,void * data,size_t data_size)92 safe_read(struct nscd_connection_ *connection, void *data, size_t data_size)
93 {
94 struct kevent eventlist;
95 size_t result;
96 ssize_t s_result;
97 struct timespec timeout;
98 int nevents;
99
100 if (data_size == 0)
101 return (0);
102
103 timeout.tv_sec = DEFAULT_NSCD_IO_TIMEOUT;
104 timeout.tv_nsec = 0;
105 result = 0;
106 do {
107 nevents = kevent(connection->read_queue, NULL, 0, &eventlist, 1,
108 &timeout);
109 if ((nevents == 1) && (eventlist.filter == EVFILT_READ)) {
110 s_result = read(connection->sockfd,
111 (char *)data + result,
112 (size_t)eventlist.data <= data_size - result ?
113 (size_t)eventlist.data : data_size - result);
114 if (s_result == -1)
115 return (-1);
116 else
117 result += s_result;
118
119 if (eventlist.flags & EV_EOF)
120 return (result < data_size ? -1 : 0);
121 } else
122 return (-1);
123 } while (result < data_size);
124
125 return (0);
126 }
127
128 static int
send_credentials(struct nscd_connection_ * connection,int type)129 send_credentials(struct nscd_connection_ *connection, int type)
130 {
131 union {
132 struct cmsghdr hdr;
133 char pad[CMSG_SPACE(sizeof(struct cmsgcred))];
134 } cmsg;
135 struct msghdr mhdr;
136 struct iovec iov;
137 struct kevent eventlist;
138 int nevents;
139 ssize_t result;
140
141 TRACE_IN(send_credentials);
142 memset(&cmsg, 0, sizeof(cmsg));
143 cmsg.hdr.cmsg_len = CMSG_LEN(sizeof(struct cmsgcred));
144 cmsg.hdr.cmsg_level = SOL_SOCKET;
145 cmsg.hdr.cmsg_type = SCM_CREDS;
146
147 memset(&mhdr, 0, sizeof(mhdr));
148 mhdr.msg_iov = &iov;
149 mhdr.msg_iovlen = 1;
150 mhdr.msg_control = &cmsg;
151 mhdr.msg_controllen = CMSG_SPACE(sizeof(struct cmsgcred));
152
153 iov.iov_base = &type;
154 iov.iov_len = sizeof(int);
155
156 EV_SET(&eventlist, connection->sockfd, EVFILT_WRITE, EV_ADD,
157 NOTE_LOWAT, sizeof(int), NULL);
158 kevent(connection->write_queue, &eventlist, 1, NULL, 0, NULL);
159
160 nevents = kevent(connection->write_queue, NULL, 0, &eventlist, 1, NULL);
161 if ((nevents == 1) && (eventlist.filter == EVFILT_WRITE)) {
162 result = sendmsg(connection->sockfd, &mhdr, 0) == -1 ? -1 : 0;
163 EV_SET(&eventlist, connection->sockfd, EVFILT_WRITE, EV_ADD,
164 0, 0, NULL);
165 kevent(connection->write_queue, &eventlist, 1, NULL, 0, NULL);
166 TRACE_OUT(send_credentials);
167 return (result);
168 } else {
169 TRACE_OUT(send_credentials);
170 return (-1);
171 }
172 }
173
174 struct nscd_connection_ *
open_nscd_connection__(struct nscd_connection_params const * params)175 open_nscd_connection__(struct nscd_connection_params const *params)
176 {
177 struct nscd_connection_ *retval;
178 struct kevent eventlist;
179 struct sockaddr_un client_address;
180 int client_address_len, client_socket;
181 int res;
182
183 TRACE_IN(open_nscd_connection);
184 assert(params != NULL);
185
186 client_socket = socket(PF_LOCAL, SOCK_STREAM, 0);
187 client_address.sun_family = PF_LOCAL;
188 strlcpy(client_address.sun_path, params->socket_path,
189 sizeof(client_address.sun_path));
190 client_address_len = sizeof(client_address.sun_family) +
191 strlen(client_address.sun_path) + 1;
192
193 res = connect(client_socket, (struct sockaddr *)&client_address,
194 client_address_len);
195 if (res == -1) {
196 close(client_socket);
197 TRACE_OUT(open_nscd_connection);
198 return (NULL);
199 }
200 fcntl(client_socket, F_SETFL, O_NONBLOCK);
201
202 retval = calloc(1, sizeof(*retval));
203 assert(retval != NULL);
204
205 retval->sockfd = client_socket;
206
207 retval->write_queue = kqueue();
208 assert(retval->write_queue != -1);
209
210 EV_SET(&eventlist, retval->sockfd, EVFILT_WRITE, EV_ADD,
211 0, 0, NULL);
212 res = kevent(retval->write_queue, &eventlist, 1, NULL, 0, NULL);
213
214 retval->read_queue = kqueue();
215 assert(retval->read_queue != -1);
216
217 EV_SET(&eventlist, retval->sockfd, EVFILT_READ, EV_ADD,
218 0, 0, NULL);
219 res = kevent(retval->read_queue, &eventlist, 1, NULL, 0, NULL);
220
221 TRACE_OUT(open_nscd_connection);
222 return (retval);
223 }
224
225 void
close_nscd_connection__(struct nscd_connection_ * connection)226 close_nscd_connection__(struct nscd_connection_ *connection)
227 {
228
229 TRACE_IN(close_nscd_connection);
230 assert(connection != NULL);
231
232 close(connection->sockfd);
233 close(connection->read_queue);
234 close(connection->write_queue);
235 free(connection);
236 TRACE_OUT(close_nscd_connection);
237 }
238
239 int
nscd_transform__(struct nscd_connection_ * connection,const char * entry_name,int transformation_type)240 nscd_transform__(struct nscd_connection_ *connection,
241 const char *entry_name, int transformation_type)
242 {
243 size_t name_size;
244 int error_code;
245 int result;
246
247 TRACE_IN(nscd_transform);
248
249 error_code = -1;
250 result = 0;
251 result = send_credentials(connection, CET_TRANSFORM_REQUEST);
252 if (result != 0)
253 goto fin;
254
255 if (entry_name != NULL)
256 name_size = strlen(entry_name);
257 else
258 name_size = 0;
259
260 result = safe_write(connection, &name_size, sizeof(size_t));
261 if (result != 0)
262 goto fin;
263
264 result = safe_write(connection, &transformation_type, sizeof(int));
265 if (result != 0)
266 goto fin;
267
268 if (entry_name != NULL) {
269 result = safe_write(connection, entry_name, name_size);
270 if (result != 0)
271 goto fin;
272 }
273
274 result = safe_read(connection, &error_code, sizeof(int));
275 if (result != 0)
276 error_code = -1;
277
278 fin:
279 TRACE_OUT(nscd_transform);
280 return (error_code);
281 }
282