raop.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979
  1. /*
  2. *
  3. * (c) Philippe 2019, philippe_44@outlook.com
  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 3 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,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU 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, see <http://www.gnu.org/licenses/>.
  17. *
  18. */
  19. #include <stdio.h>
  20. #include "platform.h"
  21. #ifdef WIN32
  22. #include <openssl/err.h>
  23. #include <openssl/rand.h>
  24. #include <openssl/rsa.h>
  25. #include <openssl/pem.h>
  26. #include <openssl/engine.h>
  27. #include "mdns.h"
  28. #include "mdnsd.h"
  29. #include "mdnssd-itf.h"
  30. #else
  31. #include "esp_pthread.h"
  32. #include "mdns.h"
  33. #include "mbedtls/version.h"
  34. #include <mbedtls/x509.h>
  35. #endif
  36. #include "util.h"
  37. #include "raop.h"
  38. #include "rtp.h"
  39. #include "dmap_parser.h"
  40. #include "log_util.h"
  41. #define RTSP_STACK_SIZE (8*1024)
  42. #define SEARCH_STACK_SIZE (2*1048)
  43. typedef struct raop_ctx_s {
  44. #ifdef WIN32
  45. struct mdns_service *svc;
  46. struct mdnsd *svr;
  47. #endif
  48. struct in_addr host; // IP of bridge
  49. short unsigned port; // RTSP port for AirPlay
  50. int sock; // socket of the above
  51. struct in_addr peer; // IP of the iDevice (airplay sender)
  52. bool running, abort;
  53. #ifdef WIN32
  54. pthread_t thread, search_thread;
  55. #else
  56. TaskHandle_t thread, search_thread, joiner;
  57. StaticTask_t *xTaskBuffer;
  58. StackType_t xStack[RTSP_STACK_SIZE] __attribute__ ((aligned (4)));
  59. #endif
  60. unsigned char mac[6];
  61. int latency;
  62. struct {
  63. char *aesiv, *aeskey;
  64. char *fmtp;
  65. } rtsp;
  66. struct rtp_s *rtp;
  67. raop_cmd_cb_t cmd_cb;
  68. raop_data_cb_t data_cb;
  69. struct {
  70. char DACPid[32], id[32];
  71. struct in_addr host;
  72. u16_t port;
  73. #ifdef WIN32
  74. struct mDNShandle_s *handle;
  75. #else
  76. bool running;
  77. TaskHandle_t thread, joiner;
  78. StaticTask_t *xTaskBuffer;
  79. StackType_t xStack[SEARCH_STACK_SIZE] __attribute__ ((aligned (4)));;
  80. SemaphoreHandle_t destroy_mutex;
  81. #endif
  82. } active_remote;
  83. void *owner;
  84. } raop_ctx_t;
  85. extern struct mdnsd* glmDNSServer;
  86. extern log_level raop_loglevel;
  87. static log_level *loglevel = &raop_loglevel;
  88. static void* rtsp_thread(void *arg);
  89. static void abort_rtsp(raop_ctx_t *ctx);
  90. static bool handle_rtsp(raop_ctx_t *ctx, int sock);
  91. static char* rsa_apply(unsigned char *input, int inlen, int *outlen, int mode);
  92. static int base64_pad(char *src, char **padded);
  93. static int base64_encode(const void *data, int size, char **str);
  94. static int base64_decode(const char *str, void *data);
  95. static void* search_remote(void *args);
  96. extern char private_key[];
  97. enum { RSA_MODE_KEY, RSA_MODE_AUTH };
  98. static void on_dmap_string(void *ctx, const char *code, const char *name, const char *buf, size_t len);
  99. /*----------------------------------------------------------------------------*/
  100. struct raop_ctx_s *raop_create(struct in_addr host, char *name,
  101. unsigned char mac[6], int latency,
  102. raop_cmd_cb_t cmd_cb, raop_data_cb_t data_cb) {
  103. struct raop_ctx_s *ctx = malloc(sizeof(struct raop_ctx_s));
  104. struct sockaddr_in addr;
  105. char id[64];
  106. #ifdef WIN32
  107. socklen_t nlen = sizeof(struct sockaddr);
  108. char *txt[] = { "am=airesp32", "tp=UDP", "sm=false", "sv=false", "ek=1",
  109. "et=0,1", "md=0,1,2", "cn=0,1", "ch=2",
  110. "ss=16", "sr=44100", "vn=3", "txtvers=1",
  111. NULL };
  112. #else
  113. mdns_txt_item_t txt[] = {
  114. {"am", "airesp32"},
  115. {"tp", "UDP"},
  116. {"sm","false"},
  117. {"sv","false"},
  118. {"ek","1"},
  119. {"et","0,1"},
  120. {"md","0,1,2"},
  121. {"cn","0,1"},
  122. {"ch","2"},
  123. {"ss","16"},
  124. {"sr","44100"},
  125. {"vn","3"},
  126. {"txtvers","1"},
  127. };
  128. #endif
  129. if (!ctx) return NULL;
  130. // make sure we have a clean context
  131. memset(ctx, 0, sizeof(raop_ctx_t));
  132. #ifdef WIN32
  133. ctx->svr = glmDNSServer;
  134. #endif
  135. ctx->host = host;
  136. ctx->sock = socket(AF_INET, SOCK_STREAM, 0);
  137. ctx->cmd_cb = cmd_cb;
  138. ctx->data_cb = data_cb;
  139. ctx->latency = min(latency, 88200);
  140. if (ctx->sock == -1) {
  141. LOG_ERROR("Cannot create listening socket", NULL);
  142. free(ctx);
  143. return NULL;
  144. }
  145. memset(&addr, 0, sizeof(addr));
  146. addr.sin_addr.s_addr = host.s_addr;
  147. addr.sin_family = AF_INET;
  148. #ifdef WIN32
  149. ctx->port = 0;
  150. addr.sin_port = htons(ctx->port);
  151. #else
  152. ctx->port = 5000;
  153. addr.sin_port = htons(ctx->port);
  154. #endif
  155. if (bind(ctx->sock, (struct sockaddr *) &addr, sizeof(addr)) < 0 || listen(ctx->sock, 1)) {
  156. LOG_ERROR("Cannot bind or listen RTSP listener: %s", strerror(errno));
  157. free(ctx);
  158. closesocket(ctx->sock);
  159. return NULL;
  160. }
  161. #ifdef WIN32
  162. getsockname(ctx->sock, (struct sockaddr *) &addr, &nlen);
  163. ctx->port = ntohs(addr.sin_port);
  164. #endif
  165. ctx->running = true;
  166. memcpy(ctx->mac, mac, 6);
  167. snprintf(id, 64, "%02X%02X%02X%02X%02X%02X@%s", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5], name);
  168. #ifdef WIN32
  169. // seems that Windows snprintf does not add NULL char if actual size > max
  170. id[63] = '\0';
  171. ctx->svc = mdnsd_register_svc(ctx->svr, id, "_raop._tcp.local", ctx->port, NULL, (const char**) txt);
  172. pthread_create(&ctx->thread, NULL, &rtsp_thread, ctx);
  173. #else
  174. LOG_INFO("starting mDNS with %s", id);
  175. ESP_ERROR_CHECK( mdns_service_add(id, "_raop", "_tcp", ctx->port, txt, sizeof(txt) / sizeof(mdns_txt_item_t)) );
  176. ctx->xTaskBuffer = (StaticTask_t*) heap_caps_malloc(sizeof(StaticTask_t), MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT);
  177. ctx->thread = xTaskCreateStatic( (TaskFunction_t) rtsp_thread, "RTSP_thread", RTSP_STACK_SIZE, ctx, ESP_TASK_PRIO_MIN + 1, ctx->xStack, ctx->xTaskBuffer);
  178. #endif
  179. return ctx;
  180. }
  181. /*----------------------------------------------------------------------------*/
  182. void raop_abort(struct raop_ctx_s *ctx) {
  183. LOG_INFO("[%p]: aborting RTSP session at next select() wakeup", ctx);
  184. ctx->abort = true;
  185. }
  186. /*----------------------------------------------------------------------------*/
  187. void raop_delete(struct raop_ctx_s *ctx) {
  188. #ifdef WIN32
  189. int sock;
  190. struct sockaddr addr;
  191. socklen_t nlen = sizeof(struct sockaddr);
  192. #endif
  193. if (!ctx) return;
  194. #ifdef WIN32
  195. ctx->running = false;
  196. // wake-up thread by connecting socket, needed for freeBSD
  197. sock = socket(AF_INET, SOCK_STREAM, 0);
  198. getsockname(ctx->sock, (struct sockaddr *) &addr, &nlen);
  199. connect(sock, (struct sockaddr*) &addr, sizeof(addr));
  200. closesocket(sock);
  201. pthread_join(ctx->thread, NULL);
  202. rtp_end(ctx->rtp);
  203. shutdown(ctx->sock, SD_BOTH);
  204. closesocket(ctx->sock);
  205. // terminate search, but do not reclaim memory of pthread if never launched
  206. if (ctx->active_remote.handle) {
  207. close_mDNS(ctx->active_remote.handle);
  208. pthread_join(ctx->search_thread, NULL);
  209. }
  210. // stop broadcasting devices
  211. mdns_service_remove(ctx->svr, ctx->svc);
  212. mdnsd_stop(ctx->svr);
  213. #else
  214. // first stop the search task if any
  215. if (ctx->active_remote.running) {
  216. ctx->active_remote.joiner = xTaskGetCurrentTaskHandle();
  217. ctx->active_remote.running = false;
  218. vTaskResume(ctx->active_remote.thread);
  219. ulTaskNotifyTake(pdFALSE, portMAX_DELAY);
  220. vTaskDelete(ctx->active_remote.thread);
  221. heap_caps_free(ctx->active_remote.xTaskBuffer);
  222. }
  223. // then the RTSP task
  224. ctx->joiner = xTaskGetCurrentTaskHandle();
  225. ctx->running = false;
  226. ulTaskNotifyTake(pdFALSE, portMAX_DELAY);
  227. vTaskDelete(ctx->thread);
  228. heap_caps_free(ctx->xTaskBuffer);
  229. rtp_end(ctx->rtp);
  230. shutdown(ctx->sock, SHUT_RDWR);
  231. closesocket(ctx->sock);
  232. mdns_service_remove("_raop", "_tcp");
  233. #endif
  234. NFREE(ctx->rtsp.aeskey);
  235. NFREE(ctx->rtsp.aesiv);
  236. NFREE(ctx->rtsp.fmtp);
  237. free(ctx);
  238. }
  239. /*----------------------------------------------------------------------------*/
  240. bool raop_cmd(struct raop_ctx_s *ctx, raop_event_t event, void *param) {
  241. struct sockaddr_in addr;
  242. int sock;
  243. char *command = NULL;
  244. // first notify the remote controller (if any)
  245. switch(event) {
  246. case RAOP_REW:
  247. command = strdup("beginrew");
  248. break;
  249. case RAOP_FWD:
  250. command = strdup("beginff");
  251. break;
  252. case RAOP_PREV:
  253. command = strdup("previtem");
  254. break;
  255. case RAOP_NEXT:
  256. command = strdup("nextitem");
  257. break;
  258. case RAOP_TOGGLE:
  259. command = strdup("playpause");
  260. break;
  261. case RAOP_PAUSE:
  262. command = strdup("pause");
  263. break;
  264. case RAOP_PLAY:
  265. command = strdup("play");
  266. break;
  267. case RAOP_RESUME:
  268. command = strdup("playresume");
  269. break;
  270. case RAOP_STOP:
  271. command = strdup("stop");
  272. break;
  273. case RAOP_VOLUME_UP:
  274. command = strdup("volumeup");
  275. break;
  276. case RAOP_VOLUME_DOWN:
  277. command = strdup("volumedown");
  278. break;
  279. case RAOP_VOLUME: {
  280. float Volume = *((float*) param);
  281. Volume = Volume ? (Volume - 1) * 30 : -144;
  282. asprintf(&command,"setproperty?dmcp.device-volume=%0.4lf", Volume);
  283. break;
  284. }
  285. default:
  286. break;
  287. }
  288. // no command to send to remote or no remote found yet
  289. if (!command || !ctx->active_remote.port) {
  290. NFREE(command);
  291. return false;
  292. }
  293. sock = socket(AF_INET, SOCK_STREAM, 0);
  294. memset(&addr, 0, sizeof(addr));
  295. addr.sin_family = AF_INET;
  296. addr.sin_addr.s_addr = S_ADDR(ctx->active_remote.host);
  297. addr.sin_port = htons(ctx->active_remote.port);
  298. if (!connect(sock, (struct sockaddr*) &addr, sizeof(addr))) {
  299. char *method, *buf, resp[512] = "";
  300. int len;
  301. key_data_t headers[4] = { {NULL, NULL} };
  302. asprintf(&method, "GET /ctrl-int/1/%s HTTP/1.0", command);
  303. kd_add(headers, "Active-Remote", ctx->active_remote.id);
  304. kd_add(headers, "Connection", "close");
  305. buf = http_send(sock, method, headers);
  306. len = recv(sock, resp, 512, 0);
  307. if (len > 0) resp[len-1] = '\0';
  308. LOG_INFO("[%p]: sending airplay remote\n%s<== received ==>\n%s", ctx, buf, resp);
  309. NFREE(method);
  310. NFREE(buf);
  311. kd_free(headers);
  312. }
  313. free(command);
  314. closesocket(sock);
  315. return true;
  316. }
  317. /*----------------------------------------------------------------------------*/
  318. static void *rtsp_thread(void *arg) {
  319. raop_ctx_t *ctx = (raop_ctx_t*) arg;
  320. int sock = -1;
  321. while (ctx->running) {
  322. fd_set rfds;
  323. struct timeval timeout = {0, 100*1000};
  324. int n;
  325. bool res = false;
  326. if (sock == -1) {
  327. struct sockaddr_in peer;
  328. socklen_t addrlen = sizeof(struct sockaddr_in);
  329. sock = accept(ctx->sock, (struct sockaddr*) &peer, &addrlen);
  330. ctx->peer.s_addr = peer.sin_addr.s_addr;
  331. ctx->abort = false;
  332. if (sock != -1 && ctx->running) {
  333. LOG_INFO("got RTSP connection %u", sock);
  334. } else continue;
  335. }
  336. FD_ZERO(&rfds);
  337. FD_SET(sock, &rfds);
  338. n = select(sock + 1, &rfds, NULL, NULL, &timeout);
  339. if (!n && !ctx->abort) continue;
  340. if (n > 0) res = handle_rtsp(ctx, sock);
  341. if (n < 0 || !res || ctx->abort) {
  342. abort_rtsp(ctx);
  343. closesocket(sock);
  344. LOG_INFO("RTSP close %u", sock);
  345. sock = -1;
  346. }
  347. }
  348. if (sock != -1) closesocket(sock);
  349. #ifndef WIN32
  350. xTaskNotifyGive(ctx->joiner);
  351. vTaskSuspend(NULL);
  352. #endif
  353. return NULL;
  354. }
  355. /*----------------------------------------------------------------------------*/
  356. static bool handle_rtsp(raop_ctx_t *ctx, int sock)
  357. {
  358. char *buf = NULL, *body = NULL, method[16] = "";
  359. key_data_t headers[16], resp[8] = { {NULL, NULL} };
  360. int len;
  361. bool success = true;
  362. if (!http_parse(sock, method, headers, &body, &len)) {
  363. NFREE(body);
  364. kd_free(headers);
  365. return false;
  366. }
  367. if (strcmp(method, "OPTIONS")) {
  368. LOG_INFO("[%p]: received %s", ctx, method);
  369. }
  370. if ((buf = kd_lookup(headers, "Apple-Challenge")) != NULL) {
  371. int n;
  372. char *buf_pad, *p, *data_b64 = NULL, data[32];
  373. LOG_INFO("[%p]: challenge %s", ctx, buf);
  374. // need to pad the base64 string as apple device don't
  375. base64_pad(buf, &buf_pad);
  376. p = data + min(base64_decode(buf_pad, data), 32-10);
  377. p = (char*) memcpy(p, &S_ADDR(ctx->host), 4) + 4;
  378. p = (char*) memcpy(p, ctx->mac, 6) + 6;
  379. memset(p, 0, 32 - (p - data));
  380. p = rsa_apply((unsigned char*) data, 32, &n, RSA_MODE_AUTH);
  381. n = base64_encode(p, n, &data_b64);
  382. // remove padding as well (seems to be optional now)
  383. for (n = strlen(data_b64) - 1; n > 0 && data_b64[n] == '='; data_b64[n--] = '\0');
  384. kd_add(resp, "Apple-Response", data_b64);
  385. NFREE(p);
  386. NFREE(buf_pad);
  387. NFREE(data_b64);
  388. }
  389. if (!strcmp(method, "OPTIONS")) {
  390. kd_add(resp, "Public", "ANNOUNCE, SETUP, RECORD, PAUSE, FLUSH, TEARDOWN, OPTIONS, GET_PARAMETER, SET_PARAMETER");
  391. } else if (!strcmp(method, "ANNOUNCE")) {
  392. char *padded, *p;
  393. NFREE(ctx->rtsp.aeskey);
  394. NFREE(ctx->rtsp.aesiv);
  395. NFREE(ctx->rtsp.fmtp);
  396. if ((p = strcasestr(body, "rsaaeskey")) != NULL) {
  397. unsigned char *aeskey;
  398. int len, outlen;
  399. p = strextract(p, ":", "\r\n");
  400. base64_pad(p, &padded);
  401. aeskey = malloc(strlen(padded));
  402. len = base64_decode(padded, aeskey);
  403. ctx->rtsp.aeskey = rsa_apply(aeskey, len, &outlen, RSA_MODE_KEY);
  404. NFREE(p);
  405. NFREE(aeskey);
  406. NFREE(padded);
  407. }
  408. if ((p = strcasestr(body, "aesiv")) != NULL) {
  409. p = strextract(p, ":", "\r\n");
  410. base64_pad(p, &padded);
  411. ctx->rtsp.aesiv = malloc(strlen(padded));
  412. base64_decode(padded, ctx->rtsp.aesiv);
  413. NFREE(p);
  414. NFREE(padded);
  415. }
  416. if ((p = strcasestr(body, "fmtp")) != NULL) {
  417. p = strextract(p, ":", "\r\n");
  418. ctx->rtsp.fmtp = strdup(p);
  419. NFREE(p);
  420. }
  421. // on announce, search remote
  422. if ((buf = kd_lookup(headers, "DACP-ID")) != NULL) strcpy(ctx->active_remote.DACPid, buf);
  423. if ((buf = kd_lookup(headers, "Active-Remote")) != NULL) strcpy(ctx->active_remote.id, buf);
  424. #ifdef WIN32
  425. ctx->active_remote.handle = init_mDNS(false, ctx->host);
  426. pthread_create(&ctx->search_thread, NULL, &search_remote, ctx);
  427. #else
  428. ctx->active_remote.running = true;
  429. ctx->active_remote.destroy_mutex = xSemaphoreCreateMutex();
  430. ctx->active_remote.xTaskBuffer = (StaticTask_t*) heap_caps_malloc(sizeof(StaticTask_t), MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT);
  431. ctx->active_remote.thread = xTaskCreateStatic( (TaskFunction_t) search_remote, "search_remote", SEARCH_STACK_SIZE, ctx, ESP_TASK_PRIO_MIN + 1, ctx->active_remote.xStack, ctx->active_remote.xTaskBuffer);
  432. #endif
  433. } else if (!strcmp(method, "SETUP") && ((buf = kd_lookup(headers, "Transport")) != NULL)) {
  434. char *p;
  435. rtp_resp_t rtp = { 0 };
  436. short unsigned tport = 0, cport = 0;
  437. // we are about to stream, do something if needed
  438. success = ctx->cmd_cb(RAOP_SETUP);
  439. if ((p = strcasestr(buf, "timing_port")) != NULL) sscanf(p, "%*[^=]=%hu", &tport);
  440. if ((p = strcasestr(buf, "control_port")) != NULL) sscanf(p, "%*[^=]=%hu", &cport);
  441. rtp = rtp_init(ctx->peer, ctx->latency, ctx->rtsp.aeskey, ctx->rtsp.aesiv,
  442. ctx->rtsp.fmtp, cport, tport, ctx->cmd_cb, ctx->data_cb);
  443. ctx->rtp = rtp.ctx;
  444. if ( (cport * tport * rtp.cport * rtp.tport * rtp.aport) != 0 && rtp.ctx) {
  445. char *transport;
  446. asprintf(&transport, "RTP/AVP/UDP;unicast;mode=record;control_port=%u;timing_port=%u;server_port=%u", rtp.cport, rtp.tport, rtp.aport);
  447. LOG_DEBUG("[%p]: audio=(%hu:%hu), timing=(%hu:%hu), control=(%hu:%hu)", ctx, 0, rtp.aport, tport, rtp.tport, cport, rtp.cport);
  448. kd_add(resp, "Transport", transport);
  449. kd_add(resp, "Session", "DEADBEEF");
  450. free(transport);
  451. } else {
  452. success = false;
  453. LOG_INFO("[%p]: cannot start session, missing ports", ctx);
  454. }
  455. } else if (!strcmp(method, "RECORD")) {
  456. unsigned short seqno = 0;
  457. unsigned rtptime = 0;
  458. char *p;
  459. if (ctx->latency) {
  460. char latency[6];
  461. snprintf(latency, 6, "%u", ctx->latency);
  462. kd_add(resp, "Audio-Latency", latency);
  463. }
  464. buf = kd_lookup(headers, "RTP-Info");
  465. if (buf && (p = strcasestr(buf, "seq")) != NULL) sscanf(p, "%*[^=]=%hu", &seqno);
  466. if (buf && (p = strcasestr(buf, "rtptime")) != NULL) sscanf(p, "%*[^=]=%u", &rtptime);
  467. if (ctx->rtp) rtp_record(ctx->rtp, seqno, rtptime);
  468. success = ctx->cmd_cb(RAOP_STREAM);
  469. } else if (!strcmp(method, "FLUSH")) {
  470. unsigned short seqno = 0;
  471. unsigned rtptime = 0;
  472. char *p;
  473. buf = kd_lookup(headers, "RTP-Info");
  474. if ((p = strcasestr(buf, "seq")) != NULL) sscanf(p, "%*[^=]=%hu", &seqno);
  475. if ((p = strcasestr(buf, "rtptime")) != NULL) sscanf(p, "%*[^=]=%u", &rtptime);
  476. // only send FLUSH if useful (discards frames above buffer head and top)
  477. if (ctx->rtp && rtp_flush(ctx->rtp, seqno, rtptime))
  478. success = ctx->cmd_cb(RAOP_FLUSH);
  479. } else if (!strcmp(method, "TEARDOWN")) {
  480. rtp_end(ctx->rtp);
  481. ctx->rtp = NULL;
  482. // need to make sure no search is on-going and reclaim pthread memory
  483. #ifdef WIN32
  484. if (ctx->active_remote.handle) close_mDNS(ctx->active_remote.handle);
  485. pthread_join(ctx->search_thread, NULL);
  486. #else
  487. ctx->active_remote.joiner = xTaskGetCurrentTaskHandle();
  488. ctx->active_remote.running = false;
  489. xSemaphoreTake(ctx->active_remote.destroy_mutex, portMAX_DELAY);
  490. vTaskDelete(ctx->active_remote.thread);
  491. vSemaphoreDelete(ctx->active_remote.thread);
  492. heap_caps_free(ctx->active_remote.xTaskBuffer);
  493. LOG_INFO("[%p]: mDNS search task terminated", ctx);
  494. #endif
  495. memset(&ctx->active_remote, 0, sizeof(ctx->active_remote));
  496. NFREE(ctx->rtsp.aeskey);
  497. NFREE(ctx->rtsp.aesiv);
  498. NFREE(ctx->rtsp.fmtp);
  499. success = ctx->cmd_cb(RAOP_STOP);
  500. } else if (!strcmp(method, "SET_PARAMETER")) {
  501. char *p;
  502. if (body && (p = strcasestr(body, "volume")) != NULL) {
  503. float volume;
  504. sscanf(p, "%*[^:]:%f", &volume);
  505. LOG_INFO("[%p]: SET PARAMETER volume %f", ctx, volume);
  506. volume = (volume == -144.0) ? 0 : (1 + volume / 30);
  507. success = ctx->cmd_cb(RAOP_VOLUME, volume);
  508. } else if (body && (p = strcasestr(body, "progress")) != NULL) {
  509. int start, current, stop = 0;
  510. // we want ms, not s
  511. sscanf(p, "%*[^:]:%u/%u/%u", &start, &current, &stop);
  512. current = ((current - start) / 44100) * 1000;
  513. if (stop) stop = ((stop - start) / 44100) * 1000;
  514. else stop = -1;
  515. LOG_INFO("[%p]: SET PARAMETER progress %u/%u %s", ctx, current, stop, p);
  516. success = ctx->cmd_cb(RAOP_PROGRESS, current, stop);
  517. }
  518. if (body && ((p = kd_lookup(headers, "Content-Type")) != NULL) && !strcasecmp(p, "application/x-dmap-tagged")) {
  519. struct metadata_s metadata;
  520. dmap_settings settings = {
  521. NULL, NULL, NULL, NULL, NULL, NULL, NULL, on_dmap_string, NULL,
  522. NULL
  523. };
  524. LOG_INFO("[%p]: received metadata");
  525. settings.ctx = &metadata;
  526. memset(&metadata, 0, sizeof(struct metadata_s));
  527. if (!dmap_parse(&settings, body, len)) {
  528. LOG_INFO("[%p]: received metadata\n\tartist: %s\n\talbum: %s\n\ttitle: %s",
  529. ctx, metadata.artist, metadata.album, metadata.title);
  530. success = ctx->cmd_cb(RAOP_METADATA, metadata.artist, metadata.album, metadata.title);
  531. free_metadata(&metadata);
  532. }
  533. }
  534. }
  535. // don't need to free "buf" because kd_lookup return a pointer, not a strdup
  536. kd_add(resp, "Audio-Jack-Status", "connected; type=analog");
  537. kd_add(resp, "CSeq", kd_lookup(headers, "CSeq"));
  538. if (success) {
  539. buf = http_send(sock, "RTSP/1.0 200 OK", resp);
  540. } else {
  541. buf = http_send(sock, "RTSP/1.0 503 ERROR", NULL);
  542. closesocket(sock);
  543. }
  544. if (strcmp(method, "OPTIONS")) {
  545. LOG_INFO("[%p]: responding:\n%s", ctx, buf ? buf : "<void>");
  546. }
  547. NFREE(body);
  548. NFREE(buf);
  549. kd_free(resp);
  550. kd_free(headers);
  551. return true;
  552. }
  553. /*----------------------------------------------------------------------------*/
  554. void abort_rtsp(raop_ctx_t *ctx) {
  555. // first stop RTP process
  556. if (ctx->rtp) {
  557. rtp_end(ctx->rtp);
  558. ctx->rtp = NULL;
  559. LOG_INFO("[%p]: RTP thread aborted", ctx);
  560. }
  561. if (ctx->active_remote.running) {
  562. // need to make sure no search is on-going and reclaim task memory
  563. ctx->active_remote.joiner = xTaskGetCurrentTaskHandle();
  564. ctx->active_remote.running = false;
  565. xSemaphoreTake(ctx->active_remote.destroy_mutex, portMAX_DELAY);
  566. vTaskDelete(ctx->active_remote.thread);
  567. vSemaphoreDelete(ctx->active_remote.thread);
  568. heap_caps_free(ctx->active_remote.xTaskBuffer);
  569. memset(&ctx->active_remote, 0, sizeof(ctx->active_remote));
  570. LOG_INFO("[%p]: Remote search thread aborted", ctx);
  571. }
  572. NFREE(ctx->rtsp.aeskey);
  573. NFREE(ctx->rtsp.aesiv);
  574. NFREE(ctx->rtsp.fmtp);
  575. }
  576. /*----------------------------------------------------------------------------*/
  577. #ifdef WIN32
  578. bool search_remote_cb(mDNSservice_t *slist, void *cookie, bool *stop) {
  579. mDNSservice_t *s;
  580. raop_ctx_t *ctx = (raop_ctx_t*) cookie;
  581. // see if we have found an active remote for our ID
  582. for (s = slist; s; s = s->next) {
  583. if (strcasestr(s->name, ctx->active_remote.DACPid)) {
  584. ctx->active_remote.host = s->addr;
  585. ctx->active_remote.port = s->port;
  586. LOG_INFO("[%p]: found ActiveRemote for %s at %s:%u", ctx, ctx->active_remote.DACPid,
  587. inet_ntoa(ctx->active_remote.host), ctx->active_remote.port);
  588. *stop = true;
  589. break;
  590. }
  591. }
  592. // let caller clear list
  593. return false;
  594. }
  595. /*----------------------------------------------------------------------------*/
  596. static void* search_remote(void *args) {
  597. raop_ctx_t *ctx = (raop_ctx_t*) args;
  598. query_mDNS(ctx->active_remote.handle, "_dacp._tcp.local", 0, 0, &search_remote_cb, (void*) ctx);
  599. return NULL;
  600. }
  601. #else
  602. /*----------------------------------------------------------------------------*/
  603. static void* search_remote(void *args) {
  604. raop_ctx_t *ctx = (raop_ctx_t*) args;
  605. bool found = false;
  606. LOG_INFO("starting remote search");
  607. while (ctx->active_remote.running && !found) {
  608. mdns_result_t *results = NULL;
  609. mdns_result_t *r;
  610. mdns_ip_addr_t *a;
  611. if (mdns_query_ptr("_dacp", "_tcp", 3000, 32, &results)) {
  612. LOG_ERROR("mDNS active remote query Failed");
  613. continue;
  614. }
  615. for (r = results; r && !strcasestr(r->instance_name, ctx->active_remote.DACPid); r = r->next);
  616. if (r) {
  617. for (a = r->addr; a && a->addr.type != IPADDR_TYPE_V4; a = a->next);
  618. if (a) {
  619. found = true;
  620. ctx->active_remote.host.s_addr = a->addr.u_addr.ip4.addr;
  621. ctx->active_remote.port = r->port;
  622. LOG_INFO("found remote %s %s:%hu", r->instance_name, inet_ntoa(ctx->active_remote.host), ctx->active_remote.port);
  623. }
  624. }
  625. mdns_query_results_free(results);
  626. }
  627. // can't use xNotifyGive as it seems LWIP is using it as well
  628. xSemaphoreGive(ctx->active_remote.destroy_mutex);
  629. vTaskSuspend(NULL);
  630. return NULL;
  631. }
  632. #endif
  633. /*----------------------------------------------------------------------------*/
  634. static char *rsa_apply(unsigned char *input, int inlen, int *outlen, int mode)
  635. {
  636. static char super_secret_key[] =
  637. "-----BEGIN RSA PRIVATE KEY-----\n"
  638. "MIIEpQIBAAKCAQEA59dE8qLieItsH1WgjrcFRKj6eUWqi+bGLOX1HL3U3GhC/j0Qg90u3sG/1CUt\n"
  639. "wC5vOYvfDmFI6oSFXi5ELabWJmT2dKHzBJKa3k9ok+8t9ucRqMd6DZHJ2YCCLlDRKSKv6kDqnw4U\n"
  640. "wPdpOMXziC/AMj3Z/lUVX1G7WSHCAWKf1zNS1eLvqr+boEjXuBOitnZ/bDzPHrTOZz0Dew0uowxf\n"
  641. "/+sG+NCK3eQJVxqcaJ/vEHKIVd2M+5qL71yJQ+87X6oV3eaYvt3zWZYD6z5vYTcrtij2VZ9Zmni/\n"
  642. "UAaHqn9JdsBWLUEpVviYnhimNVvYFZeCXg/IdTQ+x4IRdiXNv5hEewIDAQABAoIBAQDl8Axy9XfW\n"
  643. "BLmkzkEiqoSwF0PsmVrPzH9KsnwLGH+QZlvjWd8SWYGN7u1507HvhF5N3drJoVU3O14nDY4TFQAa\n"
  644. "LlJ9VM35AApXaLyY1ERrN7u9ALKd2LUwYhM7Km539O4yUFYikE2nIPscEsA5ltpxOgUGCY7b7ez5\n"
  645. "NtD6nL1ZKauw7aNXmVAvmJTcuPxWmoktF3gDJKK2wxZuNGcJE0uFQEG4Z3BrWP7yoNuSK3dii2jm\n"
  646. "lpPHr0O/KnPQtzI3eguhe0TwUem/eYSdyzMyVx/YpwkzwtYL3sR5k0o9rKQLtvLzfAqdBxBurciz\n"
  647. "aaA/L0HIgAmOit1GJA2saMxTVPNhAoGBAPfgv1oeZxgxmotiCcMXFEQEWflzhWYTsXrhUIuz5jFu\n"
  648. "a39GLS99ZEErhLdrwj8rDDViRVJ5skOp9zFvlYAHs0xh92ji1E7V/ysnKBfsMrPkk5KSKPrnjndM\n"
  649. "oPdevWnVkgJ5jxFuNgxkOLMuG9i53B4yMvDTCRiIPMQ++N2iLDaRAoGBAO9v//mU8eVkQaoANf0Z\n"
  650. "oMjW8CN4xwWA2cSEIHkd9AfFkftuv8oyLDCG3ZAf0vrhrrtkrfa7ef+AUb69DNggq4mHQAYBp7L+\n"
  651. "k5DKzJrKuO0r+R0YbY9pZD1+/g9dVt91d6LQNepUE/yY2PP5CNoFmjedpLHMOPFdVgqDzDFxU8hL\n"
  652. "AoGBANDrr7xAJbqBjHVwIzQ4To9pb4BNeqDndk5Qe7fT3+/H1njGaC0/rXE0Qb7q5ySgnsCb3DvA\n"
  653. "cJyRM9SJ7OKlGt0FMSdJD5KG0XPIpAVNwgpXXH5MDJg09KHeh0kXo+QA6viFBi21y340NonnEfdf\n"
  654. "54PX4ZGS/Xac1UK+pLkBB+zRAoGAf0AY3H3qKS2lMEI4bzEFoHeK3G895pDaK3TFBVmD7fV0Zhov\n"
  655. "17fegFPMwOII8MisYm9ZfT2Z0s5Ro3s5rkt+nvLAdfC/PYPKzTLalpGSwomSNYJcB9HNMlmhkGzc\n"
  656. "1JnLYT4iyUyx6pcZBmCd8bD0iwY/FzcgNDaUmbX9+XDvRA0CgYEAkE7pIPlE71qvfJQgoA9em0gI\n"
  657. "LAuE4Pu13aKiJnfft7hIjbK+5kyb3TysZvoyDnb3HOKvInK7vXbKuU4ISgxB2bB3HcYzQMGsz1qJ\n"
  658. "2gG0N5hvJpzwwhbhXqFKA4zaaSrw622wDniAK5MlIE0tIAKKP4yxNGjoD2QYjhBGuhvkWKY=\n"
  659. "-----END RSA PRIVATE KEY-----";
  660. #ifdef WIN32
  661. unsigned char *out;
  662. RSA *rsa;
  663. BIO *bmem = BIO_new_mem_buf(super_secret_key, -1);
  664. rsa = PEM_read_bio_RSAPrivateKey(bmem, NULL, NULL, NULL);
  665. BIO_free(bmem);
  666. out = malloc(RSA_size(rsa));
  667. switch (mode) {
  668. case RSA_MODE_AUTH:
  669. *outlen = RSA_private_encrypt(inlen, input, out, rsa,
  670. RSA_PKCS1_PADDING);
  671. break;
  672. case RSA_MODE_KEY:
  673. *outlen = RSA_private_decrypt(inlen, input, out, rsa,
  674. RSA_PKCS1_OAEP_PADDING);
  675. break;
  676. }
  677. RSA_free(rsa);
  678. return (char*) out;
  679. #else
  680. mbedtls_pk_context pkctx;
  681. mbedtls_rsa_context *trsa;
  682. size_t olen;
  683. /*
  684. we should do entropy initialization & pass a rng function but this
  685. consumes a ton of stack and there is no security concern here. Anyway,
  686. mbedtls takes a lot of stack, unfortunately ...
  687. */
  688. mbedtls_pk_init(&pkctx);
  689. mbedtls_pk_parse_key(&pkctx, (unsigned char *)super_secret_key,
  690. sizeof(super_secret_key), NULL, 0);
  691. uint8_t *outbuf = NULL;
  692. trsa = mbedtls_pk_rsa(pkctx);
  693. switch (mode) {
  694. case RSA_MODE_AUTH:
  695. mbedtls_rsa_set_padding(trsa, MBEDTLS_RSA_PKCS_V15, MBEDTLS_MD_NONE);
  696. outbuf = malloc(trsa->len);
  697. mbedtls_rsa_pkcs1_encrypt(trsa, NULL, NULL, MBEDTLS_RSA_PRIVATE, inlen, input, outbuf);
  698. *outlen = trsa->len;
  699. break;
  700. case RSA_MODE_KEY:
  701. mbedtls_rsa_set_padding(trsa, MBEDTLS_RSA_PKCS_V21, MBEDTLS_MD_SHA1);
  702. outbuf = malloc(trsa->len);
  703. mbedtls_rsa_pkcs1_decrypt(trsa, NULL, NULL, MBEDTLS_RSA_PRIVATE, &olen, input, outbuf, trsa->len);
  704. *outlen = olen;
  705. break;
  706. }
  707. mbedtls_pk_free(&pkctx);
  708. return (char*) outbuf;
  709. #endif
  710. }
  711. #define DECODE_ERROR 0xffffffff
  712. static char base64_chars[] =
  713. "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
  714. /*----------------------------------------------------------------------------*/
  715. static int base64_pad(char *src, char **padded)
  716. {
  717. int n;
  718. n = strlen(src) + strlen(src) % 4;
  719. *padded = malloc(n + 1);
  720. memset(*padded, '=', n);
  721. memcpy(*padded, src, strlen(src));
  722. (*padded)[n] = '\0';
  723. return strlen(*padded);
  724. }
  725. /*----------------------------------------------------------------------------*/
  726. static int pos(char c)
  727. {
  728. char *p;
  729. for (p = base64_chars; *p; p++)
  730. if (*p == c)
  731. return p - base64_chars;
  732. return -1;
  733. }
  734. /*----------------------------------------------------------------------------*/
  735. static int base64_encode(const void *data, int size, char **str)
  736. {
  737. char *s, *p;
  738. int i;
  739. int c;
  740. const unsigned char *q;
  741. p = s = (char *) malloc(size * 4 / 3 + 4);
  742. if (p == NULL) return -1;
  743. q = (const unsigned char *) data;
  744. i = 0;
  745. for (i = 0; i < size;) {
  746. c = q[i++];
  747. c *= 256;
  748. if (i < size) c += q[i];
  749. i++;
  750. c *= 256;
  751. if (i < size) c += q[i];
  752. i++;
  753. p[0] = base64_chars[(c & 0x00fc0000) >> 18];
  754. p[1] = base64_chars[(c & 0x0003f000) >> 12];
  755. p[2] = base64_chars[(c & 0x00000fc0) >> 6];
  756. p[3] = base64_chars[(c & 0x0000003f) >> 0];
  757. if (i > size) p[3] = '=';
  758. if (i > size + 1) p[2] = '=';
  759. p += 4;
  760. }
  761. *p = 0;
  762. *str = s;
  763. return strlen(s);
  764. }
  765. /*----------------------------------------------------------------------------*/
  766. static unsigned int token_decode(const char *token)
  767. {
  768. int i;
  769. unsigned int val = 0;
  770. int marker = 0;
  771. if (strlen(token) < 4)
  772. return DECODE_ERROR;
  773. for (i = 0; i < 4; i++) {
  774. val *= 64;
  775. if (token[i] == '=')
  776. marker++;
  777. else if (marker > 0)
  778. return DECODE_ERROR;
  779. else
  780. val += pos(token[i]);
  781. }
  782. if (marker > 2)
  783. return DECODE_ERROR;
  784. return (marker << 24) | val;
  785. }
  786. /*----------------------------------------------------------------------------*/
  787. static int base64_decode(const char *str, void *data)
  788. {
  789. const char *p;
  790. unsigned char *q;
  791. q = data;
  792. for (p = str; *p && (*p == '=' || strchr(base64_chars, *p)); p += 4) {
  793. unsigned int val = token_decode(p);
  794. unsigned int marker = (val >> 24) & 0xff;
  795. if (val == DECODE_ERROR)
  796. return -1;
  797. *q++ = (val >> 16) & 0xff;
  798. if (marker < 2)
  799. *q++ = (val >> 8) & 0xff;
  800. if (marker < 1)
  801. *q++ = val & 0xff;
  802. }
  803. return q - (unsigned char *) data;
  804. }
  805. /*----------------------------------------------------------------------------*/
  806. static void on_dmap_string(void *ctx, const char *code, const char *name, const char *buf, size_t len) {
  807. struct metadata_s *metadata = (struct metadata_s *) ctx;
  808. if (!strcasecmp(code, "asar")) metadata->artist = strndup(buf, len);
  809. else if (!strcasecmp(code, "asal")) metadata->album = strndup(buf, len);
  810. else if (!strcasecmp(code, "minm")) metadata->title = strndup(buf, len);
  811. }