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