[apple/launchd.git] / 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(launch_data_t o, FILE *w)
{
	size_t i;
	void launch_print_obj_dict_callback(launch_data_t obj, const char *key, void *context __attribute__((unused))) {
		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");
	}


        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, NULL);
		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
e91b9f68 A	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(launch_data_t o, FILE *w)
127	{
128	size_t i;
129	void launch_print_obj_dict_callback(launch_data_t obj, const char key, void context __attribute__((unused))) {
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
141	switch (launch_data_get_type(o)) {
142	case LAUNCH_DATA_DICTIONARY:
143	fprintf(w, "<ul><li>\n");
144	launch_data_dict_iterate(o, launch_print_obj_dict_callback, NULL);
145	fprintf(w, "</li></ul>\n");
146	break;
147	case LAUNCH_DATA_ARRAY:
148	fprintf(w, "<ol>\n");
149	for (i = 0; i < launch_data_array_get_count(o); i++) {
150	fprintf(w, "<li>");
151	launch_print_obj(launch_data_array_get_index(o, i), w);
152	fprintf(w, "</li>\n");
153	}
154	fprintf(w, "</ol>\n");
155	break;
156	case LAUNCH_DATA_INTEGER:
157	fprintf(w, "Number: %lld", launch_data_get_integer(o));
158	break;
159	case LAUNCH_DATA_REAL:
160	fprintf(w, "Float: %f", launch_data_get_real(o));
161	break;
162	case LAUNCH_DATA_STRING:
163	fprintf(w, "String: %s", launch_data_get_string(o));
164	break;
165	case LAUNCH_DATA_OPAQUE:
166	fprintf(w, "Opaque: %p size %zu", launch_data_get_opaque(o), launch_data_get_opaque_size(o));
167	break;
168	case LAUNCH_DATA_FD:
169	fprintf(w, "FD: %d", launch_data_get_fd(o));
170	break;
171	case LAUNCH_DATA_BOOL:
172	fprintf(w, "Bool: %s", launch_data_get_bool(o) ? "true" : "false");
173	break;
174	default:
175	fprintf(w, "type %d is unknown", launch_data_get_type(o));
176	break;
177	}
178	}