xref: /illumos-gate/usr/src/cmd/hal/addons/network-devices/addon-network-discovery.c (revision 97191cec6312a1c3a1b70c5f4a543c415c59edf7)
1 /*
2  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
3  * Use is subject to license terms.
4  *
5  * Licensed under the Academic Free License version 2.1
6  */
7 
8 
9 #ifdef HAVE_CONFIG_H
10 #include <config.h>
11 #endif
12 
13 #include <stdio.h>
14 #include <unistd.h>
15 #include <string.h>
16 #include <stdlib.h>
17 #include <fcntl.h>
18 #include <sys/dkio.h>
19 #include <sys/stat.h>
20 #include <priv.h>
21 #include <glib.h>
22 
23 #include <dbus/dbus-glib-lowlevel.h>
24 #include <libhal.h>
25 
26 #include "../../hald/logger.h"
27 
28 #include "network-discovery.h"
29 #include "printer.h"
30 
31 #define	DBUS_INTERFACE	"org.freedesktop.Hal.Device.NetworkDiscovery"
32 #define	NP(x)	(x?x:"NULL")
33 #define	STRDUP(x)	(x?strdup(x):NULL)
34 
35 typedef struct {
36 	LibHalContext *ctx;
37 	gboolean enabled;
38 	char *parent;
39 	char *community;
40 	char *network;
41 } nds_snmp_cbdata_t;
42 
43 static nds_snmp_cbdata_t *snmp_cb_data = NULL;
44 
45 static int
nds_snmp_scan(LibHalContext * ctx,char * parent,char * community,char * network)46 nds_snmp_scan(LibHalContext *ctx, char *parent, char *community, char *network)
47 {
48 	time_t start;
49 
50 	HAL_DEBUG(("nds_snmp_scan(0x%8.8x, %s, %s, %s)",
51 			ctx, NP(parent), NP(community), NP(network)));
52 	HAL_DEBUG(("NetworkDiscovery snmp scan initated"));
53 
54 	/* scan for devices */
55 	time(&start);
56 	if (network == NULL) {
57 		GList *elem, *list = broadcast_addresses();
58 
59 		for (elem = list; elem != NULL; elem = g_list_next(elem)) {
60 			scan_for_devices_using_snmp(ctx, parent, community,
61 						(char *)elem->data);
62 			free(elem->data);
63 		}
64 		g_list_free(list);
65 	} else
66 		scan_for_devices_using_snmp(ctx, parent, community, network);
67 
68 	/* remove devices that haven't been seen since before this scan */
69 	scan_for_stale_devices(ctx, start);
70 
71 	HAL_DEBUG(("NetworkDiscovery snmp scan completed"));
72 
73 	return (0);
74 }
75 
76 static gboolean
nds_snmp_scan_cb(gpointer data)77 nds_snmp_scan_cb(gpointer data)
78 {
79 	nds_snmp_cbdata_t *args = data;
80 
81 	if (args->enabled == FALSE) {
82 		if (args->parent) free(args->parent);
83 		if (args->community) free(args->community);
84 		if (args->network) free(args->network);
85 		free(args);
86 		return (FALSE);
87 	}
88 
89 	nds_snmp_scan(args->ctx, args->parent, args->community, args->network);
90 
91 	return (TRUE);
92 }
93 
94 static int
nds_EnablePrinterScanningViaSNMP(LibHalContext * ctx,char * parent,int interval,char * community,char * network)95 nds_EnablePrinterScanningViaSNMP(LibHalContext *ctx, char *parent, int interval,
96 		char *community, char *network)
97 {
98 	HAL_DEBUG(("NetworkDiscovery.EnablePrinterScanningViaSNMP(0x%8.8x, %s, %d, %s, %s)",
99 			ctx, NP(parent), interval, NP(community), NP(network)));
100 
101 	/* are we already discoverying network devices ? */
102 	if (snmp_cb_data != NULL) {
103 		snmp_cb_data->enabled = FALSE; /* cancel it */
104 	}
105 
106 	/* setup for network device discovery */
107 	if ((snmp_cb_data = calloc(1, sizeof (*snmp_cb_data))) != NULL) {
108 		snmp_cb_data->ctx = ctx;
109 		snmp_cb_data->enabled = TRUE;
110 		snmp_cb_data->parent = STRDUP(parent);
111 		snmp_cb_data->community = STRDUP(community);
112 		snmp_cb_data->network = STRDUP(network);
113 
114 		/* prime the pump with an initial scan */
115 		nds_snmp_scan(ctx, parent, community, network);
116 
117 		/* add a regular network scan */
118 		g_timeout_add(interval * 1000, nds_snmp_scan_cb, snmp_cb_data);
119 	}
120 
121 	return (0);
122 }
123 
124 static int
nds_DisablePrinterScanningViaSNMP(LibHalContext * ctx)125 nds_DisablePrinterScanningViaSNMP(LibHalContext *ctx)
126 {
127 	HAL_DEBUG(("NetworkDiscovery.DisablePrinterScanningViaSNMP(0x%8.8x)", ctx));
128 
129 	if (snmp_cb_data != NULL)
130 		snmp_cb_data->enabled = FALSE;
131 	snmp_cb_data = NULL;
132 
133 	return (0);
134 }
135 
136 static int
nds_ScanForPrintersViaSNMP(LibHalContext * ctx,char * parent,char * community,char * network)137 nds_ScanForPrintersViaSNMP(LibHalContext *ctx, char *parent, char *community,
138 		char *network)
139 {
140 	time_t start, stop;
141 
142 	HAL_DEBUG(("NetworkDiscovery.ScanForPrintersViaSNMP(0x%8.8x, %s, %s, %s)",
143 			ctx, NP(parent), NP(community), NP(network)));
144 
145 	return (nds_snmp_scan(ctx, parent, community, network));
146 }
147 
148 static DBusHandlerResult
nds_filter_function(DBusConnection * connection,DBusMessage * message,void * user_data)149 nds_filter_function(DBusConnection *connection, DBusMessage *message,
150 		void *user_data)
151 {
152 	LibHalContext *ctx = user_data;
153 	DBusMessage *reply;
154 	DBusError error;
155 	const char *member = dbus_message_get_member(message);
156 	const char *path = dbus_message_get_path(message);
157 	int rc = -1;
158 
159 	dbus_error_init(&error);
160 
161 	HAL_DEBUG(("DBus message: %s, %s ", member, path));
162 
163 	if (dbus_message_is_method_call(message,
164 				DBUS_INTERFACE, "EnablePrinterScanningViaSNMP")) {
165 		int interval = -1;
166 		char *udi = getenv("UDI");
167 		char *community = "public";
168 		char *network = "0.0.0.0";
169 
170 		dbus_message_get_args(message, &error,
171 				DBUS_TYPE_INT32, &interval,
172 				DBUS_TYPE_STRING, &community,
173 				DBUS_TYPE_STRING, &network,
174 				DBUS_TYPE_INVALID);
175 
176 		if (strcmp(network, "0.0.0.0") == 0)
177 			network = NULL;
178 
179 		rc = nds_EnablePrinterScanningViaSNMP(ctx, udi, interval,
180 				community, network);
181 	} else if (dbus_message_is_method_call(message,
182 				DBUS_INTERFACE, "ScanForPrintersViaSNMP")) {
183 		int interval = -1;
184 		char *udi = getenv("UDI");
185 		char *community = "public";
186 		char *network = "0.0.0.0";
187 
188 		dbus_message_get_args(message, &error,
189 				DBUS_TYPE_STRING, &community,
190 				DBUS_TYPE_STRING, &network,
191 				DBUS_TYPE_INVALID);
192 
193 		if (strcmp(network, "0.0.0.0") == 0)
194 			network = NULL;
195 
196 		rc = nds_ScanForPrintersViaSNMP(ctx, udi, community, network);
197 	} else if (dbus_message_is_method_call(message,
198 				DBUS_INTERFACE, "DisablePrinterScanningViaSNMP")) {
199 		rc = nds_DisablePrinterScanningViaSNMP(ctx);
200 	} else {
201 		/* bypass not-handled messages */
202 		HAL_WARNING(("Unknown DBus message: %s, %s ", member, path));
203 		return (DBUS_HANDLER_RESULT_NOT_YET_HANDLED);
204 	}
205 
206 	if (dbus_error_is_set(&error))
207 		dbus_error_free(&error);
208 
209 	if ((reply = dbus_message_new_method_return(message)) == NULL) {
210 		HAL_WARNING(("Could not allocate memory for the DBus reply"));
211 		return (FALSE);
212 	}
213 
214 	dbus_message_append_args(reply, DBUS_TYPE_INT32, &rc,
215 			DBUS_TYPE_INVALID);
216 
217 	if (!dbus_connection_send(connection, reply, NULL)) {
218 		HAL_WARNING(("Could not sent reply"));
219 	}
220 	dbus_connection_flush(connection);
221 	dbus_message_unref(reply);
222 
223 	return (DBUS_HANDLER_RESULT_HANDLED);
224 }
225 
226 static int
nds_claim_interface(LibHalContext * ctx,char * udi,DBusError * error)227 nds_claim_interface(LibHalContext *ctx, char *udi, DBusError *error)
228 {
229 	DBusConnection *connection;
230 	char *interface_xml =
231 		"<method name=\"EnablePrinterScanningViaSNMP\">\n"
232 		"  <arg name=\"interval\" direction=\"in\" type=\"i\"/>\n"
233 		"  <arg name=\"community\" direction=\"in\" type=\"s\"/>\n"
234 		"  <arg name=\"network\" direction=\"in\" type=\"s\"/>\n"
235 		"  <arg name=\"return_code\" direction=\"out\" type=\"i\"/>\n"
236 		"</method>\n"
237 		"<method name=\"DisablePrinterScanningViaSNMP\">\n"
238 		"  <arg name=\"return_code\" direction=\"out\" type=\"i\"/>\n"
239 		"</method>\n"
240 		"<method name=\"ScanForPrintersViaSNMP\">\n"
241 		"  <arg name=\"community\" direction=\"in\" type=\"s\"/>\n"
242 		"  <arg name=\"network\" direction=\"in\" type=\"s\"/>\n"
243 		"  <arg name=\"return_code\" direction=\"out\" type=\"i\"/>\n"
244 		"</method>\n";
245 
246 	HAL_DEBUG(("nds_claim_interface(0x%8.8x, %s, 0x%8.8x): %s",
247 			ctx, udi, error, DBUS_INTERFACE));
248 
249 	if ((connection = libhal_ctx_get_dbus_connection(ctx)) == NULL) {
250 		HAL_WARNING(("Could not get DBus connection"));
251 		return (-1);
252 	}
253 
254 	if (libhal_device_claim_interface(ctx, udi,
255 			DBUS_INTERFACE, interface_xml, error) == 0) {
256 		HAL_WARNING(("Could not claim interface: %s", error->message));
257 		return (-1);
258 	}
259 
260 	dbus_connection_setup_with_g_main(connection, NULL);
261 	dbus_connection_add_filter(connection, nds_filter_function, ctx, NULL);
262 	dbus_connection_set_exit_on_disconnect(connection, 0);
263 
264 	return (0);
265 }
266 
267 static void
drop_privileges()268 drop_privileges()
269 {
270 	priv_set_t *pPrivSet = NULL;
271 	priv_set_t *lPrivSet = NULL;
272 
273 	/*
274 	 * Start with the 'basic' privilege set and then remove any
275 	 * of the 'basic' privileges that will not be needed.
276 	 */
277 	if ((pPrivSet = priv_str_to_set("basic", ",", NULL)) == NULL) {
278 		return;
279 	}
280 
281 	/* Clear privileges we will not need from the 'basic' set */
282 	(void) priv_delset(pPrivSet, PRIV_FILE_LINK_ANY);
283 	(void) priv_delset(pPrivSet, PRIV_PROC_EXEC);
284 	(void) priv_delset(pPrivSet, PRIV_PROC_FORK);
285 	(void) priv_delset(pPrivSet, PRIV_PROC_INFO);
286 	(void) priv_delset(pPrivSet, PRIV_PROC_SESSION);
287 
288 	/* Set the permitted privilege set. */
289 	if (setppriv(PRIV_SET, PRIV_PERMITTED, pPrivSet) != 0) {
290 		return;
291 	}
292 
293 	/* Clear the limit set. */
294 	if ((lPrivSet = priv_allocset()) == NULL) {
295 		return;
296 	}
297 
298 	priv_emptyset(lPrivSet);
299 
300 	if (setppriv(PRIV_SET, PRIV_LIMIT, lPrivSet) != 0) {
301 		return;
302 	}
303 
304 	priv_freeset(lPrivSet);
305 }
306 
307 
308 int
main(int argc,char ** argv)309 main(int argc, char **argv)
310 {
311 	LibHalContext *ctx = NULL;
312 	DBusError error;
313 	GMainLoop *loop = g_main_loop_new(NULL, FALSE);
314 	char *udi;
315 
316 	if ((udi = getenv("UDI")) == NULL) {
317 		return (0);
318 	}
319 
320 	drop_privileges();
321 
322 	setup_logger();
323 
324 	dbus_error_init(&error);
325 
326 	if ((ctx = libhal_ctx_init_direct(&error)) == NULL) {
327 		return (0);
328 	}
329 
330 	if (!libhal_device_addon_is_ready(ctx, udi, &error)) {
331 		return (0);
332 	}
333 
334 	if (nds_claim_interface(ctx, udi, &error) != 0) {
335 		return (0);
336 	}
337 
338 	g_main_loop_run(loop);
339 
340 	/* NOTREACHED */
341 }
342