#include <sys/types.h>
#include <sys/param.h>
#include <sys/time.h>
#include <sys/socket.h>
#if HAVE_SYS_WAIT_H
# include <sys/wait.h>
#endif
#ifndef WEXITSTATUS
# define WEXITSTATUS(s) ((unsigned)(s) >> 8)
#endif
#ifndef WIFEXITED
# define WIFEXITED(s) (((s) & 255) == 0)
#endif
#ifdef IPV6_INRIA_VERSION
#include <netinet/ipsec.h>
#else
#include <netinet6/ipsec.h>
#endif
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <signal.h>
#include <sys/stat.h>
#include "libpfkey.h"
#include "var.h"
#include "misc.h"
#include "vmbuf.h"
#include "plog.h"
#include "debug.h"
#include "schedule.h"
#include "session.h"
#include "grabmyaddr.h"
#include "cfparse.h"
#include "isakmp_var.h"
#include "admin_var.h"
#include "oakley.h"
#include "pfkey.h"
#include "handler.h"
#include "localconf.h"
#include "remoteconf.h"
#include "backupsa.h"
static void close_session __P((void));
static void check_rtsock __P((void *));
static void initfds __P((void));
static void init_signal __P((void));
static int set_signal __P((int sig, RETSIGTYPE (*func) __P((int))));
static void check_sigreq __P((void));
static void check_flushsa_stub __P((void *));
static void check_flushsa __P((void));
static int close_sockets __P((void));
static fd_set mask0;
static int nfds = 0;
static int sigreq = 0;
int
session(void)
{
fd_set rfds;
struct timeval *timeout;
int error;
struct myaddrs *p;
sched_init();
init_signal();
#ifdef ENABLE_ADMINPORT
if (admin_init() < 0)
exit(1);
#endif
initmyaddr();
if (isakmp_init() < 0)
exit(1);
initfds();
sigreq = 0;
while (1) {
check_sigreq();
timeout = schedular();
rfds = mask0;
error = select(nfds, &rfds, (fd_set *)0, (fd_set *)0, timeout);
if (error < 0) {
switch (errno) {
case EINTR:
continue;
default:
plog(LLV_ERROR, LOCATION, NULL,
"failed to select (%s)\n",
strerror(errno));
return -1;
}
}
#ifdef ENABLE_ADMINPORT
if (FD_ISSET(lcconf->sock_admin, &rfds))
admin_handler();
#endif
for (p = lcconf->myaddrs; p; p = p->next) {
if (!p->addr)
continue;
if (FD_ISSET(p->sock, &rfds))
if ((error = isakmp_handler(p->sock)) == -2)
break;
#ifdef IKE_NAT_T
if (p->nattsock >= 0 && FD_ISSET(p->nattsock, &rfds))
if ((error = isakmp_natt_handler(p->nattsock)) == -2)
break;
#endif
}
if (error == -2) {
if (lcconf->autograbaddr) {
isakmp_close();
initfds();
sched_new(5, check_rtsock, NULL);
continue;
} else {
isakmp_close_sockets();
isakmp_open();
}
}
if (FD_ISSET(lcconf->sock_pfkey, &rfds))
pfkey_handler();
if (lcconf->rtsock >= 0 && FD_ISSET(lcconf->rtsock, &rfds))
if (update_myaddrs() && lcconf->autograbaddr)
sched_new(5, check_rtsock, NULL);
}
}
static void
close_session()
{
flushph1();
close_sockets();
backupsa_clean();
plog(LLV_INFO, LOCATION, NULL, "racoon shutdown\n");
exit(0);
}
static void
check_rtsock(p)
void *p;
{
grab_myaddrs();
isakmp_close_unused();
autoconf_myaddrsport();
isakmp_open();
initfds();
}
static void
initfds()
{
struct myaddrs *p;
nfds = 0;
FD_ZERO(&mask0);
#ifdef ENABLE_ADMINPORT
if (lcconf->sock_admin >= FD_SETSIZE) {
plog(LLV_ERROR, LOCATION, NULL, "fd_set overrun\n");
exit(1);
}
FD_SET(lcconf->sock_admin, &mask0);
nfds = (nfds > lcconf->sock_admin ? nfds : lcconf->sock_admin);
#endif
if (lcconf->sock_pfkey >= FD_SETSIZE) {
plog(LLV_ERROR, LOCATION, NULL, "fd_set overrun\n");
exit(1);
}
FD_SET(lcconf->sock_pfkey, &mask0);
nfds = (nfds > lcconf->sock_pfkey ? nfds : lcconf->sock_pfkey);
if (lcconf->rtsock >= 0) {
if (lcconf->rtsock >= FD_SETSIZE) {
plog(LLV_ERROR, LOCATION, NULL, "fd_set overrun\n");
exit(1);
}
FD_SET(lcconf->rtsock, &mask0);
nfds = (nfds > lcconf->rtsock ? nfds : lcconf->rtsock);
}
for (p = lcconf->myaddrs; p; p = p->next) {
if (!p->addr)
continue;
if (p->sock >= FD_SETSIZE) {
plog(LLV_ERROR, LOCATION, NULL, "fd_set overrun\n");
exit(1);
}
FD_SET(p->sock, &mask0);
nfds = (nfds > p->sock ? nfds : p->sock);
#ifdef IKE_NAT_T
if (p->nattsock >= 0) {
if (p-> nattsock >= FD_SETSIZE) {
plog(LLV_ERROR, LOCATION, NULL, "fd_set overrun\n");
exit(1);
}
FD_SET(p->nattsock, &mask0);
nfds = (nfds > p->nattsock ? nfds : p->nattsock);
}
#endif
}
nfds++;
}
static int signals[] = {
SIGHUP,
SIGINT,
SIGTERM,
SIGUSR1,
SIGUSR2,
SIGCHLD,
0
};
RETSIGTYPE
signal_handler(sig)
int sig;
{
switch (sig) {
case SIGCHLD:
{
pid_t pid;
int s;
pid = wait(&s);
}
break;
#ifdef DEBUG_RECORD_MALLOCATION
case SIGUSR2:
DRM_dump();
break;
#endif
default:
sigreq = sig;
break;
}
}
extern int cfreparse(void);
static void
check_sigreq()
{
switch (sigreq) {
case 0:
return;
case SIGHUP:
if (cfreparse()) {
plog(LLV_ERROR, LOCATION, NULL,
"configuration read failed\n");
exit(1);
}
sigreq = 0;
break;
default:
plog(LLV_INFO, LOCATION, NULL, "caught signal %d\n", sigreq);
pfkey_send_flush(lcconf->sock_pfkey, SADB_SATYPE_UNSPEC);
sched_new(1, check_flushsa_stub, NULL);
sigreq = 0;
break;
}
}
static void
check_flushsa_stub(p)
void *p;
{
check_flushsa();
}
static void
check_flushsa()
{
vchar_t *buf;
struct sadb_msg *msg, *end, *next;
struct sadb_sa *sa;
caddr_t mhp[SADB_EXT_MAX + 1];
int n;
buf = pfkey_dump_sadb(SADB_SATYPE_UNSPEC);
msg = (struct sadb_msg *)buf->v;
end = (struct sadb_msg *)(buf->v + buf->l);
n = 0;
while (msg < end) {
if (PFKEY_UNUNIT64(msg->sadb_msg_len) < sizeof(*msg))
break;
next = (struct sadb_msg *)((caddr_t)msg + PFKEY_UNUNIT64(msg->sadb_msg_len));
if (msg->sadb_msg_type != SADB_DUMP) {
msg = next;
continue;
}
if (pfkey_align(msg, mhp) || pfkey_check(mhp)) {
plog(LLV_ERROR, LOCATION, NULL,
"pfkey_check (%s)\n", ipsec_strerror());
msg = next;
continue;
}
sa = (struct sadb_sa *)(mhp[SADB_EXT_SA]);
if (!sa) {
msg = next;
continue;
}
if (sa->sadb_sa_state != SADB_SASTATE_DEAD) {
n++;
msg = next;
continue;
}
msg = next;
}
if (buf) vfree(buf);
if (n) {
sched_new(1, check_flushsa_stub, NULL);
return;
}
close_session();
}
static void
init_signal()
{
int i;
for (i = 0; signals[i] != 0; i++)
if (set_signal(signals[i], signal_handler) < 0) {
plog(LLV_ERROR, LOCATION, NULL,
"failed to set_signal (%s)\n",
strerror(errno));
exit(1);
}
}
static int
set_signal(sig, func)
int sig;
RETSIGTYPE (*func) __P((int));
{
struct sigaction sa;
memset((caddr_t)&sa, 0, sizeof(sa));
sa.sa_handler = func;
sa.sa_flags = SA_RESTART;
if (sigemptyset(&sa.sa_mask) < 0)
return -1;
if (sigaction(sig, &sa, (struct sigaction *)0) < 0)
return(-1);
return 0;
}
static int
close_sockets()
{
isakmp_close();
pfkey_close(lcconf->sock_pfkey);
#ifdef ENABLE_ADMINPORT
(void)admin_close();
#endif
return 0;
}