src/modules/rtp/sap.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of PulseAudio.
   5
   6   Copyright 2006 Lennart Poettering
   7
   8   PulseAudio is free software; you can redistribute it and/or modify
   9   it under the terms of the GNU Lesser General Public License as published
  10   by the Free Software Foundation; either version 2 of the License,
  11   or (at your option) any later version.
  12
  13   PulseAudio is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with PulseAudio; if not, write to the Free Software
  20   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  21   USA.
  22 ***/
  23
  24 #ifdef HAVE_CONFIG_H
  25 #include <config.h>
  26 #endif
  27
  28 #include <assert.h>
  29 #include <time.h>
  30 #include <stdlib.h>
  31 #include <sys/types.h>
  32 #include <sys/socket.h>
  33 #include <netinet/in.h>
  34 #include <arpa/inet.h>
  35 #include <errno.h>
  36 #include <string.h>
  37 #include <unistd.h>
  38 #include <sys/ioctl.h>
  39
  40 #ifdef HAVE_SYS_FILIO_H
  41 #include <sys/filio.h>
  42 #endif
  43
  44 #include <pulse/xmalloc.h>
  45
  46 #include <pulsecore/core-error.h>
  47 #include <pulsecore/core-util.h>
  48 #include <pulsecore/log.h>
  49
  50 #include "sap.h"
  51 #include "sdp.h"
  52
  53 #define MIME_TYPE "application/sdp"
  54
  55 pa_sap_context* pa_sap_context_init_send(pa_sap_context *c, int fd, char *sdp_data) {
  56     assert(c);
  57     assert(fd >= 0);
  58     assert(sdp_data);
  59
  60     c->fd = fd;
  61     c->sdp_data = sdp_data;
  62     c->msg_id_hash = (uint16_t) (rand()*rand());
  63
  64     return c;
  65 }
  66
  67 void pa_sap_context_destroy(pa_sap_context *c) {
  68     assert(c);
  69
  70     close(c->fd);
  71     pa_xfree(c->sdp_data);
  72 }
  73
  74 int pa_sap_send(pa_sap_context *c, int goodbye) {
  75     uint32_t header;
  76     struct sockaddr_storage sa_buf;
  77     struct sockaddr *sa = (struct sockaddr*) &sa_buf;
  78     socklen_t salen = sizeof(sa_buf);
  79     struct iovec iov[4];
  80     struct msghdr m;
  81     int k;
  82
  83     if (getsockname(c->fd, sa, &salen) < 0) {
  84         pa_log("getsockname() failed: %s\n", pa_cstrerror(errno));
  85         return -1;
  86     }
  87
  88     assert(sa->sa_family == AF_INET || sa->sa_family == AF_INET6);
  89
  90     header = htonl(((uint32_t) 1 << 29) |
  91                    (sa->sa_family == AF_INET6 ? (uint32_t) 1 << 28 : 0) |
  92                    (goodbye ? (uint32_t) 1 << 26 : 0) |
  93                    (c->msg_id_hash));
  94
  95     iov[0].iov_base = &header;
  96     iov[0].iov_len = sizeof(header);
  97
  98     iov[1].iov_base = sa->sa_family == AF_INET ? (void*) &((struct sockaddr_in*) sa)->sin_addr : (void*) &((struct sockaddr_in6*) sa)->sin6_addr;
  99     iov[1].iov_len = sa->sa_family == AF_INET ? 4 : 16;
 100
 101     iov[2].iov_base = (char*) MIME_TYPE;
 102     iov[2].iov_len = sizeof(MIME_TYPE);
 103
 104     iov[3].iov_base = c->sdp_data;
 105     iov[3].iov_len = strlen(c->sdp_data);
 106
 107     m.msg_name = NULL;
 108     m.msg_namelen = 0;
 109     m.msg_iov = iov;
 110     m.msg_iovlen = 4;
 111     m.msg_control = NULL;
 112     m.msg_controllen = 0;
 113     m.msg_flags = 0;
 114
 115     if ((k = sendmsg(c->fd, &m, MSG_DONTWAIT)) < 0)
 116         pa_log("sendmsg() failed: %s\n", pa_cstrerror(errno));
 117
 118     return k;
 119 }
 120
 121 pa_sap_context* pa_sap_context_init_recv(pa_sap_context *c, int fd) {
 122     assert(c);
 123     assert(fd >= 0);
 124
 125     c->fd = fd;
 126     c->sdp_data = NULL;
 127     return c;
 128 }
 129
 130 int pa_sap_recv(pa_sap_context *c, int *goodbye) {
 131     struct msghdr m;
 132     struct iovec iov;
 133     int size, k;
 134     char *buf = NULL, *e;
 135     uint32_t header;
 136     int six, ac;
 137     ssize_t r;
 138
 139     assert(c);
 140     assert(goodbye);
 141
 142     if (ioctl(c->fd, FIONREAD, &size) < 0) {
 143         pa_log("FIONREAD failed: %s", pa_cstrerror(errno));
 144         goto fail;
 145     }
 146
 147     if (!size)
 148         return 0;
 149
 150     buf = pa_xnew(char, size+1);
 151     buf[size] = 0;
 152
 153     iov.iov_base = buf;
 154     iov.iov_len = size;
 155
 156     m.msg_name = NULL;
 157     m.msg_namelen = 0;
 158     m.msg_iov = &iov;
 159     m.msg_iovlen = 1;
 160     m.msg_control = NULL;
 161     m.msg_controllen = 0;
 162     m.msg_flags = 0;
 163
 164     if ((r = recvmsg(c->fd, &m, 0)) != size) {
 165         pa_log("recvmsg() failed: %s", r < 0 ? pa_cstrerror(errno) : "size mismatch");
 166         goto fail;
 167     }
 168
 169     if (size < 4) {
 170         pa_log("SAP packet too short.");
 171         goto fail;
 172     }
 173
 174     memcpy(&header, buf, sizeof(uint32_t));
 175     header = ntohl(header);
 176
 177     if (header >> 29 != 1) {
 178         pa_log("Unsupported SAP version.");
 179         goto fail;
 180     }
 181
 182     if ((header >> 25) & 1) {
 183         pa_log("Encrypted SAP not supported.");
 184         goto fail;
 185     }
 186
 187     if ((header >> 24) & 1) {
 188         pa_log("Compressed SAP not supported.");
 189         goto fail;
 190     }
 191
 192     six = (header >> 28) & 1;
 193     ac = (header >> 16) & 0xFF;
 194
 195     k = 4 + (six ? 16 : 4) + ac*4;
 196     if (size < k) {
 197         pa_log("SAP packet too short (AD).");
 198         goto fail;
 199     }
 200
 201     e = buf + k;
 202     size -= k;
 203
 204     if ((unsigned) size >= sizeof(MIME_TYPE) && !strcmp(e, MIME_TYPE)) {
 205         e += sizeof(MIME_TYPE);
 206         size -= sizeof(MIME_TYPE);
 207     } else if ((unsigned) size < sizeof(PA_SDP_HEADER)-1 || strncmp(e, PA_SDP_HEADER, sizeof(PA_SDP_HEADER)-1)) {
 208         pa_log("Invalid SDP header.");
 209         goto fail;
 210     }
 211
 212     if (c->sdp_data)
 213         pa_xfree(c->sdp_data);
 214
 215     c->sdp_data = pa_xstrndup(e, size);
 216     pa_xfree(buf);
 217
 218     *goodbye = !!((header >> 26) & 1);
 219
 220     return 0;
 221
 222 fail:
 223     pa_xfree(buf);
 224
 225     return -1;
 226 }