11 #include <sys/socket.h>
12 #include <sys/types.h>
16 #include <talloc/talloc.h>
17 #include <waiter/waiter.h>
18 #include <system/system.h>
22 #include "pb-discover.h"
23 #include "device-handler.h"
26 #define DBG(fmt, args...) pb_log("DBG: " fmt, ## args)
27 #define DBGS(fmt, args...) \
28 pb_log("DBG:%s:%d: " fmt, __func__, __LINE__, ## args)
30 #define DBG(fmt, args...)
31 #define DBGS(fmt, args...)
36 struct udev_monitor *monitor;
37 struct device_handler *handler;
40 static int udev_destructor(void *p)
42 struct pb_udev *udev = p;
44 udev_monitor_unref(udev->monitor);
47 udev_unref(udev->udev);
53 static void print_device_properties(struct udev_device *dev)
55 struct udev_list_entry *list, *entry;
60 list = udev_device_get_properties_list_entry(dev);
64 udev_list_entry_foreach(entry, list)
65 DBG("property: %s - %s\n",
66 udev_list_entry_get_name(entry),
67 udev_device_get_property_value(dev,
68 udev_list_entry_get_name(entry)));
72 static int udev_handle_dev_action(struct udev_device *dev, const char *action)
79 enum event_action eva = 0;
84 devtype = udev_device_get_devtype(dev); /* DEVTYPE */
87 pb_log("udev_device_get_devtype failed\n");
91 devpath = udev_device_get_devpath(dev); /* DEVPATH */
94 pb_log("udev_device_get_devpath failed\n");
98 devnode = udev_device_get_devnode(dev); /* DEVNAME */
101 pb_log("udev_device_get_devnode failed\n");
105 print_device_properties(dev);
107 /* Ignore non disk or partition, ram, loop. */
109 if (!(strstr(devtype, "disk") || strstr(devtype, "partition"))
110 || strstr(devpath, "virtual/block/loop")
111 || strstr(devpath, "virtual/block/ram")) {
112 pb_log("SKIP: %s - %s\n", devtype, devnode);
116 if (!strcmp(action, "add")) {
117 pb_log("ADD: %s - %s\n", devtype, devnode);
118 eva = EVENT_ACTION_ADD;
119 } else if (!strcmp(action, "remove")) {
120 pb_log("REMOVE: %s - %s\n", devtype, devnode);
121 eva = EVENT_ACTION_REMOVE;
123 pb_log("SKIP: %s: %s - %s\n", action, devtype, devnode);
127 event = talloc(NULL, struct event);
129 event->type = EVENT_TYPE_UDEV;
131 event->device = devpath;
134 event->params = talloc(event, struct param);
135 event->params->name = "DEVNAME";
136 event->params->value = devnode;
138 udev = udev_get_userdata(udev_device_get_udev(dev));
141 device_handler_event(udev->handler, event);
147 static int udev_enumerate(struct udev *udev)
150 struct udev_list_entry *list, *entry;
151 struct udev_enumerate *enumerate;
153 enumerate = udev_enumerate_new(udev);
156 pb_log("udev_enumerate_new failed\n");
160 result = udev_enumerate_add_match_subsystem(enumerate, "block");
163 pb_log("udev_enumerate_add_match_subsystem failed\n");
167 udev_enumerate_scan_devices(enumerate);
169 list = udev_enumerate_get_list_entry(enumerate);
172 pb_log("udev_enumerate_get_list_entry failed\n");
176 udev_list_entry_foreach(entry, list) {
178 struct udev_device *dev;
180 syspath = udev_list_entry_get_name(entry);
181 dev = udev_device_new_from_syspath(udev, syspath);
183 udev_handle_dev_action(dev, "add");
185 udev_device_unref(dev);
188 udev_enumerate_unref(enumerate);
192 udev_enumerate_unref(enumerate);
196 static int udev_setup_monitor(struct udev *udev, struct udev_monitor **monitor)
199 struct udev_monitor *m;
202 m = udev_monitor_new_from_netlink(udev, "udev");
205 pb_log("udev_monitor_new_from_netlink failed\n");
209 result = udev_monitor_filter_add_match_subsystem_devtype(m, "block",
213 pb_log("udev_monitor_filter_add_match_subsystem_devtype failed\n");
217 result = udev_monitor_enable_receiving(m);
220 pb_log("udev_monitor_enable_receiving failed\n");
228 udev_monitor_unref(m);
233 * udev_process - waiter callback for monitor netlink.
236 static int udev_process(void *arg)
238 struct udev_monitor *monitor = arg;
239 struct udev_device *dev;
243 dev = udev_monitor_receive_device(monitor);
246 pb_log("udev_monitor_receive_device failed\n");
250 action = udev_device_get_action(dev);
253 pb_log("udev_device_get_action failed\n");
257 result = udev_handle_dev_action(dev, action);
259 udev_device_unref(dev);
263 udev_device_unref(dev);
267 static void udev_log_fn(struct udev __attribute__((unused)) *udev,
268 int __attribute__((unused)) priority, const char *file, int line,
269 const char *fn, const char *format, va_list args)
271 pb_log("libudev: %s %s:%d: ", fn, file, line);
272 vfprintf(pb_log_get_stream(), format, args);
275 struct pb_udev *udev_init(struct waitset *waitset,
276 struct device_handler *handler)
279 struct pb_udev *udev = talloc(NULL, struct pb_udev);
281 talloc_set_destructor(udev, udev_destructor);
282 udev->handler = handler;
284 udev->udev = udev_new();
287 pb_log("udev_new failed\n");
291 udev_set_userdata(udev->udev, udev);
293 udev_set_log_fn(udev->udev, udev_log_fn);
295 result = udev_enumerate(udev->udev);
300 result = udev_setup_monitor(udev->udev, &udev->monitor);
305 waiter_register(waitset, udev_monitor_get_fd(udev->monitor), WAIT_IN,
306 udev_process, udev->monitor);
308 pb_log("%s: waiting on %s\n", __func__, udev_get_sys_path(udev->udev));
314 udev_unref(udev->udev);
320 void udev_destroy(struct pb_udev *udev)