udp_in.c   [plain text]


/*
 * Copyright (c) 2004-2011 Apple Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 * 
 * @APPLE_LICENSE_HEADER_END@
 */

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/un.h>
#include <sys/uio.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
#include <netdb.h>
#include "daemon.h"

#define forever for(;;)

#define UDP_SOCKET_NAME "NetworkListener"
#define MY_ID "udp_in"
#define MAXLINE 4096

#define MAXSOCK 16
static int nsock = 0;
static int ufd[MAXSOCK];
static dispatch_source_t ufd_src[MAXSOCK];

static char uline[MAXLINE + 1];

static dispatch_source_t in_src[MAXSOCK];
static dispatch_queue_t in_queue;

#define FMT_LEGACY 0
#define FMT_ASL 1

void 
udp_in_acceptmsg(int fd)
{
	socklen_t fromlen;
	ssize_t len;
	struct sockaddr_storage from;
	char fromstr[64], *r, *p;
	struct sockaddr_in *s4;
	struct sockaddr_in6 *s6;
	aslmsg m;

	fromlen = sizeof(struct sockaddr_storage);
	memset(&from, 0, fromlen);

	len = recvfrom(fd, uline, MAXLINE, 0, (struct sockaddr *)&from, &fromlen);
	if (len <= 0) return;

	fromstr[0] = '\0';
	r = NULL;

	if (from.ss_family == AF_INET)
	{
		s4 = (struct sockaddr_in *)&from;
		inet_ntop(from.ss_family, &(s4->sin_addr), fromstr, 64);
		r = fromstr;
		asldebug("%s: fd %d recvfrom %s len %d\n", MY_ID, fd, fromstr, len);
	}
	else if (from.ss_family == AF_INET6)
	{
		s6 = (struct sockaddr_in6 *)&from;
		inet_ntop(from.ss_family, &(s6->sin6_addr), fromstr, 64);
		r = fromstr;
		asldebug("%s: fd %d recvfrom %s len %d\n", MY_ID, fd, fromstr, len);
	}

	uline[len] = '\0';

	p = strrchr(uline, '\n');
	if (p != NULL) *p = '\0';

	m = asl_input_parse(uline, len, r, SOURCE_UDP_SOCKET);
	dispatch_async(global.work_queue, ^{ process_message(m, SOURCE_UDP_SOCKET); });
}

int
udp_in_init()
{
	int i, rbufsize, len, fd;
	launch_data_t sockets_dict, fd_array, fd_dict;
	static dispatch_once_t once;

	dispatch_once(&once, ^{
		in_queue = dispatch_queue_create(MY_ID, NULL);
	});

	asldebug("%s: init\n", MY_ID);
	if (nsock > 0) return 0;

	if (global.launch_dict == NULL)
	{
		asldebug("%s: launchd dict is NULL\n", MY_ID);
		return -1;
	}

	sockets_dict = launch_data_dict_lookup(global.launch_dict, LAUNCH_JOBKEY_SOCKETS);
	if (sockets_dict == NULL)
	{
		asldebug("%s: launchd lookup of LAUNCH_JOBKEY_SOCKETS failed\n", MY_ID);
		return -1;
	}

	fd_array = launch_data_dict_lookup(sockets_dict, UDP_SOCKET_NAME);
	if (fd_array == NULL)
	{
		asldebug("%s: launchd lookup of UDP_SOCKET_NAME failed\n", MY_ID);
		return -1;
	}

	nsock = launch_data_array_get_count(fd_array);
	if (nsock <= 0)
	{
		asldebug("%s: launchd fd array is empty\n", MY_ID);
		return -1;
	}

	for (i = 0; i < nsock; i++)
	{
		ufd[i] = -1;

		fd_dict = launch_data_array_get_index(fd_array, i);
		if (fd_dict == NULL)
		{
			asldebug("%s: launchd file discriptor array element 0 is NULL\n", MY_ID);
			return -1;
		}

		fd = launch_data_get_fd(fd_dict);

		rbufsize = 128 * 1024;
		len = sizeof(rbufsize);

		if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &rbufsize, len) < 0)
		{
			asldebug("%s: couldn't set receive buffer size for file descriptor %d: %s\n", MY_ID, fd, strerror(errno));
		}

		if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0)
		{
			asldebug("%s: couldn't set O_NONBLOCK for file descriptor %d: %s\n", MY_ID, fd, strerror(errno));
		}

		ufd[i] = fd;

		in_src[i] = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, (uintptr_t)fd, 0, in_queue);
		dispatch_source_set_event_handler(in_src[i], ^{ udp_in_acceptmsg(fd); });

		dispatch_resume(in_src[i]);
	}

	return 0;
}

/* N.B. Does NOT close fds.  They "belong" to launchd. */
int
udp_in_close(void)
{
	int i;

	if (nsock == 0) return -1;

	for (i = 0; i < nsock; i++)
	{
		if (ufd_src[i] != NULL)
		{
			dispatch_source_cancel(in_src[i]);
			dispatch_release(in_src[i]);
			in_src[i] = NULL;
		}

		if (ufd[i] != -1)
		{
			ufd[i] = -1;
		}
	}

	nsock = 0;

	return 0;
}

int
udp_in_reset(void)
{
	if (udp_in_close() != 0) return -1;
	return udp_in_init();
}