summaryrefslogtreecommitdiffstats
path: root/udev-block-notify.c
diff options
context:
space:
mode:
Diffstat (limited to 'udev-block-notify.c')
-rw-r--r--udev-block-notify.c207
1 files changed, 207 insertions, 0 deletions
diff --git a/udev-block-notify.c b/udev-block-notify.c
new file mode 100644
index 0000000..5e0d6c3
--- /dev/null
+++ b/udev-block-notify.c
@@ -0,0 +1,207 @@
+/*
+ * (C) 2011-2012 by Christian Hesse <mail@eworm.de>
+ *
+ * This software may be used and distributed according to the terms
+ * of the GNU General Public License, incorporated herein by reference.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+
+#include <libnotify/notify.h>
+#include <libudev.h>
+#include <blkid.h>
+
+#define PROGNAME "udev-block-notify"
+
+#define NOTIFICATION_TIMEOUT 10000
+#ifndef DEBUG
+#define DEBUG 0
+#endif
+
+#define ICON_ADD "media-removable"
+#define ICON_REMOVE "media-removable"
+#define ICON_MOVE "media-removable"
+#define ICON_CHANGE "media-removable"
+#define ICON_DEFAULT "media-removable"
+
+#define TEXT_TOPIC "Udev Block Notification"
+#define TEXT_ADD "Device <b>%s</b> (%i:%i) <b>appeared</b>."
+#define TEXT_REMOVE "Device <b>%s</b> (%i:%i) <b>disappeared</b>."
+#define TEXT_MOVE "Device <b>%s</b> (%i:%i) was <b>renamed</b>."
+#define TEXT_CHANGE "Device <b>%s</b> (%i:%i) media <b>changed</b>."
+#define TEXT_DEFAULT "Anything happend to <b>%s</b> (%i:%i)... Don't know."
+#define TEXT_TAG "%s\n%s: <i>%s</i>"
+
+int main (int argc, char ** argv) {
+ blkid_cache cache = NULL;
+ blkid_dev blkdev = NULL;
+ blkid_tag_iterate iter = NULL;
+ char action;
+ char *device = NULL, *icon = NULL, *notifystr = NULL, *read = NULL;
+ const char *type, *value, *devname;
+ fd_set readfds;
+ GError *error = NULL;
+ int fdcount, devnum, errcount = 0;
+ NotifyNotification *notification;
+ NotifyNotification ***notificationref;
+ struct udev_device *dev = NULL;
+ struct udev_monitor *mon = NULL;
+ struct udev *udev = NULL;
+ short int *maxminor;
+ unsigned short int i, major, minor;
+
+ printf("%s: %s v%s (compiled: " __DATE__ ", " __TIME__ ")\n", argv[0], PROGNAME, VERSION);
+
+ if(!notify_init("Udev-Block-Notification")) {
+ fprintf(stderr, "%s: Can't create notify.\n", argv[0]);
+ exit(EXIT_FAILURE);
+ }
+
+ udev = udev_new();
+ if(!udev) {
+ fprintf(stderr, "%s: Can't create udev.\n", argv[0]);
+ exit(EXIT_FAILURE);
+ }
+
+ mon = udev_monitor_new_from_netlink(udev, "udev");
+ udev_monitor_filter_add_match_subsystem_devtype(mon, "block", NULL);
+ udev_monitor_enable_receiving(mon);
+
+ notificationref = malloc(256 * sizeof(size_t));
+ for(i = 0; i < 256; i++)
+ notificationref[i] = NULL;
+ maxminor = malloc(256 * sizeof(short int));
+ for(i = 0; i < 256; i++)
+ maxminor[i] = -1;
+
+ while (1) {
+ FD_ZERO(&readfds);
+ if (mon != NULL)
+ FD_SET(udev_monitor_get_fd(mon), &readfds);
+
+ fdcount = select(udev_monitor_get_fd(mon) + 1, &readfds, NULL, NULL, NULL);
+
+ if ((mon != NULL) && FD_ISSET(udev_monitor_get_fd(mon), &readfds)) {
+ dev = udev_monitor_receive_device(mon);
+ if(dev) {
+ device = (char *) udev_device_get_sysname(dev);
+ devnum = udev_device_get_devnum(dev);
+ major = devnum / 256;
+ minor = devnum - (major * 256);
+
+ action = udev_device_get_action(dev)[0];
+ switch(action) {
+ case 'a':
+ // a: add
+ notifystr = (char *) malloc(strlen(TEXT_ADD) + strlen(device));
+ sprintf(notifystr, TEXT_ADD, device, major, minor);
+ icon = ICON_ADD;
+ break;
+ case 'r':
+ // r: remove
+ notifystr = (char *) malloc(strlen(TEXT_REMOVE) + strlen(device));
+ sprintf(notifystr, TEXT_REMOVE, device, major, minor);
+ icon = ICON_REMOVE;
+ break;
+ case 'm':
+ // m: move
+ notifystr = (char *) malloc(strlen(TEXT_MOVE) + strlen(device));
+ sprintf(notifystr, TEXT_MOVE, device, major, minor);
+ icon = ICON_MOVE;
+ break;
+ case 'c':
+ // c: change
+ notifystr = (char *) malloc(strlen(TEXT_CHANGE) + strlen(device));
+ sprintf(notifystr, TEXT_CHANGE, device, major, minor);
+ icon = ICON_CHANGE;
+ break;
+ default:
+ // we should never get here I think...
+ notifystr = (char *) malloc(strlen(TEXT_DEFAULT) + strlen(device));
+ sprintf(notifystr, TEXT_CHANGE, device, major, minor);
+ icon = ICON_DEFAULT;
+ }
+
+ if (blkid_get_cache(&cache, read) != 0)
+ fprintf(stderr, "%s: Could not get blkid cache.\n", argv[0]);
+
+ if (blkid_probe_all_new(cache) != 0)
+ fprintf(stderr, "%s: Could not probe new devices.\n", argv[0]);
+
+ if (action != 'r') {
+ blkdev = blkid_get_dev(cache, udev_device_get_devnode(dev), BLKID_DEV_NORMAL);
+
+ if (blkdev) {
+ iter = blkid_tag_iterate_begin(blkdev);
+
+ while (blkid_tag_next(iter, &type, &value) == 0) {
+ notifystr = (char *) realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + strlen(type) + strlen(value));
+ sprintf(notifystr, TEXT_TAG, notifystr, type, value);
+ }
+
+ blkid_tag_iterate_end(iter);
+ blkid_put_cache(cache);
+ } else
+ fprintf(stderr, "%s: Could not get blkid device.\n", argv[0]);
+ }
+
+#if DEBUG
+ printf("%s: %s\n", argv[0], notifystr);
+#endif
+
+ if (maxminor[major] < minor) {
+ notificationref[major] = realloc(notificationref[major], (minor + 1) * sizeof(size_t));
+ while(maxminor[major] < minor)
+ notificationref[major][++maxminor[major]] = NULL;
+ }
+
+ if (notificationref[major][minor] == NULL) {
+ notification = notify_notification_new(TEXT_TOPIC, notifystr, icon);
+ notificationref[major][minor] = notification;
+ } else {
+ notification = notificationref[major][minor];
+ notify_notification_update(notification, TEXT_TOPIC, notifystr, icon);
+ }
+
+ notify_notification_set_timeout(notification, NOTIFICATION_TIMEOUT);
+ notify_notification_set_category(notification, PROGNAME);
+ notify_notification_set_urgency (notification, NOTIFY_URGENCY_NORMAL);
+
+ while(!notify_notification_show(notification, &error)) {
+ if (errcount > 1) {
+ fprintf(stderr, "%s: Looks like we can not reconnect to notification daemon... Exiting.\n", argv[0]);
+ exit(EXIT_FAILURE);
+ } else {
+ g_printerr("%s: Error \"%s\" while trying to show notification. Trying to reconnect.\n", argv[0], error->message);
+ errcount++;
+
+ g_error_free(error);
+ error = NULL;
+
+ notify_uninit();
+
+ usleep(500 * 1000);
+
+ if(!notify_init("Udev-Block-Notification")) {
+ fprintf(stderr, "%s: Can't create notify.\n", argv[0]);
+ exit(EXIT_FAILURE);
+ }
+ }
+ }
+ errcount = 0;
+
+ free(notifystr);
+ udev_device_unref(dev);
+ }
+
+ // This is not really needed... But we want to make shure not to eat 100% CPU if anything breaks. ;)
+ usleep(50 * 1000);
+ }
+ }
+
+ udev_unref(udev);
+ return EXIT_SUCCESS;
+}