/*
* $Id: ud_socket.c,v 1.6 2009/04/22 18:22:28 thockin Exp $
* A few routines for handling UNIX domain sockets
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <fcntl.h>
#include "acpid.h"
#include "log.h"
#include "ud_socket.h"
#include "libc_compat.h"
int
ud_create_socket(const char *name, mode_t socketmode)
{
int fd;
int r;
struct sockaddr_un uds_addr;
if (strnlen(name, sizeof(uds_addr.sun_path)) >
sizeof(uds_addr.sun_path) - 1) {
acpid_log(LOG_ERR, "ud_create_socket(): "
"socket filename longer than %zu characters: %s",
sizeof(uds_addr.sun_path) - 1, name);
errno = EINVAL;
return -1;
}
/* JIC */
unlink(name);
fd = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0);
if (fd < 0) {
return fd;
}
/* Clear the umask to guarantee predictable results from fchmod(). */
umask(0);
if (fchmod(fd, socketmode) < 0) {
close(fd);
acpid_log(LOG_ERR, "fchmod() on socket %s: %s",
name, strerror(errno));
return -1;
}
/* setup address struct */
memset(&uds_addr, 0, sizeof(uds_addr));
uds_addr.sun_family = AF_UNIX;
strncpy(uds_addr.sun_path, name, sizeof(uds_addr.sun_path) - 1);
/* bind it to the socket */
r = bind(fd, (struct sockaddr *)&uds_addr, sizeof(uds_addr));
if (r < 0) {
close (fd);
return r;
}
/* listen - allow 10 to queue */
r = listen(fd, 10);
if (r < 0) {
close(fd);
return r;
}
return fd;
}
int
ud_accept(int listenfd, struct ucred *cred)
{
while (1) {
int newsock = 0;
struct sockaddr_un cliaddr;
socklen_t len = sizeof(struct sockaddr_un);
newsock = TEMP_FAILURE_RETRY (accept4(listenfd, (struct sockaddr *)&cliaddr, &len, SOCK_CLOEXEC|SOCK_NONBLOCK));
if (newsock < 0) {
return newsock;
}
if (cred) {
len = sizeof(struct ucred);
getsockopt(newsock,SOL_SOCKET,SO_PEERCRED,cred,&len);
}
return newsock;
}
}
int
ud_connect(const char *name)
{
int fd;
int r;
struct sockaddr_un addr;
if (strnlen(name, sizeof(addr.sun_path)) > sizeof(addr.sun_path) - 1) {
acpid_log(LOG_ERR, "ud_connect(): "
"socket filename longer than %zu characters: %s",
sizeof(addr.sun_path) - 1, name);
errno = EINVAL;
return -1;
}
fd = socket(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
if (fd < 0) {
return fd;
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
sprintf(addr.sun_path, "%s", name);
/* safer: */
/*strncpy(addr.sun_path, name, sizeof(addr.sun_path) - 1);*/
r = connect(fd, (struct sockaddr *)&addr, sizeof(addr));
if (r < 0) {
close(fd);
return r;
}
return fd;
}
int
ud_get_peercred(int fd, struct ucred *cred)
{
socklen_t len = sizeof(struct ucred);
getsockopt(fd, SOL_SOCKET, SO_PEERCRED, cred, &len);
return 0;
}