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