#include <config.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <signal.h>
#include <ctype.h>
#include <sasl/sasl.h>
#include <sasl/saslutil.h>
#include <syslog.h>
#ifdef HAVE_STDARG_H
#include <stdarg.h>
#else
#include <varargs.h>
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/stat.h>
#include <netinet/in.h>
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
#include "prot.h"
#include "xmalloc.h"
#include "global.h"
#include "assert.h"
#include "imparse.h"
#include "iptostring.h"
#include "mupdate.h"
#include "mupdate_err.h"
#include "exitcodes.h"
static int open_kick_socket()
{
int r,s,len;
char fnamebuf[2048];
struct sockaddr_un srvaddr;
mode_t oldumask;
s = socket(AF_UNIX, SOCK_STREAM, 0);
if (s == -1) {
syslog(LOG_ERR, "socket: %m");
fatal("socket failed", EC_OSERR);
}
strlcpy(fnamebuf, config_dir, sizeof(fnamebuf));
strlcat(fnamebuf, FNAME_MUPDATE_TARGET_SOCK, sizeof(fnamebuf));
(void) unlink(fnamebuf);
memset((char *)&srvaddr, 0, sizeof(srvaddr));
srvaddr.sun_family = AF_UNIX;
strlcpy(srvaddr.sun_path, fnamebuf, sizeof(srvaddr.sun_path));
len = strlen(srvaddr.sun_path) + sizeof(srvaddr.sun_family) + 1;
oldumask = umask((mode_t) 0);
r = bind(s, (struct sockaddr *)&srvaddr, len);
umask(oldumask);
chmod(fnamebuf, 0777);
if (r == -1) {
syslog(LOG_ERR, "bind: %s: %m", fnamebuf);
fatal("bind failed", EC_OSERR);
}
r = listen(s, 10);
if (r == -1) {
syslog(LOG_ERR, "listen: %m");
fatal("listen failed", EC_OSERR);
}
return s;
}
static int get_kick_fds(int kicksock,
int *fd_list, int *num_fds, int max_fds) {
fd_set read_set;
int highest_fd = kicksock + 1;
struct timeval tv;
tv.tv_sec = 0;
tv.tv_usec = 0;
FD_ZERO(&read_set);
FD_SET(kicksock, &read_set);
*num_fds = 0;
for(*num_fds = 0; *num_fds<max_fds; (*num_fds)++) {
int gotdata;
fd_set rset;
rset = read_set;
gotdata = select(highest_fd, &rset, NULL, NULL, &tv);
if(gotdata == -1) {
syslog(LOG_ERR, "kicksock select failed");
return -1;
} else if (gotdata != 0 && FD_ISSET(kicksock, &rset)) {
struct sockaddr_un clientaddr;
int len = sizeof(clientaddr);
fd_list[*num_fds] =
accept(kicksock, (struct sockaddr *)&clientaddr, &len);
if (fd_list[*num_fds] == -1) {
syslog(LOG_WARNING, "kicksock accept() failed: %m %d", kicksock);
return -1;
}
} else {
break;
}
}
return 0;
}
#define KICK_FDS_LEN 5
static void mupdate_listen(mupdate_handle *handle, int pingtimeout)
{
int gotdata = 0;
fd_set rset, read_set;
int highest_fd, kicksock;
int waiting_for_noop = 0;
int kick_fds[KICK_FDS_LEN];
int num_kick_fds = 0;
if (!handle || !handle->saslcompleted) return;
mupdate_unready();
if(mupdate_synchronize(handle)) return;
mupdate_signal_db_synced();
mupdate_ready();
kicksock = open_kick_socket();
highest_fd = ((kicksock > handle->sock) ? kicksock : handle->sock) + 1;
FD_ZERO(&read_set);
FD_SET(handle->sock, &read_set);
FD_SET(kicksock, &read_set);
while(1) {
struct timeval tv;
tv.tv_sec = pingtimeout;
tv.tv_usec = 0;
prot_flush(handle->pout);
rset = read_set;
gotdata = select(highest_fd, &rset, NULL, NULL, &tv);
if(gotdata == -1) {
syslog(LOG_ERR, "select failed");
break;
} else if(gotdata != 0) {
if (FD_ISSET(handle->sock, &rset)) {
if (mupdate_scarf(handle, cmd_change, NULL,
waiting_for_noop, NULL) != 0) {
break;
}
}
if(waiting_for_noop) {
waiting_for_noop = 0;
for(;num_kick_fds;num_kick_fds--) {
if (write(kick_fds[num_kick_fds-1], "ok", 2) < 0) {
syslog(LOG_WARNING,
"can't write to IPC socket (ignoring)");
}
close(kick_fds[num_kick_fds-1]);
}
}
if (FD_ISSET(kicksock, &rset)) {
if(get_kick_fds(kicksock,
kick_fds, &num_kick_fds, KICK_FDS_LEN)) {
break;
}
prot_printf(handle->pout, "N%u NOOP\r\n", handle->tagn++);
prot_flush(handle->pout);
waiting_for_noop = 1;
}
} else {
if(!waiting_for_noop) {
prot_printf(handle->pout, "N%u NOOP\r\n", handle->tagn++);
prot_flush(handle->pout);
waiting_for_noop = 1;
} else {
syslog(LOG_ERR, "connection to master timed out.");
break;
}
}
}
for(;num_kick_fds;num_kick_fds--) {
close(kick_fds[num_kick_fds-1]);
}
close(kicksock);
}
void *mupdate_client_start(void *rock __attribute__((unused)))
{
mupdate_handle *h = NULL;
int retry_delay = 20, real_delay;
int ret;
srand(time(NULL) * getpid());
if(!config_mupdate_server) {
fatal("couldn't get mupdate server name", EC_UNAVAILABLE);
}
retry_delay = config_getint(IMAPOPT_MUPDATE_RETRY_DELAY);
if(retry_delay < 0) {
fatal("invalid value for mupdate_retry_delay", EC_UNAVAILABLE);
}
while(1) {
ret = mupdate_connect(config_mupdate_server, NULL, &h, NULL);
if(ret) {
syslog(LOG_ERR,"couldn't connect to mupdate server");
goto retry;
}
syslog(LOG_ERR, "successful mupdate connection to %s",
config_mupdate_server);
mupdate_listen(h, retry_delay);
retry:
if(h && h->pin) prot_free(h->pin);
if(h && h->pout) prot_free(h->pout);
if(h) close(h->sock);
if(h && h->saslconn) sasl_dispose(&h->saslconn);
free(h); h = NULL;
real_delay = retry_delay + (rand() % (retry_delay / 2));
syslog(LOG_ERR,
"retrying connection to mupdate server in %d seconds",
real_delay);
sleep(real_delay);
}
return NULL;
}
void *mupdate_placebo_kick_start(void *rock __attribute__((unused)))
{
int kicksock, kickconn = -1;
kicksock = open_kick_socket();
while(1) {
struct sockaddr_un clientaddr;
int len;
len = sizeof(clientaddr);
kickconn =
accept(kicksock, (struct sockaddr *)&clientaddr, &len);
if (kickconn == -1) {
syslog(LOG_WARNING, "accept(): %m");
break;
} else {
if (write(kickconn, "ok", 2) < 0) {
syslog(LOG_WARNING, "can't write to IPC socket?");
}
close(kickconn);
kickconn = -1;
}
}
if(kickconn >= 0) close(kickconn);
close(kicksock);
return NULL;
}