chiark / gitweb /
b680f2d49d0ed62af9af5b10d8b8b2ed1eb9ae9e
[disorder] / server / server.c
1 /*
2  * This file is part of DisOrder.
3  * Copyright (C) 2004, 2005, 2006, 2007 Richard Kettlewell
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
18  * USA
19  */
20
21 #include <config.h>
22 #include "types.h"
23
24 #include <pwd.h>
25 #include <sys/types.h>
26 #include <sys/socket.h>
27 #include <sys/time.h>
28 #include <unistd.h>
29 #include <stdlib.h>
30 #include <string.h>
31 #include <errno.h>
32 #include <fcntl.h>
33 #include <gcrypt.h>
34 #include <stddef.h>
35 #include <time.h>
36 #include <limits.h>
37 #include <pcre.h>
38 #include <netdb.h>
39 #include <netinet/in.h>
40
41 #include "event.h"
42 #include "server.h"
43 #include "syscalls.h"
44 #include "queue.h"
45 #include "server-queue.h"
46 #include "play.h"
47 #include "log.h"
48 #include "mem.h"
49 #include "state.h"
50 #include "charset.h"
51 #include "split.h"
52 #include "configuration.h"
53 #include "hex.h"
54 #include "rights.h"
55 #include "trackdb.h"
56 #include "table.h"
57 #include "kvp.h"
58 #include "mixer.h"
59 #include "sink.h"
60 #include "authhash.h"
61 #include "plugin.h"
62 #include "printf.h"
63 #include "trackname.h"
64 #include "eventlog.h"
65 #include "defs.h"
66 #include "cache.h"
67 #include "unicode.h"
68 #include "cookies.h"
69 #include "base64.h"
70
71 #ifndef NONCE_SIZE
72 # define NONCE_SIZE 16
73 #endif
74
75 int volume_left, volume_right;          /* last known volume */
76
77 /** @brief Accept all well-formed login attempts
78  *
79  * Used in debugging.
80  */
81 int wideopen;
82
83 struct listener {
84   const char *name;
85   int pf;
86 };
87
88 /** @brief One client connection */
89 struct conn {
90   /** @brief Read commands from here */
91   ev_reader *r;
92   /** @brief Send responses to here */
93   ev_writer *w;
94   /** @brief Underlying file descriptor */
95   int fd;
96   /** @brief Unique identifier for connection used in log messages */
97   unsigned tag;
98   /** @brief Login name or NULL */
99   char *who;
100   /** @brief Event loop */
101   ev_source *ev;
102   /** @brief Nonce chosen for this connection */
103   unsigned char nonce[NONCE_SIZE];
104   /** @brief Current reader callback
105    *
106    * We change this depending on whether we're servicing the @b log command
107    */
108   ev_reader_callback *reader;
109   /** @brief Event log output sending to this connection */
110   struct eventlog_output *lo;
111   /** @brief Parent listener */
112   const struct listener *l;
113   /** @brief Login cookie or NULL */
114   char *cookie;
115   /** @brief Connection rights */
116   rights_type rights;
117 };
118
119 static int reader_callback(ev_source *ev,
120                            ev_reader *reader,
121                            void *ptr,
122                            size_t bytes,
123                            int eof,
124                            void *u);
125
126 static const char *noyes[] = { "no", "yes" };
127
128 /** @brief Called when a connection's writer fails or is shut down
129  *
130  * If the connection still has a raeder that is cancelled.
131  */
132 static int writer_error(ev_source attribute((unused)) *ev,
133                         int errno_value,
134                         void *u) {
135   struct conn *c = u;
136
137   D(("server writer_error S%x %d", c->tag, errno_value));
138   if(errno_value == 0) {
139     /* writer is done */
140     D(("S%x writer completed", c->tag));
141   } else {
142     if(errno_value != EPIPE)
143       error(errno_value, "S%x write error on socket", c->tag);
144     if(c->r) {
145       D(("cancel reader"));
146       ev_reader_cancel(c->r);
147       c->r = 0;
148     }
149     D(("done cancel reader"));
150   }
151   c->w = 0;
152   ev_report(ev);
153   return 0;
154 }
155
156 /** @brief Called when a conncetion's reader fails or is shut down
157  *
158  * If connection still has a writer then it is closed.
159  */
160 static int reader_error(ev_source attribute((unused)) *ev,
161                         int errno_value,
162                         void *u) {
163   struct conn *c = u;
164
165   D(("server reader_error S%x %d", c->tag, errno_value));
166   error(errno_value, "S%x read error on socket", c->tag);
167   if(c->w)
168     ev_writer_close(c->w);
169   c->w = 0;
170   c->r = 0;
171   ev_report(ev);
172   return 0;
173 }
174
175 static int c_disable(struct conn *c, char **vec, int nvec) {
176   if(nvec == 0)
177     disable_playing(c->who);
178   else if(nvec == 1 && !strcmp(vec[0], "now"))
179     disable_playing(c->who);
180   else {
181     sink_writes(ev_writer_sink(c->w), "550 invalid argument\n");
182     return 1;                   /* completed */
183   }
184   sink_writes(ev_writer_sink(c->w), "250 OK\n");
185   return 1;                     /* completed */
186 }
187
188 static int c_enable(struct conn *c,
189                     char attribute((unused)) **vec,
190                     int attribute((unused)) nvec) {
191   enable_playing(c->who, c->ev);
192   /* Enable implicitly unpauses if there is nothing playing */
193   if(paused && !playing) resume_playing(c->who);
194   sink_writes(ev_writer_sink(c->w), "250 OK\n");
195   return 1;                     /* completed */
196 }
197
198 static int c_enabled(struct conn *c,
199                      char attribute((unused)) **vec,
200                      int attribute((unused)) nvec) {
201   sink_printf(ev_writer_sink(c->w), "252 %s\n", noyes[playing_is_enabled()]);
202   return 1;                     /* completed */
203 }
204
205 static int c_play(struct conn *c, char **vec,
206                   int attribute((unused)) nvec) {
207   const char *track;
208   struct queue_entry *q;
209   
210   if(!trackdb_exists(vec[0])) {
211     sink_writes(ev_writer_sink(c->w), "550 track is not in database\n");
212     return 1;
213   }
214   if(!(track = trackdb_resolve(vec[0]))) {
215     sink_writes(ev_writer_sink(c->w), "550 cannot resolve track\n");
216     return 1;
217   }
218   q = queue_add(track, c->who, WHERE_BEFORE_RANDOM);
219   queue_write();
220   /* If we added the first track, and something is playing, then prepare the
221    * new track.  If nothing is playing then we don't bother as it wouldn't gain
222    * anything. */
223   if(q == qhead.next && playing)
224     prepare(c->ev, q);
225   sink_printf(ev_writer_sink(c->w), "252 %s\n", q->id);
226   /* If the queue was empty but we are for some reason paused then
227    * unpause. */
228   if(!playing) resume_playing(0);
229   play(c->ev);
230   return 1;                     /* completed */
231 }
232
233 static int c_remove(struct conn *c, char **vec,
234                     int attribute((unused)) nvec) {
235   struct queue_entry *q;
236
237   if(!(q = queue_find(vec[0]))) {
238     sink_writes(ev_writer_sink(c->w), "550 no such track on the queue\n");
239     return 1;
240   }
241   if(!right_removable(c->rights, c->who, q)) {
242     error(0, "%s attempted remove but lacks required rights", c->who);
243     sink_writes(ev_writer_sink(c->w),
244                 "510 Not authorized to remove that track\n");
245     return 1;
246   }
247   queue_remove(q, c->who);
248   /* De-prepare the track. */
249   abandon(c->ev, q);
250   /* If we removed a random track then add another one. */
251   if(q->state == playing_random)
252     add_random_track();
253   /* Prepare whatever the next head track is. */
254   if(qhead.next != &qhead)
255     prepare(c->ev, qhead.next);
256   queue_write();
257   sink_writes(ev_writer_sink(c->w), "250 removed\n");
258   return 1;                     /* completed */
259 }
260
261 static int c_scratch(struct conn *c,
262                      char **vec,
263                      int nvec) {
264   if(!playing) {
265     sink_writes(ev_writer_sink(c->w), "250 nothing is playing\n");
266     return 1;                   /* completed */
267   }
268   /* TODO there is a bug here: if we specify an ID but it's not the currently
269    * playing track then you will get 550 if you weren't authorized to scratch
270    * the currently playing track. */
271   if(!right_scratchable(c->rights, c->who, playing)) {
272     error(0, "%s attempted scratch but lacks required rights", c->who);
273     sink_writes(ev_writer_sink(c->w),
274                 "510 Not authorized to scratch that track\n");
275     return 1;
276   }
277   scratch(c->who, nvec == 1 ? vec[0] : 0);
278   /* If you scratch an unpaused track then it is automatically unpaused */
279   resume_playing(0);
280   sink_writes(ev_writer_sink(c->w), "250 scratched\n");
281   return 1;                     /* completed */
282 }
283
284 static int c_pause(struct conn *c,
285                    char attribute((unused)) **vec,
286                    int attribute((unused)) nvec) {
287   if(!playing) {
288     sink_writes(ev_writer_sink(c->w), "250 nothing is playing\n");
289     return 1;                   /* completed */
290   }
291   if(paused) {
292     sink_writes(ev_writer_sink(c->w), "250 already paused\n");
293     return 1;                   /* completed */
294   }
295   if(pause_playing(c->who) < 0)
296     sink_writes(ev_writer_sink(c->w), "550 cannot pause this track\n");
297   else
298     sink_writes(ev_writer_sink(c->w), "250 paused\n");
299   return 1;
300 }
301
302 static int c_resume(struct conn *c,
303                    char attribute((unused)) **vec,
304                    int attribute((unused)) nvec) {
305   if(!paused) {
306     sink_writes(ev_writer_sink(c->w), "250 not paused\n");
307     return 1;                   /* completed */
308   }
309   resume_playing(c->who);
310   sink_writes(ev_writer_sink(c->w), "250 paused\n");
311   return 1;
312 }
313
314 static int c_shutdown(struct conn *c,
315                       char attribute((unused)) **vec,
316                       int attribute((unused)) nvec) {
317   info("S%x shut down by %s", c->tag, c->who);
318   sink_writes(ev_writer_sink(c->w), "250 shutting down\n");
319   ev_writer_flush(c->w);
320   quit(c->ev);
321 }
322
323 static int c_reconfigure(struct conn *c,
324                          char attribute((unused)) **vec,
325                          int attribute((unused)) nvec) {
326   info("S%x reconfigure by %s", c->tag, c->who);
327   if(reconfigure(c->ev, 1))
328     sink_writes(ev_writer_sink(c->w), "550 error reading new config\n");
329   else
330     sink_writes(ev_writer_sink(c->w), "250 installed new config\n");
331   return 1;                             /* completed */
332 }
333
334 static int c_rescan(struct conn *c,
335                     char attribute((unused)) **vec,
336                     int attribute((unused)) nvec) {
337   info("S%x rescan by %s", c->tag, c->who);
338   trackdb_rescan(c->ev);
339   sink_writes(ev_writer_sink(c->w), "250 initiated rescan\n");
340   return 1;                             /* completed */
341 }
342
343 static int c_version(struct conn *c,
344                      char attribute((unused)) **vec,
345                      int attribute((unused)) nvec) {
346   /* VERSION had better only use the basic character set */
347   sink_printf(ev_writer_sink(c->w), "251 %s\n", disorder_short_version_string);
348   return 1;                     /* completed */
349 }
350
351 static int c_playing(struct conn *c,
352                      char attribute((unused)) **vec,
353                      int attribute((unused)) nvec) {
354   if(playing) {
355     queue_fix_sofar(playing);
356     playing->expected = 0;
357     sink_printf(ev_writer_sink(c->w), "252 %s\n", queue_marshall(playing));
358   } else
359     sink_printf(ev_writer_sink(c->w), "259 nothing playing\n");
360   return 1;                             /* completed */
361 }
362
363 static const char *connection_host(struct conn *c) {
364   union {
365     struct sockaddr sa;
366     struct sockaddr_in in;
367     struct sockaddr_in6 in6;
368   } u;
369   socklen_t l;
370   int n;
371   char host[1024];
372
373   /* get connection data */
374   l = sizeof u;
375   if(getpeername(c->fd, &u.sa, &l) < 0) {
376     error(errno, "S%x error calling getpeername", c->tag);
377     return 0;
378   }
379   if(c->l->pf != PF_UNIX) {
380     if((n = getnameinfo(&u.sa, l,
381                         host, sizeof host, 0, 0, NI_NUMERICHOST))) {
382       error(0, "S%x error calling getnameinfo: %s", c->tag, gai_strerror(n));
383       return 0;
384     }
385     return xstrdup(host);
386   } else
387     return "local";
388 }
389
390 static int c_user(struct conn *c,
391                   char **vec,
392                   int attribute((unused)) nvec) {
393   struct kvp *k;
394   const char *res, *host, *password;
395   rights_type rights;
396
397   if(c->who) {
398     sink_writes(ev_writer_sink(c->w), "530 already authenticated\n");
399     return 1;
400   }
401   /* get connection data */
402   if(!(host = connection_host(c))) {
403     sink_writes(ev_writer_sink(c->w), "530 authentication failure\n");
404     return 1;
405   }
406   /* find the user */
407   k = trackdb_getuserinfo(vec[0]);
408   /* reject nonexistent users */
409   if(!k) {
410     error(0, "S%x unknown user '%s' from %s", c->tag, vec[0], host);
411     sink_writes(ev_writer_sink(c->w), "530 authentication failed\n");
412     return 1;
413   }
414   /* reject unconfirmed users */
415   if(kvp_get(k, "confirmation")) {
416     error(0, "S%x unconfirmed user '%s' from %s", c->tag, vec[0], host);
417     sink_writes(ev_writer_sink(c->w), "530 authentication failed\n");
418     return 1;
419   }
420   password = kvp_get(k, "password");
421   if(!password) password = "";
422   if(parse_rights(kvp_get(k, "rights"), &rights, 1)) {
423     error(0, "error parsing rights for %s", vec[0]);
424     sink_writes(ev_writer_sink(c->w), "530 authentication failed\n");
425     return 1;
426   }
427   /* check whether the response is right */
428   res = authhash(c->nonce, sizeof c->nonce, password,
429                  config->authorization_algorithm);
430   if(wideopen || (res && !strcmp(res, vec[1]))) {
431     c->who = vec[0];
432     c->rights = rights;
433     /* currently we only bother logging remote connections */
434     if(strcmp(host, "local")) {
435       info("S%x %s connected from %s", c->tag, vec[0], host);
436       c->rights |= RIGHT__LOCAL;
437     }
438     sink_writes(ev_writer_sink(c->w), "230 OK\n");
439     return 1;
440   }
441   /* oops, response was wrong */
442   info("S%x authentication failure for %s from %s", c->tag, vec[0], host);
443   sink_writes(ev_writer_sink(c->w), "530 authentication failed\n");
444   return 1;
445 }
446
447 static int c_recent(struct conn *c,
448                     char attribute((unused)) **vec,
449                     int attribute((unused)) nvec) {
450   const struct queue_entry *q;
451
452   sink_writes(ev_writer_sink(c->w), "253 Tracks follow\n");
453   for(q = phead.next; q != &phead; q = q->next)
454     sink_printf(ev_writer_sink(c->w), " %s\n", queue_marshall(q));
455   sink_writes(ev_writer_sink(c->w), ".\n");
456   return 1;                             /* completed */
457 }
458
459 static int c_queue(struct conn *c,
460                    char attribute((unused)) **vec,
461                    int attribute((unused)) nvec) {
462   struct queue_entry *q;
463   time_t when = 0;
464   const char *l;
465   long length;
466
467   sink_writes(ev_writer_sink(c->w), "253 Tracks follow\n");
468   if(playing_is_enabled() && !paused) {
469     if(playing) {
470       queue_fix_sofar(playing);
471       if((l = trackdb_get(playing->track, "_length"))
472          && (length = atol(l))) {
473         time(&when);
474         when += length - playing->sofar + config->gap;
475       }
476     } else
477       /* Nothing is playing but playing is enabled, so whatever is
478        * first in the queue can be expected to start immediately. */
479       time(&when);
480   }
481   for(q = qhead.next; q != &qhead; q = q->next) {
482     /* fill in estimated start time */
483     q->expected = when;
484     sink_printf(ev_writer_sink(c->w), " %s\n", queue_marshall(q));
485     /* update for next track */
486     if(when) {
487       if((l = trackdb_get(q->track, "_length"))
488          && (length = atol(l)))
489         when += length + config->gap;
490       else
491         when = 0;
492     }
493   }
494   sink_writes(ev_writer_sink(c->w), ".\n");
495   return 1;                             /* completed */
496 }
497
498 static int output_list(struct conn *c, char **vec) {
499   while(*vec)
500     sink_printf(ev_writer_sink(c->w), "%s\n", *vec++);
501   sink_writes(ev_writer_sink(c->w), ".\n");
502   return 1;
503 }
504
505 static int files_dirs(struct conn *c,
506                       char **vec,
507                       int nvec,
508                       enum trackdb_listable what) {
509   const char *dir, *re, *errstr;
510   int erroffset;
511   pcre *rec;
512   char **fvec, *key;
513   
514   switch(nvec) {
515   case 0: dir = 0; re = 0; break;
516   case 1: dir = vec[0]; re = 0; break;
517   case 2: dir = vec[0]; re = vec[1]; break;
518   default: abort();
519   }
520   /* A bit of a bodge to make sure the args don't trample on cache keys */
521   if(dir && strchr(dir, '\n')) {
522     sink_writes(ev_writer_sink(c->w), "550 invalid directory name\n");
523     return 1;
524   }
525   if(re && strchr(re, '\n')) {
526     sink_writes(ev_writer_sink(c->w), "550 invalid regexp\n");
527     return 1;
528   }
529   /* We bother eliminating "" because the web interface is relatively
530    * likely to send it */
531   if(re && *re) {
532     byte_xasprintf(&key, "%d\n%s\n%s", (int)what, dir ? dir : "", re);
533     fvec = (char **)cache_get(&cache_files_type, key);
534     if(fvec) {
535       /* Got a cache hit, don't store the answer in the cache */
536       key = 0;
537       ++cache_files_hits;
538       rec = 0;                          /* quieten compiler */
539     } else {
540       /* Cache miss, we'll do the lookup and key != 0 so we'll store the answer
541        * in the cache. */
542       if(!(rec = pcre_compile(re, PCRE_CASELESS|PCRE_UTF8,
543                               &errstr, &erroffset, 0))) {
544         sink_printf(ev_writer_sink(c->w), "550 Error compiling regexp: %s\n",
545                     errstr);
546         return 1;
547       }
548       /* It only counts as a miss if the regexp was valid. */
549       ++cache_files_misses;
550     }
551   } else {
552     /* No regexp, don't bother caching the result */
553     rec = 0;
554     key = 0;
555     fvec = 0;
556   }
557   if(!fvec) {
558     /* No cache hit (either because a miss, or because we did not look) so do
559      * the lookup */
560     if(dir && *dir)
561       fvec = trackdb_list(dir, 0, what, rec);
562     else
563       fvec = trackdb_list(0, 0, what, rec);
564   }
565   if(key)
566     /* Put the answer in the cache */
567     cache_put(&cache_files_type, key, fvec);
568   sink_writes(ev_writer_sink(c->w), "253 Listing follow\n");
569   return output_list(c, fvec);
570 }
571
572 static int c_files(struct conn *c,
573                   char **vec,
574                   int nvec) {
575   return files_dirs(c, vec, nvec, trackdb_files);
576 }
577
578 static int c_dirs(struct conn *c,
579                   char **vec,
580                   int nvec) {
581   return files_dirs(c, vec, nvec, trackdb_directories);
582 }
583
584 static int c_allfiles(struct conn *c,
585                       char **vec,
586                       int nvec) {
587   return files_dirs(c, vec, nvec, trackdb_directories|trackdb_files);
588 }
589
590 static int c_get(struct conn *c,
591                  char **vec,
592                  int attribute((unused)) nvec) {
593   const char *v;
594
595   if(vec[1][0] != '_' && (v = trackdb_get(vec[0], vec[1])))
596     sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(v));
597   else
598     sink_writes(ev_writer_sink(c->w), "555 not found\n");
599   return 1;
600 }
601
602 static int c_length(struct conn *c,
603                  char **vec,
604                  int attribute((unused)) nvec) {
605   const char *track, *v;
606
607   if(!(track = trackdb_resolve(vec[0]))) {
608     sink_writes(ev_writer_sink(c->w), "550 cannot resolve track\n");
609     return 1;
610   }
611   if((v = trackdb_get(track, "_length")))
612     sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(v));
613   else
614     sink_writes(ev_writer_sink(c->w), "550 not found\n");
615   return 1;
616 }
617
618 static int c_set(struct conn *c,
619                  char **vec,
620                  int attribute((unused)) nvec) {
621   if(vec[1][0] != '_' && !trackdb_set(vec[0], vec[1], vec[2]))
622     sink_writes(ev_writer_sink(c->w), "250 OK\n");
623   else
624     sink_writes(ev_writer_sink(c->w), "550 not found\n");
625   return 1;
626 }
627
628 static int c_prefs(struct conn *c,
629                    char **vec,
630                    int attribute((unused)) nvec) {
631   struct kvp *k;
632
633   k = trackdb_get_all(vec[0]);
634   sink_writes(ev_writer_sink(c->w), "253 prefs follow\n");
635   for(; k; k = k->next)
636     if(k->name[0] != '_')               /* omit internal values */
637       sink_printf(ev_writer_sink(c->w),
638                   " %s %s\n", quoteutf8(k->name), quoteutf8(k->value));
639   sink_writes(ev_writer_sink(c->w), ".\n");
640   return 1;
641 }
642
643 static int c_exists(struct conn *c,
644                     char **vec,
645                     int attribute((unused)) nvec) {
646   sink_printf(ev_writer_sink(c->w), "252 %s\n", noyes[trackdb_exists(vec[0])]);
647   return 1;
648 }
649
650 static void search_parse_error(const char *msg, void *u) {
651   *(const char **)u = msg;
652 }
653
654 static int c_search(struct conn *c,
655                           char **vec,
656                           int attribute((unused)) nvec) {
657   char **terms, **results;
658   int nterms, nresults, n;
659   const char *e = "unknown error";
660
661   /* This is a bit of a bodge.  Initially it's there to make the eclient
662    * interface a bit more convenient to add searching to, but it has the more
663    * compelling advantage that if everything uses it, then interpretation of
664    * user-supplied search strings will be the same everywhere. */
665   if(!(terms = split(vec[0], &nterms, SPLIT_QUOTES, search_parse_error, &e))) {
666     sink_printf(ev_writer_sink(c->w), "550 %s\n", e);
667   } else {
668     results = trackdb_search(terms, nterms, &nresults);
669     sink_printf(ev_writer_sink(c->w), "253 %d matches\n", nresults);
670     for(n = 0; n < nresults; ++n)
671       sink_printf(ev_writer_sink(c->w), "%s\n", results[n]);
672     sink_writes(ev_writer_sink(c->w), ".\n");
673   }
674   return 1;
675 }
676
677 static int c_random_enable(struct conn *c,
678                            char attribute((unused)) **vec,
679                            int attribute((unused)) nvec) {
680   enable_random(c->who, c->ev);
681   /* Enable implicitly unpauses if there is nothing playing */
682   if(paused && !playing) resume_playing(c->who);
683   sink_writes(ev_writer_sink(c->w), "250 OK\n");
684   return 1;                     /* completed */
685 }
686
687 static int c_random_disable(struct conn *c,
688                             char attribute((unused)) **vec,
689                             int attribute((unused)) nvec) {
690   disable_random(c->who);
691   sink_writes(ev_writer_sink(c->w), "250 OK\n");
692   return 1;                     /* completed */
693 }
694
695 static int c_random_enabled(struct conn *c,
696                             char attribute((unused)) **vec,
697                             int attribute((unused)) nvec) {
698   sink_printf(ev_writer_sink(c->w), "252 %s\n", noyes[random_is_enabled()]);
699   return 1;                     /* completed */
700 }
701
702 static void got_stats(char *stats, void *u) {
703   struct conn *const c = u;
704
705   sink_printf(ev_writer_sink(c->w), "253 stats\n%s\n.\n", stats);
706   /* Now we can start processing commands again */
707   ev_reader_enable(c->r);
708 }
709
710 static int c_stats(struct conn *c,
711                    char attribute((unused)) **vec,
712                    int attribute((unused)) nvec) {
713   trackdb_stats_subprocess(c->ev, got_stats, c);
714   return 0;                             /* not yet complete */
715 }
716
717 static int c_volume(struct conn *c,
718                     char **vec,
719                     int nvec) {
720   int l, r, set;
721   char lb[32], rb[32];
722   rights_type rights;
723
724   switch(nvec) {
725   case 0:
726     set = 0;
727     break;
728   case 1:
729     l = r = atoi(vec[0]);
730     set = 1;
731     break;
732   case 2:
733     l = atoi(vec[0]);
734     r = atoi(vec[1]);
735     set = 1;
736     break;
737   default:
738     abort();
739   }
740   rights = set ? RIGHT_VOLUME : RIGHT_READ;
741   if(!(c->rights & rights)) {
742     error(0, "%s attempted to set volume but lacks required rights", c->who);
743     sink_writes(ev_writer_sink(c->w), "510 Prohibited\n");
744     return 1;
745   }
746   if(mixer_control(&l, &r, set))
747     sink_writes(ev_writer_sink(c->w), "550 error accessing mixer\n");
748   else {
749     sink_printf(ev_writer_sink(c->w), "252 %d %d\n", l, r);
750     if(l != volume_left || r != volume_right) {
751       volume_left = l;
752       volume_right = r;
753       snprintf(lb, sizeof lb, "%d", l);
754       snprintf(rb, sizeof rb, "%d", r);
755       eventlog("volume", lb, rb, (char *)0);
756     }
757   }
758   return 1;
759 }
760
761 /** @brief Called when data arrives on a log connection
762  *
763  * We just discard all such data.  The client may occasionally send data as a
764  * keepalive.
765  */
766 static int logging_reader_callback(ev_source attribute((unused)) *ev,
767                                    ev_reader *reader,
768                                    void attribute((unused)) *ptr,
769                                    size_t bytes,
770                                    int attribute((unused)) eof,
771                                    void attribute((unused)) *u) {
772   struct conn *c = u;
773
774   ev_reader_consume(reader, bytes);
775   if(eof) {
776     /* Oops, that's all for now */
777     D(("logging reader eof"));
778     if(c->w) {
779       D(("close writer"));
780       ev_writer_close(c->w);
781       c->w = 0;
782     }
783     c->r = 0;
784   }
785   return 0;
786 }
787
788 static void logclient(const char *msg, void *user) {
789   struct conn *c = user;
790
791   if(!c->w || !c->r) {
792     /* This connection has gone up in smoke for some reason */
793     eventlog_remove(c->lo);
794     return;
795   }
796   sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" %s\n",
797               (uintmax_t)time(0), msg);
798 }
799
800 static int c_log(struct conn *c,
801                  char attribute((unused)) **vec,
802                  int attribute((unused)) nvec) {
803   time_t now;
804
805   sink_writes(ev_writer_sink(c->w), "254 OK\n");
806   /* pump out initial state */
807   time(&now);
808   sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" state %s\n",
809               (uintmax_t)now, 
810               playing_is_enabled() ? "enable_play" : "disable_play");
811   sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" state %s\n",
812               (uintmax_t)now, 
813               random_is_enabled() ? "enable_random" : "disable_random");
814   sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" state %s\n",
815               (uintmax_t)now, 
816               paused ? "pause" : "resume");
817   if(playing)
818     sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" state playing\n",
819                 (uintmax_t)now);
820   /* Initial volume */
821   sink_printf(ev_writer_sink(c->w), "%"PRIxMAX" volume %d %d\n",
822               (uintmax_t)now, volume_left, volume_right);
823   c->lo = xmalloc(sizeof *c->lo);
824   c->lo->fn = logclient;
825   c->lo->user = c;
826   eventlog_add(c->lo);
827   c->reader = logging_reader_callback;
828   return 0;
829 }
830
831 /** @brief Test whether a move is allowed
832  * @param c Connection
833  * @param qs List of IDs on queue
834  * @param nqs Number of IDs
835  * @return 0 if move is prohibited, non-0 if it is allowed
836  */
837 static int has_move_rights(struct conn *c, struct queue_entry **qs, int nqs) {
838   for(; nqs > 0; ++qs, --nqs) {
839     struct queue_entry *const q = *qs;
840
841     if(!right_movable(c->rights, c->who, q))
842       return 0;
843   }
844   return 1;
845 }
846
847 static int c_move(struct conn *c,
848                   char **vec,
849                   int attribute((unused)) nvec) {
850   struct queue_entry *q;
851   int n;
852
853   if(!(q = queue_find(vec[0]))) {
854     sink_writes(ev_writer_sink(c->w), "550 no such track on the queue\n");
855     return 1;
856   }
857   if(!has_move_rights(c, &q, 1)) {
858     error(0, "%s attempted move but lacks required rights", c->who);
859     sink_writes(ev_writer_sink(c->w),
860                 "510 Not authorized to move that track\n");
861     return 1;
862   }
863   n = queue_move(q, atoi(vec[1]), c->who);
864   sink_printf(ev_writer_sink(c->w), "252 %d\n", n);
865   /* If we've moved to the head of the queue then prepare the track. */
866   if(q == qhead.next)
867     prepare(c->ev, q);
868   return 1;
869 }
870
871 static int c_moveafter(struct conn *c,
872                        char **vec,
873                        int attribute((unused)) nvec) {
874   struct queue_entry *q, **qs;
875   int n;
876
877   if(vec[0][0]) {
878     if(!(q = queue_find(vec[0]))) {
879       sink_writes(ev_writer_sink(c->w), "550 no such track on the queue\n");
880       return 1;
881     }
882   } else
883     q = 0;
884   ++vec;
885   --nvec;
886   qs = xcalloc(nvec, sizeof *qs);
887   for(n = 0; n < nvec; ++n)
888     if(!(qs[n] = queue_find(vec[n]))) {
889       sink_writes(ev_writer_sink(c->w), "550 no such track on the queue\n");
890       return 1;
891     }
892   if(!has_move_rights(c, qs, nvec)) {
893     error(0, "%s attempted moveafter but lacks required rights", c->who);
894     sink_writes(ev_writer_sink(c->w),
895                 "510 Not authorized to move those tracks\n");
896     return 1;
897   }
898   queue_moveafter(q, nvec, qs, c->who);
899   sink_printf(ev_writer_sink(c->w), "250 Moved tracks\n");
900   /* If we've moved to the head of the queue then prepare the track. */
901   if(q == qhead.next)
902     prepare(c->ev, q);
903   return 1;
904 }
905
906 static int c_part(struct conn *c,
907                   char **vec,
908                   int attribute((unused)) nvec) {
909   sink_printf(ev_writer_sink(c->w), "252 %s\n",
910               quoteutf8(trackdb_getpart(vec[0], vec[1], vec[2])));
911   return 1;
912 }
913
914 static int c_resolve(struct conn *c,
915                      char **vec,
916                      int attribute((unused)) nvec) {
917   const char *track;
918
919   if(!(track = trackdb_resolve(vec[0]))) {
920     sink_writes(ev_writer_sink(c->w), "550 cannot resolve track\n");
921     return 1;
922   }
923   sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(track));
924   return 1;
925 }
926
927 static int c_tags(struct conn *c,
928                   char attribute((unused)) **vec,
929                   int attribute((unused)) nvec) {
930   char **tags = trackdb_alltags();
931   
932   sink_printf(ev_writer_sink(c->w), "253 Tag list follows\n");
933   while(*tags) {
934     sink_printf(ev_writer_sink(c->w), "%s%s\n",
935                 **tags == '.' ? "." : "", *tags);
936     ++tags;
937   }
938   sink_writes(ev_writer_sink(c->w), ".\n");
939   return 1;                             /* completed */
940 }
941
942 static int c_set_global(struct conn *c,
943                         char **vec,
944                         int attribute((unused)) nvec) {
945   if(vec[0][0] == '_') {
946     sink_writes(ev_writer_sink(c->w), "550 cannot set internal global preferences\n");
947     return 1;
948   }
949   trackdb_set_global(vec[0], vec[1], c->who);
950   sink_printf(ev_writer_sink(c->w), "250 OK\n");
951   return 1;
952 }
953
954 static int c_get_global(struct conn *c,
955                         char **vec,
956                         int attribute((unused)) nvec) {
957   const char *s = trackdb_get_global(vec[0]);
958
959   if(s)
960     sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(s));
961   else
962     sink_writes(ev_writer_sink(c->w), "555 not found\n");
963   return 1;
964 }
965
966 static int c_nop(struct conn *c,
967                  char attribute((unused)) **vec,
968                  int attribute((unused)) nvec) {
969   sink_printf(ev_writer_sink(c->w), "250 Quack\n");
970   return 1;
971 }
972
973 static int c_new(struct conn *c,
974                  char **vec,
975                  int nvec) {
976   char **tracks = trackdb_new(0, nvec > 0 ? atoi(vec[0]) : INT_MAX);
977
978   sink_printf(ev_writer_sink(c->w), "253 New track list follows\n");
979   while(*tracks) {
980     sink_printf(ev_writer_sink(c->w), "%s%s\n",
981                 **tracks == '.' ? "." : "", *tracks);
982     ++tracks;
983   }
984   sink_writes(ev_writer_sink(c->w), ".\n");
985   return 1;                             /* completed */
986
987 }
988
989 static int c_rtp_address(struct conn *c,
990                          char attribute((unused)) **vec,
991                          int attribute((unused)) nvec) {
992   if(config->speaker_backend == BACKEND_NETWORK) {
993     sink_printf(ev_writer_sink(c->w), "252 %s %s\n",
994                 quoteutf8(config->broadcast.s[0]),
995                 quoteutf8(config->broadcast.s[1]));
996   } else
997     sink_writes(ev_writer_sink(c->w), "550 No RTP\n");
998   return 1;
999 }
1000
1001 static int c_cookie(struct conn *c,
1002                     char **vec,
1003                     int attribute((unused)) nvec) {
1004   const char *host;
1005   char *user;
1006   rights_type rights;
1007
1008   /* Can't log in twice on the same connection */
1009   if(c->who) {
1010     sink_writes(ev_writer_sink(c->w), "530 already authenticated\n");
1011     return 1;
1012   }
1013   /* Get some kind of peer identifcation */
1014   if(!(host = connection_host(c))) {
1015     sink_writes(ev_writer_sink(c->w), "530 authentication failure\n");
1016     return 1;
1017   }
1018   /* Check the cookie */
1019   user = verify_cookie(vec[0], &rights);
1020   if(!user) {
1021     sink_writes(ev_writer_sink(c->w), "530 authentication failure\n");
1022     return 1;
1023   }
1024   /* Log in */
1025   c->who = user;
1026   c->cookie = vec[0];
1027   c->rights = rights;
1028   if(strcmp(host, "local")) {
1029     info("S%x %s connected with cookie from %s", c->tag, user, host);
1030     c->rights |= RIGHT__LOCAL;
1031   }
1032   /* Response contains username so client knows who they are acting as */
1033   sink_printf(ev_writer_sink(c->w), "232 %s\n", quoteutf8(user));
1034   return 1;
1035 }
1036
1037 static int c_make_cookie(struct conn *c,
1038                          char attribute((unused)) **vec,
1039                          int attribute((unused)) nvec) {
1040   const char *cookie = make_cookie(c->who);
1041
1042   if(cookie)
1043     sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(cookie));
1044   else
1045     sink_writes(ev_writer_sink(c->w), "550 Cannot create cookie\n");
1046   return 1;
1047 }
1048
1049 static int c_revoke(struct conn *c,
1050                     char attribute((unused)) **vec,
1051                     int attribute((unused)) nvec) {
1052   if(c->cookie) {
1053     revoke_cookie(c->cookie);
1054     sink_writes(ev_writer_sink(c->w), "250 OK\n");
1055   } else
1056     sink_writes(ev_writer_sink(c->w), "550 Did not log in with cookie\n");
1057   return 1;
1058 }
1059
1060 static int c_adduser(struct conn *c,
1061                      char **vec,
1062                      int nvec) {
1063   const char *rights;
1064
1065   if(nvec > 2) {
1066     rights = vec[2];
1067     if(parse_rights(vec[2], 0, 1)) {
1068       sink_writes(ev_writer_sink(c->w), "550 Invalid rights list\n");
1069       return -1;
1070     }
1071   } else
1072     rights = config->default_rights;
1073   if(trackdb_adduser(vec[0], vec[1], rights,
1074                      0/*email*/, 0/*confirmation*/))
1075     sink_writes(ev_writer_sink(c->w), "550 Cannot create user\n");
1076   else
1077     sink_writes(ev_writer_sink(c->w), "250 User created\n");
1078   return 1;
1079 }
1080
1081 static int c_deluser(struct conn *c,
1082                      char **vec,
1083                      int attribute((unused)) nvec) {
1084   if(trackdb_deluser(vec[0]))
1085     sink_writes(ev_writer_sink(c->w), "550 Cannot delete user\n");
1086   else
1087     sink_writes(ev_writer_sink(c->w), "250 User deleted\n");
1088   return 1;
1089 }
1090
1091 static int c_edituser(struct conn *c,
1092                       char **vec,
1093                       int attribute((unused)) nvec) {
1094   /* RIGHT_ADMIN can do anything; otherwise you can only set your own email
1095    * address and password. */
1096   if((c->rights & RIGHT_ADMIN)
1097      || (!strcmp(c->who, vec[0])
1098          && (!strcmp(vec[1], "email")
1099              || !strcmp(vec[1], "password")))) {
1100     if(trackdb_edituserinfo(vec[0], vec[1], vec[2]))
1101       sink_writes(ev_writer_sink(c->w), "550 Failed to change setting\n");
1102     else
1103       sink_writes(ev_writer_sink(c->w), "250 OK\n");
1104   } else {
1105     error(0, "%s attempted edituser but lacks required rights", c->who);
1106     sink_writes(ev_writer_sink(c->w), "510 Restricted to administrators\n");
1107   }
1108   return 1;
1109 }
1110
1111 static int c_userinfo(struct conn *c,
1112                       char attribute((unused)) **vec,
1113                       int attribute((unused)) nvec) {
1114   struct kvp *k;
1115   const char *value;
1116
1117   /* RIGHT_ADMIN allows anything; otherwise you can only get your own email
1118    * address and righst list. */
1119   if((c->rights & RIGHT_ADMIN)
1120      || (!strcmp(c->who, vec[0])
1121          && (!strcmp(vec[1], "email")
1122              || !strcmp(vec[1], "rights")))) {
1123     if((k = trackdb_getuserinfo(vec[0])))
1124       if((value = kvp_get(k, vec[1])))
1125         sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(value));
1126       else
1127         sink_writes(ev_writer_sink(c->w), "555 Not set\n");
1128     else
1129       sink_writes(ev_writer_sink(c->w), "550 No such user\n");
1130   } else {
1131     error(0, "%s attempted userinfo but lacks required rights", c->who);
1132     sink_writes(ev_writer_sink(c->w), "510 Restricted to administrators\n");
1133   }
1134   return 1;
1135 }
1136
1137 static int c_users(struct conn *c,
1138                    char attribute((unused)) **vec,
1139                    int attribute((unused)) nvec) {
1140   /* TODO de-dupe with c_tags */
1141   char **users = trackdb_listusers();
1142
1143   sink_writes(ev_writer_sink(c->w), "253 User list follows\n");
1144   while(*users) {
1145     sink_printf(ev_writer_sink(c->w), "%s%s\n",
1146                 **users == '.' ? "." : "", *users);
1147     ++users;
1148   }
1149   sink_writes(ev_writer_sink(c->w), ".\n");
1150   return 1;                             /* completed */
1151 }
1152
1153 static int c_register(struct conn *c,
1154                       char **vec,
1155                       int attribute((unused)) nvec) {
1156   char *buf, *cs;
1157   size_t bufsize;
1158   int offset;
1159
1160   /* The confirmation string is base64(username;nonce) */
1161   bufsize = strlen(vec[0]) + NONCE_SIZE + 2;
1162   buf = xmalloc_noptr(bufsize);
1163   offset = byte_snprintf(buf, bufsize, "%s;", vec[0]);
1164   gcry_randomize(buf + offset, NONCE_SIZE, GCRY_STRONG_RANDOM);
1165   cs = mime_to_base64((uint8_t *)buf, offset + NONCE_SIZE);
1166   if(trackdb_adduser(vec[0], vec[1], config->default_rights, vec[2], cs))
1167     sink_writes(ev_writer_sink(c->w), "550 Cannot create user\n");
1168   else
1169     sink_printf(ev_writer_sink(c->w), "252 %s\n", quoteutf8(cs));
1170   return 1;
1171 }
1172
1173 static int c_confirm(struct conn *c,
1174                      char **vec,
1175                      int attribute((unused)) nvec) {
1176   size_t nuser;
1177   char *user, *sep;
1178
1179   if(!(user = mime_base64(vec[0], &nuser))
1180      || !(sep = memchr(user, ';', nuser))) {
1181     sink_writes(ev_writer_sink(c->w), "550 Malformed confirmation string\n");
1182     return 1;
1183   }
1184   *sep = 0;
1185   if(trackdb_confirm(user, vec[0]))
1186     sink_writes(ev_writer_sink(c->w), "550 Incorrect confirmation string\n");
1187   else
1188     sink_writes(ev_writer_sink(c->w), "250 OK\n");
1189   return 1;
1190 }
1191  
1192 static const struct command {
1193   /** @brief Command name */
1194   const char *name;
1195
1196   /** @brief Minimum number of arguments */
1197   int minargs;
1198
1199   /** @brief Maximum number of arguments */
1200   int maxargs;
1201
1202   /** @brief Function to process command */
1203   int (*fn)(struct conn *, char **, int);
1204
1205   /** @brief Rights required to execute command
1206    *
1207    * 0 means that the command can be issued without logging in.  If multiple
1208    * bits are listed here any of those rights will do.
1209    */
1210   rights_type rights;
1211 } commands[] = {
1212   { "adduser",        2, 3,       c_adduser,        RIGHT_ADMIN|RIGHT__LOCAL },
1213   { "allfiles",       0, 2,       c_allfiles,       RIGHT_READ },
1214   { "confirm",        1, 1,       c_confirm,        0 },
1215   { "cookie",         1, 1,       c_cookie,         0 },
1216   { "deluser",        1, 1,       c_deluser,        RIGHT_ADMIN|RIGHT__LOCAL },
1217   { "dirs",           0, 2,       c_dirs,           RIGHT_READ },
1218   { "disable",        0, 1,       c_disable,        RIGHT_GLOBAL_PREFS },
1219   { "edituser",       3, 3,       c_edituser,       RIGHT_ADMIN|RIGHT_USERINFO },
1220   { "enable",         0, 0,       c_enable,         RIGHT_GLOBAL_PREFS },
1221   { "enabled",        0, 0,       c_enabled,        RIGHT_READ },
1222   { "exists",         1, 1,       c_exists,         RIGHT_READ },
1223   { "files",          0, 2,       c_files,          RIGHT_READ },
1224   { "get",            2, 2,       c_get,            RIGHT_READ },
1225   { "get-global",     1, 1,       c_get_global,     RIGHT_READ },
1226   { "length",         1, 1,       c_length,         RIGHT_READ },
1227   { "log",            0, 0,       c_log,            RIGHT_READ },
1228   { "make-cookie",    0, 0,       c_make_cookie,    RIGHT_READ },
1229   { "move",           2, 2,       c_move,           RIGHT_MOVE__MASK },
1230   { "moveafter",      1, INT_MAX, c_moveafter,      RIGHT_MOVE__MASK },
1231   { "new",            0, 1,       c_new,            RIGHT_READ },
1232   { "nop",            0, 0,       c_nop,            0 },
1233   { "part",           3, 3,       c_part,           RIGHT_READ },
1234   { "pause",          0, 0,       c_pause,          RIGHT_PAUSE },
1235   { "play",           1, 1,       c_play,           RIGHT_PLAY },
1236   { "playing",        0, 0,       c_playing,        RIGHT_READ },
1237   { "prefs",          1, 1,       c_prefs,          RIGHT_READ },
1238   { "queue",          0, 0,       c_queue,          RIGHT_READ },
1239   { "random-disable", 0, 0,       c_random_disable, RIGHT_GLOBAL_PREFS },
1240   { "random-enable",  0, 0,       c_random_enable,  RIGHT_GLOBAL_PREFS },
1241   { "random-enabled", 0, 0,       c_random_enabled, RIGHT_READ },
1242   { "recent",         0, 0,       c_recent,         RIGHT_READ },
1243   { "reconfigure",    0, 0,       c_reconfigure,    RIGHT_ADMIN },
1244   { "register",       3, 3,       c_register,       RIGHT_REGISTER|RIGHT__LOCAL },
1245   { "remove",         1, 1,       c_remove,         RIGHT_REMOVE__MASK },
1246   { "rescan",         0, 0,       c_rescan,         RIGHT_RESCAN },
1247   { "resolve",        1, 1,       c_resolve,        RIGHT_READ },
1248   { "resume",         0, 0,       c_resume,         RIGHT_PAUSE },
1249   { "revoke",         0, 0,       c_revoke,         RIGHT_READ },
1250   { "rtp-address",    0, 0,       c_rtp_address,    0 },
1251   { "scratch",        0, 1,       c_scratch,        RIGHT_SCRATCH__MASK },
1252   { "search",         1, 1,       c_search,         RIGHT_READ },
1253   { "set",            3, 3,       c_set,            RIGHT_PREFS, },
1254   { "set-global",     2, 2,       c_set_global,     RIGHT_GLOBAL_PREFS },
1255   { "shutdown",       0, 0,       c_shutdown,       RIGHT_ADMIN },
1256   { "stats",          0, 0,       c_stats,          RIGHT_READ },
1257   { "tags",           0, 0,       c_tags,           RIGHT_READ },
1258   { "unset",          2, 2,       c_set,            RIGHT_PREFS },
1259   { "unset-global",   1, 1,       c_set_global,     RIGHT_GLOBAL_PREFS },
1260   { "user",           2, 2,       c_user,           0 },
1261   { "userinfo",       2, 2,       c_userinfo,       RIGHT_READ },
1262   { "users",          0, 0,       c_users,          RIGHT_READ },
1263   { "version",        0, 0,       c_version,        RIGHT_READ },
1264   { "volume",         0, 2,       c_volume,         RIGHT_READ|RIGHT_VOLUME }
1265 };
1266
1267 static void command_error(const char *msg, void *u) {
1268   struct conn *c = u;
1269
1270   sink_printf(ev_writer_sink(c->w), "500 parse error: %s\n", msg);
1271 }
1272
1273 /* process a command.  Return 1 if complete, 0 if incomplete. */
1274 static int command(struct conn *c, char *line) {
1275   char **vec;
1276   int nvec, n;
1277
1278   D(("server command %s", line));
1279   /* We force everything into NFC as early as possible */
1280   if(!(line = utf8_compose_canon(line, strlen(line), 0))) {
1281     sink_writes(ev_writer_sink(c->w), "500 cannot normalize command\n");
1282     return 1;
1283   }
1284   if(!(vec = split(line, &nvec, SPLIT_QUOTES, command_error, c))) {
1285     sink_writes(ev_writer_sink(c->w), "500 cannot parse command\n");
1286     return 1;
1287   }
1288   if(nvec == 0) {
1289     sink_writes(ev_writer_sink(c->w), "500 do what?\n");
1290     return 1;
1291   }
1292   if((n = TABLE_FIND(commands, struct command, name, vec[0])) < 0)
1293     sink_writes(ev_writer_sink(c->w), "500 unknown command\n");
1294   else {
1295     if(commands[n].rights
1296        && !(c->rights & commands[n].rights)) {
1297       error(0, "%s attempted %s but lacks required rights", c->who ? c->who : "NULL",
1298             commands[n].name);
1299       sink_writes(ev_writer_sink(c->w), "510 Prohibited\n");
1300       return 1;
1301     }
1302     ++vec;
1303     --nvec;
1304     if(nvec < commands[n].minargs) {
1305       sink_writes(ev_writer_sink(c->w), "500 missing argument(s)\n");
1306       return 1;
1307     }
1308     if(nvec > commands[n].maxargs) {
1309       sink_writes(ev_writer_sink(c->w), "500 too many arguments\n");
1310       return 1;
1311     }
1312     return commands[n].fn(c, vec, nvec);
1313   }
1314   return 1;                     /* completed */
1315 }
1316
1317 /* redirect to the right reader callback for our current state */
1318 static int redirect_reader_callback(ev_source *ev,
1319                                     ev_reader *reader,
1320                                     void *ptr,
1321                                     size_t bytes,
1322                                     int eof,
1323                                     void *u) {
1324   struct conn *c = u;
1325
1326   return c->reader(ev, reader, ptr, bytes, eof, u);
1327 }
1328
1329 /* the main command reader */
1330 static int reader_callback(ev_source attribute((unused)) *ev,
1331                            ev_reader *reader,
1332                            void *ptr,
1333                            size_t bytes,
1334                            int eof,
1335                            void *u) {
1336   struct conn *c = u;
1337   char *eol;
1338   int complete;
1339
1340   D(("server reader_callback"));
1341   while((eol = memchr(ptr, '\n', bytes))) {
1342     *eol++ = 0;
1343     ev_reader_consume(reader, eol - (char *)ptr);
1344     complete = command(c, ptr);
1345     bytes -= (eol - (char *)ptr);
1346     ptr = eol;
1347     if(!complete) {
1348       /* the command had better have set a new reader callback */
1349       if(bytes || eof)
1350         /* there are further bytes to read, or we are at eof; arrange for the
1351          * command's reader callback to handle them */
1352         return ev_reader_incomplete(reader);
1353       /* nothing's going on right now */
1354       return 0;
1355     }
1356     /* command completed, we can go around and handle the next one */
1357   }
1358   if(eof) {
1359     if(bytes)
1360       error(0, "S%x unterminated line", c->tag);
1361     D(("normal reader close"));
1362     c->r = 0;
1363     if(c->w) {
1364       D(("close associated writer"));
1365       ev_writer_close(c->w);
1366       c->w = 0;
1367     }
1368   }
1369   return 0;
1370 }
1371
1372 static int listen_callback(ev_source *ev,
1373                            int fd,
1374                            const struct sockaddr attribute((unused)) *remote,
1375                            socklen_t attribute((unused)) rlen,
1376                            void *u) {
1377   const struct listener *l = u;
1378   struct conn *c = xmalloc(sizeof *c);
1379   static unsigned tags;
1380
1381   D(("server listen_callback fd %d (%s)", fd, l->name));
1382   nonblock(fd);
1383   cloexec(fd);
1384   c->tag = tags++;
1385   c->ev = ev;
1386   c->w = ev_writer_new(ev, fd, writer_error, c,
1387                        "client writer");
1388   c->r = ev_reader_new(ev, fd, redirect_reader_callback, reader_error, c,
1389                        "client reader");
1390   ev_tie(c->r, c->w);
1391   c->fd = fd;
1392   c->reader = reader_callback;
1393   c->l = l;
1394   c->rights = 0;
1395   gcry_randomize(c->nonce, sizeof c->nonce, GCRY_STRONG_RANDOM);
1396   sink_printf(ev_writer_sink(c->w), "231 %d %s %s\n",
1397               2,
1398               config->authorization_algorithm,
1399               hex(c->nonce, sizeof c->nonce));
1400   return 0;
1401 }
1402
1403 int server_start(ev_source *ev, int pf,
1404                  size_t socklen, const struct sockaddr *sa,
1405                  const char *name) {
1406   int fd;
1407   struct listener *l = xmalloc(sizeof *l);
1408   static const int one = 1;
1409
1410   D(("server_init socket %s", name));
1411   fd = xsocket(pf, SOCK_STREAM, 0);
1412   xsetsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof one);
1413   if(bind(fd, sa, socklen) < 0) {
1414     error(errno, "error binding to %s", name);
1415     return -1;
1416   }
1417   xlisten(fd, 128);
1418   nonblock(fd);
1419   cloexec(fd);
1420   l->name = name;
1421   l->pf = pf;
1422   if(ev_listen(ev, fd, listen_callback, l, "server listener"))
1423     exit(EXIT_FAILURE);
1424   return fd;
1425 }
1426
1427 int server_stop(ev_source *ev, int fd) {
1428   xclose(fd);
1429   return ev_listen_cancel(ev, fd);
1430 }
1431
1432 /*
1433 Local Variables:
1434 c-basic-offset:2
1435 comment-column:40
1436 fill-column:79
1437 End:
1438 */