gst-libs/gst/rtsp/gstrtspconnection.c

   1 /* GStreamer
   2  * Copyright (C) <2005,2006> Wim Taymans <wim@fluendo.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., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19 /*
  20  * Unless otherwise indicated, Source Code is licensed under MIT license.
  21  * See further explanation attached in License Statement (distributed in the file
  22  * LICENSE).
  23  *
  24  * Permission is hereby granted, free of charge, to any person obtaining a copy of
  25  * this software and associated documentation files (the "Software"), to deal in
  26  * the Software without restriction, including without limitation the rights to
  27  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
  28  * of the Software, and to permit persons to whom the Software is furnished to do
  29  * so, subject to the following conditions:
  30  *
  31  * The above copyright notice and this permission notice shall be included in all
  32  * copies or substantial portions of the Software.
  33  *
  34  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  35  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  36  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  37  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  38  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  39  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  40  * SOFTWARE.
  41  */
  42
  43 /**
  44  * SECTION:gstrtspconnection
  45  * @short_description: manage RTSP connections
  46  * @see_also: gstrtspurl
  47  *
  48  * <refsect2>
  49  * <para>
  50  * This object manages the RTSP connection to the server. It provides function
  51  * to receive and send bytes and messages.
  52  * </para>
  53  * </refsect2>
  54  *
  55  * Last reviewed on 2007-07-24 (0.10.14)
  56  */
  57
  58 #ifdef HAVE_CONFIG_H
  59 #  include <config.h>
  60 #endif
  61
  62 #include <stdio.h>
  63 #include <errno.h>
  64 #include <stdlib.h>
  65 #include <string.h>
  66 #include <time.h>
  67
  68 #ifdef HAVE_UNISTD_H
  69 #include <unistd.h>
  70 #endif
  71
  72
  73 /* we include this here to get the G_OS_* defines */
  74 #include <glib.h>
  75 #include <gst/gst.h>
  76
  77 #ifdef G_OS_WIN32
  78 #include <winsock2.h>
  79 #include <ws2tcpip.h>
  80 #define EINPROGRESS WSAEINPROGRESS
  81 #else
  82 #include <sys/ioctl.h>
  83 #include <netdb.h>
  84 #include <sys/socket.h>
  85 #include <netinet/in.h>
  86 #include <arpa/inet.h>
  87 #include <fcntl.h>
  88 #endif
  89
  90 #ifdef HAVE_FIONREAD_IN_SYS_FILIO
  91 #include <sys/filio.h>
  92 #endif
  93
  94 #include "gstrtspconnection.h"
  95 #include "gstrtspbase64.h"
  96 #include "md5.h"
  97
  98 #ifdef G_OS_WIN32
  99 #define READ_SOCKET(fd, buf, len) recv (fd, (char *)buf, len, 0)
 100 #define WRITE_SOCKET(fd, buf, len) send (fd, (const char *)buf, len, 0)
 101 #define SETSOCKOPT(sock, level, name, val, len) setsockopt (sock, level, name, (const char *)val, len)
 102 #define CLOSE_SOCKET(sock) closesocket (sock)
 103 #define ERRNO_IS_EAGAIN (WSAGetLastError () == WSAEWOULDBLOCK)
 104 #define ERRNO_IS_EINTR (WSAGetLastError () == WSAEINTR)
 105 /* According to Microsoft's connect() documentation this one returns
 106  * WSAEWOULDBLOCK and not WSAEINPROGRESS. */
 107 #define ERRNO_IS_EINPROGRESS (WSAGetLastError () == WSAEWOULDBLOCK)
 108 #else
 109 #define READ_SOCKET(fd, buf, len) read (fd, buf, len)
 110 #define WRITE_SOCKET(fd, buf, len) write (fd, buf, len)
 111 #define SETSOCKOPT(sock, level, name, val, len) setsockopt (sock, level, name, val, len)
 112 #define CLOSE_SOCKET(sock) close (sock)
 113 #define ERRNO_IS_EAGAIN (errno == EAGAIN)
 114 #define ERRNO_IS_EINTR (errno == EINTR)
 115 #define ERRNO_IS_EINPROGRESS (errno == EINPROGRESS)
 116 #endif
 117
 118 struct _GstRTSPConnection
 119 {
 120   /*< private > */
 121   /* URL for the connection */
 122   GstRTSPUrl *url;
 123
 124   /* connection state */
 125   GstPollFD fd;
 126   GstPoll *fdset;
 127   gchar *ip;
 128
 129   /* Session state */
 130   gint cseq;                    /* sequence number */
 131   gchar session_id[512];        /* session id */
 132   gint timeout;                 /* session timeout in seconds */
 133   GTimer *timer;                /* timeout timer */
 134
 135   /* Authentication */
 136   GstRTSPAuthMethod auth_method;
 137   gchar *username;
 138   gchar *passwd;
 139   GHashTable *auth_params;
 140 };
 141
 142 #ifdef G_OS_WIN32
 143 static int
 144 inet_aton (const char *c, struct in_addr *paddr)
 145 {
 146   /* note that inet_addr is deprecated on unix because
 147    * inet_addr returns -1 (INADDR_NONE) for the valid 255.255.255.255
 148    * address. */
 149   paddr->s_addr = inet_addr (c);
 150
 151   if (paddr->s_addr == INADDR_NONE)
 152     return 0;
 153
 154   return 1;
 155 }
 156 #endif
 157
 158 enum
 159 {
 160   STATE_START = 0,
 161   STATE_DATA_HEADER,
 162   STATE_DATA_BODY,
 163   STATE_READ_LINES,
 164   STATE_END,
 165   STATE_LAST
 166 };
 167
 168 /* a structure for constructing RTSPMessages */
 169 typedef struct
 170 {
 171   gint state;
 172   guint8 buffer[4096];
 173   guint offset;
 174
 175   guint line;
 176   guint8 *body_data;
 177   glong body_len;
 178 } GstRTSPBuilder;
 179
 180 static void
 181 build_reset (GstRTSPBuilder * builder)
 182 {
 183   g_free (builder->body_data);
 184   memset (builder, 0, sizeof (builder));
 185 }
 186
 187 /**
 188  * gst_rtsp_connection_create:
 189  * @url: a #GstRTSPUrl
 190  * @conn: storage for a #GstRTSPConnection
 191  *
 192  * Create a newly allocated #GstRTSPConnection from @url and store it in @conn.
 193  * The connection will not yet attempt to connect to @url, use
 194  * gst_rtsp_connection_connect().
 195  *
 196  * Returns: #GST_RTSP_OK when @conn contains a valid connection.
 197  */
 198 GstRTSPResult
 199 gst_rtsp_connection_create (GstRTSPUrl * url, GstRTSPConnection ** conn)
 200 {
 201   GstRTSPConnection *newconn;
 202
 203   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
 204
 205   newconn = g_new0 (GstRTSPConnection, 1);
 206
 207   if ((newconn->fdset = gst_poll_new (TRUE)) == NULL)
 208     goto no_fdset;
 209
 210   newconn->url = url;
 211   newconn->fd.fd = -1;
 212   newconn->timer = g_timer_new ();
 213   newconn->timeout = 60;
 214
 215   newconn->auth_method = GST_RTSP_AUTH_NONE;
 216   newconn->username = NULL;
 217   newconn->passwd = NULL;
 218   newconn->auth_params = NULL;
 219
 220   *conn = newconn;
 221
 222   return GST_RTSP_OK;
 223
 224   /* ERRORS */
 225 no_fdset:
 226   {
 227     g_free (newconn);
 228     return GST_RTSP_ESYS;
 229   }
 230 }
 231
 232 /**
 233  * gst_rtsp_connection_accept:
 234  * @sock: a socket
 235  * @conn: storage for a #GstRTSPConnection
 236  *
 237  * Accept a new connection on @sock and create a new #GstRTSPConnection for
 238  * handling communication on new socket.
 239  *
 240  * Returns: #GST_RTSP_OK when @conn contains a valid connection.
 241  *
 242  * Since: 0.10.23
 243  */
 244 GstRTSPResult
 245 gst_rtsp_connection_accept (gint sock, GstRTSPConnection ** conn)
 246 {
 247   int fd;
 248   unsigned int address_len;
 249   GstRTSPConnection *newconn;
 250   struct sockaddr_in address;
 251   GstRTSPUrl *url;
 252
 253   address_len = sizeof (address);
 254   memset (&address, 0, address_len);
 255
 256   fd = accept (sock, (struct sockaddr *) &address, &address_len);
 257   if (fd == -1)
 258     goto accept_failed;
 259
 260   /* set to non-blocking mode so that we can cancel the communication */
 261 #ifndef G_OS_WIN32
 262   fcntl (fd, F_SETFL, O_NONBLOCK);
 263 #else
 264   ioctlsocket (fd, FIONBIO, &flags);
 265 #endif /* G_OS_WIN32 */
 266
 267   /* create a url for the client address */
 268   url = g_new0 (GstRTSPUrl, 1);
 269   url->host = g_strdup_printf ("%s", inet_ntoa (address.sin_addr));
 270   url->port = address.sin_port;
 271
 272   /* now create the connection object */
 273   gst_rtsp_connection_create (url, &newconn);
 274   newconn->fd.fd = fd;
 275   gst_poll_add_fd (newconn->fdset, &newconn->fd);
 276
 277   *conn = newconn;
 278
 279   return GST_RTSP_OK;
 280
 281   /* ERRORS */
 282 accept_failed:
 283   {
 284     return GST_RTSP_ESYS;
 285   }
 286 }
 287
 288 /**
 289  * gst_rtsp_connection_connect:
 290  * @conn: a #GstRTSPConnection
 291  * @timeout: a #GTimeVal timeout
 292  *
 293  * Attempt to connect to the url of @conn made with
 294  * gst_rtsp_connection_create(). If @timeout is #NULL this function can block
 295  * forever. If @timeout contains a valid timeout, this function will return
 296  * #GST_RTSP_ETIMEOUT after the timeout expired.
 297  *
 298  * This function can be cancelled with gst_rtsp_connection_flush().
 299  *
 300  * Returns: #GST_RTSP_OK when a connection could be made.
 301  */
 302 GstRTSPResult
 303 gst_rtsp_connection_connect (GstRTSPConnection * conn, GTimeVal * timeout)
 304 {
 305   gint fd;
 306   struct sockaddr_in sa_in;
 307   struct hostent *hostinfo;
 308   const gchar *ip;
 309   struct in_addr addr;
 310   gint ret;
 311   guint16 port;
 312   GstRTSPUrl *url;
 313   GstClockTime to;
 314   gint retval;
 315
 316 #ifdef G_OS_WIN32
 317   unsigned long flags = 1;
 318   struct in_addr *addrp;
 319 #else
 320   char **addrs;
 321   gchar ipbuf[INET_ADDRSTRLEN];
 322 #endif /* G_OS_WIN32 */
 323
 324   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
 325   g_return_val_if_fail (conn->url != NULL, GST_RTSP_EINVAL);
 326   g_return_val_if_fail (conn->fd.fd < 0, GST_RTSP_EINVAL);
 327
 328   url = conn->url;
 329
 330   /* first check if it already is an IP address */
 331   if (inet_aton (url->host, &addr)) {
 332     ip = url->host;
 333   } else {
 334     hostinfo = gethostbyname (url->host);
 335     if (!hostinfo)
 336       goto not_resolved;        /* h_errno set */
 337
 338     if (hostinfo->h_addrtype != AF_INET)
 339       goto not_ip;              /* host not an IP host */
 340 #ifdef G_OS_WIN32
 341     addrp = (struct in_addr *) hostinfo->h_addr_list[0];
 342     /* this is not threadsafe */
 343     ip = inet_ntoa (*addrp);
 344 #else
 345     addrs = hostinfo->h_addr_list;
 346     ip = inet_ntop (AF_INET, (struct in_addr *) addrs[0], ipbuf,
 347         sizeof (ipbuf));
 348 #endif /* G_OS_WIN32 */
 349   }
 350
 351   /* get the port from the url */
 352   gst_rtsp_url_get_port (url, &port);
 353
 354   memset (&sa_in, 0, sizeof (sa_in));
 355   sa_in.sin_family = AF_INET;   /* network socket */
 356   sa_in.sin_port = htons (port);        /* on port */
 357   sa_in.sin_addr.s_addr = inet_addr (ip);       /* on host ip */
 358
 359   fd = socket (AF_INET, SOCK_STREAM, 0);
 360   if (fd == -1)
 361     goto sys_error;
 362
 363   /* set to non-blocking mode so that we can cancel the connect */
 364 #ifndef G_OS_WIN32
 365   fcntl (fd, F_SETFL, O_NONBLOCK);
 366 #else
 367   ioctlsocket (fd, FIONBIO, &flags);
 368 #endif /* G_OS_WIN32 */
 369
 370   /* add the socket to our fdset */
 371   conn->fd.fd = fd;
 372   gst_poll_add_fd (conn->fdset, &conn->fd);
 373
 374   /* we are going to connect ASYNC now */
 375   ret = connect (fd, (struct sockaddr *) &sa_in, sizeof (sa_in));
 376   if (ret == 0)
 377     goto done;
 378   if (!ERRNO_IS_EINPROGRESS)
 379     goto sys_error;
 380
 381   /* wait for connect to complete up to the specified timeout or until we got
 382    * interrupted. */
 383   gst_poll_fd_ctl_write (conn->fdset, &conn->fd, TRUE);
 384
 385   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
 386
 387   do {
 388     retval = gst_poll_wait (conn->fdset, to);
 389   } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
 390
 391   if (retval == 0)
 392     goto timeout;
 393   else if (retval == -1)
 394     goto sys_error;
 395
 396   /* we can still have an error connecting on windows */
 397   if (gst_poll_fd_has_error (conn->fdset, &conn->fd)) {
 398     socklen_t len = sizeof (errno);
 399     getsockopt (conn->fd.fd, SOL_SOCKET, SO_ERROR, &errno, &len);
 400     goto sys_error;
 401   }
 402
 403   gst_poll_fd_ignored (conn->fdset, &conn->fd);
 404
 405 done:
 406   conn->ip = g_strdup (ip);
 407
 408   return GST_RTSP_OK;
 409
 410 sys_error:
 411   {
 412     GST_ERROR ("system error %d (%s)", errno, g_strerror (errno));
 413     if (conn->fd.fd >= 0) {
 414       GST_DEBUG ("remove fd %d", conn->fd.fd);
 415       gst_poll_remove_fd (conn->fdset, &conn->fd);
 416       conn->fd.fd = -1;
 417     }
 418     if (fd >= 0)
 419       CLOSE_SOCKET (fd);
 420     return GST_RTSP_ESYS;
 421   }
 422 not_resolved:
 423   {
 424     GST_ERROR ("could not resolve %s", url->host);
 425     return GST_RTSP_ENET;
 426   }
 427 not_ip:
 428   {
 429     GST_ERROR ("not an IP address");
 430     return GST_RTSP_ENOTIP;
 431   }
 432 timeout:
 433   {
 434     GST_ERROR ("timeout");
 435     if (conn->fd.fd >= 0) {
 436       GST_DEBUG ("remove fd %d", conn->fd.fd);
 437       gst_poll_remove_fd (conn->fdset, &conn->fd);
 438       conn->fd.fd = -1;
 439     }
 440     if (fd >= 0)
 441       CLOSE_SOCKET (fd);
 442     return GST_RTSP_ETIMEOUT;
 443   }
 444 }
 445
 446 static void
 447 md5_digest_to_hex_string (unsigned char digest[16], char string[33])
 448 {
 449   static const char hexdigits[] = "0123456789abcdef";
 450   int i;
 451
 452   for (i = 0; i < 16; i++) {
 453     string[i * 2] = hexdigits[(digest[i] >> 4) & 0x0f];
 454     string[i * 2 + 1] = hexdigits[digest[i] & 0x0f];
 455   }
 456   string[32] = 0;
 457 }
 458
 459 static void
 460 auth_digest_compute_hex_urp (const gchar * username,
 461     const gchar * realm, const gchar * password, gchar hex_urp[33])
 462 {
 463   struct MD5Context md5_context;
 464   unsigned char digest[16];
 465
 466   MD5Init (&md5_context);
 467   MD5Update (&md5_context, username, strlen (username));
 468   MD5Update (&md5_context, ":", 1);
 469   MD5Update (&md5_context, realm, strlen (realm));
 470   MD5Update (&md5_context, ":", 1);
 471   MD5Update (&md5_context, password, strlen (password));
 472   MD5Final (digest, &md5_context);
 473   md5_digest_to_hex_string (digest, hex_urp);
 474 }
 475
 476 static void
 477 auth_digest_compute_response (const gchar * method,
 478     const gchar * uri, const gchar * hex_a1, const gchar * nonce,
 479     gchar response[33])
 480 {
 481   char hex_a2[33];
 482   struct MD5Context md5_context;
 483   unsigned char digest[16];
 484
 485   /* compute A2 */
 486   MD5Init (&md5_context);
 487   MD5Update (&md5_context, method, strlen (method));
 488   MD5Update (&md5_context, ":", 1);
 489   MD5Update (&md5_context, uri, strlen (uri));
 490   MD5Final (digest, &md5_context);
 491   md5_digest_to_hex_string (digest, hex_a2);
 492
 493   /* compute KD */
 494   MD5Init (&md5_context);
 495   MD5Update (&md5_context, hex_a1, strlen (hex_a1));
 496   MD5Update (&md5_context, ":", 1);
 497   MD5Update (&md5_context, nonce, strlen (nonce));
 498   MD5Update (&md5_context, ":", 1);
 499
 500   MD5Update (&md5_context, hex_a2, 32);
 501   MD5Final (digest, &md5_context);
 502   md5_digest_to_hex_string (digest, response);
 503 }
 504
 505 static void
 506 add_auth_header (GstRTSPConnection * conn, GstRTSPMessage * message)
 507 {
 508   switch (conn->auth_method) {
 509     case GST_RTSP_AUTH_BASIC:{
 510       gchar *user_pass =
 511           g_strdup_printf ("%s:%s", conn->username, conn->passwd);
 512       gchar *user_pass64 =
 513           gst_rtsp_base64_encode (user_pass, strlen (user_pass));
 514       gchar *auth_string = g_strdup_printf ("Basic %s", user_pass64);
 515
 516       gst_rtsp_message_take_header (message, GST_RTSP_HDR_AUTHORIZATION,
 517           auth_string);
 518
 519       g_free (user_pass);
 520       g_free (user_pass64);
 521       break;
 522     }
 523     case GST_RTSP_AUTH_DIGEST:{
 524       gchar response[33], hex_urp[33];
 525       gchar *auth_string, *auth_string2;
 526       gchar *realm;
 527       gchar *nonce;
 528       gchar *opaque;
 529       const gchar *uri;
 530       const gchar *method;
 531
 532       /* we need to have some params set */
 533       if (conn->auth_params == NULL)
 534         break;
 535
 536       /* we need the realm and nonce */
 537       realm = (gchar *) g_hash_table_lookup (conn->auth_params, "realm");
 538       nonce = (gchar *) g_hash_table_lookup (conn->auth_params, "nonce");
 539       if (realm == NULL || nonce == NULL)
 540         break;
 541
 542       auth_digest_compute_hex_urp (conn->username, realm, conn->passwd,
 543           hex_urp);
 544
 545       method = gst_rtsp_method_as_text (message->type_data.request.method);
 546       uri = message->type_data.request.uri;
 547
 548       /* Assume no qop, algorithm=md5, stale=false */
 549       /* For algorithm MD5, a1 = urp. */
 550       auth_digest_compute_response (method, uri, hex_urp, nonce, response);
 551       auth_string = g_strdup_printf ("Digest username=\"%s\", "
 552           "realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
 553           conn->username, realm, nonce, uri, response);
 554
 555       opaque = (gchar *) g_hash_table_lookup (conn->auth_params, "opaque");
 556       if (opaque) {
 557         auth_string2 = g_strdup_printf ("%s, opaque=\"%s\"", auth_string,
 558             opaque);
 559         g_free (auth_string);
 560         auth_string = auth_string2;
 561       }
 562       gst_rtsp_message_take_header (message, GST_RTSP_HDR_AUTHORIZATION,
 563           auth_string);
 564       break;
 565     }
 566     default:
 567       /* Nothing to do */
 568       break;
 569   }
 570 }
 571
 572 static void
 573 add_date_header (GstRTSPMessage * message)
 574 {
 575   GTimeVal tv;
 576   gchar date_string[100];
 577   time_t t;
 578
 579 #ifdef HAVE_GMTIME_R
 580   struct tm tm_;
 581 #endif
 582
 583   g_get_current_time (&tv);
 584   t = (time_t) tv.tv_sec;
 585
 586 #ifdef HAVE_GMTIME_R
 587   strftime (date_string, sizeof (date_string), "%a, %d %b %Y %H:%M:%S GMT",
 588       gmtime_r (&t, &tm_));
 589 #else
 590   strftime (date_string, sizeof (date_string), "%a, %d %b %Y %H:%M:%S GMT",
 591       gmtime (&t));
 592 #endif
 593
 594   gst_rtsp_message_add_header (message, GST_RTSP_HDR_DATE, date_string);
 595 }
 596
 597 static GstRTSPResult
 598 write_bytes (gint fd, const guint8 * buffer, guint * idx, guint size)
 599 {
 600   guint left;
 601
 602   if (*idx > size)
 603     return GST_RTSP_ERROR;
 604
 605   left = size - *idx;
 606
 607   while (left) {
 608     gint r;
 609
 610     r = WRITE_SOCKET (fd, &buffer[*idx], left);
 611     if (r == 0) {
 612       return GST_RTSP_EINTR;
 613     } else if (r < 0) {
 614       if (ERRNO_IS_EAGAIN)
 615         return GST_RTSP_EINTR;
 616       if (!ERRNO_IS_EINTR)
 617         return GST_RTSP_ESYS;
 618     } else {
 619       left -= r;
 620       *idx += r;
 621     }
 622   }
 623   return GST_RTSP_OK;
 624 }
 625
 626 static GstRTSPResult
 627 read_bytes (gint fd, guint8 * buffer, guint * idx, guint size)
 628 {
 629   guint left;
 630
 631   if (*idx > size)
 632     return GST_RTSP_ERROR;
 633
 634   left = size - *idx;
 635
 636   while (left) {
 637     gint r;
 638
 639     r = READ_SOCKET (fd, &buffer[*idx], left);
 640     if (r == 0) {
 641       return GST_RTSP_EEOF;
 642     } else if (r < 0) {
 643       if (ERRNO_IS_EAGAIN)
 644         return GST_RTSP_EINTR;
 645       if (!ERRNO_IS_EINTR)
 646         return GST_RTSP_ESYS;
 647     } else {
 648       left -= r;
 649       *idx += r;
 650     }
 651   }
 652   return GST_RTSP_OK;
 653 }
 654
 655 static GstRTSPResult
 656 read_line (gint fd, guint8 * buffer, guint * idx, guint size)
 657 {
 658   while (TRUE) {
 659     guint8 c;
 660     gint r;
 661
 662     r = READ_SOCKET (fd, &c, 1);
 663     if (r == 0) {
 664       return GST_RTSP_EEOF;
 665     } else if (r < 0) {
 666       if (ERRNO_IS_EAGAIN)
 667         return GST_RTSP_EINTR;
 668       if (!ERRNO_IS_EINTR)
 669         return GST_RTSP_ESYS;
 670     } else {
 671       if (c == '\n')            /* end on \n */
 672         break;
 673       if (c == '\r')            /* ignore \r */
 674         continue;
 675
 676       if (*idx < size - 1)
 677         buffer[(*idx)++] = c;
 678     }
 679   }
 680   buffer[*idx] = '\0';
 681
 682   return GST_RTSP_OK;
 683 }
 684
 685 /**
 686  * gst_rtsp_connection_write:
 687  * @conn: a #GstRTSPConnection
 688  * @data: the data to write
 689  * @size: the size of @data
 690  * @timeout: a timeout value or #NULL
 691  *
 692  * Attempt to write @size bytes of @data to the connected @conn, blocking up to
 693  * the specified @timeout. @timeout can be #NULL, in which case this function
 694  * might block forever.
 695  *
 696  * This function can be cancelled with gst_rtsp_connection_flush().
 697  *
 698  * Returns: #GST_RTSP_OK on success.
 699  */
 700 GstRTSPResult
 701 gst_rtsp_connection_write (GstRTSPConnection * conn, const guint8 * data,
 702     guint size, GTimeVal * timeout)
 703 {
 704   guint offset;
 705   gint retval;
 706   GstClockTime to;
 707   GstRTSPResult res;
 708
 709   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
 710   g_return_val_if_fail (data != NULL || size == 0, GST_RTSP_EINVAL);
 711   g_return_val_if_fail (conn->fd.fd >= 0, GST_RTSP_EINVAL);
 712
 713   gst_poll_set_controllable (conn->fdset, TRUE);
 714   gst_poll_fd_ctl_write (conn->fdset, &conn->fd, TRUE);
 715   gst_poll_fd_ctl_read (conn->fdset, &conn->fd, FALSE);
 716   /* clear all previous poll results */
 717   gst_poll_fd_ignored (conn->fdset, &conn->fd);
 718
 719   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
 720
 721   offset = 0;
 722
 723   while (TRUE) {
 724     /* try to write */
 725     res = write_bytes (conn->fd.fd, data, &offset, size);
 726     if (res == GST_RTSP_OK)
 727       break;
 728     if (res != GST_RTSP_EINTR)
 729       goto write_error;
 730
 731     /* not all is written, wait until we can write more */
 732     do {
 733       retval = gst_poll_wait (conn->fdset, to);
 734     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
 735
 736     if (retval == 0)
 737       goto timeout;
 738
 739     if (retval == -1) {
 740       if (errno == EBUSY)
 741         goto stopped;
 742       else
 743         goto select_error;
 744     }
 745   }
 746   return GST_RTSP_OK;
 747
 748   /* ERRORS */
 749 timeout:
 750   {
 751     return GST_RTSP_ETIMEOUT;
 752   }
 753 select_error:
 754   {
 755     return GST_RTSP_ESYS;
 756   }
 757 stopped:
 758   {
 759     return GST_RTSP_EINTR;
 760   }
 761 write_error:
 762   {
 763     return res;
 764   }
 765 }
 766
 767 static GString *
 768 message_to_string (GstRTSPConnection * conn, GstRTSPMessage * message)
 769 {
 770   GString *str = NULL;
 771
 772   str = g_string_new ("");
 773
 774   switch (message->type) {
 775     case GST_RTSP_MESSAGE_REQUEST:
 776       /* create request string, add CSeq */
 777       g_string_append_printf (str, "%s %s RTSP/1.0\r\n"
 778           "CSeq: %d\r\n",
 779           gst_rtsp_method_as_text (message->type_data.request.method),
 780           message->type_data.request.uri, conn->cseq++);
 781       /* add session id if we have one */
 782       if (conn->session_id[0] != '\0') {
 783         gst_rtsp_message_add_header (message, GST_RTSP_HDR_SESSION,
 784             conn->session_id);
 785       }
 786       /* add any authentication headers */
 787       add_auth_header (conn, message);
 788       break;
 789     case GST_RTSP_MESSAGE_RESPONSE:
 790       /* create response string */
 791       g_string_append_printf (str, "RTSP/1.0 %d %s\r\n",
 792           message->type_data.response.code, message->type_data.response.reason);
 793       break;
 794     case GST_RTSP_MESSAGE_DATA:
 795     {
 796       guint8 data_header[4];
 797
 798       /* prepare data header */
 799       data_header[0] = '$';
 800       data_header[1] = message->type_data.data.channel;
 801       data_header[2] = (message->body_size >> 8) & 0xff;
 802       data_header[3] = message->body_size & 0xff;
 803
 804       /* create string with header and data */
 805       str = g_string_append_len (str, (gchar *) data_header, 4);
 806       str =
 807           g_string_append_len (str, (gchar *) message->body,
 808           message->body_size);
 809       break;
 810     }
 811     default:
 812       g_string_free (str, TRUE);
 813       g_return_val_if_reached (NULL);
 814       break;
 815   }
 816
 817   /* append headers and body */
 818   if (message->type != GST_RTSP_MESSAGE_DATA) {
 819     /* add date header */
 820     add_date_header (message);
 821
 822     /* append headers */
 823     gst_rtsp_message_append_headers (message, str);
 824
 825     /* append Content-Length and body if needed */
 826     if (message->body != NULL && message->body_size > 0) {
 827       gchar *len;
 828
 829       len = g_strdup_printf ("%d", message->body_size);
 830       g_string_append_printf (str, "%s: %s\r\n",
 831           gst_rtsp_header_as_text (GST_RTSP_HDR_CONTENT_LENGTH), len);
 832       g_free (len);
 833       /* header ends here */
 834       g_string_append (str, "\r\n");
 835       str =
 836           g_string_append_len (str, (gchar *) message->body,
 837           message->body_size);
 838     } else {
 839       /* just end headers */
 840       g_string_append (str, "\r\n");
 841     }
 842   }
 843
 844   return str;
 845 }
 846
 847 /**
 848  * gst_rtsp_connection_send:
 849  * @conn: a #GstRTSPConnection
 850  * @message: the message to send
 851  * @timeout: a timeout value or #NULL
 852  *
 853  * Attempt to send @message to the connected @conn, blocking up to
 854  * the specified @timeout. @timeout can be #NULL, in which case this function
 855  * might block forever.
 856  *
 857  * This function can be cancelled with gst_rtsp_connection_flush().
 858  *
 859  * Returns: #GST_RTSP_OK on success.
 860  */
 861 GstRTSPResult
 862 gst_rtsp_connection_send (GstRTSPConnection * conn, GstRTSPMessage * message,
 863     GTimeVal * timeout)
 864 {
 865   GString *str = NULL;
 866   GstRTSPResult res;
 867
 868 #ifdef G_OS_WIN32
 869   WSADATA w;
 870   int error;
 871 #endif
 872
 873   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
 874   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
 875
 876 #ifdef G_OS_WIN32
 877   error = WSAStartup (0x0202, &w);
 878
 879   if (error)
 880     goto startup_error;
 881
 882   if (w.wVersion != 0x0202)
 883     goto version_error;
 884 #endif
 885
 886   if (!(str = message_to_string (conn, message)))
 887     goto no_message;
 888
 889   /* write request */
 890   res =
 891       gst_rtsp_connection_write (conn, (guint8 *) str->str, str->len, timeout);
 892
 893   g_string_free (str, TRUE);
 894
 895   return res;
 896
 897 no_message:
 898   {
 899     g_warning ("Wrong message");
 900     return GST_RTSP_EINVAL;
 901   }
 902 #ifdef G_OS_WIN32
 903 startup_error:
 904   {
 905     g_warning ("Error %d on WSAStartup", error);
 906     return GST_RTSP_EWSASTART;
 907   }
 908 version_error:
 909   {
 910     g_warning ("Windows sockets are not version 0x202 (current 0x%x)",
 911         w.wVersion);
 912     WSACleanup ();
 913     return GST_RTSP_EWSAVERSION;
 914   }
 915 #endif
 916 }
 917
 918 static void
 919 parse_string (gchar * dest, gint size, gchar ** src)
 920 {
 921   gint idx;
 922
 923   idx = 0;
 924   /* skip spaces */
 925   while (g_ascii_isspace (**src))
 926     (*src)++;
 927
 928   while (!g_ascii_isspace (**src) && **src != '\0') {
 929     if (idx < size - 1)
 930       dest[idx++] = **src;
 931     (*src)++;
 932   }
 933   if (size > 0)
 934     dest[idx] = '\0';
 935 }
 936
 937 static void
 938 parse_key (gchar * dest, gint size, gchar ** src)
 939 {
 940   gint idx;
 941
 942   idx = 0;
 943   while (**src != ':' && **src != '\0') {
 944     if (idx < size - 1)
 945       dest[idx++] = **src;
 946     (*src)++;
 947   }
 948   if (size > 0)
 949     dest[idx] = '\0';
 950 }
 951
 952 static GstRTSPResult
 953 parse_response_status (guint8 * buffer, GstRTSPMessage * msg)
 954 {
 955   GstRTSPResult res;
 956   gchar versionstr[20];
 957   gchar codestr[4];
 958   gint code;
 959   gchar *bptr;
 960
 961   bptr = (gchar *) buffer;
 962
 963   parse_string (versionstr, sizeof (versionstr), &bptr);
 964   parse_string (codestr, sizeof (codestr), &bptr);
 965   code = atoi (codestr);
 966
 967   while (g_ascii_isspace (*bptr))
 968     bptr++;
 969
 970   if (strcmp (versionstr, "RTSP/1.0") == 0)
 971     GST_RTSP_CHECK (gst_rtsp_message_init_response (msg, code, bptr, NULL),
 972         parse_error);
 973   else if (strncmp (versionstr, "RTSP/", 5) == 0) {
 974     GST_RTSP_CHECK (gst_rtsp_message_init_response (msg, code, bptr, NULL),
 975         parse_error);
 976     msg->type_data.response.version = GST_RTSP_VERSION_INVALID;
 977   } else
 978     goto parse_error;
 979
 980   return GST_RTSP_OK;
 981
 982 parse_error:
 983   {
 984     return GST_RTSP_EPARSE;
 985   }
 986 }
 987
 988 static GstRTSPResult
 989 parse_request_line (guint8 * buffer, GstRTSPMessage * msg)
 990 {
 991   GstRTSPResult res = GST_RTSP_OK;
 992   gchar versionstr[20];
 993   gchar methodstr[20];
 994   gchar urlstr[4096];
 995   gchar *bptr;
 996   GstRTSPMethod method;
 997
 998   bptr = (gchar *) buffer;
 999
1000   parse_string (methodstr, sizeof (methodstr), &bptr);
1001   method = gst_rtsp_find_method (methodstr);
1002
1003   parse_string (urlstr, sizeof (urlstr), &bptr);
1004   if (*urlstr == '\0')
1005     res = GST_RTSP_EPARSE;
1006
1007   parse_string (versionstr, sizeof (versionstr), &bptr);
1008
1009   if (*bptr != '\0')
1010     res = GST_RTSP_EPARSE;
1011
1012   if (strcmp (versionstr, "RTSP/1.0") == 0) {
1013     if (gst_rtsp_message_init_request (msg, method, urlstr) != GST_RTSP_OK)
1014       res = GST_RTSP_EPARSE;
1015   } else if (strncmp (versionstr, "RTSP/", 5) == 0) {
1016     if (gst_rtsp_message_init_request (msg, method, urlstr) != GST_RTSP_OK)
1017       res = GST_RTSP_EPARSE;
1018     msg->type_data.request.version = GST_RTSP_VERSION_INVALID;
1019   } else {
1020     gst_rtsp_message_init_request (msg, method, urlstr);
1021     msg->type_data.request.version = GST_RTSP_VERSION_INVALID;
1022     res = GST_RTSP_EPARSE;
1023   }
1024
1025   return res;
1026 }
1027
1028 /* parsing lines means reading a Key: Value pair */
1029 static GstRTSPResult
1030 parse_line (guint8 * buffer, GstRTSPMessage * msg)
1031 {
1032   gchar key[32];
1033   gchar *bptr;
1034   GstRTSPHeaderField field;
1035
1036   bptr = (gchar *) buffer;
1037
1038   /* read key */
1039   parse_key (key, sizeof (key), &bptr);
1040   if (*bptr != ':')
1041     goto no_column;
1042
1043   bptr++;
1044
1045   field = gst_rtsp_find_header_field (key);
1046   if (field != GST_RTSP_HDR_INVALID) {
1047     while (g_ascii_isspace (*bptr))
1048       bptr++;
1049     gst_rtsp_message_add_header (msg, field, bptr);
1050   }
1051
1052   return GST_RTSP_OK;
1053
1054 no_column:
1055   {
1056     return GST_RTSP_EPARSE;
1057   }
1058 }
1059
1060 /* returns:
1061  *  GST_RTSP_OK when a complete message was read.
1062  *  GST_RTSP_EEOF: when the socket is closed
1063  *  GST_RTSP_EINTR: when more data is needed.
1064  *  GST_RTSP_..: some other error occured.
1065  */
1066 static GstRTSPResult
1067 build_next (GstRTSPBuilder * builder, GstRTSPMessage * message,
1068     GstRTSPConnection * conn)
1069 {
1070   GstRTSPResult res;
1071
1072   while (TRUE) {
1073     switch (builder->state) {
1074       case STATE_START:
1075         builder->offset = 0;
1076         res =
1077             read_bytes (conn->fd.fd, (guint8 *) builder->buffer,
1078             &builder->offset, 1);
1079         if (res != GST_RTSP_OK)
1080           goto done;
1081
1082         /* we have 1 bytes now and we can see if this is a data message or
1083          * not */
1084         if (builder->buffer[0] == '$') {
1085           /* data message, prepare for the header */
1086           builder->state = STATE_DATA_HEADER;
1087         } else {
1088           builder->line = 0;
1089           builder->state = STATE_READ_LINES;
1090         }
1091         break;
1092       case STATE_DATA_HEADER:
1093       {
1094         res =
1095             read_bytes (conn->fd.fd, (guint8 *) builder->buffer,
1096             &builder->offset, 4);
1097         if (res != GST_RTSP_OK)
1098           goto done;
1099
1100         gst_rtsp_message_init_data (message, builder->buffer[1]);
1101
1102         builder->body_len = (builder->buffer[2] << 8) | builder->buffer[3];
1103         builder->body_data = g_malloc (builder->body_len + 1);
1104         builder->body_data[builder->body_len] = '\0';
1105         builder->offset = 0;
1106         builder->state = STATE_DATA_BODY;
1107         break;
1108       }
1109       case STATE_DATA_BODY:
1110       {
1111         res = read_bytes (conn->fd.fd, builder->body_data, &builder->offset,
1112             builder->body_len);
1113         if (res != GST_RTSP_OK)
1114           goto done;
1115
1116         /* we have the complete body now */
1117         gst_rtsp_message_take_body (message,
1118             (guint8 *) builder->body_data, builder->body_len);
1119         builder->body_data = NULL;
1120         builder->body_len = 0;
1121
1122         builder->state = STATE_END;
1123         break;
1124       }
1125       case STATE_READ_LINES:
1126       {
1127         res = read_line (conn->fd.fd, builder->buffer, &builder->offset,
1128             sizeof (builder->buffer));
1129         if (res != GST_RTSP_OK)
1130           goto done;
1131
1132         /* we have a regular response */
1133         if (builder->buffer[0] == '\r') {
1134           builder->buffer[0] = '\0';
1135         }
1136
1137         if (builder->buffer[0] == '\0') {
1138           gchar *hdrval;
1139
1140           /* empty line, end of message header */
1141           /* see if there is a Content-Length header */
1142           if (gst_rtsp_message_get_header (message,
1143                   GST_RTSP_HDR_CONTENT_LENGTH, &hdrval, 0) == GST_RTSP_OK) {
1144             /* there is, prepare to read the body */
1145             builder->body_len = atol (hdrval);
1146             builder->body_data = g_malloc (builder->body_len + 1);
1147             builder->body_data[builder->body_len] = '\0';
1148             builder->offset = 0;
1149             builder->state = STATE_DATA_BODY;
1150           } else {
1151             builder->state = STATE_END;
1152           }
1153           break;
1154         }
1155
1156         /* we have a line */
1157         if (builder->line == 0) {
1158           /* first line, check for response status */
1159           if (memcmp (builder->buffer, "RTSP", 4) == 0) {
1160             res = parse_response_status (builder->buffer, message);
1161           } else {
1162             res = parse_request_line (builder->buffer, message);
1163           }
1164         } else {
1165           /* else just parse the line */
1166           parse_line (builder->buffer, message);
1167         }
1168         builder->line++;
1169         builder->offset = 0;
1170         break;
1171       }
1172       case STATE_END:
1173       {
1174         gchar *session_id;
1175
1176         /* save session id in the connection for further use */
1177         if (gst_rtsp_message_get_header (message, GST_RTSP_HDR_SESSION,
1178                 &session_id, 0) == GST_RTSP_OK) {
1179           gint maxlen, i;
1180
1181           maxlen = sizeof (conn->session_id) - 1;
1182           /* the sessionid can have attributes marked with ;
1183            * Make sure we strip them */
1184           for (i = 0; session_id[i] != '\0'; i++) {
1185             if (session_id[i] == ';') {
1186               maxlen = i;
1187               /* parse timeout */
1188               do {
1189                 i++;
1190               } while (g_ascii_isspace (session_id[i]));
1191               if (g_str_has_prefix (&session_id[i], "timeout=")) {
1192                 gint to;
1193
1194                 /* if we parsed something valid, configure */
1195                 if ((to = atoi (&session_id[i + 9])) > 0)
1196                   conn->timeout = to;
1197               }
1198               break;
1199             }
1200           }
1201
1202           /* make sure to not overflow */
1203           strncpy (conn->session_id, session_id, maxlen);
1204           conn->session_id[maxlen] = '\0';
1205         }
1206         res = GST_RTSP_OK;
1207         goto done;
1208       }
1209       default:
1210         res = GST_RTSP_ERROR;
1211         break;
1212     }
1213   }
1214 done:
1215   return res;
1216 }
1217
1218 /**
1219  * gst_rtsp_connection_read:
1220  * @conn: a #GstRTSPConnection
1221  * @data: the data to read
1222  * @size: the size of @data
1223  * @timeout: a timeout value or #NULL
1224  *
1225  * Attempt to read @size bytes into @data from the connected @conn, blocking up to
1226  * the specified @timeout. @timeout can be #NULL, in which case this function
1227  * might block forever.
1228  *
1229  * This function can be cancelled with gst_rtsp_connection_flush().
1230  *
1231  * Returns: #GST_RTSP_OK on success.
1232  */
1233 GstRTSPResult
1234 gst_rtsp_connection_read (GstRTSPConnection * conn, guint8 * data, guint size,
1235     GTimeVal * timeout)
1236 {
1237   guint offset;
1238   gint retval;
1239   GstClockTime to;
1240   GstRTSPResult res;
1241
1242   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1243   g_return_val_if_fail (data != NULL, GST_RTSP_EINVAL);
1244   g_return_val_if_fail (conn->fd.fd >= 0, GST_RTSP_EINVAL);
1245
1246   if (size == 0)
1247     return GST_RTSP_OK;
1248
1249   offset = 0;
1250
1251   /* configure timeout if any */
1252   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1253
1254   gst_poll_set_controllable (conn->fdset, TRUE);
1255   gst_poll_fd_ctl_write (conn->fdset, &conn->fd, FALSE);
1256   gst_poll_fd_ctl_read (conn->fdset, &conn->fd, TRUE);
1257
1258   while (TRUE) {
1259     res = read_bytes (conn->fd.fd, data, &offset, size);
1260     if (res == GST_RTSP_EEOF)
1261       goto eof;
1262     if (res == GST_RTSP_OK)
1263       break;
1264     if (res != GST_RTSP_EINTR)
1265       goto read_error;
1266
1267     do {
1268       retval = gst_poll_wait (conn->fdset, to);
1269     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1270
1271     /* check for timeout */
1272     if (retval == 0)
1273       goto select_timeout;
1274
1275     if (retval == -1) {
1276       if (errno == EBUSY)
1277         goto stopped;
1278       else
1279         goto select_error;
1280     }
1281     gst_poll_set_controllable (conn->fdset, FALSE);
1282   }
1283   return GST_RTSP_OK;
1284
1285   /* ERRORS */
1286 select_error:
1287   {
1288     return GST_RTSP_ESYS;
1289   }
1290 select_timeout:
1291   {
1292     return GST_RTSP_ETIMEOUT;
1293   }
1294 stopped:
1295   {
1296     return GST_RTSP_EINTR;
1297   }
1298 eof:
1299   {
1300     return GST_RTSP_EEOF;
1301   }
1302 read_error:
1303   {
1304     return res;
1305   }
1306 }
1307
1308
1309 /**
1310  * gst_rtsp_connection_receive:
1311  * @conn: a #GstRTSPConnection
1312  * @message: the message to read
1313  * @timeout: a timeout value or #NULL
1314  *
1315  * Attempt to read into @message from the connected @conn, blocking up to
1316  * the specified @timeout. @timeout can be #NULL, in which case this function
1317  * might block forever.
1318  *
1319  * This function can be cancelled with gst_rtsp_connection_flush().
1320  *
1321  * Returns: #GST_RTSP_OK on success.
1322  */
1323 GstRTSPResult
1324 gst_rtsp_connection_receive (GstRTSPConnection * conn, GstRTSPMessage * message,
1325     GTimeVal * timeout)
1326 {
1327   GstRTSPResult res;
1328   GstRTSPBuilder builder = { 0 };
1329   gint retval;
1330   GstClockTime to;
1331
1332   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1333   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
1334
1335   /* configure timeout if any */
1336   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1337
1338   gst_poll_set_controllable (conn->fdset, TRUE);
1339   gst_poll_fd_ctl_write (conn->fdset, &conn->fd, FALSE);
1340   gst_poll_fd_ctl_read (conn->fdset, &conn->fd, TRUE);
1341
1342   while (TRUE) {
1343     res = build_next (&builder, message, conn);
1344     if (res == GST_RTSP_EEOF)
1345       goto eof;
1346     if (res == GST_RTSP_OK)
1347       break;
1348     if (res != GST_RTSP_EINTR)
1349       goto read_error;
1350
1351     do {
1352       retval = gst_poll_wait (conn->fdset, to);
1353     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1354
1355     /* check for timeout */
1356     if (retval == 0)
1357       goto select_timeout;
1358
1359     if (retval == -1) {
1360       if (errno == EBUSY)
1361         goto stopped;
1362       else
1363         goto select_error;
1364     }
1365     gst_poll_set_controllable (conn->fdset, FALSE);
1366   }
1367
1368   /* we have a message here */
1369   build_reset (&builder);
1370
1371   return GST_RTSP_OK;
1372
1373   /* ERRORS */
1374 select_error:
1375   {
1376     res = GST_RTSP_ESYS;
1377     goto cleanup;
1378   }
1379 select_timeout:
1380   {
1381     res = GST_RTSP_ETIMEOUT;
1382     goto cleanup;
1383   }
1384 stopped:
1385   {
1386     res = GST_RTSP_EINTR;
1387     goto cleanup;
1388   }
1389 eof:
1390   {
1391     res = GST_RTSP_EEOF;
1392     goto cleanup;
1393   }
1394 read_error:
1395 cleanup:
1396   {
1397     build_reset (&builder);
1398     gst_rtsp_message_unset (message);
1399     return res;
1400   }
1401 }
1402
1403 /**
1404  * gst_rtsp_connection_close:
1405  * @conn: a #GstRTSPConnection
1406  *
1407  * Close the connected @conn.
1408  *
1409  * Returns: #GST_RTSP_OK on success.
1410  */
1411 GstRTSPResult
1412 gst_rtsp_connection_close (GstRTSPConnection * conn)
1413 {
1414   gint res;
1415
1416   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1417
1418   g_free (conn->ip);
1419   conn->ip = NULL;
1420
1421   if (conn->fd.fd != -1) {
1422     gst_poll_remove_fd (conn->fdset, &conn->fd);
1423     res = CLOSE_SOCKET (conn->fd.fd);
1424     conn->fd.fd = -1;
1425 #ifdef G_OS_WIN32
1426     WSACleanup ();
1427 #endif
1428     if (res != 0)
1429       goto sys_error;
1430   }
1431
1432   return GST_RTSP_OK;
1433
1434 sys_error:
1435   {
1436     return GST_RTSP_ESYS;
1437   }
1438 }
1439
1440 /**
1441  * gst_rtsp_connection_free:
1442  * @conn: a #GstRTSPConnection
1443  *
1444  * Close and free @conn.
1445  *
1446  * Returns: #GST_RTSP_OK on success.
1447  */
1448 GstRTSPResult
1449 gst_rtsp_connection_free (GstRTSPConnection * conn)
1450 {
1451   GstRTSPResult res;
1452
1453   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1454
1455   res = gst_rtsp_connection_close (conn);
1456   gst_poll_free (conn->fdset);
1457 #ifdef G_OS_WIN32
1458   WSACleanup ();
1459 #endif
1460   g_timer_destroy (conn->timer);
1461   g_free (conn->username);
1462   g_free (conn->passwd);
1463   gst_rtsp_connection_clear_auth_params (conn);
1464   g_free (conn);
1465
1466   return res;
1467 }
1468
1469 /**
1470  * gst_rtsp_connection_poll:
1471  * @conn: a #GstRTSPConnection
1472  * @events: a bitmask of #GstRTSPEvent flags to check
1473  * @revents: location for result flags
1474  * @timeout: a timeout
1475  *
1476  * Wait up to the specified @timeout for the connection to become available for
1477  * at least one of the operations specified in @events. When the function returns
1478  * with #GST_RTSP_OK, @revents will contain a bitmask of available operations on
1479  * @conn.
1480  *
1481  * @timeout can be #NULL, in which case this function might block forever.
1482  *
1483  * This function can be cancelled with gst_rtsp_connection_flush().
1484  *
1485  * Returns: #GST_RTSP_OK on success.
1486  *
1487  * Since: 0.10.15
1488  */
1489 GstRTSPResult
1490 gst_rtsp_connection_poll (GstRTSPConnection * conn, GstRTSPEvent events,
1491     GstRTSPEvent * revents, GTimeVal * timeout)
1492 {
1493   GstClockTime to;
1494   gint retval;
1495
1496   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1497   g_return_val_if_fail (events != 0, GST_RTSP_EINVAL);
1498   g_return_val_if_fail (revents != NULL, GST_RTSP_EINVAL);
1499   g_return_val_if_fail (conn->fd.fd >= 0, GST_RTSP_EINVAL);
1500
1501   gst_poll_set_controllable (conn->fdset, TRUE);
1502
1503   /* add fd to writer set when asked to */
1504   gst_poll_fd_ctl_write (conn->fdset, &conn->fd, events & GST_RTSP_EV_WRITE);
1505
1506   /* add fd to reader set when asked to */
1507   gst_poll_fd_ctl_read (conn->fdset, &conn->fd, events & GST_RTSP_EV_READ);
1508
1509   /* configure timeout if any */
1510   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1511
1512   do {
1513     retval = gst_poll_wait (conn->fdset, to);
1514   } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1515
1516   if (retval == 0)
1517     goto select_timeout;
1518
1519   if (retval == -1) {
1520     if (errno == EBUSY)
1521       goto stopped;
1522     else
1523       goto select_error;
1524   }
1525
1526   *revents = 0;
1527   if (events & GST_RTSP_EV_READ) {
1528     if (gst_poll_fd_can_read (conn->fdset, &conn->fd))
1529       *revents |= GST_RTSP_EV_READ;
1530   }
1531   if (events & GST_RTSP_EV_WRITE) {
1532     if (gst_poll_fd_can_write (conn->fdset, &conn->fd))
1533       *revents |= GST_RTSP_EV_WRITE;
1534   }
1535   return GST_RTSP_OK;
1536
1537   /* ERRORS */
1538 select_timeout:
1539   {
1540     return GST_RTSP_ETIMEOUT;
1541   }
1542 select_error:
1543   {
1544     return GST_RTSP_ESYS;
1545   }
1546 stopped:
1547   {
1548     return GST_RTSP_EINTR;
1549   }
1550 }
1551
1552 /**
1553  * gst_rtsp_connection_next_timeout:
1554  * @conn: a #GstRTSPConnection
1555  * @timeout: a timeout
1556  *
1557  * Calculate the next timeout for @conn, storing the result in @timeout.
1558  *
1559  * Returns: #GST_RTSP_OK.
1560  */
1561 GstRTSPResult
1562 gst_rtsp_connection_next_timeout (GstRTSPConnection * conn, GTimeVal * timeout)
1563 {
1564   gdouble elapsed;
1565   glong sec;
1566   gulong usec;
1567
1568   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1569   g_return_val_if_fail (timeout != NULL, GST_RTSP_EINVAL);
1570
1571   elapsed = g_timer_elapsed (conn->timer, &usec);
1572   if (elapsed >= conn->timeout) {
1573     sec = 0;
1574     usec = 0;
1575   } else {
1576     sec = conn->timeout - elapsed;
1577   }
1578
1579   timeout->tv_sec = sec;
1580   timeout->tv_usec = usec;
1581
1582   return GST_RTSP_OK;
1583 }
1584
1585 /**
1586  * gst_rtsp_connection_reset_timeout:
1587  * @conn: a #GstRTSPConnection
1588  *
1589  * Reset the timeout of @conn.
1590  *
1591  * Returns: #GST_RTSP_OK.
1592  */
1593 GstRTSPResult
1594 gst_rtsp_connection_reset_timeout (GstRTSPConnection * conn)
1595 {
1596   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1597
1598   g_timer_start (conn->timer);
1599
1600   return GST_RTSP_OK;
1601 }
1602
1603 /**
1604  * gst_rtsp_connection_flush:
1605  * @conn: a #GstRTSPConnection
1606  * @flush: start or stop the flush
1607  *
1608  * Start or stop the flushing action on @conn. When flushing, all current
1609  * and future actions on @conn will return #GST_RTSP_EINTR until the connection
1610  * is set to non-flushing mode again.
1611  *
1612  * Returns: #GST_RTSP_OK.
1613  */
1614 GstRTSPResult
1615 gst_rtsp_connection_flush (GstRTSPConnection * conn, gboolean flush)
1616 {
1617   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1618
1619   gst_poll_set_flushing (conn->fdset, flush);
1620
1621   return GST_RTSP_OK;
1622 }
1623
1624 /**
1625  * gst_rtsp_connection_set_auth:
1626  * @conn: a #GstRTSPConnection
1627  * @method: authentication method
1628  * @user: the user
1629  * @pass: the password
1630  *
1631  * Configure @conn for authentication mode @method with @user and @pass as the
1632  * user and password respectively.
1633  *
1634  * Returns: #GST_RTSP_OK.
1635  */
1636 GstRTSPResult
1637 gst_rtsp_connection_set_auth (GstRTSPConnection * conn,
1638     GstRTSPAuthMethod method, const gchar * user, const gchar * pass)
1639 {
1640   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1641
1642   if (method == GST_RTSP_AUTH_DIGEST && ((user == NULL || pass == NULL)
1643           || g_strrstr (user, ":") != NULL))
1644     return GST_RTSP_EINVAL;
1645
1646   /* Make sure the username and passwd are being set for authentication */
1647   if (method == GST_RTSP_AUTH_NONE && (user == NULL || pass == NULL))
1648     return GST_RTSP_EINVAL;
1649
1650   /* ":" chars are not allowed in usernames for basic auth */
1651   if (method == GST_RTSP_AUTH_BASIC && g_strrstr (user, ":") != NULL)
1652     return GST_RTSP_EINVAL;
1653
1654   g_free (conn->username);
1655   g_free (conn->passwd);
1656
1657   conn->auth_method = method;
1658   conn->username = g_strdup (user);
1659   conn->passwd = g_strdup (pass);
1660
1661   return GST_RTSP_OK;
1662 }
1663
1664 /**
1665  * str_case_hash:
1666  * @key: ASCII string to hash
1667  *
1668  * Hashes @key in a case-insensitive manner.
1669  *
1670  * Returns: the hash code.
1671  **/
1672 static guint
1673 str_case_hash (gconstpointer key)
1674 {
1675   const char *p = key;
1676   guint h = g_ascii_toupper (*p);
1677
1678   if (h)
1679     for (p += 1; *p != '\0'; p++)
1680       h = (h << 5) - h + g_ascii_toupper (*p);
1681
1682   return h;
1683 }
1684
1685 /**
1686  * str_case_equal:
1687  * @v1: an ASCII string
1688  * @v2: another ASCII string
1689  *
1690  * Compares @v1 and @v2 in a case-insensitive manner
1691  *
1692  * Returns: %TRUE if they are equal (modulo case)
1693  **/
1694 static gboolean
1695 str_case_equal (gconstpointer v1, gconstpointer v2)
1696 {
1697   const char *string1 = v1;
1698   const char *string2 = v2;
1699
1700   return g_ascii_strcasecmp (string1, string2) == 0;
1701 }
1702
1703 /**
1704  * gst_rtsp_connection_set_auth_param:
1705  * @conn: a #GstRTSPConnection
1706  * @param: authentication directive
1707  * @value: value
1708  *
1709  * Setup @conn with authentication directives. This is not necesary for
1710  * methods #GST_RTSP_AUTH_NONE and #GST_RTSP_AUTH_BASIC. For
1711  * #GST_RTSP_AUTH_DIGEST, directives should be taken from the digest challenge
1712  * in the WWW-Authenticate response header and can include realm, domain,
1713  * nonce, opaque, stale, algorithm, qop as per RFC2617.
1714  *
1715  * Since: 0.10.20
1716  */
1717 void
1718 gst_rtsp_connection_set_auth_param (GstRTSPConnection * conn,
1719     const gchar * param, const gchar * value)
1720 {
1721   g_return_if_fail (conn != NULL);
1722   g_return_if_fail (param != NULL);
1723
1724   if (conn->auth_params == NULL) {
1725     conn->auth_params =
1726         g_hash_table_new_full (str_case_hash, str_case_equal, g_free, g_free);
1727   }
1728   g_hash_table_insert (conn->auth_params, g_strdup (param), g_strdup (value));
1729 }
1730
1731 /**
1732  * gst_rtsp_connection_clear_auth_params:
1733  * @conn: a #GstRTSPConnection
1734  *
1735  * Clear the list of authentication directives stored in @conn.
1736  *
1737  * Since: 0.10.20
1738  */
1739 void
1740 gst_rtsp_connection_clear_auth_params (GstRTSPConnection * conn)
1741 {
1742   g_return_if_fail (conn != NULL);
1743
1744   if (conn->auth_params != NULL) {
1745     g_hash_table_destroy (conn->auth_params);
1746     conn->auth_params = NULL;
1747   }
1748 }
1749
1750 /**
1751  * gst_rtsp_connection_set_qos_dscp:
1752  * @conn: a #GstRTSPConnection
1753  * @qos_dscp: DSCP value
1754  *
1755  * Configure @conn to use the specified DSCP value.
1756  *
1757  * Returns: #GST_RTSP_OK on success.
1758  *
1759  * Since: 0.10.20
1760  */
1761 GstRTSPResult
1762 gst_rtsp_connection_set_qos_dscp (GstRTSPConnection * conn, guint qos_dscp)
1763 {
1764   struct sockaddr_storage sa_s;
1765   socklen_t sa_sl = sizeof (sa_s);
1766   gint af;
1767   gint tos;
1768
1769   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1770   g_return_val_if_fail (conn->fd.fd >= 0, GST_RTSP_EINVAL);
1771
1772   if (getsockname (conn->fd.fd, (struct sockaddr *) &sa_s, &sa_sl) < 0)
1773     goto no_getsockname;
1774
1775   af = sa_s.ss_family;
1776
1777   /* if this is an IPv4-mapped address then do IPv4 QoS */
1778   if (af == AF_INET6) {
1779     struct sockaddr_in6 *saddr6 = (struct sockaddr_in6 *) &sa_s;
1780
1781     if (IN6_IS_ADDR_V4MAPPED (&saddr6->sin6_addr))
1782       af = AF_INET;
1783   }
1784
1785   /* extract and shift 6 bits of the DSCP */
1786   tos = (qos_dscp & 0x3f) << 2;
1787
1788   switch (af) {
1789     case AF_INET:
1790       if (SETSOCKOPT (conn->fd.fd, IPPROTO_IP, IP_TOS, &tos, sizeof (tos)) < 0)
1791         goto no_setsockopt;
1792       break;
1793     case AF_INET6:
1794 #ifdef IPV6_TCLASS
1795       if (SETSOCKOPT (conn->fd.fd, IPPROTO_IPV6, IPV6_TCLASS, &tos,
1796               sizeof (tos)) < 0)
1797         goto no_setsockopt;
1798       break;
1799 #endif
1800     default:
1801       goto wrong_family;
1802   }
1803
1804   return GST_RTSP_OK;
1805
1806   /* ERRORS */
1807 no_getsockname:
1808 no_setsockopt:
1809   {
1810     return GST_RTSP_ESYS;
1811   }
1812
1813 wrong_family:
1814   {
1815     return GST_RTSP_ERROR;
1816   }
1817 }
1818
1819 /**
1820  * gst_rtsp_connection_get_ip:
1821  * @conn: a #GstRTSPConnection
1822  *
1823  * Retrieve the IP address of the other end of @conn.
1824  *
1825  * Returns: The IP address as a string. this value remains valid until the
1826  * connection is closed.
1827  *
1828  * Since: 0.10.20
1829  */
1830 const gchar *
1831 gst_rtsp_connection_get_ip (const GstRTSPConnection * conn)
1832 {
1833   g_return_val_if_fail (conn != NULL, NULL);
1834
1835   return conn->ip;
1836 }
1837
1838 #define READ_COND   (G_IO_IN | G_IO_HUP | G_IO_ERR)
1839 #define WRITE_COND  (G_IO_OUT | G_IO_ERR)
1840
1841 typedef struct
1842 {
1843   GString *str;
1844   guint cseq;
1845 } GstRTSPRec;
1846
1847 /* async functions */
1848 struct _GstRTSPWatch
1849 {
1850   GSource source;
1851
1852   GstRTSPConnection *conn;
1853
1854   GstRTSPBuilder builder;
1855   GstRTSPMessage message;
1856
1857   GPollFD readfd;
1858   GPollFD writefd;
1859   gboolean write_added;
1860
1861   /* queued message for transmission */
1862   GList *messages;
1863   guint8 *write_data;
1864   guint write_off;
1865   guint write_len;
1866   guint write_cseq;
1867
1868   GstRTSPWatchFuncs funcs;
1869
1870   gpointer user_data;
1871   GDestroyNotify notify;
1872 };
1873
1874 static gboolean
1875 gst_rtsp_source_prepare (GSource * source, gint * timeout)
1876 {
1877   GstRTSPWatch *watch = (GstRTSPWatch *) source;
1878
1879   *timeout = (watch->conn->timeout * 1000);
1880
1881   return FALSE;
1882 }
1883
1884 static gboolean
1885 gst_rtsp_source_check (GSource * source)
1886 {
1887   GstRTSPWatch *watch = (GstRTSPWatch *) source;
1888
1889   if (watch->readfd.revents & READ_COND)
1890     return TRUE;
1891
1892   if (watch->writefd.revents & WRITE_COND)
1893     return TRUE;
1894
1895   return FALSE;
1896 }
1897
1898 static gboolean
1899 gst_rtsp_source_dispatch (GSource * source, GSourceFunc callback,
1900     gpointer user_data)
1901 {
1902   GstRTSPWatch *watch = (GstRTSPWatch *) source;
1903   GstRTSPResult res;
1904
1905   /* first read as much as we can */
1906   if (watch->readfd.revents & READ_COND) {
1907     do {
1908       res = build_next (&watch->builder, &watch->message, watch->conn);
1909       if (res == GST_RTSP_EINTR)
1910         break;
1911       if (res == GST_RTSP_EEOF)
1912         goto eof;
1913       if (res != GST_RTSP_OK)
1914         goto error;
1915
1916       if (watch->funcs.message_received)
1917         watch->funcs.message_received (watch, &watch->message,
1918             watch->user_data);
1919
1920       gst_rtsp_message_unset (&watch->message);
1921       build_reset (&watch->builder);
1922     } while (FALSE);
1923   }
1924
1925   if (watch->writefd.revents & WRITE_COND) {
1926     do {
1927       if (watch->write_data == NULL) {
1928         GstRTSPRec *data;
1929
1930         if (!watch->messages)
1931           goto done;
1932
1933         /* no data, get a new message from the queue */
1934         data = watch->messages->data;
1935         watch->messages = g_list_delete_link (watch->messages, watch->messages);
1936
1937         watch->write_off = 0;
1938         watch->write_len = data->str->len;
1939         watch->write_data = (guint8 *) g_string_free (data->str, FALSE);
1940         watch->write_cseq = data->cseq;
1941
1942         g_slice_free (GstRTSPRec, data);
1943       }
1944
1945       res = write_bytes (watch->writefd.fd, watch->write_data,
1946           &watch->write_off, watch->write_len);
1947       if (res == GST_RTSP_EINTR)
1948         break;
1949       if (res != GST_RTSP_OK)
1950         goto error;
1951
1952       if (watch->funcs.message_sent)
1953         watch->funcs.message_sent (watch, watch->write_cseq, watch->user_data);
1954
1955     done:
1956       if (watch->messages == NULL && watch->write_added) {
1957         g_source_remove_poll ((GSource *) watch, &watch->writefd);
1958         watch->write_added = FALSE;
1959         watch->writefd.revents = 0;
1960       }
1961       g_free (watch->write_data);
1962       watch->write_data = NULL;
1963     } while (FALSE);
1964   }
1965
1966   return TRUE;
1967
1968   /* ERRORS */
1969 eof:
1970   {
1971     if (watch->funcs.closed)
1972       watch->funcs.closed (watch, watch->user_data);
1973     return FALSE;
1974   }
1975 error:
1976   {
1977     if (watch->funcs.error)
1978       watch->funcs.error (watch, res, watch->user_data);
1979     return FALSE;
1980   }
1981 }
1982
1983 static void
1984 gst_rtsp_source_finalize (GSource * source)
1985 {
1986   GstRTSPWatch *watch = (GstRTSPWatch *) source;
1987   GList *walk;
1988
1989   build_reset (&watch->builder);
1990
1991   for (walk = watch->messages; walk; walk = g_list_next (walk)) {
1992     GstRTSPRec *data = walk->data;
1993
1994     g_string_free (data->str, TRUE);
1995     g_slice_free (GstRTSPRec, data);
1996   }
1997   g_list_free (watch->messages);
1998   g_free (watch->write_data);
1999
2000   if (watch->notify)
2001     watch->notify (watch->user_data);
2002 }
2003
2004 static GSourceFuncs gst_rtsp_source_funcs = {
2005   gst_rtsp_source_prepare,
2006   gst_rtsp_source_check,
2007   gst_rtsp_source_dispatch,
2008   gst_rtsp_source_finalize
2009 };
2010
2011 /**
2012  * gst_rtsp_watch_new:
2013  * @conn: a #GstRTSPConnection
2014  * @funcs: watch functions
2015  * @user_data: user data to pass to @funcs
2016  *
2017  * Create a watch object for @conn. The functions provided in @funcs will be
2018  * called with @user_data when activity happened on the watch.
2019  *
2020  * The new watch is usually created so that it can be attached to a
2021  * maincontext with gst_rtsp_watch_attach().
2022  *
2023  * @conn must exist for the entire lifetime of the watch.
2024  *
2025  * Returns: a #GstRTSPWatch that can be used for asynchronous RTSP
2026  * communication. Free with gst_rtsp_watch_unref () after usage.
2027  *
2028  * Since: 0.10.23
2029  */
2030 GstRTSPWatch *
2031 gst_rtsp_watch_new (GstRTSPConnection * conn,
2032     GstRTSPWatchFuncs * funcs, gpointer user_data, GDestroyNotify notify)
2033 {
2034   GstRTSPWatch *result;
2035
2036   g_return_val_if_fail (conn != NULL, NULL);
2037   g_return_val_if_fail (funcs != NULL, NULL);
2038
2039   result = (GstRTSPWatch *) g_source_new (&gst_rtsp_source_funcs,
2040       sizeof (GstRTSPWatch));
2041
2042   result->conn = conn;
2043   result->builder.state = STATE_START;
2044
2045   result->readfd.fd = conn->fd.fd;
2046   result->readfd.events = READ_COND;
2047   result->readfd.revents = 0;
2048
2049   result->writefd.fd = conn->fd.fd;
2050   result->writefd.events = WRITE_COND;
2051   result->writefd.revents = 0;
2052   result->write_added = FALSE;
2053
2054   result->funcs = *funcs;
2055   result->user_data = user_data;
2056   result->notify = notify;
2057
2058   /* only add the read fd, the write fd is only added when we have data
2059    * to send. */
2060   g_source_add_poll ((GSource *) result, &result->readfd);
2061
2062   return result;
2063 }
2064
2065 /**
2066  * gst_rtsp_watch_attach:
2067  * @watch: a #GstRTSPWatch
2068  * @context: a GMainContext (if NULL, the default context will be used)
2069  *
2070  * Adds a #GstRTSPWatch to a context so that it will be executed within that context.
2071  *
2072  * Returns: the ID (greater than 0) for the watch within the GMainContext.
2073  *
2074  * Since: 0.10.23
2075  */
2076 guint
2077 gst_rtsp_watch_attach (GstRTSPWatch * watch, GMainContext * context)
2078 {
2079   g_return_val_if_fail (watch != NULL, 0);
2080
2081   return g_source_attach ((GSource *) watch, context);
2082 }
2083
2084 /**
2085  * gst_rtsp_watch_free:
2086  * @watch: a #GstRTSPWatch
2087  *
2088  * Decreases the reference count of @watch by one. If the resulting reference
2089  * count is zero the watch and associated memory will be destroyed.
2090  *
2091  * Since: 0.10.23
2092  */
2093 void
2094 gst_rtsp_watch_unref (GstRTSPWatch * watch)
2095 {
2096   g_return_if_fail (watch != NULL);
2097
2098   g_source_unref ((GSource *) watch);
2099 }
2100
2101 /**
2102  * gst_rtsp_watch_queue_message:
2103  * @watch: a #GstRTSPWatch
2104  * @message: a #GstRTSPMessage
2105  *
2106  * Queue a @message for transmission in @watch. The contents of this
2107  * message will be serialized and transmitted when the connection of the
2108  * watch becomes writable.
2109  *
2110  * The return value of this function will be returned as the cseq argument in
2111  * the message_sent callback.
2112  *
2113  * Returns: the sequence number of the message or -1 if the cseq could not be
2114  * determined.
2115  *
2116  * Since: 0.10.23
2117  */
2118 guint
2119 gst_rtsp_watch_queue_message (GstRTSPWatch * watch, GstRTSPMessage * message)
2120 {
2121   GstRTSPRec *data;
2122   gchar *header;
2123   guint cseq;
2124
2125   g_return_val_if_fail (watch != NULL, GST_RTSP_EINVAL);
2126   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
2127
2128   /* get the cseq from the message, when we finish writing this message on the
2129    * socket we will have to pass the cseq to the callback. */
2130   if (gst_rtsp_message_get_header (message, GST_RTSP_HDR_CSEQ, &header,
2131           0) == GST_RTSP_OK) {
2132     cseq = atoi (header);
2133   } else {
2134     cseq = -1;
2135   }
2136
2137   /* make a record with the message as a string ans cseq */
2138   data = g_slice_new (GstRTSPRec);
2139   data->str = message_to_string (watch->conn, message);
2140   data->cseq = cseq;
2141
2142   /* add the record to a queue */
2143   watch->messages = g_list_append (watch->messages, data);
2144
2145   /* make sure the main context will now also check for writability on the
2146    * socket */
2147   if (!watch->write_added) {
2148     g_source_add_poll ((GSource *) watch, &watch->writefd);
2149     watch->write_added = TRUE;
2150   }
2151   return cseq;
2152 }