launchd/src/launchdebugd.c

/*
 * Copyright (c) 2005 Apple Computer, 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/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 launch_print_obj(launch_data_t o, FILE *w);

static int kq = -1;

static void find_fds(launch_data_t o, const char *key __attribute__((unused)), void *context __attribute__((unused)))
{
	struct kevent kev;
	size_t i;

	switch (launch_data_get_type(o)) {
	case LAUNCH_DATA_FD:
		EV_SET(&kev, launch_data_get_fd(o), EVFILT_READ, EV_ADD, 0, 0, NULL);
		if (kevent(kq, &kev, 1, NULL, 0, NULL) == -1)
			syslog(LOG_DEBUG, "kevent(): %m");
		break;
	case LAUNCH_DATA_ARRAY:
		for (i = 0; i < launch_data_array_get_count(o); i++)
			find_fds(launch_data_array_get_index(o, i), NULL, NULL);
		break;
	case LAUNCH_DATA_DICTIONARY:
		launch_data_dict_iterate(o, find_fds, NULL);
		break;
	default:
		break;
	}
}

int main(void)
{
	int r;
	struct sockaddr_storage ss;
	socklen_t slen = sizeof(ss);
	struct kevent kev;
	FILE *c;
	launch_data_t tmp, resp, msg = launch_data_alloc(LAUNCH_DATA_STRING);

	kq = kqueue();

	launch_data_set_string(msg, LAUNCH_KEY_CHECKIN);

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

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

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

	launch_data_free(resp);

	if ((r = kevent(kq, NULL, 0, &kev, 1, NULL)) == -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");
		exit(EXIT_FAILURE);
	}

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

	fprintf(c, "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\n\r\n");

	fprintf(c, "<html>\n<body>\n");

	launch_data_set_string(msg, LAUNCH_KEY_GETJOBS);

	resp = launch_msg(msg);

	launch_print_obj(resp, c);

	fprintf(c, "</body>\n</html>\n");

	fclose(c);

	exit(EXIT_SUCCESS);
}

static void launch_print_obj_dict_callback(launch_data_t obj, const char *key, void *context)
{
	FILE *w = context;

	fprintf(w, "<i>%s</i>\n", key);
	if (launch_data_get_type(obj) != LAUNCH_DATA_ARRAY &&
			launch_data_get_type(obj) != LAUNCH_DATA_DICTIONARY)
		fprintf(w, "<ul><li>\n");
	launch_print_obj(obj, w);
	if (launch_data_get_type(obj) != LAUNCH_DATA_ARRAY &&
			launch_data_get_type(obj) != LAUNCH_DATA_DICTIONARY)
		fprintf(w, "</li></ul>\n");
}

static void launch_print_obj(launch_data_t o, FILE *w)
{
	size_t i;


        switch (launch_data_get_type(o)) {
        case LAUNCH_DATA_DICTIONARY:
		fprintf(w, "<ul><li>\n");
		launch_data_dict_iterate(o, launch_print_obj_dict_callback, w);
		fprintf(w, "</li></ul>\n");
                break;
        case LAUNCH_DATA_ARRAY:
		fprintf(w, "<ol>\n");
                for (i = 0; i < launch_data_array_get_count(o); i++) {
			fprintf(w, "<li>");
                        launch_print_obj(launch_data_array_get_index(o, i), w);
			fprintf(w, "</li>\n");
		}
		fprintf(w, "</ol>\n");
                break;
        case LAUNCH_DATA_INTEGER:
                fprintf(w, "Number: %lld", launch_data_get_integer(o));
                break;
        case LAUNCH_DATA_REAL:
                fprintf(w, "Float: %f", launch_data_get_real(o));
                break;
        case LAUNCH_DATA_STRING:
                fprintf(w, "String: %s", launch_data_get_string(o));
                break;
        case LAUNCH_DATA_OPAQUE:
                fprintf(w, "Opaque: %p size %zu", launch_data_get_opaque(o), launch_data_get_opaque_size(o));
                break;
        case LAUNCH_DATA_FD:
                fprintf(w, "FD: %d", launch_data_get_fd(o));
                break;
        case LAUNCH_DATA_BOOL:
                fprintf(w, "Bool: %s", launch_data_get_bool(o) ? "true" : "false");
                break;
        default:
                fprintf(w, "type %d is unknown", launch_data_get_type(o));
                break;
        }
}
Commit	Line	Data
	1	/*
	2	* Copyright (c) 2005 Apple Computer, Inc. All rights reserved.
	3	*
	4	* @APPLE_LICENSE_HEADER_START@
	5	*
	6	* This file contains Original Code and/or Modifications of Original Code
	7	* as defined in and that are subject to the Apple Public Source License
	8	* Version 2.0 (the 'License'). You may not use this file except in
	9	* compliance with the License. Please obtain a copy of the License at
	10	* http://www.opensource.apple.com/apsl/ and read it before using this
	11	* file.
	12	*
	13	* The Original Code and all software distributed under the License are
	14	* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
	15	* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
	16	* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
	17	* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
	18	* Please see the License for the specific language governing rights and
	19	* limitations under the License.
	20	*
	21	* @APPLE_LICENSE_HEADER_END@
	22	*/
	23	#include <sys/types.h>
	24	#include <sys/event.h>
	25	#include <sys/socket.h>
	26	#include <sys/time.h>
	27	#include <stdio.h>
	28	#include <stdlib.h>
	29	#include <stdbool.h>
	30	#include <unistd.h>
	31	#include <string.h>
	32	#include <errno.h>
	33	#include <fcntl.h>
	34	#include <syslog.h>
	35	#include <libgen.h>
	36
	37	#include "launch.h"
	38
	39	static void launch_print_obj(launch_data_t o, FILE *w);
	40
	41	static int kq = -1;
	42
	43	static void find_fds(launch_data_t o, const char key __attribute__((unused)), void context __attribute__((unused)))
	44	{
	45	struct kevent kev;
	46	size_t i;
	47
	48	switch (launch_data_get_type(o)) {
	49	case LAUNCH_DATA_FD:
	50	EV_SET(&kev, launch_data_get_fd(o), EVFILT_READ, EV_ADD, 0, 0, NULL);
	51	if (kevent(kq, &kev, 1, NULL, 0, NULL) == -1)
	52	syslog(LOG_DEBUG, "kevent(): %m");
	53	break;
	54	case LAUNCH_DATA_ARRAY:
	55	for (i = 0; i < launch_data_array_get_count(o); i++)
	56	find_fds(launch_data_array_get_index(o, i), NULL, NULL);
	57	break;
	58	case LAUNCH_DATA_DICTIONARY:
	59	launch_data_dict_iterate(o, find_fds, NULL);
	60	break;
	61	default:
	62	break;
	63	}
	64	}
	65
	66	int main(void)
	67	{
	68	int r;
	69	struct sockaddr_storage ss;
	70	socklen_t slen = sizeof(ss);
	71	struct kevent kev;
	72	FILE *c;
	73	launch_data_t tmp, resp, msg = launch_data_alloc(LAUNCH_DATA_STRING);
	74
	75	kq = kqueue();
	76
	77	launch_data_set_string(msg, LAUNCH_KEY_CHECKIN);
	78
	79	openlog(getprogname(), LOG_PERROR\|LOG_PID\|LOG_CONS, LOG_DAEMON);
	80
	81	if ((resp = launch_msg(msg)) == NULL) {
	82	syslog(LOG_ERR, "launch_msg(\"" LAUNCH_KEY_CHECKIN "\"): %m");
	83	exit(EXIT_FAILURE);
	84	}
	85
	86	tmp = launch_data_dict_lookup(resp, LAUNCH_JOBKEY_SOCKETS);
	87	if (tmp) {
	88	find_fds(tmp, NULL, NULL);
	89	} else {
	90	syslog(LOG_ERR, "No FDs found to answer requests on!");
	91	exit(EXIT_FAILURE);
	92	}
	93
	94	launch_data_free(resp);
	95
	96	if ((r = kevent(kq, NULL, 0, &kev, 1, NULL)) == -1) {
	97	syslog(LOG_ERR, "kevent(): %m");
	98	exit(EXIT_FAILURE);
	99	} else if (r == 0) {
	100	exit(EXIT_SUCCESS);
	101	}
	102	if ((r = accept(kev.ident, (struct sockaddr *)&ss, &slen)) == -1) {
	103	syslog(LOG_ERR, "accept(): %m");
	104	exit(EXIT_FAILURE);
	105	}
	106
	107	c = fdopen(r, "r+");
	108
	109	fprintf(c, "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\n\r\n");
	110
	111	fprintf(c, "<html>\n<body>\n");
	112
	113	launch_data_set_string(msg, LAUNCH_KEY_GETJOBS);
	114
	115	resp = launch_msg(msg);
	116
	117	launch_print_obj(resp, c);
	118
	119	fprintf(c, "</body>\n</html>\n");
	120
	121	fclose(c);
	122
	123	exit(EXIT_SUCCESS);
	124	}
	125
	126	static void launch_print_obj_dict_callback(launch_data_t obj, const char key, void context)
	127	{
	128	FILE *w = context;
	129
	130	fprintf(w, "<i>%s</i>\n", key);
	131	if (launch_data_get_type(obj) != LAUNCH_DATA_ARRAY &&
	132	launch_data_get_type(obj) != LAUNCH_DATA_DICTIONARY)
	133	fprintf(w, "<ul><li>\n");
	134	launch_print_obj(obj, w);
	135	if (launch_data_get_type(obj) != LAUNCH_DATA_ARRAY &&
	136	launch_data_get_type(obj) != LAUNCH_DATA_DICTIONARY)
	137	fprintf(w, "</li></ul>\n");
	138	}
	139
	140	static void launch_print_obj(launch_data_t o, FILE *w)
	141	{
	142	size_t i;
	143
	144
	145	switch (launch_data_get_type(o)) {
	146	case LAUNCH_DATA_DICTIONARY:
	147	fprintf(w, "<ul><li>\n");
	148	launch_data_dict_iterate(o, launch_print_obj_dict_callback, w);
	149	fprintf(w, "</li></ul>\n");
	150	break;
	151	case LAUNCH_DATA_ARRAY:
	152	fprintf(w, "<ol>\n");
	153	for (i = 0; i < launch_data_array_get_count(o); i++) {
	154	fprintf(w, "<li>");
	155	launch_print_obj(launch_data_array_get_index(o, i), w);
	156	fprintf(w, "</li>\n");
	157	}
	158	fprintf(w, "</ol>\n");
	159	break;
	160	case LAUNCH_DATA_INTEGER:
	161	fprintf(w, "Number: %lld", launch_data_get_integer(o));
	162	break;
	163	case LAUNCH_DATA_REAL:
	164	fprintf(w, "Float: %f", launch_data_get_real(o));
	165	break;
	166	case LAUNCH_DATA_STRING:
	167	fprintf(w, "String: %s", launch_data_get_string(o));
	168	break;
	169	case LAUNCH_DATA_OPAQUE:
	170	fprintf(w, "Opaque: %p size %zu", launch_data_get_opaque(o), launch_data_get_opaque_size(o));
	171	break;
	172	case LAUNCH_DATA_FD:
	173	fprintf(w, "FD: %d", launch_data_get_fd(o));
	174	break;
	175	case LAUNCH_DATA_BOOL:
	176	fprintf(w, "Bool: %s", launch_data_get_bool(o) ? "true" : "false");
	177	break;
	178	default:
	179	fprintf(w, "type %d is unknown", launch_data_get_type(o));
	180	break;
	181	}
	182	}