launchd-329.3.2.tar.gz

[apple/launchd.git] / launchd / doc / sampled.c

#include <mach/mach.h>
#include <sys/types.h>
#include <sys/event.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <fcntl.h>
#include <syslog.h>
#include <libgen.h>

#include "launch.h"

static void
ack_mach_port(launch_data_t o, const char *name, void *context __attribute__((unused)))
{
	mach_port_t p = launch_data_get_machport(o);

	mach_port_deallocate(mach_task_self(), p);

	syslog(LOG_NOTICE, "Ignoring Mach service: %s", name);
}

int
main(void)
{
	struct timespec timeout = { 60, 0 };
	struct sockaddr_storage ss;
	socklen_t slen = sizeof(ss);
	struct kevent kev;
	launch_data_t tmp, resp, msg = launch_data_new_string(LAUNCH_KEY_CHECKIN);
	size_t i;
	int kq;

	openlog(getprogname(), LOG_PERROR|LOG_PID|LOG_CONS, LOG_DAEMON);

	if (-1 == (kq = kqueue())) {
		syslog(LOG_ERR, "kqueue(): %m");
		exit(EXIT_FAILURE);
	}

	if ((resp = launch_msg(msg)) == NULL) {
		syslog(LOG_ERR, "launch_msg(\"" LAUNCH_KEY_CHECKIN "\") IPC failure: %m");
		exit(EXIT_FAILURE);
	}

	if (LAUNCH_DATA_ERRNO == launch_data_get_type(resp)) {
		errno = launch_data_get_errno(resp);
		if (errno == EACCES)
			syslog(LOG_ERR, "Check-in failed. Did you forget to set ServiceIPC == true in your plist?");
		else
			syslog(LOG_ERR, "Check-in failed: %m");
		exit(EXIT_FAILURE);
	}

	tmp = launch_data_dict_lookup(resp, LAUNCH_JOBKEY_TIMEOUT);
	if (tmp)
		timeout.tv_sec = launch_data_get_integer(tmp);

	tmp = launch_data_dict_lookup(resp, LAUNCH_JOBKEY_MACHSERVICES);
	if (tmp) {
		launch_data_dict_iterate(tmp, ack_mach_port, NULL);
	}

	tmp = launch_data_dict_lookup(resp, LAUNCH_JOBKEY_SOCKETS);
	if (NULL == tmp) {
		syslog(LOG_ERR, "No sockets found to answer requests on!");
		exit(EXIT_FAILURE);
	}

	if (launch_data_dict_get_count(tmp) > 1) {
		syslog(LOG_WARNING, "Some sockets will be ignored!");
	}

	tmp = launch_data_dict_lookup(tmp, "SampleListeners");
	if (NULL == tmp) {
		syslog(LOG_ERR, "No known sockets found to answer requests on!");
		exit(EXIT_FAILURE);
	}

	for (i = 0; i < launch_data_array_get_count(tmp); i++) {
		launch_data_t tmpi = launch_data_array_get_index(tmp, i);

		EV_SET(&kev, launch_data_get_fd(tmpi), EVFILT_READ, EV_ADD, 0, 0, NULL);
		if (kevent(kq, &kev, 1, NULL, 0, NULL) == -1) {
			syslog(LOG_DEBUG, "kevent(): %m");
			exit(EXIT_FAILURE);
		}
	}

	launch_data_free(msg);
	launch_data_free(resp);

	for (;;) {
		FILE *c;
		int r;

		if ((r = kevent(kq, NULL, 0, &kev, 1, &timeout)) == -1) {
			syslog(LOG_ERR, "kevent(): %m");
			exit(EXIT_FAILURE);
		} else if (r == 0) {
			exit(EXIT_SUCCESS);
		}

		if ((r = accept(kev.ident, (struct sockaddr *)&ss, &slen)) == -1) {
			syslog(LOG_ERR, "accept(): %m");
			continue; /* this isn't fatal */
		}

		c = fdopen(r, "r+");

		if (c) {
			fprintf(c, "hello world!\n");
			fclose(c);
		} else {
			close(r);
		}
	}
}