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