102 lines
2.0 KiB
C
102 lines
2.0 KiB
C
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include <sys/time.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include "logging.h"
|
|
#include "multicast.h"
|
|
#include "unixsock.h"
|
|
|
|
#define BUF_SIZE 256
|
|
|
|
void selector()
|
|
{
|
|
fd_set fdsel;
|
|
|
|
char *buf = malloc(BUF_SIZE);
|
|
if (buf == NULL) {
|
|
log_print(LOG_ERROR, "selector: out of memory");
|
|
return;
|
|
}
|
|
|
|
int usock = sock_init();
|
|
if (usock < 0)
|
|
return;
|
|
|
|
int msock = mcast_init();
|
|
if (msock < 0)
|
|
return;
|
|
|
|
int usock_con = -1;
|
|
|
|
while (1) {
|
|
struct timeval tv;
|
|
|
|
FD_ZERO(&fdsel);
|
|
FD_SET(usock, &fdsel);
|
|
FD_SET(msock, &fdsel);
|
|
|
|
if (usock_con != -1)
|
|
FD_SET(usock_con, &fdsel);
|
|
|
|
tv.tv_sec = 60;
|
|
tv.tv_usec = 0;
|
|
|
|
int ret = select(64, &fdsel, NULL, NULL, &tv);
|
|
if (ret < 0) {
|
|
log_print(LOG_ERROR, "selector: select()");
|
|
continue;
|
|
|
|
} else if (ret == 0) {
|
|
mcast_send(msock, "KEEPALIVE", 10);
|
|
|
|
} else if (FD_ISSET(usock, &fdsel)) {
|
|
if (usock_con != -1) {
|
|
log_print(LOG_INFO, "selector: unix socket overflow");
|
|
close(usock_con);
|
|
}
|
|
|
|
usock_con = accept(usock, NULL, NULL);
|
|
if (usock_con == -1)
|
|
log_print(LOG_ERROR, "selector: accept()");
|
|
|
|
} else if (usock_con != -1 && FD_ISSET(usock_con, &fdsel)) {
|
|
int len = read(usock_con, buf, BUF_SIZE);
|
|
if (len > 0)
|
|
mcast_send(msock, buf, len);
|
|
|
|
if (!strncmp(buf, "DELETE ", 7)) {
|
|
log_print(LOG_DEBUG, "delete '%s'", buf +7);
|
|
// exec unlink
|
|
|
|
} else {
|
|
log_print(LOG_DEBUG, "recv unknown cmd via unix socket: '%s'", buf);
|
|
}
|
|
|
|
close(usock_con);
|
|
usock_con = -1;
|
|
|
|
} else if (FD_ISSET(msock, &fdsel)) {
|
|
int len = read(msock, buf, BUF_SIZE);
|
|
if (len <= 0) {
|
|
log_print(LOG_ERROR, "selector: multicast sock closed?");
|
|
|
|
} else if (!strncmp(buf, "KEEPALIVE", 10)) {
|
|
|
|
} else if (!strncmp(buf, "DELETE ", 7)) {
|
|
log_print(LOG_DEBUG, "delete '%s'", buf +7);
|
|
// exec unlink
|
|
|
|
} else {
|
|
log_print(LOG_DEBUG, "recv unknown cmd via multicast: '%s'", buf);
|
|
}
|
|
}
|
|
}
|
|
|
|
msock_close(msock);
|
|
sock_close(usock);
|
|
}
|