nscdcli.c revision 158115
1158115Sume/*- 2158115Sume * Copyright (c) 2005 Michael Bushkov <bushman@rsu.ru> 3158115Sume * All rights reserved. 4158115Sume * 5158115Sume * Redistribution and use in source and binary forms, with or without 6158115Sume * modification, are permitted provided that the following conditions 7158115Sume * are met: 8158115Sume * 1. Redistributions of source code must retain the above copyright 9158115Sume * notice, this list of conditions and the following disclaimer. 10158115Sume * 2. Redistributions in binary form must reproduce the above copyright 11158115Sume * notice, this list of conditions and the following disclaimer in the 12158115Sume * documentation and/or other materials provided with the distribution. 13158115Sume * 14158115Sume * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15158115Sume * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16158115Sume * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17158115Sume * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18158115Sume * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19158115Sume * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20158115Sume * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21158115Sume * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22158115Sume * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23158115Sume * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24158115Sume * SUCH DAMAGE. 25158115Sume * 26158115Sume */ 27158115Sume 28158115Sume#include <sys/cdefs.h> 29158115Sume__FBSDID("$FreeBSD: head/usr.sbin/nscd/nscdcli.c 158115 2006-04-28 12:03:38Z ume $"); 30158115Sume 31158115Sume#include <sys/types.h> 32158115Sume#include <sys/socket.h> 33158115Sume#include <sys/event.h> 34158115Sume#include <sys/uio.h> 35158115Sume#include <sys/un.h> 36158115Sume#include <assert.h> 37158115Sume#include <errno.h> 38158115Sume#include <fcntl.h> 39158115Sume#include <stdlib.h> 40158115Sume#include <string.h> 41158115Sume#include <unistd.h> 42158115Sume 43158115Sume#include "debug.h" 44158115Sume#include "cachedcli.h" 45158115Sume#include "protocol.h" 46158115Sume 47158115Sume#define DEFAULT_CACHED_IO_TIMEOUT 4 48158115Sume 49158115Sumestatic int safe_write(struct cached_connection_ *, const void *, size_t); 50158115Sumestatic int safe_read(struct cached_connection_ *, void *, size_t); 51158115Sumestatic int send_credentials(struct cached_connection_ *, int); 52158115Sume 53158115Sumestatic int 54158115Sumesafe_write(struct cached_connection_ *connection, const void *data, 55158115Sume size_t data_size) 56158115Sume{ 57158115Sume struct kevent eventlist; 58158115Sume int nevents; 59158115Sume size_t result; 60158115Sume ssize_t s_result; 61158115Sume struct timespec timeout; 62158115Sume 63158115Sume if (data_size == 0) 64158115Sume return (0); 65158115Sume 66158115Sume timeout.tv_sec = DEFAULT_CACHED_IO_TIMEOUT; 67158115Sume timeout.tv_nsec = 0; 68158115Sume result = 0; 69158115Sume do { 70158115Sume nevents = kevent(connection->write_queue, NULL, 0, &eventlist, 71158115Sume 1, &timeout); 72158115Sume if ((nevents == 1) && (eventlist.filter == EVFILT_WRITE)) { 73158115Sume s_result = write(connection->sockfd, data + result, 74158115Sume eventlist.data < data_size - result ? 75158115Sume eventlist.data : data_size - result); 76158115Sume if (s_result == -1) 77158115Sume return (-1); 78158115Sume else 79158115Sume result += s_result; 80158115Sume 81158115Sume if (eventlist.flags & EV_EOF) 82158115Sume return (result < data_size ? -1 : 0); 83158115Sume } else 84158115Sume return (-1); 85158115Sume } while (result < data_size); 86158115Sume 87158115Sume return (0); 88158115Sume} 89158115Sume 90158115Sumestatic int 91158115Sumesafe_read(struct cached_connection_ *connection, void *data, size_t data_size) 92158115Sume{ 93158115Sume struct kevent eventlist; 94158115Sume size_t result; 95158115Sume ssize_t s_result; 96158115Sume struct timespec timeout; 97158115Sume int nevents; 98158115Sume 99158115Sume if (data_size == 0) 100158115Sume return (0); 101158115Sume 102158115Sume timeout.tv_sec = DEFAULT_CACHED_IO_TIMEOUT; 103158115Sume timeout.tv_nsec = 0; 104158115Sume result = 0; 105158115Sume do { 106158115Sume nevents = kevent(connection->read_queue, NULL, 0, &eventlist, 1, 107158115Sume &timeout); 108158115Sume if ((nevents == 1) && (eventlist.filter == EVFILT_READ)) { 109158115Sume s_result = read(connection->sockfd, data + result, 110158115Sume eventlist.data <= data_size - result ? eventlist.data : 111158115Sume data_size - result); 112158115Sume if (s_result == -1) 113158115Sume return (-1); 114158115Sume else 115158115Sume result += s_result; 116158115Sume 117158115Sume if (eventlist.flags & EV_EOF) 118158115Sume return (result < data_size ? -1 : 0); 119158115Sume } else 120158115Sume return (-1); 121158115Sume } while (result < data_size); 122158115Sume 123158115Sume return (0); 124158115Sume} 125158115Sume 126158115Sumestatic int 127158115Sumesend_credentials(struct cached_connection_ *connection, int type) 128158115Sume{ 129158115Sume struct kevent eventlist; 130158115Sume int nevents; 131158115Sume ssize_t result; 132158115Sume int res; 133158115Sume 134158115Sume struct msghdr cred_hdr; 135158115Sume struct iovec iov; 136158115Sume 137158115Sume struct { 138158115Sume struct cmsghdr hdr; 139158115Sume struct cmsgcred creds; 140158115Sume } cmsg; 141158115Sume 142158115Sume TRACE_IN(send_credentials); 143158115Sume memset(&cmsg, 0, sizeof(cmsg)); 144158115Sume cmsg.hdr.cmsg_len = sizeof(cmsg); 145158115Sume cmsg.hdr.cmsg_level = SOL_SOCKET; 146158115Sume cmsg.hdr.cmsg_type = SCM_CREDS; 147158115Sume 148158115Sume memset(&cred_hdr, 0, sizeof(struct msghdr)); 149158115Sume cred_hdr.msg_iov = &iov; 150158115Sume cred_hdr.msg_iovlen = 1; 151158115Sume cred_hdr.msg_control = &cmsg; 152158115Sume cred_hdr.msg_controllen = sizeof(cmsg); 153158115Sume 154158115Sume iov.iov_base = &type; 155158115Sume iov.iov_len = sizeof(int); 156158115Sume 157158115Sume EV_SET(&eventlist, connection->sockfd, EVFILT_WRITE, EV_ADD, 158158115Sume NOTE_LOWAT, sizeof(int), NULL); 159158115Sume res = kevent(connection->write_queue, &eventlist, 1, NULL, 0, NULL); 160158115Sume 161158115Sume nevents = kevent(connection->write_queue, NULL, 0, &eventlist, 1, NULL); 162158115Sume if ((nevents == 1) && (eventlist.filter == EVFILT_WRITE)) { 163158115Sume result = (sendmsg(connection->sockfd, &cred_hdr, 0) == -1) ? -1 164158115Sume : 0; 165158115Sume EV_SET(&eventlist, connection->sockfd, EVFILT_WRITE, EV_ADD, 166158115Sume 0, 0, NULL); 167158115Sume kevent(connection->write_queue, &eventlist, 1, NULL, 0, NULL); 168158115Sume TRACE_OUT(send_credentials); 169158115Sume return (result); 170158115Sume } else { 171158115Sume TRACE_OUT(send_credentials); 172158115Sume return (-1); 173158115Sume } 174158115Sume} 175158115Sume 176158115Sumestruct cached_connection_ * 177158115Sumeopen_cached_connection__(struct cached_connection_params const *params) 178158115Sume{ 179158115Sume struct cached_connection_ *retval; 180158115Sume struct kevent eventlist; 181158115Sume struct sockaddr_un client_address; 182158115Sume int client_address_len, client_socket; 183158115Sume int res; 184158115Sume 185158115Sume TRACE_IN(open_cached_connection); 186158115Sume assert(params != NULL); 187158115Sume 188158115Sume client_socket = socket(PF_LOCAL, SOCK_STREAM, 0); 189158115Sume client_address.sun_family = PF_LOCAL; 190158115Sume strncpy(client_address.sun_path, params->socket_path, 191158115Sume sizeof(client_address.sun_path)); 192158115Sume client_address_len = sizeof(client_address.sun_family) + 193158115Sume strlen(client_address.sun_path) + 1; 194158115Sume 195158115Sume res = connect(client_socket, (struct sockaddr *)&client_address, 196158115Sume client_address_len); 197158115Sume if (res == -1) { 198158115Sume close(client_socket); 199158115Sume TRACE_OUT(open_cached_connection); 200158115Sume return (NULL); 201158115Sume } 202158115Sume fcntl(client_socket, F_SETFL, O_NONBLOCK); 203158115Sume 204158115Sume retval = malloc(sizeof(struct cached_connection_)); 205158115Sume assert(retval != NULL); 206158115Sume memset(retval, 0, sizeof(struct cached_connection_)); 207158115Sume 208158115Sume retval->sockfd = client_socket; 209158115Sume 210158115Sume retval->write_queue = kqueue(); 211158115Sume assert(retval->write_queue != -1); 212158115Sume 213158115Sume EV_SET(&eventlist, retval->sockfd, EVFILT_WRITE, EV_ADD, 214158115Sume 0, 0, NULL); 215158115Sume res = kevent(retval->write_queue, &eventlist, 1, NULL, 0, NULL); 216158115Sume 217158115Sume retval->read_queue = kqueue(); 218158115Sume assert(retval->read_queue != -1); 219158115Sume 220158115Sume EV_SET(&eventlist, retval->sockfd, EVFILT_READ, EV_ADD, 221158115Sume 0, 0, NULL); 222158115Sume res = kevent(retval->read_queue, &eventlist, 1, NULL, 0, NULL); 223158115Sume 224158115Sume TRACE_OUT(open_cached_connection); 225158115Sume return (retval); 226158115Sume} 227158115Sume 228158115Sumevoid 229158115Sumeclose_cached_connection__(struct cached_connection_ *connection) 230158115Sume{ 231158115Sume 232158115Sume TRACE_IN(close_cached_connection); 233158115Sume assert(connection != NULL); 234158115Sume 235158115Sume close(connection->sockfd); 236158115Sume close(connection->read_queue); 237158115Sume close(connection->write_queue); 238158115Sume free(connection); 239158115Sume TRACE_OUT(close_cached_connection); 240158115Sume} 241158115Sume 242158115Sumeint 243158115Sumecached_transform__(struct cached_connection_ *connection, 244158115Sume const char *entry_name, int transformation_type) 245158115Sume{ 246158115Sume size_t name_size; 247158115Sume int error_code; 248158115Sume int result; 249158115Sume 250158115Sume TRACE_IN(cached_transform); 251158115Sume 252158115Sume error_code = -1; 253158115Sume result = 0; 254158115Sume result = send_credentials(connection, CET_TRANSFORM_REQUEST); 255158115Sume if (result != 0) 256158115Sume goto fin; 257158115Sume 258158115Sume if (entry_name != NULL) 259158115Sume name_size = strlen(entry_name); 260158115Sume else 261158115Sume name_size = 0; 262158115Sume 263158115Sume result = safe_write(connection, &name_size, sizeof(size_t)); 264158115Sume if (result != 0) 265158115Sume goto fin; 266158115Sume 267158115Sume result = safe_write(connection, &transformation_type, sizeof(int)); 268158115Sume if (result != 0) 269158115Sume goto fin; 270158115Sume 271158115Sume if (entry_name != NULL) { 272158115Sume result = safe_write(connection, entry_name, name_size); 273158115Sume if (result != 0) 274158115Sume goto fin; 275158115Sume } 276158115Sume 277158115Sume result = safe_read(connection, &error_code, sizeof(int)); 278158115Sume if (result != 0) 279158115Sume error_code = -1; 280158115Sume 281158115Sumefin: 282158115Sume TRACE_OUT(cached_transform); 283158115Sume return (error_code); 284158115Sume} 285