From f615773fe6108221543495fb193a66eee7b8d85d Mon Sep 17 00:00:00 2001 From: Christian Hesse Date: Wed, 17 Apr 2013 09:42:33 +0200 Subject: handle strings in functions --- udev-block-notify.c | 71 +++++++++++++++++++++++++---------------------------- 1 file changed, 33 insertions(+), 38 deletions(-) diff --git a/udev-block-notify.c b/udev-block-notify.c index 2952db3..636d647 100644 --- a/udev-block-notify.c +++ b/udev-block-notify.c @@ -34,6 +34,20 @@ #define TEXT_DEFAULT "Anything happend to %s (%i:%i)... Don't know." #define TEXT_TAG "\n%s: %s" +char * newstr(char *text, char *notifystr, char *device, unsigned short int major, unsigned short int minor) { + notifystr = malloc(strlen(text) + strlen(device) + 10 /* max string length 2* unsigned short int */); + sprintf(notifystr, text, device, major, minor); + + return notifystr; +} + +char * appendstr(char *text, char *notifystr, char *property, const char *value) { + notifystr = realloc(notifystr, strlen(text) + strlen(notifystr) + strlen(property) + strlen(value)); + sprintf(notifystr + strlen(notifystr), text, property, value); + + return notifystr; +} + int main (int argc, char ** argv) { char action; char *device = NULL, *icon = NULL, *notifystr = NULL; @@ -92,64 +106,45 @@ int main (int argc, char ** argv) { switch(action) { case 'a': // a: add - notifystr = malloc(strlen(TEXT_ADD) + strlen(device) + 5); - sprintf(notifystr, TEXT_ADD, device, major, minor); + notifystr = newstr(TEXT_ADD, notifystr, device, major, minor); icon = ICON_ADD; break; case 'r': // r: remove - notifystr = malloc(strlen(TEXT_REMOVE) + strlen(device) + 5); - sprintf(notifystr, TEXT_REMOVE, device, major, minor); + notifystr = newstr(TEXT_REMOVE, notifystr, device, major, minor); icon = ICON_REMOVE; break; case 'm': // m: move - notifystr = malloc(strlen(TEXT_MOVE) + strlen(device) + 5); - sprintf(notifystr, TEXT_MOVE, device, major, minor); + notifystr = newstr(TEXT_MOVE, notifystr, device, major, minor); icon = ICON_MOVE; break; case 'c': // c: change - notifystr = malloc(strlen(TEXT_CHANGE) + strlen(device) + 5); - sprintf(notifystr, TEXT_CHANGE, device, major, minor); + notifystr = newstr(TEXT_CHANGE, notifystr, device, major, minor); icon = ICON_CHANGE; break; default: // we should never get here I think... - notifystr = malloc(strlen(TEXT_DEFAULT) + strlen(device) + 5); - sprintf(notifystr, TEXT_CHANGE, device, major, minor); + notifystr = newstr(TEXT_DEFAULT, notifystr, device, major, minor); icon = ICON_DEFAULT; } if (action != 'r') { - if ((value = udev_device_get_property_value(dev, "ID_FS_LABEL")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 5 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Label", value); - } - if ((value = udev_device_get_property_value(dev, "ID_FS_TYPE")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 4 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Type", value); - } - if ((value = udev_device_get_property_value(dev, "ID_FS_USAGE")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 5 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Usage", value); - } - if ((value = udev_device_get_property_value(dev, "ID_FS_UUID")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 4 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "UUID", value); - } - if ((value = udev_device_get_property_value(dev, "ID_PART_TABLE_TYPE")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 14 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Partition Type", value); - } - if ((value = udev_device_get_property_value(dev, "ID_PART_TABLE_NAME")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 14 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Partition Name", value); - } - if ((value = udev_device_get_property_value(dev, "ID_PART_ENTRY_TYPE")) != NULL) { - notifystr = realloc(notifystr, strlen(TEXT_TAG) + strlen(notifystr) + 14 + strlen(value)); - sprintf(notifystr + strlen(notifystr), TEXT_TAG, "Partition UUID", value); - } + if ((value = udev_device_get_property_value(dev, "ID_FS_LABEL")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Label", value); + if ((value = udev_device_get_property_value(dev, "ID_FS_TYPE")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Type", value); + if ((value = udev_device_get_property_value(dev, "ID_FS_USAGE")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Usage", value); + if ((value = udev_device_get_property_value(dev, "ID_FS_UUID")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "UUID", value); + if ((value = udev_device_get_property_value(dev, "ID_PART_TABLE_TYPE")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Partition Type", value); + if ((value = udev_device_get_property_value(dev, "ID_PART_TABLE_NAME")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Partition Name", value); + if ((value = udev_device_get_property_value(dev, "ID_PART_ENTRY_TYPE")) != NULL) + notifystr = appendstr(TEXT_TAG, notifystr, "Partition UUID", value); } #if DEBUG -- cgit v1.2.3-54-g00ecf