2 * Copyright (C) 1999-2007 Andre Noll <maan@systemlinux.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
19 /** \file mysql_selector.c para_server's mysql-based audio file selector */
21 /** \cond some internal constants */
22 #define MEDIUM_BLOB_SIZE 16777220 /* (2**24 + 4) */
23 #define BLOB_SIZE 65539 /* (2**16 + 3) */
25 #include "server.cmdline.h"
29 #include <mysql/mysql.h>
30 #include <mysql/mysql_version.h>
35 #include "user_list.h"
36 #include "mysql_selector_command_list.h"
38 /** pointer to the shared memory area */
39 extern struct misc_meta_data *mmd;
41 static void *mysql_ptr = NULL;
44 * contains name/replacement pairs used by s_a_r_list()
49 /** the name of the macro */
51 /** the replacement text */
52 const char *replacement;
55 static const struct para_macro mysql_macro_list[] = {
57 .replacement = "(data.%s != '1')"
60 .replacement = "(data.%s = '1')"
63 .replacement = "%sdata.Pic_Id"
66 .replacement = "(data.name like '%s')"
69 .replacement = "%sFLOOR((UNIX_TIMESTAMP(now())"
70 "-UNIX_TIMESTAMP(data.Lastplayed))/60)"
73 .replacement = "%sdata.Numplayed"
80 * simple search and replace routine
82 * \param src source string
83 * \param macro_name the name of the macro
84 * \param replacement the replacement format string
86 * In \p src, replace each occurence of \p macro_name(arg) by the string
87 * determined by the \p replacement format string. \p replacement may (but
88 * needs not) contain a single string conversion specifier (%s) which gets
91 * \return A string in which all matches in \p src are replaced, or \p NULL if
92 * an syntax error was encountered. Caller must free the result.
96 __must_check __malloc static char *s_a_r(const char *src, const char* macro_name,
97 const char *replacement)
101 regmatch_t pmatch[1];
104 const char *bufptr = src;
106 if (!macro_name || !replacement || !src)
107 return para_strdup(src);
108 regcomp(&preg, macro_name, 0);
109 while (regexec(&preg, bufptr, nmatch, pmatch, eflags)
111 char *tmp, *arg, *o_bracket, *c_bracket;
113 o_bracket = strchr(bufptr + pmatch[0].rm_so, '(');
114 c_bracket = o_bracket? strchr(o_bracket, ')') : NULL;
117 tmp = para_strdup(bufptr);
118 tmp[pmatch[0].rm_so] = '\0';
119 dest = para_strcat(dest, tmp);
122 arg = para_strdup(o_bracket + 1);
123 arg[c_bracket - o_bracket - 1] = '\0';
124 tmp = make_message(replacement, arg);
126 dest = para_strcat(dest, tmp);
131 dest = para_strcat(dest, bufptr);
132 // PARA_DEBUG_LOG("%s: returning %s\n", __func__, dest);
139 * replace a string according to a list of macros
141 * \param macro_list the array containing a macro/replacement pairs.
142 * \param src the source string
144 * This function just calls s_a_r() for each element of \p macro_list.
146 * \return \p NULL if one of the underlying calls to \p s_a_r returned \p NULL.
147 * Otherwise the completely expanded version of \p src is returned.
149 __must_check __malloc static char *s_a_r_list(const struct para_macro *macro_list,
152 const struct para_macro *mp = macro_list;
153 char *ret = NULL, *tmp = para_strdup(src);
156 ret = s_a_r(tmp, mp->name, mp->replacement);
158 if (!ret) /* syntax error */
163 //PARA_DEBUG_LOG("%s: returning %s\n", __func__, dest);
167 static int real_query(const char *query)
171 PARA_DEBUG_LOG("%s\n", query);
172 if (mysql_real_query(mysql_ptr, query, strlen(query))) {
173 PARA_ERROR_LOG("real_query error (%s)\n",
174 mysql_error(mysql_ptr));
181 * Use open connection given by mysql_ptr to query server. Returns a
182 * result pointer on succes and NULL on errors
184 static struct MYSQL_RES *get_result(const char *query)
188 if (real_query(query) < 0)
190 result = mysql_store_result(mysql_ptr);
192 PARA_ERROR_LOG("%s", "store_result error\n");
196 * write input from fd to dynamically allocated char array,
197 * but maximal max_size byte. Return size.
199 static int fd2buf(int fd, char **buf_ptr, size_t max_size)
201 const size_t chunk_size = 1024;
203 char *buf = para_malloc(size * sizeof(char)), *p = buf;
206 while ((ret = recv_bin_buffer(fd, p, chunk_size)) > 0) {
208 if ((p - buf) + chunk_size >= size) {
212 if (size > max_size) {
216 tmp = para_realloc(buf, size);
231 static char *escape_blob(const char* old, int size)
235 if (!mysql_ptr || size < 0)
237 new = para_malloc(2 * size * sizeof(char) + 1);
238 mysql_real_escape_string(mysql_ptr, new, old, size);
242 static char *escape_str(const char* old)
244 return escape_blob(old, strlen(old));
247 static char *escaped_basename(const char *name)
249 char *esc, *bn = para_basename(name);
253 esc = escape_str(bn);
261 int com_na(__a_unused int fd, int argc, char *argv[])
267 return -E_MYSQL_SYNTAX;
268 tmp = escape_str(argv[1]);
271 q = make_message("alter table data add %s char(1) "
272 "not null default 0", tmp);
282 int com_da(__a_unused int fd, int argc, char *argv[])
288 return -E_MYSQL_SYNTAX;
289 tmp = escape_str(argv[1]);
292 q = make_message("alter table data drop %s", tmp);
300 static int com_stradd_picadd(int fd, int argc, char *argv[])
302 char *blob = NULL, *esc_blob = NULL, *q = NULL, *tmp = NULL;
303 const char *fmt, *del_fmt;
304 int ret, stradd = strcmp(argv[0], "picadd");
308 return -E_MYSQL_SYNTAX;
309 if (strlen(argv[1]) >= MAXLINE - 1)
310 return -E_NAMETOOLONG;
315 fmt = "insert into streams (name, def) values ('%s','%s')";
316 del_fmt="delete from streams where name='%s'";
318 size = MEDIUM_BLOB_SIZE;
319 fmt = "insert into pics (name, pic) values ('%s','%s')";
320 del_fmt="delete from pics where pic='%s'";
322 tmp = escape_str(argv[1]);
325 q = make_message(del_fmt, tmp);
331 if ((ret = send_buffer(fd, AWAITING_DATA_MSG) < 0))
333 if ((ret = fd2buf(fd, &blob, size)) < 0)
339 esc_blob = escape_blob(blob, size);
342 tmp = escape_str(argv[1]);
345 q = make_message(fmt, tmp, esc_blob);
356 int com_stradd(int fd, int argc, char *argv[])
358 return com_stradd_picadd(fd, argc, argv);
362 int com_picadd(int fd, int argc, char *argv[])
364 return com_stradd_picadd(fd, argc, argv);
368 * print results to fd
370 static int print_results(int fd, void *result,
371 unsigned int top, unsigned int left,
372 unsigned int bottom, unsigned int right)
378 for (i = top; i <= bottom; i++) {
379 row = mysql_fetch_row(result);
382 for (j = left; j <= right; j++) {
383 ret = send_va_buffer(fd, j == left? "%s" : "\t%s",
384 row[j]? row[j] : "NULL");
388 ret = send_buffer(fd, "\n");
398 int com_verb(int fd, int argc, char *argv[])
402 unsigned int num_rows, num_fields;
406 return -E_MYSQL_SYNTAX;
407 tmp = escape_str(argv[1]);
410 result = get_result(tmp);
413 /* return success, because it's ok to have no results */
415 num_fields = mysql_field_count(mysql_ptr);
416 num_rows = mysql_num_rows(result);
418 if (num_fields && num_rows)
419 ret = print_results(fd, result, 0, 0, num_rows - 1,
421 mysql_free_result(result);
425 /* returns NULL on errors or if there are no atts defined yet */
426 static void *get_all_attributes(void)
428 void *result = get_result("desc data");
429 unsigned int num_rows;
433 num_rows = mysql_num_rows(result);
435 mysql_free_result(result);
438 mysql_data_seek(result, 4); /* skip Lastplayed, Numplayed... */
443 * list all attributes
445 int com_laa(int fd, int argc, __a_unused char *argv[])
451 return -E_MYSQL_SYNTAX;
452 result = get_all_attributes();
455 ret = print_results(fd, result, 0, 0, mysql_num_rows(result) - 5, 0);
456 mysql_free_result(result);
463 int com_hist(int fd, int argc, char *argv[])
468 unsigned int num_rows;
471 return -E_MYSQL_SYNTAX;
473 char *tmp = escape_str(argv[1]);
476 atts = make_message("where %s = '1'", tmp);
479 atts = para_strdup(NULL);
481 q = make_message("select name, to_days(now()) - to_days(lastplayed) from "
482 "data %s order by lastplayed", atts);
484 result = get_result(q);
488 num_rows = mysql_num_rows(result);
491 ret = print_results(fd, result, 0, 0, num_rows - 1, 1);
492 mysql_free_result(result);
497 * get last num audio files
499 int com_last(int fd, int argc, char *argv[])
510 return -E_MYSQL_SYNTAX;
511 q = make_message("select name from data order by lastplayed desc "
513 result = get_result(q);
517 ret = print_results(fd, result, 0, 0, mysql_num_rows(result) - 1, 0);
518 mysql_free_result(result);
522 int com_mbox(int fd, int argc, char *argv[])
527 unsigned int num_rows, num_fields;
528 char *query = para_strdup("select concat('From foo@localhost ', "
529 "date_format(Lastplayed, '%a %b %e %T %Y'), "
530 "'\nReceived: from\nTo: bar\n");
533 result = get_all_attributes();
537 while ((row = mysql_fetch_row(result))) {
542 tmp = make_message("%sX-Attribute-%s: ', %s, '\n", query,
547 query = para_strcat(query,
555 char *esc = escape_str(argv[1]), *tmp;
559 tmp = make_message("%s where name LIKE '%s'", query, esc);
564 mysql_free_result(result);
566 result = get_result(query);
569 ret = -E_EMPTY_RESULT;
570 num_fields = mysql_field_count(mysql_ptr);
571 num_rows = mysql_num_rows(result);
572 if (!num_fields || !num_rows)
574 ret = print_results(fd, result, 0, 0, num_rows - 1, num_fields - 1);
578 mysql_free_result(result);
583 * get attributes by name. If verbose is not 0, this function returns a string
584 * of the form 'att1="0",att2="1"'... which is used in com_cam() for
585 * constructing a mysql update query. Otherwise the space-separated list of all
586 * attributes which are set in the audio file given by name is returned. Never
587 * returns NULL in *NON VERBOSE* mode.
589 static char *get_atts(char *name, int verbose)
591 char *atts = NULL, *buf, *ebn;
592 void *result = NULL, *result2 = NULL;
597 result2 = get_all_attributes();
600 ebn = escaped_basename(name);
603 buf = make_message("select * from data where name='%s'", ebn);
605 result = get_result(buf);
609 num_fields = mysql_num_fields(result);
612 mysql_data_seek(result2, 4); /* skip Lastplayed, Numplayed... */
613 row = mysql_fetch_row(result);
616 for (i = 4; i < num_fields; i++) {
617 int is_set = row[i] && !strcmp(row[i], "1");
618 row2 = mysql_fetch_row(result2);
619 if (!row2 || !row2[0])
621 if (atts && (verbose || is_set))
622 atts = para_strcat(atts, verbose? "," : " ");
623 if (is_set || verbose)
624 atts = para_strcat(atts, row2[0]);
626 atts = para_strcat(atts, is_set? "=\"1\"" : "=\"0\"");
630 mysql_free_result(result2);
632 mysql_free_result(result);
633 if (!atts && !verbose)
634 atts = para_strdup("(none)");
638 /* never returns NULL in verbose mode */
639 static char *get_meta(char *name, int verbose)
643 char *ebn, *q, *ret = NULL;
644 const char *verbose_fmt =
645 "select concat('lastplayed: ', "
646 "(to_days(now()) - to_days(lastplayed)),"
647 "' day(s). numplayed: ', numplayed, "
648 "', pic: ', pic_id) "
649 "from data where name = '%s'";
650 /* is that really needed? */
651 const char *fmt = "select concat('lastplayed=\\'', lastplayed, "
652 "'\\', numplayed=\\'', numplayed, "
653 "'\\', pic_id=\\'', pic_id, '\\'') "
654 "from data where name = '%s'";
656 if (!(ebn = escaped_basename(name)))
658 q = make_message(verbose? verbose_fmt : fmt, ebn);
660 result = get_result(q);
664 row = mysql_fetch_row(result);
667 ret = para_strdup(row[0]);
670 mysql_free_result(result);
672 ret = para_strdup("(not yet played)");
676 static char *get_dir(char *name)
678 char *ret = NULL, *q, *ebn;
682 if (!(ebn = escaped_basename(name)))
684 q = make_message("select dir from dir where name = '%s'", ebn);
686 result = get_result(q);
690 row = mysql_fetch_row(result);
692 ret = para_strdup(row[0]);
693 mysql_free_result(result);
697 /* never returns NULL */
698 static char *get_current_stream(void)
702 void *result = get_result("select def from streams where "
703 "name = 'current_stream'");
707 row = mysql_fetch_row(result);
710 ret = para_strdup(row[0]);
711 mysql_free_result(result);
715 mysql_free_result(result);
716 return para_strdup("(none)");
720 * Read stream definition of stream streamname and construct mysql
721 * query. Return NULL on errors. If streamname is NULL, use current
722 * stream. If that is also NULL, use query that selects everything.
723 * If filename is NULL, query will list everything, otherwise only
724 * the score of given file.
726 static char *get_query(char *streamname, char *filename, int with_path)
728 char *accept_opts = NULL, *deny_opts = NULL, *score = NULL;
729 char *where_clause, *order, *query;
730 char command[255] = ""; /* buffer for sscanf */
734 char *select_clause = NULL;
736 tmp = get_current_stream();
738 tmp = escape_str(streamname);
742 if (!strcmp(tmp, "(none)")) {
745 char *ret, *ebn = escaped_basename(filename);
748 ret = make_message("select to_days(now()) - "
749 "to_days(lastplayed) from data "
750 "where name = '%s'", ebn);
756 "select concat(dir.dir, '/', dir.name) "
757 "from data, dir where dir.name = data.name "
758 "order by data.lastplayed"
761 "select name from data where name is not NULL "
762 "order by lastplayed"
766 query = make_message("select def from streams where name = '%s'",
768 result = get_result(query);
773 row = mysql_fetch_row(result);
779 char *arg, *line = end;
781 if (!(end = strchr(line, '\n')))
785 if (sscanf(line, "%200s%n", command, &n) < 1)
788 if (!strcmp(command, "accept:")) {
789 char *tmp2 = s_a_r_list(mysql_macro_list, arg);
791 accept_opts = para_strcat(
792 accept_opts, " or ");
793 accept_opts = para_strcat(accept_opts, tmp2);
797 if (!strcmp(command, "deny:")) {
798 char *tmp2 = s_a_r_list(mysql_macro_list, arg);
800 deny_opts = para_strcat(deny_opts, " or ");
801 deny_opts = para_strcat(deny_opts, tmp2);
805 if (!score && !strcmp(command, "score:"))
806 score = s_a_r_list(mysql_macro_list, arg);
809 score = s_a_r_list(mysql_macro_list, conf.mysql_default_score_arg);
814 char *ebn = escaped_basename(filename);
817 select_clause = make_message("select %s from data ", score);
819 where_clause = make_message( "where name = '%s' ", ebn);
821 order = para_strdup("");
824 select_clause = para_strdup(with_path?
825 "select concat(dir.dir, '/', dir.name) from data, dir "
826 "where dir.name = data.name "
828 "select name from data where name is not NULL");
829 order = make_message("order by -(%s)", score);
831 if (accept_opts && deny_opts) {
832 where_clause = make_message("and ((%s) and not (%s)) ",
833 accept_opts, deny_opts);
836 if (accept_opts && !deny_opts) {
837 where_clause = make_message("and (%s) ", accept_opts);
840 if (!accept_opts && deny_opts) {
841 where_clause = make_message("and not (%s) ", deny_opts);
844 where_clause = para_strdup("");
846 query = make_message("%s %s %s", select_clause, where_clause, order);
856 mysql_free_result(result);
863 * This is called from server and from some commands. Name must not be NULL
864 * Never returns NULL.
866 static char *get_selector_info(char *name)
868 char *meta, *atts, *info, *dir, *query, *stream;
870 MYSQL_ROW row = NULL;
873 return para_strdup("(none)");
874 stream = get_current_stream();
875 meta = get_meta(name, 1);
876 atts = get_atts(name, 0);
879 query = get_query(stream, name, 0); /* FIXME: pass stream == NULL instead? */
882 result = get_result(query);
885 row = mysql_fetch_row(result);
887 info = make_message("dbinfo1:dir: %s\n"
888 "dbinfo2:stream: %s, %s, score: %s\n"
890 dir? dir : "(not contained in table)",
892 (result && row && row[0])? row[0] : "(no score)",
899 mysql_free_result(result);
903 /* might return NULL */
904 static char *get_current_audio_file(void)
908 name = para_basename(mmd->filename);
913 /* If called as child, mmd_lock must be held */
914 static void update_mmd(char *info)
916 PARA_DEBUG_LOG("%s", "updating shared memory area\n");
917 strncpy(mmd->selector_info, info, MMD_INFO_SIZE - 1);
918 mmd->selector_info[MMD_INFO_SIZE - 1] = '\0';
921 static void refresh_selector_info(void)
923 char *name = get_current_audio_file();
928 info = get_selector_info(name);
936 /* list attributes / print database info */
937 static int com_la_info(int fd, int argc, char *argv[])
939 char *name = NULL, *meta = NULL, *atts = NULL, *dir = NULL;
940 int ret, la = strcmp(argv[0], "info");
943 ret = -E_GET_AUDIO_FILE;
944 if (!(name = get_current_audio_file()))
946 ret = send_va_buffer(fd, "%s\n", name);
951 if (!(name = escaped_basename(argv[1])))
954 meta = get_meta(name, 1);
955 atts = get_atts(name, 0);
958 ret = send_va_buffer(fd, "%s\n", atts);
960 ret = send_va_buffer(fd, "dir: %s\n" "%s\n" "attributes: %s\n",
961 dir? dir : "(not contained in table)", meta, atts);
970 /* list attributes */
971 int com_la(int fd, int argc, char *argv[])
973 return com_la_info(fd, argc, argv);
976 /* print database info */
977 int com_info(int fd, int argc, char *argv[])
979 return com_la_info(fd, argc, argv);
982 static int change_stream(const char *stream)
986 query = make_message("update streams set def='%s' "
987 "where name = 'current_stream'", stream);
988 ret = real_query(query);
993 static int get_pic_id_by_name(char *name)
1000 if (!(ebn = escaped_basename(name)))
1002 q = make_message("select pic_id from data where name = '%s'", ebn);
1004 result = get_result(q);
1008 row = mysql_fetch_row(result);
1012 mysql_free_result(result);
1016 static int remove_entry(const char *name)
1018 char *q, *ebn = escaped_basename(name);
1019 int ret = -E_ESCAPE;
1023 q = make_message("delete from data where name = '%s'", ebn);
1024 real_query(q); /* ignore errors */
1026 q = make_message("delete from dir where name = '%s'", ebn);
1027 real_query(q); /* ignore errors */
1035 static int add_entry(const char *name)
1037 char *q, *dn, *ebn = NULL, *edn = NULL;
1040 if (!name || !*name)
1041 return -E_MYSQL_SYNTAX;
1042 ebn = escaped_basename(name);
1045 ret = -E_MYSQL_SYNTAX;
1046 dn = para_dirname(name);
1050 edn = escape_str(dn);
1054 q = make_message("insert into data (name, pic_id) values "
1055 "('%s', '%s')", ebn, "1");
1056 ret = real_query(q);
1057 // ret = 1; PARA_DEBUG_LOG("q: %s\n", q);
1061 q = make_message("insert into dir (name, dir) values "
1062 "('%s', '%s')", ebn, edn);
1063 // ret = 1; PARA_DEBUG_LOG("q: %s\n", q);
1064 ret = real_query(q);
1075 * remove/add entries
1077 static int com_rm_ne(__a_unused int fd, int argc, char *argv[])
1079 int ne = !strcmp(argv[0], "ne");
1082 return -E_MYSQL_SYNTAX;
1083 for (i = 1; i < argc; i++) {
1084 ret = remove_entry(argv[i]);
1089 ret = add_entry(argv[i]);
1099 int com_rm(int fd, int argc, char *argv[])
1101 return com_rm_ne(fd, argc, argv);
1107 int com_ne(int fd, int argc, char *argv[])
1109 return com_ne(fd, argc, argv);
1115 int com_mv(__a_unused int fd, int argc, char *argv[])
1117 char *q, *dn, *ebn1 = NULL, *ebn2 = NULL, *edn = NULL;
1121 return -E_MYSQL_SYNTAX;
1123 ebn1 = escaped_basename(argv[1]);
1124 ebn2 = escaped_basename(argv[2]);
1125 if (!ebn1 || !ebn2 || !*ebn1 || !*ebn2)
1127 ret = -E_MYSQL_SYNTAX;
1128 if (!strcmp(ebn1, ebn2))
1130 remove_entry(argv[2]); /* no need to escape, ignore error */
1131 q = make_message("update data set name = '%s' where name = '%s'",
1133 ret = real_query(q);
1137 ret = -E_AUDIO_FILE;
1138 if (!mysql_affected_rows(mysql_ptr))
1140 q = make_message("update dir set name = '%s' where name = '%s'",
1142 ret = real_query(q);
1148 dn = para_dirname(argv[2]);
1152 edn = escape_str(dn);
1159 q = make_message("update dir set dir = '%s' where name = '%s'",
1161 ret = real_query(q);
1173 static int com_set(__a_unused int fd, int argc, char *argv[])
1178 const char *field = strcmp(argv[0], "picass")? "numplayed" : "pic_id";
1181 return -E_MYSQL_SYNTAX;
1183 for (i = 2; i < argc; i++) {
1184 ebn = escaped_basename(argv[i]);
1187 q = make_message("update data set %s = %lu "
1188 "where name = '%s'", field, id, ebn);
1190 ret = real_query(q);
1199 * snp: set numplayed
1201 int com_picass(int fd, int argc, char *argv[])
1203 return com_set(fd, argc, argv);
1207 * snp: set numplayed
1209 int com_snp(int fd, int argc, char *argv[])
1211 int ret = com_set(fd, argc, argv);
1213 refresh_selector_info();
1218 * picch: change entry's name in pics table
1220 int com_picch(__a_unused int fd, int argc, char *argv[])
1227 return -E_MYSQL_SYNTAX;
1230 tmp = escape_str(argv[2]);
1233 q = make_message("update pics set name = '%s' where id = %lu", tmp, id);
1235 ret = real_query(q);
1241 * piclist: print list of pics in db
1243 int com_piclist(__a_unused int fd, int argc, __a_unused char *argv[])
1245 void *result = NULL;
1247 unsigned long *length;
1251 return -E_MYSQL_SYNTAX;
1252 result = get_result("select id,name,pic from pics order by id");
1255 while ((row = mysql_fetch_row(result))) {
1256 length = mysql_fetch_lengths(result);
1257 if (!row || !row[0] || !row[1] || !row[2])
1259 ret = send_va_buffer(fd, "%s\t%lu\t%s\n", row[0], length[2], row[1]);
1265 mysql_free_result(result);
1270 * picdel: delete picture from database
1272 int com_picdel(int fd, int argc, char *argv[])
1280 return -E_MYSQL_SYNTAX;
1281 for (i = 1; i < argc; i++) {
1283 q = make_message("delete from pics where id = %lu", id);
1284 ret = real_query(q);
1288 aff = mysql_affected_rows(mysql_ptr);
1290 ret = send_va_buffer(fd, "No such id: %lu\n", id);
1295 q = make_message("update data set pic_id = 1 where pic_id = %lu", id);
1296 ret = real_query(q);
1302 * pic: get picture by name or by number
1304 int com_pic(int fd, int argc, char *argv[])
1306 void *result = NULL;
1308 unsigned long *length, id;
1310 char *q, *name = NULL;
1313 ret = -E_GET_AUDIO_FILE;
1314 name = get_current_audio_file();
1317 name = escaped_basename(argv[1]);
1322 id = atoi(name + 1);
1324 id = get_pic_id_by_name(name);
1328 q = make_message("select pic from pics where id = '%lu'", id);
1329 result = get_result(q);
1333 row = mysql_fetch_row(result);
1335 if (!row || !row[0])
1337 length = mysql_fetch_lengths(result);
1338 ret = send_bin_buffer(fd, row[0], *length);
1340 mysql_free_result(result);
1345 int com_strdel(__a_unused int fd, int argc, char *argv[])
1351 return -E_MYSQL_SYNTAX;
1352 tmp = escape_str(argv[1]);
1355 q = make_message("delete from streams where name='%s'", tmp);
1357 ret = real_query(q);
1365 int com_ls(int fd, int argc, char *argv[])
1370 unsigned int num_rows;
1373 return -E_MYSQL_SYNTAX;
1375 char *tmp = escape_str(argv[1]);
1378 q = make_message("select name from data where name like '%s'",
1382 q = para_strdup("select name from data");
1383 result = get_result(q);
1387 num_rows = mysql_num_rows(result);
1390 ret = print_results(fd, result, 0, 0, num_rows - 1, 0);
1391 mysql_free_result(result);
1398 int com_summary(__a_unused int fd, int argc, __a_unused char *argv[])
1403 void *result2 = NULL;
1404 const char *fmt = "select count(name) from data where %s='1'";
1405 int ret = -E_NORESULT;
1408 return -E_MYSQL_SYNTAX;
1409 result = get_all_attributes();
1412 while ((row = mysql_fetch_row(result))) {
1419 buf = make_message(fmt, row[0]);
1420 result2 = get_result(buf);
1425 row2 = mysql_fetch_row(result2);
1426 if (!row2 || !row2[0])
1428 ret = send_va_buffer(fd, "%s\t%s\n", row[0], row2[0]);
1435 mysql_free_result(result2);
1437 mysql_free_result(result);
1441 static int get_numplayed(char *name)
1445 const char *fmt = "select numplayed from data where name = '%s'";
1446 char *buf = make_message(fmt, name);
1447 int ret = -E_NORESULT;
1449 result = get_result(buf);
1454 row = mysql_fetch_row(result);
1455 if (!row || !row[0])
1460 mysql_free_result(result);
1464 static int update_audio_file(char *name)
1467 const char *fmt1 = "update data set lastplayed = now() where name = '%s'";
1468 const char *fmt2 = "update data set numplayed = %i where name = '%s'";
1470 char *ebn = escaped_basename(name);
1475 q = make_message(fmt1, ebn);
1476 ret = real_query(q);
1480 ret = get_numplayed(ebn);
1483 q = make_message(fmt2, ret + 1, ebn);
1484 ret = real_query(q);
1491 static void update_audio_file_server_handler(char *name)
1494 info = get_selector_info(name);
1497 update_audio_file(name);
1500 int com_us(__a_unused int fd, int argc, char *argv[])
1506 return -E_MYSQL_SYNTAX;
1507 tmp = escape_str(argv[1]);
1510 ret = update_audio_file(argv[1]);
1513 refresh_selector_info();
1517 /* select previous / next stream */
1518 static int com_ps_ns(__a_unused int fd, int argc, char *argv[])
1520 char *query, *stream = get_current_stream();
1521 void *result = get_result("select name from streams");
1523 int match = -1, ret, i;
1524 unsigned int num_rows;
1527 return -E_MYSQL_SYNTAX;
1531 num_rows = mysql_num_rows(result);
1532 ret = -E_EMPTY_RESULT;
1536 for (i = 0; i < num_rows; i++) {
1537 row = mysql_fetch_row(result);
1538 if (!row || !row[0])
1540 if (!strcmp(row[0], "current_stream"))
1542 if (!strcmp(row[0], stream)) {
1550 if (!strcmp(argv[0], "ps"))
1551 i = match > 0? match - 1 : num_rows - 1;
1553 i = match < num_rows - 1? match + 1 : 0;
1555 mysql_data_seek(result, i);
1556 row = mysql_fetch_row(result);
1557 if (!row || !row[0])
1559 if (!strcmp(row[0], "current_stream")) {
1560 if (!strcmp(argv[0], "ps")) {
1562 i = i < 0? i + num_rows : i;
1565 i = i > num_rows - 1? i - num_rows : i;
1567 mysql_data_seek(result, i);
1568 row = mysql_fetch_row(result);
1569 if (!row || !row[0])
1572 query = make_message("update streams set def='%s' where name = "
1573 "'current_stream'", row[0]);
1574 ret = real_query(query);
1576 refresh_selector_info();
1580 mysql_free_result(result);
1584 /* select previous stream */
1585 int com_ps(int fd, int argc, char *argv[])
1587 return com_ps_ns(fd, argc, argv);
1590 /* select next stream */
1591 int com_ns(int fd, int argc, char *argv[])
1593 return com_ps_ns(fd, argc, argv);
1597 int com_streams(int fd, int argc, __a_unused char *argv[])
1599 unsigned int num_rows;
1600 int i, ret = -E_NORESULT;
1604 if (argc > 1 && strcmp(argv[1], "current_stream"))
1605 return -E_MYSQL_SYNTAX;
1607 char *cs = get_current_stream();
1608 ret = send_va_buffer(fd, "%s\n", cs);
1612 result = get_result("select name from streams");
1615 num_rows = mysql_num_rows(result);
1620 for (i = 0; i < num_rows; i++) {
1621 row = mysql_fetch_row(result);
1622 if (!row || !row[0])
1624 if (strcmp(row[0], "current_stream"))
1625 send_va_buffer(fd, "%s\n", row[0]);
1630 mysql_free_result(result);
1634 /* query stream definition */
1635 int com_strq(int fd, int argc, char *argv[])
1643 ret = -E_GET_STREAM;
1644 name = get_current_stream();
1647 name = escaped_basename(argv[1]);
1652 query = make_message("select def from streams where name='%s'", name);
1654 result = get_result(query);
1659 row = mysql_fetch_row(result);
1660 if (!row || !row[0])
1662 /* no '\n' needed */
1663 ret = send_buffer(fd, row[0]);
1666 mysql_free_result(result);
1670 /* change stream / change stream and play */
1671 static int com_cs_csp(int fd, int argc, char *argv[])
1673 int ret, stream_change;
1674 char *query, *stream = NULL;
1675 char *old_stream = get_current_stream();
1676 int csp = !strcmp(argv[0], "csp");
1678 ret = -E_MYSQL_SYNTAX;
1684 ret = send_va_buffer(fd, "%s\n", old_stream);
1688 /* test if stream is valid, no need to escape argv[1] */
1689 query = get_query(argv[1], NULL, 0);
1694 stream = escape_str(argv[1]);
1697 stream_change = strcmp(stream, old_stream);
1698 if (stream_change) {
1699 ret = change_stream(stream);
1702 refresh_selector_info();
1706 mmd->new_vss_status_flags |= VSS_PLAYING;
1708 mmd->new_vss_status_flags |= VSS_NEXT;
1719 int com_cs(int fd, int argc, char *argv[])
1721 return com_cs_csp(fd, argc, argv);
1724 /* change stream and play */
1725 int com_csp(int fd, int argc, char *argv[])
1727 return com_cs_csp(fd, argc, argv);
1730 /* score list / skip */
1731 static int com_sl_skip(int fd, int argc, char *argv[])
1733 void *result = NULL;
1735 int ret, i, skip = !strcmp(argv[0], "skip");
1736 char *query, *stream, *tmp;
1737 unsigned int num_rows, num;
1740 return -E_MYSQL_SYNTAX;
1741 num = atoi(argv[1]);
1743 return -E_MYSQL_SYNTAX;
1745 stream = get_current_stream();
1747 return -E_GET_STREAM;
1749 stream = escape_str(argv[2]);
1753 tmp = get_query(stream, NULL, 0);
1756 return -E_GET_QUERY;
1757 query = make_message("%s limit %d", tmp, num);
1760 result = get_result(query);
1764 ret = -E_EMPTY_RESULT;
1765 num_rows = mysql_num_rows(result);
1768 for (i = 0; i < num_rows && i < num; i++) {
1769 row = mysql_fetch_row(result);
1771 send_va_buffer(fd, "Skipping %s\n", row[0]);
1772 update_audio_file(row[0]);
1774 send_va_buffer(fd, "%s\n", row[0]? row[0]: "BUG");
1779 mysql_free_result(result);
1784 int com_sl(int fd, int argc, char *argv[])
1786 return com_sl_skip(fd, argc, argv);
1790 int com_skip(int fd, int argc, char *argv[])
1792 return com_sl_skip(fd, argc, argv);
1796 * update attributes of name
1798 static int update_atts(int fd, char *name, char *atts)
1801 char *ebn, *q, *old, *new = NULL;
1805 ebn = escaped_basename(name);
1808 q = make_message("update data set %s where name = '%s'", atts, ebn);
1809 old = get_atts(ebn, 0);
1810 send_va_buffer(fd, "old: %s\n", old);
1812 ret = real_query(q);
1816 new = get_atts(ebn, 0);
1817 ret = send_va_buffer(fd, "new: %s\n", new);
1827 int com_sa(int fd, int argc, char *argv[])
1830 char *atts = NULL, *name;
1833 return -E_MYSQL_SYNTAX;
1834 for (i = 1; i < argc; i++) {
1836 char *esc, *tmp, *p =argv[i];
1837 int len = strlen(p);
1841 switch (p[len - 1]) {
1852 esc = escape_str(p);
1855 tmp = make_message("%s%s='%s'", atts? "," : "", esc,
1858 atts = para_strcat(atts, tmp);
1864 if (i >= argc) { /* no name given, use current af */
1865 ret = -E_GET_AUDIO_FILE;
1866 if (!(name = get_current_audio_file()))
1868 ret = update_atts(fd, name, atts);
1872 for (; argv[i] && ret >= 0; i++)
1873 ret = update_atts(fd, argv[i], atts);
1875 refresh_selector_info();
1884 int com_cam(int fd, int argc, char *argv[])
1886 char *name = NULL, *meta = NULL, *atts = NULL;
1890 return -E_MYSQL_SYNTAX;
1891 if (!(name = escaped_basename(argv[1])))
1894 if (!(atts = get_atts(name, 1)))
1897 if (!(meta = get_meta(name, 0)))
1899 for (i = 2; i < argc; i++) {
1902 if (!(ebn = escaped_basename(argv[i])))
1904 ret = send_va_buffer(fd, "updating %s\n", ebn);
1909 q = make_message("update data set %s where name = '%s'",
1911 if ((ret = update_atts(fd, ebn, atts)) >= 0)
1912 ret = real_query(q);
1932 static int com_vrfy_clean(int fd, int argc, __a_unused char *argv[])
1935 int ret, vrfy_mode = strcmp(argv[0], "clean");
1936 void *result = NULL;
1937 unsigned int num_rows;
1942 return -E_MYSQL_SYNTAX;
1944 result = get_result("select data.name from data left join dir on "
1945 "dir.name = data.name where dir.name is NULL");
1948 num_rows = mysql_num_rows(result);
1950 ret = send_buffer(fd, "No invalid entries\n");
1954 send_va_buffer(fd, "found %i invalid entr%s\n", num_rows,
1955 num_rows == 1? "y" : "ies");
1956 ret = print_results(fd, result, 0, 0, num_rows - 1, 0);
1959 while ((row = mysql_fetch_row(result))) {
1964 escaped_name = escape_str(row[0]);
1967 send_va_buffer(fd, "deleting %s\n", escaped_name);
1968 query = make_message("delete from data where name = '%s'",
1970 ret = real_query(query);
1978 mysql_free_result(result);
1985 int com_vrfy(int fd, int argc, char **argv)
1987 return com_vrfy_clean(fd, argc, argv);
1993 int com_clean(int fd, int argc, char **argv)
1995 return com_vrfy_clean(fd, argc, argv);
1998 static FILE *out_file;
2000 static int mysql_write_tmp_file(const char *dir, const char *name)
2002 int ret = -E_TMPFILE;
2003 char *msg = make_message("%s\t%s\n", dir, name);
2004 if (fputs(msg, out_file) != EOF)
2013 int com_upd(int fd, int argc, __a_unused char *argv[])
2015 char *tempname = NULL, *query = NULL;
2016 int ret, out_fd = -1, num = 0;
2017 void *result = NULL;
2018 unsigned int num_rows;
2022 return -E_MYSQL_SYNTAX;
2024 tempname = para_strdup("/tmp/mysql.tmp.XXXXXX");
2025 ret = para_mkstemp(tempname, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
2029 out_file = fdopen(out_fd, "w");
2034 if (find_audio_files(conf.mysql_audio_file_dir_arg, mysql_write_tmp_file) < 0)
2036 num = ftell(out_file);
2038 * we have to make sure the file hit the disk before we call
2043 PARA_DEBUG_LOG("wrote tempfile %s (%d bytes)\n", tempname, num);
2046 if ((ret = real_query("delete from dir")) < 0)
2048 query = make_message("load data infile '%s' ignore into table dir "
2049 "fields terminated by '\t' lines terminated by '\n' "
2050 "(dir, name)", tempname);
2051 ret = real_query(query);
2055 result = get_result("select dir.name from dir left join data on "
2056 "data.name = dir.name where data.name is NULL");
2060 num_rows = mysql_num_rows(result);
2062 ret = send_buffer(fd, "no new entries\n");
2065 while ((row = mysql_fetch_row(result))) {
2070 send_va_buffer(fd, "new entry: %s\n", row[0]);
2071 erow = escape_str(row[0]);
2074 query = make_message("insert into data (name, pic_id) values "
2075 "('%s','%s')", erow, "1");
2077 ret = real_query(query);
2090 mysql_free_result(result);
2094 static char **server_get_audio_file_list(unsigned int num)
2096 char **list = para_malloc((num + 1) * sizeof(char *));
2097 char *tmp, *query, *stream = get_current_stream();
2098 void *result = NULL;
2099 unsigned int num_rows;
2103 tmp = get_query(stream, NULL, 1);
2107 query = make_message("%s limit %d", tmp, num);
2109 result = get_result(query);
2113 num_rows = mysql_num_rows(result);
2116 for (i = 0; i < num_rows && i < num; i++) {
2117 row = mysql_fetch_row(result);
2118 if (!row || !row[0])
2120 list[i] = para_strdup(row[0]);
2133 mysql_free_result(result);
2138 * connect to mysql server, return mysql pointer on success, -E_NOTCONN
2139 * on errors. Called from parent on startup and also from com_cdb().
2141 static int init_mysql_server(void)
2143 char *u = conf.mysql_user_arg? conf.mysql_user_arg : para_logname();
2145 mysql_ptr = mysql_init(NULL);
2147 PARA_CRIT_LOG("%s", "mysql init error\n");
2150 PARA_DEBUG_LOG("connecting: %s@%s:%d\n", u, conf.mysql_host_arg,
2151 conf.mysql_port_arg);
2152 if (!conf.mysql_user_arg)
2155 * If host is NULL a connection to the local host is assumed,
2156 * If user is NULL, the current user is assumed
2158 if (!(mysql_ptr = mysql_real_connect(mysql_ptr,
2159 conf.mysql_host_arg,
2160 conf.mysql_user_arg,
2161 conf.mysql_passwd_arg,
2162 conf.mysql_database_arg,
2163 conf.mysql_port_arg, NULL, 0))) {
2164 PARA_CRIT_LOG("%s", "connect error\n");
2167 PARA_INFO_LOG("%s", "success\n");
2171 /* mmd lock must be held */
2172 static void write_msg2mmd(int success)
2174 sprintf(mmd->selector_info, "dbinfo1:%s\ndbinfo2:mysql-%s\ndbinfo3:\n",
2175 success < 0? PARA_STRERROR(-success) :
2176 "successfully connected to mysql server",
2177 success < 0? "" : mysql_get_server_info(mysql_ptr));
2180 /* create database */
2181 int com_cdb(int fd, int argc, char *argv[])
2187 PARA_INFO_LOG("%s", "closing database\n");
2188 mysql_close(mysql_ptr);
2190 /* dont use any database */
2191 conf.mysql_database_arg = NULL; /* leak? */
2192 ret = -E_MYSQL_INIT;
2193 if (init_mysql_server() < 0 || !mysql_ptr)
2196 conf.mysql_database_arg = para_strdup("paraslash");
2199 conf.mysql_database_arg = escape_str(argv[1]);
2200 if (!conf.mysql_database_arg)
2203 query = make_message("create database %s", conf.mysql_database_arg);
2204 ret = real_query(query);
2208 /* reconnect with database just created */
2209 mysql_close(mysql_ptr);
2210 ret = -E_MYSQL_INIT;
2211 if (init_mysql_server() < 0 || !mysql_ptr)
2217 if (real_query("create table data (name varchar(255) binary not null "
2219 "lastplayed datetime not null default "
2221 "numplayed int not null default 0, "
2222 "pic_id bigint unsigned not null default 1)") < 0)
2224 if (real_query("create table dir (name varchar(255) binary not null "
2225 "primary key, dir varchar(255) default null)") < 0)
2227 if (real_query("create table pics ("
2228 "id bigint(20) unsigned not null primary key "
2230 "name varchar(255) binary not null, "
2231 "pic mediumblob not null)") < 0)
2233 if (real_query("create table streams ("
2234 "name varchar(255) binary not null primary key, "
2235 "def blob not null)") < 0)
2237 if (real_query("insert into streams (name, def) values "
2238 "('current_stream', '(none)')") < 0)
2240 ret = send_va_buffer(fd, "successfully created database %s\n",
2241 conf.mysql_database_arg);
2246 static void shutdown_connection(void)
2249 PARA_NOTICE_LOG("%s", "shutting down mysql connection\n");
2250 mysql_close(mysql_ptr);
2256 * the init function of the mysql-based audio file selector
2258 * \param db pointer to the struct to initialize
2260 * Check the command line options and initialize all function pointers of \a
2261 * db. Connect to the mysql server and initialize the info string.
2263 * \return This function returns success even if it could not connect
2264 * to the mysql server. This is because the connect is expected to fail
2265 * if there the paraslash database is not yet created. This gives the
2266 * user a chance to send the "cdb" to create the database.
2268 * \sa struct audio_file_selector, misc_meta_data::selector_info,
2271 int mysql_selector_init(struct audio_file_selector *db)
2275 if (!conf.mysql_passwd_given)
2276 return -E_NO_MYSQL_PASSWD;
2277 if (!conf.mysql_audio_file_dir_given)
2278 return -E_NO_AF_DIR;
2280 db->cmd_list = mysql_selector_cmds;
2281 db->get_audio_file_list = server_get_audio_file_list;
2282 db->update_audio_file = update_audio_file_server_handler;
2283 db->shutdown = shutdown_connection;
2284 ret = init_mysql_server();
2286 PARA_WARNING_LOG("%s\n", PARA_STRERROR(-ret));
2288 return 1; /* return success even if connect failed to give the
2289 * user the chance to exec com_cdb