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