[apple/ipsec.git] / ipsec-tools / racoon / session.c

/*	$NetBSD: session.c,v 1.7.6.2 2007/08/01 11:52:22 vanhu Exp $	*/

/*	$KAME: session.c,v 1.32 2003/09/24 02:01:17 jinmei Exp $	*/

/*
 * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the project nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "config.h"

#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

#ifndef HAVE_NETINET6_IPSEC
#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 <paths.h>

#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/ip_icmp.h>

#include <resolv.h>
#include <TargetConditionals.h>
#include <vproc_priv.h>
#include <dispatch/dispatch.h>

#include "libpfkey.h"

#include "var.h"
#include "misc.h"
#include "vmbuf.h"
#include "plog.h"
#include "debug.h"
#include "plog.h"

#include "schedule.h"
#include "session.h"
#include "grabmyaddr.h"
#include "cfparse_proto.h"
#include "isakmp_var.h"
#include "isakmp_xauth.h"
#include "isakmp_cfg.h"
#include "oakley.h"
#include "pfkey.h"
#include "handler.h"
#include "localconf.h"
#include "remoteconf.h"
#ifdef ENABLE_NATT
#include "nattraversal.h"
#endif
#include "vpn_control_var.h"
#include "policy.h"
#include "algorithm.h" /* XXX ??? */

#include "sainfo.h"
#include "power_mgmt.h"

#include <NetworkExtension/NEPolicy.h>
#include <sys/proc_info.h>
#include <libproc.h>


extern pid_t racoon_pid;
extern int launchdlaunched;
static void close_session (int);
static int init_signal (void);
static int set_signal (int sig, RETSIGTYPE (*func) (int, siginfo_t *, void *));
static void check_sigreq (void);
static void check_flushsa_stub (void *);
static void check_flushsa (void);
static void auto_exit_do (void *);
static int close_sockets (void);

static volatile sig_atomic_t sigreq[NSIG + 1];
int terminated = 0;

static int64_t racoon_keepalive = -1;

dispatch_queue_t main_queue;

static NEPolicySessionRef policySession = NULL;

/*
 * This is used to (manually) update racoon's launchd keepalive, which is needed because racoon is (mostly) 
 * launched on demand and for <rdar://problem/8768510> requires a keepalive on dirty/failure exits.
 * The launchd plist can't be used for this because RunOnLoad is required to have keepalive on a failure exit.
 */
int64_t
launchd_update_racoon_keepalive (Boolean enabled)
{
	if (launchdlaunched) {		
		int64_t     val = (__typeof__(val))enabled;
		/* Set our own KEEPALIVE value */
		if (vproc_swap_integer(NULL,
							   VPROC_GSK_BASIC_KEEPALIVE,
							   &val,
							   &racoon_keepalive)) {
			plog(ASL_LEVEL_ERR, 
				 "failed to swap launchd keepalive integer %d\n", enabled);
		}
	}
	return racoon_keepalive;
}

static CFUUIDRef
copy_racoon_proc_uuid(void)
{
	struct proc_uniqidentifierinfo procu;
	CFUUIDBytes uuidBytes;
	int size = 0;

	memset(&procu, 0, sizeof(procu));
	size = proc_pidinfo(getpid(), PROC_PIDUNIQIDENTIFIERINFO, 1, &procu, PROC_PIDUNIQIDENTIFIERINFO_SIZE);
	if (size != PROC_PIDUNIQIDENTIFIERINFO_SIZE) {
		return (NULL);
	}

	memcpy(&uuidBytes, procu.p_uuid, sizeof(CFUUIDBytes));
	return CFUUIDCreateFromUUIDBytes(kCFAllocatorDefault, uuidBytes);
}

static bool
policy_session_init(void)
{
	bool success = true;
	policySession = NEPolicyCreateSession(kCFAllocatorDefault, CFSTR("racoon"), NULL, NULL);
	if (policySession == NULL) {
		return false;
	}
	
	CFUUIDRef proc_uuid = copy_racoon_proc_uuid();
	if (proc_uuid == NULL) {
		return false;
	}
	
	CFMutableArrayRef conditions = CFArrayCreateMutable(kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks);
	if (conditions) {
		CFMutableDictionaryRef uuidCondition = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
		if (uuidCondition) {
			CFDictionarySetValue(uuidCondition, kNEPolicyConditionType, kNEPolicyValPolicyConditionTypeApplication);
			CFDictionarySetValue(uuidCondition, kNEPolicyApplicationUUID, proc_uuid);
			CFArrayAppendValue(conditions, uuidCondition);
			CFRelease(uuidCondition);
		} else {
			success = false;
		}
		
		CFMutableDictionaryRef interfacesCondition = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
		if (interfacesCondition) {
			CFDictionarySetValue(interfacesCondition, kNEPolicyConditionType, kNEPolicyValPolicyConditionTypeAllInterfaces);
			CFArrayAppendValue(conditions, interfacesCondition);
			CFRelease(interfacesCondition);
		} else {
			success = false;
		}
	} else {
		success = false;
	}
	
	CFMutableDictionaryRef result = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
	if (result) {
		CFDictionaryAddValue(result, kNEPolicyResult, kNEPolicyValPolicyResultPass);
	} else {
		success = false;
	}
	
	if (success) {
		success = (NEPolicyAdd(policySession, 0, conditions, result, NULL) != kNEPolicyIDInvalid);
	}
	
	if (result) {
		CFRelease(result);
	}
	if (conditions) {
		CFRelease(conditions);
	}
	if (proc_uuid) {
		CFRelease(proc_uuid);
	}
	
	return (success && NEPolicyApply(policySession));
}

//
// Session
// 
// Initialize listening sockets, timers, vpn control etc.,
// write the PID file and call dispatch_main.
//
void
session(void)
{
	char pid_file[MAXPATHLEN];
	FILE *fp;
	int i;
    
    main_queue = dispatch_get_main_queue();

	/* initialize schedular */
	sched_init();

	/* needs to be called after schedular */
	if (init_power_mgmt() < 0) {
        plog(ASL_LEVEL_ERR, 
             "failed to initialize power-mgmt.");
		exit(1);
	}

    if (lcconf->autograbaddr == 1)
        if (pfroute_init()) {
            plog(ASL_LEVEL_ERR, "failed to initialize route socket.\n");
            exit(1);
        }
	
	if (!policy_session_init()) {
		plog(ASL_LEVEL_ERR, "failed to initialize NEPolicy session.\n");
	}
	
    if (initmyaddr()) {
        plog(ASL_LEVEL_ERR, "failed to initialize listening addresses.\n");
        exit(1);
    }
	if (isakmp_init()) {
		plog(ASL_LEVEL_ERR, "failed to initialize isakmp");
		exit(1);
	}    
#ifdef ENABLE_VPNCONTROL_PORT
	if (vpncontrol_init()) {
		plog(ASL_LEVEL_ERR, "failed to initialize vpn control port");
		//exit(1);
	}
#endif

	if (init_signal()) {
        plog(ASL_LEVEL_ERR, "failed to initialize signals.\n");
		exit(1); 
    }
    
	for (i = 0; i <= NSIG; i++)
		sigreq[i] = 0;

	/* write .pid file */
	if (!f_foreground) {
		racoon_pid = getpid();
		if (lcconf->pathinfo[LC_PATHTYPE_PIDFILE] == NULL) 
			strlcpy(pid_file, _PATH_VARRUN "racoon.pid", sizeof(pid_file));
		else if (lcconf->pathinfo[LC_PATHTYPE_PIDFILE][0] == '/') 
			strlcpy(pid_file, lcconf->pathinfo[LC_PATHTYPE_PIDFILE], sizeof(pid_file));
		else {
			strlcat(pid_file, _PATH_VARRUN, sizeof(pid_file));
			strlcat(pid_file, lcconf->pathinfo[LC_PATHTYPE_PIDFILE], sizeof(pid_file));
		} 
		fp = fopen(pid_file, "w");
		if (fp) {
			if (fchmod(fileno(fp),
				S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH) == -1) {
				plog(ASL_LEVEL_ERR, "%s", strerror(errno));
				fclose(fp);
				exit(1);
			}
			fprintf(fp, "%ld\n", (long)racoon_pid);
			fclose(fp);
		} else {
			plog(ASL_LEVEL_ERR, 
				"cannot open %s", pid_file);
		}
	}
#if !TARGET_OS_EMBEDDED
	// enable keepalive for recovery (from crashes and bad exits... after init)
	(void)launchd_update_racoon_keepalive(true);
#endif // !TARGET_OS_EMBEDDED
		
    // Off to the races!
    if (!terminated) {
        dispatch_main();
    }
            
    exit(1);    // should not be reached!!!
}


/* clear all status and exit program. */
static void
close_session(int error)
{
    sched_killall();    
    cleanup_power_mgmt();
	if ( terminated )
		ike_session_flush_all_phase2(false);
	ike_session_flush_all_phase1(false);
	close_sockets();

#if !TARGET_OS_EMBEDDED
	// a clean exit, so disable launchd keepalive
	(void)launchd_update_racoon_keepalive(false);
#endif // !TARGET_OS_EMBEDDED

	plog(ASL_LEVEL_INFO, "racoon shutdown\n");
	exit(0);
}


/*
 * waiting the termination of processing until sending DELETE message
 * for all inbound SA will complete.
 */
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);
	if (buf == NULL) {
		plog(ASL_LEVEL_DEBUG, 
		    "pfkey_dump_sadb: returned nothing.\n");
		return;
	}

	msg = ALIGNED_CAST(struct sadb_msg *)buf->v; 
	end = ALIGNED_CAST(struct sadb_msg *)(buf->v + buf->l);

	/* counting SA except of dead one. */
	n = 0;
	while (msg < end) {
		if (PFKEY_UNUNIT64(msg->sadb_msg_len) < sizeof(*msg))
			break;
		next = ALIGNED_CAST(struct sadb_msg *)((caddr_t)msg + PFKEY_UNUNIT64(msg->sadb_msg_len));    // Wcast-align fix (void*) - aligned buffer + multiple of 64
		if (msg->sadb_msg_type != SADB_DUMP) {
			msg = next;
			continue;
		}

		if (pfkey_align(msg, mhp) || pfkey_check(mhp)) {
			plog(ASL_LEVEL_ERR, 
				"pfkey_check (%s)\n", ipsec_strerror());
			msg = next;
			continue;
		}

		sa = ALIGNED_CAST(struct sadb_sa *)(mhp[SADB_EXT_SA]);       // Wcast-align fix (void*) - mhp contains pointers to aligned structs
		if (!sa) {
			msg = next;
			continue;
		}

		if (sa->sadb_sa_state != SADB_SASTATE_DEAD) {
			n++;
			msg = next;
			continue;
		}

		msg = next;
	}

	if (buf != NULL)
		vfree(buf);

	if (n) {
		sched_new(1, check_flushsa_stub, NULL);
		return;
	}

#if !TARGET_OS_EMBEDDED
	if (lcconf->vt)
		vproc_transaction_end(NULL, lcconf->vt);
#endif
    close_session(0);
}

void
auto_exit_do(void *p)
{
	plog(ASL_LEVEL_DEBUG, 
				"performing auto exit\n");
#if ENABLE_NO_SA_FLUSH
	close_session(0);
#else
	pfkey_send_flush(lcconf->sock_pfkey, SADB_SATYPE_UNSPEC);
	sched_new(1, check_flushsa_stub, NULL);
	dying();
#endif /* ENABLE_NO_SA_FLUSH */
}

void
check_auto_exit(void)
{
	if (lcconf->auto_exit_sched) {	/* exit scheduled? */
		if (lcconf->auto_exit_state != LC_AUTOEXITSTATE_ENABLED
			|| vpn_control_connected()          /* vpn control connected */
			|| policies_installed()             /* policies installed in kernel */
			|| !no_remote_configs(FALSE))	{	/* remote or anonymous configs */
			SCHED_KILL(lcconf->auto_exit_sched);
        }
	} else {								/* exit not scheduled */
		if (lcconf->auto_exit_state == LC_AUTOEXITSTATE_ENABLED
			&& !vpn_control_connected()
			&& !policies_installed()
			&& no_remote_configs(FALSE)) {
            if (lcconf->auto_exit_delay == 0) {
                auto_exit_do(NULL);		/* immediate exit */
            } else {
                lcconf->auto_exit_sched = sched_new(lcconf->auto_exit_delay, auto_exit_do, NULL);
            }
        }
	}
}

static int signals[] = {
	SIGHUP,
	SIGINT,
	SIGTERM,
	SIGUSR1,
	SIGUSR2,
	SIGPIPE,
	0
};


static void
check_sigreq()
{
	int sig;
    
	/* 
	 * XXX We are not able to tell if we got 
	 * several time the same signal. This is
	 * not a problem for the current code, 
	 * but we shall remember this limitation.
	 */
	for (sig = 0; sig <= NSIG; sig++) {
		if (sigreq[sig] == 0)
			continue;
        
		sigreq[sig]--;
		switch(sig) {
            case 0:
                return;
                
                /* Catch up childs, mainly scripts.
                 */
                                
            case SIGUSR1:
            case SIGHUP:
#ifdef ENABLE_HYBRID
                if ((isakmp_cfg_init(ISAKMP_CFG_INIT_WARM)) != 0) {
                    plog(ASL_LEVEL_ERR, 
                         "ISAKMP mode config structure reset failed, "
                         "not reloading\n");
                    return;
                }
#endif
                if ( terminated )
                    break;
				
                /*
                 * if we got a HUP... try graceful teardown of sessions before we close and reopen sockets...
                 * so that info-deletes notifications can make it to the peer.
                 */
                if (sig == SIGHUP) {
                    ike_session_flush_all_phase2(true);
                    ike_session_flush_all_phase1(true);
                }		

                /* Save old configuration, load new one...  */
                if (cfreparse(sig)) {
                    plog(ASL_LEVEL_ERR, 
                         "configuration read failed\n");
                    exit(1);
                }
                if (lcconf->logfile_param == NULL && logFileStr[0] == 0)
                    plogresetfile(lcconf->pathinfo[LC_PATHTYPE_LOGFILE]);
				            
#if TARGET_OS_EMBEDDED
                if (no_remote_configs(TRUE)) {
#if ENABLE_NO_SA_FLUSH
                    close_session(0);
#else
                    pfkey_send_flush(lcconf->sock_pfkey, SADB_SATYPE_UNSPEC);
#ifdef ENABLE_FASTQUIT
                    close_session(0);
#else
                    sched_new(1, check_flushsa_stub, NULL);
#endif /* ENABLE_FASTQUIT */
                    dying();
#endif /* ENABLE_NO_SA_FLUSH */
                }
#endif

                break;
                
            case SIGINT:
            case SIGTERM:			
                plog(ASL_LEVEL_INFO, 
                     "caught signal %d\n", sig);
#if ENABLE_NO_SA_FLUSH
                close_session(0);
#else
                pfkey_send_flush(lcconf->sock_pfkey,
                                 SADB_SATYPE_UNSPEC);
                if ( sig == SIGTERM ){
                    terminated = 1;			/* in case if it hasn't been set yet */
                    close_session(0);
                }
                else
                    sched_new(1, check_flushsa_stub, NULL);
                
				dying();
#endif /* ENABLE_NO_SA_FLUSH */
                break;
                
            default:
                plog(ASL_LEVEL_INFO, 
                     "caught signal %d\n", sig);
                break;
		}
	}
}

    
/*
 * asynchronous requests will actually dispatched in the
 * main loop in session().
 */
RETSIGTYPE
signal_handler(int sig, siginfo_t *sigi, void *ctx)
{
#if 0
    plog(ASL_LEVEL_DEBUG, 
         "%s received signal %d from pid %d uid %d\n\n",
         __FUNCTION__, sig, sigi->si_pid, sigi->si_uid);
#endif
    
    /* Do not just set it to 1, because we may miss some signals by just setting
     * values to 0/1
     */
    sigreq[sig]++;
    if ( sig == SIGTERM ){
        terminated = 1;
    }
    dispatch_async(main_queue, 
                   ^{
                        check_sigreq(); 
                   });
}


static int
init_signal()
{
	int i;

	for (i = 0; signals[i] != 0; i++) {
		if (set_signal(signals[i], signal_handler) < 0) {
			plog(ASL_LEVEL_ERR, 
				"failed to set_signal (%s)\n",
				strerror(errno));
			return (1);
		}
    }
    return 0;
}

static int
set_signal(int sig, RETSIGTYPE (*func) (int, siginfo_t *, void *))
{
	struct sigaction sa;

	memset((caddr_t)&sa, 0, sizeof(sa));
    sa.sa_sigaction = func;
	sa.sa_flags = SA_RESTART | SA_SIGINFO;

	if (sigemptyset(&sa.sa_mask) < 0)
		return -1;

	if (sigaction(sig, &sa, (struct sigaction *)0) < 0)
		return(-1);

	return 0;
}

void
fatal_error(int error)
{
    close_session(error == 0 ? -1 : error);
}

/* suspend all socket sources except pf_key */
void
dying(void)
{
    if (lcconf->rt_source)
        dispatch_suspend(lcconf->rt_source);
    if (lcconf->vpncontrol_source)
        dispatch_suspend(lcconf->vpncontrol_source);
    isakmp_suspend_sockets();
}
    
static int
close_sockets()
{
    pfroute_close();
	isakmp_close();
	pfkey_close();
#ifdef ENABLE_VPNCONTROL_PORT
	vpncontrol_close();
#endif
    
	return 0;
}
Commit	Line	Data
d1e348cf A	1	/* $NetBSD: session.c,v 1.7.6.2 2007/08/01 11:52:22 vanhu Exp $ */
d1e348cf A	2
52b7d2ce A	3	/* $KAME: session.c,v 1.32 2003/09/24 02:01:17 jinmei Exp $ */
	4
	5	/*
	6	* Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
	7	* All rights reserved.
	8	*
	9	* Redistribution and use in source and binary forms, with or without
	10	* modification, are permitted provided that the following conditions
	11	* are met:
	12	* 1. Redistributions of source code must retain the above copyright
	13	* notice, this list of conditions and the following disclaimer.
	14	* 2. Redistributions in binary form must reproduce the above copyright
	15	* notice, this list of conditions and the following disclaimer in the
	16	* documentation and/or other materials provided with the distribution.
	17	* 3. Neither the name of the project nor the names of its contributors
	18	* may be used to endorse or promote products derived from this software
	19	* without specific prior written permission.
	20	*
	21	* THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
	22	* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	23	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	24	* ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
	25	* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	26	* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
	27	* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
	28	* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
	29	* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
	30	* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
	31	* SUCH DAMAGE.
	32	*/
	33
	34	#include "config.h"
	35
	36	#include <sys/types.h>
	37	#include <sys/param.h>
	38	#include <sys/time.h>
	39	#include <sys/socket.h>
	40	#if HAVE_SYS_WAIT_H
	41	# include <sys/wait.h>
	42	#endif
	43	#ifndef WEXITSTATUS
	44	# define WEXITSTATUS(s) ((unsigned)(s) >> 8)
	45	#endif
	46	#ifndef WIFEXITED
	47	# define WIFEXITED(s) (((s) & 255) == 0)
	48	#endif
	49
	50	#ifndef HAVE_NETINET6_IPSEC
	51	#include <netinet/ipsec.h>
	52	#else
	53	#include <netinet6/ipsec.h>
	54	#endif
	55
	56	#include <stdlib.h>
	57	#include <stdio.h>
	58	#include <string.h>
	59	#include <errno.h>
	60	#ifdef HAVE_UNISTD_H
	61	#include <unistd.h>
	62	#endif
	63	#include <signal.h>
	64	#include <sys/stat.h>
	65	#include <paths.h>
	66
d1e348cf A	67	#include <netinet/in.h>
	68	#include <netinet/ip.h>
	69	#include <netinet/ip_icmp.h>
	70
	71	#include <resolv.h>
	72	#include <TargetConditionals.h>
e97d2cf9	73	#include <vproc_priv.h>
65c25746	74	#include <dispatch/dispatch.h>
d1e348cf	75
52b7d2ce A	76	#include "libpfkey.h"
	77
	78	#include "var.h"
	79	#include "misc.h"
	80	#include "vmbuf.h"
	81	#include "plog.h"
	82	#include "debug.h"
65c25746	83	#include "plog.h"
52b7d2ce A	84
	85	#include "schedule.h"
	86	#include "session.h"
	87	#include "grabmyaddr.h"
52b7d2ce A	88	#include "cfparse_proto.h"
52b7d2ce A	89	#include "isakmp_var.h"
d1e348cf A	90	#include "isakmp_xauth.h"
d1e348cf A	91	#include "isakmp_cfg.h"
52b7d2ce A	92	#include "oakley.h"
	93	#include "pfkey.h"
	94	#include "handler.h"
	95	#include "localconf.h"
	96	#include "remoteconf.h"
52b7d2ce A	97	#ifdef ENABLE_NATT
	98	#include "nattraversal.h"
	99	#endif
	100	#include "vpn_control_var.h"
	101	#include "policy.h"
d1e348cf A	102	#include "algorithm.h" /* XXX ??? */
	103
	104	#include "sainfo.h"
e8d9021d	105	#include "power_mgmt.h"
d1e348cf	106
d9c572c0 A	107	#include <NetworkExtension/NEPolicy.h>
	108	#include <sys/proc_info.h>
	109	#include <libproc.h>
d1e348cf	110
52b7d2ce A	111
52b7d2ce A	112	extern pid_t racoon_pid;
85f41bec	113	extern int launchdlaunched;
65c25746 A	114	static void close_session (int);
	115	static int init_signal (void);
	116	static int set_signal (int sig, RETSIGTYPE (func) (int, siginfo_t , void *));
	117	static void check_sigreq (void);
	118	static void check_flushsa_stub (void *);
	119	static void check_flushsa (void);
	120	static void auto_exit_do (void *);
	121	static int close_sockets (void);
	122
52b7d2ce	123	static volatile sig_atomic_t sigreq[NSIG + 1];
d1e348cf	124	int terminated = 0;
52b7d2ce	125
e97d2cf9 A	126	static int64_t racoon_keepalive = -1;
e97d2cf9 A	127
65c25746 A	128	dispatch_queue_t main_queue;
65c25746 A	129
d9c572c0 A	130	static NEPolicySessionRef policySession = NULL;
d9c572c0 A	131
e97d2cf9 A	132	/*
e97d2cf9 A	133	* This is used to (manually) update racoon's launchd keepalive, which is needed because racoon is (mostly)
e8d9021d	134	* launched on demand and for <rdar://problem/8768510> requires a keepalive on dirty/failure exits.
e97d2cf9 A	135	* The launchd plist can't be used for this because RunOnLoad is required to have keepalive on a failure exit.
	136	*/
	137	int64_t
	138	launchd_update_racoon_keepalive (Boolean enabled)
	139	{
65c25746 A	140	if (launchdlaunched) {
	141	int64_t val = (__typeof__(val))enabled;
	142	/* Set our own KEEPALIVE value */
	143	if (vproc_swap_integer(NULL,
	144	VPROC_GSK_BASIC_KEEPALIVE,
	145	&val,
	146	&racoon_keepalive)) {
	147	plog(ASL_LEVEL_ERR,
	148	"failed to swap launchd keepalive integer %d\n", enabled);
e97d2cf9 A	149	}
	150	}
	151	return racoon_keepalive;
	152	}
e97d2cf9	153
d9c572c0 A	154	static CFUUIDRef
	155	copy_racoon_proc_uuid(void)
	156	{
	157	struct proc_uniqidentifierinfo procu;
	158	CFUUIDBytes uuidBytes;
	159	int size = 0;
	160
	161	memset(&procu, 0, sizeof(procu));
	162	size = proc_pidinfo(getpid(), PROC_PIDUNIQIDENTIFIERINFO, 1, &procu, PROC_PIDUNIQIDENTIFIERINFO_SIZE);
	163	if (size != PROC_PIDUNIQIDENTIFIERINFO_SIZE) {
	164	return (NULL);
	165	}
	166
	167	memcpy(&uuidBytes, procu.p_uuid, sizeof(CFUUIDBytes));
	168	return CFUUIDCreateFromUUIDBytes(kCFAllocatorDefault, uuidBytes);
	169	}
	170
	171	static bool
	172	policy_session_init(void)
	173	{
	174	bool success = true;
	175	policySession = NEPolicyCreateSession(kCFAllocatorDefault, CFSTR("racoon"), NULL, NULL);
	176	if (policySession == NULL) {
	177	return false;
	178	}
	179
	180	CFUUIDRef proc_uuid = copy_racoon_proc_uuid();
	181	if (proc_uuid == NULL) {
	182	return false;
	183	}
	184
	185	CFMutableArrayRef conditions = CFArrayCreateMutable(kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks);
	186	if (conditions) {
	187	CFMutableDictionaryRef uuidCondition = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
	188	if (uuidCondition) {
	189	CFDictionarySetValue(uuidCondition, kNEPolicyConditionType, kNEPolicyValPolicyConditionTypeApplication);
	190	CFDictionarySetValue(uuidCondition, kNEPolicyApplicationUUID, proc_uuid);
	191	CFArrayAppendValue(conditions, uuidCondition);
	192	CFRelease(uuidCondition);
	193	} else {
	194	success = false;
	195	}
	196
	197	CFMutableDictionaryRef interfacesCondition = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
	198	if (interfacesCondition) {
	199	CFDictionarySetValue(interfacesCondition, kNEPolicyConditionType, kNEPolicyValPolicyConditionTypeAllInterfaces);
	200	CFArrayAppendValue(conditions, interfacesCondition);
	201	CFRelease(interfacesCondition);
	202	} else {
	203	success = false;
	204	}
	205	} else {
	206	success = false;
	207	}
	208
	209	CFMutableDictionaryRef result = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
	210	if (result) {
	211	CFDictionaryAddValue(result, kNEPolicyResult, kNEPolicyValPolicyResultPass);
	212	} else {
	213	success = false;
	214	}
	215
	216	if (success) {
	217	success = (NEPolicyAdd(policySession, 0, conditions, result, NULL) != kNEPolicyIDInvalid);
218	}
219
220	if (result) {
221	CFRelease(result);
222	}
223	if (conditions) {
224	CFRelease(conditions);
225	}
226	if (proc_uuid) {
227	CFRelease(proc_uuid);
228	}
229
230	return (success && NEPolicyApply(policySession));
231	}
232
65c25746 A	233	//
	234	// Session
	235	//
	236	// Initialize listening sockets, timers, vpn control etc.,
	237	// write the PID file and call dispatch_main.
	238	//
	239	void
52b7d2ce A	240	session(void)
52b7d2ce A	241	{
52b7d2ce A	242	char pid_file[MAXPATHLEN];
52b7d2ce A	243	FILE *fp;
65c25746 A	244	int i;
	245
	246	main_queue = dispatch_get_main_queue();
52b7d2ce A	247
	248	/* initialize schedular */
	249	sched_init();
	250
e8d9021d A	251	/* needs to be called after schedular */
e8d9021d A	252	if (init_power_mgmt() < 0) {
65c25746 A	253	plog(ASL_LEVEL_ERR,
65c25746 A	254	"failed to initialize power-mgmt.");
52b7d2ce A	255	exit(1);
52b7d2ce A	256	}
65c25746 A	257
	258	if (lcconf->autograbaddr == 1)
	259	if (pfroute_init()) {
	260	plog(ASL_LEVEL_ERR, "failed to initialize route socket.\n");
	261	exit(1);
	262	}
d9c572c0 A	263
	264	if (!policy_session_init()) {
	265	plog(ASL_LEVEL_ERR, "failed to initialize NEPolicy session.\n");
	266	}
	267
65c25746 A	268	if (initmyaddr()) {
	269	plog(ASL_LEVEL_ERR, "failed to initialize listening addresses.\n");
	270	exit(1);
	271	}
	272	if (isakmp_init()) {
	273	plog(ASL_LEVEL_ERR, "failed to initialize isakmp");
52b7d2ce	274	exit(1);
65c25746	275	}
52b7d2ce	276	#ifdef ENABLE_VPNCONTROL_PORT
65c25746 A	277	if (vpncontrol_init()) {
	278	plog(ASL_LEVEL_ERR, "failed to initialize vpn control port");
	279	//exit(1);
52b7d2ce A	280	}
	281	#endif
	282
65c25746 A	283	if (init_signal()) {
	284	plog(ASL_LEVEL_ERR, "failed to initialize signals.\n");
	285	exit(1);
	286	}
	287
52b7d2ce A	288	for (i = 0; i <= NSIG; i++)
	289	sigreq[i] = 0;
	290
	291	/* write .pid file */
	292	if (!f_foreground) {
	293	racoon_pid = getpid();
	294	if (lcconf->pathinfo[LC_PATHTYPE_PIDFILE] == NULL)
d1e348cf	295	strlcpy(pid_file, _PATH_VARRUN "racoon.pid", sizeof(pid_file));
52b7d2ce	296	else if (lcconf->pathinfo[LC_PATHTYPE_PIDFILE][0] == '/')
d1e348cf	297	strlcpy(pid_file, lcconf->pathinfo[LC_PATHTYPE_PIDFILE], sizeof(pid_file));
52b7d2ce	298	else {
d1e348cf A	299	strlcat(pid_file, _PATH_VARRUN, sizeof(pid_file));
d1e348cf A	300	strlcat(pid_file, lcconf->pathinfo[LC_PATHTYPE_PIDFILE], sizeof(pid_file));
52b7d2ce A	301	}
	302	fp = fopen(pid_file, "w");
	303	if (fp) {
	304	if (fchmod(fileno(fp),
	305	S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IROTH) == -1) {
65c25746	306	plog(ASL_LEVEL_ERR, "%s", strerror(errno));
52b7d2ce A	307	fclose(fp);
	308	exit(1);
	309	}
	310	fprintf(fp, "%ld\n", (long)racoon_pid);
	311	fclose(fp);
	312	} else {
65c25746	313	plog(ASL_LEVEL_ERR,
52b7d2ce A	314	"cannot open %s", pid_file);
	315	}
	316	}
e97d2cf9 A	317	#if !TARGET_OS_EMBEDDED
	318	// enable keepalive for recovery (from crashes and bad exits... after init)
	319	(void)launchd_update_racoon_keepalive(true);
	320	#endif // !TARGET_OS_EMBEDDED
e97d2cf9	321
65c25746 A	322	// Off to the races!
	323	if (!terminated) {
	324	dispatch_main();
	325	}
	326
	327	exit(1); // should not be reached!!!
52b7d2ce A	328	}
	329
	330
	331	/* clear all status and exit program. */
	332	static void
65c25746	333	close_session(int error)
52b7d2ce	334	{
65c25746 A	335	sched_killall();
65c25746 A	336	cleanup_power_mgmt();
d1e348cf	337	if ( terminated )
65c25746 A	338	ike_session_flush_all_phase2(false);
65c25746 A	339	ike_session_flush_all_phase1(false);
52b7d2ce	340	close_sockets();
52b7d2ce	341
e97d2cf9 A	342	#if !TARGET_OS_EMBEDDED
	343	// a clean exit, so disable launchd keepalive
	344	(void)launchd_update_racoon_keepalive(false);
	345	#endif // !TARGET_OS_EMBEDDED
e97d2cf9	346
65c25746	347	plog(ASL_LEVEL_INFO, "racoon shutdown\n");
52b7d2ce A	348	exit(0);
	349	}
	350
52b7d2ce A	351
	352	/*
	353	* waiting the termination of processing until sending DELETE message
	354	* for all inbound SA will complete.
	355	*/
	356	static void
	357	check_flushsa_stub(p)
	358	void *p;
	359	{
52b7d2ce A	360	check_flushsa();
	361	}
	362
	363	static void
	364	check_flushsa()
	365	{
	366	vchar_t *buf;
	367	struct sadb_msg msg, end, *next;
	368	struct sadb_sa *sa;
	369	caddr_t mhp[SADB_EXT_MAX + 1];
	370	int n;
	371
	372	buf = pfkey_dump_sadb(SADB_SATYPE_UNSPEC);
	373	if (buf == NULL) {
65c25746	374	plog(ASL_LEVEL_DEBUG,
52b7d2ce A	375	"pfkey_dump_sadb: returned nothing.\n");
	376	return;
	377	}
	378
85f41bec A	379	msg = ALIGNED_CAST(struct sadb_msg *)buf->v;
85f41bec A	380	end = ALIGNED_CAST(struct sadb_msg *)(buf->v + buf->l);
52b7d2ce A	381
	382	/* counting SA except of dead one. */
	383	n = 0;
	384	while (msg < end) {
	385	if (PFKEY_UNUNIT64(msg->sadb_msg_len) < sizeof(*msg))
	386	break;
85f41bec	387	next = ALIGNED_CAST(struct sadb_msg )((caddr_t)msg + PFKEY_UNUNIT64(msg->sadb_msg_len)); // Wcast-align fix (void) - aligned buffer + multiple of 64
52b7d2ce A	388	if (msg->sadb_msg_type != SADB_DUMP) {
	389	msg = next;
	390	continue;
	391	}
	392
	393	if (pfkey_align(msg, mhp) \|\| pfkey_check(mhp)) {
65c25746	394	plog(ASL_LEVEL_ERR,
52b7d2ce A	395	"pfkey_check (%s)\n", ipsec_strerror());
	396	msg = next;
	397	continue;
	398	}
	399
85f41bec	400	sa = ALIGNED_CAST(struct sadb_sa )(mhp[SADB_EXT_SA]); // Wcast-align fix (void) - mhp contains pointers to aligned structs
52b7d2ce A	401	if (!sa) {
	402	msg = next;
	403	continue;
	404	}
	405
	406	if (sa->sadb_sa_state != SADB_SASTATE_DEAD) {
	407	n++;
	408	msg = next;
	409	continue;
	410	}
	411
	412	msg = next;
	413	}
	414
	415	if (buf != NULL)
	416	vfree(buf);
	417
	418	if (n) {
	419	sched_new(1, check_flushsa_stub, NULL);
	420	return;
	421	}
	422
d1e348cf A	423	#if !TARGET_OS_EMBEDDED
	424	if (lcconf->vt)
	425	vproc_transaction_end(NULL, lcconf->vt);
	426	#endif
65c25746	427	close_session(0);
52b7d2ce A	428	}
	429
	430	void
	431	auto_exit_do(void *p)
	432	{
65c25746	433	plog(ASL_LEVEL_DEBUG,
d1e348cf	434	"performing auto exit\n");
d9c572c0 A	435	#if ENABLE_NO_SA_FLUSH
	436	close_session(0);
	437	#else
52b7d2ce A	438	pfkey_send_flush(lcconf->sock_pfkey, SADB_SATYPE_UNSPEC);
52b7d2ce A	439	sched_new(1, check_flushsa_stub, NULL);
65c25746	440	dying();
d9c572c0	441	#endif /* ENABLE_NO_SA_FLUSH */
52b7d2ce A	442	}
	443
	444	void
	445	check_auto_exit(void)
	446	{
65c25746	447	if (lcconf->auto_exit_sched) { /* exit scheduled? */
52b7d2ce	448	if (lcconf->auto_exit_state != LC_AUTOEXITSTATE_ENABLED
65c25746 A	449	\|\| vpn_control_connected() /* vpn control connected */
	450	\|\| policies_installed() /* policies installed in kernel */
	451	\|\| !no_remote_configs(FALSE)) { /* remote or anonymous configs */
52b7d2ce	452	SCHED_KILL(lcconf->auto_exit_sched);
65c25746	453	}
52b7d2ce A	454	} else { /* exit not scheduled */
52b7d2ce A	455	if (lcconf->auto_exit_state == LC_AUTOEXITSTATE_ENABLED
65c25746	456	&& !vpn_control_connected()
93762ec7	457	&& !policies_installed()
65c25746 A	458	&& no_remote_configs(FALSE)) {
	459	if (lcconf->auto_exit_delay == 0) {
	460	auto_exit_do(NULL); /* immediate exit */
	461	} else {
	462	lcconf->auto_exit_sched = sched_new(lcconf->auto_exit_delay, auto_exit_do, NULL);
	463	}
	464	}
52b7d2ce A	465	}
	466	}
	467
65c25746 A	468	static int signals[] = {
	469	SIGHUP,
	470	SIGINT,
	471	SIGTERM,
	472	SIGUSR1,
	473	SIGUSR2,
	474	SIGPIPE,
	475	0
	476	};
	477
52b7d2ce A	478
52b7d2ce A	479	static void
65c25746 A	480	check_sigreq()
	481	{
	482	int sig;
	483
	484	/*
	485	* XXX We are not able to tell if we got
	486	* several time the same signal. This is
	487	* not a problem for the current code,
	488	* but we shall remember this limitation.
	489	*/
	490	for (sig = 0; sig <= NSIG; sig++) {
	491	if (sigreq[sig] == 0)
	492	continue;
	493
	494	sigreq[sig]--;
	495	switch(sig) {
	496	case 0:
	497	return;
	498
	499	/* Catch up childs, mainly scripts.
	500	*/
	501
	502	case SIGUSR1:
	503	case SIGHUP:
	504	#ifdef ENABLE_HYBRID
	505	if ((isakmp_cfg_init(ISAKMP_CFG_INIT_WARM)) != 0) {
	506	plog(ASL_LEVEL_ERR,
	507	"ISAKMP mode config structure reset failed, "
	508	"not reloading\n");
	509	return;
	510	}
	511	#endif
	512	if ( terminated )
	513	break;
	514
	515	/*
	516	* if we got a HUP... try graceful teardown of sessions before we close and reopen sockets...
	517	* so that info-deletes notifications can make it to the peer.
	518	*/
	519	if (sig == SIGHUP) {
	520	ike_session_flush_all_phase2(true);
	521	ike_session_flush_all_phase1(true);
	522	}
	523
	524	/* Save old configuration, load new one... */
	525	if (cfreparse(sig)) {
	526	plog(ASL_LEVEL_ERR,
	527	"configuration read failed\n");
	528	exit(1);
	529	}
	530	if (lcconf->logfile_param == NULL && logFileStr[0] == 0)
	531	plogresetfile(lcconf->pathinfo[LC_PATHTYPE_LOGFILE]);
	532
	533	#if TARGET_OS_EMBEDDED
	534	if (no_remote_configs(TRUE)) {
d9c572c0 A	535	#if ENABLE_NO_SA_FLUSH
	536	close_session(0);
	537	#else
65c25746 A	538	pfkey_send_flush(lcconf->sock_pfkey, SADB_SATYPE_UNSPEC);
	539	#ifdef ENABLE_FASTQUIT
	540	close_session(0);
	541	#else
	542	sched_new(1, check_flushsa_stub, NULL);
d9c572c0	543	#endif /* ENABLE_FASTQUIT */
65c25746	544	dying();
d9c572c0	545	#endif /* ENABLE_NO_SA_FLUSH */
65c25746 A	546	}
	547	#endif
	548
	549	break;
	550
	551	case SIGINT:
	552	case SIGTERM:
	553	plog(ASL_LEVEL_INFO,
	554	"caught signal %d\n", sig);
d9c572c0 A	555	#if ENABLE_NO_SA_FLUSH
	556	close_session(0);
	557	#else
	558	pfkey_send_flush(lcconf->sock_pfkey,
65c25746 A	559	SADB_SATYPE_UNSPEC);
	560	if ( sig == SIGTERM ){
	561	terminated = 1; /* in case if it hasn't been set yet */
	562	close_session(0);
	563	}
	564	else
	565	sched_new(1, check_flushsa_stub, NULL);
	566
	567	dying();
d9c572c0	568	#endif /* ENABLE_NO_SA_FLUSH */
65c25746 A	569	break;
	570
	571	default:
	572	plog(ASL_LEVEL_INFO,
	573	"caught signal %d\n", sig);
	574	break;
	575	}
	576	}
	577	}
	578
	579
	580	/*
	581	* asynchronous requests will actually dispatched in the
	582	* main loop in session().
	583	*/
	584	RETSIGTYPE
	585	signal_handler(int sig, siginfo_t sigi, void ctx)
	586	{
	587	#if 0
	588	plog(ASL_LEVEL_DEBUG,
	589	"%s received signal %d from pid %d uid %d\n\n",
	590	__FUNCTION__, sig, sigi->si_pid, sigi->si_uid);
	591	#endif
	592
	593	/* Do not just set it to 1, because we may miss some signals by just setting
	594	* values to 0/1
	595	*/
	596	sigreq[sig]++;
	597	if ( sig == SIGTERM ){
	598	terminated = 1;
	599	}
	600	dispatch_async(main_queue,
	601	^{
	602	check_sigreq();
	603	});
	604	}
	605
	606
	607	static int
52b7d2ce A	608	init_signal()
	609	{
	610	int i;
	611
65c25746	612	for (i = 0; signals[i] != 0; i++) {
52b7d2ce	613	if (set_signal(signals[i], signal_handler) < 0) {
65c25746	614	plog(ASL_LEVEL_ERR,
52b7d2ce A	615	"failed to set_signal (%s)\n",
52b7d2ce A	616	strerror(errno));
65c25746	617	return (1);
52b7d2ce	618	}
65c25746 A	619	}
65c25746 A	620	return 0;
52b7d2ce A	621	}
	622
	623	static int
65c25746	624	set_signal(int sig, RETSIGTYPE (func) (int, siginfo_t , void *))
52b7d2ce A	625	{
	626	struct sigaction sa;
	627
	628	memset((caddr_t)&sa, 0, sizeof(sa));
65c25746	629	sa.sa_sigaction = func;
e8d9021d	630	sa.sa_flags = SA_RESTART \| SA_SIGINFO;
52b7d2ce A	631
	632	if (sigemptyset(&sa.sa_mask) < 0)
	633	return -1;
	634
	635	if (sigaction(sig, &sa, (struct sigaction *)0) < 0)
	636	return(-1);
	637
	638	return 0;
	639	}
	640
65c25746 A	641	void
	642	fatal_error(int error)
	643	{
	644	close_session(error == 0 ? -1 : error);
	645	}
	646
	647	/* suspend all socket sources except pf_key */
	648	void
	649	dying(void)
	650	{
	651	if (lcconf->rt_source)
	652	dispatch_suspend(lcconf->rt_source);
	653	if (lcconf->vpncontrol_source)
	654	dispatch_suspend(lcconf->vpncontrol_source);
	655	isakmp_suspend_sockets();
	656	}
	657
52b7d2ce A	658	static int
	659	close_sockets()
	660	{
65c25746	661	pfroute_close();
52b7d2ce	662	isakmp_close();
65c25746	663	pfkey_close();
52b7d2ce A	664	#ifdef ENABLE_VPNCONTROL_PORT
	665	vpncontrol_close();
	666	#endif
65c25746	667
52b7d2ce A	668	return 0;
	669	}
	670
d1e348cf	671