nscdcli.c revision 183770
14Srgrimes/*- 21549Srgrimes * Copyright (c) 2005 Michael Bushkov <bushman@rsu.ru> 31549Srgrimes * All rights reserved. 44Srgrimes * 54Srgrimes * Redistribution and use in source and binary forms, with or without 64Srgrimes * modification, are permitted provided that the following conditions 74Srgrimes * are met: 84Srgrimes * 1. Redistributions of source code must retain the above copyright 94Srgrimes * notice, this list of conditions and the following disclaimer. 104Srgrimes * 2. Redistributions in binary form must reproduce the above copyright 114Srgrimes * notice, this list of conditions and the following disclaimer in the 124Srgrimes * documentation and/or other materials provided with the distribution. 134Srgrimes * 144Srgrimes * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 154Srgrimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 164Srgrimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 174Srgrimes * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 184Srgrimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 194Srgrimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 204Srgrimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 214Srgrimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 224Srgrimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 234Srgrimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 244Srgrimes * SUCH DAMAGE. 254Srgrimes * 264Srgrimes */ 274Srgrimes 284Srgrimes#include <sys/cdefs.h> 294Srgrimes__FBSDID("$FreeBSD: head/usr.sbin/nscd/nscdcli.c 183770 2008-10-12 00:44:27Z delphij $"); 304Srgrimes 314Srgrimes#include <sys/types.h> 324Srgrimes#include <sys/socket.h> 331549Srgrimes#include <sys/event.h> 3450477Speter#include <sys/uio.h> 354Srgrimes#include <sys/un.h> 364Srgrimes#include <assert.h> 37719Swollman#include <errno.h> 386165Sbde#include <fcntl.h> 39719Swollman#include <stdlib.h> 404Srgrimes#include <string.h> 411549Srgrimes#include <unistd.h> 424Srgrimes 431549Srgrimes#include "debug.h" 441549Srgrimes#include "nscdcli.h" 451549Srgrimes#include "protocol.h" 461549Srgrimes 471549Srgrimes#define DEFAULT_NSCD_IO_TIMEOUT 4 481549Srgrimes 491549Srgrimesstatic int safe_write(struct nscd_connection_ *, const void *, size_t); 501549Srgrimesstatic int safe_read(struct nscd_connection_ *, void *, size_t); 511549Srgrimesstatic int send_credentials(struct nscd_connection_ *, int); 521549Srgrimes 531549Srgrimesstatic int 541549Srgrimessafe_write(struct nscd_connection_ *connection, const void *data, 554Srgrimes size_t data_size) 561549Srgrimes{ 571549Srgrimes struct kevent eventlist; 5847347Sache int nevents; 5947389Sbde size_t result; 6047389Sbde ssize_t s_result; 6147347Sache struct timespec timeout; 6247389Sbde 6347389Sbde if (data_size == 0) 6447347Sache return (0); 6547347Sache 661549Srgrimes timeout.tv_sec = DEFAULT_NSCD_IO_TIMEOUT; 671549Srgrimes timeout.tv_nsec = 0; 681549Srgrimes result = 0; 691549Srgrimes do { 704Srgrimes nevents = kevent(connection->write_queue, NULL, 0, &eventlist, 711549Srgrimes 1, &timeout); 721549Srgrimes if ((nevents == 1) && (eventlist.filter == EVFILT_WRITE)) { 734Srgrimes s_result = write(connection->sockfd, data + result, 744Srgrimes eventlist.data < data_size - result ? 751549Srgrimes eventlist.data : data_size - result); 761549Srgrimes if (s_result == -1) 77880Salm return (-1); 781549Srgrimes else 791549Srgrimes result += s_result; 80880Salm 811549Srgrimes if (eventlist.flags & EV_EOF) 821549Srgrimes return (result < data_size ? -1 : 0); 831549Srgrimes } else 841549Srgrimes return (-1); 851549Srgrimes } while (result < data_size); 861549Srgrimes 871549Srgrimes return (0); 881549Srgrimes} 891549Srgrimes 901549Srgrimesstatic int 911549Srgrimessafe_read(struct nscd_connection_ *connection, void *data, size_t data_size) 921549Srgrimes{ 936165Sbde 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 126static int 127send_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 176struct nscd_connection_ * 177open_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 strncpy(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 227void 228close_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 241int 242nscd_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 280fin: 281 TRACE_OUT(nscd_transform); 282 return (error_code); 283} 284