Send command output via sideband.
[paraslash.git] / aft.c
1 /*
2  * Copyright (C) 2007-2012 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         ret = get_audio_file_path_of_row(aft_row, &path);
1283         if (ret < 0)
1284                 return ret;
1285         if (!(options->flags & LS_FLAG_FULL_PATH)) {
1286                 char *p = strrchr(path, '/');
1287                 if (p)
1288                         path = p + 1;
1289         }
1290         if (options->num_patterns) {
1291                 for (i = 0; i < options->num_patterns; i++) {
1292                         ret = fnmatch(options->patterns[i], path, 0);
1293                         if (!ret)
1294                                 break;
1295                         if (ret == FNM_NOMATCH)
1296                                 continue;
1297                         return -E_FNMATCH;
1298                 }
1299                 if (i >= options->num_patterns) /* no match */
1300                         return 1;
1301         }
1302         tmp = options->num_matching_paths++;
1303         if (options->num_matching_paths > options->array_size) {
1304                 options->array_size++;
1305                 options->array_size *= 2;
1306                 options->data = para_realloc(options->data, options->array_size
1307                         * sizeof(*options->data));
1308         }
1309         d = options->data + tmp;
1310         ret = get_afsi_of_row(aft_row, &d->afsi);
1311         if (ret < 0)
1312                 return ret;
1313         ret = get_afhi_of_row(aft_row, &d->afhi);
1314         if (ret < 0)
1315                 return ret;
1316         d->path = path;
1317         ret = get_hash_of_row(aft_row, &d->hash);
1318         if (ret < 0)
1319                 goto err;
1320         w = &options->widths;
1321         GET_NUM_DIGITS(d->afsi.image_id, &num_digits);
1322         w->image_id_width = PARA_MAX(w->image_id_width, num_digits);
1323         GET_NUM_DIGITS(d->afsi.lyrics_id, &num_digits);
1324         w->lyrics_id_width = PARA_MAX(w->lyrics_id_width, num_digits);
1325         GET_NUM_DIGITS(d->afhi.bitrate, &num_digits);
1326         w->bitrate_width = PARA_MAX(w->bitrate_width, num_digits);
1327         GET_NUM_DIGITS(d->afhi.frequency, &num_digits);
1328         w->frequency_width = PARA_MAX(w->frequency_width, num_digits);
1329         GET_NUM_DIGITS(d->afsi.num_played, &num_digits);
1330         w->num_played_width = PARA_MAX(w->num_played_width, num_digits);
1331         /* get the number of chars to print this amount of time */
1332         num_digits = get_duration_width(d->afhi.seconds_total);
1333         w->duration_width = PARA_MAX(w->duration_width, num_digits);
1334         GET_NUM_DIGITS(d->afsi.amp, &num_digits);
1335         w->amp_width = PARA_MAX(w->amp_width, num_digits);
1336         num_digits = strlen(audio_format_name(d->afsi.audio_format_id));
1337         w->audio_format_width = PARA_MAX(w->audio_format_width, num_digits);
1338         if (options->flags & LS_FLAG_ADMISSIBLE_ONLY) {
1339                 GET_NUM_DIGITS(score, &num_digits);
1340                 num_digits++; /* add one for the sign (space or "-") */
1341                 w->score_width = PARA_MAX(w->score_width, num_digits);
1342                 d->score = score;
1343         }
1344         return 1;
1345 err:
1346         return ret;
1347 }
1348
1349 static void com_ls_callback(int fd, const struct osl_object *query)
1350 {
1351         struct ls_options *opts = query->data;
1352         char *p, *pattern_start = (char *)query->data + sizeof(*opts);
1353         struct para_buffer b = {
1354                 .max_size = shm_get_shmmax(),
1355                 .flags = (opts->mode == LS_MODE_PARSER)? PBF_SIZE_PREFIX : 0,
1356                 .max_size_handler = afs_max_size_handler,
1357                 .private_data = &(struct afs_max_size_handler_data) {
1358                         .fd = fd,
1359                         .band = SBD_OUTPUT
1360                 }
1361         };
1362         int i = 0, ret;
1363         time_t current_time;
1364
1365         if (opts->num_patterns) {
1366                 opts->patterns = para_malloc(opts->num_patterns * sizeof(char *));
1367                 for (i = 0, p = pattern_start; i < opts->num_patterns; i++) {
1368                         opts->patterns[i] = p;
1369                         p += strlen(p) + 1;
1370                 }
1371         } else
1372                 opts->patterns = NULL;
1373         if (opts->flags & LS_FLAG_ADMISSIBLE_ONLY)
1374                 ret = admissible_file_loop(opts, prepare_ls_row);
1375         else
1376                 ret = osl(osl_rbtree_loop(audio_file_table, AFTCOL_PATH, opts,
1377                         prepare_ls_row));
1378         if (ret < 0)
1379                 goto out;
1380         if (opts->num_matching_paths == 0) {
1381                 if (opts->num_patterns > 0)
1382                         para_printf(&b, "no matches\n");
1383                 goto out;
1384         }
1385         ret = sort_matching_paths(opts);
1386         if (ret < 0)
1387                 goto out;
1388         time(&current_time);
1389         if (opts->flags & LS_FLAG_REVERSE)
1390                 for (i = opts->num_matching_paths - 1; i >= 0; i--) {
1391                         ret = print_list_item(opts->data_ptr[i], opts, &b, current_time);
1392                         if (ret < 0)
1393                                 goto out;
1394                 }
1395         else
1396                 for (i = 0; i < opts->num_matching_paths; i++) {
1397                         ret = print_list_item(opts->data_ptr[i], opts, &b, current_time);
1398                         if (ret < 0)
1399                                 goto out;
1400                 }
1401 out:
1402         if (b.offset)
1403                 pass_buffer_as_shm(fd, SBD_OUTPUT, b.buf, b.offset);
1404         free(b.buf);
1405         free(opts->data);
1406         free(opts->data_ptr);
1407         free(opts->patterns);
1408 }
1409
1410 /*
1411  * TODO: flags -h (sort by hash)
1412  */
1413 int com_ls(struct command_context *cc)
1414 {
1415         int i, ret;
1416         unsigned flags = 0;
1417         enum ls_sorting_method sort = LS_SORT_BY_PATH;
1418         enum ls_listing_mode mode = LS_MODE_SHORT;
1419         struct ls_options opts = {.patterns = NULL};
1420         struct osl_object query = {.data = &opts, .size = sizeof(opts)};
1421
1422         for (i = 1; i < cc->argc; i++) {
1423                 const char *arg = cc->argv[i];
1424                 if (arg[0] != '-')
1425                         break;
1426                 if (!strcmp(arg, "--")) {
1427                         i++;
1428                         break;
1429                 }
1430                 if (!strncmp(arg, "-l", 2)) {
1431                         if (!*(arg + 2)) {
1432                                 mode = LS_MODE_LONG;
1433                                 continue;
1434                         }
1435                         if (*(arg + 3))
1436                                 return -E_AFT_SYNTAX;
1437                         switch(*(arg + 2)) {
1438                         case 's':
1439                                 mode = LS_MODE_SHORT;
1440                                 continue;
1441                         case 'l':
1442                                 mode = LS_MODE_LONG;
1443                                 continue;
1444                         case 'v':
1445                                 mode = LS_MODE_VERBOSE;
1446                                 continue;
1447                         case 'm':
1448                                 mode = LS_MODE_MBOX;
1449                                 continue;
1450                         case 'c':
1451                                 mode = LS_MODE_CHUNKS;
1452                                 continue;
1453                         case 'p':
1454                                 mode = LS_MODE_PARSER;
1455                                 continue;
1456                         default:
1457                                 return -E_AFT_SYNTAX;
1458                         }
1459                 }
1460                 if (!strcmp(arg, "-p")) {
1461                         flags |= LS_FLAG_FULL_PATH;
1462                         continue;
1463                 }
1464                 if (!strcmp(arg, "-a")) {
1465                         flags |= LS_FLAG_ADMISSIBLE_ONLY;
1466                         continue;
1467                 }
1468                 if (!strcmp(arg, "-r")) {
1469                         flags |= LS_FLAG_REVERSE;
1470                         continue;
1471                 }
1472                 if (!strcmp(arg, "-d")) {
1473                         flags |= LS_FLAG_UNIXDATE;
1474                         continue;
1475                 }
1476                 if (!strncmp(arg, "-s", 2)) {
1477                         if (!*(arg + 2) || *(arg + 3))
1478                                 return -E_AFT_SYNTAX;
1479                         switch(*(arg + 2)) {
1480                         case 'p':
1481                                 sort = LS_SORT_BY_PATH;
1482                                 continue;
1483                         case 's': /* -ss implies -a */
1484                                 sort = LS_SORT_BY_SCORE;
1485                                 flags |= LS_FLAG_ADMISSIBLE_ONLY;
1486                                 continue;
1487                         case 'l':
1488                                 sort = LS_SORT_BY_LAST_PLAYED;
1489                                 continue;
1490                         case 'n':
1491                                 sort = LS_SORT_BY_NUM_PLAYED;
1492                                 continue;
1493                         case 'f':
1494                                 sort = LS_SORT_BY_FREQUENCY;
1495                                 continue;
1496                         case 'c':
1497                                 sort = LS_SORT_BY_CHANNELS;
1498                                 continue;
1499                         case 'i':
1500                                 sort = LS_SORT_BY_IMAGE_ID;
1501                                 continue;
1502                         case 'y':
1503                                 sort = LS_SORT_BY_LYRICS_ID;
1504                                 continue;
1505                         case 'b':
1506                                 sort = LS_SORT_BY_BITRATE;
1507                                 continue;
1508                         case 'd':
1509                                 sort = LS_SORT_BY_DURATION;
1510                                 continue;
1511                         case 'a':
1512                                 sort = LS_SORT_BY_AUDIO_FORMAT;
1513                                 continue;
1514                         default:
1515                                 return -E_AFT_SYNTAX;
1516                         }
1517                 }
1518                 return -E_AFT_SYNTAX;
1519         }
1520         opts.flags = flags;
1521         opts.sorting = sort;
1522         opts.mode = mode;
1523         opts.num_patterns = cc->argc - i;
1524         ret = send_option_arg_callback_request(&query, opts.num_patterns,
1525                 cc->argv + i, com_ls_callback, afs_cb_result_handler, cc);
1526         return ret;
1527 }
1528
1529 /**
1530  * Call the given function for each file in the audio file table.
1531  *
1532  * \param private_data An arbitrary data pointer, passed to \a func.
1533  * \param func The custom function to be called.
1534  *
1535  * \return Standard.
1536  */
1537 int audio_file_loop(void *private_data, osl_rbtree_loop_func *func)
1538 {
1539         return osl(osl_rbtree_loop(audio_file_table, AFTCOL_HASH, private_data,
1540                 func));
1541 }
1542
1543 static struct osl_row *find_hash_sister(unsigned char *hash)
1544 {
1545         const struct osl_object obj = {.data = hash, .size = HASH_SIZE};
1546         struct osl_row *row;
1547
1548         osl_get_row(audio_file_table, AFTCOL_HASH, &obj, &row);
1549         return row;
1550 }
1551
1552 /** The format of the data stored by save_audio_file_data(). */
1553 enum com_add_buffer_offsets {
1554         /** afhi (if present) starts here. */
1555         CAB_AFHI_OFFSET_POS = 0,
1556         /** Start of the chunk table (if present). */
1557         CAB_CHUNKS_OFFSET_POS = 2,
1558         /** Audio format id. */
1559         CAB_AUDIO_FORMAT_OFFSET = 4,
1560         /** Flags given to the add command. */
1561         CAB_FLAGS_OFFSET = 5,
1562         /** The hash of the audio file being added. */
1563         CAB_HASH_OFFSET = 9,
1564         /** Start of the path of the audio file. */
1565         CAB_PATH_OFFSET = (CAB_HASH_OFFSET + HASH_SIZE),
1566 };
1567
1568 /*
1569  * Store the given data to a single buffer. Doesn't need the audio file selector
1570  * info struct as the server knows it as well.
1571  *
1572  * It's OK to call this with afhi == NULL. In this case, the audio format
1573  * handler info won't be stored in the buffer.
1574  */
1575 static void save_add_callback_buffer(unsigned char *hash, const char *path,
1576                 struct afh_info *afhi, uint32_t flags,
1577                 uint8_t audio_format_num, struct osl_object *obj)
1578 {
1579         size_t path_len = strlen(path) + 1;
1580         size_t afhi_size = sizeof_afhi_buf(afhi);
1581         size_t size = CAB_PATH_OFFSET + path_len + afhi_size
1582                 + sizeof_chunk_table(afhi);
1583         char *buf = para_malloc(size);
1584         uint16_t pos;
1585
1586         write_u8(buf + CAB_AUDIO_FORMAT_OFFSET, audio_format_num);
1587         write_u32(buf + CAB_FLAGS_OFFSET, flags);
1588
1589         memcpy(buf + CAB_HASH_OFFSET, hash, HASH_SIZE);
1590         strcpy(buf + CAB_PATH_OFFSET, path);
1591
1592         pos = CAB_PATH_OFFSET + path_len;
1593         PARA_DEBUG_LOG("size: %zu, afhi starts at %d\n", size, pos);
1594         PARA_DEBUG_LOG("last afhi byte: %p, pos %zu\n", buf + pos + afhi_size - 1,
1595                 pos + afhi_size - 1);
1596         write_u16(buf + CAB_AFHI_OFFSET_POS, pos);
1597         save_afhi(afhi, buf + pos);
1598
1599         pos += afhi_size;
1600         PARA_DEBUG_LOG("size: %zu, chunks start at %d\n", size, pos);
1601         write_u16(buf + CAB_CHUNKS_OFFSET_POS, pos);
1602         if (afhi)
1603                 save_chunk_table(afhi, buf + pos);
1604         PARA_DEBUG_LOG("last byte in buf: %p\n", buf + size - 1);
1605         obj->data = buf;
1606         obj->size = size;
1607 }
1608
1609 /*
1610
1611 Overview of the add command.
1612
1613 Input: What was passed to the callback by the command handler.
1614 ~~~~~~
1615 HS:     Hash sister. Whether an audio file with identical hash
1616         already exists in the osl database.
1617
1618 PB:     Path brother. Whether a file with the given path exists
1619         in the table.
1620
1621 F:      Force flag given. Whether add was called with -f.
1622
1623 output: Action performed by the callback.
1624 ~~~~~~~
1625 AFHI:   Whether afhi and chunk table are computed and sent.
1626 ACTION: Table modifications to be done by the callback.
1627
1628 +----+----+---+------+---------------------------------------------------+
1629 | HS | PB | F | AFHI | ACTION
1630 +----+----+---+------+---------------------------------------------------+
1631 | Y  |  Y | Y |  Y   | if HS != PB: remove PB. HS: force afhi update,
1632 |                    | update path, keep afsi
1633 +----+----+---+------+---------------------------------------------------+
1634 | Y  |  Y | N |  N   | if HS == PB: do not send callback request at all.
1635 |                    | otherwise: remove PB, HS: update path, keep afhi,
1636 |                    | afsi.
1637 +----+----+---+------+---------------------------------------------------+
1638 | Y  |  N | Y |  Y   | (rename) force afhi update of HS, update path of
1639 |                    | HS, keep afsi
1640 +----+----+---+------+---------------------------------------------------+
1641 | Y  |  N | N |  N   | (file rename) update path of HS, keep afsi, afhi
1642 +----+----+---+------+---------------------------------------------------+
1643 | N  |  Y | Y |  Y   | (file change) update afhi, hash, of PB, keep afsi
1644 |                    | (force has no effect)
1645 +----+----+---+------+---------------------------------------------------+
1646 | N  |  Y | N |  Y   | (file change) update afhi, hash of PB, keep afsi
1647 +----+----+---+------+---------------------------------------------------+
1648 | N  |  N | Y |  Y   | (new file) create new entry (force has no effect)
1649 +----+----+---+------+---------------------------------------------------+
1650 |  N |  N | N |  Y   | (new file) create new entry
1651 +----+----+---+------+---------------------------------------------------+
1652
1653 Notes:
1654
1655         afhi <=> force or no HS
1656         F => AFHI
1657
1658 */
1659
1660 /** Flags passed to the add command. */
1661 enum com_add_flags {
1662         /** Skip paths that exist already. */
1663         ADD_FLAG_LAZY = 1,
1664         /** Force adding. */
1665         ADD_FLAG_FORCE = 2,
1666         /** Print what is being done. */
1667         ADD_FLAG_VERBOSE = 4,
1668         /** Try to add files with unknown suffixes. */
1669         ADD_FLAG_ALL = 8,
1670 };
1671
1672 static void com_add_callback(int fd, const struct osl_object *query)
1673 {
1674         char *buf = query->data, *path;
1675         struct osl_row *pb, *aft_row;
1676         struct osl_row *hs;
1677         struct osl_object objs[NUM_AFT_COLUMNS];
1678         unsigned char *hash;
1679         char asc[2 * HASH_SIZE + 1];
1680         int ret;
1681         char afsi_buf[AFSI_SIZE];
1682         uint32_t flags = read_u32(buf + CAB_FLAGS_OFFSET);
1683         struct afs_info default_afsi = {.last_played = 0};
1684         struct para_buffer msg = {
1685                 .max_size = shm_get_shmmax(),
1686                 .max_size_handler = afs_max_size_handler,
1687                 .private_data = &(struct afs_max_size_handler_data) {
1688                         .fd = fd,
1689                         .band = SBD_OUTPUT
1690                 }
1691         };
1692         uint16_t afhi_offset, chunks_offset;
1693
1694         hash = (unsigned char *)buf + CAB_HASH_OFFSET;
1695         hash_to_asc(hash, asc);;
1696         objs[AFTCOL_HASH].data = buf + CAB_HASH_OFFSET;
1697         objs[AFTCOL_HASH].size = HASH_SIZE;
1698
1699         path = buf + CAB_PATH_OFFSET;
1700         objs[AFTCOL_PATH].data = path;
1701         objs[AFTCOL_PATH].size = strlen(path) + 1;
1702
1703         PARA_INFO_LOG("request to add %s\n", path);
1704         hs = find_hash_sister(hash);
1705         ret = aft_get_row_of_path(path, &pb);
1706         if (ret < 0 && ret != -OSL_ERRNO_TO_PARA_ERROR(E_OSL_RB_KEY_NOT_FOUND))
1707                 goto out;
1708         if (hs && pb && hs == pb && !(flags & ADD_FLAG_FORCE)) {
1709                 if (flags & ADD_FLAG_VERBOSE)
1710                         ret = para_printf(&msg, "ignoring duplicate\n");
1711                 else
1712                         ret = 1;
1713                 goto out;
1714         }
1715         if (hs && hs != pb) {
1716                 struct osl_object obj;
1717                 if (pb) { /* hs trumps pb, remove pb */
1718                         if (flags & ADD_FLAG_VERBOSE) {
1719                                 ret = para_printf(&msg, "removing path brother\n");
1720                                 if (ret < 0)
1721                                         goto out;
1722                         }
1723                         ret = osl(osl_del_row(audio_file_table, pb));
1724                         if (ret < 0)
1725                                 goto out;
1726                         pb = NULL;
1727                 }
1728                 /* file rename, update hs' path */
1729                 if (flags & ADD_FLAG_VERBOSE) {
1730                         ret = osl(osl_get_object(audio_file_table, hs,
1731                                 AFTCOL_PATH, &obj));
1732                         if (ret < 0)
1733                                 goto out;
1734                         ret = para_printf(&msg, "renamed from %s\n", (char *)obj.data);
1735                         if (ret < 0)
1736                                 goto out;
1737                 }
1738                 ret = osl(osl_update_object(audio_file_table, hs, AFTCOL_PATH,
1739                         &objs[AFTCOL_PATH]));
1740                 if (ret < 0)
1741                         goto out;
1742                 afs_event(AUDIO_FILE_RENAME, &msg, hs);
1743                 if (!(flags & ADD_FLAG_FORCE))
1744                         goto out;
1745         }
1746         /* no hs or force mode, child must have sent afhi */
1747         afhi_offset = read_u16(buf + CAB_AFHI_OFFSET_POS);
1748         chunks_offset = read_u16(buf + CAB_CHUNKS_OFFSET_POS);
1749
1750         objs[AFTCOL_AFHI].data = buf + afhi_offset;
1751         objs[AFTCOL_AFHI].size = chunks_offset - afhi_offset;
1752         ret = -E_NO_AFHI;
1753         if (!objs[AFTCOL_AFHI].size) /* "impossible" */
1754                 goto out;
1755         objs[AFTCOL_CHUNKS].data = buf + chunks_offset;
1756         objs[AFTCOL_CHUNKS].size = query->size - chunks_offset;
1757         if (pb && !hs) { /* update pb's hash */
1758                 char old_asc[2 * HASH_SIZE + 1];
1759                 unsigned char *old_hash;
1760                 ret = get_hash_of_row(pb, &old_hash);
1761                 if (ret < 0)
1762                         goto out;
1763                 hash_to_asc(old_hash, old_asc);
1764                 if (flags & ADD_FLAG_VERBOSE) {
1765                         ret = para_printf(&msg, "file change: %s -> %s\n",
1766                                 old_asc, asc);
1767                         if (ret < 0)
1768                                 goto out;
1769                 }
1770                 ret = osl_update_object(audio_file_table, pb, AFTCOL_HASH,
1771                         &objs[AFTCOL_HASH]);
1772                 if (ret < 0)
1773                         goto out;
1774         }
1775         if (hs || pb) { /* (hs != NULL and pb != NULL) implies hs == pb */
1776                 struct osl_row *row = pb? pb : hs;
1777                 /* update afhi and chunk_table */
1778                 if (flags & ADD_FLAG_VERBOSE) {
1779                         ret = para_printf(&msg, "updating afhi and chunk table\n");
1780                         if (ret < 0)
1781                                 goto out;
1782                 }
1783                 ret = osl(osl_update_object(audio_file_table, row, AFTCOL_AFHI,
1784                         &objs[AFTCOL_AFHI]));
1785                 if (ret < 0)
1786                         goto out;
1787                 ret = osl(osl_update_object(audio_file_table, row, AFTCOL_CHUNKS,
1788                         &objs[AFTCOL_CHUNKS]));
1789                 if (ret < 0)
1790                         goto out;
1791                 afs_event(AFHI_CHANGE, &msg, row);
1792                 goto out;
1793         }
1794         /* new entry, use default afsi */
1795         if (flags & ADD_FLAG_VERBOSE) {
1796                 ret = para_printf(&msg, "new file\n");
1797                 if (ret < 0)
1798                         goto out;
1799         }
1800         default_afsi.last_played = time(NULL) - 365 * 24 * 60 * 60;
1801         default_afsi.audio_format_id = read_u8(buf + CAB_AUDIO_FORMAT_OFFSET);
1802
1803         objs[AFTCOL_AFSI].data = &afsi_buf;
1804         objs[AFTCOL_AFSI].size = AFSI_SIZE;
1805         save_afsi(&default_afsi, &objs[AFTCOL_AFSI]);
1806         ret = osl(osl_add_and_get_row(audio_file_table, objs, &aft_row));
1807         afs_event(AUDIO_FILE_ADD, &msg, aft_row);
1808 out:
1809         if (ret < 0)
1810                 para_printf(&msg, "%s\n", para_strerror(-ret));
1811         if (msg.offset)
1812                 pass_buffer_as_shm(fd, SBD_OUTPUT, msg.buf, msg.offset);
1813         free(msg.buf);
1814 }
1815
1816 /** Used by com_add(). */
1817 struct private_add_data {
1818         /** The pointer passed to the original command handler. */
1819         struct command_context *cc;
1820         /** The given add flags. */
1821         uint32_t flags;
1822 };
1823
1824 static void path_brother_callback(int fd, const struct osl_object *query)
1825 {
1826         char *path = query->data;
1827         struct osl_row *path_brother;
1828         int ret = aft_get_row_of_path(path, &path_brother);
1829         if (ret < 0)
1830                 return;
1831         pass_buffer_as_shm(fd, SBD_OUTPUT, (char *)&path_brother,
1832                 sizeof(path_brother));
1833 }
1834
1835 static void hash_sister_callback(int fd, const struct osl_object *query)
1836 {
1837         unsigned char *hash = query->data;
1838         struct osl_row *hash_sister;
1839
1840         hash_sister = find_hash_sister(hash);
1841         if (!hash_sister)
1842                 return;
1843         pass_buffer_as_shm(fd, SBD_OUTPUT, (char *)&hash_sister,
1844                 sizeof(hash_sister));
1845 }
1846
1847 static int get_row_pointer_from_result(struct osl_object *result,
1848                 __a_unused uint8_t band, void *private)
1849 {
1850         struct osl_row **row = private;
1851         *row = *(struct osl_row **)(result->data);
1852         return 1;
1853 }
1854
1855 static int add_one_audio_file(const char *path, void *private_data)
1856 {
1857         int ret, send_ret = 1, fd;
1858         uint8_t format_num = -1;
1859         struct private_add_data *pad = private_data;
1860         struct afh_info afhi, *afhi_ptr = NULL;
1861         struct osl_row *pb = NULL, *hs = NULL; /* path brother/hash sister */
1862         struct osl_object map, obj = {.data = NULL}, query;
1863         unsigned char hash[HASH_SIZE];
1864
1865         ret = guess_audio_format(path);
1866         if (ret < 0 && !(pad->flags & ADD_FLAG_ALL))
1867                 goto out_free;
1868         query.data = (char *)path;
1869         query.size = strlen(path) + 1;
1870         ret = send_callback_request(path_brother_callback, &query,
1871                 get_row_pointer_from_result, &pb);
1872         if (ret < 0 && ret != -OSL_ERRNO_TO_PARA_ERROR(E_OSL_RB_KEY_NOT_FOUND))
1873                 goto out_free;
1874         ret = 1;
1875         if (pb && (pad->flags & ADD_FLAG_LAZY)) { /* lazy is really cheap */
1876                 if (pad->flags & ADD_FLAG_VERBOSE)
1877                         send_ret = pad->cc->use_sideband?
1878                                 send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1879                                         "lazy-ignore: %s\n", path)
1880                         :
1881                                 sc_send_va_buffer(&pad->cc->scc,
1882                                         "lazy-ignore: %s\n", path);
1883                 goto out_free;
1884         }
1885         /* We still want to add this file. Compute its hash. */
1886         ret = mmap_full_file(path, O_RDONLY, &map.data, &map.size, &fd);
1887         if (ret < 0)
1888                 goto out_free;
1889         hash_function(map.data, map.size, hash);
1890
1891         /* Check whether the database contains a file with the same hash. */
1892         query.data = hash;
1893         query.size = HASH_SIZE;
1894         ret = send_callback_request(hash_sister_callback, &query,
1895                 get_row_pointer_from_result, &hs);
1896         if (ret < 0)
1897                 goto out_unmap;
1898         /* Return success if we already know this file. */
1899         ret = 1;
1900         if (pb && hs && hs == pb && !(pad->flags & ADD_FLAG_FORCE)) {
1901                 if (pad->flags & ADD_FLAG_VERBOSE)
1902                         send_ret = pad->cc->use_sideband?
1903                                 send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1904                                         "%s exists, not forcing update\n", path)
1905                         :
1906                                 sc_send_va_buffer(&pad->cc->scc,
1907                                         "%s exists, not forcing update\n", path);
1908                 goto out_unmap;
1909         }
1910         /*
1911          * We won't recalculate the audio format info and the chunk table if
1912          * there is a hash sister and FORCE was not given.
1913          */
1914         if (!hs || (pad->flags & ADD_FLAG_FORCE)) {
1915                 ret = compute_afhi(path, map.data, map.size, fd, &afhi);
1916                 if (ret < 0)
1917                         goto out_unmap;
1918                 format_num = ret;
1919                 afhi_ptr = &afhi;
1920         }
1921         munmap(map.data, map.size);
1922         close(fd);
1923         if (pad->flags & ADD_FLAG_VERBOSE) {
1924                 send_ret = pad->cc->use_sideband?
1925                         send_sb_va(&pad->cc->scc, SBD_OUTPUT,
1926                                 "adding %s\n", path)
1927                 :
1928                         sc_send_va_buffer(&pad->cc->scc,
1929                                 "adding %s\n", path);
1930                 if (send_ret < 0)
1931                         goto out_free;
1932         }
1933         save_add_callback_buffer(hash, path, afhi_ptr, pad->flags, format_num, &obj);
1934         /* Ask afs to consider this entry for adding. */
1935         ret = send_callback_request(com_add_callback, &obj,
1936                 afs_cb_result_handler, pad->cc);
1937         goto out_free;
1938
1939 out_unmap:
1940         close(fd);
1941         munmap(map.data, map.size);
1942 out_free:
1943         if (ret < 0 && send_ret >= 0)
1944                 send_ret = pad->cc->use_sideband?
1945                         send_sb_va(&pad->cc->scc, SBD_ERROR_LOG,
1946                                 "failed to add %s (%s)\n", path,
1947                                 para_strerror(-ret))
1948                 :
1949                         sc_send_va_buffer(&pad->cc->scc,
1950                                 "failed to add %s (%s)\n", path,
1951                                 para_strerror(-ret));
1952         free(obj.data);
1953         clear_afhi(afhi_ptr);
1954         /* Stop adding files only on send errors. */
1955         return send_ret;
1956 }
1957
1958 int com_add(struct command_context *cc)
1959 {
1960         int i, ret;
1961         struct private_add_data pad = {.cc = cc, .flags = 0};
1962         struct stat statbuf;
1963
1964         for (i = 1; i < cc->argc; i++) {
1965                 const char *arg = cc->argv[i];
1966                 if (arg[0] != '-')
1967                         break;
1968                 if (!strcmp(arg, "--")) {
1969                         i++;
1970                         break;
1971                 }
1972                 if (!strcmp(arg, "-a")) {
1973                         pad.flags |= ADD_FLAG_ALL;
1974                         continue;
1975                 }
1976                 if (!strcmp(arg, "-l")) {
1977                         pad.flags |= ADD_FLAG_LAZY;
1978                         continue;
1979                 }
1980                 if (!strcmp(arg, "-f")) {
1981                         pad.flags |= ADD_FLAG_FORCE;
1982                         continue;
1983                 }
1984                 if (!strcmp(arg, "-v")) {
1985                         pad.flags |= ADD_FLAG_VERBOSE;
1986                         continue;
1987                 }
1988         }
1989         if (cc->argc <= i)
1990                 return -E_AFT_SYNTAX;
1991         for (; i < cc->argc; i++) {
1992                 char *path;
1993                 ret = verify_path(cc->argv[i], &path);
1994                 if (ret < 0) {
1995                         ret = cc->use_sideband?
1996                                 send_sb_va(&cc->scc, SBD_ERROR_LOG, "%s: %s\n",
1997                                 cc->argv[i], para_strerror(-ret))
1998                         :
1999                                 sc_send_va_buffer(&cc->scc, "%s: %s\n",
2000                                 cc->argv[i], para_strerror(-ret));
2001                         if (ret < 0)
2002                                 return ret;
2003                         continue;
2004                 }
2005                 ret = stat(path, &statbuf);
2006                 if (ret < 0) {
2007                         ret = cc->use_sideband?
2008                                 send_sb_va(&cc->scc, SBD_ERROR_LOG,
2009                                         "failed to stat %s (%s)\n", path,
2010                                         strerror(errno))
2011                                 :
2012                                 sc_send_va_buffer(&cc->scc,
2013                                         "failed to stat %s (%s)\n", path,
2014                                         strerror(errno));
2015                         free(path);
2016                         if (ret < 0)
2017                                 return ret;
2018                         continue;
2019                 }
2020                 if (S_ISDIR(statbuf.st_mode))
2021                         ret = for_each_file_in_dir(path, add_one_audio_file,
2022                                 &pad);
2023                 else
2024                         ret = add_one_audio_file(path, &pad);
2025                 if (ret < 0) {
2026                         if (cc->use_sideband)
2027                                 send_sb_va(&cc->scc, SBD_OUTPUT, "%s: %s\n", path,
2028                                         para_strerror(-ret));
2029                         else
2030                                 sc_send_va_buffer(&cc->scc, "%s: %s\n", path,
2031                                         para_strerror(-ret));
2032                         free(path);
2033                         return ret;
2034                 }
2035                 free(path);
2036         }
2037         return 1;
2038
2039 }
2040
2041 /**
2042  * Flags used by the touch command.
2043  *
2044  * \sa com_touch().
2045  */
2046 enum touch_flags {
2047         /** Whether the \p FNM_PATHNAME flag should be passed to fnmatch(). */
2048         TOUCH_FLAG_FNM_PATHNAME = 1,
2049         /** Activates verbose mode. */
2050         TOUCH_FLAG_VERBOSE = 2
2051 };
2052
2053 /** Options used by com_touch(). */
2054 struct com_touch_options {
2055         /** New num_played value. */
2056         int32_t num_played;
2057         /** New last played count. */
2058         int64_t last_played;
2059         /** New lyrics id. */
2060         int32_t lyrics_id;
2061         /** New image id. */
2062         int32_t image_id;
2063         /** New amplification value. */
2064         int32_t amp;
2065         /** Command line flags (see \ref touch_flags). */
2066         unsigned flags;
2067 };
2068
2069 /** Data passed to the action handler of com_touch(). */
2070 struct touch_action_data {
2071         /** Command line options (see \ref com_touch_options). */
2072         struct com_touch_options *cto;
2073         /** Message buffer. */
2074         struct para_buffer pb;
2075 };
2076
2077 static int touch_audio_file(__a_unused struct osl_table *table,
2078                 struct osl_row *row, const char *name, void *data)
2079 {
2080         struct touch_action_data *tad = data;
2081         struct osl_object obj;
2082         struct afs_info old_afsi, new_afsi;
2083         int ret, no_options = tad->cto->num_played < 0 && tad->cto->last_played < 0 &&
2084                 tad->cto->lyrics_id < 0 && tad->cto->image_id < 0 && tad->cto->amp < 0;
2085         struct afsi_change_event_data aced;
2086
2087         ret = get_afsi_object_of_row(row, &obj);
2088         if (ret < 0)
2089                 return para_printf(&tad->pb, "%s: %s\n", name, para_strerror(-ret));
2090         ret = load_afsi(&old_afsi, &obj);
2091         if (ret < 0)
2092                 return para_printf(&tad->pb, "%s: %s\n", name, para_strerror(-ret));
2093         new_afsi = old_afsi;
2094         if (no_options) {
2095                 new_afsi.num_played++;
2096                 new_afsi.last_played = time(NULL);
2097                 if (tad->cto->flags & TOUCH_FLAG_VERBOSE) {
2098                         ret = para_printf(&tad->pb, "%s: num_played = %u, "
2099                                 "last_played = now()\n", name,
2100                                 new_afsi.num_played);
2101                         if (ret < 0)
2102                                 return ret;
2103                 }
2104         } else {
2105                 if (tad->cto->flags & TOUCH_FLAG_VERBOSE) {
2106                         ret = para_printf(&tad->pb, "touching %s\n", name);
2107                         if (ret < 0)
2108                                 return ret;
2109                 }
2110                 if (tad->cto->lyrics_id >= 0)
2111                         new_afsi.lyrics_id = tad->cto->lyrics_id;
2112                 if (tad->cto->image_id >= 0)
2113                         new_afsi.image_id = tad->cto->image_id;
2114                 if (tad->cto->num_played >= 0)
2115                         new_afsi.num_played = tad->cto->num_played;
2116                 if (tad->cto->last_played >= 0)
2117                         new_afsi.last_played = tad->cto->last_played;
2118                 if (tad->cto->amp >= 0)
2119                         new_afsi.amp = tad->cto->amp;
2120         }
2121         save_afsi(&new_afsi, &obj); /* in-place update */
2122         aced.aft_row = row;
2123         aced.old_afsi = &old_afsi;
2124         afs_event(AFSI_CHANGE, &tad->pb, &aced);
2125         return 1;
2126 }
2127
2128 static void com_touch_callback(int fd, const struct osl_object *query)
2129 {
2130         struct touch_action_data tad = {.cto = query->data,
2131                 .pb = {
2132                         .max_size = shm_get_shmmax(),
2133                         .private_data = &fd,
2134                         .max_size_handler = afs_max_size_handler
2135                 }
2136         };
2137         int ret, ret2 = 0;
2138         struct pattern_match_data pmd = {
2139                 .table = audio_file_table,
2140                 .loop_col_num = AFTCOL_HASH,
2141                 .match_col_num = AFTCOL_PATH,
2142                 .patterns = {.data = (char *)query->data + sizeof(*tad.cto),
2143                         .size = query->size - sizeof(*tad.cto)},
2144                 .data = &tad,
2145                 .action = touch_audio_file
2146         };
2147         if (tad.cto->flags & TOUCH_FLAG_FNM_PATHNAME)
2148                 pmd.fnmatch_flags |= FNM_PATHNAME;
2149         ret = for_each_matching_row(&pmd);
2150         if (ret < 0)
2151                 ret2 = para_printf(&tad.pb, "%s\n", para_strerror(-ret));
2152         else if (pmd.num_matches == 0)
2153                 ret2 = para_printf(&tad.pb, "no matches\n");
2154         if (ret2 >= 0 && tad.pb.offset)
2155                 pass_buffer_as_shm(fd, SBD_OUTPUT, tad.pb.buf, tad.pb.offset);
2156         free(tad.pb.buf);
2157 }
2158
2159 int com_touch(struct command_context *cc)
2160 {
2161         struct com_touch_options cto = {
2162                 .num_played = -1,
2163                 .last_played = -1,
2164                 .lyrics_id = -1,
2165                 .image_id = -1,
2166                 .amp = -1,
2167         };
2168         struct osl_object query = {.data = &cto, .size = sizeof(cto)};
2169         int i, ret;
2170
2171
2172         for (i = 1; i < cc->argc; i++) {
2173                 const char *arg = cc->argv[i];
2174                 if (arg[0] != '-')
2175                         break;
2176                 if (!strcmp(arg, "--")) {
2177                         i++;
2178                         break;
2179                 }
2180                 if (!strncmp(arg, "-n=", 3)) {
2181                         ret = para_atoi32(arg + 3, &cto.num_played);
2182                         if (ret < 0)
2183                                 return ret;
2184                         continue;
2185                 }
2186                 if (!strncmp(arg, "-l=", 3)) {
2187                         ret = para_atoi64(arg + 3, &cto.last_played);
2188                         if (ret < 0)
2189                                 return ret;
2190                         continue;
2191                 }
2192                 if (!strncmp(arg, "-y=", 3)) {
2193                         ret = para_atoi32(arg + 3, &cto.lyrics_id);
2194                         if (ret < 0)
2195                                 return ret;
2196                         continue;
2197                 }
2198                 if (!strncmp(arg, "-i=", 3)) {
2199                         ret = para_atoi32(arg + 3, &cto.image_id);
2200                         if (ret < 0)
2201                                 return ret;
2202                         continue;
2203                 }
2204                 if (!strncmp(arg, "-a=", 3)) {
2205                         int32_t val;
2206                         ret = para_atoi32(arg + 3, &val);
2207                         if (ret < 0)
2208                                 return ret;
2209                         if (val < 0 || val > 255)
2210                                 return -ERRNO_TO_PARA_ERROR(EINVAL);
2211                         cto.amp = val;
2212                         continue;
2213                 }
2214                 if (!strcmp(arg, "-p")) {
2215                         cto.flags |= TOUCH_FLAG_FNM_PATHNAME;
2216                         continue;
2217                 }
2218                 if (!strcmp(arg, "-v")) {
2219                         cto.flags |= TOUCH_FLAG_VERBOSE;
2220                         continue;
2221                 }
2222                 break; /* non-option starting with dash */
2223         }
2224         if (i >= cc->argc)
2225                 return -E_AFT_SYNTAX;
2226         ret = send_option_arg_callback_request(&query, cc->argc - i,
2227                 cc->argv + i, com_touch_callback, afs_cb_result_handler, cc);
2228         if (ret < 0)
2229                 send_strerror(cc, -ret);
2230         return ret;
2231 }
2232
2233 /** Flags for com_rm(). */
2234 enum rm_flags {
2235         /** -v */
2236         RM_FLAG_VERBOSE = 1,
2237         /** -f */
2238         RM_FLAG_FORCE = 2,
2239         /** -p */
2240         RM_FLAG_FNM_PATHNAME = 4
2241 };
2242
2243 /** Data passed to the action handler of com_rm(). */
2244 struct com_rm_action_data {
2245         /** Command line flags ((see \ref rm_flags). */
2246         uint32_t flags;
2247         /** Message buffer. */
2248         struct para_buffer pb;
2249 };
2250
2251 static int remove_audio_file(__a_unused struct osl_table *table,
2252                 struct osl_row *row, const char *name, void *data)
2253 {
2254         struct com_rm_action_data *crd = data;
2255         int ret;
2256
2257         if (crd->flags & RM_FLAG_VERBOSE) {
2258                 ret = para_printf(&crd->pb, "removing %s\n", name);
2259                 if (ret < 0)
2260                         return ret;
2261         }
2262         afs_event(AUDIO_FILE_REMOVE, &crd->pb, row);
2263         ret = osl(osl_del_row(audio_file_table, row));
2264         if (ret < 0)
2265                 para_printf(&crd->pb, "%s: %s\n", name, para_strerror(-ret));
2266         return ret;
2267 }
2268
2269 static void com_rm_callback(int fd, const struct osl_object *query)
2270 {
2271         struct com_rm_action_data crd = {.flags = *(uint32_t *)query->data,
2272                 .pb = {
2273                         .max_size = shm_get_shmmax(),
2274                         .private_data = &fd,
2275                         .max_size_handler = afs_max_size_handler
2276                 }
2277         };
2278         int ret;
2279         struct pattern_match_data pmd = {
2280                 .table = audio_file_table,
2281                 .loop_col_num = AFTCOL_HASH,
2282                 .match_col_num = AFTCOL_PATH,
2283                 .patterns = {.data = (char *)query->data + sizeof(uint32_t),
2284                         .size = query->size - sizeof(uint32_t)},
2285                 .data = &crd,
2286                 .action = remove_audio_file
2287         };
2288         if (crd.flags & RM_FLAG_FNM_PATHNAME)
2289                 pmd.fnmatch_flags |= FNM_PATHNAME;
2290         ret = for_each_matching_row(&pmd);
2291         if (ret < 0) {
2292                 para_printf(&crd.pb, "%s\n", para_strerror(-ret));
2293                 return;
2294         }
2295         if ((pmd.num_matches == 0) && !(crd.flags & RM_FLAG_FORCE))
2296                 ret = para_printf(&crd.pb, "no matches -- nothing removed\n");
2297         else if (crd.flags & RM_FLAG_VERBOSE) {
2298                 ret = para_printf(&crd.pb, "removed %u files\n",
2299                         pmd.num_matches);
2300         }
2301         if (ret >= 0 && crd.pb.offset)
2302                 pass_buffer_as_shm(fd, SBD_OUTPUT, crd.pb.buf, crd.pb.offset);
2303         free(crd.pb.buf);
2304 }
2305
2306 /* TODO options: -r (recursive) */
2307 int com_rm(struct command_context *cc)
2308 {
2309         uint32_t flags = 0;
2310         struct osl_object query = {.data = &flags, .size = sizeof(flags)};
2311         int i, ret;
2312
2313         for (i = 1; i < cc->argc; i++) {
2314                 const char *arg = cc->argv[i];
2315                 if (arg[0] != '-')
2316                         break;
2317                 if (!strcmp(arg, "--")) {
2318                         i++;
2319                         break;
2320                 }
2321                 if (!strcmp(arg, "-f")) {
2322                         flags |= RM_FLAG_FORCE;
2323                         continue;
2324                 }
2325                 if (!strcmp(arg, "-p")) {
2326                         flags |= RM_FLAG_FNM_PATHNAME;
2327                         continue;
2328                 }
2329                 if (!strcmp(arg, "-v")) {
2330                         flags |= RM_FLAG_VERBOSE;
2331                         continue;
2332                 }
2333                 break;
2334         }
2335         if (i >= cc->argc)
2336                 return -E_AFT_SYNTAX;
2337         ret = send_option_arg_callback_request(&query, cc->argc - i,
2338                 cc->argv + i, com_rm_callback, afs_cb_result_handler, cc);
2339         if (ret < 0)
2340                 send_strerror(cc, -ret);
2341         return ret;
2342 }
2343
2344 /**
2345  * Flags used by the cpsi command.
2346  *
2347  * \sa com_cpsi().
2348  */
2349 enum cpsi_flags {
2350         /** Whether the lyrics id should be copied. */
2351         CPSI_FLAG_COPY_LYRICS_ID = 1,
2352         /** Whether the image id should be copied. */
2353         CPSI_FLAG_COPY_IMAGE_ID = 2,
2354         /** Whether the lastplayed time should be copied. */
2355         CPSI_FLAG_COPY_LASTPLAYED = 4,
2356         /** Whether the numplayed count should be copied. */
2357         CPSI_FLAG_COPY_NUMPLAYED = 8,
2358         /** Whether the attributes should be copied. */
2359         CPSI_FLAG_COPY_ATTRIBUTES = 16,
2360         /** Activates verbose mode. */
2361         CPSI_FLAG_VERBOSE = 32,
2362 };
2363
2364 /** Data passed to the action handler of com_cpsi(). */
2365 struct cpsi_action_data {
2366         /** command line flags (see \ref cpsi_flags). */
2367         unsigned flags;
2368         /** Message buffer. */
2369         struct para_buffer pb;
2370         /** Values are copied from here. */
2371         struct afs_info source_afsi;
2372 };
2373
2374 static int copy_selector_info(__a_unused struct osl_table *table,
2375                 struct osl_row *row, const char *name, void *data)
2376 {
2377         struct cpsi_action_data *cad = data;
2378         struct osl_object target_afsi_obj;
2379         int ret;
2380         struct afs_info old_afsi, target_afsi;
2381         struct afsi_change_event_data aced;
2382
2383         ret = get_afsi_object_of_row(row, &target_afsi_obj);
2384         if (ret < 0)
2385                 return ret;
2386         load_afsi(&target_afsi, &target_afsi_obj);
2387         old_afsi = target_afsi;
2388         if (cad->flags & CPSI_FLAG_COPY_LYRICS_ID)
2389                 target_afsi.lyrics_id = cad->source_afsi.lyrics_id;
2390         if (cad->flags & CPSI_FLAG_COPY_IMAGE_ID)
2391                 target_afsi.image_id = cad->source_afsi.image_id;
2392         if (cad->flags & CPSI_FLAG_COPY_LASTPLAYED)
2393                 target_afsi.last_played = cad->source_afsi.last_played;
2394         if (cad->flags & CPSI_FLAG_COPY_NUMPLAYED)
2395                 target_afsi.num_played = cad->source_afsi.num_played;
2396         if (cad->flags & CPSI_FLAG_COPY_ATTRIBUTES)
2397                 target_afsi.attributes = cad->source_afsi.attributes;
2398         save_afsi(&target_afsi, &target_afsi_obj); /* in-place update */
2399         if (cad->flags & CPSI_FLAG_VERBOSE) {
2400                 ret = para_printf(&cad->pb, "copied afsi to %s\n", name);
2401                 if (ret < 0)
2402                         return ret;
2403         }
2404         aced.aft_row = row;
2405         aced.old_afsi = &old_afsi;
2406         afs_event(AFSI_CHANGE, &cad->pb, &aced);
2407         return 1;
2408 }
2409
2410 static void com_cpsi_callback(int fd, const struct osl_object *query)
2411 {
2412         struct cpsi_action_data cad = {
2413                 .flags = *(unsigned *)query->data,
2414                 .pb = {
2415                         .max_size = shm_get_shmmax(),
2416                         .private_data = &fd,
2417                         .max_size_handler = afs_max_size_handler
2418                 }
2419         };
2420         int ret;
2421         char *source_path = (char *)query->data + sizeof(cad.flags);
2422         struct pattern_match_data pmd = {
2423                 .table = audio_file_table,
2424                 .loop_col_num = AFTCOL_HASH,
2425                 .match_col_num = AFTCOL_PATH,
2426                 .patterns = {.data = source_path + strlen(source_path) + 1,
2427                         .size = query->size - sizeof(cad.flags)
2428                                 - strlen(source_path) - 1},
2429                 .data = &cad,
2430                 .action = copy_selector_info
2431         };
2432
2433         ret = get_afsi_of_path(source_path, &cad.source_afsi);
2434         if (ret < 0)
2435                 goto out;
2436         ret = for_each_matching_row(&pmd);
2437 out:
2438         if (ret < 0)
2439                 para_printf(&cad.pb, "%s\n", para_strerror(-ret));
2440         else if (cad.flags & CPSI_FLAG_VERBOSE) {
2441                 if (pmd.num_matches > 0)
2442                         para_printf(&cad.pb, "copied requested afsi from %s "
2443                                 "to %u files\n", source_path, pmd.num_matches);
2444                 else
2445                         para_printf(&cad.pb, "nothing copied\n");
2446         }
2447         if (cad.pb.offset)
2448                 pass_buffer_as_shm(fd, SBD_OUTPUT, cad.pb.buf, cad.pb.offset);
2449         free(cad.pb.buf);
2450 }
2451
2452 int com_cpsi(struct command_context *cc)
2453 {
2454         unsigned flags = 0;
2455         int i, ret;
2456         struct osl_object options = {.data = &flags, .size = sizeof(flags)};
2457
2458         for (i = 1; i < cc->argc; i++) {
2459                 const char *arg = cc->argv[i];
2460                 if (arg[0] != '-')
2461                         break;
2462                 if (!strcmp(arg, "--")) {
2463                         i++;
2464                         break;
2465                 }
2466                 if (!strcmp(arg, "-y")) {
2467                         flags |= CPSI_FLAG_COPY_LYRICS_ID;
2468                         continue;
2469                 }
2470                 if (!strcmp(arg, "-i")) {
2471                         flags |= CPSI_FLAG_COPY_IMAGE_ID;
2472                         continue;
2473                 }
2474                 if (!strcmp(arg, "-l")) {
2475                         flags |= CPSI_FLAG_COPY_LASTPLAYED;
2476                         continue;
2477                 }
2478                 if (!strcmp(arg, "-n")) {
2479                         flags |= CPSI_FLAG_COPY_NUMPLAYED;
2480                         continue;
2481                 }
2482                 if (!strcmp(arg, "-a")) {
2483                         flags |= CPSI_FLAG_COPY_ATTRIBUTES;
2484                         continue;
2485                 }
2486                 if (!strcmp(arg, "-v")) {
2487                         flags |= CPSI_FLAG_VERBOSE;
2488                         continue;
2489                 }
2490                 break;
2491         }
2492         if (i + 1 >= cc->argc) /* need at least source file and pattern */
2493                 return -E_AFT_SYNTAX;
2494         if (!(flags & ~CPSI_FLAG_VERBOSE)) /* no copy flags given */
2495                 flags = ~(unsigned)CPSI_FLAG_VERBOSE | flags;
2496         ret = send_option_arg_callback_request(&options, cc->argc - i,
2497                 cc->argv + i, com_cpsi_callback, afs_cb_result_handler, cc);
2498         if (ret < 0)
2499                 send_strerror(cc, -ret);
2500         return ret;
2501 }
2502
2503 static void afs_stat_callback(int fd, const struct osl_object *query)
2504 {
2505         int *parser_friendly = query->data;
2506         char *buf = *parser_friendly?
2507                 parser_friendly_status_items : status_items;
2508
2509         if (!buf)
2510                 return;
2511         pass_buffer_as_shm(fd, SBD_OUTPUT, buf, strlen(buf));
2512 }
2513
2514 /**
2515  * Get the current afs status items from the afs process and send it.
2516  *
2517  * \param cc The command context, used e.g. for data encryption.
2518  * \param parser_friendly Whether parser-friendly output format should be used.
2519  *
2520  * As the contents of the afs status items change in time and the command
2521  * handler only has a COW version created at fork time, it can not send
2522  * up-to-date afs status items directly. Therefore the usual callback mechanism
2523  * is used to pass the status items from the afs process to the command handler
2524  * via a shared memory area and a pipe.
2525  *
2526  * \return The return value of the underyling call to \ref send_callback_request().
2527  */
2528 int send_afs_status(struct command_context *cc, int parser_friendly)
2529 {
2530         struct osl_object query = {.data = &parser_friendly,
2531                 .size = sizeof(parser_friendly)};
2532
2533         return send_callback_request(afs_stat_callback, &query,
2534                 afs_cb_result_handler, cc);
2535 }
2536
2537 /* TODO: optionally fix problems by removing offending rows */
2538 static int check_audio_file(struct osl_row *row, void *data)
2539 {
2540         char *path;
2541         struct para_buffer *pb = data;
2542         struct stat statbuf;
2543         int ret = get_audio_file_path_of_row(row, &path);
2544         struct afs_info afsi;
2545         char *blob_name;
2546
2547         if (ret < 0)
2548                 return para_printf(pb, "%s\n", para_strerror(-ret));
2549         if (stat(path, &statbuf) < 0) {
2550                 ret = para_printf(pb, "%s: stat error (%s)\n", path, strerror(errno));
2551                 if (ret < 0)
2552                         return ret;
2553         } else {
2554                 if (!S_ISREG(statbuf.st_mode)) {
2555                         ret = para_printf(pb, "%s: not a regular file\n", path);
2556                         if (ret < 0)
2557                                 return ret;
2558                 }
2559         }
2560         ret = get_afsi_of_row(row, &afsi);
2561         if (ret < 0)
2562                 return para_printf(pb, "%s: %s\n", path, para_strerror(-ret));
2563         ret = lyr_get_name_by_id(afsi.lyrics_id, &blob_name);
2564         if (ret < 0) {
2565                 ret = para_printf(pb, "%s lyrics id %u: %s\n", path, afsi.lyrics_id,
2566                         para_strerror(-ret));
2567                 if (ret < 0)
2568                         return ret;
2569         }
2570         ret = img_get_name_by_id(afsi.image_id, &blob_name);
2571         if (ret < 0)
2572                 ret = para_printf(pb, "%s image id %u: %s\n", path, afsi.image_id,
2573                         para_strerror(-ret));
2574         return ret;
2575 }
2576
2577 /**
2578  * Check the audio file table for inconsistencies.
2579  *
2580  * \param fd The afs socket.
2581  * \param query Unused.
2582  *
2583  * This function always succeeds.
2584  *
2585  * \sa com_check().
2586  */
2587 void aft_check_callback(int fd, __a_unused const struct osl_object *query)
2588 {
2589         struct para_buffer pb = {
2590                 .max_size = shm_get_shmmax(),
2591                 .private_data = &(struct afs_max_size_handler_data) {
2592                         .fd = fd,
2593                         .band = SBD_OUTPUT
2594                 },
2595                 .max_size_handler = afs_max_size_handler
2596         };
2597         int ret = para_printf(&pb, "checking audio file table...\n");
2598
2599         if (ret < 0)
2600                 return;
2601         audio_file_loop(&pb, check_audio_file);
2602         if (pb.offset)
2603                 pass_buffer_as_shm(fd, SBD_OUTPUT, pb.buf, pb.offset);
2604         free(pb.buf);
2605 }
2606
2607 /**
2608  * Close the audio file table.
2609  *
2610  * \param flags Usual flags that are passed to osl_close_table().
2611  *
2612  * \sa osl_close_table().
2613  */
2614 static void aft_close(void)
2615 {
2616         osl_close_table(audio_file_table, OSL_MARK_CLEAN);
2617         audio_file_table = NULL;
2618 }
2619
2620 /**
2621  * Open the audio file table.
2622  *
2623  * \param dir The database directory.
2624  *
2625  * \return Standard.
2626  *
2627  * \sa osl_open_table().
2628  */
2629 static int aft_open(const char *dir)
2630 {
2631         int ret;
2632
2633         audio_file_table_desc.dir = dir;
2634         ret = osl(osl_open_table(&audio_file_table_desc, &audio_file_table));
2635         if (ret >= 0) {
2636                 unsigned num;
2637                 osl_get_num_rows(audio_file_table, &num);
2638                 PARA_INFO_LOG("audio file table contains %d files\n", num);
2639                 return ret;
2640         }
2641         PARA_INFO_LOG("failed to open audio file table\n");
2642         audio_file_table = NULL;
2643         if (ret >= 0 || ret == -OSL_ERRNO_TO_PARA_ERROR(E_OSL_NOENT))
2644                 return 1;
2645         return ret;
2646 }
2647
2648 static int aft_create(const char *dir)
2649 {
2650         audio_file_table_desc.dir = dir;
2651         return osl(osl_create_table(&audio_file_table_desc));
2652 }
2653
2654 static int clear_attribute(struct osl_row *row, void *data)
2655 {
2656         struct rmatt_event_data *red = data;
2657         struct afs_info afsi;
2658         struct osl_object obj;
2659         int ret = get_afsi_object_of_row(row, &obj);
2660         uint64_t mask = ~(1ULL << red->bitnum);
2661
2662         if (ret < 0)
2663                 return ret;
2664         ret = load_afsi(&afsi, &obj);
2665         if (ret < 0)
2666                 return ret;
2667         afsi.attributes &= mask;
2668         save_afsi(&afsi, &obj);
2669         return 1;
2670 }
2671
2672 static int aft_event_handler(enum afs_events event, struct para_buffer *pb,
2673                 void *data)
2674 {
2675         int ret;
2676
2677         switch(event) {
2678         case ATTRIBUTE_REMOVE: {
2679                 const struct rmatt_event_data *red = data;
2680                 ret = para_printf(pb, "clearing attribute %s (bit %u) from all "
2681                         "entries in the audio file table\n", red->name,
2682                         red->bitnum);
2683                 if (ret < 0)
2684                         return ret;
2685                 return audio_file_loop(data, clear_attribute);
2686                 }
2687         default:
2688                 return 1;
2689         }
2690 }
2691
2692 /**
2693  * Initialize the audio file table.
2694  *
2695  * \param t Pointer to the structure to be initialized.
2696  */
2697 void aft_init(struct afs_table *t)
2698 {
2699         t->open = aft_open;
2700         t->close = aft_close;
2701         t->create = aft_create;
2702         t->event_handler = aft_event_handler;
2703 }