Make afs commands work.
[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, const char **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 const char **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 const char **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 const char **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, const char **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(__a_unused int fd, int argc, const char **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 const struct osl_table_description *desc,
356 struct table_info *ti)
357 {
358 int ret;
359
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) \
373 { \
374 return blob_init(&table_name ## _table, \
375 &table_name ## _table_desc, ti); \
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 */