From ee6296abb65f348e17464145b12d388dd970deee Mon Sep 17 00:00:00 2001 From: Christian Hesse Date: Mon, 15 Apr 2013 22:36:55 +0200 Subject: initial commit --- netlink-notify.c | 171 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 171 insertions(+) create mode 100644 netlink-notify.c (limited to 'netlink-notify.c') diff --git a/netlink-notify.c b/netlink-notify.c new file mode 100644 index 0000000..9679a45 --- /dev/null +++ b/netlink-notify.c @@ -0,0 +1,171 @@ +/* + * (C) 2011-2013 by Christian Hesse + * + * This software may be used and distributed according to the terms + * of the GNU General Public License, incorporated herein by reference. + */ + +#include +#include +#include +#include + +#include + +#include +#include +#include +#include + +#define PROGNAME "netlink-notify" + +#define NOTIFICATION_TIMEOUT 10000 +#ifndef DEBUG +#define DEBUG 0 +#endif + +#define ICON_NETWORK_ONLINE "network-transmit-receive" +#define ICON_NETWORK_OFFLINE "network-error" + +#define TEXT_TOPIC "Netlink Notification" +#define TEXT_NOTIFICATION "Interface %s is %s." +#define TEXT_NOTIFICATION_DEBUG "%s: Interface %s (index %d) is %s.\n" + +// we need these to be global... +unsigned int netlinksize = 1; // never use 0 and avoid overwriting the main pointer... +size_t * notificationref; +char * program = NULL; + +static int data_attr_cb(const struct nlattr * attr, void * data) { + const struct nlattr ** tb = data; + int type = mnl_attr_get_type(attr); + + if (mnl_attr_type_valid(attr, IFLA_MAX) < 0) + return MNL_CB_OK; + + switch(type) { + case IFLA_MTU: + if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0) { + fprintf(stderr, "%s: Invalid netlink attribute.\n", program); + return MNL_CB_ERROR; + } + break; + case IFLA_IFNAME: + if (mnl_attr_validate(attr, MNL_TYPE_STRING) < 0) { + fprintf(stderr, "%s: Invalid netlink attribute.\n", program); + return MNL_CB_ERROR; + } + break; + } + tb[type] = attr; + return MNL_CB_OK; +} + +static int data_cb(const struct nlmsghdr * nlh, void * data) { + struct nlattr * tb[IFLA_MAX + 1] = {}; + struct ifinfomsg * ifm = mnl_nlmsg_get_payload(nlh); + + char * notifystr = NULL; + const char * interface = NULL; + NotifyNotification * notification = NULL; + unsigned int errcount = 0; + + gboolean res = FALSE; + GError * error = NULL; + + mnl_attr_parse(nlh, sizeof(* ifm), data_attr_cb, tb); + + interface = mnl_attr_get_str(tb[IFLA_IFNAME]); + notifystr = malloc(strlen(interface) + strlen(TEXT_NOTIFICATION) + 1); // 2* %s is enough for "down", but we need an additional byte for \n + sprintf(notifystr, TEXT_NOTIFICATION, interface, (ifm->ifi_flags & IFF_RUNNING ? "up" : "down")); +#if DEBUG + printf(TEXT_NOTIFICATION_DEBUG, program, interface, ifm->ifi_index, (ifm->ifi_flags & IFF_RUNNING ? "up" : "down")); +#endif + + if (netlinksize < ifm->ifi_index) { + notificationref = realloc(notificationref, (ifm->ifi_index + 1) * sizeof(size_t)); + while(netlinksize < ifm->ifi_index) + notificationref[++netlinksize] = 0; + } + + if (notificationref[ifm->ifi_index] == 0) { + notification = notify_notification_new(TEXT_TOPIC, notifystr, (ifm->ifi_flags & IFF_RUNNING ? ICON_NETWORK_ONLINE : ICON_NETWORK_OFFLINE)); + notificationref[ifm->ifi_index] = (size_t)notification; + } else { + notification = (NotifyNotification *)notificationref[ifm->ifi_index]; + notify_notification_update(notification, TEXT_TOPIC, notifystr, (ifm->ifi_flags & IFF_RUNNING ? ICON_NETWORK_ONLINE : ICON_NETWORK_OFFLINE)); + } + + 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", program); + exit(EXIT_FAILURE); + } else { + g_printerr("%s: Error \"%s\" while trying to show notification. Trying to reconnect.\n", program, error->message); + errcount++; + + g_error_free(error); + error = NULL; + + notify_uninit(); + + usleep(500 * 1000); + + if(!notify_init(PROGNAME)) { + fprintf(stderr, "%s: Can't create notify.\n", program); + exit(EXIT_FAILURE); + } + } + } + errcount = 0; + + free(notifystr); + + return MNL_CB_OK; +} + +int main(int argc, char ** argv) { + struct mnl_socket * nl; + char buf[MNL_SOCKET_BUFFER_SIZE]; + int ret; + + program = argv[0]; + + printf("%s: %s v%s (compiled: " __DATE__ ", " __TIME__ ")\n", argv[0], PROGNAME, VERSION); + + if(!notify_init(PROGNAME)) { + fprintf(stderr, "%s: Can't create notify.\n", argv[0]); + exit(EXIT_FAILURE); + } + + nl = mnl_socket_open(NETLINK_ROUTE); + if (!nl) { + fprintf(stderr, "%s: Can't create netlink socket.\n", argv[0]); + exit(EXIT_FAILURE); + } + + if (mnl_socket_bind(nl, RTMGRP_LINK, MNL_SOCKET_AUTOPID) < 0) { + fprintf(stderr, "%s: Can't bind netlink socket.\n", argv[0]); + exit(EXIT_FAILURE); + } + + ret = mnl_socket_recvfrom(nl, buf, sizeof(buf)); + while (ret > 0) { + ret = mnl_cb_run(buf, ret, 0, 0, data_cb, NULL); + if (ret <= 0) + break; + ret = mnl_socket_recvfrom(nl, buf, sizeof(buf)); + } + if (ret == -1) { + fprintf(stderr, "%s: An error occured reading from netlink socket.\n", argv[0]); + exit(EXIT_FAILURE); + } + + mnl_socket_close(nl); + + return EXIT_SUCCESS; +} -- cgit v1.2.3-54-g00ecf