Fix com_lsblob().
[paraslash.git] / 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) {
142 send_buffer(fd, (char *)ls_output.data);
143 free(ls_output.data);
144 }
145 return ret;
146 }
147
148 static int com_catblob_callback(struct osl_table *table,
149 const struct osl_object *query, struct osl_object *output)
150 {
151 struct osl_object obj;
152 int ret;
153 struct osl_row *row;
154
155 ret = osl_get_row(table, BLOBCOL_NAME, query, &row);
156 if (ret < 0)
157 return ret;
158 ret = osl_open_disk_object(table, row, BLOBCOL_DEF, &obj);
159 if (ret < 0)
160 return ret;
161 output->data = para_malloc(obj.size);
162 output->size = obj.size;
163 memcpy(output->data, obj.data, obj.size);
164 return osl_close_disk_object(&obj);
165 }
166 static int com_catblob(callback_function *f, int fd, int argc,
167 char * const * const argv)
168 {
169 struct osl_object cat_output = {.data = NULL};
170 int ret;
171
172 if (argc != 2)
173 return -E_BLOB_SYNTAX;
174 if (!*argv[1]) /* empty name is reserved of the dummy row */
175 return -E_BLOB_SYNTAX;
176 ret = send_standard_callback_request(1, argv + 1, f, &cat_output);
177 if (ret >= 0 && cat_output.data)
178 ret = send_buffer(fd, (char *)cat_output.data);
179 free(cat_output.data);
180 return ret;
181
182 }
183
184 static int com_addblob_callback(struct osl_table *table,
185 const struct osl_object *query,
186 __a_unused struct osl_object *result)
187 {
188 struct osl_object objs[NUM_BLOB_COLUMNS];
189 char *name = query->data;
190 size_t name_len = strlen(name) + 1;
191 uint32_t id;
192 unsigned num_rows;
193 int ret;
194
195 ret = osl_get_num_rows(table, &num_rows);
196 if (ret < 0)
197 return ret;
198 if (!num_rows) { /* this is the first entry ever added */
199 /* insert dummy row containing the id */
200 id = 2; /* this entry will be entry #1, so 2 is the next */
201 objs[BLOBCOL_ID].data = &id;
202 objs[BLOBCOL_ID].size = sizeof(id);
203 objs[BLOBCOL_NAME].data = "";
204 objs[BLOBCOL_NAME].size = 1;
205 objs[BLOBCOL_DEF].data = "";
206 objs[BLOBCOL_DEF].size = 1;
207 ret = osl_add_row(table, objs);
208 if (ret < 0)
209 return ret;
210 } else { /* get id of the dummy row and increment it */
211 struct osl_row *row;
212 struct osl_object obj = {.data = "", .size = 1};
213 ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
214 if (ret < 0)
215 return ret;
216 ret = osl_get_object(table, row, BLOBCOL_ID, &obj);
217 if (ret < 0)
218 return ret;
219 id = *(uint32_t *)obj.data + 1;
220 obj.data = &id;
221 ret = osl_update_object(table, row, BLOBCOL_ID, &obj);
222 if (ret < 0)
223 return ret;
224 }
225 id--;
226 objs[BLOBCOL_ID].data = &id;
227 objs[BLOBCOL_ID].size = sizeof(id);
228 objs[BLOBCOL_NAME].data = name;
229 objs[BLOBCOL_NAME].size = name_len;
230 objs[BLOBCOL_DEF].data = name + name_len;
231 objs[BLOBCOL_DEF].size = query->size - name_len;
232 return osl_add_row(table, objs);
233 }
234
235 static int com_addblob(callback_function *f, __a_unused int fd, int argc,
236 char * const * const argv)
237 {
238 struct osl_object arg_obj;
239
240 if (argc != 2)
241 return -E_BLOB_SYNTAX;
242 if (!*argv[1]) /* empty name is reserved for the dummy row */
243 return -E_BLOB_SYNTAX;
244 PARA_NOTICE_LOG("argv[1]: %s\n", argv[1]);
245 arg_obj.size = strlen(argv[1]) + 1;
246 arg_obj.data = (char *)argv[1];
247 return stdin_command(&arg_obj, f, 10 * 1024 * 1024, NULL);
248 }
249
250 static int com_rmblob_callback(struct osl_table *table,
251 const struct osl_object *query,
252 __a_unused struct osl_object *result)
253 {
254 char *p = query->data;
255 size_t len;
256 int ret;
257
258 for (; p < (char *)query->data + query->size; p += len + 1) {
259 struct osl_row *row;
260 struct osl_object obj;
261
262 len = strlen(p);
263 obj.data = p;
264 obj.size = len + 1;
265 ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
266 if (ret < 0)
267 return ret;
268 ret = osl_del_row(table, row);
269 if (ret < 0)
270 return ret;
271 }
272 return 1;
273 }
274
275 static int com_rmblob(callback_function *f, __a_unused int fd, int argc,
276 char * const * const argv)
277 {
278 if (argc < 2)
279 return -E_MOOD_SYNTAX;
280 return send_option_arg_callback_request(NULL, argc - 1, argv + 1, f,
281 NULL);
282 }
283
284 static int com_mvblob_callback(struct osl_table *table,
285 const struct osl_object *query,
286 __a_unused struct osl_object *result)
287 {
288 char *src = (char *) query->data;
289 struct osl_object obj = {.data = src, .size = strlen(src) + 1};
290 char *dest = src + obj.size;
291 struct osl_row *row;
292 int ret = osl_get_row(table, BLOBCOL_NAME, &obj, &row);
293
294 if (ret < 0)
295 return ret;
296 obj.data = dest;
297 obj.size = strlen(dest) + 1;
298 return osl_update_object(table, row, BLOBCOL_NAME, &obj);
299 }
300
301 static int com_mvblob(callback_function *f, __a_unused int fd,
302 int argc, char * const * const argv)
303 {
304 if (argc != 3)
305 return -E_MOOD_SYNTAX;
306 return send_option_arg_callback_request(NULL, argc - 1, argv + 1, f,
307 NULL);
308 }
309
310 #define DEFINE_BLOB_COMMAND(cmd_name, table_name, cmd_prefix) \
311 static int com_ ## cmd_name ## cmd_prefix ## _callback(const struct osl_object *query, \
312 struct osl_object *output) \
313 { \
314 return com_ ## cmd_name ## blob_callback(table_name ## _table, query, output); \
315 } \
316 int com_ ## cmd_name ## cmd_prefix(int fd, int argc, char * const * const argv) \
317 { \
318 return com_ ## cmd_name ## blob(com_ ## cmd_name ## cmd_prefix ## _callback, fd, argc, argv); \
319 }
320
321 static int blob_get_name_by_id(struct osl_table *table, uint32_t id,
322 char **name)
323 {
324 struct osl_row *row;
325 struct osl_object obj = {.data = &id, .size = sizeof(id)};
326 int ret;
327
328 *name = NULL;
329 if (!id)
330 return 1;
331 ret = osl_get_row(table, BLOBCOL_ID, &obj, &row);
332 if (ret < 0)
333 return ret;
334 ret = osl_get_object(table, row, BLOBCOL_NAME, &obj);
335 if (ret < 0)
336 return ret;
337 *name = (char *)obj.data;
338 return 1;
339 }
340 /** Define the \p get_name_by_id function for this blob type. */
341 #define DEFINE_GET_NAME_BY_ID(table_name, cmd_prefix) \
342 int cmd_prefix ## _get_name_by_id(uint32_t id, char **name) \
343 { \
344 return blob_get_name_by_id(table_name ## _table, id, name); \
345 }
346
347 /** Define the \p shutdown function for this blob type. */
348 #define DEFINE_BLOB_SHUTDOWN(table_name) \
349 void table_name ## _shutdown(enum osl_close_flags flags) \
350 { \
351 osl_close_table(table_name ## _table, flags); \
352 table_name ## _table = NULL; \
353 }
354
355 static int blob_init(struct osl_table **table,
356 struct osl_table_description *desc,
357 struct table_info *ti, const char *db)
358 {
359 int ret;
360 desc->dir = db;
361 ti->desc = desc;
362 ret = osl_open_table(ti->desc, &ti->table);
363 if (ret >= 0) {
364 *table = ti->table;
365 return ret;
366 }
367 *table = NULL;
368 return ret == -E_NOENT? 1 : ret;
369 }
370
371 /** Define the \p init function for this blob type. */
372 #define DEFINE_BLOB_INIT(table_name) \
373 int table_name ## _init(struct table_info *ti, const char *db) \
374 { \
375 return blob_init(&table_name ## _table, \
376 &table_name ## _table_desc, ti, db); \
377 }
378
379
380 /** Define all functions for this blob type. */
381 #define DEFINE_BLOB_FUNCTIONS(table_name, cmd_prefix) \
382 DEFINE_BLOB_COMMAND(ls, table_name, cmd_prefix) \
383 DEFINE_BLOB_COMMAND(cat, table_name, cmd_prefix) \
384 DEFINE_BLOB_COMMAND(add, table_name, cmd_prefix) \
385 DEFINE_BLOB_COMMAND(rm, table_name, cmd_prefix) \
386 DEFINE_BLOB_COMMAND(mv, table_name, cmd_prefix) \
387 DEFINE_GET_NAME_BY_ID(table_name, cmd_prefix); \
388 DEFINE_BLOB_SHUTDOWN(table_name); \
389 DEFINE_BLOB_INIT(table_name);
390
391 /** \cond doxygen isn't smart enough to recognize these */
392 DEFINE_BLOB_FUNCTIONS(lyrics, lyr);
393 DEFINE_BLOB_FUNCTIONS(images, img);
394 DEFINE_BLOB_FUNCTIONS(moods, mood);
395 DEFINE_BLOB_FUNCTIONS(playlists, pl);
396 /** \endcond */