Add signal handlers to shairplay to make it shut down gracefully
[deb_shairplay.git] / src / lib / raop.c
1 /**
2 * Copyright (C) 2011-2012 Juho Vähä-Herttua
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library 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 GNU
12 * Lesser General Public License for more details.
13 */
14
15 #include <stdlib.h>
16 #include <stdio.h>
17 #include <string.h>
18 #include <assert.h>
19
20 #include "raop.h"
21 #include "raop_rtp.h"
22 #include "rsakey.h"
23 #include "digest.h"
24 #include "httpd.h"
25 #include "sdp.h"
26
27 #include "global.h"
28 #include "utils.h"
29 #include "netutils.h"
30 #include "logger.h"
31 #include "compat.h"
32
33 /* Actually 345 bytes for 2048-bit key */
34 #define MAX_SIGNATURE_LEN 512
35
36 /* Let's just decide on some length */
37 #define MAX_PASSWORD_LEN 64
38
39 /* MD5 as hex fits here */
40 #define MAX_NONCE_LEN 32
41
42 struct raop_s {
43 /* Callbacks for audio */
44 raop_callbacks_t callbacks;
45
46 /* Logger instance */
47 logger_t *logger;
48
49 /* HTTP daemon and RSA key */
50 httpd_t *httpd;
51 rsakey_t *rsakey;
52
53 /* Hardware address information */
54 unsigned char hwaddr[MAX_HWADDR_LEN];
55 int hwaddrlen;
56
57 /* Password information */
58 char password[MAX_PASSWORD_LEN+1];
59 };
60
61 struct raop_conn_s {
62 raop_t *raop;
63 raop_rtp_t *raop_rtp;
64
65 unsigned char *local;
66 int locallen;
67
68 unsigned char *remote;
69 int remotelen;
70
71 char nonce[MAX_NONCE_LEN+1];
72 };
73 typedef struct raop_conn_s raop_conn_t;
74
75 static void *
76 conn_init(void *opaque, unsigned char *local, int locallen, unsigned char *remote, int remotelen)
77 {
78 raop_conn_t *conn;
79
80 conn = calloc(1, sizeof(raop_conn_t));
81 if (!conn) {
82 return NULL;
83 }
84 conn->raop = opaque;
85 conn->raop_rtp = NULL;
86
87 if (locallen == 4) {
88 logger_log(conn->raop->logger, LOGGER_INFO,
89 "Local: %d.%d.%d.%d",
90 local[0], local[1], local[2], local[3]);
91 } else if (locallen == 16) {
92 logger_log(conn->raop->logger, LOGGER_INFO,
93 "Local: %02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x",
94 local[0], local[1], local[2], local[3], local[4], local[5], local[6], local[7],
95 local[8], local[9], local[10], local[11], local[12], local[13], local[14], local[15]);
96 }
97 if (remotelen == 4) {
98 logger_log(conn->raop->logger, LOGGER_INFO,
99 "Remote: %d.%d.%d.%d",
100 remote[0], remote[1], remote[2], remote[3]);
101 } else if (remotelen == 16) {
102 logger_log(conn->raop->logger, LOGGER_INFO,
103 "Remote: %02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x",
104 remote[0], remote[1], remote[2], remote[3], remote[4], remote[5], remote[6], remote[7],
105 remote[8], remote[9], remote[10], remote[11], remote[12], remote[13], remote[14], remote[15]);
106 }
107
108 conn->local = malloc(locallen);
109 assert(conn->local);
110 memcpy(conn->local, local, locallen);
111
112 conn->remote = malloc(remotelen);
113 assert(conn->remote);
114 memcpy(conn->remote, remote, remotelen);
115
116 conn->locallen = locallen;
117 conn->remotelen = remotelen;
118
119 digest_generate_nonce(conn->nonce, sizeof(conn->nonce));
120 return conn;
121 }
122
123 static void
124 conn_request(void *ptr, http_request_t *request, http_response_t **response)
125 {
126 raop_conn_t *conn = ptr;
127 raop_t *raop = conn->raop;
128
129 http_response_t *res;
130 const char *method;
131 const char *cseq;
132 const char *challenge;
133 int require_auth = 0;
134
135 method = http_request_get_method(request);
136 cseq = http_request_get_header(request, "CSeq");
137 if (!method || !cseq) {
138 return;
139 }
140
141 res = http_response_init("RTSP/1.0", 200, "OK");
142 if (strlen(raop->password)) {
143 const char *authorization;
144
145 authorization = http_request_get_header(request, "Authorization");
146 if (authorization) {
147 logger_log(conn->raop->logger, LOGGER_DEBUG, "Our nonce: %s", conn->nonce);
148 logger_log(conn->raop->logger, LOGGER_DEBUG, "Authorization: %s", authorization);
149 }
150 if (!digest_is_valid("AppleTV", raop->password, conn->nonce, method, http_request_get_url(request), authorization)) {
151 char *authstr;
152 int authstrlen;
153
154 /* Allocate the authenticate string */
155 authstrlen = sizeof("Digest realm=\"AppleTV\", nonce=\"\"") + sizeof(conn->nonce) + 1;
156 authstr = malloc(authstrlen);
157
158 /* Concatenate the authenticate string */
159 memset(authstr, 0, authstrlen);
160 strcat(authstr, "Digest realm=\"AppleTV\", nonce=\"");
161 strcat(authstr, conn->nonce);
162 strcat(authstr, "\"");
163
164 /* Construct a new response */
165 require_auth = 1;
166 http_response_destroy(res);
167 res = http_response_init("RTSP/1.0", 401, "Unauthorized");
168 http_response_add_header(res, "WWW-Authenticate", authstr);
169 free(authstr);
170 } else {
171 logger_log(conn->raop->logger, LOGGER_DEBUG, "AUTHENTICATION SUCCESS!");
172 }
173 }
174
175 http_response_add_header(res, "CSeq", cseq);
176 http_response_add_header(res, "Apple-Jack-Status", "connected; type=analog");
177
178 challenge = http_request_get_header(request, "Apple-Challenge");
179 if (!require_auth && challenge) {
180 char signature[MAX_SIGNATURE_LEN];
181
182 memset(signature, 0, sizeof(signature));
183 rsakey_sign(raop->rsakey, signature, sizeof(signature), challenge,
184 conn->local, conn->locallen, raop->hwaddr, raop->hwaddrlen);
185 http_response_add_header(res, "Apple-Response", signature);
186
187 logger_log(conn->raop->logger, LOGGER_DEBUG, "Got challenge: %s", challenge);
188 logger_log(conn->raop->logger, LOGGER_DEBUG, "Got response: %s", signature);
189 }
190
191 if (require_auth) {
192 /* Do nothing in case of authentication request */
193 } else if (!strcmp(method, "OPTIONS")) {
194 http_response_add_header(res, "Public", "ANNOUNCE, SETUP, RECORD, PAUSE, FLUSH, TEARDOWN, OPTIONS, GET_PARAMETER, SET_PARAMETER");
195 } else if (!strcmp(method, "ANNOUNCE")) {
196 const char *data;
197 int datalen;
198
199 unsigned char aeskey[16];
200 unsigned char aesiv[16];
201 int aeskeylen, aesivlen;
202
203 data = http_request_get_data(request, &datalen);
204 if (data) {
205 sdp_t *sdp;
206 const char *remotestr, *rtpmapstr, *fmtpstr, *aeskeystr, *aesivstr;
207
208 sdp = sdp_init(data, datalen);
209 remotestr = sdp_get_connection(sdp);
210 rtpmapstr = sdp_get_rtpmap(sdp);
211 fmtpstr = sdp_get_fmtp(sdp);
212 aeskeystr = sdp_get_rsaaeskey(sdp);
213 aesivstr = sdp_get_aesiv(sdp);
214
215 logger_log(conn->raop->logger, LOGGER_DEBUG, "connection: %s", remotestr);
216 logger_log(conn->raop->logger, LOGGER_DEBUG, "rtpmap: %s", rtpmapstr);
217 logger_log(conn->raop->logger, LOGGER_DEBUG, "fmtp: %s", fmtpstr);
218 logger_log(conn->raop->logger, LOGGER_DEBUG, "rsaaeskey: %s", aeskeystr);
219 logger_log(conn->raop->logger, LOGGER_DEBUG, "aesiv: %s", aesivstr);
220
221 aeskeylen = rsakey_decrypt(raop->rsakey, aeskey, sizeof(aeskey), aeskeystr);
222 aesivlen = rsakey_parseiv(raop->rsakey, aesiv, sizeof(aesiv), aesivstr);
223 logger_log(conn->raop->logger, LOGGER_DEBUG, "aeskeylen: %d", aeskeylen);
224 logger_log(conn->raop->logger, LOGGER_DEBUG, "aesivlen: %d", aesivlen);
225
226 if (conn->raop_rtp) {
227 /* This should never happen */
228 raop_rtp_destroy(conn->raop_rtp);
229 conn->raop_rtp = NULL;
230 }
231 conn->raop_rtp = raop_rtp_init(raop->logger, &raop->callbacks, remotestr, rtpmapstr, fmtpstr, aeskey, aesiv);
232 if (!conn->raop_rtp) {
233 logger_log(conn->raop->logger, LOGGER_ERR, "Error initializing the audio decoder");
234 http_response_set_disconnect(res, 1);
235 }
236 sdp_destroy(sdp);
237 }
238 } else if (!strcmp(method, "SETUP")) {
239 unsigned short remote_cport=0, remote_tport=0;
240 unsigned short cport=0, tport=0, dport=0;
241 const char *transport;
242 char buffer[1024];
243 int use_udp;
244
245 transport = http_request_get_header(request, "Transport");
246 assert(transport);
247
248 logger_log(conn->raop->logger, LOGGER_INFO, "Transport: %s", transport);
249 use_udp = strncmp(transport, "RTP/AVP/TCP", 11);
250 if (use_udp) {
251 char *original, *current, *tmpstr;
252
253 current = original = strdup(transport);
254 if (original) {
255 while ((tmpstr = utils_strsep(&current, ";")) != NULL) {
256 unsigned short value;
257 int ret;
258
259 ret = sscanf(tmpstr, "control_port=%hu", &value);
260 if (ret == 1) {
261 logger_log(conn->raop->logger, LOGGER_DEBUG, "Found remote control port: %hu", value);
262 remote_cport = value;
263 }
264 ret = sscanf(tmpstr, "timing_port=%hu", &value);
265 if (ret == 1) {
266 logger_log(conn->raop->logger, LOGGER_DEBUG, "Found remote timing port: %hu", value);
267 remote_tport = value;
268 }
269 }
270 }
271 free(original);
272 }
273 if (conn->raop_rtp) {
274 raop_rtp_start(conn->raop_rtp, use_udp, remote_cport, remote_tport, &cport, &tport, &dport);
275 } else {
276 logger_log(conn->raop->logger, LOGGER_ERR, "RAOP not initialized at SETUP, playing will fail!");
277 http_response_set_disconnect(res, 1);
278 }
279
280 memset(buffer, 0, sizeof(buffer));
281 if (use_udp) {
282 snprintf(buffer, sizeof(buffer)-1,
283 "RTP/AVP/UDP;unicast;mode=record;timing_port=%hu;events;control_port=%hu;server_port=%hu",
284 tport, cport, dport);
285 } else {
286 snprintf(buffer, sizeof(buffer)-1,
287 "RTP/AVP/TCP;unicast;interleaved=0-1;mode=record;server_port=%u",
288 dport);
289 }
290 logger_log(conn->raop->logger, LOGGER_INFO, "Responding with %s", buffer);
291 http_response_add_header(res, "Transport", buffer);
292 http_response_add_header(res, "Session", "DEADBEEF");
293 } else if (!strcmp(method, "SET_PARAMETER")) {
294 const char *content_type;
295 const char *data;
296 int datalen;
297
298 content_type = http_request_get_header(request, "Content-Type");
299 data = http_request_get_data(request, &datalen);
300 if (!strcmp(content_type, "text/parameters")) {
301 char *datastr;
302 datastr = calloc(1, datalen+1);
303 if (data && datastr && conn->raop_rtp) {
304 memcpy(datastr, data, datalen);
305 if (!strncmp(datastr, "volume: ", 8)) {
306 float vol = 0.0;
307 sscanf(datastr+8, "%f", &vol);
308 raop_rtp_set_volume(conn->raop_rtp, vol);
309 }
310 } else if (!conn->raop_rtp) {
311 logger_log(conn->raop->logger, LOGGER_WARNING, "RAOP not initialized at SET_PARAMETER volume");
312 }
313 free(datastr);
314 } else if (!strcmp(content_type, "image/jpeg")) {
315 logger_log(conn->raop->logger, LOGGER_INFO, "Got image data of %d bytes", datalen);
316 if (conn->raop_rtp) {
317 raop_rtp_set_coverart(conn->raop_rtp, data, datalen);
318 } else {
319 logger_log(conn->raop->logger, LOGGER_WARNING, "RAOP not initialized at SET_PARAMETER coverart");
320 }
321 } else if (!strcmp(content_type, "application/x-dmap-tagged")) {
322 logger_log(conn->raop->logger, LOGGER_INFO, "Got metadata of %d bytes", datalen);
323 if (conn->raop_rtp) {
324 raop_rtp_set_metadata(conn->raop_rtp, data, datalen);
325 } else {
326 logger_log(conn->raop->logger, LOGGER_WARNING, "RAOP not initialized at SET_PARAMETER metadata");
327 }
328 }
329 } else if (!strcmp(method, "FLUSH")) {
330 const char *rtpinfo;
331 int next_seq = -1;
332
333 rtpinfo = http_request_get_header(request, "RTP-Info");
334 if (rtpinfo) {
335 logger_log(conn->raop->logger, LOGGER_INFO, "Flush with RTP-Info: %s", rtpinfo);
336 if (!strncmp(rtpinfo, "seq=", 4)) {
337 next_seq = strtol(rtpinfo+4, NULL, 10);
338 }
339 }
340 if (conn->raop_rtp) {
341 raop_rtp_flush(conn->raop_rtp, next_seq);
342 } else {
343 logger_log(conn->raop->logger, LOGGER_WARNING, "RAOP not initialized at FLUSH");
344 }
345 } else if (!strcmp(method, "TEARDOWN")) {
346 http_response_add_header(res, "Connection", "close");
347 if (conn->raop_rtp) {
348 /* Destroy our RTP session */
349 raop_rtp_stop(conn->raop_rtp);
350 raop_rtp_destroy(conn->raop_rtp);
351 conn->raop_rtp = NULL;
352 }
353 }
354 http_response_finish(res, NULL, 0);
355
356 logger_log(conn->raop->logger, LOGGER_DEBUG, "Handled request %s with URL %s", method, http_request_get_url(request));
357 *response = res;
358 }
359
360 static void
361 conn_destroy(void *ptr)
362 {
363 raop_conn_t *conn = ptr;
364
365 if (conn->raop_rtp) {
366 /* This is done in case TEARDOWN was not called */
367 raop_rtp_destroy(conn->raop_rtp);
368 }
369 free(conn->local);
370 free(conn->remote);
371 free(conn);
372 }
373
374 raop_t *
375 raop_init(int max_clients, raop_callbacks_t *callbacks, const char *pemkey, int *error)
376 {
377 raop_t *raop;
378 httpd_t *httpd;
379 rsakey_t *rsakey;
380 httpd_callbacks_t httpd_cbs;
381
382 assert(callbacks);
383 assert(max_clients > 0);
384 assert(max_clients < 100);
385 assert(pemkey);
386
387 /* Initialize the network */
388 if (netutils_init() < 0) {
389 return NULL;
390 }
391
392 /* Validate the callbacks structure */
393 if (!callbacks->audio_init ||
394 !callbacks->audio_process ||
395 !callbacks->audio_destroy) {
396 return NULL;
397 }
398
399 /* Allocate the raop_t structure */
400 raop = calloc(1, sizeof(raop_t));
401 if (!raop) {
402 return NULL;
403 }
404
405 /* Initialize the logger */
406 raop->logger = logger_init();
407
408 /* Set HTTP callbacks to our handlers */
409 memset(&httpd_cbs, 0, sizeof(httpd_cbs));
410 httpd_cbs.opaque = raop;
411 httpd_cbs.conn_init = &conn_init;
412 httpd_cbs.conn_request = &conn_request;
413 httpd_cbs.conn_destroy = &conn_destroy;
414
415 /* Initialize the http daemon */
416 httpd = httpd_init(raop->logger, &httpd_cbs, max_clients);
417 if (!httpd) {
418 free(raop);
419 return NULL;
420 }
421
422 /* Copy callbacks structure */
423 memcpy(&raop->callbacks, callbacks, sizeof(raop_callbacks_t));
424
425 /* Initialize RSA key handler */
426 rsakey = rsakey_init_pem(pemkey);
427 if (!rsakey) {
428 free(httpd);
429 free(raop);
430 return NULL;
431 }
432
433 raop->httpd = httpd;
434 raop->rsakey = rsakey;
435
436 return raop;
437 }
438
439 raop_t *
440 raop_init_from_keyfile(int max_clients, raop_callbacks_t *callbacks, const char *keyfile, int *error)
441 {
442 raop_t *raop;
443 char *pemstr;
444
445 if (utils_read_file(&pemstr, keyfile) < 0) {
446 return NULL;
447 }
448 raop = raop_init(max_clients, callbacks, pemstr, error);
449 free(pemstr);
450 return raop;
451 }
452
453 void
454 raop_destroy(raop_t *raop)
455 {
456 if (raop) {
457 raop_stop(raop);
458
459 httpd_destroy(raop->httpd);
460 rsakey_destroy(raop->rsakey);
461 logger_destroy(raop->logger);
462 free(raop);
463
464 /* Cleanup the network */
465 netutils_cleanup();
466 }
467 }
468
469 int
470 raop_is_running(raop_t *raop)
471 {
472 assert(raop);
473
474 return httpd_is_running(raop->httpd);
475 }
476
477 void
478 raop_set_log_level(raop_t *raop, int level)
479 {
480 assert(raop);
481
482 logger_set_level(raop->logger, level);
483 }
484
485 void
486 raop_set_log_callback(raop_t *raop, raop_log_callback_t callback, void *cls)
487 {
488 assert(raop);
489
490 logger_set_callback(raop->logger, callback, cls);
491 }
492
493 int
494 raop_start(raop_t *raop, unsigned short *port, const char *hwaddr, int hwaddrlen, const char *password)
495 {
496 assert(raop);
497 assert(port);
498 assert(hwaddr);
499
500 /* Validate hardware address */
501 if (hwaddrlen > MAX_HWADDR_LEN) {
502 return -1;
503 }
504
505 memset(raop->password, 0, sizeof(raop->password));
506 if (password) {
507 /* Validate password */
508 if (strlen(password) > MAX_PASSWORD_LEN) {
509 return -1;
510 }
511
512 /* Copy password to the raop structure */
513 strncpy(raop->password, password, MAX_PASSWORD_LEN);
514 }
515
516 /* Copy hwaddr to the raop structure */
517 memcpy(raop->hwaddr, hwaddr, hwaddrlen);
518 raop->hwaddrlen = hwaddrlen;
519
520 return httpd_start(raop->httpd, port);
521 }
522
523 void
524 raop_stop(raop_t *raop)
525 {
526 assert(raop);
527
528 httpd_stop(raop->httpd);
529 }
530