/*
	 * Copyright (C) 2010,2013 Red Hat, Inc.
	 *
	 * Author: Angus Salkeld <asalkeld@redhat.com>
	 *
	 * This file is part of libqb.
	 *
	 * libqb is free software: you can redistribute it and/or modify
	 * it under the terms of the GNU Lesser General Public License as published by
	 * the Free Software Foundation, either version 2.1 of the License, or
	 * (at your option) any later version.
	 *
	 * libqb is distributed in the hope that it will be useful,
	 * but WITHOUT ANY WARRANTY; without even the implied warranty of
	 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	 * GNU Lesser General Public License for more details.
	 *
	 * You should have received a copy of the GNU Lesser General Public License
	 * along with libqb.  If not, see <http://www.gnu.org/licenses/>.
	 */
	#include "os_base.h"
	#include <poll.h>
	
	#ifdef HAVE_SYS_UN_H
	#include <sys/un.h>
	#endif /* HAVE_SYS_UN_H */
	#ifdef HAVE_SYS_MMAN_H
	#include <sys/mman.h>
	#endif
	
	#include <qb/qbatomic.h>
	#include <qb/qbipcs.h>
	#include <qb/qbloop.h>
	#include <qb/qbdefs.h>
	
	#include "util_int.h"
	#include "ipc_int.h"
	
	struct ipc_us_control {
		int32_t sent;
		int32_t flow_control;
	};
	#define SHM_CONTROL_SIZE (3 * sizeof(struct ipc_us_control))
	
	int use_filesystem_sockets(void)
	{
		static int need_init = 1;
		static int filesystem_sockets = 0;
	
		if (need_init) {
	#if defined(QB_LINUX) || defined(QB_CYGWIN)
			struct stat buf;
	
			if (stat(FORCESOCKETSFILE, &buf) == 0) {
				filesystem_sockets = 1;
			}
	#else
			filesystem_sockets = 1;
	#endif
			need_init = 0;
		}
		return filesystem_sockets;
	}
	
	static void
	set_sock_addr(struct sockaddr_un *address, const char *socket_name)
	{
		memset(address, 0, sizeof(struct sockaddr_un));
		address->sun_family = AF_UNIX;
	#ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
		address->sun_len = QB_SUN_LEN(address);
	#endif
	
		if (socket_name[0] == '/' || !use_filesystem_sockets()) {
			snprintf(address->sun_path + 1, UNIX_PATH_MAX - 1, "%s", socket_name);
		} else {
			snprintf(address->sun_path, sizeof(address->sun_path), "%s/%s", SOCKETDIR,
				 socket_name);
		}
	}
	
	static int32_t
	qb_ipc_dgram_sock_setup(const char *base_name,
				const char *service_name, int32_t * sock_pt,
				gid_t gid)
	{
		int32_t request_fd;
		struct sockaddr_un local_address;
		int32_t res = 0;
		char sock_path[PATH_MAX];
	
		request_fd = socket(PF_UNIX, SOCK_DGRAM, 0);
		if (request_fd == -1) {
			return -errno;
		}
	
		qb_socket_nosigpipe(request_fd);
		res = qb_sys_fd_nonblock_cloexec_set(request_fd);
		if (res < 0) {
			goto error_connect;
		}
		snprintf(sock_path, PATH_MAX, "%s-%s", base_name, service_name);
		set_sock_addr(&local_address, sock_path);
		if (use_filesystem_sockets()) {
			(void)unlink(local_address.sun_path);
		}
		res = bind(request_fd, (struct sockaddr *)&local_address,
			   sizeof(local_address));
		if (res < 0) {
			goto error_connect;
		}
	
		if (use_filesystem_sockets()) {
			if (chmod(local_address.sun_path, 0660) != 0) {
				res = 0;
				qb_util_perror(LOG_ERR, "failed to chmod socket (%s)",
					       local_address.sun_path);
			}
			/* chown may fail if not root, but log it */
			if (chown(local_address.sun_path, -1, gid) != 0) {
				qb_util_perror(LOG_WARNING, "failed to chown socket (%s)",
					       local_address.sun_path);
			}
		}
	
		*sock_pt = request_fd;
		return 0;
	
	error_connect:
		close(request_fd);
		*sock_pt = -1;
	
		return res;
	}
	
	static int32_t
	set_sock_size(int sockfd, size_t max_msg_size)
	{
		int32_t rc;
		unsigned int optval;
		socklen_t optlen = sizeof(optval);
	
		rc = getsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &optval, &optlen);
	
		qb_util_log(LOG_TRACE, "%d: getsockopt(%d, SO_SNDBUF, needed:%d) actual:%d",
			rc, sockfd, max_msg_size, optval);
	
		/* The optval <= max_msg_size check is weird...
		 * during testing it was discovered in some instances if the
		 * default optval is exactly equal to our max_msg_size, we couldn't
		 * actually send a message that large unless we explicitly set
		 * it using setsockopt... there is no good explaination for this. Most
		 * likely this is hitting some sort of "off by one" error in the kernel. */
		if (rc == 0 && optval <= max_msg_size) {
			optval = max_msg_size;
			optlen = sizeof(optval);
			rc = setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &optval, optlen);
		}
	
		if (rc != 0) {
			return -errno;
		}
	
		rc = getsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &optval, &optlen);
	
		qb_util_log(LOG_TRACE, "%d: getsockopt(%d, SO_RCVBUF, needed:%d) actual:%d",
			rc, sockfd, max_msg_size, optval);
	
		/* Set the sockets receive buffer size to match the send buffer.  On
		 * FreeBSD without this calls to sendto() will result in an ENOBUFS error
		 * if the message is larger than net.local.dgram.recvspace sysctl. */
		if (rc == 0 && optval <= max_msg_size) {
			optval = max_msg_size;
			optlen = sizeof(optval);
			rc = setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &optval, optlen);
		}
	
		if (rc != 0) {
			return -errno;
		}
	
		return rc;
	}
	
	static int32_t
	dgram_verify_msg_size(size_t max_msg_size)
	{
		int32_t rc = -1;
		int32_t sockets[2];
		int32_t tries = 0;
		int32_t write_passed = 0;
		int32_t read_passed = 0;
		char buf[max_msg_size];
		memset (buf, 0, max_msg_size);
	
		if (socketpair(AF_UNIX, SOCK_DGRAM, 0, sockets) < 0) {
			qb_util_perror(LOG_DEBUG, "error calling socketpair()");
			goto cleanup_socks;
		}
	
		if (set_sock_size(sockets[0], max_msg_size) != 0) {
			qb_util_log(LOG_DEBUG, "error set_sock_size(sockets[0],%#x)",
				    max_msg_size);
			goto cleanup_socks;
		}
		if (set_sock_size(sockets[1], max_msg_size) != 0) {
			qb_util_log(LOG_DEBUG, "error set_sock_size(sockets[1],%#x)",
				    max_msg_size);
			goto cleanup_socks;
		}
	
		for (tries = 0; tries < 3; tries++) {
	
			if (write_passed == 0) {
				rc = write(sockets[1], buf, max_msg_size);
	
				if (rc < 0 && (errno == EAGAIN || errno == EINTR)) {
					continue;
				} else if (rc == max_msg_size) {
					write_passed = 1;
				} else {
					break;
				}
			}
	
			if (read_passed == 0) {
				rc = read(sockets[0], buf, max_msg_size);
	
				if (rc < 0 && (errno == EAGAIN || errno == EINTR)) {
					continue;
				} else if (rc == max_msg_size) {
					read_passed = 1;
				} else {
					break;
				}
			}
	
			if (read_passed && write_passed) {
				rc = 0;
				break;
			}
		}
	
	
	cleanup_socks:
		close(sockets[0]);
		close(sockets[1]);
		return rc;
	}
	
	int32_t
	qb_ipcc_verify_dgram_max_msg_size(size_t max_msg_size)
	{
		int32_t i;

		int32_t last = -1;
		int32_t inc = 2048;
	

		if (dgram_verify_msg_size(max_msg_size) == 0) {
			return max_msg_size;
		}
	

		for (i = inc; i < max_msg_size; i+=inc) {
			if (dgram_verify_msg_size(i) == 0) {
				last = i;
			} else if (inc >= 512) {
				i-=inc;
				inc = inc/2;
			} else {
				break;
			}
		}
	

		return last;
	}
	
	/*
	 * bind to "base_name-local_name"
	 * connect to "base_name-remote_name"
	 * output sock_pt
	 */
	static int32_t
	qb_ipc_dgram_sock_connect(const char *base_name,
				  const char *local_name,
				  const char *remote_name,
				  int32_t max_msg_size, int32_t * sock_pt, gid_t gid)
	{
		char sock_path[PATH_MAX];
		struct sockaddr_un remote_address;
		int32_t res = qb_ipc_dgram_sock_setup(base_name, local_name,
						      sock_pt, gid);
		if (res < 0) {
			return res;
		}
	
		snprintf(sock_path, PATH_MAX, "%s-%s", base_name, remote_name);
		set_sock_addr(&remote_address, sock_path);
		if (connect(*sock_pt, (struct sockaddr *)&remote_address,
			    QB_SUN_LEN(&remote_address)) == -1) {
			res = -errno;
			goto error_connect;
		}
	
		return set_sock_size(*sock_pt, max_msg_size);
	
	error_connect:
		close(*sock_pt);
		*sock_pt = -1;
	
		return res;
	}
	
	static int32_t
	_finish_connecting(struct qb_ipc_one_way *one_way)
	{
		struct sockaddr_un remote_address;
		int res;
		int error;
		int retry = 0;
	
		set_sock_addr(&remote_address, one_way->u.us.sock_name);
	
		/* this retry loop is here to help connecting when trying to send
		 * an event right after connection setup.
		 */
		do {
			errno = 0;
			res = connect(one_way->u.us.sock,
				      (struct sockaddr *)&remote_address,
				      QB_SUN_LEN(&remote_address));
			if (res == -1) {
				error = -errno;
				qb_util_perror(LOG_DEBUG, "error calling connect()");
				retry++;
				usleep(100000);
			}
		} while (res == -1 && retry < 10);
		if (res == -1) {
			return error;
		}
	
		/* Beside disposing no longer needed value, this also signals that
		   we are done with connect-on-send arrangement at the server side
		   (i.e. for response and event channels). */
		free(one_way->u.us.sock_name);
		one_way->u.us.sock_name = NULL;
	
		return set_sock_size(one_way->u.us.sock, one_way->max_msg_size);
	}
	
	/*
	 * client functions
	 * --------------------------------------------------------
	 */
	static void
	qb_ipcc_us_disconnect(struct qb_ipcc_connection *c)
	{
		munmap(c->request.u.us.shared_data, SHM_CONTROL_SIZE);
		unlink(c->request.u.us.shared_file_name);
	
		if (use_filesystem_sockets()) {
			struct sockaddr_un un_addr;
			socklen_t un_addr_len = sizeof(struct sockaddr_un);
			char *base_name;
			char sock_name[PATH_MAX];
			size_t length;
			if (getsockname(c->response.u.us.sock, (struct sockaddr *)&un_addr, &un_addr_len) == 0) {
				length = strlen(un_addr.sun_path);
				base_name = strndup(un_addr.sun_path,
						    length - /* strlen("-response") */ 9);
				qb_util_log(LOG_DEBUG, "unlinking socket bound files with base_name=%s length=%d",base_name,length);
				snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"request");
				qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
				unlink(sock_name);
				snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"event");
				qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
				unlink(sock_name);
				snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"event-tx");
				qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
				unlink(sock_name);
				snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"response");
				qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
				unlink(sock_name);
				free(base_name);
			}
		}
	
		/* Last-ditch attempt to tidy up after ourself */
		remove_tempdir(c->request.u.us.shared_file_name);
	
		qb_ipcc_us_sock_close(c->event.u.us.sock);
		qb_ipcc_us_sock_close(c->request.u.us.sock);
		qb_ipcc_us_sock_close(c->setup.u.us.sock);
	}
	
	static ssize_t
	qb_ipc_socket_send(struct qb_ipc_one_way *one_way,
			   const void *msg_ptr, size_t msg_len)
	{
		ssize_t rc = 0;
		struct ipc_us_control *ctl;
		ctl = (struct ipc_us_control *)one_way->u.us.shared_data;
	
		if (one_way->u.us.sock_name) {
			rc = _finish_connecting(one_way);
			if (rc < 0) {
				qb_util_log(LOG_ERR, "socket connect-on-send");
				return rc;
			}
		}
	
		qb_sigpipe_ctl(QB_SIGPIPE_IGNORE);
		rc = send(one_way->u.us.sock, msg_ptr, msg_len, MSG_NOSIGNAL);
		if (rc == -1) {
			rc = -errno;
			if (errno != EAGAIN && errno != ENOBUFS) {
				qb_util_perror(LOG_DEBUG, "socket_send:send");
			}
		}
		qb_sigpipe_ctl(QB_SIGPIPE_DEFAULT);
	
		if (ctl && rc == msg_len) {
			qb_atomic_int_inc(&ctl->sent);
		}
	
		return rc;
	}
	
	static ssize_t
	qb_ipc_socket_sendv(struct qb_ipc_one_way *one_way, const struct iovec *iov,
			    size_t iov_len)
	{
		int32_t rc;
		struct ipc_us_control *ctl;
		ctl = (struct ipc_us_control *)one_way->u.us.shared_data;
	
		qb_sigpipe_ctl(QB_SIGPIPE_IGNORE);
	
		if (one_way->u.us.sock_name) {
			rc = _finish_connecting(one_way);
			if (rc < 0) {
				qb_util_perror(LOG_ERR, "socket connect-on-sendv");
				qb_sigpipe_ctl(QB_SIGPIPE_DEFAULT);
				return rc;
			}
		}
	
		rc = writev(one_way->u.us.sock, iov, iov_len);
	
		if (rc == -1) {
			rc = -errno;
			if (errno != EAGAIN && errno != ENOBUFS) {
				qb_util_perror(LOG_DEBUG, "socket_sendv:writev %d",
					       one_way->u.us.sock);
			}
		}
	
		qb_sigpipe_ctl(QB_SIGPIPE_DEFAULT);
	
		if (ctl && rc > 0) {
			qb_atomic_int_inc(&ctl->sent);
		}
		return rc;
	}
	
	/*
	 * recv a message of unknown size.
	 */
	static ssize_t
	qb_ipc_us_recv_at_most(struct qb_ipc_one_way *one_way,
			       void *msg, size_t len, int32_t timeout)
	{
		int32_t result;
		int32_t final_rc = 0;
		int32_t to_recv = 0;
		char *data = msg;
		struct ipc_us_control *ctl = NULL;
		int32_t time_waited = 0;
		int32_t time_to_wait = timeout;
	
		if (timeout == -1) {
			time_to_wait = 1000;
		}
	
		qb_sigpipe_ctl(QB_SIGPIPE_IGNORE);
	
	retry_peek:
		result = recv(one_way->u.us.sock, data,
			      sizeof(struct qb_ipc_request_header),
			      MSG_NOSIGNAL | MSG_PEEK);
	
		if (result == -1) {
	
			if (errno != EAGAIN) {
				final_rc = -errno;
				if (use_filesystem_sockets()) {
					if (errno == ECONNRESET || errno == EPIPE) {
						final_rc = -ENOTCONN;
					}
				}
				goto cleanup_sigpipe;
			}
	
			/* check to see if we have enough time left to try again */
			if (time_waited < timeout || timeout == -1) {
				result = qb_ipc_us_ready(one_way, NULL, time_to_wait, POLLIN);
				if (qb_ipc_us_sock_error_is_disconnected(result)) {
					final_rc = result;
					goto cleanup_sigpipe;
				}
				time_waited += time_to_wait;
				goto retry_peek;
			} else if (time_waited >= timeout) {
				final_rc = -ETIMEDOUT;
				goto cleanup_sigpipe;
			}
		}
		if (result >= sizeof(struct qb_ipc_request_header)) {
			struct qb_ipc_request_header *hdr = NULL;
			hdr = (struct qb_ipc_request_header *)msg;
			to_recv = hdr->size;
		}
	
		result = recv(one_way->u.us.sock, data, to_recv,
			      MSG_NOSIGNAL | MSG_WAITALL);
		if (result == -1) {
			final_rc = -errno;
			goto cleanup_sigpipe;
		} else if (result == 0) {
			qb_util_log(LOG_DEBUG, "recv == 0 -> ENOTCONN");
	
			final_rc = -ENOTCONN;
			goto cleanup_sigpipe;
		}
	
		final_rc = result;
	
		ctl = (struct ipc_us_control *)one_way->u.us.shared_data;
		if (ctl) {
			(void)qb_atomic_int_dec_and_test(&ctl->sent);
		}
	
	cleanup_sigpipe:
		qb_sigpipe_ctl(QB_SIGPIPE_DEFAULT);
		return final_rc;
	}
	
	static void
	qb_ipc_us_fc_set(struct qb_ipc_one_way *one_way, int32_t fc_enable)
	{
		struct ipc_us_control *ctl =
		    (struct ipc_us_control *)one_way->u.us.shared_data;
	
		qb_util_log(LOG_TRACE, "setting fc to %d", fc_enable);
		qb_atomic_int_set(&ctl->flow_control, fc_enable);
	}
	
	static int32_t
	qb_ipc_us_fc_get(struct qb_ipc_one_way *one_way)
	{
		struct ipc_us_control *ctl =
		    (struct ipc_us_control *)one_way->u.us.shared_data;
	
		return qb_atomic_int_get(&ctl->flow_control);
	}
	
	static ssize_t
	qb_ipc_us_q_len_get(struct qb_ipc_one_way *one_way)
	{
		struct ipc_us_control *ctl =
		    (struct ipc_us_control *)one_way->u.us.shared_data;
		return qb_atomic_int_get(&ctl->sent);
	}
	
	int32_t
	qb_ipcc_us_connect(struct qb_ipcc_connection * c,
			   struct qb_ipc_connection_response * r)
	{
		int32_t res;
		char path[PATH_MAX];
		int32_t fd_hdr;
		char *shm_ptr;
	
		qb_atomic_init();
	
		c->needs_sock_for_poll = QB_FALSE;
		c->funcs.send = qb_ipc_socket_send;
		c->funcs.sendv = qb_ipc_socket_sendv;
		c->funcs.recv = qb_ipc_us_recv_at_most;
		c->funcs.fc_get = qb_ipc_us_fc_get;
		c->funcs.disconnect = qb_ipcc_us_disconnect;
	
		fd_hdr = qb_sys_mmap_file_open(path, r->request,
					       SHM_CONTROL_SIZE, O_RDWR);
		if (fd_hdr < 0) {
			res = fd_hdr;
			errno = -fd_hdr;
			qb_util_perror(LOG_ERR, "couldn't open file for mmap");
			return res;
		}
		(void)strlcpy(c->request.u.us.shared_file_name, r->request, NAME_MAX);
		shm_ptr = mmap(0, SHM_CONTROL_SIZE,
			       PROT_READ | PROT_WRITE, MAP_SHARED, fd_hdr, 0);
	
		if (shm_ptr == MAP_FAILED) {
			res = -errno;
			qb_util_perror(LOG_ERR, "couldn't create mmap for header");
			goto cleanup_hdr;
		}
		c->request.u.us.shared_data = shm_ptr;
		c->response.u.us.shared_data = shm_ptr + sizeof(struct ipc_us_control);
		c->event.u.us.shared_data =  shm_ptr + (2 * sizeof(struct ipc_us_control));
	
		close(fd_hdr);
		fd_hdr = -1;
	
		res = qb_ipc_dgram_sock_connect(r->response, "response", "request",
						r->max_msg_size, &c->request.u.us.sock, c->egid);
		if (res != 0) {
			goto cleanup_hdr;
		}
		c->response.u.us.sock = c->request.u.us.sock;
	
		res = qb_ipc_dgram_sock_connect(r->response, "event", "event-tx",
						r->max_msg_size, &c->event.u.us.sock, c->egid);
		if (res != 0) {
			goto cleanup_hdr;
		}
	
		return 0;
	
	cleanup_hdr:
		if (fd_hdr >= 0) {
			close(fd_hdr);
		}
		close(c->event.u.us.sock);
		close(c->request.u.us.sock);
		unlink(r->request);
		munmap(c->request.u.us.shared_data, SHM_CONTROL_SIZE);
		return res;
	}
	
	/*
	 * service functions
	 * --------------------------------------------------------
	 */
	static int32_t
	_sock_connection_liveliness(int32_t fd, int32_t revents, void *data)
	{
		struct qb_ipcs_connection *c = (struct qb_ipcs_connection *)data;
	
		qb_util_log(LOG_DEBUG, "LIVENESS: fd %d event %d conn (%s)",
			    fd, revents, c->description);
		if (revents & POLLNVAL) {
			qb_util_log(LOG_DEBUG, "NVAL conn (%s)", c->description);
			qb_ipcs_disconnect(c);
			return -EINVAL;
		}
		if (revents & POLLHUP) {
			qb_util_log(LOG_DEBUG, "HUP conn (%s)", c->description);
			qb_ipcs_disconnect(c);
			return -ESHUTDOWN;
		}
	
		/* If we actually get POLLIN for some reason here, it most
		 * certainly means EOF. Do a recv on the fd to detect eof and
		 * then disconnect */
		if (revents & POLLIN) {
			char buf[10];
			int res;
	
			res = recv(fd, buf, sizeof(buf), MSG_DONTWAIT);
			if (res < 0 && errno != EAGAIN && errno != EWOULDBLOCK) {
				res = -errno;
			} else if (res == 0) {
				qb_util_log(LOG_DEBUG, "EOF conn (%s)", c->description);
				res = -ESHUTDOWN;
			}
	
			if (res < 0) {
				qb_ipcs_disconnect(c);
				return res;
			}
		}
	
		return 0;
	}
	
	static int32_t
	_sock_add_to_mainloop(struct qb_ipcs_connection *c)
	{
		int res;
	
		res = c->service->poll_fns.dispatch_add(c->service->poll_priority,
							c->request.u.us.sock,
							POLLIN | POLLPRI | POLLNVAL,
							c,
							qb_ipcs_dispatch_connection_request);
	
		if (res < 0) {
			qb_util_log(LOG_ERR,
				    "Error adding socket to mainloop (%s).",
				    c->description);
			return res;
		}
	
		res = c->service->poll_fns.dispatch_add(c->service->poll_priority,
							c->setup.u.us.sock,
							POLLIN | POLLPRI | POLLNVAL,
							c, _sock_connection_liveliness);
		qb_util_log(LOG_DEBUG, "added %d to poll loop (liveness)",
			    c->setup.u.us.sock);
		if (res < 0) {
			qb_util_perror(LOG_ERR, "Error adding setupfd to mainloop");
			(void)c->service->poll_fns.dispatch_del(c->request.u.us.sock);
			return res;
		}
		return res;
	}
	
	static void
	_sock_rm_from_mainloop(struct qb_ipcs_connection *c)
	{
		(void)c->service->poll_fns.dispatch_del(c->request.u.us.sock);
		(void)c->service->poll_fns.dispatch_del(c->setup.u.us.sock);
	}
	
	static void
	qb_ipcs_us_disconnect(struct qb_ipcs_connection *c)
	{
		qb_enter();
	
		if (c->state == QB_IPCS_CONNECTION_ESTABLISHED ||
		    c->state == QB_IPCS_CONNECTION_ACTIVE) {
			_sock_rm_from_mainloop(c);
	
			/* Free the temporaries denoting which respective socket
			   name on the client's side to connect upon the first
			   send operation -- normally the variable is free'd once
			   the connection is established but there may have been
			   no chance for that.  */
			free(c->response.u.us.sock_name);
			c->response.u.us.sock_name = NULL;
	
			free(c->event.u.us.sock_name);
			c->event.u.us.sock_name = NULL;
	
			if (use_filesystem_sockets()) {
				struct sockaddr_un un_addr;
				socklen_t un_addr_len = sizeof(struct sockaddr_un);
				char *base_name;
				char sock_name[PATH_MAX];
				size_t length;
				if (getsockname(c->request.u.us.sock, (struct sockaddr *)&un_addr, &un_addr_len) == 0) {
					length = strlen(un_addr.sun_path);
					base_name = strndup(un_addr.sun_path,
							    length - /* strlen("-request") */ 8);
					qb_util_log(LOG_DEBUG, "unlinking socket bound files with base_name=%s length=%d",base_name,length);
					snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"request");
					qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
					unlink(sock_name);
					snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"event");
					qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
					unlink(sock_name);
					snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"event-tx");
					qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
					unlink(sock_name);
					snprintf(sock_name,PATH_MAX,"%s-%s",base_name,"response");
					qb_util_log(LOG_DEBUG, "unlink sock_name=%s",sock_name);
					unlink(sock_name);
					free(base_name);
				}
			}
			qb_ipcc_us_sock_close(c->setup.u.us.sock);
			qb_ipcc_us_sock_close(c->request.u.us.sock);
			qb_ipcc_us_sock_close(c->event.u.us.sock);
		}
		if (c->state == QB_IPCS_CONNECTION_SHUTTING_DOWN ||
		    c->state == QB_IPCS_CONNECTION_ACTIVE) {
			munmap(c->request.u.us.shared_data, SHM_CONTROL_SIZE);
			unlink(c->request.u.us.shared_file_name);
	
	
		}
		remove_tempdir(c->description);
	}
	
	static int32_t
	qb_ipcs_us_connect(struct qb_ipcs_service *s,
			   struct qb_ipcs_connection *c,
			   struct qb_ipc_connection_response *r)
	{
		char path[PATH_MAX];
		int32_t fd_hdr;
		int32_t res = 0;
		struct ipc_us_control *ctl;
		char *shm_ptr;
	
		qb_util_log(LOG_DEBUG, "connecting to client (%s)", c->description);
	
		c->request.u.us.sock = c->setup.u.us.sock;
		c->response.u.us.sock = c->setup.u.us.sock;
	
		snprintf(r->request, NAME_MAX, "%s-control-%s",
			 c->description, s->name);
		snprintf(r->response, NAME_MAX, "%s-%s", c->description, s->name);
	
		fd_hdr = qb_sys_mmap_file_open(path, r->request,
					       SHM_CONTROL_SIZE,
					       O_CREAT | O_TRUNC | O_RDWR | O_EXCL);
		if (fd_hdr < 0) {
			res = fd_hdr;
			errno = -fd_hdr;
			qb_util_perror(LOG_ERR, "couldn't create file for mmap (%s)",
				       c->description);
			return res;
		}
		(void)strlcpy(r->request, path, PATH_MAX);
		(void)strlcpy(c->request.u.us.shared_file_name, r->request, NAME_MAX);
		/* These aren't failures, but are worth logging */
		res = chown(r->request, c->auth.uid, c->auth.gid);
		if (res != 0) {
			res = 0;
			qb_util_perror(LOG_ERR, "failed to chown shared memory file (%s)",
				       r->request);
			goto cleanup_hdr;
		}
		res = chmod(r->request, c->auth.mode);
		if (res != 0) {
			res = 0;
			qb_util_perror(LOG_ERR, "failed to chmod shared memory file (%s)",
				       r->request);
			goto cleanup_hdr;
		}
	
		shm_ptr = mmap(0, SHM_CONTROL_SIZE,
			       PROT_READ | PROT_WRITE, MAP_SHARED, fd_hdr, 0);
	
		if (shm_ptr == MAP_FAILED) {
			res = -errno;
			qb_util_perror(LOG_ERR, "couldn't create mmap for header (%s)",
				       c->description);
			goto cleanup_hdr;
		}
		c->request.u.us.shared_data = shm_ptr;
		c->response.u.us.shared_data = shm_ptr + sizeof(struct ipc_us_control);
		c->event.u.us.shared_data =  shm_ptr + (2 * sizeof(struct ipc_us_control));
	
		ctl = (struct ipc_us_control *)c->request.u.us.shared_data;
		ctl->sent = 0;
		ctl->flow_control = 0;
		ctl = (struct ipc_us_control *)c->response.u.us.shared_data;
		ctl->sent = 0;
		ctl->flow_control = 0;
		ctl = (struct ipc_us_control *)c->event.u.us.shared_data;
		ctl->sent = 0;
		ctl->flow_control = 0;
	
		close(fd_hdr);
		fd_hdr = -1;
	
		/* request channel */
		res = qb_ipc_dgram_sock_setup(r->response, "request",
					      &c->request.u.us.sock, c->egid);
		if (res < 0) {
			goto cleanup_hdr;
		}
	
		res = set_sock_size(c->request.u.us.sock, c->request.max_msg_size);
		if (res != 0) {
			goto cleanup_hdr;
		}
	
		c->setup.u.us.sock_name = NULL;
		c->request.u.us.sock_name = NULL;
	
		/* response channel */
		c->response.u.us.sock = c->request.u.us.sock;
		snprintf(path, PATH_MAX, "%s-%s", r->response, "response");
		c->response.u.us.sock_name = strdup(path);
		if (c->response.u.us.sock_name == NULL) {
			res = -ENOMEM;
			goto cleanup_hdr;
		}
	
		/* event channel */
		res = qb_ipc_dgram_sock_setup(r->response, "event-tx",
					      &c->event.u.us.sock, c->egid);
		if (res < 0) {
			goto cleanup_hdr;
		}
	
		res = set_sock_size(c->event.u.us.sock, c->event.max_msg_size);
		if (res != 0) {
			goto cleanup_hdr;
		}
	
		snprintf(path, PATH_MAX, "%s-%s", r->response, "event");
		c->event.u.us.sock_name = strdup(path);
		if (c->event.u.us.sock_name == NULL) {
			res = -ENOMEM;
			goto cleanup_hdr;
		}
	
		res = _sock_add_to_mainloop(c);
		if (res < 0) {
			goto cleanup_hdr;
		}
	
		return res;
	
	cleanup_hdr:
		free(c->response.u.us.sock_name);
		free(c->event.u.us.sock_name);
	
		if (fd_hdr >= 0) {
			close(fd_hdr);
		}
		unlink(r->request);
		munmap(c->request.u.us.shared_data, SHM_CONTROL_SIZE);
		return res;
	}
	
	void
	qb_ipcs_us_init(struct qb_ipcs_service *s)
	{
		s->funcs.connect = qb_ipcs_us_connect;
		s->funcs.disconnect = qb_ipcs_us_disconnect;
	
		s->funcs.recv = qb_ipc_us_recv_at_most;
		s->funcs.peek = NULL;
		s->funcs.reclaim = NULL;
		s->funcs.send = qb_ipc_socket_send;
		s->funcs.sendv = qb_ipc_socket_sendv;
	
		s->funcs.fc_set = qb_ipc_us_fc_set;
		s->funcs.q_len_get = qb_ipc_us_q_len_get;
	
		s->needs_sock_for_poll = QB_FALSE;
	
		qb_atomic_init();
	}