]> git.tuebingen.mpg.de Git - paraslash.git/blob - command.c
Introduce lsu.{c,h}, implement help --long for para_server.
[paraslash.git] / command.c
1 /* Copyright (C) 1997 Andre Noll <maan@tuebingen.mpg.de>, see file COPYING. */
2
3 /** \file command.c Client authentication and server commands. */
4
5 #include <netinet/in.h>
6 #include <sys/socket.h>
7 #include <regex.h>
8 #include <signal.h>
9 #include <sys/types.h>
10 #include <osl.h>
11 #include <arpa/inet.h>
12 #include <sys/un.h>
13 #include <netdb.h>
14 #include <lopsub.h>
15
16 #include "server.lsg.h"
17 #include "para.h"
18 #include "error.h"
19 #include "lsu.h"
20 #include "crypt.h"
21 #include "sideband.h"
22 #include "command.h"
23 #include "string.h"
24 #include "afh.h"
25 #include "afs.h"
26 #include "net.h"
27 #include "server.h"
28 #include "list.h"
29 #include "send.h"
30 #include "sched.h"
31 #include "vss.h"
32 #include "daemon.h"
33 #include "fd.h"
34 #include "ipc.h"
35 #include "server_cmd.lsg.h"
36 #include "user_list.h"
37 #include "signal.h"
38 #include "version.h"
39
40 #define SERVER_CMD_AUX_INFO(_arg) _arg,
41 static const unsigned server_command_perms[] = {LSG_SERVER_CMD_AUX_INFOS};
42 #undef SERVER_CMD_AUX_INFO
43 #define SERVER_CMD_AUX_INFO(_arg) #_arg,
44 static const char * const server_command_perms_txt[] = {LSG_SERVER_CMD_AUX_INFOS};
45 #undef SERVER_CMD_AUX_INFO
46
47 /** Commands including options must be shorter than this. */
48 #define MAX_COMMAND_LEN 32768
49
50 extern int mmd_mutex;
51 extern struct misc_meta_data *mmd;
52 extern struct sender senders[];
53 int send_afs_status(struct command_context *cc, int parser_friendly);
54
55 static void dummy(__a_unused int s)
56 {
57 }
58
59 /*
60  * Compute human readable vss status text.
61  *
62  * We can't call vss_playing() and friends here because those functions read
63  * the flags from the primary mmd structure, so calling them from command
64  * handler context would require to take the mmd lock. At the time the function
65  * is called we already took a copy of the mmd structure and want to use the
66  * flags value of the copy for computing the vss status text.
67  */
68 static char *vss_status_tohuman(unsigned int flags)
69 {
70         if (flags & VSS_PLAYING)
71                 return para_strdup("playing");
72         if (flags & VSS_NEXT)
73                 return para_strdup("stopped");
74         return para_strdup("paused");
75 }
76
77 /*
78  * Never returns NULL.
79  */
80 static char *vss_get_status_flags(unsigned int flags)
81 {
82         char *msg = para_malloc(5 * sizeof(char));
83
84         msg[0] = (flags & VSS_PLAYING)? 'P' : '_';
85         msg[1] = (flags & VSS_NOMORE)? 'O' : '_';
86         msg[2] = (flags & VSS_NEXT)? 'N' : '_';
87         msg[3] = (flags & VSS_REPOS)? 'R' : '_';
88         msg[4] = '\0';
89         return msg;
90 }
91
92 static unsigned get_status(struct misc_meta_data *nmmd, bool parser_friendly,
93                 char **result)
94 {
95         char *status, *flags; /* vss status info */
96         /* nobody updates our version of "now" */
97         long offset = (nmmd->offset + 500) / 1000;
98         struct timeval current_time;
99         struct para_buffer b = {.flags = parser_friendly? PBF_SIZE_PREFIX : 0};
100
101         /* report real status */
102         status = vss_status_tohuman(nmmd->vss_status_flags);
103         flags = vss_get_status_flags(nmmd->vss_status_flags);
104         clock_get_realtime(&current_time);
105         /*
106          * The calls to WRITE_STATUS_ITEM() below never fail because
107          * b->max_size is zero (unlimited), see \ref para_printf(). However,
108          * clang is not smart enough to prove this and complains nevertheless.
109          * Casting the return value to void silences clang.
110          */
111         (void)WRITE_STATUS_ITEM(&b, SI_status, "%s\n", status);
112         (void)WRITE_STATUS_ITEM(&b, SI_status_flags, "%s\n", flags);
113         (void)WRITE_STATUS_ITEM(&b, SI_offset, "%li\n", offset);
114         (void)WRITE_STATUS_ITEM(&b, SI_afs_mode, "%s\n", mmd->afs_mode_string);
115         (void)WRITE_STATUS_ITEM(&b, SI_stream_start, "%lu.%lu\n",
116                 (long unsigned)nmmd->stream_start.tv_sec,
117                 (long unsigned)nmmd->stream_start.tv_usec);
118         (void)WRITE_STATUS_ITEM(&b, SI_current_time, "%lu.%lu\n",
119                 (long unsigned)current_time.tv_sec,
120                 (long unsigned)current_time.tv_usec);
121         free(flags);
122         free(status);
123         *result = b.buf;
124         return b.offset;
125 }
126
127 /**
128  * Send a sideband packet through a blocking file descriptor.
129  *
130  * \param scc fd and crypto keys.
131  * \param buf The buffer to send.
132  * \param numbytes The size of \a buf.
133  * \param band The sideband designator of this packet.
134  * \param dont_free If true, never deallocate \a buf.
135  *
136  * The nonblock flag must be disabled for the file descriptor given by \a scc.
137  *
138  * Stream cipher encryption is automatically activated if necessary via the
139  * sideband transformation, depending on the value of \a band.
140  *
141  * \return Standard.
142  *
143  * \sa \ref send_sb_va().
144  */
145 int send_sb(struct stream_cipher_context *scc, void *buf, size_t numbytes,
146                 int band, bool dont_free)
147 {
148         int ret;
149         struct sb_context *sbc;
150         struct iovec iov[2];
151         sb_transformation trafo = band < SBD_PROCEED? NULL : sc_trafo;
152         struct sb_buffer sbb = SBB_INIT(band, buf, numbytes);
153
154         sbc = sb_new_send(&sbb, dont_free, trafo, scc->send);
155         do {
156                 ret = sb_get_send_buffers(sbc, iov);
157                 ret = xwritev(scc->fd, iov, ret);
158                 if (ret < 0)
159                         goto fail;
160         } while (sb_sent(sbc, ret) == false);
161         return 1;
162 fail:
163         sb_free(sbc);
164         return ret;
165 }
166
167 /**
168  * Create a variable sized buffer and send it as a sideband packet.
169  *
170  * \param scc Passed to \ref send_sb.
171  * \param band See \ref send_sb.
172  * \param fmt The format string.
173  *
174  * \return The return value of the underlying call to \ref send_sb.
175  */
176 __printf_3_4 int send_sb_va(struct stream_cipher_context *scc, int band,
177                 const char *fmt, ...)
178 {
179         va_list ap;
180         char *msg;
181         int ret;
182
183         va_start(ap, fmt);
184         ret = xvasprintf(&msg, fmt, ap);
185         va_end(ap);
186         return send_sb(scc, msg, ret, band, false);
187 }
188
189 /**
190  * Send an error message to a client.
191  *
192  * \param cc Client info.
193  * \param err The (positive) error code.
194  *
195  * \return The return value of the underlying call to send_sb_va().
196  */
197 int send_strerror(struct command_context *cc, int err)
198 {
199         return send_sb_va(&cc->scc, SBD_ERROR_LOG, "%s\n", para_strerror(err));
200 }
201
202 /**
203  * Send an error context to a client,
204  *
205  * \param cc Client info.
206  * \param errctx The error context string.
207  *
208  * \return The return value of the underlying call to send_sb_va().
209  *
210  * This function frees the error context string after it was sent.
211  */
212 int send_errctx(struct command_context *cc, char *errctx)
213 {
214         int ret;
215
216         if (!errctx)
217                 return 0;
218         ret = send_sb_va(&cc->scc, SBD_ERROR_LOG, "%s\n", errctx);
219         free(errctx);
220         return ret;
221 }
222
223 static int check_sender_args(struct command_context *cc,
224                 struct lls_parse_result *lpr, struct sender_command_data *scd)
225 {
226         int i, ret;
227         const char * const subcmds[] = {SENDER_SUBCOMMANDS};
228         const char *arg;
229         char *errctx;
230         unsigned num_inputs = lls_num_inputs(lpr);
231
232         scd->sender_num = -1;
233         ret = lls(lls_check_arg_count(lpr, 2, INT_MAX, &errctx));
234         if (ret < 0) {
235                 send_errctx(cc, errctx);
236                 return ret;
237         }
238         arg = lls_input(0, lpr);
239         for (i = 0; senders[i].name; i++)
240                 if (!strcmp(senders[i].name, arg))
241                         break;
242         if (!senders[i].name)
243                 return -E_COMMAND_SYNTAX;
244         scd->sender_num = i;
245         arg = lls_input(1, lpr);
246         for (i = 0; i < NUM_SENDER_CMDS; i++)
247                 if (!strcmp(subcmds[i], arg))
248                         break;
249         if (i == NUM_SENDER_CMDS)
250                 return -E_COMMAND_SYNTAX;
251         scd->cmd_num = i;
252         if (!senders[scd->sender_num].client_cmds[scd->cmd_num])
253                 return -E_SENDER_CMD;
254         switch (scd->cmd_num) {
255         case SENDER_on:
256         case SENDER_off:
257                 if (num_inputs != 2)
258                         return -E_COMMAND_SYNTAX;
259                 break;
260         case SENDER_deny:
261         case SENDER_allow:
262                 if (num_inputs != 3 || parse_cidr(lls_input(2, lpr), scd->host,
263                                 sizeof(scd->host), &scd->netmask) == NULL)
264                         return -E_COMMAND_SYNTAX;
265                 break;
266         case SENDER_add:
267         case SENDER_delete:
268                 if (num_inputs != 3)
269                         return -E_COMMAND_SYNTAX;
270                 return parse_fec_url(lls_input(2, lpr), scd);
271         default:
272                 return -E_COMMAND_SYNTAX;
273         }
274         return 1;
275 }
276
277 /**
278  * Send a sideband packet through a blocking file descriptor.
279  *
280  * \param scc fd and crypto keys.
281  * \param expected_band The expected band designator.
282  * \param max_size Passed to \ref sb_new_recv().
283  * \param result Body of the sideband packet is returned here.
284  *
285  * If \a expected_band is not \p SBD_ANY, the band designator of the received
286  * sideband packet is compared to \a expected_band and a mismatch is considered
287  * an error.
288  *
289  * \return Standard.
290  */
291 int recv_sb(struct stream_cipher_context *scc,
292                 enum sb_designator expected_band,
293                 size_t max_size, struct iovec *result)
294 {
295         int ret;
296         struct sb_context *sbc;
297         struct iovec iov;
298         struct sb_buffer sbb;
299         sb_transformation trafo;
300
301         trafo = expected_band != SBD_ANY && expected_band < SBD_PROCEED?
302                 NULL : sc_trafo;
303         sbc = sb_new_recv(max_size, trafo, scc->recv);
304         for (;;) {
305                 sb_get_recv_buffer(sbc, &iov);
306                 ret = recv_bin_buffer(scc->fd, iov.iov_base, iov.iov_len);
307                 if (ret == 0)
308                         ret = -E_EOF;
309                 if (ret < 0)
310                         goto fail;
311                 ret = sb_received(sbc, ret, &sbb);
312                 if (ret < 0)
313                         goto fail;
314                 if (ret > 0)
315                         break;
316         }
317         ret = -E_BAD_BAND;
318         if (expected_band != SBD_ANY && sbb.band != expected_band)
319                 goto fail;
320         *result = sbb.iov;
321         return 1;
322 fail:
323         sb_free(sbc);
324         return ret;
325 }
326
327 static int com_sender(struct command_context *cc, struct lls_parse_result *lpr)
328 {
329         int i, ret = 0;
330         char *msg = NULL;
331         struct sender_command_data scd;
332
333         if (lls_num_inputs(lpr) == 0) {
334                 for (i = 0; senders[i].name; i++) {
335                         char *tmp;
336                         ret = xasprintf(&tmp, "%s%s\n", msg? msg : "",
337                                 senders[i].name);
338                         free(msg);
339                         msg = tmp;
340                 }
341                 return send_sb(&cc->scc, msg, ret, SBD_OUTPUT, false);
342         }
343         ret = check_sender_args(cc, lpr, &scd);
344         if (ret < 0) {
345                 if (scd.sender_num < 0)
346                         return ret;
347                 if (strcmp(lls_input(1, lpr), "status") == 0)
348                         msg = senders[scd.sender_num].status();
349                 else
350                         msg = senders[scd.sender_num].help();
351                 return send_sb(&cc->scc, msg, strlen(msg), SBD_OUTPUT, false);
352         }
353
354         switch (scd.cmd_num) {
355         case SENDER_add:
356         case SENDER_delete:
357                 assert(senders[scd.sender_num].resolve_target);
358                 ret = senders[scd.sender_num].resolve_target(lls_input(2, lpr),
359                         &scd);
360                 if (ret < 0)
361                         return ret;
362         }
363
364         for (i = 0; i < 10; i++) {
365                 mutex_lock(mmd_mutex);
366                 if (mmd->sender_cmd_data.cmd_num >= 0) {
367                         /* another sender command is active, retry in 100ms */
368                         struct timespec ts = {.tv_nsec = 100 * 1000 * 1000};
369                         mutex_unlock(mmd_mutex);
370                         nanosleep(&ts, NULL);
371                         continue;
372                 }
373                 mmd->sender_cmd_data = scd;
374                 mutex_unlock(mmd_mutex);
375                 break;
376         }
377         return (i < 10)? 1 : -E_LOCK;
378 }
379 EXPORT_SERVER_CMD_HANDLER(sender);
380
381 static int com_si(struct command_context *cc,
382                 __a_unused struct lls_parse_result *lpr)
383 {
384         char *msg, *ut;
385         int ret;
386
387         ut = daemon_get_uptime_str(now);
388         mutex_lock(mmd_mutex);
389         ret = xasprintf(&msg,
390                 "up: %s\nplayed: %u\n"
391                 "server_pid: %d\n"
392                 "afs_pid: %d\n"
393                 "connections (active/accepted/total): %u/%u/%u\n"
394                 "current loglevel: %s\n"
395                 "supported audio formats: %s\n",
396                 ut, mmd->num_played,
397                 (int)getppid(),
398                 (int)afs_pid,
399                 mmd->active_connections,
400                 mmd->num_commands,
401                 mmd->num_connects,
402                 ENUM_STRING_VAL(LOGLEVEL),
403                 AUDIO_FORMAT_HANDLERS
404         );
405         mutex_unlock(mmd_mutex);
406         free(ut);
407         return send_sb(&cc->scc, msg, ret, SBD_OUTPUT, false);
408 }
409 EXPORT_SERVER_CMD_HANDLER(si);
410
411 static int com_version(struct command_context *cc, struct lls_parse_result *lpr)
412 {
413         char *msg;
414         size_t len;
415
416         if (SERVER_CMD_OPT_GIVEN(VERSION, VERBOSE, lpr))
417                 len = xasprintf(&msg, "%s", version_text("server"));
418         else
419                 len = xasprintf(&msg, "%s\n", version_single_line("server"));
420         return send_sb(&cc->scc, msg, len, SBD_OUTPUT, false);
421 }
422 EXPORT_SERVER_CMD_HANDLER(version);
423
424 /** These status items are cleared if no audio file is currently open. */
425 #define EMPTY_STATUS_ITEMS \
426         ITEM(path) \
427         ITEM(directory) \
428         ITEM(basename) \
429         ITEM(score) \
430         ITEM(attributes_bitmap) \
431         ITEM(attributes_txt) \
432         ITEM(hash) \
433         ITEM(image_id) \
434         ITEM(image_name) \
435         ITEM(lyrics_id) \
436         ITEM(lyrics_name) \
437         ITEM(bitrate) \
438         ITEM(format) \
439         ITEM(frequency) \
440         ITEM(channels) \
441         ITEM(duration) \
442         ITEM(seconds_total) \
443         ITEM(num_played) \
444         ITEM(last_played) \
445         ITEM(techinfo) \
446         ITEM(artist) \
447         ITEM(title) \
448         ITEM(year) \
449         ITEM(album) \
450         ITEM(comment) \
451         ITEM(mtime) \
452         ITEM(file_size) \
453         ITEM(chunk_time) \
454         ITEM(num_chunks) \
455         ITEM(amplification) \
456
457 /*
458  * Create a set of audio-file related status items with empty values. These are
459  * written to stat clients when no audio file is open.
460  */
461 static unsigned empty_status_items(bool parser_friendly, char **result)
462 {
463         char *esi;
464         unsigned len;
465
466         if (parser_friendly)
467                 len = xasprintf(&esi,
468                         #define ITEM(x) "0004 %02x:\n"
469                         EMPTY_STATUS_ITEMS
470                         #undef ITEM
471                         #define ITEM(x) , (unsigned) SI_ ## x
472                         EMPTY_STATUS_ITEMS
473                         #undef ITEM
474                 );
475         else
476                 len = xasprintf(&esi,
477                         #define ITEM(x) "%s:\n"
478                         EMPTY_STATUS_ITEMS
479                         #undef ITEM
480                         #define ITEM(x) ,status_item_list[SI_ ## x]
481                         EMPTY_STATUS_ITEMS
482                         #undef ITEM
483                 );
484         *result = esi;
485         return len;
486 }
487 #undef EMPTY_STATUS_ITEMS
488
489 static int com_stat(struct command_context *cc, struct lls_parse_result *lpr)
490 {
491         int ret;
492         struct misc_meta_data tmp, *nmmd = &tmp;
493         char *s;
494         bool parser_friendly = SERVER_CMD_OPT_GIVEN(STAT, PARSER_FRIENDLY,
495                 lpr) > 0;
496         uint32_t num = SERVER_CMD_UINT32_VAL(STAT, NUM, lpr);
497
498         para_sigaction(SIGUSR1, dummy);
499         for (;;) {
500                 /*
501                  * Copy the mmd structure to minimize the time we hold the mmd
502                  * lock.
503                  */
504                 mutex_lock(mmd_mutex);
505                 *nmmd = *mmd;
506                 mutex_unlock(mmd_mutex);
507                 ret = get_status(nmmd, parser_friendly, &s);
508                 ret = send_sb(&cc->scc, s, ret, SBD_OUTPUT, false);
509                 if (ret < 0)
510                         goto out;
511                 if (nmmd->vss_status_flags & VSS_NEXT) {
512                         char *esi;
513                         ret = empty_status_items(parser_friendly, &esi);
514                         ret = send_sb(&cc->scc, esi, ret, SBD_OUTPUT, false);
515                         if (ret < 0)
516                                 goto out;
517                 } else
518                         send_afs_status(cc, parser_friendly);
519                 ret = 1;
520                 if (num > 0 && !--num)
521                         goto out;
522                 sleep(50);
523                 ret = -E_SERVER_CRASH;
524                 if (getppid() == 1)
525                         goto out;
526         }
527 out:
528         return ret;
529 }
530 EXPORT_SERVER_CMD_HANDLER(stat);
531
532 const char *aux_info_cb(unsigned cmd_num, bool verbose)
533 {
534         static char result[80];
535         unsigned perms = server_command_perms[cmd_num];
536
537         if (verbose) {
538                 /* permissions: VSS_READ | VSS_WRITE */
539                 sprintf(result, "permissions: %s",
540                         server_command_perms_txt[cmd_num]);
541         } else {
542                 result[0] = perms & AFS_READ? 'a' : '-';
543                 result[1] = perms & AFS_WRITE? 'A' : '-';
544                 result[2] = perms & VSS_READ? 'v' : '-';
545                 result[3] = perms & VSS_WRITE? 'V' : '-';
546                 result[4] = '\0';
547         }
548         return result;
549 }
550
551 static int com_help(struct command_context *cc, struct lls_parse_result *lpr)
552 {
553         char *buf;
554         int ret;
555         unsigned n;
556         bool long_help = SERVER_CMD_OPT_GIVEN(HELP, LONG, lpr);
557
558         lsu_com_help(long_help, lpr, server_cmd_suite, aux_info_cb, &buf, &n);
559         ret = send_sb(&cc->scc, buf, n, SBD_OUTPUT, false);
560         return ret;
561 }
562 EXPORT_SERVER_CMD_HANDLER(help);
563
564 static int com_hup(__a_unused struct command_context *cc,
565                 __a_unused struct lls_parse_result *lpr)
566 {
567         kill(getppid(), SIGHUP);
568         return 1;
569 }
570 EXPORT_SERVER_CMD_HANDLER(hup);
571
572 static int com_term(__a_unused struct command_context *cc,
573                 __a_unused struct lls_parse_result *lpr)
574 {
575         kill(getppid(), SIGTERM);
576         return 1;
577 }
578 EXPORT_SERVER_CMD_HANDLER(term);
579
580 static int com_play(__a_unused struct command_context *cc,
581                 __a_unused struct lls_parse_result *lpr)
582 {
583         mutex_lock(mmd_mutex);
584         mmd->new_vss_status_flags |= VSS_PLAYING;
585         mmd->new_vss_status_flags &= ~VSS_NOMORE;
586         mutex_unlock(mmd_mutex);
587         return 1;
588 }
589 EXPORT_SERVER_CMD_HANDLER(play);
590
591 static int com_stop(__a_unused struct command_context *cc,
592                 __a_unused struct lls_parse_result *lpr)
593 {
594         mutex_lock(mmd_mutex);
595         mmd->new_vss_status_flags &= ~VSS_PLAYING;
596         mmd->new_vss_status_flags &= ~VSS_REPOS;
597         mmd->new_vss_status_flags |= VSS_NEXT;
598         mutex_unlock(mmd_mutex);
599         return 1;
600 }
601 EXPORT_SERVER_CMD_HANDLER(stop);
602
603 static int com_pause(__a_unused struct command_context *cc,
604                 __a_unused struct lls_parse_result *lpr)
605 {
606         mutex_lock(mmd_mutex);
607         if (!vss_paused() && !vss_stopped()) {
608                 mmd->events++;
609                 mmd->new_vss_status_flags &= ~VSS_PLAYING;
610                 mmd->new_vss_status_flags &= ~VSS_NEXT;
611         }
612         mutex_unlock(mmd_mutex);
613         return 1;
614 }
615 EXPORT_SERVER_CMD_HANDLER(pause);
616
617 static int com_next(__a_unused struct command_context *cc,
618                 __a_unused struct lls_parse_result *lpr)
619 {
620         mutex_lock(mmd_mutex);
621         mmd->events++;
622         mmd->new_vss_status_flags |= VSS_NEXT;
623         mutex_unlock(mmd_mutex);
624         return 1;
625 }
626 EXPORT_SERVER_CMD_HANDLER(next);
627
628 static int com_nomore(__a_unused struct command_context *cc,
629                 __a_unused struct lls_parse_result *lpr)
630 {
631         mutex_lock(mmd_mutex);
632         if (vss_playing() || vss_paused())
633                 mmd->new_vss_status_flags |= VSS_NOMORE;
634         mutex_unlock(mmd_mutex);
635         return 1;
636 }
637 EXPORT_SERVER_CMD_HANDLER(nomore);
638
639 static int com_ff(__a_unused struct command_context *cc,
640                 struct lls_parse_result *lpr)
641 {
642         long promille;
643         int ret, backwards = 0;
644         unsigned i;
645         char c, *errctx;
646
647         ret = lls(lls_check_arg_count(lpr, 1, 1, &errctx));
648         if (ret < 0) {
649                 send_errctx(cc, errctx);
650                 return ret;
651         }
652         if (!(ret = sscanf(lls_input(0, lpr), "%u%c", &i, &c)))
653                 return -E_COMMAND_SYNTAX;
654         if (ret > 1 && c == '-')
655                 backwards = 1; /* jmp backwards */
656         mutex_lock(mmd_mutex);
657         ret = -E_NO_AUDIO_FILE;
658         if (!mmd->afd.afhi.chunks_total || !mmd->afd.afhi.seconds_total)
659                 goto out;
660         promille = (1000 * mmd->current_chunk) / mmd->afd.afhi.chunks_total;
661         if (backwards)
662                 promille -= 1000 * i / mmd->afd.afhi.seconds_total;
663         else
664                 promille += 1000 * i / mmd->afd.afhi.seconds_total;
665         if (promille < 0)
666                 promille = 0;
667         if (promille > 1000) {
668                 mmd->new_vss_status_flags |= VSS_NEXT;
669                 goto out;
670         }
671         mmd->repos_request = (mmd->afd.afhi.chunks_total * promille) / 1000;
672         mmd->new_vss_status_flags |= VSS_REPOS;
673         mmd->new_vss_status_flags &= ~VSS_NEXT;
674         mmd->events++;
675         ret = 1;
676 out:
677         mutex_unlock(mmd_mutex);
678         return ret;
679 }
680 EXPORT_SERVER_CMD_HANDLER(ff);
681
682 static int com_jmp(__a_unused struct command_context *cc,
683                 struct lls_parse_result *lpr)
684 {
685         long unsigned int i;
686         int ret;
687         char *errctx;
688
689         ret = lls(lls_check_arg_count(lpr, 1, 1, &errctx));
690         if (ret < 0) {
691                 send_errctx(cc, errctx);
692                 return ret;
693         }
694         if (sscanf(lls_input(0, lpr), "%lu", &i) <= 0)
695                 return -ERRNO_TO_PARA_ERROR(EINVAL);
696         mutex_lock(mmd_mutex);
697         ret = -E_NO_AUDIO_FILE;
698         if (!mmd->afd.afhi.chunks_total)
699                 goto out;
700         if (i > 100)
701                 i = 100;
702         PARA_INFO_LOG("jumping to %lu%%\n", i);
703         mmd->repos_request = (mmd->afd.afhi.chunks_total * i + 50) / 100;
704         PARA_INFO_LOG("sent: %lu, offset before jmp: %li\n",
705                 mmd->chunks_sent, mmd->offset);
706         mmd->new_vss_status_flags |= VSS_REPOS;
707         mmd->new_vss_status_flags &= ~VSS_NEXT;
708         ret = 1;
709         mmd->events++;
710 out:
711         mutex_unlock(mmd_mutex);
712         return ret;
713 }
714 EXPORT_SERVER_CMD_HANDLER(jmp);
715
716 /* deprecated, does nothing */
717 static int com_tasks(__a_unused struct command_context *cc,
718                 __a_unused struct lls_parse_result *lpr)
719 {
720         return 1;
721 }
722 EXPORT_SERVER_CMD_HANDLER(tasks);
723
724 static void reset_signals(void)
725 {
726         para_sigaction(SIGCHLD, SIG_IGN);
727         para_sigaction(SIGINT, SIG_DFL);
728         para_sigaction(SIGTERM, SIG_DFL);
729         para_sigaction(SIGHUP, SIG_DFL);
730 }
731
732 struct connection_features {
733         int dummy; /* none at the moment */
734 };
735
736 static int parse_auth_request(char *buf, int len, struct user **u,
737                 struct connection_features *cf)
738 {
739         int ret;
740         char *p, *username, **features = NULL;
741         size_t auth_rq_len = strlen(AUTH_REQUEST_MSG);
742
743         *u = NULL;
744         memset(cf, 0, sizeof(*cf));
745         if (len < auth_rq_len + 2)
746                 return -E_AUTH_REQUEST;
747         if (strncmp(buf, AUTH_REQUEST_MSG, auth_rq_len) != 0)
748                 return -E_AUTH_REQUEST;
749         username = buf + auth_rq_len;
750         p = strchr(username, ' ');
751         if (p) {
752                 int i;
753                 if (p == username)
754                         return -E_AUTH_REQUEST;
755                 *p = '\0';
756                 p++;
757                 create_argv(p, ",", &features);
758                 for (i = 0; features[i]; i++) {
759                         if (strcmp(features[i], "sideband") == 0)
760                                 continue;
761                         if (strcmp(features[i], "aes_ctr128") == 0)
762                                 continue;
763                         else {
764                                 ret = -E_BAD_FEATURE;
765                                 goto out;
766                         }
767                 }
768         }
769         PARA_DEBUG_LOG("received auth request for user %s\n", username);
770         *u = lookup_user(username);
771         ret = 1;
772 out:
773         free_argv(features);
774         return ret;
775 }
776
777 #define HANDSHAKE_BUFSIZE 4096
778
779 static int run_command(struct command_context *cc, struct iovec *iov)
780 {
781         int ret, i, argc;
782         char *p, *end, **argv;
783         const struct lls_command *lcmd = NULL;
784         unsigned perms;
785         struct lls_parse_result *lpr;
786         char *errctx;
787
788         if (iov->iov_base == NULL || iov->iov_len == 0)
789                 return -ERRNO_TO_PARA_ERROR(EINVAL);
790         p = iov->iov_base;
791         p[iov->iov_len - 1] = '\0'; /* just to be sure */
792
793         ret = lls(lls_lookup_subcmd(p, server_cmd_suite, &errctx));
794         if (ret < 0) {
795                 send_errctx(cc, errctx);
796                 return ret;
797         }
798         perms = server_command_perms[ret];
799         if ((perms & cc->u->perms) != perms)
800                 return -E_PERM;
801         lcmd = lls_cmd(ret, server_cmd_suite);
802         end = iov->iov_base + iov->iov_len;
803         for (i = 0; p < end; i++)
804                 p += strlen(p) + 1;
805         argc = i;
806         argv = para_malloc((argc + 1) * sizeof(char *));
807         for (i = 0, p = iov->iov_base; p < end; i++) {
808                 argv[i] = para_strdup(p);
809                 p += strlen(p) + 1;
810         }
811         argv[argc] = NULL;
812         PARA_NOTICE_LOG("calling com_%s() for user %s\n",
813                 lls_command_name(lcmd), cc->u->name);
814         ret = lls(lls_parse(argc, argv, lcmd, &lpr, &errctx));
815         if (ret >= 0) {
816                 const struct server_cmd_user_data *ud = lls_user_data(lcmd);
817                 ret = ud->handler(cc, lpr);
818                 lls_free_parse_result(lpr, lcmd);
819         } else
820                 send_errctx(cc, errctx);
821         free_argv(argv);
822         mutex_lock(mmd_mutex);
823         mmd->num_commands++;
824         if (ret >= 0 && (perms & AFS_WRITE))
825                 mmd->events++;
826         mutex_unlock(mmd_mutex);
827         return ret;
828 }
829
830 /**
831  * Perform user authentication and execute a command.
832  *
833  * \param fd The file descriptor to send output to.
834  *
835  * Whenever para_server accepts an incoming tcp connection on the port it
836  * listens on, it forks and the resulting child calls this function.
837  *
838  * An RSA-based challenge/response is used to authenticate the peer. It that
839  * authentication succeeds, a random session key is generated and sent back to
840  * the peer, encrypted with its RSA public key. From this point on, all
841  * transfers are crypted with this session key.
842  *
843  * Next it is checked if the peer supplied a valid server command or a command
844  * for the audio file selector. If yes, and if the user has sufficient
845  * permissions to execute that command, the function calls the corresponding
846  * command handler which does argument checking and further processing.
847  *
848  * In order to cope with DOS attacks, a timeout is set up which terminates
849  * the function if the connection was not authenticated when the timeout
850  * expires.
851  *
852  * \sa alarm(2), \ref crypt.c, \ref crypt.h.
853  */
854 __noreturn void handle_connect(int fd)
855 {
856         int ret;
857         unsigned char rand_buf[CHALLENGE_SIZE + 2 * SESSION_KEY_LEN];
858         unsigned char challenge_hash[HASH_SIZE];
859         char *command = NULL, *buf = para_malloc(HANDSHAKE_BUFSIZE) /* must be on the heap */;
860         size_t numbytes;
861         struct command_context cc_struct = {.u = NULL}, *cc = &cc_struct;
862         struct iovec iov;
863         struct connection_features cf;
864
865         cc->scc.fd = fd;
866         reset_signals();
867         /* we need a blocking fd here as recv() might return EAGAIN otherwise. */
868         ret = mark_fd_blocking(fd);
869         if (ret < 0)
870                 goto net_err;
871         /* send Welcome message */
872         ret = write_va_buffer(fd, "This is para_server, version "
873                 PACKAGE_VERSION ".\n"
874                 "Features: sideband,aes_ctr128\n"
875         );
876         if (ret < 0)
877                 goto net_err;
878         /* recv auth request line */
879         ret = recv_buffer(fd, buf, HANDSHAKE_BUFSIZE);
880         if (ret < 0)
881                 goto net_err;
882         ret = parse_auth_request(buf, ret, &cc->u, &cf);
883         if (ret < 0)
884                 goto net_err;
885         if (cc->u) {
886                 get_random_bytes_or_die(rand_buf, sizeof(rand_buf));
887                 ret = pub_encrypt(cc->u->pubkey, rand_buf, sizeof(rand_buf),
888                         (unsigned char *)buf);
889                 if (ret < 0)
890                         goto net_err;
891                 numbytes = ret;
892         } else {
893                 /*
894                  * We don't want to reveal our user names, so we send a
895                  * challenge to the client even if the user does not exist, and
896                  * fail the authentication later.
897                  */
898                 numbytes = 256;
899                 get_random_bytes_or_die((unsigned char *)buf, numbytes);
900         }
901         PARA_DEBUG_LOG("sending %d byte challenge + session key (%zu bytes)\n",
902                 CHALLENGE_SIZE, numbytes);
903         ret = send_sb(&cc->scc, buf, numbytes, SBD_CHALLENGE, false);
904         buf = NULL;
905         if (ret < 0)
906                 goto net_err;
907         ret = recv_sb(&cc->scc, SBD_CHALLENGE_RESPONSE,
908                 HANDSHAKE_BUFSIZE, &iov);
909         if (ret < 0)
910                 goto net_err;
911         buf = iov.iov_base;
912         numbytes = iov.iov_len;
913         PARA_DEBUG_LOG("received %zu bytes challenge response\n", numbytes);
914         ret = -E_BAD_USER;
915         if (!cc->u)
916                 goto net_err;
917         /*
918          * The correct response is the hash of the first CHALLENGE_SIZE bytes
919          * of the random data.
920          */
921         ret = -E_BAD_AUTH;
922         if (numbytes != HASH_SIZE)
923                 goto net_err;
924         hash_function((char *)rand_buf, CHALLENGE_SIZE, challenge_hash);
925         if (memcmp(challenge_hash, buf, HASH_SIZE))
926                 goto net_err;
927         /* auth successful */
928         alarm(0);
929         PARA_INFO_LOG("good auth for %s\n", cc->u->name);
930         /* init stream cipher keys with the second part of the random buffer */
931         cc->scc.recv = sc_new(rand_buf + CHALLENGE_SIZE, SESSION_KEY_LEN);
932         cc->scc.send = sc_new(rand_buf + CHALLENGE_SIZE + SESSION_KEY_LEN,
933                 SESSION_KEY_LEN);
934         ret = send_sb(&cc->scc, NULL, 0, SBD_PROCEED, false);
935         if (ret < 0)
936                 goto net_err;
937         ret = recv_sb(&cc->scc, SBD_COMMAND, MAX_COMMAND_LEN, &iov);
938         if (ret < 0)
939                 goto net_err;
940         ret = run_command(cc, &iov);
941         free(iov.iov_base);
942         if (ret < 0)
943                 goto err_out;
944         if (ret >= 0)
945                 goto out;
946 err_out:
947         if (send_strerror(cc, -ret) >= 0)
948                 send_sb(&cc->scc, NULL, 0, SBD_EXIT__FAILURE, true);
949 net_err:
950         PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
951 out:
952         free(buf);
953         free(command);
954         mutex_lock(mmd_mutex);
955         mmd->active_connections--;
956         mutex_unlock(mmd_mutex);
957         if (ret >= 0) {
958                 ret = send_sb(&cc->scc, NULL, 0, SBD_EXIT__SUCCESS, true);
959                 if (ret < 0)
960                         PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
961         }
962         sc_free(cc->scc.recv);
963         sc_free(cc->scc.send);
964         exit(ret < 0? EXIT_FAILURE : EXIT_SUCCESS);
965 }