sched: Only call pre/post select hooks if tasks are not in error state.
[paraslash.git] / attribute.c
index 801d907..e7fba90 100644 (file)
@@ -1,14 +1,17 @@
 /*
- * Copyright (C) 1997-2007 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 1997-2008 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
+
+/** \file attribute.c Attribute handling functions. */
 #include "para.h"
 #include "error.h"
+#include "string.h"
 #include "afh.h"
 #include "afs.h"
-#include "string.h"
 #include "net.h"
+#include "ipc.h"
 
 static struct osl_table *attribute_table;
 static int greatest_att_bitnum;
@@ -69,7 +72,7 @@ static void find_greatest_att_bitnum(void)
                        return;
                }
        } while (c--);
-       PARA_INFO_LOG("%s\n", "no attributes");
+       PARA_INFO_LOG("no attributes\n");
        greatest_att_bitnum = -E_NO_ATTRIBUTES;
 }
 
@@ -97,63 +100,82 @@ int get_attribute_bitnum_by_name(const char *att_name, unsigned char *bitnum)
        return 1;
 }
 
-/** Whether "-a" was given for the lsatt command. */
-#define LSATT_FLAG_ALPHA 1
-/** Whether "-l" was given for the lsatt command. */
-#define LSATT_FLAG_LONG 2
+/**
+ * Flags used by the lsatt command.
+ *
+ * \param \sa com_lsatt().
+ */
+enum lsatt_flags {
+       /** Whether "-a" was given for the lsatt command. */
+       LSATT_FLAG_SORT_BY_ID = 1,
+       /** Whether "-l" was given for the lsatt command. */
+       LSATT_FLAG_LONG = 2,
+       /** Reverse sort order. */
+       LSATT_FLAG_REVERSE = 4
+};
 
-/** Data passed via osl_rbtree_loop(). */
-struct private_lsatt_data {
+/** Data passed to the action function of lsatt */
+struct lsatt_action_data {
+       /** The result buffer. */
+       struct para_buffer pb;
        /** The given flags for the lsatt command. */
        unsigned flags;
-       /** The result buffer. */
-       struct para_buffer b;
 };
 
-static int print_attribute(struct osl_row *row, void *private_data)
+static int print_attribute(struct osl_table *table, struct osl_row *row,
+               const char *name, void *data)
 {
-       struct private_lsatt_data *pld = private_data;
-       int ret;
-       struct osl_object name_obj, bitnum_obj;
-
-       ret = osl_get_object(attribute_table, row, ATTCOL_NAME, &name_obj);
-       if (ret < 0)
+       struct lsatt_action_data *laad = data;
+       struct osl_object bitnum_obj;
+       int ret, ret2;
+
+       if (!(laad->flags & LSATT_FLAG_LONG))
+               return para_printf(&laad->pb, "%s\n", name);
+       ret = osl_get_object(table, row, ATTCOL_BITNUM, &bitnum_obj);
+       if (ret < 0) {
+               ret2 = para_printf(&laad->pb, "%s: %s\n", name, para_strerror(-ret));
                return ret;
-       if (!(pld->flags & LSATT_FLAG_LONG)) {
-               para_printf(&pld->b, "%s\n", (char *)name_obj.data);
-               return 1;
        }
-       ret = osl_get_object(attribute_table, row, ATTCOL_BITNUM, &bitnum_obj);
-       if (ret < 0)
-               return ret;
-       para_printf(&pld->b, "%u\t%s\n", *(unsigned char*)bitnum_obj.data,
-               (char *)name_obj.data);
-       return 1;
+       return para_printf(&laad->pb, "%u\t%s\n", *(unsigned char*)bitnum_obj.data,
+               name);
 }
 
-static int com_lsatt_callback(const struct osl_object *query,
-               struct osl_object *result)
+static void com_lsatt_callback(int fd, const struct osl_object *query)
 {
-       struct private_lsatt_data pld = {.flags = *(uint32_t *) query->data};
-       int ret;
+       struct lsatt_action_data laad = {
+               .flags = *(unsigned *) query->data,
+               .pb = {
+                       .max_size = SHMMAX,
+                       .private_data = &fd,
+                       .max_size_handler = pass_buffer_as_shm
+               }
 
-       if (pld.flags & LSATT_FLAG_ALPHA)
-               ret = osl_rbtree_loop(attribute_table, ATTCOL_NAME,
-                       &pld, print_attribute);
-       else
-               ret = osl_rbtree_loop(attribute_table, ATTCOL_BITNUM,
-                       &pld, print_attribute);
-       result->data = pld.b.buf;
-       result->size = pld.b.size;
-       return ret;
+       };
+       struct pattern_match_data pmd = {
+               .table = attribute_table,
+               .loop_col_num = ATTCOL_BITNUM,
+               .match_col_num = ATTCOL_NAME,
+               .patterns = {.data = (char *)query->data + sizeof(laad.flags),
+                       .size = query->size - sizeof(laad.flags)},
+               .pm_flags = PM_NO_PATTERN_MATCHES_EVERYTHING,
+               .data = &laad,
+               .action = print_attribute
+       };
+       if (laad.flags & LSATT_FLAG_SORT_BY_ID)
+               pmd.loop_col_num = ATTCOL_NAME;
+       if (laad.flags & LSATT_FLAG_REVERSE)
+               pmd.pm_flags |= PM_REVERSE_LOOP;
+       for_each_matching_row(&pmd);
+       if (laad.pb.offset)
+               pass_buffer_as_shm(laad.pb.buf, laad.pb.offset, &fd);
+       free(laad.pb.buf);
 }
 
-
 int com_lsatt(int fd, int argc, char * const * const argv)
 {
+       unsigned flags = 0;
+       struct osl_object options = {.data = &flags, .size = sizeof(flags)};
        int ret, i;
-       uint32_t flags = 0;
-       struct osl_object query, result;
 
        for (i = 1; i < argc; i++) {
                const char *arg = argv[i];
@@ -163,29 +185,30 @@ int com_lsatt(int fd, int argc, char * const * const argv)
                        i++;
                        break;
                }
-               if (!strcmp(arg, "-a")) {
-                       flags |= LSATT_FLAG_ALPHA;
+               if (!strcmp(arg, "-i")) {
+                       flags |= LSATT_FLAG_SORT_BY_ID;
                        continue;
                }
                if (!strcmp(arg, "-l")) {
                        flags |= LSATT_FLAG_LONG;
                        continue;
                }
+               if (!strcmp(arg, "-r")) {
+                       flags |= LSATT_FLAG_REVERSE;
+                       continue;
+               }
        }
-       if (argc > i)
-               return -E_ATTR_SYNTAX;
-       query.data = &flags;
-       query.size = sizeof(flags);
-       ret = send_callback_request(com_lsatt_callback, &query, &result);
-       if (ret > 0) {
-               ret = send_buffer(fd, (char *)result.data);
-               free(result.data);
-       }
+       ret = send_option_arg_callback_request(&options, argc - i, argv + i,
+               com_lsatt_callback, send_result, &fd);
+       if (!ret) {
+               if (argc > 1)
+                       ret = send_va_buffer(fd, "no matches\n");
+       } else if (ret < 0)
+               send_va_buffer(fd, "%s\n", para_strerror(-ret));
        return ret;
 }
 
-static int com_setatt_callback(const struct osl_object *query,
-               __a_unused struct osl_object *result)
+static void com_setatt_callback(__a_unused int fd, const struct osl_object *query)
 {
        char *p;
        uint64_t add_mask = 0, del_mask = 0;
@@ -198,8 +221,9 @@ static int com_setatt_callback(const struct osl_object *query,
                char c;
 
                len = strlen(p);
+               ret = -E_ATTR_SYNTAX;
                if (!*p)
-                       return -E_ATTR_SYNTAX;
+                       goto out;
                c = p[len - 1];
                if (c != '+' && c != '-')
                        break;
@@ -208,92 +232,95 @@ static int com_setatt_callback(const struct osl_object *query,
                obj.size = len + 1;
                ret = osl_get_row(attribute_table, ATTCOL_NAME, &obj, &row);
                if (ret < 0)
-                       return ret;
+                       goto out;
                ret = osl_get_object(attribute_table, row, ATTCOL_BITNUM,
                        &obj);
                if (ret < 0)
-                       return ret;
+                       goto out;
                if (c == '+')
                        add_mask |= (1UL << *(unsigned char *)obj.data);
                else
                        del_mask |= (1UL << *(unsigned char *)obj.data);
        }
+       ret = -E_ATTR_SYNTAX;
        if (!add_mask && !del_mask)
-               return -E_ATTR_SYNTAX;
+               goto out;
        PARA_DEBUG_LOG("masks: %llx:%llx\n",(long long unsigned)add_mask,
                (long long unsigned)del_mask);
        for (; p < (char *)query->data + query->size; p += len + 1) { /* TODO: fnmatch */
                struct afs_info old_afsi, new_afsi;
-               struct osl_row *aft_row;
+               struct afsi_change_event_data aced = {.old_afsi = &old_afsi};
 
                len = strlen(p);
-               ret = aft_get_row_of_path(p, &aft_row);
+               ret = aft_get_row_of_path(p, &aced.aft_row);
                if (ret < 0)
-                       return ret;
-               ret = get_afsi_object_of_row(aft_row, &obj);
+                       goto out;
+               ret = get_afsi_object_of_row(aced.aft_row, &obj);
                if (ret < 0)
-                       return ret;
+                       goto out;
                ret = load_afsi(&old_afsi, &obj);
                if (ret < 0)
-                       return ret;
+                       goto out;
                new_afsi = old_afsi;
                new_afsi.attributes |= add_mask;
                new_afsi.attributes &= ~del_mask;
                save_afsi(&new_afsi, &obj); /* in-place update */
-//             ret = mood_update_audio_file(aft_row, &old_afsi);
-//             if (ret < 0)
-//                     return ret;
+               afs_event(AFSI_CHANGE, NULL, &aced);
        }
-       return 1;
+out:
+       if (ret < 0)
+               PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
 }
 
 int com_setatt(__a_unused int fd, int argc, char * const * const argv)
 {
-       if (argc < 2)
+       if (argc < 3)
                return -E_ATTR_SYNTAX;
        return send_standard_callback_request(argc - 1, argv + 1, com_setatt_callback,
-               NULL);
+               NULL, NULL);
 }
 
-/* TODO: make it faster by only extracting the attribute member from afsi */
-static int logical_and_attribute(struct osl_row *aft_row, void *attribute_ptr)
-{
-       struct afs_info afsi;
-       uint64_t *att = attribute_ptr;
-       struct osl_object obj;
-       int ret = get_afsi_object_of_row(aft_row, &obj);
-       if (ret < 0)
-               return ret;
-       ret = load_afsi(&afsi, &obj);
-       if (ret < 0)
-               return ret;
-       afsi.attributes &= *att;
-       save_afsi(&afsi, &obj);
-       return 1;
-}
+struct addatt_event_data {
+       const char *name;
+       unsigned char bitnum;
+};
 
-static int com_addatt_callback(const struct osl_object *query,
-               __a_unused struct osl_object *result)
+
+static void com_addatt_callback(int fd, const struct osl_object *query)
 {
        char *p = query->data;
-       uint64_t atts_added = 0;
-       int ret;
+       int ret = 1, ret2 = 0;
+       struct para_buffer pb = {
+               .max_size = SHMMAX,
+               .private_data = &fd,
+               .max_size_handler = pass_buffer_as_shm
+       };
+       size_t len;
 
-       while (p < (char *)query->data + query->size) {
+       for (p = query->data; p < (char *)query->data + query->size; p += len + 1) {
                struct osl_object objs[NUM_ATT_COLUMNS];
                struct osl_row *row;
                unsigned char bitnum;
+               len = strlen(p);
+               struct addatt_event_data aed;
 
-               objs[ATTCOL_BITNUM].size = 1;
-               objs[ATTCOL_NAME].data = p;
-               objs[ATTCOL_NAME].size = strlen(p) + 1;
-               ret = osl_get_row(attribute_table, ATTCOL_NAME,
-                       &objs[ATTCOL_NAME], &row); /* expected to fail */
-               if (ret >= 0)
-                       return -E_ATTR_EXISTS;
+               if (!len || p[len - 1] == '-' || p[len - 1] == '+') {
+                       ret2 = para_printf(&pb, "invalid attribute name: %s\n", p);
+                       if (ret2 < 0)
+                               goto out;
+                       continue;
+               }
+               ret = get_attribute_bitnum_by_name(p, &bitnum);
+               if (ret >= 0) {
+                       ret2 = para_printf(&pb, "attribute \"%s\" already exists\n", p);
+                       if (ret2 < 0)
+                               goto out;
+                       continue;
+               }
                if (ret != -E_RB_KEY_NOT_FOUND) /* error */
-                       return ret;
-               /* find smallest non-used attribute */
+                       goto out;
+               objs[ATTCOL_BITNUM].size = 1;
+               /* find smallest unused attribute */
                for (bitnum = 0; bitnum < 64; bitnum++) {
                        objs[ATTCOL_BITNUM].data = &bitnum;
                        ret = osl_get_row(attribute_table, ATTCOL_BITNUM,
@@ -301,73 +328,161 @@ static int com_addatt_callback(const struct osl_object *query,
                        if (ret == -E_RB_KEY_NOT_FOUND)
                                break; /* this bitnum is unused, use it */
                        if (ret < 0) /* error */
-                               return ret;
+                               goto out;
                        /* this bit is already in use, try next bit */
                }
-               if (bitnum == 64)
-                       return -E_ATTR_TABLE_FULL;
+               if (bitnum == 64) {
+                       ret = -E_ATT_TABLE_FULL;
+                       goto out;
+               }
+               objs[ATTCOL_NAME].data = p;
+               objs[ATTCOL_NAME].size = len + 1;
                ret = osl_add_row(attribute_table, objs);
                if (ret < 0)
-                       return ret;
+                       goto out;
+               aed.name = p;
+               aed.bitnum = bitnum;
+               afs_event(ATTRIBUTE_ADD, &pb, &aed);
                greatest_att_bitnum = PARA_MAX(greatest_att_bitnum, bitnum);
-               atts_added |= 1 << bitnum;
-               p += strlen(p) + 1;
        }
-       if (!atts_added)
-               return 1;
-       atts_added = ~atts_added;
-       ret = audio_file_loop(&atts_added, logical_and_attribute);
-       if (ret < 0)
-               return ret;
-       find_greatest_att_bitnum();
-       return reload_current_mood(); /* FIXME: mood_reload() returns an error */
+out:
+       if (ret < 0 && ret2 >= 0)
+               ret = para_printf(&pb, "%s: %s\n", p, para_strerror(-ret));
+       if (pb.offset)
+               pass_buffer_as_shm(pb.buf, pb.offset, &fd);
+       free(pb.buf);
 }
 
-int com_addatt(__a_unused int fd, int argc, char * const * const argv)
+int com_addatt(int fd, int argc, char * const * const argv)
 {
+       int ret;
+
        if (argc < 2)
                return -E_ATTR_SYNTAX;
-       return send_standard_callback_request(argc - 1, argv + 1, com_addatt_callback,
-               NULL);
+       ret = send_standard_callback_request(argc - 1, argv + 1, com_addatt_callback,
+               send_result, &fd);
+       if (ret < 0)
+               send_va_buffer(fd, "%s\n", para_strerror(-ret));
+       return ret;
 }
 
-static int com_rmatt_callback(const struct osl_object *query,
-               __a_unused struct osl_object *result)
+static void com_mvatt_callback(int fd, const struct osl_object *query)
 {
-       char *p = query->data;
-       int ret, atts_removed = 0;
-       while (p < (char *)query->data + query->size) {
-               struct osl_object obj = {
-                       .data = p,
-                       .size = strlen(p) + 1
-               };
-               struct osl_row *row;
-               ret = osl_get_row(attribute_table, ATTCOL_NAME,
-                       &obj, &row);
-               if (ret < 0)
-                       return ret;
-               ret = osl_del_row(attribute_table, row);
-               if (ret < 0)
-                       return ret;
-               atts_removed++;
-               p += strlen(p) + 1;
-       }
-       find_greatest_att_bitnum();
-       if (!atts_removed)
-               return 1;
-       return reload_current_mood();
+       char *old = query->data;
+       size_t size = strlen(old) + 1;
+       char *new = old + size;
+       struct osl_object obj = {.data = old, .size = size};
+       struct osl_row *row;
+       struct para_buffer pb = {
+               .max_size = SHMMAX,
+               .private_data = &fd,
+               .max_size_handler = pass_buffer_as_shm
+       };
+       int ret;
+
+       ret = osl_get_row(attribute_table, ATTCOL_NAME, &obj, &row);
+       if (ret < 0)
+               goto out;
+       obj.data = new;
+       obj.size = strlen(new) + 1;
+       ret = osl_update_object(attribute_table, row, ATTCOL_NAME, &obj);
+out:
+       if (ret < 0)
+               ret = para_printf(&pb, "%s\n", para_strerror(-ret));
+       else
+               afs_event(ATTRIBUTE_RENAME, &pb, NULL);
+       if (pb.offset)
+               pass_buffer_as_shm(pb.buf, pb.offset, &fd);
+       free(pb.buf);
 }
 
-int com_rmatt(__a_unused int fd, int argc, char * const * const argv)
+int com_mvatt(int fd, int argc, char * const * const argv)
+{
+       int ret;
+
+       if (argc != 3)
+               return -E_ATTR_SYNTAX;
+       ret = send_standard_callback_request(argc - 1, argv + 1, com_mvatt_callback,
+               send_result, &fd);
+       if (ret < 0)
+               send_va_buffer(fd, "%s\n", para_strerror(-ret));
+       return ret;
+}
+
+/** Data passed to the action handler of com_rmatt(). */
+struct remove_attribute_action_data {
+       /** Message buffer. */
+       struct para_buffer pb;
+       /** Numver of attributes removed. */
+       int num_removed;
+       /** Bitwise "or" of the removed attributes. */
+       uint64_t mask_of_removed_atts;
+};
+
+static int remove_attribute(struct osl_table *table, struct osl_row *row,
+               const char *name, void *data)
 {
+       struct remove_attribute_action_data *raad = data;
+       int ret;
+       struct rmatt_event_data red = {.name = name};
+
+       ret = get_attribute_bitnum_by_name(name, &red.bitnum);
+       if (ret < 0)
+               return para_printf(&raad->pb, "%s: %s\n", name, para_strerror(-ret));
+       ret = osl_del_row(table, row);
+       if (ret < 0)
+               return para_printf(&raad->pb, "%s: %s\n", name, para_strerror(-ret));
+       ret = para_printf(&raad->pb, "removed attribute %s\n", name);
+       raad->num_removed++;
+       raad->mask_of_removed_atts |= (1 << red.bitnum);
+       afs_event(ATTRIBUTE_REMOVE, &raad->pb, &red);
+       return ret;
+}
+
+static void com_rmatt_callback(int fd, const struct osl_object *query)
+{
+       struct remove_attribute_action_data raad = {
+               .num_removed = 0,
+               .pb = {
+                       .max_size = SHMMAX,
+                       .private_data = &fd,
+                       .max_size_handler = pass_buffer_as_shm
+               }
+       };
+       int ret, ret2 = 0;
+       struct pattern_match_data pmd = {
+               .table = attribute_table,
+               .patterns = *query,
+               .loop_col_num = ATTCOL_BITNUM,
+               .match_col_num = ATTCOL_NAME,
+               .data = &raad,
+               .action = remove_attribute
+       };
+       ret = for_each_matching_row(&pmd);
+       if (ret < 0)
+               ret2 = para_printf(&raad.pb, "%s\n", para_strerror(-ret));
+       else if (!raad.num_removed)
+               ret2 = para_printf(&raad.pb, "no match -- nothing removed\n");
+       if (ret2 >= 0 && raad.pb.offset)
+               pass_buffer_as_shm(raad.pb.buf, raad.pb.offset, &fd);
+       free(raad.pb.buf);
+}
+
+int com_rmatt(int fd, int argc, char * const * const argv)
+{
+       int ret;
+
        if (argc < 2)
                return -E_ATTR_SYNTAX;
-       return send_standard_callback_request(argc - 1, argv + 1, com_rmatt_callback,
-               NULL);
+       ret = send_standard_callback_request(argc - 1, argv + 1, com_rmatt_callback,
+               send_result, &fd);
+       if (ret < 0)
+               send_va_buffer(fd, "%s\n", para_strerror(-ret));
+       return ret;
 }
 
 /**
- * Return a binary representation of the geiven attribute value.
+ * Return a binary representation of the given attribute value.
  *
  * \param atts Pointer to the attribute value.
  * \param buf Result.
@@ -407,8 +522,10 @@ int get_attribute_text(uint64_t *atts, const char *delim, char **text)
        const uint64_t one = 1;
 
        *text = NULL;
-       if (greatest_att_bitnum < 0) /* no attributes available */
+       if (greatest_att_bitnum < 0) { /* no attributes available */
+               *text = para_strdup("(no attributes available)");
                return 1;
+       }
        for (i = 0; i <= greatest_att_bitnum; i++) {
                unsigned char bn = i;
                struct osl_object obj = {.data = &bn, .size = 1};
@@ -440,13 +557,11 @@ err:
 /**
  * Close the attribute table.
  *
- * \param flags Ususal flags that are passed to osl_close_table().
- *
  * \sa osl_close_table().
  */
-void attribute_shutdown(enum osl_close_flags flags)
+void attribute_close(void)
 {
-       osl_close_table(attribute_table, flags);
+       osl_close_table(attribute_table, OSL_MARK_CLEAN);
        attribute_table = NULL;
 }
 
@@ -460,20 +575,34 @@ void attribute_shutdown(enum osl_close_flags flags)
  *
  * \sa osl_open_table().
  */
-int attribute_init(struct table_info *ti, const char *db)
+static int attribute_open(const char *dir)
 {
        int ret;
 
-       attribute_table_desc.dir = db;
-       ti->desc = &attribute_table_desc;
-       ret = osl_open_table(ti->desc, &attribute_table);
+       attribute_table_desc.dir = dir;
+       ret = osl_open_table(&attribute_table_desc, &attribute_table);
        greatest_att_bitnum = -1; /* no atts available */
        if (ret >= 0) {
                find_greatest_att_bitnum();
                return ret;
        }
        attribute_table = NULL;
-       if (ret == -E_NOENT)
+       if (ret >= 0 || is_errno(-ret, ENOENT))
                return 1;
        return ret;
 }
+
+static int attribute_create(const char *dir)
+{
+       attribute_table_desc.dir = dir;
+       return osl_create_table(&attribute_table_desc);
+}
+
+
+void attribute_init(struct afs_table *t)
+{
+       t->name = attribute_table_desc.name;
+       t->open = attribute_open;
+       t->close = attribute_close;
+       t->create = attribute_create;
+}