blob.c

   1 #include "para.h"
   2 #include "error.h"
   3 #include "afh.h"
   4 #include "afs.h"
   5 #include "string.h"
   6 #include "net.h"
   7
   8 /** \file blob.c Macros and functions for blob handling. */
   9
  10 static struct osl_column_description blob_cols[] = {
  11         [BLOBCOL_ID] = {
  12                 .storage_type = OSL_MAPPED_STORAGE,
  13                 .storage_flags = OSL_RBTREE | OSL_UNIQUE | OSL_FIXED_SIZE,
  14                 .name = "id",
  15                 .data_size = 4,
  16                 .compare_function = uint32_compare
  17         },
  18         [BLOBCOL_NAME] = {
  19                 .storage_type = OSL_MAPPED_STORAGE,
  20                 .storage_flags = OSL_RBTREE | OSL_UNIQUE,
  21                 .name = "name",
  22                 .compare_function = string_compare
  23         },
  24         [BLOBCOL_DEF] = {
  25                 .storage_type = OSL_DISK_STORAGE,
  26                 .storage_flags = 0,
  27                 .name = "definition"
  28         }
  29 };
  30
  31 /** \cond doxygen isn't smart enough to recognize these */
  32 INIT_BLOB_TABLE(lyrics);
  33 INIT_BLOB_TABLE(images);
  34 INIT_BLOB_TABLE(moods);
  35 INIT_BLOB_TABLE(playlists);
  36 /** \endcond */
  37
  38 /** Flags that may be passed to the \p ls functions of each blob  type. */
  39 enum blob_ls_flags {
  40         /** List both id and name. */
  41         BLOB_LS_FLAG_LONG = 1,
  42         /** Reverse sort order. */
  43         BLOB_LS_FLAG_REVERSE = 2,
  44         /** Sort by id instead of name. */
  45         BLOB_LS_FLAG_SORT_BY_ID = 4,
  46 };
  47
  48 /** Data passed to \p com_lsbob_callback(). */
  49 struct com_lsblob_options {
  50         /** Given flags for the ls command. */
  51         uint32_t flags;
  52 };
  53
  54 /** Structure passed to the \p print_blob loop function. */
  55 struct lsblob_loop_data {
  56         struct com_lsblob_options *opts;
  57         struct para_buffer *pb;
  58         struct osl_table *table;
  59 };
  60
  61 static int print_blob(struct osl_row *row, void *loop_data)
  62 {
  63         struct osl_object obj;
  64         char *name;
  65         uint32_t id;
  66         int ret;
  67         struct lsblob_loop_data *lld = loop_data;
  68
  69         ret = osl_get_object(lld->table, row, BLOBCOL_NAME, &obj);
  70         if (ret < 0)
  71                 return ret;
  72         name = obj.data;
  73         if (!*name) /* ignore dummy row */
  74                 return 1;
  75         ret = osl_get_object(lld->table, row, BLOBCOL_ID, &obj);
  76         if (ret < 0)
  77                 return ret;
  78         id = *(uint32_t *)obj.data;
  79         if (lld->opts->flags & BLOB_LS_FLAG_LONG)
  80                 para_printf(lld->pb, "%u\t%s\n", id, name);
  81         else
  82                 para_printf(lld->pb, "%s\n", name);
  83         return 1;
  84 }
  85
  86 int com_lsblob_callback(struct osl_table *table,
  87                 const struct osl_object *query, struct osl_object *ls_output)
  88 {
  89         struct para_buffer pb = {.buf = NULL};
  90         struct lsblob_loop_data lld = {.opts = query->data, .pb = &pb, .table = table};
  91         int ret;
  92
  93         if (lld.opts->flags & BLOB_LS_FLAG_REVERSE) {
  94                 if (lld.opts->flags & BLOB_LS_FLAG_SORT_BY_ID)
  95                         ret = osl_rbtree_loop(lld.table, BLOBCOL_ID, &lld, print_blob);
  96                 else
  97                         ret = osl_rbtree_loop_reverse(lld.table, BLOBCOL_NAME, &lld, print_blob);
  98         } else {
  99                 if (lld.opts->flags & BLOB_LS_FLAG_SORT_BY_ID)
 100                         ret = osl_rbtree_loop_reverse(lld.table, BLOBCOL_ID, &lld, print_blob);
 101                 else
 102                         ret = osl_rbtree_loop(lld.table, BLOBCOL_NAME, &lld, print_blob);
 103         }
 104         ls_output->data = pb.buf;
 105         ls_output->size = pb.size;
 106         return ret;
 107 }
 108
 109 static int com_lsblob(callback_function *f, int fd, int argc, char * const * const argv)
 110 {
 111         struct com_lsblob_options clbo = {.flags = 0};
 112         struct osl_object query = {.data = &clbo, .size = sizeof(clbo)},
 113                 ls_output;
 114         int i, ret;
 115
 116         for (i = 1; i < argc; i++) {
 117                 const char *arg = argv[i];
 118                 if (arg[0] != '-')
 119                         break;
 120                 if (!strcmp(arg, "--")) {
 121                         i++;
 122                         break;
 123                 }
 124                 if (!strcmp(arg, "-l")) {
 125                         clbo.flags |= BLOB_LS_FLAG_LONG;
 126                         continue;
 127                 }
 128                 if (!strcmp(arg, "-i")) {
 129                         clbo.flags |= BLOB_LS_FLAG_SORT_BY_ID;
 130                         continue;
 131                 }
 132                 if (!strcmp(arg, "-r")) {
 133                         clbo.flags |= BLOB_LS_FLAG_REVERSE;
 134                         continue;
 135                 }
 136         }
 137         if (argc > i)
 138                 return -E_BLOB_SYNTAX;
 139         ret = send_option_arg_callback_request(&query, argc - i,
 140                 argv + i, f, &ls_output);
 141         if (ret >= 0 && ls_output.data)
 142                 send_buffer(fd, (char *)ls_output.data);
 143         free(ls_output.data);
 144         return ret;
 145 }
 146
 147 static int com_catblob_callback(struct osl_table *table,
 148                 const struct osl_object *query, struct osl_object *output)
 149 {
 150         struct osl_object obj;
 151         int ret;
 152         struct osl_row *row;
 153
 154         ret = osl_get_row(table, BLOBCOL_NAME, query, &row);
 155         if (ret < 0)
 156                 return ret;
 157         ret = osl_open_disk_object(table, row, BLOBCOL_DEF, &obj);
 158         if (ret < 0)
 159                 return ret;
 160         output->data = para_malloc(obj.size);
 161         output->size = obj.size;
 162         memcpy(output->data, obj.data, obj.size);
 163         return osl_close_disk_object(&obj);
 164 }
 165 static int com_catblob(callback_function *f, int fd, int argc,
 166                 char * const * const argv)
 167 {
 168         struct osl_object cat_output = {.data = NULL};
 169         int ret;
 170
 171         if (argc != 2)
 172                 return -E_BLOB_SYNTAX;
 173         if (!*argv[1]) /* empty name is reserved of the dummy row */
 174                 return -E_BLOB_SYNTAX;
 175         ret = send_standard_callback_request(1, argv + 1, f, &cat_output);
 176         if (ret >= 0 && cat_output.data)
 177                 ret = send_buffer(fd, (char *)cat_output.data);
 178         free(cat_output.data);
 179         return ret;
 180
 181 }
 182
 183 static int com_addblob_callback(struct osl_table *table,
 184                 const struct osl_object *query,
 185                 __a_unused struct osl_object *result)
 186 {
 187         struct osl_object objs[NUM_BLOB_COLUMNS];
 188         char *name = query->data;
 189         size_t name_len = strlen(name) + 1;
 190         uint32_t id;
 191         unsigned num_rows;
 192         int ret;
 193
 194         ret = osl_get_num_rows(table, &num_rows);
 195         if (ret < 0)
 196                 return ret;
 197         if (!num_rows) { /* this is the first entry ever added */
 198                 /* insert dummy row containing the id */
 199                 id = 2; /* this entry will be entry #1, so 2 is the next */
 200                 objs[BLOBCOL_ID].data = &id;
 201                 objs[BLOBCOL_ID].size = sizeof(id);
 202                 objs[BLOBCOL_NAME].data = "";
 203                 objs[BLOBCOL_NAME].size = 1;
 204                 objs[BLOBCOL_DEF].data = "";
 205                 objs[BLOBCOL_DEF].size = 1;
 206                 ret = osl_add_row(table, objs);
 207                 if (ret < 0)
 208                         return ret;
 209         } else { /* get id of the dummy row and increment it */
 210                 struct osl_row *row;
 211                 struct osl_object obj = {.data = "", .size = 1};
 212                 ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
 213                 if (ret < 0)
 214                         return ret;
 215                 ret = osl_get_object(table, row, BLOBCOL_ID, &obj);
 216                 if (ret < 0)
 217                         return ret;
 218                 id = *(uint32_t *)obj.data + 1;
 219                 obj.data = &id;
 220                 ret = osl_update_object(table, row, BLOBCOL_ID, &obj);
 221                 if (ret < 0)
 222                         return ret;
 223         }
 224         id--;
 225         objs[BLOBCOL_ID].data = &id;
 226         objs[BLOBCOL_ID].size = sizeof(id);
 227         objs[BLOBCOL_NAME].data = name;
 228         objs[BLOBCOL_NAME].size = name_len;
 229         objs[BLOBCOL_DEF].data = name + name_len;
 230         objs[BLOBCOL_DEF].size = query->size - name_len;
 231         return osl_add_row(table, objs);
 232 }
 233
 234 static int com_addblob(callback_function *f, __a_unused int fd, int argc,
 235                 char * const * const argv)
 236 {
 237         struct osl_object arg_obj;
 238
 239         if (argc != 2)
 240                 return -E_BLOB_SYNTAX;
 241         if (!*argv[1]) /* empty name is reserved for the dummy row */
 242                 return -E_BLOB_SYNTAX;
 243         PARA_NOTICE_LOG("argv[1]: %s\n", argv[1]);
 244         arg_obj.size = strlen(argv[1]) + 1;
 245         arg_obj.data = (char *)argv[1];
 246         return stdin_command(&arg_obj, f, 10 * 1024 * 1024, NULL);
 247 }
 248
 249 static int com_rmblob_callback(struct osl_table *table,
 250                 const struct osl_object *query,
 251                 __a_unused struct osl_object *result)
 252 {
 253         char *p = query->data;
 254         size_t len;
 255         int ret;
 256
 257         for (; p < (char *)query->data + query->size; p += len + 1) {
 258                 struct osl_row *row;
 259                 struct osl_object obj;
 260
 261                 len = strlen(p);
 262                 obj.data = p;
 263                 obj.size = len + 1;
 264                 ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
 265                 if (ret < 0)
 266                         return ret;
 267                 ret = osl_del_row(table, row);
 268                 if (ret < 0)
 269                         return ret;
 270         }
 271         return 1;
 272 }
 273
 274 static int com_rmblob(callback_function *f, __a_unused int fd, int argc,
 275                 char * const * const argv)
 276 {
 277         if (argc < 2)
 278                 return -E_MOOD_SYNTAX;
 279         return send_option_arg_callback_request(NULL, argc - 1, argv + 1, f,
 280                 NULL);
 281 }
 282
 283 static int com_mvblob_callback(struct osl_table *table,
 284                 const struct osl_object *query,
 285                 __a_unused struct osl_object *result)
 286 {
 287         char *src = (char *) query->data;
 288         struct osl_object obj = {.data = src, .size = strlen(src) + 1};
 289         char *dest = src + obj.size;
 290         struct osl_row *row;
 291         int ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
 292
 293         if (ret < 0)
 294                 return ret;
 295         obj.data = dest;
 296         obj.size = strlen(dest) + 1;
 297         return osl_update_object(table, row, BLOBCOL_NAME, &obj);
 298 }
 299
 300 static int com_mvblob(callback_function *f,  __a_unused int fd,
 301                 int argc, char * const * const argv)
 302 {
 303         if (argc != 3)
 304                 return -E_MOOD_SYNTAX;
 305         return send_option_arg_callback_request(NULL, argc - 1, argv + 1, f,
 306                 NULL);
 307 }
 308
 309 #define DEFINE_BLOB_COMMAND(cmd_name, table_name, cmd_prefix) \
 310         static int com_ ## cmd_name ## cmd_prefix ## _callback(const struct osl_object *query, \
 311                         struct osl_object *output) \
 312         { \
 313                 return com_ ## cmd_name ## blob_callback(table_name ## _table, query, output); \
 314         } \
 315         int com_ ## cmd_name ## cmd_prefix(int fd, int argc, char * const * const argv) \
 316         { \
 317                 return com_ ## cmd_name ## blob(com_ ## cmd_name ## cmd_prefix ## _callback, fd, argc, argv); \
 318         }
 319
 320 static int blob_get_name_by_id(struct osl_table *table, uint32_t id,
 321                 char **name)
 322 {
 323         struct osl_row *row;
 324         struct osl_object obj = {.data = &id, .size = sizeof(id)};
 325         int ret;
 326
 327         *name = NULL;
 328         if (!id)
 329                 return 1;
 330         ret = osl_get_row(table, BLOBCOL_ID, &obj, &row);
 331         if (ret < 0)
 332                 return ret;
 333         ret = osl_get_object(table, row, BLOBCOL_NAME, &obj);
 334         if (ret < 0)
 335                 return ret;
 336         *name = (char *)obj.data;
 337         return 1;
 338 }
 339 /** Define the \p get_name_by_id function for this blob type. */
 340 #define DEFINE_GET_NAME_BY_ID(table_name, cmd_prefix) \
 341         int cmd_prefix ## _get_name_by_id(uint32_t id, char **name) \
 342         { \
 343                 return blob_get_name_by_id(table_name ## _table, id, name); \
 344         }
 345
 346 /** Define the \p shutdown function for this blob type. */
 347 #define DEFINE_BLOB_SHUTDOWN(table_name) \
 348         void table_name ## _shutdown(enum osl_close_flags flags) \
 349         { \
 350                 osl_close_table(table_name ## _table, flags); \
 351                 table_name ## _table = NULL; \
 352         }
 353
 354 static int blob_init(struct osl_table **table,
 355                 struct osl_table_description *desc,
 356                 struct table_info *ti, const char *db)
 357 {
 358         int ret;
 359         desc->dir = db;
 360         ti->desc = desc;
 361         ret = osl_open_table(ti->desc, &ti->table);
 362         if (ret >= 0) {
 363                 *table = ti->table;
 364                 return ret;
 365         }
 366         *table = NULL;
 367         return ret == -E_NOENT? 1 : ret;
 368 }
 369
 370 /** Define the \p init function for this blob type. */
 371 #define DEFINE_BLOB_INIT(table_name) \
 372         int table_name ## _init(struct table_info *ti, const char *db) \
 373         { \
 374                 return blob_init(&table_name ## _table, \
 375                         &table_name ## _table_desc, ti, db); \
 376         }
 377
 378
 379 /** Define all functions for this blob type. */
 380 #define DEFINE_BLOB_FUNCTIONS(table_name, cmd_prefix) \
 381         DEFINE_BLOB_COMMAND(ls, table_name, cmd_prefix) \
 382         DEFINE_BLOB_COMMAND(cat, table_name, cmd_prefix) \
 383         DEFINE_BLOB_COMMAND(add, table_name, cmd_prefix) \
 384         DEFINE_BLOB_COMMAND(rm, table_name, cmd_prefix) \
 385         DEFINE_BLOB_COMMAND(mv, table_name, cmd_prefix) \
 386         DEFINE_GET_NAME_BY_ID(table_name, cmd_prefix); \
 387         DEFINE_BLOB_SHUTDOWN(table_name); \
 388         DEFINE_BLOB_INIT(table_name);
 389
 390 /** \cond doxygen isn't smart enough to recognize these */
 391 DEFINE_BLOB_FUNCTIONS(lyrics, lyr);
 392 DEFINE_BLOB_FUNCTIONS(images, img);
 393 DEFINE_BLOB_FUNCTIONS(moods, mood);
 394 DEFINE_BLOB_FUNCTIONS(playlists, pl);
 395 /** \endcond */