gst/rtsp-server/rtsp-sdp.c

   1 /* GStreamer
   2  * Copyright (C) 2008 Wim Taymans <wim.taymans at gmail.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 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  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19 /**
  20  * SECTION:rtsp-sdp
  21  * @short_description: Make SDP messages
  22  * @see_also: #GstRTSPMedia
  23  *
  24  * Last reviewed on 2013-07-11 (1.0.0)
  25  */
  26
  27 #include <string.h>
  28
  29 #include <gst/sdp/gstmikey.h>
  30
  31 #include "rtsp-sdp.h"
  32
  33 static gboolean
  34 get_info_from_tags (GstPad * pad, GstEvent ** event, gpointer user_data)
  35 {
  36   GstSDPMedia *media = (GstSDPMedia *) user_data;
  37
  38   if (GST_EVENT_TYPE (*event) == GST_EVENT_TAG) {
  39     GstTagList *tags;
  40     guint bitrate = 0;
  41
  42     gst_event_parse_tag (*event, &tags);
  43
  44     if (gst_tag_list_get_scope (tags) != GST_TAG_SCOPE_STREAM)
  45       return TRUE;
  46
  47     if (!gst_tag_list_get_uint (tags, GST_TAG_MAXIMUM_BITRATE,
  48             &bitrate) || bitrate == 0)
  49       if (!gst_tag_list_get_uint (tags, GST_TAG_BITRATE, &bitrate) ||
  50           bitrate == 0)
  51         return TRUE;
  52
  53     /* set bandwidth (kbits/s) */
  54     gst_sdp_media_add_bandwidth (media, GST_SDP_BWTYPE_AS, bitrate / 1000);
  55
  56     return FALSE;
  57
  58   }
  59
  60   return TRUE;
  61 }
  62
  63 static void
  64 update_sdp_from_tags (GstRTSPStream * stream, GstSDPMedia * stream_media)
  65 {
  66   GstPad *src_pad;
  67
  68   src_pad = gst_rtsp_stream_get_srcpad (stream);
  69
  70   gst_pad_sticky_events_foreach (src_pad, get_info_from_tags, stream_media);
  71
  72   gst_object_unref (src_pad);
  73 }
  74
  75 static void
  76 make_media (GstSDPMessage * sdp, GstSDPInfo * info, GstRTSPMedia * media,
  77     GstRTSPStream * stream, GstStructure * s, GstRTSPProfile profile)
  78 {
  79   GstSDPMedia *smedia;
  80   const gchar *caps_str, *caps_enc, *caps_params;
  81   gchar *tmp;
  82   gint caps_pt, caps_rate;
  83   guint n_fields, j;
  84   gboolean first;
  85   GString *fmtp;
  86   GstRTSPLowerTrans ltrans;
  87   GSocketFamily family;
  88   const gchar *addrtype, *proto;
  89   gchar *address;
  90   guint ttl;
  91
  92   gst_sdp_media_new (&smedia);
  93
  94   /* get media type and payload for the m= line */
  95   caps_str = gst_structure_get_string (s, "media");
  96   gst_sdp_media_set_media (smedia, caps_str);
  97
  98   gst_structure_get_int (s, "payload", &caps_pt);
  99   tmp = g_strdup_printf ("%d", caps_pt);
 100   gst_sdp_media_add_format (smedia, tmp);
 101   g_free (tmp);
 102
 103   gst_sdp_media_set_port_info (smedia, 0, 1);
 104
 105   switch (profile) {
 106     case GST_RTSP_PROFILE_AVP:
 107       proto = "RTP/AVP";
 108       break;
 109     case GST_RTSP_PROFILE_AVPF:
 110       proto = "RTP/AVPF";
 111       break;
 112     case GST_RTSP_PROFILE_SAVP:
 113       proto = "RTP/SAVP";
 114       break;
 115     case GST_RTSP_PROFILE_SAVPF:
 116       proto = "RTP/SAVPF";
 117       break;
 118     default:
 119       proto = "udp";
 120       break;
 121   }
 122   gst_sdp_media_set_proto (smedia, proto);
 123
 124   if (info->is_ipv6) {
 125     addrtype = "IP6";
 126     family = G_SOCKET_FAMILY_IPV6;
 127   } else {
 128     addrtype = "IP4";
 129     family = G_SOCKET_FAMILY_IPV4;
 130   }
 131
 132   ltrans = gst_rtsp_stream_get_protocols (stream);
 133   if (ltrans == GST_RTSP_LOWER_TRANS_UDP_MCAST) {
 134     GstRTSPAddress *addr;
 135
 136     addr = gst_rtsp_stream_get_multicast_address (stream, family);
 137     if (addr == NULL)
 138       goto no_multicast;
 139
 140     address = g_strdup (addr->address);
 141     ttl = addr->ttl;
 142     gst_rtsp_address_free (addr);
 143   } else {
 144     ttl = 16;
 145     if (info->is_ipv6)
 146       address = g_strdup ("::");
 147     else
 148       address = g_strdup ("0.0.0.0");
 149   }
 150
 151   /* for the c= line */
 152   gst_sdp_media_add_connection (smedia, "IN", addrtype, address, ttl, 1);
 153   g_free (address);
 154
 155   /* get clock-rate, media type and params for the rtpmap attribute */
 156   gst_structure_get_int (s, "clock-rate", &caps_rate);
 157   caps_enc = gst_structure_get_string (s, "encoding-name");
 158   caps_params = gst_structure_get_string (s, "encoding-params");
 159
 160   if (caps_enc) {
 161     if (caps_params)
 162       tmp = g_strdup_printf ("%d %s/%d/%s", caps_pt, caps_enc, caps_rate,
 163           caps_params);
 164     else
 165       tmp = g_strdup_printf ("%d %s/%d", caps_pt, caps_enc, caps_rate);
 166
 167     gst_sdp_media_add_attribute (smedia, "rtpmap", tmp);
 168     g_free (tmp);
 169   }
 170
 171   /* the config uri */
 172   tmp = gst_rtsp_stream_get_control (stream);
 173   gst_sdp_media_add_attribute (smedia, "control", tmp);
 174   g_free (tmp);
 175
 176
 177   /* check for srtp */
 178   do {
 179     GstBuffer *srtpkey;
 180     const GValue *val;
 181     const gchar *srtpcipher, *srtpauth, *srtcpcipher, *srtcpauth;
 182     GstMIKEYMessage *msg;
 183     GstMIKEYPayload *payload, *pkd;
 184     GBytes *bytes;
 185     GstMapInfo info;
 186     const guint8 *data;
 187     gsize size;
 188     gchar *base64;
 189     guint8 byte;
 190     guint32 ssrc;
 191
 192     val = gst_structure_get_value (s, "srtp-key");
 193     if (val == NULL)
 194       break;
 195
 196     srtpkey = gst_value_get_buffer (val);
 197     if (srtpkey == NULL)
 198       break;
 199
 200     srtpcipher = gst_structure_get_string (s, "srtp-cipher");
 201     srtpauth = gst_structure_get_string (s, "srtp-auth");
 202     srtcpcipher = gst_structure_get_string (s, "srtcp-cipher");
 203     srtcpauth = gst_structure_get_string (s, "srtcp-auth");
 204
 205     if (srtpcipher == NULL || srtpauth == NULL || srtcpcipher == NULL ||
 206         srtcpauth == NULL)
 207       break;
 208
 209     msg = gst_mikey_message_new ();
 210     /* unencrypted MIKEY message, we send this over TLS so this is allowed */
 211     gst_mikey_message_set_info (msg, GST_MIKEY_VERSION, GST_MIKEY_TYPE_PSK_INIT,
 212         FALSE, GST_MIKEY_PRF_MIKEY_1, 0, GST_MIKEY_MAP_TYPE_SRTP);
 213     /* add policy '0' for our SSRC */
 214     gst_rtsp_stream_get_ssrc (stream, &ssrc);
 215     gst_mikey_message_add_cs_srtp (msg, 0, ssrc, 0);
 216     /* timestamp is now */
 217     gst_mikey_message_add_t_now_ntp_utc (msg);
 218     /* add some random data */
 219     gst_mikey_message_add_rand_len (msg, 16);
 220
 221     /* the policy '0' is SRTP with the above discovered algorithms */
 222     payload = gst_mikey_payload_new (GST_MIKEY_PT_SP);
 223     gst_mikey_payload_sp_set (payload, 0, GST_MIKEY_SEC_PROTO_SRTP);
 224
 225     /* only AES-CM is supported */
 226     byte = 1;
 227     gst_mikey_payload_sp_add_param (payload, GST_MIKEY_SP_SRTP_ENC_ALG, 1,
 228         &byte);
 229     /* only HMAC-SHA1 */
 230     gst_mikey_payload_sp_add_param (payload, GST_MIKEY_SP_SRTP_AUTH_ALG, 1,
 231         &byte);
 232     /* we enable encryption on RTP and RTCP */
 233     gst_mikey_payload_sp_add_param (payload, GST_MIKEY_SP_SRTP_SRTP_ENC, 1,
 234         &byte);
 235     gst_mikey_payload_sp_add_param (payload, GST_MIKEY_SP_SRTP_SRTCP_ENC, 1,
 236         &byte);
 237     /* we enable authentication on RTP and RTCP */
 238     gst_mikey_payload_sp_add_param (payload, GST_MIKEY_SP_SRTP_SRTP_AUTH, 1,
 239         &byte);
 240     gst_mikey_message_add_payload (msg, payload);
 241
 242     /* make unencrypted KEMAC */
 243     payload = gst_mikey_payload_new (GST_MIKEY_PT_KEMAC);
 244     gst_mikey_payload_kemac_set (payload, GST_MIKEY_ENC_NULL,
 245         GST_MIKEY_MAC_NULL);
 246
 247     /* add the key in key data */
 248     pkd = gst_mikey_payload_new (GST_MIKEY_PT_KEY_DATA);
 249     gst_buffer_map (srtpkey, &info, GST_MAP_READ);
 250     gst_mikey_payload_key_data_set_key (pkd, GST_MIKEY_KD_TEK, info.size,
 251         info.data);
 252     gst_buffer_unmap (srtpkey, &info);
 253     /* add key data to KEMAC */
 254     gst_mikey_payload_kemac_add_sub (payload, pkd);
 255     gst_mikey_message_add_payload (msg, payload);
 256
 257     /* now serialize this to bytes */
 258     bytes = gst_mikey_message_to_bytes (msg, NULL, NULL);
 259     gst_mikey_message_free (msg);
 260     /* and make it into base64 */
 261     data = g_bytes_get_data (bytes, &size);
 262     base64 = g_base64_encode (data, size);
 263     g_bytes_unref (bytes);
 264
 265     tmp = g_strdup_printf ("mikey %s", base64);
 266     g_free (base64);
 267
 268     gst_sdp_media_add_attribute (smedia, "key-mgmt", tmp);
 269     g_free (tmp);
 270   } while (FALSE);
 271
 272   /* collect all other properties and add them to fmtp or attributes */
 273   fmtp = g_string_new ("");
 274   g_string_append_printf (fmtp, "%d ", caps_pt);
 275   first = TRUE;
 276   n_fields = gst_structure_n_fields (s);
 277   for (j = 0; j < n_fields; j++) {
 278     const gchar *fname, *fval;
 279
 280     fname = gst_structure_nth_field_name (s, j);
 281
 282     /* filter out standard properties */
 283     if (!strcmp (fname, "media"))
 284       continue;
 285     if (!strcmp (fname, "payload"))
 286       continue;
 287     if (!strcmp (fname, "clock-rate"))
 288       continue;
 289     if (!strcmp (fname, "encoding-name"))
 290       continue;
 291     if (!strcmp (fname, "encoding-params"))
 292       continue;
 293     if (!strcmp (fname, "ssrc"))
 294       continue;
 295     if (!strcmp (fname, "clock-base"))
 296       continue;
 297     if (!strcmp (fname, "seqnum-base"))
 298       continue;
 299     if (g_str_has_prefix (fname, "srtp-"))
 300       continue;
 301     if (g_str_has_prefix (fname, "srtcp-"))
 302       continue;
 303
 304     if (g_str_has_prefix (fname, "a-")) {
 305       /* attribute */
 306       if ((fval = gst_structure_get_string (s, fname)))
 307         gst_sdp_media_add_attribute (smedia, fname + 2, fval);
 308       continue;
 309     }
 310     if (g_str_has_prefix (fname, "x-")) {
 311       /* attribute */
 312       if ((fval = gst_structure_get_string (s, fname)))
 313         gst_sdp_media_add_attribute (smedia, fname, fval);
 314       continue;
 315     }
 316
 317     if ((fval = gst_structure_get_string (s, fname))) {
 318       g_string_append_printf (fmtp, "%s%s=%s", first ? "" : ";", fname, fval);
 319       first = FALSE;
 320     }
 321   }
 322   if (!first) {
 323     tmp = g_string_free (fmtp, FALSE);
 324     gst_sdp_media_add_attribute (smedia, "fmtp", tmp);
 325     g_free (tmp);
 326   } else {
 327     g_string_free (fmtp, TRUE);
 328   }
 329
 330   update_sdp_from_tags (stream, smedia);
 331
 332   gst_sdp_message_add_media (sdp, smedia);
 333   gst_sdp_media_free (smedia);
 334
 335   return;
 336
 337   /* ERRORS */
 338 no_multicast:
 339   {
 340     gst_sdp_media_free (smedia);
 341     g_warning ("ignoring stream %d without multicast address",
 342         gst_rtsp_stream_get_index (stream));
 343     return;
 344   }
 345 }
 346
 347 /**
 348  * gst_rtsp_sdp_from_media:
 349  * @sdp: a #GstSDPMessage
 350  * @info: (transfer none): a #GstSDPInfo
 351  * @media: (transfer none): a #GstRTSPMedia
 352  *
 353  * Add @media specific info to @sdp. @info is used to configure the connection
 354  * information in the SDP.
 355  *
 356  * Returns: TRUE on success.
 357  */
 358 gboolean
 359 gst_rtsp_sdp_from_media (GstSDPMessage * sdp, GstSDPInfo * info,
 360     GstRTSPMedia * media)
 361 {
 362   guint i, n_streams;
 363   gchar *rangestr;
 364
 365   n_streams = gst_rtsp_media_n_streams (media);
 366
 367   rangestr = gst_rtsp_media_get_range_string (media, FALSE, GST_RTSP_RANGE_NPT);
 368   if (rangestr == NULL)
 369     goto not_prepared;
 370
 371   gst_sdp_message_add_attribute (sdp, "range", rangestr);
 372   g_free (rangestr);
 373
 374   for (i = 0; i < n_streams; i++) {
 375     GstRTSPStream *stream;
 376     GstCaps *caps;
 377     GstStructure *s;
 378     GstRTSPProfile profiles;
 379     guint mask;
 380
 381     stream = gst_rtsp_media_get_stream (media, i);
 382     caps = gst_rtsp_stream_get_caps (stream);
 383
 384     if (caps == NULL) {
 385       g_warning ("ignoring stream %d without media type", i);
 386       continue;
 387     }
 388
 389     s = gst_caps_get_structure (caps, 0);
 390     if (s == NULL) {
 391       gst_caps_unref (caps);
 392       g_warning ("ignoring stream %d without media type", i);
 393       continue;
 394     }
 395
 396     /* make a new media for each profile */
 397     profiles = gst_rtsp_stream_get_profiles (stream);
 398     mask = 1;
 399     while (profiles >= mask) {
 400       GstRTSPProfile prof = profiles & mask;
 401
 402       if (prof)
 403         make_media (sdp, info, media, stream, s, prof);
 404
 405       mask <<= 1;
 406     }
 407     gst_caps_unref (caps);
 408   }
 409
 410   {
 411     GstNetTimeProvider *provider;
 412
 413     if ((provider =
 414             gst_rtsp_media_get_time_provider (media, info->server_ip, 0))) {
 415       GstClock *clock;
 416       gchar *address, *str;
 417       gint port;
 418
 419       g_object_get (provider, "clock", &clock, "address", &address, "port",
 420           &port, NULL);
 421
 422       str = g_strdup_printf ("GstNetTimeProvider %s %s:%d %" G_GUINT64_FORMAT,
 423           g_type_name (G_TYPE_FROM_INSTANCE (clock)), address, port,
 424           gst_clock_get_time (clock));
 425
 426       gst_sdp_message_add_attribute (sdp, "x-gst-clock", str);
 427       g_free (str);
 428       gst_object_unref (clock);
 429       g_free (address);
 430       gst_object_unref (provider);
 431     }
 432   }
 433
 434   return TRUE;
 435
 436   /* ERRORS */
 437 not_prepared:
 438   {
 439     GST_ERROR ("media %p is not prepared", media);
 440     return FALSE;
 441   }
 442 }