build: Create section for para_write.
[paraslash.git] / aft.c
1 /*
2  * Copyright (C) 2007-2013 Andre Noll <maan@systemlinux.org>
3  *
4  * Licensed under the GPL v2. For licencing details see COPYING.
5  */
6
7 /** \file aft.c Audio file table functions. */
8
9 #include <regex.h>
10 #include <sys/mman.h>
11 #include <fnmatch.h>
12 #include <sys/shm.h>
13 #include <osl.h>
14
15 #include "para.h"
16 #include "error.h"
17 #include "crypt.h"
18 #include "string.h"
19 #include "afh.h"
20 #include "afs.h"
21 #include "fd.h"
22 #include "ipc.h"
23 #include "portable_io.h"
24 #include "sideband.h"
25 #include "command.h"
26
27 static struct osl_table *audio_file_table;
28 static char *status_items;
29 static char *parser_friendly_status_items;
30
31 /** The different sorting methods of the ls command. */
32 enum ls_sorting_method {
33         /** -sp (default) */
34         LS_SORT_BY_PATH,
35         /** -ss */
36         LS_SORT_BY_SCORE,
37         /** -sl */
38         LS_SORT_BY_LAST_PLAYED,
39         /** -sn */
40         LS_SORT_BY_NUM_PLAYED,
41         /** -sf */
42         LS_SORT_BY_FREQUENCY,
43         /** -sc */
44         LS_SORT_BY_CHANNELS,
45         /** -si */
46         LS_SORT_BY_IMAGE_ID,
47         /** -sy */
48         LS_SORT_BY_LYRICS_ID,
49         /** -sb */
50         LS_SORT_BY_BITRATE,
51         /** -sd */
52         LS_SORT_BY_DURATION,
53         /** -sa */
54         LS_SORT_BY_AUDIO_FORMAT,
55         /** -sh */
56         LS_SORT_BY_HASH,
57 };
58
59 /** The different listing modes of the ls command. */
60 enum ls_listing_mode {
61         /** Default listing mode. */
62         LS_MODE_SHORT,
63         /** -l or -ll */
64         LS_MODE_LONG,
65         /** -lv */
66         LS_MODE_VERBOSE,
67         /** -lm */
68         LS_MODE_MBOX,
69         /** -lc */
70         LS_MODE_CHUNKS,
71         /** -lp */
72         LS_MODE_PARSER,
73 };
74
75 /** The flags accepted by the ls command. */
76 enum ls_flags {
77         /** -p */
78         LS_FLAG_FULL_PATH = 1,
79         /** -a */
80         LS_FLAG_ADMISSIBLE_ONLY = 2,
81         /** -r */
82         LS_FLAG_REVERSE = 4,
83         /** -d */
84         LS_FLAG_UNIXDATE = 8,
85 };
86
87 /**
88  * The size of the individual output fields of the ls command.
89  *
90  * These depend on the actual content being listed. If, for instance only files
91  * with duration less than an hour are being listed, then the duration with is
92  * made smaller because then the duration is listed as mm:ss rather than
93  * hh:mm:ss.
94  */
95 struct ls_widths {
96         /** size of the score field. */
97         unsigned short score_width;
98         /** size of the image id field. */
99         unsigned short image_id_width;
100         /** size of the lyrics id field. */
101         unsigned short lyrics_id_width;
102         /** size of the bitrate field. */
103         unsigned short bitrate_width;
104         /** size of the frequency field. */
105         unsigned short frequency_width;
106         /** size of the duration field. */
107         unsigned short duration_width;
108         /** size of the num played field. */
109         unsigned short num_played_width;
110         /** size of the amp field. */
111         unsigned short amp_width;
112         /** size of the audio format field. */
113         unsigned short audio_format_width;
114 };
115
116 /** Data passed from the ls command handler to its callback function. */
117 struct ls_options {
118         /** The given command line flags. */
119         unsigned flags;
120         /** The sorting method given at the command line. */
121         enum ls_sorting_method sorting;
122         /** The given listing mode (short, long, verbose, mbox). */
123         enum ls_listing_mode mode;
124         /** The arguments passed to the ls command. */
125         char **patterns;
126         /** Number of non-option arguments. */
127         int num_patterns;
128         /** Used for long listing mode to align the output fields. */
129         struct ls_widths widths;
130         /** Size of the \a data array. */
131         uint32_t array_size;
132         /** Number of used entries in the data array. */
133         uint32_t num_matching_paths;
134         /** Array of matching entries. */
135         struct ls_data *data;
136         /** Used to sort the array. */
137         struct ls_data **data_ptr;
138 };
139
140 /**
141  * Describes the layout of the mmapped-afs info struct.
142  *
143  * \sa struct afs_info.
144  */
145 enum afsi_offsets {
146         /** Where .last_played is stored. */
147         AFSI_LAST_PLAYED_OFFSET = 0,
148         /** Storage position of the attributes bitmap. */
149         AFSI_ATTRIBUTES_OFFSET = 8,
150         /** Storage position of the .num_played field. */
151         AFSI_NUM_PLAYED_OFFSET = 16,
152         /** Storage position of the .image_id field. */
153         AFSI_IMAGE_ID_OFFSET = 20,
154         /** Storage position of the .lyrics_id field. */
155         AFSI_LYRICS_ID_OFFSET = 24,
156         /** Storage position of the .audio_format_id field. */
157         AFSI_AUDIO_FORMAT_ID_OFFSET = 28,
158         /** Storage position of the amplification field. */
159         AFSI_AMP_OFFSET = 29,
160         /** 2 bytes reserved space for future usage. */
161         AFSI_AUDIO_FORMAT_UNUSED_OFFSET = 30,
162         /** On-disk storage space needed. */
163         AFSI_SIZE = 32
164 };
165
166 /**
167  * Convert a struct afs_info to an osl object.
168  *
169  * \param afsi Pointer to the audio file info to be converted.
170  * \param obj Result pointer.
171  *
172  * \sa load_afsi().
173  */
174 void save_afsi(struct afs_info *afsi, struct osl_object *obj)
175 {
176         char *buf = obj->data;
177
178         write_u64(buf + AFSI_LAST_PLAYED_OFFSET, afsi->last_played);
179         write_u64(buf + AFSI_ATTRIBUTES_OFFSET, afsi->attributes);
180         write_u32(buf + AFSI_NUM_PLAYED_OFFSET, afsi->num_played);
181         write_u32(buf + AFSI_IMAGE_ID_OFFSET, afsi->image_id);
182         write_u32(buf + AFSI_LYRICS_ID_OFFSET, afsi->lyrics_id);
183         write_u8(buf + AFSI_AUDIO_FORMAT_ID_OFFSET,
184                 afsi->audio_format_id);
185         write_u8(buf + AFSI_AMP_OFFSET, afsi->amp);
186         memset(buf + AFSI_AUDIO_FORMAT_UNUSED_OFFSET, 0, 2);
187 }
188
189 /**
190  *  Get the audio file selector info struct stored in an osl object.
191  *
192  * \param afsi Points to the audio_file info structure to be filled in.
193  * \param obj The osl object holding the data.
194  *
195  * \return Positive on success, negative on errors. Possible errors: \p E_BAD_AFS.
196  *
197  * \sa save_afsi().
198  */
199 int load_afsi(struct afs_info *afsi, struct osl_object *obj)
200 {
201         char *buf = obj->data;
202         if (obj->size < AFSI_SIZE)
203                 return -E_BAD_AFSI;
204         afsi->last_played = read_u64(buf + AFSI_LAST_PLAYED_OFFSET);
205         afsi->attributes = read_u64(buf + AFSI_ATTRIBUTES_OFFSET);
206         afsi->num_played = read_u32(buf + AFSI_NUM_PLAYED_OFFSET);
207         afsi->image_id = read_u32(buf + AFSI_IMAGE_ID_OFFSET);
208         afsi->lyrics_id = read_u32(buf + AFSI_LYRICS_ID_OFFSET);
209         afsi->audio_format_id = read_u8(buf +
210                 AFSI_AUDIO_FORMAT_ID_OFFSET);
211         afsi->amp = read_u8(buf + AFSI_AMP_OFFSET);
212         return 1;
213 }
214
215 /** The columns of the audio file table. */
216 enum audio_file_table_columns {
217         /** The hash on the content of the audio file. */
218         AFTCOL_HASH,
219         /** The full path in the filesystem. */
220         AFTCOL_PATH,
221         /** The audio file selector info. */
222         AFTCOL_AFSI,
223         /** The audio format handler info. */
224         AFTCOL_AFHI,
225         /** The chunk table info and the chunk table of the audio file. */
226         AFTCOL_CHUNKS,
227         /** The number of columns of this table. */
228         NUM_AFT_COLUMNS
229 };
230
231 /**
232  * Compare two osl objects pointing to hash values.
233  *
234  * \param obj1 Pointer to the first hash object.
235  * \param obj2 Pointer to the second hash object.
236  *
237  * \return The values required for an osl compare function.
238  *
239  * \sa osl_compare_func, uint32_compare().
240  */
241 static int aft_hash_compare(const struct osl_object *obj1, const struct osl_object *obj2)
242 {
243         return hash_compare((unsigned char *)obj1->data, (unsigned char *)obj2->data);
244 }
245
246 static struct osl_column_description aft_cols[] = {
247         [AFTCOL_HASH] = {
248                 .storage_type = OSL_MAPPED_STORAGE,
249                 .storage_flags = OSL_RBTREE | OSL_FIXED_SIZE | OSL_UNIQUE,
250                 .name = "hash",
251                 .compare_function = aft_hash_compare,
252                 .data_size = HASH_SIZE
253         },
254         [AFTCOL_PATH] = {
255                 .storage_type = OSL_MAPPED_STORAGE,
256                 .storage_flags = OSL_RBTREE | OSL_UNIQUE,
257                 .name = "path",
258                 .compare_function = string_compare,
259         },
260         [AFTCOL_AFSI] = {
261                 .storage_type = OSL_MAPPED_STORAGE,
262                 .storage_flags = OSL_FIXED_SIZE,
263                 .name = "afs_info",
264                 .data_size = AFSI_SIZE
265         },
266         [AFTCOL_AFHI] = {
267                 .storage_type = OSL_MAPPED_STORAGE,
268                 .name = "afh_info",
269         },
270         [AFTCOL_CHUNKS] = {
271                 .storage_type = OSL_DISK_STORAGE,
272                 .name = "chunks",
273         }
274 };
275
276 static struct osl_table_description audio_file_table_desc = {
277         .name = "audio_files",
278         .num_columns = NUM_AFT_COLUMNS,
279         .flags = OSL_LARGE_TABLE,
280         .column_descriptions = aft_cols
281 };
282
283 /* We don't want dot or dot-dot anywhere. */
284 static int verify_dotfile(const char *rest)
285 {
286         /*
287          * The first character was '.', but that has already been discarded, we
288          * now test the rest.
289          */
290         switch (*rest) {
291         case '\0': case '/': /* /foo/. and /foo/./bar are not ok */
292                 return -1;
293         case '.': /* path start with /foo/.. */
294                 if (rest[1] == '\0' || rest[1] == '/')
295                         return -1; /* /foo/.. or /foo/../bar are not ok */
296                 /* /foo/..bar is ok */
297         }
298         return 1;
299 }
300
301 /*
302  * We fundamentally don't like some paths: We don't want double slashes or
303  * slashes at the end that can make pathnames ambiguous.
304  */
305 static int verify_path(const char *orig_path, char **resolved_path)
306 {
307         char c;
308         size_t len;
309         char *path;
310
311         if (*orig_path != '/') /* we only accept absolute paths */
312                 return -E_BAD_PATH;
313         len = strlen(orig_path);
314         *resolved_path = para_strdup(orig_path);
315         path = *resolved_path;
316         while (len > 1 && path[--len] == '/')
317                 path[len] = '\0'; /* remove slash at the end */
318         c = *path++;
319         while (c) {
320                 if (c == '/') {
321                         c = *path++;
322                         switch (c) {
323                         case '/': /* double slash */
324                                 goto bad_path;
325                         case '.':
326                                 if (verify_dotfile(path) < 0)
327                                         goto bad_path;
328                         default:
329                                 continue;
330                         }
331                 }
332                 c = *path++;
333         }
334         return 1;
335 bad_path:
336         free(*resolved_path);
337         return -E_BAD_PATH;
338 }
339
340 /** The on-disk layout of a afhi struct. */
341 enum afhi_offsets {
342         /** Where the number of seconds is stored. */
343         AFHI_SECONDS_TOTAL_OFFSET = 0,
344         /** Position of the bitrate. */
345         AFHI_BITRATE_OFFSET = 4,
346         /** Position of the frequency. */
347         AFHI_FREQUENCY_OFFSET = 8,
348         /** Was: Location of the audio file header. */
349         AFHI_UNUSED1_OFFSET = 12,
350         /* Length of the audio file header. Zero means: No header. */
351         AFHI_HEADER_LEN_OFFSET = 16,
352         /** The total number of chunks (4 bytes). */
353         CHUNKS_TOTAL_OFFSET = 20,
354         /** The length of the audio file header (4 bytes). */
355         HEADER_LEN_OFFSET = 24,
356         /** Was: The start of the audio file header (4 bytes). */
357         AFHI_UNUSED2_OFFSET = 28,
358         /** The seconds part of the chunk time (4 bytes). */
359         CHUNK_TV_TV_SEC_OFFSET = 32,
360         /** The microseconds part of the chunk time (4 bytes). */
361         CHUNK_TV_TV_USEC_OFFSET = 36,
362         /** Number of channels is stored here. (1 byte) */
363         AFHI_CHANNELS_OFFSET = 40,
364         /** The tag info position. */
365         AFHI_INFO_STRING_OFFSET = 41,
366         /** Minimal on-disk size of a valid afhi struct. */
367         MIN_AFHI_SIZE = 47, /* at least 6 null bytes for techinfo/tags */
368 };
369
370 static unsigned sizeof_afhi_buf(const struct afh_info *afhi)
371 {
372         if (!afhi)
373                 return 0;
374         return MIN_AFHI_SIZE
375                 + strlen(afhi->techinfo)
376                 + strlen(afhi->tags.artist)
377                 + strlen(afhi->tags.title)
378                 + strlen(afhi->tags.year)
379                 + strlen(afhi->tags.album)
380                 + strlen(afhi->tags.comment);
381 }
382
383 static void save_afhi(struct afh_info *afhi, char *buf)
384 {
385         char *p;
386
387         if (!afhi)
388                 return;
389         write_u32(buf + AFHI_SECONDS_TOTAL_OFFSET, afhi->seconds_total);
390         write_u32(buf + AFHI_BITRATE_OFFSET, afhi->bitrate);
391         write_u32(buf + AFHI_FREQUENCY_OFFSET, afhi->frequency);
392         write_u32(buf + AFHI_UNUSED1_OFFSET, 0);
393         write_u32(buf + AFHI_HEADER_LEN_OFFSET, afhi->header_len);
394         write_u8(buf + AFHI_CHANNELS_OFFSET, afhi->channels);
395         write_u32(buf + CHUNKS_TOTAL_OFFSET, afhi->chunks_total);
396         write_u32(buf + HEADER_LEN_OFFSET, afhi->header_len);
397         write_u32(buf + AFHI_UNUSED2_OFFSET, 0);
398         write_u32(buf + CHUNK_TV_TV_SEC_OFFSET, afhi->chunk_tv.tv_sec);
399         write_u32(buf + CHUNK_TV_TV_USEC_OFFSET, afhi->chunk_tv.tv_usec);
400         p = buf + AFHI_INFO_STRING_OFFSET;
401         /* The sprintf's below are OK as our caller made sure that buf is large enough */
402         p += sprintf(p, "%s", afhi->techinfo) + 1;
403         p += sprintf(p, "%s", afhi->tags.artist) + 1;
404         p += sprintf(p, "%s", afhi->tags.title) + 1;
405         p += sprintf(p, "%s", afhi->tags.year) + 1;
406         p += sprintf(p, "%s", afhi->tags.album) + 1;
407         sprintf(p, "%s", afhi->tags.comment);
408 }
409
410 static void load_afhi(const char *buf, struct afh_info *afhi)
411 {
412         afhi->seconds_total = read_u32(buf + AFHI_SECONDS_TOTAL_OFFSET);
413         afhi->bitrate = read_u32(buf + AFHI_BITRATE_OFFSET);
414         afhi->frequency = read_u32(buf + AFHI_FREQUENCY_OFFSET);
415         afhi->header_len = read_u32(buf + AFHI_HEADER_LEN_OFFSET);
416         afhi->channels = read_u8(buf + AFHI_CHANNELS_OFFSET);
417         afhi->chunks_total = read_u32(buf + CHUNKS_TOTAL_OFFSET);
418         afhi->header_len = read_u32(buf + HEADER_LEN_OFFSET);
419         afhi->chunk_tv.tv_sec = read_u32(buf + CHUNK_TV_TV_SEC_OFFSET);
420         afhi->chunk_tv.tv_usec = read_u32(buf + CHUNK_TV_TV_USEC_OFFSET);
421         afhi->techinfo = (char *)buf + AFHI_INFO_STRING_OFFSET;
422         afhi->tags.artist = afhi->techinfo + strlen(afhi->techinfo) + 1;
423         afhi->tags.title = afhi->tags.artist + strlen(afhi->tags.artist) + 1;
424         afhi->tags.year = afhi->tags.title + strlen(afhi->tags.title) + 1;
425         afhi->tags.album = afhi->tags.year + strlen(afhi->tags.year) + 1;
426         afhi->tags.comment = afhi->tags.album + strlen(afhi->tags.album) + 1;
427 }
428
429 static unsigned sizeof_chunk_table(struct afh_info *afhi)
430 {
431         if (!afhi)
432                 return 0;
433         return 4 * (afhi->chunks_total + 1);
434 }
435
436 static uint32_t save_chunk_table(struct afh_info *afhi, char *buf)
437 {
438         int i;
439         uint32_t max = 0, old = 0;
440
441         for (i = 0; i <= afhi->chunks_total; i++) {
442                 uint32_t val = afhi->chunk_table[i];
443                 write_u32(buf + 4 * i, val);
444                 /*
445                  * If the first chunk is the header, do not consider it for the
446                  * calculation of the largest chunk size.
447                  */
448                 if (i == 0 || (i == 1 && afhi->header_len > 0)) {
449                         old = val;
450                         continue;
451                 }
452                 max = PARA_MAX(max, val - old);
453                 old = val;
454         }
455         return max;
456 }
457
458 static void load_chunk_table(struct afh_info *afhi, char *buf)
459 {
460         int i;
461
462         afhi->chunk_table = para_malloc(sizeof_chunk_table(afhi));
463         for (i = 0; i <= afhi->chunks_total; i++)
464                 afhi->chunk_table[i] = read_u32(buf + 4 * i);
465 }
466
467 /**
468  * Get the row of the audio file table corresponding to the given path.
469  *
470  * \param path The full path of the audio file.
471  * \param row Result pointer.
472  *
473  * \return Standard.
474  */
475 int aft_get_row_of_path(const char *path, struct osl_row **row)
476 {
477         struct osl_object obj = {.data = (char *)path, .size = strlen(path) + 1};
478
479         return osl(osl_get_row(audio_file_table, AFTCOL_PATH, &obj, row));
480 }
481
482 /**
483  * Get the row of the audio file table corresponding to the given hash value.
484  *
485  * \param hash The hash value of the desired audio file.
486  * \param row Result pointer.
487  *
488  * \return Standard.
489  */
490 static int aft_get_row_of_hash(unsigned char *hash, struct osl_row **row)
491 {
492         const struct osl_object obj = {.data = hash, .size = HASH_SIZE};
493         return osl(osl_get_row(audio_file_table, AFTCOL_HASH, &obj, row));
494 }
495
496 /**
497  * Get the osl object holding the audio file selector info of a row.
498  *
499  * \param row Pointer to a row in the audio file table.
500  * \param obj Result pointer.
501  *
502  * \return Standard.
503  */
504 int get_afsi_object_of_row(const struct osl_row *row, struct osl_object *obj)
505 {
506         return osl(osl_get_object(audio_file_table, row, AFTCOL_AFSI, obj));
507 }
508
509 /**
510  * Get the osl object holding the audio file selector info, given a path.
511  *
512  *
513  * \param path The full path of the audio file.
514  * \param obj Result pointer.
515  *
516  * \return Positive on success, negative on errors.
517  */
518 static int get_afsi_object_of_path(const char *path, struct osl_object *obj)
519 {
520         struct osl_row *row;
521         int ret = aft_get_row_of_path(path, &row);
522         if (ret < 0)
523                 return ret;
524         return get_afsi_object_of_row(row, obj);
525 }
526
527 /**
528  * Get the audio file selector info, given a row of the audio file table.
529  *
530  * \param row Pointer to a row in the audio file table.
531  * \param afsi Result pointer.
532  *
533  * \return Positive on success, negative on errors.
534  */
535 int get_afsi_of_row(const struct osl_row *row, struct afs_info *afsi)
536 {
537         struct osl_object obj;
538         int ret = get_afsi_object_of_row(row, &obj);
539         if (ret < 0)
540                 return ret;
541         return load_afsi(afsi, &obj);
542 }
543
544 /**
545  * Get the audio file selector info, given the path of an audio table.
546  *
547  * \param path The full path of the audio file.
548  * \param afsi Result pointer.
549  *
550  * \return Positive on success, negative on errors.
551  */
552 int get_afsi_of_path(const char *path, struct afs_info *afsi)
553 {
554         struct osl_object obj;
555         int ret = get_afsi_object_of_path(path, &obj);
556         if (ret < 0)
557                 return ret;
558         return load_afsi(afsi, &obj);
559 }
560
561 /**
562  * Get the path of an audio file, given a row of the audio file table.
563  *
564  * \param row Pointer to a row in the audio file table.
565  * \param path Result pointer.
566  *
567  * The result is a pointer to mmapped data. The caller must not attempt
568  * to free it.
569  *
570  * \return Standard.
571  */
572 int get_audio_file_path_of_row(const struct osl_row *row, char **path)
573 {
574         struct osl_object path_obj;
575         int ret = osl(osl_get_object(audio_file_table, row, AFTCOL_PATH,
576                 &path_obj));
577         if (ret < 0)
578                 return ret;
579         *path = path_obj.data;
580         return 1;
581 }
582
583 /**
584  * Get the object containing the hash value of an audio file, given a row.
585  *
586  * \param row Pointer to a row of the audio file table.
587  * \param obj Result pointer.
588  *
589  * \return The return value of the underlying call to osl_get_object().
590  *
591  * \sa get_hash_of_row().
592  */
593 static int get_hash_object_of_aft_row(const struct osl_row *row,
594                 struct osl_object *obj)
595 {
596         return osl(osl_get_object(audio_file_table, row, AFTCOL_HASH, obj));
597 }
598
599 /**
600  * Get the hash value of an audio file, given a row of the audio file table.
601  *
602  * \param row Pointer to a row of the audio file table.
603  * \param hash Result pointer.
604  *
605  * \a hash points to mapped data and must not be freed by the caller.
606  *
607  * \return The return value of the underlying call to
608  * get_hash_object_of_aft_row().
609  */
610 static int get_hash_of_row(const struct osl_row *row, unsigned char **hash)
611 {
612         struct osl_object obj;
613         int ret = get_hash_object_of_aft_row(row, &obj);
614
615         if (ret < 0)
616                 return ret;
617         *hash = obj.data;
618         return 1;
619 }
620
621 /**
622  * Get the audio format handler info, given a row of the audio file table.
623  *
624  * \param row Pointer to a row of the audio file table.
625  * \param afhi Result pointer.
626  *
627  * \return The return value of the underlying call to osl_get_object().
628  *
629  * \sa get_chunk_table_of_row().
630  */
631 int get_afhi_of_row(const struct osl_row *row, struct afh_info *afhi)
632 {
633         struct osl_object obj;
634         int ret = osl(osl_get_object(audio_file_table, row, AFTCOL_AFHI,
635                 &obj));
636         if (ret < 0)
637                 return ret;
638         load_afhi(obj.data, afhi);
639         return 1;
640 }
641
642 /* returns shmid on success */
643 static int save_afd(struct audio_file_data *afd)
644 {
645         size_t size = sizeof(*afd) + sizeof_chunk_table(&afd->afhi);
646         int shmid, ret = shm_new(size);
647         void *shm_afd;
648         char *buf;
649
650         if (ret < 0)
651                 return ret;
652         shmid = ret;
653         ret = shm_attach(shmid, ATTACH_RW, &shm_afd);
654         if (ret < 0)
655                 goto err;
656         buf = shm_afd;
657         buf += sizeof(*afd);
658         afd->max_chunk_size = save_chunk_table(&afd->afhi, buf);
659         *(struct audio_file_data *)shm_afd = *afd;
660         shm_detach(shm_afd);
661         return shmid;
662 err:
663         shm_destroy(shmid);
664         return ret;
665 }
666
667 /**
668  * Extract a afd stored in a shared memory area.
669  *
670  * Attach the shared memory area given by \a shmid, load the audio file data
671  * stored therein and detach the area afterwards.  Called by vss, after
672  * receiving a positive response to the request for the next audio file.
673  +
674  * \param shmid The identifier of the shared memory area containing the afd.
675  * \param afd Result pointer.
676  *
677  * \return Standard.
678  */
679 int load_afd(int shmid, struct audio_file_data *afd)
680 {
681         void *shm_afd;
682         char *buf;
683         int ret;
684
685         ret = shm_attach(shmid, ATTACH_RO, &shm_afd);
686         if (ret < 0)
687                 return ret;
688         *afd = *(struct audio_file_data *)shm_afd;
689         buf = shm_afd;
690         buf += sizeof(*afd);
691         load_chunk_table(&afd->afhi, buf);
692         shm_detach(shm_afd);
693         return 1;
694 }
695
696 static int get_local_time(uint64_t *seconds, char *buf, size_t size,
697         time_t current_time, enum ls_listing_mode lm)
698 {
699         struct tm t;
700
701         if (!localtime_r((time_t *)seconds, &t))
702                 return -E_LOCALTIME;
703         if (lm == LS_MODE_MBOX) {
704                 if (!strftime(buf, size, "%c", &t))
705                         return -E_STRFTIME;
706                 return 1;
707         }
708         if (*seconds + 6 * 30 * 24 * 3600 > current_time) {
709                 if (!strftime(buf, size, "%b %e %k:%M", &t))
710                         return -E_STRFTIME;
711                 return 1;
712         }
713         if (!strftime(buf, size, "%b %e  %Y", &t))
714                 return -E_STRFTIME;
715         return 1;
716 }
717
718 /** Compute the number of (decimal) digits of a number. */
719 #define GET_NUM_DIGITS(x, num) { \
720         typeof((x)) _tmp = PARA_ABS(x); \
721         *num = 1; \
722         if ((_tmp)) \
723                 while ((_tmp) > 9) { \
724                         (_tmp) /= 10; \
725                         (*num)++; \
726                 } \
727         }
728
729 __a_const static short unsigned get_duration_width(int seconds)
730 {
731         short unsigned width;
732         unsigned hours = seconds / 3600, mins = (seconds % 3600) / 60;
733
734         if (!hours) /* less than one hour => m:ss or mm:ss => 4 or 5 digits */
735                 return 4 + (mins > 9);
736         /* more than one hour => h:mm:ss, hh:mm:ss, hhh:mm:ss, ... */
737         GET_NUM_DIGITS(hours, &width);
738         return width + 6;
739 }
740
741 static void get_duration_buf(int seconds, char *buf, struct ls_options *opts)
742 {
743         unsigned hours = seconds / 3600, mins = (seconds % 3600) / 60;
744         short unsigned max_width;
745
746         if (!hours) { /* m:ss or mm:ss */
747                 max_width = opts->mode == LS_MODE_LONG?
748                         opts->widths.duration_width : 4;
749                 sprintf(buf, "%*u:%02u", max_width - 3, mins, seconds % 60);
750         } else { /* more than one hour => h:mm:ss, hh:mm:ss, hhh:mm:ss, ... */
751                 max_width = opts->mode == LS_MODE_LONG?
752                         opts->widths.duration_width : 7;
753                 sprintf(buf, "%*u:%02u:%02u", max_width - 6, hours, mins,
754                         seconds % 60);
755         }
756 }
757
758
759 static int write_attribute_items(struct para_buffer *b,
760                 const char *att_bitmap, struct afs_info *afsi)
761 {
762         char *att_text;
763         int ret;
764
765         ret = WRITE_STATUS_ITEM(b, SI_ATTRIBUTES_BITMAP, "%s\n", att_bitmap);
766         if (ret < 0)
767                 return ret;
768         ret = get_attribute_text(&afsi->attributes, " ", &att_text);
769         if (ret < 0)
770                 return ret;
771         ret = WRITE_STATUS_ITEM(b, SI_ATTRIBUTES_TXT, "%s\n", att_text);
772         free(att_text);
773         return ret;
774 }
775
776 static int write_lyrics_items(struct para_buffer *b, struct afs_info *afsi)
777 {
778         char *lyrics_name;
779         int ret;
780
781         ret = WRITE_STATUS_ITEM(b, SI_LYRICS_ID, "%u\n", afsi->lyrics_id);
782         if (ret < 0)
783                 return ret;
784         lyr_get_name_by_id(afsi->lyrics_id, &lyrics_name);
785         return WRITE_STATUS_ITEM(b, SI_LYRICS_NAME, "%s\n", lyrics_name?
786                 lyrics_name : "(none)");
787 }
788
789 static int write_image_items(struct para_buffer *b, struct afs_info *afsi)
790 {
791         char *image_name;
792         int ret;
793
794         ret = WRITE_STATUS_ITEM(b, SI_IMAGE_ID, "%u\n", afsi->image_id);
795         if (ret < 0)
796                 return ret;
797         img_get_name_by_id(afsi->image_id, &image_name);
798         return WRITE_STATUS_ITEM(b, SI_IMAGE_NAME, "%s\n", image_name?
799                 image_name : "(none)");
800 }
801
802 static int write_filename_items(struct para_buffer *b, const char *path,
803                 unsigned flags)
804 {
805         char *val;
806         int ret;
807
808         if (!(flags & LS_FLAG_FULL_PATH))
809                 return WRITE_STATUS_ITEM(b, SI_BASENAME, "%s\n", path);
810         ret = WRITE_STATUS_ITEM(b, SI_PATH, "%s\n", path);
811         if (ret < 0)
812                 return ret;
813         val = para_basename(path);
814         ret = WRITE_STATUS_ITEM(b, SI_BASENAME, "%s\n", val? val : "");
815         if (ret < 0)
816                 return ret;
817         val = para_dirname(path);
818         ret = WRITE_STATUS_ITEM(b, SI_DIRECTORY, "%s\n", val? val : "");
819         free(val);
820         return ret;
821 }
822
823 static int print_chunk_table(struct ls_data *d, struct para_buffer *b)
824 {
825         struct osl_object chunk_table_obj;
826         struct osl_row *aft_row;
827         int ret, i;
828         char *buf;
829
830         ret = aft_get_row_of_hash(d->hash, &aft_row);
831         if (ret < 0)
832                 return ret;
833         ret = osl_open_disk_object(audio_file_table, aft_row,
834                 AFTCOL_CHUNKS, &chunk_table_obj);
835         if (ret < 0)
836                 return ret;
837         ret = para_printf(b, "%s\n"
838                 "chunk_time: %lu:%lu\nchunk_offsets: ",
839                 d->path,
840                 (long unsigned) d->afhi.chunk_tv.tv_sec,
841                 (long unsigned) d->afhi.chunk_tv.tv_usec
842         );
843         if (ret < 0)
844                 goto out;
845         buf = chunk_table_obj.data;
846         for (i = 0; i <= d->afhi.chunks_total; i++) {
847                 ret = para_printf(b, "%u ", (unsigned) read_u32(buf + 4 * i));
848                 if (ret < 0)
849                         goto out;
850         }
851         ret = para_printf(b, "\n");
852 out:
853         osl_close_disk_object(&chunk_table_obj);
854         return ret;
855 }
856
857 static int write_score(struct para_buffer *b, struct ls_data *d,
858                 struct ls_options *opts)
859 {
860         if (!(opts->flags & LS_FLAG_ADMISSIBLE_ONLY)) /* no score*/
861                 return 0;
862         return WRITE_STATUS_ITEM(b, SI_SCORE, "%li\n", d->score);
863 }
864
865 static int print_list_item(struct ls_data *d, struct ls_options *opts,
866         struct para_buffer *b, time_t current_time)
867 {
868         int ret;
869         char att_buf[65];
870         char last_played_time[30];
871         char duration_buf[30]; /* nobody has an audio file long enough to overflow this */
872         struct afs_info *afsi = &d->afsi;
873         struct afh_info *afhi = &d->afhi;
874         char asc_hash[2 * HASH_SIZE + 1];
875
876         if (opts->mode == LS_MODE_SHORT) {
877                 ret = para_printf(b, "%s\n", d->path);
878                 goto out;
879         }
880         if (opts->mode == LS_MODE_CHUNKS) {
881                 ret = print_chunk_table(d, b);
882                 goto out;
883         }
884         get_attribute_bitmap(&afsi->attributes, att_buf);
885         if (opts->flags & LS_FLAG_UNIXDATE)
886                 sprintf(last_played_time, "%llu",
887                         (long long unsigned)afsi->last_played);
888         else {
889                 ret = get_local_time(&afsi->last_played, last_played_time,
890                         sizeof(last_played_time), current_time, opts->mode);
891                 if (ret < 0)
892                         goto out;
893         }
894         get_duration_buf(afhi->seconds_total, duration_buf, opts);
895         if (opts->mode == LS_MODE_LONG) {
896                 struct ls_widths *w = &opts->widths;
897                 if (opts->flags & LS_FLAG_ADMISSIBLE_ONLY) {
898                         ret = para_printf(b, "%*li ",
899                                 opts->widths.score_width, d->score);
900                         if (ret < 0)
901                                 goto out;
902                 }
903                 ret = para_printf(b,
904                         "%s "   /* attributes */
905                         "%*u "  /* amp */
906                         "%*d "  /* image_id  */
907                         "%*d "  /* lyrics_id */
908                         "%*d "  /* bitrate */
909                         "%*s "  /* audio format */
910                         "%*d "  /* frequency */
911                         "%d "   /* channels */
912                         "%s "   /* duration */
913                         "%*d "  /* num_played */
914                         "%s "   /* last_played */
915                         "%s\n", /* path */
916                         att_buf,
917                         w->amp_width, afsi->amp,
918                         w->image_id_width, afsi->image_id,
919                         w->lyrics_id_width, afsi->lyrics_id,
920                         w->bitrate_width, afhi->bitrate,
921                         w->audio_format_width,
922                         audio_format_name(afsi->audio_format_id),
923                         w->frequency_width, afhi->frequency,
924                         afhi->channels,
925                         duration_buf,
926                         w->num_played_width, afsi->num_played,
927                         last_played_time,
928                         d->path
929                 );
930                 goto out;
931         }
932         if (opts->mode == LS_MODE_MBOX) {
933                 const char *bn = para_basename(d->path);
934                 ret = para_printf(b,
935                         "From foo@localhost %s\n"
936                         "Received: from\nTo: bar\nFrom: a\n"
937                         "Subject: %s\n\n",
938                         last_played_time,
939                         bn? bn : "?");
940                 if (ret < 0)
941                         goto out;
942         }
943         ret = write_filename_items(b, d->path, opts->flags);
944         if (ret < 0)
945                 goto out;
946         ret = write_score(b, d, opts);
947         if (ret < 0)
948                 goto out;
949         ret = write_attribute_items(b, att_buf, afsi);
950         if (ret < 0)
951                 goto out;
952         ret = write_image_items(b, afsi);
953         if (ret < 0)
954                 goto out;
955         ret = write_lyrics_items(b, afsi);
956         if (ret < 0)
957                 goto out;
958         hash_to_asc(d->hash, asc_hash);
959         ret = WRITE_STATUS_ITEM(b, SI_HASH, "%s\n", asc_hash);
960         if (ret < 0)
961                 goto out;
962         ret = WRITE_STATUS_ITEM(b, SI_BITRATE, "%dkbit/s\n", afhi->bitrate);
963         if (ret < 0)
964                 goto out;
965         ret = WRITE_STATUS_ITEM(b, SI_FORMAT, "%s\n",
966                 audio_format_name(afsi->audio_format_id));
967         if (ret < 0)
968                 goto out;
969         ret = WRITE_STATUS_ITEM(b, SI_FREQUENCY, "%dHz\n", afhi->frequency);
970         if (ret < 0)
971                 goto out;
972         ret = WRITE_STATUS_ITEM(b, SI_CHANNELS, "%d\n", afhi->channels);
973         if (ret < 0)
974                 goto out;
975         ret = WRITE_STATUS_ITEM(b, SI_DURATION, "%s\n", duration_buf);
976         if (ret < 0)
977                 goto out;
978         ret = WRITE_STATUS_ITEM(b, SI_SECONDS_TOTAL, "%lu\n",
979                 afhi->seconds_total);
980         if (ret < 0)
981                 goto out;
982         ret = WRITE_STATUS_ITEM(b, SI_LAST_PLAYED, "%s\n", last_played_time);
983         if (ret < 0)
984                 goto out;
985         ret = WRITE_STATUS_ITEM(b, SI_NUM_PLAYED, "%d\n", afsi->num_played);
986         if (ret < 0)
987                 goto out;
988         ret = WRITE_STATUS_ITEM(b, SI_AMPLIFICATION, "%u\n", afsi->amp);
989         if (ret < 0)
990                 goto out;
991         ret = WRITE_STATUS_ITEM(b, SI_CHUNK_TIME, "%lu\n",
992                 tv2ms(&afhi->chunk_tv));
993         if (ret < 0)
994                 goto out;
995         ret = WRITE_STATUS_ITEM(b, SI_NUM_CHUNKS, "%lu\n",
996                 afhi->chunks_total);
997         if (ret < 0)
998                 goto out;
999         ret = WRITE_STATUS_ITEM(b, SI_TECHINFO, "%s\n", afhi->techinfo);
1000         if (ret < 0)
1001                 goto out;
1002         ret = WRITE_STATUS_ITEM(b, SI_ARTIST, "%s\n", afhi->tags.artist);
1003         if (ret < 0)
1004                 goto out;
1005         ret = WRITE_STATUS_ITEM(b, SI_TITLE, "%s\n", afhi->tags.title);
1006         if (ret < 0)
1007                 goto out;
1008         ret = WRITE_STATUS_ITEM(b, SI_YEAR, "%s\n", afhi->tags.year);
1009         if (ret < 0)
1010                 goto out;
1011         ret = WRITE_STATUS_ITEM(b, SI_ALBUM, "%s\n", afhi->tags.album);
1012         if (ret < 0)
1013                 goto out;
1014         ret = WRITE_STATUS_ITEM(b, SI_COMMENT, "%s\n", afhi->tags.comment);
1015         if (ret < 0)
1016                 goto out;
1017         if (opts->mode == LS_MODE_MBOX) {
1018                 struct osl_object lyrics_def;
1019                 lyr_get_def_by_id(afsi->lyrics_id, &lyrics_def);
1020                 if (lyrics_def.data) {
1021                         ret = para_printf(b, "Lyrics:\n~~~~~~~\n%s",
1022                                 (char *)lyrics_def.data);
1023                         osl_close_disk_object(&lyrics_def);
1024                 }
1025         }
1026 out:
1027         return ret;
1028 }
1029
1030 static int make_status_items(struct audio_file_data *afd,
1031                 struct afs_info *afsi, char *path, long score,
1032                 unsigned char *hash)
1033 {
1034         struct ls_data d = {
1035                 .afhi = afd->afhi,
1036                 .afsi = *afsi,
1037                 .path = path,
1038                 .score = score,
1039                 .hash = hash
1040         };
1041         struct ls_options opts = {
1042                 .flags = LS_FLAG_FULL_PATH | LS_FLAG_ADMISSIBLE_ONLY,
1043                 .mode = LS_MODE_VERBOSE,
1044         };
1045         struct para_buffer pb = {.max_size = shm_get_shmmax() - 1};
1046         time_t current_time;
1047         int ret;
1048
1049         time(&current_time);
1050         ret = print_list_item(&d, &opts, &pb, current_time);
1051         if (ret < 0)
1052                 return ret;
1053         free(status_items);
1054         status_items = pb.buf;
1055         memset(&pb, 0, sizeof(pb));
1056         pb.max_size = shm_get_shmmax() - 1;
1057         pb.flags = PBF_SIZE_PREFIX;
1058         ret = print_list_item(&d, &opts, &pb, current_time);
1059         if (ret < 0) {
1060                 free(status_items);
1061                 status_items = NULL;
1062                 return ret;
1063         }
1064         free(parser_friendly_status_items);
1065         parser_friendly_status_items = pb.buf;
1066         return 1;
1067 }
1068
1069 /**
1070  * Mmap the given audio file and update statistics.
1071  *
1072  * \param aft_row Determines the audio file to be opened and updated.
1073  * \param score The score of the audio file.
1074  * \param afd Result pointer.
1075  *
1076  * On success, the numplayed field of the audio file selector info is increased
1077  * and the lastplayed time is set to the current time. Finally, the score of
1078  * the audio file is updated.
1079  *
1080  * \return Positive shmid on success, negative on errors.
1081  */
1082 int open_and_update_audio_file(struct osl_row *aft_row, long score,
1083         struct audio_file_data *afd)
1084 {
1085         unsigned char *aft_hash, file_hash[HASH_SIZE];
1086         struct osl_object afsi_obj;
1087         struct afs_info old_afsi, new_afsi;
1088         int ret = get_hash_of_row(aft_row, &aft_hash);
1089         struct afsi_change_event_data aced;
1090         struct osl_object map, chunk_table_obj;
1091         char *path;
1092
1093         if (ret < 0)
1094                 return ret;
1095         ret = get_audio_file_path_of_row(aft_row, &path);
1096         if (ret < 0)
1097                 return ret;
1098         PARA_NOTICE_LOG("%s\n", path);
1099         ret = get_afsi_object_of_row(aft_row, &afsi_obj);
1100         if (ret < 0)
1101                 return ret;
1102         ret = load_afsi(&old_afsi, &afsi_obj);
1103         if (ret < 0)
1104                 return ret;
1105         ret = get_afhi_of_row(aft_row, &afd->afhi);
1106         if (ret < 0)
1107                 return ret;
1108         afd->afhi.chunk_table = NULL;
1109         ret = osl_open_disk_object(audio_file_table, aft_row,
1110                 AFTCOL_CHUNKS, &chunk_table_obj);
1111         if (ret < 0)
1112                 goto err;
1113         ret = mmap_full_file(path, O_RDONLY, &map.data,
1114                 &map.size, &afd->fd);
1115         if (ret < 0)
1116                 goto err;
1117         hash_function(map.data, map.size, file_hash);
1118         ret = hash_compare(file_hash, aft_hash);
1119         para_munmap(map.data, map.size);
1120         if (ret) {
1121                 ret = -E_HASH_MISMATCH;
1122                 goto err;
1123         }
1124         new_afsi = old_afsi;
1125         new_afsi.num_played++;
1126         new_afsi.last_played = time(NULL);
1127         save_afsi(&new_afsi, &afsi_obj); /* in-place update */
1128
1129         afd->audio_format_id = old_afsi.audio_format_id;
1130         load_chunk_table(&afd->afhi, chunk_table_obj.data);
1131         ret = make_status_items(afd, &old_afsi, path, score, file_hash);
1132         if (ret < 0)
1133                 goto err;
1134         aced.aft_row = aft_row;
1135         aced.old_afsi = &old_afsi;
1136         afs_event(AFSI_CHANGE, NULL, &aced);
1137         ret = save_afd(afd);
1138 err:
1139         free(afd->afhi.chunk_table);
1140         osl_close_disk_object(&chunk_table_obj);
1141         if (ret < 0)
1142                 PARA_ERROR_LOG("%s: %s\n", path, para_strerror(-ret));
1143         return ret;
1144 }
1145
1146 static int ls_audio_format_compare(const void *a, const void *b)
1147 {
1148         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1149         return NUM_COMPARE(d1->afsi.audio_format_id, d2->afsi.audio_format_id);
1150 }
1151
1152 static int ls_duration_compare(const void *a, const void *b)
1153 {
1154         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1155         return NUM_COMPARE(d1->afhi.seconds_total, d2->afhi.seconds_total);
1156 }
1157
1158 static int ls_bitrate_compare(const void *a, const void *b)
1159 {
1160         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1161         return NUM_COMPARE(d1->afhi.bitrate, d2->afhi.bitrate);
1162 }
1163
1164 static int ls_lyrics_id_compare(const void *a, const void *b)
1165 {
1166         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1167         return NUM_COMPARE(d1->afsi.lyrics_id, d2->afsi.lyrics_id);
1168 }
1169
1170 static int ls_image_id_compare(const void *a, const void *b)
1171 {
1172         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1173         return NUM_COMPARE(d1->afsi.image_id, d2->afsi.image_id);
1174 }
1175
1176 static int ls_channels_compare(const void *a, const void *b)
1177 {
1178         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1179         return NUM_COMPARE(d1->afhi.channels, d2->afhi.channels);
1180 }
1181
1182 static int ls_frequency_compare(const void *a, const void *b)
1183 {
1184         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1185         return NUM_COMPARE(d1->afhi.frequency, d2->afhi.frequency);
1186 }
1187
1188 static int ls_num_played_compare(const void *a, const void *b)
1189 {
1190         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1191         return NUM_COMPARE(d1->afsi.num_played, d2->afsi.num_played);
1192 }
1193
1194 static int ls_last_played_compare(const void *a, const void *b)
1195 {
1196         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1197         return NUM_COMPARE(d1->afsi.last_played, d2->afsi.last_played);
1198 }
1199
1200 static int ls_score_compare(const void *a, const void *b)
1201 {
1202         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1203         return NUM_COMPARE(d1->score, d2->score);
1204 }
1205
1206 static int ls_path_compare(const void *a, const void *b)
1207 {
1208         struct ls_data *d1 = *(struct ls_data **)a, *d2 = *(struct ls_data **)b;
1209         return strcmp(d1->path, d2->path);
1210 }
1211
1212 static int sort_matching_paths(struct ls_options *options)
1213 {
1214         size_t nmemb = options->num_matching_paths;
1215         size_t size = sizeof(*options->data_ptr);
1216         int (*compar)(const void *, const void *);
1217         int i;
1218
1219         options->data_ptr = para_malloc(nmemb * sizeof(*options->data_ptr));
1220         for (i = 0; i < nmemb; i++)
1221                 options->data_ptr[i] = options->data + i;
1222
1223         /* In these cases the array is already sorted */
1224         if (options->sorting == LS_SORT_BY_PATH
1225                 && !(options->flags & LS_FLAG_ADMISSIBLE_ONLY)
1226                 && (options->flags & LS_FLAG_FULL_PATH))
1227                 return 1;
1228         if (options->sorting == LS_SORT_BY_SCORE &&
1229                         options->flags & LS_FLAG_ADMISSIBLE_ONLY)
1230                 return 1;
1231
1232         switch (options->sorting) {
1233         case LS_SORT_BY_PATH:
1234                 compar = ls_path_compare; break;
1235         case LS_SORT_BY_SCORE:
1236                 compar = ls_score_compare; break;
1237         case LS_SORT_BY_LAST_PLAYED:
1238                 compar = ls_last_played_compare; break;
1239         case LS_SORT_BY_NUM_PLAYED:
1240                 compar = ls_num_played_compare; break;
1241         case LS_SORT_BY_FREQUENCY:
1242                 compar = ls_frequency_compare; break;
1243         case LS_SORT_BY_CHANNELS:
1244                 compar = ls_channels_compare; break;
1245         case LS_SORT_BY_IMAGE_ID:
1246                 compar = ls_image_id_compare; break;
1247         case LS_SORT_BY_LYRICS_ID:
1248                 compar = ls_lyrics_id_compare; break;
1249         case LS_SORT_BY_BITRATE:
1250                 compar = ls_bitrate_compare; break;
1251         case LS_SORT_BY_DURATION:
1252                 compar = ls_duration_compare; break;
1253         case LS_SORT_BY_AUDIO_FORMAT:
1254                 compar = ls_audio_format_compare; break;
1255         default:
1256                 return -E_BAD_SORT;
1257         }
1258         qsort(options->data_ptr, nmemb, size, compar);
1259         return 1;
1260 }
1261
1262 /* row is either an aft_row or a row of the score table */
1263 /* TODO: Only compute widths if we need them */
1264 static int prepare_ls_row(struct osl_row *row, void *ls_opts)
1265 {
1266         int ret, i;
1267         struct ls_options *options = ls_opts;
1268         struct ls_data *d;
1269         struct ls_widths *w;
1270         unsigned short num_digits;
1271         unsigned tmp;
1272         struct osl_row *aft_row;
1273         long score;
1274         char *path;
1275
1276         if (options->flags & LS_FLAG_ADMISSIBLE_ONLY) {
1277                 ret = get_score_and_aft_row(row, &score, &aft_row);
1278                 if (ret < 0)
1279                         return ret;
1280         } else {
1281                 aft_row = row;
1282                 score = 0;
1283         }
1284         ret = get_audio_file_path_of_row(aft_row, &path);
1285         if (ret < 0)
1286                 return ret;
1287         if (!(options->flags & LS_FLAG_FULL_PATH)) {
1288                 char *p = strrchr(path, '/');
1289                 if (p)
1290                         path = p + 1;
1291         }
1292         if (options->num_patterns) {
1293                 for (i = 0; i < options->num_patterns; i++) {
1294                         ret = fnmatch(options->patterns[i], path, 0);
1295                         if (!ret)
1296                                 break;
1297                         if (ret == FNM_NOMATCH)
1298                                 continue;
1299                         return -E_FNMATCH;
1300                 }
1301                 if (i >= options->num_patterns) /* no match */
1302                         return 1;
1303         }
1304         tmp = options->num_matching_paths++;
1305         if (options->num_matching_paths > options->array_size) {
1306                 options->array_size++;
1307                 options->array_size *= 2;
1308                 options->data = para_realloc(options->data, options->array_size
1309                         * sizeof(*options->data));
1310         }
1311         d = options->data + tmp;
1312         ret = get_afsi_of_row(aft_row, &d->afsi);
1313         if (ret < 0)
1314                 return ret;
1315         ret = get_afhi_of_row(aft_row, &d->afhi);
1316         if (ret < 0)
1317                 return ret;
1318         d->path = path;
1319         ret = get_hash_of_row(aft_row, &d->hash);
1320         if (ret < 0)
1321                 goto err;
1322         w = &options->widths;
1323         GET_NUM_DIGITS(d->afsi.image_id, &num_digits);
1324         w->image_id_width = PARA_MAX(w->image_id_width, num_digits);
1325         GET_NUM_DIGITS(d->afsi.lyrics_id, &num_digits);
1326         w->lyrics_id_width = PARA_MAX(w->lyrics_id_width, num_digits);
1327         GET_NUM_DIGITS(d->afhi.bitrate, &num_digits);
1328         w->bitrate_width = PARA_MAX(w->bitrate_width, num_digits);
1329         GET_NUM_DIGITS(d->afhi.frequency, &num_digits);
1330         w->frequency_width = PARA_MAX(w->frequency_width, num_digits);
1331         GET_NUM_DIGITS(d->afsi.num_played, &num_digits);
1332         w->num_played_width = PARA_MAX(w->num_played_width, num_digits);
1333         /* get the number of chars to print this amount of time */
1334         num_digits = get_duration_width(d->afhi.seconds_total);
1335         w->duration_width = PARA_MAX(w->duration_width, num_digits);
1336         GET_NUM_DIGITS(d->afsi.amp, &num_digits);
1337         w->amp_width = PARA_MAX(w->amp_width, num_digits);
1338         num_digits = strlen(audio_format_name(d->afsi.audio_format_id));
1339         w->audio_format_width = PARA_MAX(w->audio_format_width, num_digits);
1340         if (options->flags & LS_FLAG_ADMISSIBLE_ONLY) {
1341                 GET_NUM_DIGITS(score, &num_digits);
1342                 num_digits++; /* add one for the sign (space or "-") */
1343                 w->score_width = PARA_MAX(w->score_width, num_digits);
1344                 d->score = score;
1345         }
1346         return 1;
1347 err:
1348         return ret;
1349 }
1350
1351 static void com_ls_callback(int fd, const struct osl_object *query)
1352 {
1353         struct ls_options *opts = query->data;
1354         char *p, *pattern_start = (char *)query->data + sizeof(*opts);
1355         struct para_buffer b = {
1356                 .max_size = shm_get_shmmax(),
1357                 .flags = (opts->mode == LS_MODE_PARSER)? PBF_SIZE_PREFIX : 0,
1358                 .max_size_handler = afs_max_size_handler,
1359                 .private_data = &(struct afs_max_size_handler_data) {
1360                         .fd = fd,
1361                         .band = SBD_OUTPUT
1362                 }
1363         };
1364         int i = 0, ret;
1365         time_t current_time;
1366
1367         if (opts->num_patterns) {
1368                 opts->patterns = para_malloc(opts->num_patterns * sizeof(char *));
1369                 for (i = 0, p = pattern_start; i < opts->num_patterns; i++) {
1370                         opts->patterns[i] = p;
1371                         p += strlen(p) + 1;
1372                 }
1373         } else
1374                 opts->patterns = NULL;
1375         if (opts->flags & LS_FLAG_ADMISSIBLE_ONLY)
1376                 ret = admissible_file_loop(opts, prepare_ls_row);
1377         else
1378                 ret = osl(osl_rbtree_loop(audio_file_table, AFTCOL_PATH, opts,
1379                         prepare_ls_row));
1380         if (ret < 0)
1381                 goto out;
1382         if (opts->num_matching_paths == 0) {
1383                 if (opts->num_patterns > 0)
1384                         para_printf(&b, "no matches\n");
1385                 goto out;
1386         }
1387         ret = sort_matching_paths(opts);
1388         if (ret < 0)
1389                 goto out;
1390         time(&current_time);
1391         if (opts->flags & LS_FLAG_REVERSE)
1392                 for (i = opts->num_matching_paths - 1; i >= 0; i--) {
1393                         ret = print_list_item(opts->data_ptr[i], opts, &b, current_time);
1394                         if (ret < 0)
1395                                 goto out;
1396                 }
1397         else
1398                 for (i = 0; i < opts->num_matching_paths; i++) {
1399                         ret = print_list_item(opts->data_ptr[i], opts, &b, current_time);
1400                         if (ret < 0)
1401                                 goto out;
1402                 }
1403 out:
1404         if (b.offset)
1405                 pass_buffer_as_shm(fd, SBD_OUTPUT, b.buf, b.offset);
1406         free(b.buf);
1407         free(opts->data);
1408         free(opts->data_ptr);
1409         free(opts->patterns);
1410 }
1411
1412 /*
1413  * TODO: flags -h (sort by hash)
1414  */
1415 int com_ls(struct command_context *cc)
1416 {
1417         int i, ret;
1418         unsigned flags = 0;
1419         enum ls_sorting_method sort = LS_SORT_BY_PATH;
1420         enum ls_listing_mode mode = LS_MODE_SHORT;
1421         struct ls_options opts = {.patterns = NULL};
1422         struct osl_object query = {.data = &opts, .size = sizeof(opts)};
1423
1424         for (i = 1; i < cc->argc; i++) {
1425                 const char *arg = cc->argv[i];
1426                 if (arg[0] != '-')
1427                         break;
1428                 if (!strcmp(arg, "--")) {
1429                         i++;
1430                         break;
1431                 }
1432                 if (!strncmp(arg, "-l", 2)) {
1433                         if (!*(arg + 2)) {
1434                                 mode = LS_MODE_LONG;
1435                                 continue;
1436                         }
1437                         if (*(arg + 3))
1438                                 return -E_AFT_SYNTAX;
1439                         switch(*(arg + 2)) {
1440                         case 's':
1441                                 mode = LS_MODE_SHORT;
1442                                 continue;
1443                         case 'l':
1444                                 mode = LS_MODE_LONG;
1445                                 continue;
1446                         case 'v':
1447                                 mode = LS_MODE_VERBOSE;
1448                                 continue;
1449                         case 'm':
1450                                 mode = LS_MODE_MBOX;
1451                                 continue;
1452                         case 'c':
1453                                 mode = LS_MODE_CHUNKS;
1454                                 continue;
1455                         case 'p':
1456                                 mode = LS_MODE_PARSER;
1457                                 continue;
1458                         default:
1459                                 return -E_AFT_SYNTAX;
1460                         }
1461                 }
1462                 if (!strcmp(arg, "-p")) {
1463                         flags |= LS_FLAG_FULL_PATH;
1464                         continue;
1465                 }
1466                 if (!strcmp(arg, "-a")) {
1467                         flags |= LS_FLAG_ADMISSIBLE_ONLY;
1468                         continue;
1469                 }
1470                 if (!strcmp(arg, "-r")) {
1471                         flags |= LS_FLAG_REVERSE;
1472                         continue;
1473                 }
1474                 if (!strcmp(arg, "-d")) {
1475                         flags |= LS_FLAG_UNIXDATE;
1476                         continue;
1477                 }
1478                 if (!strncmp(arg, "-s", 2)) {
1479                         if (!*(arg + 2) || *(arg + 3))
1480                                 return -E_AFT_SYNTAX;
1481                         switch(*(arg + 2)) {
1482                         case 'p':
1483                                 sort = LS_SORT_BY_PATH;
1484                                 continue;
1485                         case 's': /* -ss implies -a */
1486                                 sort = LS_SORT_BY_SCORE;
1487                                 flags |= LS_FLAG_ADMISSIBLE_ONLY;
1488                                 continue;
1489                         case 'l':
1490                                 sort = LS_SORT_BY_LAST_PLAYED;
1491                                 continue;
1492                         case 'n':
1493                                 sort = LS_SORT_BY_NUM_PLAYED;
1494                                 continue;
1495                         case 'f':
1496                                 sort = LS_SORT_BY_FREQUENCY;
1497                                 continue;
1498                         case 'c':
1499                                 sort = LS_SORT_BY_CHANNELS;
1500                                 continue;
1501                         case 'i':
1502                                 sort = LS_SORT_BY_IMAGE_ID;
1503                                 continue;
1504                         case 'y':
1505                                 sort = LS_SORT_BY_LYRICS_ID;
1506                                 continue;
1507                         case 'b':
1508                                 sort = LS_SORT_BY_BITRATE;
1509                                 continue;
1510                         case 'd':
1511                                 sort = LS_SORT_BY_DURATION;
1512                                 continue;
1513                         case 'a':
1514                                 sort = LS_SORT_BY_AUDIO_FORMAT;
1515                                 continue;
1516                         default:
1517                                 return -E_AFT_SYNTAX;
1518                         }
1519                 }
1520                 return -E_AFT_SYNTAX;
1521         }
1522         opts.flags = flags;
1523         opts.sorting = sort;
1524         opts.mode = mode;
1525         opts.num_patterns = cc->argc - i;
1526         ret = send_option_arg_callback_request(&query, opts.num_patterns,
1527                 cc->argv + i, com_ls_callback, afs_cb_result_handler, cc);
1528         return ret;
1529 }
1530
1531 /**
1532  * Call the given function for each file in the audio file table.
1533  *
1534  * \param private_data An arbitrary data pointer, passed to \a func.
1535  * \param func The custom function to be called.
1536  *
1537  * \return Standard.
1538  */
1539 int audio_file_loop(void *private_data, osl_rbtree_loop_func *func)
1540 {
1541         return osl(osl_rbtree_loop(audio_file_table, AFTCOL_HASH, private_data,
1542                 func));
1543 }
1544
1545 static struct osl_row *find_hash_sister(unsigned char *hash)
1546 {
1547         const struct osl_object obj = {.data = hash, .size = HASH_SIZE};
1548         struct osl_row *row;
1549
1550         osl_get_row(audio_file_table, AFTCOL_HASH, &obj, &row);
1551         return row;
1552 }
1553
1554 /** The format of the data stored by save_audio_file_data(). */
1555 enum com_add_buffer_offsets {
1556         /** afhi (if present) starts here. */
1557         CAB_AFHI_OFFSET_POS = 0,
1558         /** Start of the chunk table (if present). */
1559         CAB_CHUNKS_OFFSET_POS = 2,
1560         /** Audio format id. */
1561         CAB_AUDIO_FORMAT_OFFSET = 4,
1562         /** Flags given to the add command. */
1563         CAB_FLAGS_OFFSET = 5,
1564         /** The hash of the audio file being added. */
1565         CAB_HASH_OFFSET = 9,
1566         /** Start of the path of the audio file. */
1567         CAB_PATH_OFFSET = (CAB_HASH_OFFSET + HASH_SIZE),
1568 };
1569
1570 /*
1571  * Store the given data to a single buffer. Doesn't need the audio file selector
1572  * info struct as the server knows it as well.
1573  *
1574  * It's OK to call this with afhi == NULL. In this case, the audio format
1575  * handler info won't be stored in the buffer.
1576  */
1577 static void save_add_callback_buffer(unsigned char *hash, const char *path,
1578                 struct afh_info *afhi, uint32_t flags,
1579                 uint8_t audio_format_num, struct osl_object *obj)
1580 {
1581         size_t path_len = strlen(path) + 1;
1582         size_t afhi_size = sizeof_afhi_buf(afhi);
1583         size_t size = CAB_PATH_OFFSET + path_len + afhi_size
1584                 + sizeof_chunk_table(afhi);
1585         char *buf = para_malloc(size);
1586         uint16_t pos;
1587
1588         write_u8(buf + CAB_AUDIO_FORMAT_OFFSET, audio_format_num);
1589         write_u32(buf + CAB_FLAGS_OFFSET, flags);
1590
1591         memcpy(buf + CAB_HASH_OFFSET, hash, HASH_SIZE);
1592         strcpy(buf + CAB_PATH_OFFSET, path);
1593
1594         pos = CAB_PATH_OFFSET + path_len;
1595         PARA_DEBUG_LOG("size: %zu, afhi starts at %d\n", size, pos);
1596         PARA_DEBUG_LOG("last afhi byte: %p, pos %zu\n", buf + pos + afhi_size - 1,
1597                 pos + afhi_size - 1);
1598         write_u16(buf + CAB_AFHI_OFFSET_POS, pos);
1599         save_afhi(afhi, buf + pos);
1600
1601         pos += afhi_size;
1602         PARA_DEBUG_LOG("size: %zu, chunks start at %d\n", size, pos);
1603         write_u16(buf + CAB_CHUNKS_OFFSET_POS, pos);
1604         if (afhi)
1605                 save_chunk_table(afhi, buf + pos);
1606         PARA_DEBUG_LOG("last byte in buf: %p\n", buf + size - 1);
1607         obj->data = buf;
1608         obj->size = size;
1609 }
1610
1611 /*
1612
1613 Overview of the add command.
1614
1615 Input: What was passed to the callback by the command handler.
1616 ~~~~~~
1617 HS:     Hash sister. Whether an audio file with identical hash
1618         already exists in the osl database.
1619
1620 PB:     Path brother. Whether a file with the given path exists
1621         in the table.
1622
1623 F:      Force flag given. Whether add was called with -f.
1624
1625 output: Action performed by the callback.
1626 ~~~~~~~
1627 AFHI:   Whether afhi and chunk table are computed and sent.
1628 ACTION: Table modifications to be done by the callback.
1629
1630 +----+----+---+------+---------------------------------------------------+
1631 | HS | PB | F | AFHI | ACTION
1632 +----+----+---+------+---------------------------------------------------+
1633 | Y  |  Y | Y |  Y   | if HS != PB: remove PB. HS: force afhi update,
1634 |                    | update path, keep afsi
1635 +----+----+---+------+---------------------------------------------------+
1636 | Y  |  Y | N |  N   | if HS == PB: do not send callback request at all.
1637 |                    | otherwise: remove PB, HS: update path, keep afhi,
1638 |                    | afsi.
1639 +----+----+---+------+---------------------------------------------------+
1640 | Y  |  N | Y |  Y   | (rename) force afhi update of HS, update path of
1641 |                    | HS, keep afsi
1642 +----+----+---+------+---------------------------------------------------+
1643 | Y  |  N | N |  N   | (file rename) update path of HS, keep afsi, afhi
1644 +----+----+---+------+---------------------------------------------------+
1645 | N  |  Y | Y |  Y   | (file change) update afhi, hash, of PB, keep afsi
1646 |                    | (force has no effect)
1647 +----+----+---+------+---------------------------------------------------+
1648 | N  |  Y | N |  Y   | (file change) update afhi, hash of PB, keep afsi
1649 +----+----+---+------+---------------------------------------------------+
1650 | N  |  N | Y |  Y   | (new file) create new entry (force has no effect)
1651 +----+----+---+------+---------------------------------------------------+
1652 |  N |  N | N |  Y   | (new file) create new entry
1653 +----+----+---+------+---------------------------------------------------+
1654
1655 Notes:
1656
1657         afhi <=> force or no HS
1658         F => AFHI
1659
1660 */
1661
1662 /** Flags passed to the add command. */
1663 enum com_add_flags {
1664         /** Skip paths that exist already. */
1665         ADD_FLAG_LAZY = 1,
1666         /** Force adding. */
1667         ADD_FLAG_FORCE = 2,
1668         /** Print what is being done. */
1669         ADD_FLAG_VERBOSE = 4,
1670         /** Try to add files with unknown suffixes. */
1671         ADD_FLAG_ALL = 8,
1672 };
1673
1674 static void com_add_callback(int fd, const struct osl_object *query)
1675 {
1676         char *buf = query->data, *path;
1677         struct osl_row *pb, *aft_row;
1678         struct osl_row *hs;
1679         struct osl_object objs[NUM_AFT_COLUMNS];
1680         unsigned char *hash;
1681         char asc[2 * HASH_SIZE + 1];
1682         int ret;
1683         char afsi_buf[AFSI_SIZE];
1684         uint32_t flags = read_u32(buf + CAB_FLAGS_OFFSET);
1685         struct afs_info default_afsi = {.last_played = 0};
1686         struct para_buffer msg = {
1687                 .max_size = shm_get_shmmax(),
1688                 .max_size_handler = afs_max_size_handler,
1689                 .private_data = &(struct afs_max_size_handler_data) {
1690                         .fd = fd,
1691                         .band = SBD_OUTPUT
1692                 }
1693         };
1694         uint16_t afhi_offset, chunks_offset;
1695
1696         hash = (unsigned char *)buf + CAB_HASH_OFFSET;
1697         hash_to_asc(hash, asc);
1698         objs[AFTCOL_HASH].data = buf + CAB_HASH_OFFSET;
1699         objs[AFTCOL_HASH].size = HASH_SIZE;
1700
1701         path = buf + CAB_PATH_OFFSET;
1702         objs[AFTCOL_PATH].data = path;
1703         objs[AFTCOL_PATH].size = strlen(path) + 1;
1704
1705         PARA_INFO_LOG("request to add %s\n", path);
1706         hs = find_hash_sister(hash);
1707         ret = aft_get_row_of_path(path, &pb);
1708         if (ret < 0 && ret != -OSL_ERRNO_TO_PARA_ERROR(E_OSL_RB_KEY_NOT_FOUND))
1709                 goto out;
1710         if (hs && pb && hs == pb && !(flags & ADD_FLAG_FORCE)) {
1711                 if (flags & ADD_FLAG_VERBOSE)
1712                         ret = para_printf(&msg, "ignoring duplicate\n");
1713                 else
1714                         ret = 1;
1715                 goto out;
1716         }
1717         if (hs && hs != pb) {
1718                 struct osl_object obj;
1719                 if (pb) { /* hs trumps pb, remove pb */
1720                         if (flags & ADD_FLAG_VERBOSE) {
1721                                 ret = para_printf(&msg, "removing path brother\n");
1722                                 if (ret < 0)
1723                                         goto out;
1724                         }
1725                         ret = osl(osl_del_row(audio_file_table, pb));
1726                         if (ret < 0)
1727                                 goto out;
1728                         pb = NULL;
1729                 }
1730                 /* file rename, update hs' path */
1731                 if (flags & ADD_FLAG_VERBOSE) {
1732                         ret = osl(osl_get_object(audio_file_table, hs,
1733                                 AFTCOL_PATH, &obj));
1734                         if (ret < 0)
1735                                 goto out;
1736                         ret = para_printf(&msg, "renamed from %s\n", (char *)obj.data);
1737                         if (ret < 0)
1738                                 goto out;
1739                 }
1740                 ret = osl(osl_update_object(audio_file_table, hs, AFTCOL_PATH,
1741                         &objs[AFTCOL_PATH]));
1742                 if (ret < 0)
1743                         goto out;
1744                 afs_event(AUDIO_FILE_RENAME, &msg, hs);
1745                 if (!(flags & ADD_FLAG_FORCE))
1746                         goto out;
1747         }
1748         /* no hs or force mode, child must have sent afhi */
1749         afhi_offset = read_u16(buf + CAB_AFHI_OFFSET_POS);
1750         chunks_offset = read_u16(buf + CAB_CHUNKS_OFFSET_POS);
1751
1752         objs[AFTCOL_AFHI].data = buf + afhi_offset;
1753         objs[AFTCOL_AFHI].size = chunks_offset - afhi_offset;
1754         ret = -E_NO_AFHI;
1755         if (!objs[AFTCOL_AFHI].size) /* "impossible" */
1756                 goto out;
1757         objs[AFTCOL_CHUNKS].data = buf + chunks_offset;
1758         objs[AFTCOL_CHUNKS].size = query->size - chunks_offset;
1759         if (pb && !hs) { /* update pb's hash */
1760                 char old_asc[2 * HASH_SIZE + 1];
1761                 unsigned char *old_hash;
1762                 ret = get_hash_of_row(pb, &old_hash);
1763                 if (ret < 0)
1764                         goto out;
1765                 hash_to_asc(old_hash, old_asc);
1766                 if (flags & ADD_FLAG_VERBOSE) {
1767                         ret = para_printf(&msg, "file change: %s -> %s\n",
1768                                 old_asc, asc);
1769                         if (ret < 0)
1770                                 goto out;
1771                 }
1772                 ret = osl_update_object(audio_file_table, pb, AFTCOL_HASH,
1773                         &objs[AFTCOL_HASH]);
1774                 if (ret < 0)
1775                         goto out;
1776         }
1777         if (hs || pb) { /* (hs != NULL and pb != NULL) implies hs == pb */
1778                 struct osl_row *row = pb? pb : hs;
1779                 /* update afhi and chunk_table */
1780                 if (flags & ADD_FLAG_VERBOSE) {
1781                         ret = para_printf(&msg, "updating afhi and chunk table\n");
1782                         if (ret < 0)
1783                                 goto out;
1784                 }
1785                 ret = osl(osl_update_object(audio_file_table, row, AFTCOL_AFHI,
1786                         &objs[AFTCOL_AFHI]));
1787                 if (ret < 0)
1788                         goto out;
1789                 ret = osl(osl_update_object(audio_file_table, row, AFTCOL_CHUNKS,
1790                         &objs[AFTCOL_CHUNKS]));
1791                 if (ret < 0)
1792                         goto out;
1793                 afs_event(AFHI_CHANGE, &msg, row);
1794                 goto out;
1795         }
1796         /* new entry, use default afsi */
1797         if (flags & ADD_FLAG_VERBOSE) {
1798                 ret = para_printf(&msg, "new file\n");
1799                 if (ret < 0)
1800                         goto out;
1801         }
1802         default_afsi.last_played = time(NULL) - 365 * 24 * 60 * 60;
1803         default_afsi.audio_format_id = read_u8(buf + CAB_AUDIO_FORMAT_OFFSET);
1804
1805         objs[AFTCOL_AFSI].data = &afsi_buf;
1806         objs[AFTCOL_AFSI].size = AFSI_SIZE;
1807         save_afsi(&default_afsi, &objs[AFTCOL_AFSI]);
1808         ret = osl(osl_add_and_get_row(audio_file_table, objs, &aft_row));
1809         afs_event(AUDIO_FILE_ADD, &msg, aft_row);
1810 out:
1811         if (ret < 0)
1812                 para_printf(&msg, "%s\n", para_strerror(-ret));
1813         if (msg.offset)
1814                 pass_buffer_as_shm(fd, SBD_OUTPUT, msg.buf, msg.offset);
1815         free(msg.buf);
1816 }
1817
1818 /** Used by com_add(). */
1819 struct private_add_data {
1820         /** The pointer passed to the original command handler. */
1821         struct command_context *cc;
1822         /** The given add flags. */
1823         uint32_t flags;
1824 };
1825
1826 static void path_brother_callback(int fd, const struct osl_object *query)
1827 {
1828         char *path = query->data;
1829         struct osl_row *path_brother;
1830         int ret = aft_get_row_of_path(path, &path_brother);
1831         if (ret < 0)
1832                 return;
1833         pass_buffer_as_shm(fd, SBD_OUTPUT, (char *)&path_brother,
1834                 sizeof(path_brother));
1835 }
1836
1837 static void hash_sister_callback(int fd, const struct osl_object *query)
1838 {
1839         unsigned char *hash = query->data;
1840         struct osl_row *hash_sister;
1841
1842         hash_sister = find_hash_sister(hash);
1843         if (!hash_sister)
1844                 return;
1845         pass_buffer_as_shm(fd, SBD_OUTPUT, (char *)&hash_sister,
1846                 sizeof(hash_sister));
1847 }
1848
1849 static int get_row_pointer_from_result(struct osl_object *result,
1850                 __a_unused uint8_t band, void *private)
1851 {
1852         struct osl_row **row = private;
1853         *row = *(struct osl_row **)(result->data);
1854         return 1;
1855 }
1856
1857 static int add_one_audio_file(const char *path, void *private_data)
1858 {
1859         int ret, send_ret = 1, fd;
1860         uint8_t format_num = -1;
1861         struct private_add_data *pad = private_data;
1862         struct afh_info afhi, *afhi_ptr = NULL;
1863         struct osl_row *pb = NULL, *hs = NULL; /* path brother/hash sister */
1864         struct osl_object map, obj = {.data = NULL}, query;
1865         unsigned char hash[HASH_SIZE];
1866
1867         ret = guess_audio_format(path);
1868         if (ret < 0 && !(pad->flags & ADD_FLAG_ALL))
1869                 goto out_free;
1870         query.data = (char *)path;
1871         query.size = strlen(path) + 1;
1872         ret = send_callback_request(path_brother_callback, &query,
1873                 get_row_pointer_from_result, &pb);
1874         if (ret < 0 && ret != -OSL_ERRNO_TO_PARA_ERROR(E_OSL_RB_KEY_NOT_FOUND))
1875                 goto out_free;
1876         ret = 1;
1877         if (pb && (pad->flags & ADD_FLAG_LAZY)) { /* lazy is really cheap */
1878                 if (pad->flags & ADD_FLAG_VERBOSE)
1879                         send_ret = send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1880                                 "lazy-ignore: %s\n", path);
1881                 goto out_free;
1882         }
1883         /* We still want to add this file. Compute its hash. */
1884         ret = mmap_full_file(path, O_RDONLY, &map.data, &map.size, &fd);
1885         if (ret < 0)
1886                 goto out_free;
1887         hash_function(map.data, map.size, hash);
1888
1889         /* Check whether the database contains a file with the same hash. */
1890         query.data = hash;
1891         query.size = HASH_SIZE;
1892         ret = send_callback_request(hash_sister_callback, &query,
1893                 get_row_pointer_from_result, &hs);
1894         if (ret < 0)
1895                 goto out_unmap;
1896         /* Return success if we already know this file. */
1897         ret = 1;
1898         if (pb && hs && hs == pb && !(pad->flags & ADD_FLAG_FORCE)) {
1899                 if (pad->flags & ADD_FLAG_VERBOSE)
1900                         send_ret = send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1901                                 "%s exists, not forcing update\n", path);
1902                 goto out_unmap;
1903         }
1904         /*
1905          * We won't recalculate the audio format info and the chunk table if
1906          * there is a hash sister and FORCE was not given.
1907          */
1908         if (!hs || (pad->flags & ADD_FLAG_FORCE)) {
1909                 ret = compute_afhi(path, map.data, map.size, fd, &afhi);
1910                 if (ret < 0)
1911                         goto out_unmap;
1912                 format_num = ret;
1913                 afhi_ptr = &afhi;
1914         }
1915         munmap(map.data, map.size);
1916         close(fd);
1917         if (pad->flags & ADD_FLAG_VERBOSE) {
1918                 send_ret = send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1919                         "adding %s\n", path);
1920                 if (send_ret < 0)
1921                         goto out_free;
1922         }
1923         save_add_callback_buffer(hash, path, afhi_ptr, pad->flags, format_num, &obj);
1924         /* Ask afs to consider this entry for adding. */
1925         ret = send_callback_request(com_add_callback, &obj,
1926                 afs_cb_result_handler, pad->cc);
1927         goto out_free;
1928
1929 out_unmap:
1930         close(fd);
1931         munmap(map.data, map.size);
1932 out_free:
1933         if (ret < 0 && send_ret >= 0)
1934                 send_ret = send_sb_va(&pad->cc->scc, SBD_ERROR_LOG,
1935                         "failed to add %s (%s)\n", path, para_strerror(-ret));
1936         free(obj.data);
1937         clear_afhi(afhi_ptr);
1938         /* Stop adding files only on send errors. */
1939         return send_ret;
1940 }
1941
1942 int com_add(struct command_context *cc)
1943 {
1944         int i, ret;
1945         struct private_add_data pad = {.cc = cc, .flags = 0};
1946         struct stat statbuf;
1947
1948         for (i = 1; i < cc->argc; i++) {
1949                 const char *arg = cc->argv[i];
1950                 if (arg[0] != '-')
1951                         break;
1952                 if (!strcmp(arg, "--")) {
1953                         i++;
1954                         break;
1955                 }
1956                 if (!strcmp(arg, "-a")) {
1957                         pad.flags |= ADD_FLAG_ALL;
1958                         continue;
1959                 }
1960                 if (!strcmp(arg, "-l")) {
1961                         pad.flags |= ADD_FLAG_LAZY;
1962                         continue;
1963                 }
1964                 if (!strcmp(arg, "-f")) {
1965                         pad.flags |= ADD_FLAG_FORCE;
1966                         continue;
1967                 }
1968                 if (!strcmp(arg, "-v")) {
1969                         pad.flags |= ADD_FLAG_VERBOSE;
1970                         continue;
1971                 }
1972         }
1973         if (cc->argc <= i)
1974                 return -E_AFT_SYNTAX;
1975         for (; i < cc->argc; i++) {
1976                 char *path;
1977                 ret = verify_path(cc->argv[i], &path);
1978                 if (ret < 0) {
1979                         ret = send_sb_va(&cc->scc, SBD_ERROR_LOG, "%s: %s\n",
1980                                 cc->argv[i], para_strerror(-ret));
1981                         if (ret < 0)
1982                                 return ret;
1983                         continue;
1984                 }
1985                 ret = stat(path, &statbuf);
1986                 if (ret < 0) {
1987                         ret = send_sb_va(&cc->scc, SBD_ERROR_LOG,
1988                                 "failed to stat %s (%s)\n", path,
1989                                 strerror(errno));
1990                         free(path);
1991                         if (ret < 0)
1992                                 return ret;
1993                         continue;
1994                 }
1995                 if (S_ISDIR(statbuf.st_mode))
1996                         ret = for_each_file_in_dir(path, add_one_audio_file,
1997                                 &pad);
1998                 else
1999                         ret = add_one_audio_file(path, &pad);
2000                 if (ret < 0) {
2001                         send_sb_va(&cc->scc, SBD_OUTPUT, "%s: %s\n", path,
2002                                 para_strerror(-ret));
2003                         free(path);
2004                         return ret;
2005                 }
2006                 free(path);
2007         }
2008         return 1;
2009
2010 }
2011
2012 /**
2013  * Flags used by the touch command.
2014  *
2015  * \sa com_touch().
2016  */
2017 enum touch_flags {
2018         /** Whether the \p FNM_PATHNAME flag should be passed to fnmatch(). */
2019         TOUCH_FLAG_FNM_PATHNAME = 1,
2020         /** Activates verbose mode. */
2021         TOUCH_FLAG_VERBOSE = 2
2022 };
2023
2024 /** Options used by com_touch(). */
2025 struct com_touch_options {
2026         /** New num_played value. */
2027         int32_t num_played;
2028         /** New last played count. */
2029         int64_t last_played;
2030         /** New lyrics id. */
2031         int32_t lyrics_id;
2032         /** New image id. */
2033         int32_t image_id;
2034         /** New amplification value. */
2035         int32_t amp;
2036         /** Command line flags (see \ref touch_flags). */
2037         unsigned flags;
2038 };
2039
2040 /** Data passed to the action handler of com_touch(). */
2041 struct touch_action_data {
2042         /** Command line options (see \ref com_touch_options). */
2043         struct com_touch_options *cto;
2044         /** Message buffer. */
2045         struct para_buffer pb;
2046 };
2047
2048 static int touch_audio_file(__a_unused struct osl_table *table,
2049                 struct osl_row *row, const char *name, void *data)
2050 {
2051         struct touch_action_data *tad = data;
2052         struct osl_object obj;
2053         struct afs_info old_afsi, new_afsi;
2054         int ret, no_options = tad->cto->num_played < 0 && tad->cto->last_played < 0 &&
2055                 tad->cto->lyrics_id < 0 && tad->cto->image_id < 0 && tad->cto->amp < 0;
2056         struct afsi_change_event_data aced;
2057
2058         ret = get_afsi_object_of_row(row, &obj);
2059         if (ret < 0)
2060                 return para_printf(&tad->pb, "%s: %s\n", name, para_strerror(-ret));
2061         ret = load_afsi(&old_afsi, &obj);
2062         if (ret < 0)
2063                 return para_printf(&tad->pb, "%s: %s\n", name, para_strerror(-ret));
2064         new_afsi = old_afsi;
2065         if (no_options) {
2066                 new_afsi.num_played++;
2067                 new_afsi.last_played = time(NULL);
2068                 if (tad->cto->flags & TOUCH_FLAG_VERBOSE) {
2069                         ret = para_printf(&tad->pb, "%s: num_played = %u, "
2070                                 "last_played = now()\n", name,
2071                                 new_afsi.num_played);
2072                         if (ret < 0)
2073                                 return ret;
2074                 }
2075         } else {
2076                 if (tad->cto->flags & TOUCH_FLAG_VERBOSE) {
2077                         ret = para_printf(&tad->pb, "touching %s\n", name);
2078                         if (ret < 0)
2079                                 return ret;
2080                 }
2081                 if (tad->cto->lyrics_id >= 0)
2082                         new_afsi.lyrics_id = tad->cto->lyrics_id;
2083                 if (tad->cto->image_id >= 0)
2084                         new_afsi.image_id = tad->cto->image_id;
2085                 if (tad->cto->num_played >= 0)
2086                         new_afsi.num_played = tad->cto->num_played;
2087                 if (tad->cto->last_played >= 0)
2088                         new_afsi.last_played = tad->cto->last_played;
2089                 if (tad->cto->amp >= 0)
2090                         new_afsi.amp = tad->cto->amp;
2091         }
2092         save_afsi(&new_afsi, &obj); /* in-place update */
2093         aced.aft_row = row;
2094         aced.old_afsi = &old_afsi;
2095         afs_event(AFSI_CHANGE, &tad->pb, &aced);
2096         return 1;
2097 }
2098
2099 static void com_touch_callback(int fd, const struct osl_object *query)
2100 {
2101         struct touch_action_data tad = {.cto = query->data,
2102                 .pb = {
2103                         .max_size = shm_get_shmmax(),
2104                         .private_data = &fd,
2105                         .max_size_handler = afs_max_size_handler
2106                 }
2107         };
2108         int ret, ret2 = 0;
2109         struct pattern_match_data pmd = {
2110                 .table = audio_file_table,
2111                 .loop_col_num = AFTCOL_HASH,
2112                 .match_col_num = AFTCOL_PATH,
2113                 .patterns = {.data = (char *)query->data + sizeof(*tad.cto),
2114                         .size = query->size - sizeof(*tad.cto)},
2115                 .data = &tad,
2116                 .action = touch_audio_file
2117         };
2118         if (tad.cto->flags & TOUCH_FLAG_FNM_PATHNAME)
2119                 pmd.fnmatch_flags |= FNM_PATHNAME;
2120         ret = for_each_matching_row(&pmd);
2121         if (ret < 0)
2122                 ret2 = para_printf(&tad.pb, "%s\n", para_strerror(-ret));
2123         else if (pmd.num_matches == 0)
2124                 ret2 = para_printf(&tad.pb, "no matches\n");
2125         if (ret2 >= 0 && tad.pb.offset)
2126                 pass_buffer_as_shm(fd, SBD_OUTPUT, tad.pb.buf, tad.pb.offset);
2127         free(tad.pb.buf);
2128 }
2129
2130 int com_touch(struct command_context *cc)
2131 {
2132         struct com_touch_options cto = {
2133                 .num_played = -1,
2134                 .last_played = -1,
2135                 .lyrics_id = -1,
2136                 .image_id = -1,
2137                 .amp = -1,
2138         };
2139         struct osl_object query = {.data = &cto, .size = sizeof(cto)};
2140         int i, ret;
2141
2142
2143         for (i = 1; i < cc->argc; i++) {
2144                 const char *arg = cc->argv[i];
2145                 if (arg[0] != '-')
2146                         break;
2147                 if (!strcmp(arg, "--")) {
2148                         i++;
2149                         break;
2150                 }
2151                 if (!strncmp(arg, "-n=", 3)) {
2152                         ret = para_atoi32(arg + 3, &cto.num_played);
2153                         if (ret < 0)
2154                                 return ret;
2155                         continue;
2156                 }
2157                 if (!strncmp(arg, "-l=", 3)) {
2158                         ret = para_atoi64(arg + 3, &cto.last_played);
2159                         if (ret < 0)
2160                                 return ret;
2161                         continue;
2162                 }
2163                 if (!strncmp(arg, "-y=", 3)) {
2164                         ret = para_atoi32(arg + 3, &cto.lyrics_id);
2165                         if (ret < 0)
2166                                 return ret;
2167                         continue;
2168                 }
2169                 if (!strncmp(arg, "-i=", 3)) {
2170                         ret = para_atoi32(arg + 3, &cto.image_id);
2171                         if (ret < 0)
2172                                 return ret;
2173                         continue;
2174                 }
2175                 if (!strncmp(arg, "-a=", 3)) {
2176                         int32_t val;
2177                         ret = para_atoi32(arg + 3, &val);
2178                         if (ret < 0)
2179                                 return ret;
2180                         if (val < 0 || val > 255)
2181                                 return -ERRNO_TO_PARA_ERROR(EINVAL);
2182                         cto.amp = val;
2183                         continue;
2184                 }
2185                 if (!strcmp(arg, "-p")) {
2186                         cto.flags |= TOUCH_FLAG_FNM_PATHNAME;
2187                         continue;
2188                 }
2189                 if (!strcmp(arg, "-v")) {
2190                         cto.flags |= TOUCH_FLAG_VERBOSE;
2191                         continue;
2192                 }
2193                 break; /* non-option starting with dash */
2194         }
2195         if (i >= cc->argc)
2196                 return -E_AFT_SYNTAX;
2197         ret = send_option_arg_callback_request(&query, cc->argc - i,
2198                 cc->argv + i, com_touch_callback, afs_cb_result_handler, cc);
2199         if (ret < 0)
2200                 send_strerror(cc, -ret);
2201         return ret;
2202 }
2203
2204 /** Flags for com_rm(). */
2205 enum rm_flags {
2206         /** -v */
2207         RM_FLAG_VERBOSE = 1,
2208         /** -f */
2209         RM_FLAG_FORCE = 2,
2210         /** -p */
2211         RM_FLAG_FNM_PATHNAME = 4
2212 };
2213
2214 /** Data passed to the action handler of com_rm(). */
2215 struct com_rm_action_data {
2216         /** Command line flags ((see \ref rm_flags). */
2217         uint32_t flags;
2218         /** Message buffer. */
2219         struct para_buffer pb;
2220 };
2221
2222 static int remove_audio_file(__a_unused struct osl_table *table,
2223                 struct osl_row *row, const char *name, void *data)
2224 {
2225         struct com_rm_action_data *crd = data;
2226         int ret;
2227
2228         if (crd->flags & RM_FLAG_VERBOSE) {
2229                 ret = para_printf(&crd->pb, "removing %s\n", name);
2230                 if (ret < 0)
2231                         return ret;
2232         }
2233         afs_event(AUDIO_FILE_REMOVE, &crd->pb, row);
2234         ret = osl(osl_del_row(audio_file_table, row));
2235         if (ret < 0)
2236                 para_printf(&crd->pb, "%s: %s\n", name, para_strerror(-ret));
2237         return ret;
2238 }
2239
2240 static void com_rm_callback(int fd, const struct osl_object *query)
2241 {
2242         struct com_rm_action_data crd = {.flags = *(uint32_t *)query->data,
2243                 .pb = {
2244                         .max_size = shm_get_shmmax(),
2245                         .private_data = &fd,
2246                         .max_size_handler = afs_max_size_handler
2247                 }
2248         };
2249         int ret;
2250         struct pattern_match_data pmd = {
2251                 .table = audio_file_table,
2252                 .loop_col_num = AFTCOL_HASH,
2253                 .match_col_num = AFTCOL_PATH,
2254                 .patterns = {.data = (char *)query->data + sizeof(uint32_t),
2255                         .size = query->size - sizeof(uint32_t)},
2256                 .data = &crd,
2257                 .action = remove_audio_file
2258         };
2259         if (crd.flags & RM_FLAG_FNM_PATHNAME)
2260                 pmd.fnmatch_flags |= FNM_PATHNAME;
2261         ret = for_each_matching_row(&pmd);
2262         if (ret < 0) {
2263                 para_printf(&crd.pb, "%s\n", para_strerror(-ret));
2264                 return;
2265         }
2266         if ((pmd.num_matches == 0) && !(crd.flags & RM_FLAG_FORCE))
2267                 ret = para_printf(&crd.pb, "no matches -- nothing removed\n");
2268         else if (crd.flags & RM_FLAG_VERBOSE) {
2269                 ret = para_printf(&crd.pb, "removed %u files\n",
2270                         pmd.num_matches);
2271         }
2272         if (ret >= 0 && crd.pb.offset)
2273                 pass_buffer_as_shm(fd, SBD_OUTPUT, crd.pb.buf, crd.pb.offset);
2274         free(crd.pb.buf);
2275 }
2276
2277 /* TODO options: -r (recursive) */
2278 int com_rm(struct command_context *cc)
2279 {
2280         uint32_t flags = 0;
2281         struct osl_object query = {.data = &flags, .size = sizeof(flags)};
2282         int i, ret;
2283
2284         for (i = 1; i < cc->argc; i++) {
2285                 const char *arg = cc->argv[i];
2286                 if (arg[0] != '-')
2287                         break;
2288                 if (!strcmp(arg, "--")) {
2289                         i++;
2290                         break;
2291                 }
2292                 if (!strcmp(arg, "-f")) {
2293                         flags |= RM_FLAG_FORCE;
2294                         continue;
2295                 }
2296                 if (!strcmp(arg, "-p")) {
2297                         flags |= RM_FLAG_FNM_PATHNAME;
2298                         continue;
2299                 }
2300                 if (!strcmp(arg, "-v")) {
2301                         flags |= RM_FLAG_VERBOSE;
2302                         continue;
2303                 }
2304                 break;
2305         }
2306         if (i >= cc->argc)
2307                 return -E_AFT_SYNTAX;
2308         ret = send_option_arg_callback_request(&query, cc->argc - i,
2309                 cc->argv + i, com_rm_callback, afs_cb_result_handler, cc);
2310         if (ret < 0)
2311                 send_strerror(cc, -ret);
2312         return ret;
2313 }
2314
2315 /**
2316  * Flags used by the cpsi command.
2317  *
2318  * \sa com_cpsi().
2319  */
2320 enum cpsi_flags {
2321         /** Whether the lyrics id should be copied. */
2322         CPSI_FLAG_COPY_LYRICS_ID = 1,
2323         /** Whether the image id should be copied. */
2324         CPSI_FLAG_COPY_IMAGE_ID = 2,
2325         /** Whether the lastplayed time should be copied. */
2326         CPSI_FLAG_COPY_LASTPLAYED = 4,
2327         /** Whether the numplayed count should be copied. */
2328         CPSI_FLAG_COPY_NUMPLAYED = 8,
2329         /** Whether the attributes should be copied. */
2330         CPSI_FLAG_COPY_ATTRIBUTES = 16,
2331         /** Activates verbose mode. */
2332         CPSI_FLAG_VERBOSE = 32,
2333 };
2334
2335 /** Data passed to the action handler of com_cpsi(). */
2336 struct cpsi_action_data {
2337         /** command line flags (see \ref cpsi_flags). */
2338         unsigned flags;
2339         /** Message buffer. */
2340         struct para_buffer pb;
2341         /** Values are copied from here. */
2342         struct afs_info source_afsi;
2343 };
2344
2345 static int copy_selector_info(__a_unused struct osl_table *table,
2346                 struct osl_row *row, const char *name, void *data)
2347 {
2348         struct cpsi_action_data *cad = data;
2349         struct osl_object target_afsi_obj;
2350         int ret;
2351         struct afs_info old_afsi, target_afsi;
2352         struct afsi_change_event_data aced;
2353
2354         ret = get_afsi_object_of_row(row, &target_afsi_obj);
2355         if (ret < 0)
2356                 return ret;
2357         load_afsi(&target_afsi, &target_afsi_obj);
2358         old_afsi = target_afsi;
2359         if (cad->flags & CPSI_FLAG_COPY_LYRICS_ID)
2360                 target_afsi.lyrics_id = cad->source_afsi.lyrics_id;
2361         if (cad->flags & CPSI_FLAG_COPY_IMAGE_ID)
2362                 target_afsi.image_id = cad->source_afsi.image_id;
2363         if (cad->flags & CPSI_FLAG_COPY_LASTPLAYED)
2364                 target_afsi.last_played = cad->source_afsi.last_played;
2365         if (cad->flags & CPSI_FLAG_COPY_NUMPLAYED)
2366                 target_afsi.num_played = cad->source_afsi.num_played;
2367         if (cad->flags & CPSI_FLAG_COPY_ATTRIBUTES)
2368                 target_afsi.attributes = cad->source_afsi.attributes;
2369         save_afsi(&target_afsi, &target_afsi_obj); /* in-place update */
2370         if (cad->flags & CPSI_FLAG_VERBOSE) {
2371                 ret = para_printf(&cad->pb, "copied afsi to %s\n", name);
2372                 if (ret < 0)
2373                         return ret;
2374         }
2375         aced.aft_row = row;
2376         aced.old_afsi = &old_afsi;
2377         afs_event(AFSI_CHANGE, &cad->pb, &aced);
2378         return 1;
2379 }
2380
2381 static void com_cpsi_callback(int fd, const struct osl_object *query)
2382 {
2383         struct cpsi_action_data cad = {
2384                 .flags = *(unsigned *)query->data,
2385                 .pb = {
2386                         .max_size = shm_get_shmmax(),
2387                         .private_data = &fd,
2388                         .max_size_handler = afs_max_size_handler
2389                 }
2390         };
2391         int ret;
2392         char *source_path = (char *)query->data + sizeof(cad.flags);
2393         struct pattern_match_data pmd = {
2394                 .table = audio_file_table,
2395                 .loop_col_num = AFTCOL_HASH,
2396                 .match_col_num = AFTCOL_PATH,
2397                 .patterns = {.data = source_path + strlen(source_path) + 1,
2398                         .size = query->size - sizeof(cad.flags)
2399                                 - strlen(source_path) - 1},
2400                 .data = &cad,
2401                 .action = copy_selector_info
2402         };
2403
2404         ret = get_afsi_of_path(source_path, &cad.source_afsi);
2405         if (ret < 0)
2406                 goto out;
2407         ret = for_each_matching_row(&pmd);
2408 out:
2409         if (ret < 0)
2410                 para_printf(&cad.pb, "%s\n", para_strerror(-ret));
2411         else if (cad.flags & CPSI_FLAG_VERBOSE) {
2412                 if (pmd.num_matches > 0)
2413                         para_printf(&cad.pb, "copied requested afsi from %s "
2414                                 "to %u files\n", source_path, pmd.num_matches);
2415                 else
2416                         para_printf(&cad.pb, "nothing copied\n");
2417         }
2418         if (cad.pb.offset)
2419                 pass_buffer_as_shm(fd, SBD_OUTPUT, cad.pb.buf, cad.pb.offset);
2420         free(cad.pb.buf);
2421 }
2422
2423 int com_cpsi(struct command_context *cc)
2424 {
2425         unsigned flags = 0;
2426         int i, ret;
2427         struct osl_object options = {.data = &flags, .size = sizeof(flags)};
2428
2429         for (i = 1; i < cc->argc; i++) {
2430                 const char *arg = cc->argv[i];
2431                 if (arg[0] != '-')
2432                         break;
2433                 if (!strcmp(arg, "--")) {
2434                         i++;
2435                         break;
2436                 }
2437                 if (!strcmp(arg, "-y")) {
2438                         flags |= CPSI_FLAG_COPY_LYRICS_ID;
2439                         continue;
2440                 }
2441                 if (!strcmp(arg, "-i")) {
2442                         flags |= CPSI_FLAG_COPY_IMAGE_ID;
2443                         continue;
2444                 }
2445                 if (!strcmp(arg, "-l")) {
2446                         flags |= CPSI_FLAG_COPY_LASTPLAYED;
2447                         continue;
2448                 }
2449                 if (!strcmp(arg, "-n")) {
2450                         flags |= CPSI_FLAG_COPY_NUMPLAYED;
2451                         continue;
2452                 }
2453                 if (!strcmp(arg, "-a")) {
2454                         flags |= CPSI_FLAG_COPY_ATTRIBUTES;
2455                         continue;
2456                 }
2457                 if (!strcmp(arg, "-v")) {
2458                         flags |= CPSI_FLAG_VERBOSE;
2459                         continue;
2460                 }
2461                 break;
2462         }
2463         if (i + 1 >= cc->argc) /* need at least source file and pattern */
2464                 return -E_AFT_SYNTAX;
2465         if (!(flags & ~CPSI_FLAG_VERBOSE)) /* no copy flags given */
2466                 flags = ~(unsigned)CPSI_FLAG_VERBOSE | flags;
2467         ret = send_option_arg_callback_request(&options, cc->argc - i,
2468                 cc->argv + i, com_cpsi_callback, afs_cb_result_handler, cc);
2469         if (ret < 0)
2470                 send_strerror(cc, -ret);
2471         return ret;
2472 }
2473
2474 struct change_atts_data {
2475         uint64_t add_mask, del_mask;
2476         struct para_buffer pb;
2477 };
2478
2479 static int change_atts(__a_unused struct osl_table *table,
2480                 struct osl_row *row, __a_unused const char *name, void *data)
2481 {
2482         int ret;
2483         struct osl_object obj;
2484         struct afs_info old_afsi, new_afsi;
2485         struct afsi_change_event_data aced = {
2486                 .aft_row = row,
2487                 .old_afsi = &old_afsi
2488         };
2489         struct change_atts_data *cad = data;
2490
2491         ret = get_afsi_object_of_row(row, &obj);
2492         if (ret < 0)
2493                 return ret;
2494         ret = load_afsi(&old_afsi, &obj);
2495         if (ret < 0)
2496                 return ret;
2497         new_afsi = old_afsi;
2498         new_afsi.attributes |= cad->add_mask;
2499         new_afsi.attributes &= ~cad->del_mask;
2500         save_afsi(&new_afsi, &obj); /* in-place update */
2501         afs_event(AFSI_CHANGE, &cad->pb, &aced);
2502         return 1;
2503 }
2504
2505 static void com_setatt_callback(int fd, const struct osl_object *query)
2506 {
2507         char *p;
2508         int ret;
2509         size_t len;
2510         struct change_atts_data cad = {
2511                 .pb = {
2512                         .max_size = shm_get_shmmax(),
2513                         .max_size_handler = afs_max_size_handler,
2514                         .private_data = &(struct afs_max_size_handler_data) {
2515                                 .fd = fd,
2516                                 .band = SBD_OUTPUT
2517                         }
2518                 }
2519         };
2520         struct pattern_match_data pmd = {
2521                 .table = audio_file_table,
2522                 .loop_col_num = AFTCOL_HASH,
2523                 .match_col_num = AFTCOL_PATH,
2524                 .pm_flags = PM_SKIP_EMPTY_NAME,
2525                 .data = &cad,
2526                 .action = change_atts
2527         };
2528
2529         for (p = query->data; p < (char *)query->data + query->size; p += len + 1) {
2530                 char c;
2531                 unsigned char bitnum;
2532
2533                 len = strlen(p);
2534                 ret = -E_ATTR_SYNTAX;
2535                 if (len == 0)
2536                         goto out;
2537                 c = p[len - 1];
2538                 if (c != '+' && c != '-')
2539                         break;
2540                 p[len - 1] = '\0';
2541                 ret = get_attribute_bitnum_by_name(p, &bitnum);
2542                 if (ret < 0)
2543                         goto out;
2544                 if (c == '+')
2545                         cad.add_mask |= (1UL << bitnum);
2546                 else
2547                         cad.del_mask |= (1UL << bitnum);
2548         }
2549         ret = -E_ATTR_SYNTAX;
2550         if (!cad.add_mask && !cad.del_mask)
2551                 goto out;
2552         pmd.patterns.data = p;
2553         assert(p < (char *)query->data + query->size);
2554         pmd.patterns.size = (char *)query->data + query->size - p;
2555         ret = for_each_matching_row(&pmd);
2556         if (ret < 0)
2557                 goto out;
2558         if (pmd.num_matches == 0)
2559                 para_printf(&cad.pb, "no matches\n");
2560 out:
2561         if (ret < 0)
2562                 para_printf(&cad.pb, "%s\n", para_strerror(-ret));
2563         if (cad.pb.offset)
2564                 pass_buffer_as_shm(fd, SBD_OUTPUT, cad.pb.buf, cad.pb.offset);
2565         free(cad.pb.buf);
2566 }
2567
2568 int com_setatt(struct command_context *cc)
2569 {
2570         if (cc->argc < 3)
2571                 return -E_ATTR_SYNTAX;
2572         return send_standard_callback_request(cc->argc - 1, cc->argv + 1,
2573                 com_setatt_callback, afs_cb_result_handler, cc);
2574 }
2575
2576 static void afs_stat_callback(int fd, const struct osl_object *query)
2577 {
2578         int *parser_friendly = query->data;
2579         char *buf = *parser_friendly?
2580                 parser_friendly_status_items : status_items;
2581
2582         if (!buf)
2583                 return;
2584         pass_buffer_as_shm(fd, SBD_OUTPUT, buf, strlen(buf));
2585 }
2586
2587 /**
2588  * Get the current afs status items from the afs process and send it.
2589  *
2590  * \param cc The command context, used e.g. for data encryption.
2591  * \param parser_friendly Whether parser-friendly output format should be used.
2592  *
2593  * As the contents of the afs status items change in time and the command
2594  * handler only has a COW version created at fork time, it can not send
2595  * up-to-date afs status items directly. Therefore the usual callback mechanism
2596  * is used to pass the status items from the afs process to the command handler
2597  * via a shared memory area and a pipe.
2598  *
2599  * \return The return value of the underyling call to \ref send_callback_request().
2600  */
2601 int send_afs_status(struct command_context *cc, int parser_friendly)
2602 {
2603         struct osl_object query = {.data = &parser_friendly,
2604                 .size = sizeof(parser_friendly)};
2605
2606         return send_callback_request(afs_stat_callback, &query,
2607                 afs_cb_result_handler, cc);
2608 }
2609
2610 /* TODO: optionally fix problems by removing offending rows */
2611 static int check_audio_file(struct osl_row *row, void *data)
2612 {
2613         char *path;
2614         struct para_buffer *pb = data;
2615         struct stat statbuf;
2616         int ret = get_audio_file_path_of_row(row, &path);
2617         struct afs_info afsi;
2618         char *blob_name;
2619
2620         if (ret < 0)
2621                 return para_printf(pb, "%s\n", para_strerror(-ret));
2622         if (stat(path, &statbuf) < 0) {
2623                 ret = para_printf(pb, "%s: stat error (%s)\n", path, strerror(errno));
2624                 if (ret < 0)
2625                         return ret;
2626         } else {
2627                 if (!S_ISREG(statbuf.st_mode)) {
2628                         ret = para_printf(pb, "%s: not a regular file\n", path);
2629                         if (ret < 0)
2630                                 return ret;
2631                 }
2632         }
2633         ret = get_afsi_of_row(row, &afsi);
2634         if (ret < 0)
2635                 return para_printf(pb, "%s: %s\n", path, para_strerror(-ret));
2636         ret = lyr_get_name_by_id(afsi.lyrics_id, &blob_name);
2637         if (ret < 0) {
2638                 ret = para_printf(pb, "%s lyrics id %u: %s\n", path, afsi.lyrics_id,
2639                         para_strerror(-ret));
2640                 if (ret < 0)
2641                         return ret;
2642         }
2643         ret = img_get_name_by_id(afsi.image_id, &blob_name);
2644         if (ret < 0)
2645                 ret = para_printf(pb, "%s image id %u: %s\n", path, afsi.image_id,
2646                         para_strerror(-ret));
2647         return ret;
2648 }
2649
2650 /**
2651  * Check the audio file table for inconsistencies.
2652  *
2653  * \param fd The afs socket.
2654  * \param query Unused.
2655  *
2656  * This function always succeeds.
2657  *
2658  * \sa com_check().
2659  */
2660 void aft_check_callback(int fd, __a_unused const struct osl_object *query)
2661 {
2662         struct para_buffer pb = {
2663                 .max_size = shm_get_shmmax(),
2664                 .private_data = &(struct afs_max_size_handler_data) {
2665                         .fd = fd,
2666                         .band = SBD_OUTPUT
2667                 },
2668                 .max_size_handler = afs_max_size_handler
2669         };
2670         int ret = para_printf(&pb, "checking audio file table...\n");
2671
2672         if (ret < 0)
2673                 return;
2674         audio_file_loop(&pb, check_audio_file);
2675         if (pb.offset)
2676                 pass_buffer_as_shm(fd, SBD_OUTPUT, pb.buf, pb.offset);
2677         free(pb.buf);
2678 }
2679
2680 /**
2681  * Close the audio file table.
2682  *
2683  * \param flags Usual flags that are passed to osl_close_table().
2684  *
2685  * \sa osl_close_table().
2686  */
2687 static void aft_close(void)
2688 {
2689         osl_close_table(audio_file_table, OSL_MARK_CLEAN);
2690         audio_file_table = NULL;
2691 }
2692
2693 /**
2694  * Open the audio file table.
2695  *
2696  * \param dir The database directory.
2697  *
2698  * \return Standard.
2699  *
2700  * \sa osl_open_table().
2701  */
2702 static int aft_open(const char *dir)
2703 {
2704         int ret;
2705
2706         audio_file_table_desc.dir = dir;
2707         ret = osl(osl_open_table(&audio_file_table_desc, &audio_file_table));
2708         if (ret >= 0) {
2709                 unsigned num;
2710                 osl_get_num_rows(audio_file_table, &num);
2711                 PARA_INFO_LOG("audio file table contains %d files\n", num);
2712                 return ret;
2713         }
2714         PARA_INFO_LOG("failed to open audio file table\n");
2715         audio_file_table = NULL;
2716         if (ret >= 0 || ret == -OSL_ERRNO_TO_PARA_ERROR(E_OSL_NOENT))
2717                 return 1;
2718         return ret;
2719 }
2720
2721 static int aft_create(const char *dir)
2722 {
2723         audio_file_table_desc.dir = dir;
2724         return osl(osl_create_table(&audio_file_table_desc));
2725 }
2726
2727 static int clear_attribute(struct osl_row *row, void *data)
2728 {
2729         struct rmatt_event_data *red = data;
2730         struct afs_info afsi;
2731         struct osl_object obj;
2732         int ret = get_afsi_object_of_row(row, &obj);
2733         uint64_t mask = ~(1ULL << red->bitnum);
2734
2735         if (ret < 0)
2736                 return ret;
2737         ret = load_afsi(&afsi, &obj);
2738         if (ret < 0)
2739                 return ret;
2740         afsi.attributes &= mask;
2741         save_afsi(&afsi, &obj);
2742         return 1;
2743 }
2744
2745 static int aft_event_handler(enum afs_events event, struct para_buffer *pb,
2746                 void *data)
2747 {
2748         int ret;
2749
2750         switch(event) {
2751         case ATTRIBUTE_REMOVE: {
2752                 const struct rmatt_event_data *red = data;
2753                 ret = para_printf(pb, "clearing attribute %s (bit %u) from all "
2754                         "entries in the audio file table\n", red->name,
2755                         red->bitnum);
2756                 if (ret < 0)
2757                         return ret;
2758                 return audio_file_loop(data, clear_attribute);
2759                 }
2760         default:
2761                 return 1;
2762         }
2763 }
2764
2765 /**
2766  * Initialize the audio file table.
2767  *
2768  * \param t Pointer to the structure to be initialized.
2769  */
2770 void aft_init(struct afs_table *t)
2771 {
2772         t->open = aft_open;
2773         t->close = aft_close;
2774         t->create = aft_create;
2775         t->event_handler = aft_event_handler;
2776 }