RTSP: Add support for server tunneling
[platform/upstream/gstreamer.git] / 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 typedef struct
99 {
100   gint state;
101   guint save;
102   gchar in[4];
103   guint cin;
104   gchar out[3];
105   guint cout;
106 } DecodeCtx;
107
108 static GstRTSPResult read_line (gint fd, guint8 * buffer, guint * idx,
109     guint size, DecodeCtx * ctxp);
110 static GstRTSPResult parse_key_value (guint8 * buffer, gchar * key,
111     guint keysize, gchar ** value);
112 static void parse_string (gchar * dest, gint size, gchar ** src);
113
114 #ifdef G_OS_WIN32
115 #define READ_SOCKET(fd, buf, len) recv (fd, (char *)buf, len, 0)
116 #define WRITE_SOCKET(fd, buf, len) send (fd, (const char *)buf, len, 0)
117 #define SETSOCKOPT(sock, level, name, val, len) setsockopt (sock, level, name, (const char *)val, len)
118 #define CLOSE_SOCKET(sock) closesocket (sock)
119 #define ERRNO_IS_EAGAIN (WSAGetLastError () == WSAEWOULDBLOCK)
120 #define ERRNO_IS_EINTR (WSAGetLastError () == WSAEINTR)
121 /* According to Microsoft's connect() documentation this one returns
122  * WSAEWOULDBLOCK and not WSAEINPROGRESS. */
123 #define ERRNO_IS_EINPROGRESS (WSAGetLastError () == WSAEWOULDBLOCK)
124 #else
125 #define READ_SOCKET(fd, buf, len) read (fd, buf, len)
126 #define WRITE_SOCKET(fd, buf, len) write (fd, buf, len)
127 #define SETSOCKOPT(sock, level, name, val, len) setsockopt (sock, level, name, val, len)
128 #define CLOSE_SOCKET(sock) close (sock)
129 #define ERRNO_IS_EAGAIN (errno == EAGAIN)
130 #define ERRNO_IS_EINTR (errno == EINTR)
131 #define ERRNO_IS_EINPROGRESS (errno == EINPROGRESS)
132 #endif
133
134 #define ADD_POLLFD(fdset, pfd, fd)        \
135 G_STMT_START {                            \
136   (pfd)->fd = fd;                         \
137   gst_poll_add_fd (fdset, pfd);           \
138 } G_STMT_END
139
140 #define REMOVE_POLLFD(fdset, pfd)          \
141 G_STMT_START {                             \
142   if ((pfd)->fd != -1) {                   \
143     GST_DEBUG ("remove fd %d", (pfd)->fd); \
144     gst_poll_remove_fd (fdset, pfd);       \
145     CLOSE_SOCKET ((pfd)->fd);              \
146     (pfd)->fd = -1;                        \
147   }                                        \
148 } G_STMT_END
149
150 typedef enum
151 {
152   TUNNEL_STATE_NONE,
153   TUNNEL_STATE_GET,
154   TUNNEL_STATE_POST,
155   TUNNEL_STATE_COMPLETE
156 } GstRTSPTunnelState;
157
158 #define TUNNELID_LEN   24
159
160 struct _GstRTSPConnection
161 {
162   /*< private > */
163   /* URL for the connection */
164   GstRTSPUrl *url;
165
166   /* connection state */
167   GstPollFD fd0;
168   GstPollFD fd1;
169
170   GstPollFD *readfd;
171   GstPollFD *writefd;
172
173   gchar tunnelid[TUNNELID_LEN];
174   gboolean tunneled;
175   GstRTSPTunnelState tstate;
176
177   GstPoll *fdset;
178   gchar *ip;
179
180   /* Session state */
181   gint cseq;                    /* sequence number */
182   gchar session_id[512];        /* session id */
183   gint timeout;                 /* session timeout in seconds */
184   GTimer *timer;                /* timeout timer */
185
186   /* Authentication */
187   GstRTSPAuthMethod auth_method;
188   gchar *username;
189   gchar *passwd;
190   GHashTable *auth_params;
191
192   DecodeCtx ctx;
193   DecodeCtx *ctxp;
194 };
195
196 #ifdef G_OS_WIN32
197 static int
198 inet_aton (const char *c, struct in_addr *paddr)
199 {
200   /* note that inet_addr is deprecated on unix because
201    * inet_addr returns -1 (INADDR_NONE) for the valid 255.255.255.255
202    * address. */
203   paddr->s_addr = inet_addr (c);
204
205   if (paddr->s_addr == INADDR_NONE)
206     return 0;
207
208   return 1;
209 }
210 #endif
211
212 enum
213 {
214   STATE_START = 0,
215   STATE_DATA_HEADER,
216   STATE_DATA_BODY,
217   STATE_READ_LINES,
218   STATE_END,
219   STATE_LAST
220 };
221
222 /* a structure for constructing RTSPMessages */
223 typedef struct
224 {
225   gint state;
226   guint8 buffer[4096];
227   guint offset;
228
229   guint line;
230   guint8 *body_data;
231   glong body_len;
232 } GstRTSPBuilder;
233
234 static void
235 build_reset (GstRTSPBuilder * builder)
236 {
237   g_free (builder->body_data);
238   memset (builder, 0, sizeof (builder));
239 }
240
241 /**
242  * gst_rtsp_connection_create:
243  * @url: a #GstRTSPUrl 
244  * @conn: storage for a #GstRTSPConnection
245  *
246  * Create a newly allocated #GstRTSPConnection from @url and store it in @conn.
247  * The connection will not yet attempt to connect to @url, use
248  * gst_rtsp_connection_connect().
249  *
250  * Returns: #GST_RTSP_OK when @conn contains a valid connection.
251  */
252 GstRTSPResult
253 gst_rtsp_connection_create (GstRTSPUrl * url, GstRTSPConnection ** conn)
254 {
255   GstRTSPConnection *newconn;
256 #ifdef G_OS_WIN32
257   WSADATA w;
258   int error;
259 #endif
260
261   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
262
263 #ifdef G_OS_WIN32
264   error = WSAStartup (0x0202, &w);
265
266   if (error)
267     goto startup_error;
268
269   if (w.wVersion != 0x0202)
270     goto version_error;
271 #endif
272
273   newconn = g_new0 (GstRTSPConnection, 1);
274
275   if ((newconn->fdset = gst_poll_new (TRUE)) == NULL)
276     goto no_fdset;
277
278   newconn->url = url;
279   newconn->fd0.fd = -1;
280   newconn->fd1.fd = -1;
281   newconn->timer = g_timer_new ();
282   newconn->timeout = 60;
283   newconn->tunneled = FALSE;
284
285   newconn->auth_method = GST_RTSP_AUTH_NONE;
286   newconn->username = NULL;
287   newconn->passwd = NULL;
288   newconn->auth_params = NULL;
289
290   *conn = newconn;
291
292   return GST_RTSP_OK;
293
294   /* ERRORS */
295 #ifdef G_OS_WIN32
296 startup_error:
297   {
298     g_warning ("Error %d on WSAStartup", error);
299     return GST_RTSP_EWSASTART;
300   }
301 version_error:
302   {
303     g_warning ("Windows sockets are not version 0x202 (current 0x%x)",
304         w.wVersion);
305     WSACleanup ();
306     return GST_RTSP_EWSAVERSION;
307   }
308 #endif
309 no_fdset:
310   {
311     g_free (newconn);
312 #ifdef G_OS_WIN32
313     WSACleanup ();
314 #endif
315     return GST_RTSP_ESYS;
316   }
317 }
318
319 /**
320  * gst_rtsp_connection_accept:
321  * @sock: a socket
322  * @conn: storage for a #GstRTSPConnection
323  *
324  * Accept a new connection on @sock and create a new #GstRTSPConnection for
325  * handling communication on new socket.
326  *
327  * Returns: #GST_RTSP_OK when @conn contains a valid connection.
328  *
329  * Since: 0.10.23
330  */
331 GstRTSPResult
332 gst_rtsp_connection_accept (gint sock, GstRTSPConnection ** conn)
333 {
334   int fd;
335   unsigned int address_len;
336   GstRTSPConnection *newconn = NULL;
337   struct sockaddr_in address;
338   GstRTSPUrl *url;
339 #ifdef G_OS_WIN32
340   gulong flags = 1;
341 #endif
342
343   address_len = sizeof (address);
344   memset (&address, 0, address_len);
345
346 #ifndef G_OS_WIN32
347   fd = accept (sock, (struct sockaddr *) &address, &address_len);
348 #else
349   fd = accept (sock, (struct sockaddr *) &address, (gint *) & address_len);
350 #endif /* G_OS_WIN32 */
351   if (fd == -1)
352     goto accept_failed;
353
354   /* set to non-blocking mode so that we can cancel the communication */
355 #ifndef G_OS_WIN32
356   fcntl (fd, F_SETFL, O_NONBLOCK);
357 #else
358   ioctlsocket (fd, FIONBIO, &flags);
359 #endif /* G_OS_WIN32 */
360
361   /* create a url for the client address */
362   url = g_new0 (GstRTSPUrl, 1);
363   url->host = g_strdup_printf ("%s", inet_ntoa (address.sin_addr));
364   url->port = address.sin_port;
365
366   /* now create the connection object */
367   gst_rtsp_connection_create (url, &newconn);
368   ADD_POLLFD (newconn->fdset, &newconn->fd0, fd);
369
370   /* both read and write initially */
371   newconn->readfd = &newconn->fd0;
372   newconn->writefd = &newconn->fd0;
373
374   *conn = newconn;
375
376   return GST_RTSP_OK;
377
378   /* ERRORS */
379 accept_failed:
380   {
381     return GST_RTSP_ESYS;
382   }
383 }
384
385 static const gchar *
386 do_resolve (const gchar * host)
387 {
388   struct hostent *hostinfo;
389   struct in_addr addr;
390   const gchar *ip;
391 #ifdef G_OS_WIN32
392   struct in_addr *addrp;
393 #else
394   char **addrs;
395   gchar ipbuf[INET_ADDRSTRLEN];
396 #endif /* G_OS_WIN32 */
397
398   ip = NULL;
399
400   /* first check if it already is an IP address */
401   if (inet_aton (host, &addr)) {
402     ip = host;
403   } else {
404     hostinfo = gethostbyname (host);
405     if (!hostinfo)
406       goto not_resolved;        /* h_errno set */
407
408     if (hostinfo->h_addrtype != AF_INET)
409       goto not_ip;              /* host not an IP host */
410 #ifdef G_OS_WIN32
411     addrp = (struct in_addr *) hostinfo->h_addr_list[0];
412     /* this is not threadsafe */
413     ip = inet_ntoa (*addrp);
414 #else
415     addrs = hostinfo->h_addr_list;
416     ip = inet_ntop (AF_INET, (struct in_addr *) addrs[0], ipbuf,
417         sizeof (ipbuf));
418 #endif /* G_OS_WIN32 */
419   }
420   return ip;
421
422   /* ERRORS */
423 not_resolved:
424   {
425     GST_ERROR ("could not resolve %s", host);
426     return NULL;
427   }
428 not_ip:
429   {
430     GST_ERROR ("not an IP address");
431     return NULL;
432   }
433 }
434
435 static GstRTSPResult
436 do_connect (const gchar * ip, guint16 port, GstPollFD * fdout,
437     GstPoll * fdset, GTimeVal * timeout)
438 {
439   gint fd;
440   struct sockaddr_in sa_in;
441   gint ret;
442 #ifdef G_OS_WIN32
443   unsigned long flags = 1;
444 #endif /* G_OS_WIN32 */
445   GstClockTime to;
446   gint retval;
447
448   memset (&sa_in, 0, sizeof (sa_in));
449   sa_in.sin_family = AF_INET;   /* network socket */
450   sa_in.sin_port = htons (port);        /* on port */
451   sa_in.sin_addr.s_addr = inet_addr (ip);       /* on host ip */
452
453   fd = socket (AF_INET, SOCK_STREAM, 0);
454   if (fd == -1)
455     goto no_socket;
456
457   /* set to non-blocking mode so that we can cancel the connect */
458 #ifndef G_OS_WIN32
459   fcntl (fd, F_SETFL, O_NONBLOCK);
460 #else
461   ioctlsocket (fd, FIONBIO, &flags);
462 #endif /* G_OS_WIN32 */
463
464   /* add the socket to our fdset */
465   ADD_POLLFD (fdset, fdout, fd);
466
467   /* we are going to connect ASYNC now */
468   ret = connect (fd, (struct sockaddr *) &sa_in, sizeof (sa_in));
469   if (ret == 0)
470     goto done;
471   if (!ERRNO_IS_EINPROGRESS)
472     goto sys_error;
473
474   /* wait for connect to complete up to the specified timeout or until we got
475    * interrupted. */
476   gst_poll_fd_ctl_write (fdset, fdout, TRUE);
477
478   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
479
480   do {
481     retval = gst_poll_wait (fdset, to);
482   } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
483
484   if (retval == 0)
485     goto timeout;
486   else if (retval == -1)
487     goto sys_error;
488
489   /* we can still have an error connecting on windows */
490   if (gst_poll_fd_has_error (fdset, fdout)) {
491     socklen_t len = sizeof (errno);
492 #ifndef G_OS_WIN32
493     getsockopt (fd, SOL_SOCKET, SO_ERROR, &errno, &len);
494 #else
495     getsockopt (fd, SOL_SOCKET, SO_ERROR, (char *) &errno, &len);
496 #endif
497     goto sys_error;
498   }
499
500   gst_poll_fd_ignored (fdset, fdout);
501 done:
502
503   return GST_RTSP_OK;
504
505   /* ERRORS */
506 no_socket:
507   {
508     GST_ERROR ("no socket %d (%s)", errno, g_strerror (errno));
509     return GST_RTSP_ESYS;
510   }
511 sys_error:
512   {
513     GST_ERROR ("system error %d (%s)", errno, g_strerror (errno));
514     REMOVE_POLLFD (fdset, fdout);
515     return GST_RTSP_ESYS;
516   }
517 timeout:
518   {
519     GST_ERROR ("timeout");
520     REMOVE_POLLFD (fdset, fdout);
521     return GST_RTSP_ETIMEOUT;
522   }
523 }
524
525 static GstRTSPResult
526 setup_tunneling (GstRTSPConnection * conn, GTimeVal * timeout)
527 {
528   gint i;
529   GstRTSPResult res;
530   gchar *str;
531   guint idx, line;
532   gint retval;
533   GstClockTime to;
534   const gchar *ip;
535   guint16 port;
536   gchar codestr[4], *resultstr;
537   gint code;
538   GstRTSPUrl *url;
539
540   /* create a random sessionid */
541   for (i = 0; i < TUNNELID_LEN; i++)
542     conn->tunnelid[i] = g_random_int_range ('a', 'z');
543   conn->tunnelid[TUNNELID_LEN - 1] = '\0';
544
545   url = conn->url;
546
547   /* */
548   str = g_strdup_printf ("GET %s%s%s HTTP/1.0\r\n"
549       "x-sessioncookie: %s\r\n"
550       "Accept: application/x-rtsp-tunnelled\r\n"
551       "Pragma: no-cache\r\n"
552       "Cache-Control: no-cache\r\n" "\r\n",
553       url->abspath, url->query ? "?" : "", url->query ? url->query : "",
554       conn->tunnelid);
555
556   /* we start by writing to this fd */
557   conn->writefd = &conn->fd0;
558
559   res = gst_rtsp_connection_write (conn, (guint8 *) str, strlen (str), timeout);
560   g_free (str);
561   if (res != GST_RTSP_OK)
562     goto write_failed;
563
564   gst_poll_fd_ctl_write (conn->fdset, &conn->fd0, FALSE);
565   gst_poll_fd_ctl_read (conn->fdset, &conn->fd0, TRUE);
566
567   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
568
569   line = 0;
570   while (TRUE) {
571     guint8 buffer[4096];
572
573     idx = 0;
574     while (TRUE) {
575       res = read_line (conn->fd0.fd, buffer, &idx, sizeof (buffer), NULL);
576       if (res == GST_RTSP_EEOF)
577         goto eof;
578       if (res == GST_RTSP_OK)
579         break;
580       if (res != GST_RTSP_EINTR)
581         goto read_error;
582
583       do {
584         retval = gst_poll_wait (conn->fdset, to);
585       } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
586
587       /* check for timeout */
588       if (retval == 0)
589         goto timeout;
590
591       if (retval == -1) {
592         if (errno == EBUSY)
593           goto stopped;
594         else
595           goto select_error;
596       }
597     }
598
599     /* check for last line */
600     if (buffer[0] == '\r')
601       buffer[0] = '\0';
602     if (buffer[0] == '\0')
603       break;
604
605     if (line == 0) {
606       /* first line, parse response */
607       gchar versionstr[20];
608       gchar *bptr;
609
610       bptr = (gchar *) buffer;
611
612       parse_string (versionstr, sizeof (versionstr), &bptr);
613       parse_string (codestr, sizeof (codestr), &bptr);
614       code = atoi (codestr);
615
616       while (g_ascii_isspace (*bptr))
617         bptr++;
618
619       resultstr = bptr;
620
621       if (code != 200)
622         goto wrong_result;
623     } else {
624       gchar key[32];
625       gchar *value;
626
627       /* other lines, parse key/value */
628       res = parse_key_value (buffer, key, sizeof (key), &value);
629       if (res == GST_RTSP_OK) {
630         /* we got a new ip address */
631         if (g_ascii_strcasecmp (key, "x-server-ip-address") == 0) {
632           g_free (conn->ip);
633           conn->ip = g_strdup (value);
634         }
635       }
636     }
637     line++;
638   }
639
640   if (!(ip = do_resolve (conn->ip)))
641     goto not_resolved;
642
643   /* get the port from the url */
644   gst_rtsp_url_get_port (conn->url, &port);
645
646   /* connect to the host/port */
647   res = do_connect (ip, port, &conn->fd1, conn->fdset, timeout);
648   if (res != GST_RTSP_OK)
649     goto connect_failed;
650
651   /* this is now our writing socket */
652   conn->writefd = &conn->fd1;
653
654   /* */
655   str = g_strdup_printf ("POST %s%s%s HTTP/1.0\r\n"
656       "x-sessioncookie: %s\r\n"
657       "Content-Type: application/x-rtsp-tunnelled\r\n"
658       "Pragma: no-cache\r\n"
659       "Cache-Control: no-cache\r\n"
660       "Content-Length: 32767\r\n"
661       "Expires: Sun, 9 Jan 1972 00:00:00 GMT\r\n"
662       "\r\n",
663       url->abspath, url->query ? "?" : "", url->query ? url->query : "",
664       conn->tunnelid);
665
666   /* we start by writing to this fd */
667   conn->writefd = &conn->fd1;
668
669   res = gst_rtsp_connection_write (conn, (guint8 *) str, strlen (str), timeout);
670   g_free (str);
671   if (res != GST_RTSP_OK)
672     goto write_failed;
673
674   return res;
675
676   /* ERRORS */
677 write_failed:
678   {
679     GST_ERROR ("write failed", res);
680     return res;
681   }
682 eof:
683   {
684     return GST_RTSP_EEOF;
685   }
686 read_error:
687   {
688     return res;
689   }
690 timeout:
691   {
692     return GST_RTSP_ETIMEOUT;
693   }
694 select_error:
695   {
696     return GST_RTSP_ESYS;
697   }
698 stopped:
699   {
700     return GST_RTSP_EINTR;
701   }
702 wrong_result:
703   {
704     GST_ERROR ("got failure response %d %s", code, resultstr);
705     return GST_RTSP_ERROR;
706   }
707 not_resolved:
708   {
709     GST_ERROR ("could not resolve %s", conn->ip);
710     return GST_RTSP_ENET;
711   }
712 connect_failed:
713   {
714     GST_ERROR ("failed to connect");
715     return res;
716   }
717 }
718
719 /**
720  * gst_rtsp_connection_connect:
721  * @conn: a #GstRTSPConnection 
722  * @timeout: a #GTimeVal timeout
723  *
724  * Attempt to connect to the url of @conn made with
725  * gst_rtsp_connection_create(). If @timeout is #NULL this function can block
726  * forever. If @timeout contains a valid timeout, this function will return
727  * #GST_RTSP_ETIMEOUT after the timeout expired.
728  *
729  * This function can be cancelled with gst_rtsp_connection_flush().
730  *
731  * Returns: #GST_RTSP_OK when a connection could be made.
732  */
733 GstRTSPResult
734 gst_rtsp_connection_connect (GstRTSPConnection * conn, GTimeVal * timeout)
735 {
736   GstRTSPResult res;
737   const gchar *ip;
738   guint16 port;
739   GstRTSPUrl *url;
740 #ifdef G_OS_WIN32
741   unsigned long flags = 1;
742 #endif /* G_OS_WIN32 */
743
744   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
745   g_return_val_if_fail (conn->url != NULL, GST_RTSP_EINVAL);
746   g_return_val_if_fail (conn->fd0.fd < 0, GST_RTSP_EINVAL);
747
748   url = conn->url;
749
750   if (!(ip = do_resolve (url->host)))
751     goto not_resolved;
752
753   /* get the port from the url */
754   gst_rtsp_url_get_port (url, &port);
755
756   /* connect to the host/port */
757   res = do_connect (ip, port, &conn->fd0, conn->fdset, timeout);
758   if (res != GST_RTSP_OK)
759     goto connect_failed;
760
761   g_free (conn->ip);
762   conn->ip = g_strdup (ip);
763
764   /* this is our read URL */
765   conn->readfd = &conn->fd0;
766
767   if (conn->tunneled) {
768     res = setup_tunneling (conn, timeout);
769     if (res != GST_RTSP_OK)
770       goto tunneling_failed;
771   } else {
772     conn->writefd = &conn->fd0;
773   }
774
775   return GST_RTSP_OK;
776
777 not_resolved:
778   {
779     GST_ERROR ("could not resolve %s", url->host);
780     return GST_RTSP_ENET;
781   }
782 connect_failed:
783   {
784     GST_ERROR ("failed to connect");
785     return res;
786   }
787 tunneling_failed:
788   {
789     GST_ERROR ("failed to setup tunneling");
790     return res;
791   }
792 }
793
794 static void
795 md5_digest_to_hex_string (unsigned char digest[16], char string[33])
796 {
797   static const char hexdigits[] = "0123456789abcdef";
798   int i;
799
800   for (i = 0; i < 16; i++) {
801     string[i * 2] = hexdigits[(digest[i] >> 4) & 0x0f];
802     string[i * 2 + 1] = hexdigits[digest[i] & 0x0f];
803   }
804   string[32] = 0;
805 }
806
807 static void
808 auth_digest_compute_hex_urp (const gchar * username,
809     const gchar * realm, const gchar * password, gchar hex_urp[33])
810 {
811   struct MD5Context md5_context;
812   unsigned char digest[16];
813
814   MD5Init (&md5_context);
815   MD5Update (&md5_context, username, strlen (username));
816   MD5Update (&md5_context, ":", 1);
817   MD5Update (&md5_context, realm, strlen (realm));
818   MD5Update (&md5_context, ":", 1);
819   MD5Update (&md5_context, password, strlen (password));
820   MD5Final (digest, &md5_context);
821   md5_digest_to_hex_string (digest, hex_urp);
822 }
823
824 static void
825 auth_digest_compute_response (const gchar * method,
826     const gchar * uri, const gchar * hex_a1, const gchar * nonce,
827     gchar response[33])
828 {
829   char hex_a2[33];
830   struct MD5Context md5_context;
831   unsigned char digest[16];
832
833   /* compute A2 */
834   MD5Init (&md5_context);
835   MD5Update (&md5_context, method, strlen (method));
836   MD5Update (&md5_context, ":", 1);
837   MD5Update (&md5_context, uri, strlen (uri));
838   MD5Final (digest, &md5_context);
839   md5_digest_to_hex_string (digest, hex_a2);
840
841   /* compute KD */
842   MD5Init (&md5_context);
843   MD5Update (&md5_context, hex_a1, strlen (hex_a1));
844   MD5Update (&md5_context, ":", 1);
845   MD5Update (&md5_context, nonce, strlen (nonce));
846   MD5Update (&md5_context, ":", 1);
847
848   MD5Update (&md5_context, hex_a2, 32);
849   MD5Final (digest, &md5_context);
850   md5_digest_to_hex_string (digest, response);
851 }
852
853 static void
854 add_auth_header (GstRTSPConnection * conn, GstRTSPMessage * message)
855 {
856   switch (conn->auth_method) {
857     case GST_RTSP_AUTH_BASIC:{
858       gchar *user_pass =
859           g_strdup_printf ("%s:%s", conn->username, conn->passwd);
860       gchar *user_pass64 =
861           gst_rtsp_base64_encode (user_pass, strlen (user_pass));
862       gchar *auth_string = g_strdup_printf ("Basic %s", user_pass64);
863
864       gst_rtsp_message_take_header (message, GST_RTSP_HDR_AUTHORIZATION,
865           auth_string);
866
867       g_free (user_pass);
868       g_free (user_pass64);
869       break;
870     }
871     case GST_RTSP_AUTH_DIGEST:{
872       gchar response[33], hex_urp[33];
873       gchar *auth_string, *auth_string2;
874       gchar *realm;
875       gchar *nonce;
876       gchar *opaque;
877       const gchar *uri;
878       const gchar *method;
879
880       /* we need to have some params set */
881       if (conn->auth_params == NULL)
882         break;
883
884       /* we need the realm and nonce */
885       realm = (gchar *) g_hash_table_lookup (conn->auth_params, "realm");
886       nonce = (gchar *) g_hash_table_lookup (conn->auth_params, "nonce");
887       if (realm == NULL || nonce == NULL)
888         break;
889
890       auth_digest_compute_hex_urp (conn->username, realm, conn->passwd,
891           hex_urp);
892
893       method = gst_rtsp_method_as_text (message->type_data.request.method);
894       uri = message->type_data.request.uri;
895
896       /* Assume no qop, algorithm=md5, stale=false */
897       /* For algorithm MD5, a1 = urp. */
898       auth_digest_compute_response (method, uri, hex_urp, nonce, response);
899       auth_string = g_strdup_printf ("Digest username=\"%s\", "
900           "realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
901           conn->username, realm, nonce, uri, response);
902
903       opaque = (gchar *) g_hash_table_lookup (conn->auth_params, "opaque");
904       if (opaque) {
905         auth_string2 = g_strdup_printf ("%s, opaque=\"%s\"", auth_string,
906             opaque);
907         g_free (auth_string);
908         auth_string = auth_string2;
909       }
910       gst_rtsp_message_take_header (message, GST_RTSP_HDR_AUTHORIZATION,
911           auth_string);
912       break;
913     }
914     default:
915       /* Nothing to do */
916       break;
917   }
918 }
919
920 static void
921 gen_date_string (gchar * date_string, guint len)
922 {
923   GTimeVal tv;
924   time_t t;
925 #ifdef HAVE_GMTIME_R
926   struct tm tm_;
927 #endif
928
929   g_get_current_time (&tv);
930   t = (time_t) tv.tv_sec;
931
932 #ifdef HAVE_GMTIME_R
933   strftime (date_string, len, "%a, %d %b %Y %H:%M:%S GMT", gmtime_r (&t, &tm_));
934 #else
935   strftime (date_string, len, "%a, %d %b %Y %H:%M:%S GMT", gmtime (&t));
936 #endif
937 }
938
939 static GstRTSPResult
940 write_bytes (gint fd, const guint8 * buffer, guint * idx, guint size)
941 {
942   guint left;
943
944   if (*idx > size)
945     return GST_RTSP_ERROR;
946
947   left = size - *idx;
948
949   while (left) {
950     gint r;
951
952     r = WRITE_SOCKET (fd, &buffer[*idx], left);
953     if (r == 0) {
954       return GST_RTSP_EINTR;
955     } else if (r < 0) {
956       if (ERRNO_IS_EAGAIN)
957         return GST_RTSP_EINTR;
958       if (!ERRNO_IS_EINTR)
959         return GST_RTSP_ESYS;
960     } else {
961       left -= r;
962       *idx += r;
963     }
964   }
965   return GST_RTSP_OK;
966 }
967
968 static gint
969 fill_bytes (gint fd, guint8 * buffer, guint size, DecodeCtx * ctx)
970 {
971   gint out = 0;
972
973   if (ctx) {
974     gint r;
975
976     while (size > 0) {
977       while (size > 0 && ctx->cout < 3) {
978         /* we have some leftover bytes */
979         *buffer++ = ctx->out[ctx->cout];
980         ctx->cout++;
981         size--;
982         out++;
983       }
984       /* nothing in the buffer */
985       if (size == 0)
986         break;
987
988       /* try to read more bytes */
989       r = READ_SOCKET (fd, &ctx->in[ctx->cin], 4 - ctx->cin);
990       if (r <= 0) {
991         if (out == 0)
992           out = r;
993         break;
994       }
995
996       ctx->cin += r;
997       if (ctx->cin == 4) {
998         r = g_base64_decode_step ((const gchar *) ctx->in, 4,
999             (guchar *) ctx->out, &ctx->state, &ctx->save);
1000         ctx->cout = 0;
1001         ctx->cin = 0;
1002       }
1003     }
1004   } else {
1005     out = READ_SOCKET (fd, buffer, size);
1006   }
1007
1008   return out;
1009 }
1010
1011 static GstRTSPResult
1012 read_bytes (gint fd, guint8 * buffer, guint * idx, guint size, DecodeCtx * ctx)
1013 {
1014   guint left;
1015
1016   if (*idx > size)
1017     return GST_RTSP_ERROR;
1018
1019   left = size - *idx;
1020
1021   while (left) {
1022     gint r;
1023
1024     r = fill_bytes (fd, &buffer[*idx], left, ctx);
1025     if (r == 0) {
1026       return GST_RTSP_EEOF;
1027     } else if (r < 0) {
1028       if (ERRNO_IS_EAGAIN)
1029         return GST_RTSP_EINTR;
1030       if (!ERRNO_IS_EINTR)
1031         return GST_RTSP_ESYS;
1032     } else {
1033       left -= r;
1034       *idx += r;
1035     }
1036   }
1037   return GST_RTSP_OK;
1038 }
1039
1040 static GstRTSPResult
1041 read_line (gint fd, guint8 * buffer, guint * idx, guint size, DecodeCtx * ctx)
1042 {
1043   while (TRUE) {
1044     guint8 c;
1045     gint r;
1046
1047     r = fill_bytes (fd, &c, 1, ctx);
1048     if (r == 0) {
1049       return GST_RTSP_EEOF;
1050     } else if (r < 0) {
1051       if (ERRNO_IS_EAGAIN)
1052         return GST_RTSP_EINTR;
1053       if (!ERRNO_IS_EINTR)
1054         return GST_RTSP_ESYS;
1055     } else {
1056       if (c == '\n')            /* end on \n */
1057         break;
1058       if (c == '\r')            /* ignore \r */
1059         continue;
1060
1061       if (*idx < size - 1)
1062         buffer[(*idx)++] = c;
1063     }
1064   }
1065   buffer[*idx] = '\0';
1066
1067   return GST_RTSP_OK;
1068 }
1069
1070 /**
1071  * gst_rtsp_connection_write:
1072  * @conn: a #GstRTSPConnection
1073  * @data: the data to write
1074  * @size: the size of @data
1075  * @timeout: a timeout value or #NULL
1076  *
1077  * Attempt to write @size bytes of @data to the connected @conn, blocking up to
1078  * the specified @timeout. @timeout can be #NULL, in which case this function
1079  * might block forever.
1080  * 
1081  * This function can be cancelled with gst_rtsp_connection_flush().
1082  *
1083  * Returns: #GST_RTSP_OK on success.
1084  */
1085 GstRTSPResult
1086 gst_rtsp_connection_write (GstRTSPConnection * conn, const guint8 * data,
1087     guint size, GTimeVal * timeout)
1088 {
1089   guint offset;
1090   gint retval;
1091   GstClockTime to;
1092   GstRTSPResult res;
1093
1094   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1095   g_return_val_if_fail (data != NULL || size == 0, GST_RTSP_EINVAL);
1096   g_return_val_if_fail (conn->writefd != NULL, GST_RTSP_EINVAL);
1097
1098   gst_poll_set_controllable (conn->fdset, TRUE);
1099   gst_poll_fd_ctl_write (conn->fdset, conn->writefd, TRUE);
1100   gst_poll_fd_ctl_read (conn->fdset, conn->readfd, FALSE);
1101   /* clear all previous poll results */
1102   gst_poll_fd_ignored (conn->fdset, conn->writefd);
1103   gst_poll_fd_ignored (conn->fdset, conn->readfd);
1104
1105   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1106
1107   offset = 0;
1108
1109   while (TRUE) {
1110     /* try to write */
1111     res = write_bytes (conn->writefd->fd, data, &offset, size);
1112     if (res == GST_RTSP_OK)
1113       break;
1114     if (res != GST_RTSP_EINTR)
1115       goto write_error;
1116
1117     /* not all is written, wait until we can write more */
1118     do {
1119       retval = gst_poll_wait (conn->fdset, to);
1120     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1121
1122     if (retval == 0)
1123       goto timeout;
1124
1125     if (retval == -1) {
1126       if (errno == EBUSY)
1127         goto stopped;
1128       else
1129         goto select_error;
1130     }
1131   }
1132   return GST_RTSP_OK;
1133
1134   /* ERRORS */
1135 timeout:
1136   {
1137     return GST_RTSP_ETIMEOUT;
1138   }
1139 select_error:
1140   {
1141     return GST_RTSP_ESYS;
1142   }
1143 stopped:
1144   {
1145     return GST_RTSP_EINTR;
1146   }
1147 write_error:
1148   {
1149     return res;
1150   }
1151 }
1152
1153 static GString *
1154 message_to_string (GstRTSPConnection * conn, GstRTSPMessage * message)
1155 {
1156   GString *str = NULL;
1157
1158   str = g_string_new ("");
1159
1160   switch (message->type) {
1161     case GST_RTSP_MESSAGE_REQUEST:
1162       /* create request string, add CSeq */
1163       g_string_append_printf (str, "%s %s RTSP/1.0\r\n"
1164           "CSeq: %d\r\n",
1165           gst_rtsp_method_as_text (message->type_data.request.method),
1166           message->type_data.request.uri, conn->cseq++);
1167       /* add session id if we have one */
1168       if (conn->session_id[0] != '\0') {
1169         gst_rtsp_message_add_header (message, GST_RTSP_HDR_SESSION,
1170             conn->session_id);
1171       }
1172       /* add any authentication headers */
1173       add_auth_header (conn, message);
1174       break;
1175     case GST_RTSP_MESSAGE_RESPONSE:
1176       /* create response string */
1177       g_string_append_printf (str, "RTSP/1.0 %d %s\r\n",
1178           message->type_data.response.code, message->type_data.response.reason);
1179       break;
1180     case GST_RTSP_MESSAGE_DATA:
1181     {
1182       guint8 data_header[4];
1183
1184       /* prepare data header */
1185       data_header[0] = '$';
1186       data_header[1] = message->type_data.data.channel;
1187       data_header[2] = (message->body_size >> 8) & 0xff;
1188       data_header[3] = message->body_size & 0xff;
1189
1190       /* create string with header and data */
1191       str = g_string_append_len (str, (gchar *) data_header, 4);
1192       str =
1193           g_string_append_len (str, (gchar *) message->body,
1194           message->body_size);
1195       break;
1196     }
1197     default:
1198       g_string_free (str, TRUE);
1199       g_return_val_if_reached (NULL);
1200       break;
1201   }
1202
1203   /* append headers and body */
1204   if (message->type != GST_RTSP_MESSAGE_DATA) {
1205     gchar date_string[100];
1206
1207     gen_date_string (date_string, sizeof (date_string));
1208
1209     /* add date header */
1210     gst_rtsp_message_add_header (message, GST_RTSP_HDR_DATE, date_string);
1211
1212     /* append headers */
1213     gst_rtsp_message_append_headers (message, str);
1214
1215     /* append Content-Length and body if needed */
1216     if (message->body != NULL && message->body_size > 0) {
1217       gchar *len;
1218
1219       len = g_strdup_printf ("%d", message->body_size);
1220       g_string_append_printf (str, "%s: %s\r\n",
1221           gst_rtsp_header_as_text (GST_RTSP_HDR_CONTENT_LENGTH), len);
1222       g_free (len);
1223       /* header ends here */
1224       g_string_append (str, "\r\n");
1225       str =
1226           g_string_append_len (str, (gchar *) message->body,
1227           message->body_size);
1228     } else {
1229       /* just end headers */
1230       g_string_append (str, "\r\n");
1231     }
1232   }
1233
1234   return str;
1235 }
1236
1237 /**
1238  * gst_rtsp_connection_send:
1239  * @conn: a #GstRTSPConnection
1240  * @message: the message to send
1241  * @timeout: a timeout value or #NULL
1242  *
1243  * Attempt to send @message to the connected @conn, blocking up to
1244  * the specified @timeout. @timeout can be #NULL, in which case this function
1245  * might block forever.
1246  * 
1247  * This function can be cancelled with gst_rtsp_connection_flush().
1248  *
1249  * Returns: #GST_RTSP_OK on success.
1250  */
1251 GstRTSPResult
1252 gst_rtsp_connection_send (GstRTSPConnection * conn, GstRTSPMessage * message,
1253     GTimeVal * timeout)
1254 {
1255   GString *string = NULL;
1256   GstRTSPResult res;
1257   gchar *str;
1258   gsize len;
1259
1260   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1261   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
1262
1263   if (!(string = message_to_string (conn, message)))
1264     goto no_message;
1265
1266   if (conn->tunneled) {
1267     str = g_base64_encode ((const guchar *) string->str, string->len);
1268     g_string_free (string, TRUE);
1269     len = strlen (str);
1270   } else {
1271     str = string->str;
1272     len = string->len;
1273     g_string_free (string, FALSE);
1274   }
1275
1276   /* write request */
1277   res = gst_rtsp_connection_write (conn, (guint8 *) str, len, timeout);
1278
1279   g_free (str);
1280
1281   return res;
1282
1283 no_message:
1284   {
1285     g_warning ("Wrong message");
1286     return GST_RTSP_EINVAL;
1287   }
1288 }
1289
1290 static void
1291 parse_string (gchar * dest, gint size, gchar ** src)
1292 {
1293   gint idx;
1294
1295   idx = 0;
1296   /* skip spaces */
1297   while (g_ascii_isspace (**src))
1298     (*src)++;
1299
1300   while (!g_ascii_isspace (**src) && **src != '\0') {
1301     if (idx < size - 1)
1302       dest[idx++] = **src;
1303     (*src)++;
1304   }
1305   if (size > 0)
1306     dest[idx] = '\0';
1307 }
1308
1309 static void
1310 parse_key (gchar * dest, gint size, gchar ** src)
1311 {
1312   gint idx;
1313
1314   idx = 0;
1315   while (**src != ':' && **src != '\0') {
1316     if (idx < size - 1)
1317       dest[idx++] = **src;
1318     (*src)++;
1319   }
1320   if (size > 0)
1321     dest[idx] = '\0';
1322 }
1323
1324 static GstRTSPResult
1325 parse_response_status (guint8 * buffer, GstRTSPMessage * msg)
1326 {
1327   GstRTSPResult res;
1328   gchar versionstr[20];
1329   gchar codestr[4];
1330   gint code;
1331   gchar *bptr;
1332
1333   bptr = (gchar *) buffer;
1334
1335   parse_string (versionstr, sizeof (versionstr), &bptr);
1336   parse_string (codestr, sizeof (codestr), &bptr);
1337   code = atoi (codestr);
1338
1339   while (g_ascii_isspace (*bptr))
1340     bptr++;
1341
1342   if (strcmp (versionstr, "RTSP/1.0") == 0)
1343     GST_RTSP_CHECK (gst_rtsp_message_init_response (msg, code, bptr, NULL),
1344         parse_error);
1345   else if (strncmp (versionstr, "RTSP/", 5) == 0) {
1346     GST_RTSP_CHECK (gst_rtsp_message_init_response (msg, code, bptr, NULL),
1347         parse_error);
1348     msg->type_data.response.version = GST_RTSP_VERSION_INVALID;
1349   } else
1350     goto parse_error;
1351
1352   return GST_RTSP_OK;
1353
1354 parse_error:
1355   {
1356     return GST_RTSP_EPARSE;
1357   }
1358 }
1359
1360 static GstRTSPResult
1361 parse_request_line (GstRTSPConnection * conn, guint8 * buffer,
1362     GstRTSPMessage * msg)
1363 {
1364   GstRTSPResult res = GST_RTSP_OK;
1365   gchar versionstr[20];
1366   gchar methodstr[20];
1367   gchar urlstr[4096];
1368   gchar *bptr;
1369   GstRTSPMethod method;
1370   GstRTSPTunnelState tstate = TUNNEL_STATE_NONE;
1371
1372   bptr = (gchar *) buffer;
1373
1374   parse_string (methodstr, sizeof (methodstr), &bptr);
1375   method = gst_rtsp_find_method (methodstr);
1376   if (method == GST_RTSP_INVALID) {
1377     /* a tunnel request is allowed when we don't have one yet */
1378     if (conn->tstate != TUNNEL_STATE_NONE)
1379       goto invalid_method;
1380     /* we need GET or POST for a valid tunnel request */
1381     if (!strcmp (methodstr, "GET"))
1382       tstate = TUNNEL_STATE_GET;
1383     else if (!strcmp (methodstr, "POST"))
1384       tstate = TUNNEL_STATE_POST;
1385     else
1386       goto invalid_method;
1387   }
1388
1389   parse_string (urlstr, sizeof (urlstr), &bptr);
1390   if (*urlstr == '\0')
1391     goto invalid_url;
1392
1393   parse_string (versionstr, sizeof (versionstr), &bptr);
1394
1395   if (*bptr != '\0')
1396     goto invalid_version;
1397
1398   if (strcmp (versionstr, "RTSP/1.0") == 0) {
1399     res = gst_rtsp_message_init_request (msg, method, urlstr);
1400   } else if (strncmp (versionstr, "RTSP/", 5) == 0) {
1401     res = gst_rtsp_message_init_request (msg, method, urlstr);
1402     msg->type_data.request.version = GST_RTSP_VERSION_INVALID;
1403   } else if (strcmp (versionstr, "HTTP/1.0") == 0) {
1404     /* tunnel request, we need a tunnel method */
1405     if (tstate == TUNNEL_STATE_NONE) {
1406       res = GST_RTSP_EPARSE;
1407     } else {
1408       conn->tstate = tstate;
1409     }
1410   } else {
1411     res = GST_RTSP_EPARSE;
1412   }
1413
1414   return res;
1415
1416   /* ERRORS */
1417 invalid_method:
1418   {
1419     GST_ERROR ("invalid method %s", methodstr);
1420     return GST_RTSP_EPARSE;
1421   }
1422 invalid_url:
1423   {
1424     GST_ERROR ("invalid url %s", urlstr);
1425     return GST_RTSP_EPARSE;
1426   }
1427 invalid_version:
1428   {
1429     GST_ERROR ("invalid version");
1430     return GST_RTSP_EPARSE;
1431   }
1432 }
1433
1434 static GstRTSPResult
1435 parse_key_value (guint8 * buffer, gchar * key, guint keysize, gchar ** value)
1436 {
1437   gchar *bptr;
1438
1439   bptr = (gchar *) buffer;
1440
1441   /* read key */
1442   parse_key (key, keysize, &bptr);
1443   if (*bptr != ':')
1444     goto no_column;
1445
1446   bptr++;
1447   while (g_ascii_isspace (*bptr))
1448     bptr++;
1449
1450   *value = bptr;
1451
1452   return GST_RTSP_OK;
1453
1454   /* ERRORS */
1455 no_column:
1456   {
1457     return GST_RTSP_EPARSE;
1458   }
1459 }
1460
1461 /* parsing lines means reading a Key: Value pair */
1462 static GstRTSPResult
1463 parse_line (GstRTSPConnection * conn, guint8 * buffer, GstRTSPMessage * msg)
1464 {
1465   GstRTSPResult res;
1466   gchar key[32];
1467   gchar *value;
1468   GstRTSPHeaderField field;
1469
1470   res = parse_key_value (buffer, key, sizeof (key), &value);
1471   if (res != GST_RTSP_OK)
1472     goto parse_error;
1473
1474   if (conn->tstate == TUNNEL_STATE_GET || conn->tstate == TUNNEL_STATE_POST) {
1475     /* save the tunnel session in the connection */
1476     if (!strcmp (key, "x-sessioncookie")) {
1477       strncpy (conn->tunnelid, value, TUNNELID_LEN);
1478       conn->tunnelid[TUNNELID_LEN - 1] = '\0';
1479       conn->tunneled = TRUE;
1480     }
1481   } else {
1482     field = gst_rtsp_find_header_field (key);
1483     if (field != GST_RTSP_HDR_INVALID)
1484       gst_rtsp_message_add_header (msg, field, value);
1485   }
1486
1487   return GST_RTSP_OK;
1488
1489   /* ERRORS */
1490 parse_error:
1491   {
1492     return res;
1493   }
1494 }
1495
1496 /* returns:
1497  *  GST_RTSP_OK when a complete message was read.
1498  *  GST_RTSP_EEOF: when the socket is closed
1499  *  GST_RTSP_EINTR: when more data is needed.
1500  *  GST_RTSP_..: some other error occured.
1501  */
1502 static GstRTSPResult
1503 build_next (GstRTSPBuilder * builder, GstRTSPMessage * message,
1504     GstRTSPConnection * conn)
1505 {
1506   GstRTSPResult res;
1507
1508   while (TRUE) {
1509     switch (builder->state) {
1510       case STATE_START:
1511         builder->offset = 0;
1512         res =
1513             read_bytes (conn->readfd->fd, (guint8 *) builder->buffer,
1514             &builder->offset, 1, conn->ctxp);
1515         if (res != GST_RTSP_OK)
1516           goto done;
1517
1518         /* we have 1 bytes now and we can see if this is a data message or
1519          * not */
1520         if (builder->buffer[0] == '$') {
1521           /* data message, prepare for the header */
1522           builder->state = STATE_DATA_HEADER;
1523         } else {
1524           builder->line = 0;
1525           builder->state = STATE_READ_LINES;
1526         }
1527         break;
1528       case STATE_DATA_HEADER:
1529       {
1530         res =
1531             read_bytes (conn->readfd->fd, (guint8 *) builder->buffer,
1532             &builder->offset, 4, conn->ctxp);
1533         if (res != GST_RTSP_OK)
1534           goto done;
1535
1536         gst_rtsp_message_init_data (message, builder->buffer[1]);
1537
1538         builder->body_len = (builder->buffer[2] << 8) | builder->buffer[3];
1539         builder->body_data = g_malloc (builder->body_len + 1);
1540         builder->body_data[builder->body_len] = '\0';
1541         builder->offset = 0;
1542         builder->state = STATE_DATA_BODY;
1543         break;
1544       }
1545       case STATE_DATA_BODY:
1546       {
1547         res =
1548             read_bytes (conn->readfd->fd, builder->body_data, &builder->offset,
1549             builder->body_len, conn->ctxp);
1550         if (res != GST_RTSP_OK)
1551           goto done;
1552
1553         /* we have the complete body now, store in the message adjusting the
1554          * length to include the traling '\0' */
1555         gst_rtsp_message_take_body (message,
1556             (guint8 *) builder->body_data, builder->body_len + 1);
1557         builder->body_data = NULL;
1558         builder->body_len = 0;
1559
1560         builder->state = STATE_END;
1561         break;
1562       }
1563       case STATE_READ_LINES:
1564       {
1565         res = read_line (conn->readfd->fd, builder->buffer, &builder->offset,
1566             sizeof (builder->buffer), conn->ctxp);
1567         if (res != GST_RTSP_OK)
1568           goto done;
1569
1570         /* we have a regular response */
1571         if (builder->buffer[0] == '\r') {
1572           builder->buffer[0] = '\0';
1573         }
1574
1575         if (builder->buffer[0] == '\0') {
1576           gchar *hdrval;
1577
1578           /* empty line, end of message header */
1579           /* see if there is a Content-Length header */
1580           if (gst_rtsp_message_get_header (message,
1581                   GST_RTSP_HDR_CONTENT_LENGTH, &hdrval, 0) == GST_RTSP_OK) {
1582             /* there is, prepare to read the body */
1583             builder->body_len = atol (hdrval);
1584             builder->body_data = g_malloc (builder->body_len + 1);
1585             builder->body_data[builder->body_len] = '\0';
1586             builder->offset = 0;
1587             builder->state = STATE_DATA_BODY;
1588           } else {
1589             builder->state = STATE_END;
1590           }
1591           break;
1592         }
1593
1594         /* we have a line */
1595         if (builder->line == 0) {
1596           /* first line, check for response status */
1597           if (memcmp (builder->buffer, "RTSP", 4) == 0) {
1598             res = parse_response_status (builder->buffer, message);
1599           } else {
1600             res = parse_request_line (conn, builder->buffer, message);
1601           }
1602           /* the first line must parse without errors */
1603           if (res != GST_RTSP_OK)
1604             goto done;
1605         } else {
1606           /* else just parse the line, ignore errors */
1607           parse_line (conn, builder->buffer, message);
1608         }
1609         builder->line++;
1610         builder->offset = 0;
1611         break;
1612       }
1613       case STATE_END:
1614       {
1615         gchar *session_id;
1616
1617         if (conn->tstate == TUNNEL_STATE_GET) {
1618           res = GST_RTSP_ETGET;
1619           goto done;
1620         } else if (conn->tstate == TUNNEL_STATE_POST) {
1621           res = GST_RTSP_ETPOST;
1622           goto done;
1623         }
1624
1625         if (message->type == GST_RTSP_MESSAGE_DATA) {
1626           /* data messages don't have headers */
1627           res = GST_RTSP_OK;
1628           goto done;
1629         }
1630
1631         /* save session id in the connection for further use */
1632         if (gst_rtsp_message_get_header (message, GST_RTSP_HDR_SESSION,
1633                 &session_id, 0) == GST_RTSP_OK) {
1634           gint maxlen, i;
1635
1636           maxlen = sizeof (conn->session_id) - 1;
1637           /* the sessionid can have attributes marked with ;
1638            * Make sure we strip them */
1639           for (i = 0; session_id[i] != '\0'; i++) {
1640             if (session_id[i] == ';') {
1641               maxlen = i;
1642               /* parse timeout */
1643               do {
1644                 i++;
1645               } while (g_ascii_isspace (session_id[i]));
1646               if (g_str_has_prefix (&session_id[i], "timeout=")) {
1647                 gint to;
1648
1649                 /* if we parsed something valid, configure */
1650                 if ((to = atoi (&session_id[i + 9])) > 0)
1651                   conn->timeout = to;
1652               }
1653               break;
1654             }
1655           }
1656
1657           /* make sure to not overflow */
1658           strncpy (conn->session_id, session_id, maxlen);
1659           conn->session_id[maxlen] = '\0';
1660         }
1661         res = GST_RTSP_OK;
1662         goto done;
1663       }
1664       default:
1665         res = GST_RTSP_ERROR;
1666         break;
1667     }
1668   }
1669 done:
1670   return res;
1671 }
1672
1673 /**
1674  * gst_rtsp_connection_read:
1675  * @conn: a #GstRTSPConnection
1676  * @data: the data to read
1677  * @size: the size of @data
1678  * @timeout: a timeout value or #NULL
1679  *
1680  * Attempt to read @size bytes into @data from the connected @conn, blocking up to
1681  * the specified @timeout. @timeout can be #NULL, in which case this function
1682  * might block forever.
1683  *
1684  * This function can be cancelled with gst_rtsp_connection_flush().
1685  *
1686  * Returns: #GST_RTSP_OK on success.
1687  */
1688 GstRTSPResult
1689 gst_rtsp_connection_read (GstRTSPConnection * conn, guint8 * data, guint size,
1690     GTimeVal * timeout)
1691 {
1692   guint offset;
1693   gint retval;
1694   GstClockTime to;
1695   GstRTSPResult res;
1696
1697   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1698   g_return_val_if_fail (data != NULL, GST_RTSP_EINVAL);
1699   g_return_val_if_fail (conn->readfd != NULL, GST_RTSP_EINVAL);
1700
1701   if (size == 0)
1702     return GST_RTSP_OK;
1703
1704   offset = 0;
1705
1706   /* configure timeout if any */
1707   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1708
1709   gst_poll_set_controllable (conn->fdset, TRUE);
1710   gst_poll_fd_ctl_write (conn->fdset, conn->writefd, FALSE);
1711   gst_poll_fd_ctl_read (conn->fdset, conn->readfd, TRUE);
1712
1713   while (TRUE) {
1714     res = read_bytes (conn->readfd->fd, data, &offset, size, conn->ctxp);
1715     if (res == GST_RTSP_EEOF)
1716       goto eof;
1717     if (res == GST_RTSP_OK)
1718       break;
1719     if (res != GST_RTSP_EINTR)
1720       goto read_error;
1721
1722     do {
1723       retval = gst_poll_wait (conn->fdset, to);
1724     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1725
1726     /* check for timeout */
1727     if (retval == 0)
1728       goto select_timeout;
1729
1730     if (retval == -1) {
1731       if (errno == EBUSY)
1732         goto stopped;
1733       else
1734         goto select_error;
1735     }
1736     gst_poll_set_controllable (conn->fdset, FALSE);
1737   }
1738   return GST_RTSP_OK;
1739
1740   /* ERRORS */
1741 select_error:
1742   {
1743     return GST_RTSP_ESYS;
1744   }
1745 select_timeout:
1746   {
1747     return GST_RTSP_ETIMEOUT;
1748   }
1749 stopped:
1750   {
1751     return GST_RTSP_EINTR;
1752   }
1753 eof:
1754   {
1755     return GST_RTSP_EEOF;
1756   }
1757 read_error:
1758   {
1759     return res;
1760   }
1761 }
1762
1763 static GString *
1764 gen_tunnel_reply (GstRTSPConnection * conn, GstRTSPStatusCode code)
1765 {
1766   GString *str;
1767   gchar date_string[100];
1768   const gchar *status;
1769
1770   gen_date_string (date_string, sizeof (date_string));
1771
1772   status = gst_rtsp_status_as_text (code);
1773   if (status == NULL) {
1774     code = 500;
1775     status = "Internal Server Error";
1776   }
1777
1778   str = g_string_new ("");
1779
1780   /* */
1781   g_string_append_printf (str, "HTTP/1.0 %d %s\r\n", code, status);
1782   g_string_append_printf (str,
1783       "Server: GStreamer RTSP Server\r\n"
1784       "Date: %s\r\n"
1785       "Connection: close\r\n"
1786       "Cache-Control: no-store\r\n" "Pragma: no-cache\r\n", date_string);
1787   if (code == 200) {
1788     if (conn->ip)
1789       g_string_append_printf (str, "x-server-ip-address: %s\r\n", conn->ip);
1790     g_string_append_printf (str,
1791         "Content-Type: application/x-rtsp-tunnelled\r\n" "\r\n");
1792   }
1793   return str;
1794 }
1795
1796 /**
1797  * gst_rtsp_connection_receive:
1798  * @conn: a #GstRTSPConnection
1799  * @message: the message to read
1800  * @timeout: a timeout value or #NULL
1801  *
1802  * Attempt to read into @message from the connected @conn, blocking up to
1803  * the specified @timeout. @timeout can be #NULL, in which case this function
1804  * might block forever.
1805  * 
1806  * This function can be cancelled with gst_rtsp_connection_flush().
1807  *
1808  * Returns: #GST_RTSP_OK on success.
1809  */
1810 GstRTSPResult
1811 gst_rtsp_connection_receive (GstRTSPConnection * conn, GstRTSPMessage * message,
1812     GTimeVal * timeout)
1813 {
1814   GstRTSPResult res;
1815   GstRTSPBuilder builder = { 0 };
1816   gint retval;
1817   GstClockTime to;
1818
1819   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1820   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
1821   g_return_val_if_fail (conn->readfd != NULL, GST_RTSP_EINVAL);
1822
1823   /* configure timeout if any */
1824   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
1825
1826   gst_poll_set_controllable (conn->fdset, TRUE);
1827   gst_poll_fd_ctl_write (conn->fdset, conn->writefd, FALSE);
1828   gst_poll_fd_ctl_read (conn->fdset, conn->readfd, TRUE);
1829
1830   while (TRUE) {
1831     res = build_next (&builder, message, conn);
1832     if (res == GST_RTSP_EEOF)
1833       goto eof;
1834     if (res == GST_RTSP_OK)
1835       break;
1836     if (res == GST_RTSP_ETGET) {
1837       GString *str;
1838
1839       /* tunnel GET request, we can reply now */
1840       str = gen_tunnel_reply (conn, GST_RTSP_STS_OK);
1841       res =
1842           gst_rtsp_connection_write (conn, (guint8 *) str->str, str->len,
1843           timeout);
1844       g_string_free (str, TRUE);
1845     } else if (res == GST_RTSP_ETPOST) {
1846       /* tunnel POST request, return the value, the caller now has to link the
1847        * two connections. */
1848       break;
1849     } else if (res != GST_RTSP_EINTR)
1850       goto read_error;
1851
1852     do {
1853       retval = gst_poll_wait (conn->fdset, to);
1854     } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
1855
1856     /* check for timeout */
1857     if (retval == 0)
1858       goto select_timeout;
1859
1860     if (retval == -1) {
1861       if (errno == EBUSY)
1862         goto stopped;
1863       else
1864         goto select_error;
1865     }
1866     gst_poll_set_controllable (conn->fdset, FALSE);
1867   }
1868
1869   /* we have a message here */
1870   build_reset (&builder);
1871
1872   return GST_RTSP_OK;
1873
1874   /* ERRORS */
1875 select_error:
1876   {
1877     res = GST_RTSP_ESYS;
1878     goto cleanup;
1879   }
1880 select_timeout:
1881   {
1882     res = GST_RTSP_ETIMEOUT;
1883     goto cleanup;
1884   }
1885 stopped:
1886   {
1887     res = GST_RTSP_EINTR;
1888     goto cleanup;
1889   }
1890 eof:
1891   {
1892     res = GST_RTSP_EEOF;
1893     goto cleanup;
1894   }
1895 read_error:
1896 cleanup:
1897   {
1898     build_reset (&builder);
1899     gst_rtsp_message_unset (message);
1900     return res;
1901   }
1902 }
1903
1904 /**
1905  * gst_rtsp_connection_close:
1906  * @conn: a #GstRTSPConnection
1907  *
1908  * Close the connected @conn.
1909  * 
1910  * Returns: #GST_RTSP_OK on success.
1911  */
1912 GstRTSPResult
1913 gst_rtsp_connection_close (GstRTSPConnection * conn)
1914 {
1915   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1916
1917   g_free (conn->ip);
1918   conn->ip = NULL;
1919
1920   REMOVE_POLLFD (conn->fdset, &conn->fd0);
1921   REMOVE_POLLFD (conn->fdset, &conn->fd1);
1922   conn->writefd = NULL;
1923   conn->readfd = NULL;
1924
1925   return GST_RTSP_OK;
1926 }
1927
1928 /**
1929  * gst_rtsp_connection_free:
1930  * @conn: a #GstRTSPConnection
1931  *
1932  * Close and free @conn.
1933  * 
1934  * Returns: #GST_RTSP_OK on success.
1935  */
1936 GstRTSPResult
1937 gst_rtsp_connection_free (GstRTSPConnection * conn)
1938 {
1939   GstRTSPResult res;
1940
1941   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1942
1943   res = gst_rtsp_connection_close (conn);
1944   gst_poll_free (conn->fdset);
1945   g_timer_destroy (conn->timer);
1946   g_free (conn->username);
1947   g_free (conn->passwd);
1948   gst_rtsp_connection_clear_auth_params (conn);
1949   g_free (conn);
1950 #ifdef G_OS_WIN32
1951   WSACleanup ();
1952 #endif
1953
1954   return res;
1955 }
1956
1957 /**
1958  * gst_rtsp_connection_poll:
1959  * @conn: a #GstRTSPConnection
1960  * @events: a bitmask of #GstRTSPEvent flags to check
1961  * @revents: location for result flags 
1962  * @timeout: a timeout
1963  *
1964  * Wait up to the specified @timeout for the connection to become available for
1965  * at least one of the operations specified in @events. When the function returns
1966  * with #GST_RTSP_OK, @revents will contain a bitmask of available operations on
1967  * @conn.
1968  *
1969  * @timeout can be #NULL, in which case this function might block forever.
1970  *
1971  * This function can be cancelled with gst_rtsp_connection_flush().
1972  * 
1973  * Returns: #GST_RTSP_OK on success.
1974  *
1975  * Since: 0.10.15
1976  */
1977 GstRTSPResult
1978 gst_rtsp_connection_poll (GstRTSPConnection * conn, GstRTSPEvent events,
1979     GstRTSPEvent * revents, GTimeVal * timeout)
1980 {
1981   GstClockTime to;
1982   gint retval;
1983
1984   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
1985   g_return_val_if_fail (events != 0, GST_RTSP_EINVAL);
1986   g_return_val_if_fail (revents != NULL, GST_RTSP_EINVAL);
1987   g_return_val_if_fail (conn->readfd != NULL, GST_RTSP_EINVAL);
1988   g_return_val_if_fail (conn->writefd != NULL, GST_RTSP_EINVAL);
1989
1990   gst_poll_set_controllable (conn->fdset, TRUE);
1991
1992   /* add fd to writer set when asked to */
1993   gst_poll_fd_ctl_write (conn->fdset, conn->writefd,
1994       events & GST_RTSP_EV_WRITE);
1995
1996   /* add fd to reader set when asked to */
1997   gst_poll_fd_ctl_read (conn->fdset, conn->readfd, events & GST_RTSP_EV_READ);
1998
1999   /* configure timeout if any */
2000   to = timeout ? GST_TIMEVAL_TO_TIME (*timeout) : GST_CLOCK_TIME_NONE;
2001
2002   do {
2003     retval = gst_poll_wait (conn->fdset, to);
2004   } while (retval == -1 && (errno == EINTR || errno == EAGAIN));
2005
2006   if (retval == 0)
2007     goto select_timeout;
2008
2009   if (retval == -1) {
2010     if (errno == EBUSY)
2011       goto stopped;
2012     else
2013       goto select_error;
2014   }
2015
2016   *revents = 0;
2017   if (events & GST_RTSP_EV_READ) {
2018     if (gst_poll_fd_can_read (conn->fdset, conn->readfd))
2019       *revents |= GST_RTSP_EV_READ;
2020   }
2021   if (events & GST_RTSP_EV_WRITE) {
2022     if (gst_poll_fd_can_write (conn->fdset, conn->writefd))
2023       *revents |= GST_RTSP_EV_WRITE;
2024   }
2025   return GST_RTSP_OK;
2026
2027   /* ERRORS */
2028 select_timeout:
2029   {
2030     return GST_RTSP_ETIMEOUT;
2031   }
2032 select_error:
2033   {
2034     return GST_RTSP_ESYS;
2035   }
2036 stopped:
2037   {
2038     return GST_RTSP_EINTR;
2039   }
2040 }
2041
2042 /**
2043  * gst_rtsp_connection_next_timeout:
2044  * @conn: a #GstRTSPConnection
2045  * @timeout: a timeout
2046  *
2047  * Calculate the next timeout for @conn, storing the result in @timeout.
2048  * 
2049  * Returns: #GST_RTSP_OK.
2050  */
2051 GstRTSPResult
2052 gst_rtsp_connection_next_timeout (GstRTSPConnection * conn, GTimeVal * timeout)
2053 {
2054   gdouble elapsed;
2055   glong sec;
2056   gulong usec;
2057
2058   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2059   g_return_val_if_fail (timeout != NULL, GST_RTSP_EINVAL);
2060
2061   elapsed = g_timer_elapsed (conn->timer, &usec);
2062   if (elapsed >= conn->timeout) {
2063     sec = 0;
2064     usec = 0;
2065   } else {
2066     sec = conn->timeout - elapsed;
2067   }
2068
2069   timeout->tv_sec = sec;
2070   timeout->tv_usec = usec;
2071
2072   return GST_RTSP_OK;
2073 }
2074
2075 /**
2076  * gst_rtsp_connection_reset_timeout:
2077  * @conn: a #GstRTSPConnection
2078  *
2079  * Reset the timeout of @conn.
2080  * 
2081  * Returns: #GST_RTSP_OK.
2082  */
2083 GstRTSPResult
2084 gst_rtsp_connection_reset_timeout (GstRTSPConnection * conn)
2085 {
2086   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2087
2088   g_timer_start (conn->timer);
2089
2090   return GST_RTSP_OK;
2091 }
2092
2093 /**
2094  * gst_rtsp_connection_flush:
2095  * @conn: a #GstRTSPConnection
2096  * @flush: start or stop the flush
2097  *
2098  * Start or stop the flushing action on @conn. When flushing, all current
2099  * and future actions on @conn will return #GST_RTSP_EINTR until the connection
2100  * is set to non-flushing mode again.
2101  * 
2102  * Returns: #GST_RTSP_OK.
2103  */
2104 GstRTSPResult
2105 gst_rtsp_connection_flush (GstRTSPConnection * conn, gboolean flush)
2106 {
2107   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2108
2109   gst_poll_set_flushing (conn->fdset, flush);
2110
2111   return GST_RTSP_OK;
2112 }
2113
2114 /**
2115  * gst_rtsp_connection_set_auth:
2116  * @conn: a #GstRTSPConnection
2117  * @method: authentication method
2118  * @user: the user
2119  * @pass: the password
2120  *
2121  * Configure @conn for authentication mode @method with @user and @pass as the
2122  * user and password respectively.
2123  * 
2124  * Returns: #GST_RTSP_OK.
2125  */
2126 GstRTSPResult
2127 gst_rtsp_connection_set_auth (GstRTSPConnection * conn,
2128     GstRTSPAuthMethod method, const gchar * user, const gchar * pass)
2129 {
2130   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2131
2132   if (method == GST_RTSP_AUTH_DIGEST && ((user == NULL || pass == NULL)
2133           || g_strrstr (user, ":") != NULL))
2134     return GST_RTSP_EINVAL;
2135
2136   /* Make sure the username and passwd are being set for authentication */
2137   if (method == GST_RTSP_AUTH_NONE && (user == NULL || pass == NULL))
2138     return GST_RTSP_EINVAL;
2139
2140   /* ":" chars are not allowed in usernames for basic auth */
2141   if (method == GST_RTSP_AUTH_BASIC && g_strrstr (user, ":") != NULL)
2142     return GST_RTSP_EINVAL;
2143
2144   g_free (conn->username);
2145   g_free (conn->passwd);
2146
2147   conn->auth_method = method;
2148   conn->username = g_strdup (user);
2149   conn->passwd = g_strdup (pass);
2150
2151   return GST_RTSP_OK;
2152 }
2153
2154 /**
2155  * str_case_hash:
2156  * @key: ASCII string to hash
2157  *
2158  * Hashes @key in a case-insensitive manner.
2159  *
2160  * Returns: the hash code.
2161  **/
2162 static guint
2163 str_case_hash (gconstpointer key)
2164 {
2165   const char *p = key;
2166   guint h = g_ascii_toupper (*p);
2167
2168   if (h)
2169     for (p += 1; *p != '\0'; p++)
2170       h = (h << 5) - h + g_ascii_toupper (*p);
2171
2172   return h;
2173 }
2174
2175 /**
2176  * str_case_equal:
2177  * @v1: an ASCII string
2178  * @v2: another ASCII string
2179  *
2180  * Compares @v1 and @v2 in a case-insensitive manner
2181  *
2182  * Returns: %TRUE if they are equal (modulo case)
2183  **/
2184 static gboolean
2185 str_case_equal (gconstpointer v1, gconstpointer v2)
2186 {
2187   const char *string1 = v1;
2188   const char *string2 = v2;
2189
2190   return g_ascii_strcasecmp (string1, string2) == 0;
2191 }
2192
2193 /**
2194  * gst_rtsp_connection_set_auth_param:
2195  * @conn: a #GstRTSPConnection
2196  * @param: authentication directive
2197  * @value: value
2198  *
2199  * Setup @conn with authentication directives. This is not necesary for
2200  * methods #GST_RTSP_AUTH_NONE and #GST_RTSP_AUTH_BASIC. For
2201  * #GST_RTSP_AUTH_DIGEST, directives should be taken from the digest challenge
2202  * in the WWW-Authenticate response header and can include realm, domain,
2203  * nonce, opaque, stale, algorithm, qop as per RFC2617.
2204  * 
2205  * Since: 0.10.20
2206  */
2207 void
2208 gst_rtsp_connection_set_auth_param (GstRTSPConnection * conn,
2209     const gchar * param, const gchar * value)
2210 {
2211   g_return_if_fail (conn != NULL);
2212   g_return_if_fail (param != NULL);
2213
2214   if (conn->auth_params == NULL) {
2215     conn->auth_params =
2216         g_hash_table_new_full (str_case_hash, str_case_equal, g_free, g_free);
2217   }
2218   g_hash_table_insert (conn->auth_params, g_strdup (param), g_strdup (value));
2219 }
2220
2221 /**
2222  * gst_rtsp_connection_clear_auth_params:
2223  * @conn: a #GstRTSPConnection
2224  *
2225  * Clear the list of authentication directives stored in @conn.
2226  *
2227  * Since: 0.10.20
2228  */
2229 void
2230 gst_rtsp_connection_clear_auth_params (GstRTSPConnection * conn)
2231 {
2232   g_return_if_fail (conn != NULL);
2233
2234   if (conn->auth_params != NULL) {
2235     g_hash_table_destroy (conn->auth_params);
2236     conn->auth_params = NULL;
2237   }
2238 }
2239
2240 static GstRTSPResult
2241 set_qos_dscp (gint fd, guint qos_dscp)
2242 {
2243   union gst_sockaddr
2244   {
2245     struct sockaddr sa;
2246     struct sockaddr_in6 sa_in6;
2247     struct sockaddr_storage sa_stor;
2248   } sa;
2249   socklen_t slen = sizeof (sa);
2250   gint af;
2251   gint tos;
2252
2253   if (fd == -1)
2254     return GST_RTSP_OK;
2255
2256   if (getsockname (fd, &sa.sa, &slen) < 0)
2257     goto no_getsockname;
2258
2259   af = sa.sa.sa_family;
2260
2261   /* if this is an IPv4-mapped address then do IPv4 QoS */
2262   if (af == AF_INET6) {
2263     if (IN6_IS_ADDR_V4MAPPED (&sa.sa_in6.sin6_addr))
2264       af = AF_INET;
2265   }
2266
2267   /* extract and shift 6 bits of the DSCP */
2268   tos = (qos_dscp & 0x3f) << 2;
2269
2270   switch (af) {
2271     case AF_INET:
2272       if (SETSOCKOPT (fd, IPPROTO_IP, IP_TOS, &tos, sizeof (tos)) < 0)
2273         goto no_setsockopt;
2274       break;
2275     case AF_INET6:
2276 #ifdef IPV6_TCLASS
2277       if (SETSOCKOPT (fd, IPPROTO_IPV6, IPV6_TCLASS, &tos, sizeof (tos)) < 0)
2278         goto no_setsockopt;
2279       break;
2280 #endif
2281     default:
2282       goto wrong_family;
2283   }
2284
2285   return GST_RTSP_OK;
2286
2287   /* ERRORS */
2288 no_getsockname:
2289 no_setsockopt:
2290   {
2291     return GST_RTSP_ESYS;
2292   }
2293
2294 wrong_family:
2295   {
2296     return GST_RTSP_ERROR;
2297   }
2298 }
2299
2300 /**
2301  * gst_rtsp_connection_set_qos_dscp:
2302  * @conn: a #GstRTSPConnection
2303  * @qos_dscp: DSCP value
2304  *
2305  * Configure @conn to use the specified DSCP value.
2306  *
2307  * Returns: #GST_RTSP_OK on success.
2308  *
2309  * Since: 0.10.20
2310  */
2311 GstRTSPResult
2312 gst_rtsp_connection_set_qos_dscp (GstRTSPConnection * conn, guint qos_dscp)
2313 {
2314   GstRTSPResult res;
2315
2316   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2317   g_return_val_if_fail (conn->readfd != NULL, GST_RTSP_EINVAL);
2318   g_return_val_if_fail (conn->writefd != NULL, GST_RTSP_EINVAL);
2319
2320   res = set_qos_dscp (conn->fd0.fd, qos_dscp);
2321   if (res == GST_RTSP_OK)
2322     res = set_qos_dscp (conn->fd1.fd, qos_dscp);
2323
2324   return res;
2325 }
2326
2327
2328 /**
2329  * gst_rtsp_connection_get_url:
2330  * @conn: a #GstRTSPConnection
2331  *
2332  * Retrieve the URL of the other end of @conn.
2333  *
2334  * Returns: The URL. This value remains valid until the
2335  * connection is closed.
2336  *
2337  * Since: 0.10.23
2338  */
2339 GstRTSPUrl *
2340 gst_rtsp_connection_get_url (const GstRTSPConnection * conn)
2341 {
2342   g_return_val_if_fail (conn != NULL, NULL);
2343
2344   return conn->url;
2345 }
2346
2347 /**
2348  * gst_rtsp_connection_get_ip:
2349  * @conn: a #GstRTSPConnection
2350  *
2351  * Retrieve the IP address of the other end of @conn.
2352  *
2353  * Returns: The IP address as a string. this value remains valid until the
2354  * connection is closed.
2355  *
2356  * Since: 0.10.20
2357  */
2358 const gchar *
2359 gst_rtsp_connection_get_ip (const GstRTSPConnection * conn)
2360 {
2361   g_return_val_if_fail (conn != NULL, NULL);
2362
2363   return conn->ip;
2364 }
2365
2366 /**
2367  * gst_rtsp_connection_set_ip:
2368  * @conn: a #GstRTSPConnection
2369  * @ip: an ip address
2370  *
2371  * Set the IP address of the server.
2372  *
2373  * Since: 0.10.23
2374  */
2375 void
2376 gst_rtsp_connection_set_ip (GstRTSPConnection * conn, const gchar * ip)
2377 {
2378   g_return_if_fail (conn != NULL);
2379
2380   g_free (conn->ip);
2381   conn->ip = g_strdup (ip);
2382 }
2383
2384 /**
2385  * gst_rtsp_connection_set_tunneled:
2386  * @conn: a #GstRTSPConnection
2387  * @tunneled: the new state
2388  *
2389  * Set the HTTP tunneling state of the connection. This must be configured before
2390  * the @conn is connected.
2391  *
2392  * Since: 0.10.23
2393  */
2394 void
2395 gst_rtsp_connection_set_tunneled (GstRTSPConnection * conn, gboolean tunneled)
2396 {
2397   g_return_if_fail (conn != NULL);
2398   g_return_if_fail (conn->readfd == NULL);
2399   g_return_if_fail (conn->writefd == NULL);
2400
2401   conn->tunneled = tunneled;
2402 }
2403
2404 /**
2405  * gst_rtsp_connection_is_tunneled:
2406  * @conn: a #GstRTSPConnection
2407  *
2408  * Get the tunneling state of the connection. 
2409  *
2410  * Returns: if @conn is using HTTP tunneling.
2411  *
2412  * Since: 0.10.23
2413  */
2414 gboolean
2415 gst_rtsp_connection_is_tunneled (const GstRTSPConnection * conn)
2416 {
2417   g_return_val_if_fail (conn != NULL, FALSE);
2418
2419   return conn->tunneled;
2420 }
2421
2422 /**
2423  * gst_rtsp_connection_get_tunnelid:
2424  * @conn: a #GstRTSPConnection
2425  *
2426  * Get the tunnel session id the connection. 
2427  *
2428  * Returns: returns a non-empty string if @conn is being tunneled over HTTP.
2429  *
2430  * Since: 0.10.23
2431  */
2432 const gchar *
2433 gst_rtsp_connection_get_tunnelid (const GstRTSPConnection * conn)
2434 {
2435   g_return_val_if_fail (conn != NULL, NULL);
2436
2437   if (!conn->tunneled)
2438     return NULL;
2439
2440   return conn->tunnelid;
2441 }
2442
2443 /**
2444  * gst_rtsp_connection_do_tunnel:
2445  * @conn: a #GstRTSPConnection
2446  * @conn2: a #GstRTSPConnection
2447  *
2448  * If @conn received the first tunnel connection and @conn2 received
2449  * the second tunnel connection, link the two connections together so that
2450  * @conn manages the tunneled connection.
2451  *
2452  * After this call, @conn2 cannot be used anymore and must be freed with
2453  * gst_rtsp_connection_free().
2454  *
2455  * Returns: return GST_RTSP_OK on success.
2456  *
2457  * Since: 0.10.23
2458  */
2459 GstRTSPResult
2460 gst_rtsp_connection_do_tunnel (GstRTSPConnection * conn,
2461     GstRTSPConnection * conn2)
2462 {
2463   g_return_val_if_fail (conn != NULL, GST_RTSP_EINVAL);
2464   g_return_val_if_fail (conn2 != NULL, GST_RTSP_EINVAL);
2465   g_return_val_if_fail (conn->tstate == TUNNEL_STATE_GET, GST_RTSP_EINVAL);
2466   g_return_val_if_fail (conn2->tstate == TUNNEL_STATE_POST, GST_RTSP_EINVAL);
2467   g_return_val_if_fail (!memcmp (conn2->tunnelid, conn->tunnelid, TUNNELID_LEN),
2468       GST_RTSP_EINVAL);
2469
2470   /* both connections have fd0 as the read/write socket. start by taking the
2471    * socket from conn2 and set it as the socket in conn */
2472   conn->fd1 = conn2->fd0;
2473
2474   /* clean up some of the state of conn2 */
2475   gst_poll_remove_fd (conn2->fdset, &conn2->fd0);
2476   conn2->fd0.fd = -1;
2477   conn2->readfd = conn2->writefd = NULL;
2478
2479   /* We make fd0 the write socket and fd1 the read socket. */
2480   conn->writefd = &conn->fd0;
2481   conn->readfd = &conn->fd1;
2482
2483   conn->tstate = TUNNEL_STATE_COMPLETE;
2484
2485   /* we need base64 decoding for the readfd */
2486   conn->ctx.state = 0;
2487   conn->ctx.cin = 0;
2488   conn->ctx.cout = 3;
2489   conn->ctx.save = 0;
2490   conn->ctxp = &conn->ctx;
2491
2492   return GST_RTSP_OK;
2493 }
2494
2495 #define READ_COND   (G_IO_IN | G_IO_HUP | G_IO_ERR)
2496 #define WRITE_COND  (G_IO_OUT | G_IO_ERR)
2497
2498 typedef struct
2499 {
2500   GString *str;
2501   guint cseq;
2502 } GstRTSPRec;
2503
2504 static GstRTSPRec *queue_response (GstRTSPWatch * watch, GString * str,
2505     guint cseq);
2506
2507 /* async functions */
2508 struct _GstRTSPWatch
2509 {
2510   GSource source;
2511
2512   GstRTSPConnection *conn;
2513
2514   GstRTSPBuilder builder;
2515   GstRTSPMessage message;
2516
2517   GPollFD readfd;
2518   GPollFD writefd;
2519   gboolean write_added;
2520
2521   /* queued message for transmission */
2522   GList *messages;
2523   guint8 *write_data;
2524   guint write_off;
2525   guint write_len;
2526   guint write_cseq;
2527
2528   GstRTSPWatchFuncs funcs;
2529
2530   gpointer user_data;
2531   GDestroyNotify notify;
2532 };
2533
2534 static gboolean
2535 gst_rtsp_source_prepare (GSource * source, gint * timeout)
2536 {
2537   GstRTSPWatch *watch = (GstRTSPWatch *) source;
2538
2539   *timeout = (watch->conn->timeout * 1000);
2540
2541   return FALSE;
2542 }
2543
2544 static gboolean
2545 gst_rtsp_source_check (GSource * source)
2546 {
2547   GstRTSPWatch *watch = (GstRTSPWatch *) source;
2548
2549   if (watch->readfd.revents & READ_COND)
2550     return TRUE;
2551
2552   if (watch->writefd.revents & WRITE_COND)
2553     return TRUE;
2554
2555   return FALSE;
2556 }
2557
2558 static gboolean
2559 gst_rtsp_source_dispatch (GSource * source, GSourceFunc callback,
2560     gpointer user_data)
2561 {
2562   GstRTSPWatch *watch = (GstRTSPWatch *) source;
2563   GstRTSPResult res;
2564
2565   /* first read as much as we can */
2566   if (watch->readfd.revents & READ_COND) {
2567     do {
2568       res = build_next (&watch->builder, &watch->message, watch->conn);
2569       if (res == GST_RTSP_EINTR)
2570         break;
2571       if (res == GST_RTSP_EEOF)
2572         goto eof;
2573       if (res == GST_RTSP_ETGET) {
2574         GString *str;
2575         GstRTSPStatusCode code;
2576
2577         if (watch->funcs.tunnel_start)
2578           code = watch->funcs.tunnel_start (watch, watch->user_data);
2579         else
2580           code = GST_RTSP_STS_OK;
2581
2582         /* queue the response string */
2583         str = gen_tunnel_reply (watch->conn, code);
2584         queue_response (watch, str, -1);
2585       } else if (res == GST_RTSP_ETPOST) {
2586         /* in the callback the connection should be tunneled with the
2587          * GET connection */
2588         if (watch->funcs.tunnel_complete)
2589           watch->funcs.tunnel_complete (watch, watch->user_data);
2590       } else if (res != GST_RTSP_OK)
2591         goto error;
2592
2593       if (res == GST_RTSP_OK) {
2594         if (watch->funcs.message_received)
2595           watch->funcs.message_received (watch, &watch->message,
2596               watch->user_data);
2597
2598         gst_rtsp_message_unset (&watch->message);
2599       }
2600       build_reset (&watch->builder);
2601     } while (FALSE);
2602   }
2603
2604   if (watch->writefd.revents & WRITE_COND) {
2605     do {
2606       if (watch->write_data == NULL) {
2607         GstRTSPRec *data;
2608
2609         if (!watch->messages)
2610           goto done;
2611
2612         /* no data, get a new message from the queue */
2613         data = watch->messages->data;
2614         watch->messages = g_list_delete_link (watch->messages, watch->messages);
2615
2616         watch->write_off = 0;
2617         watch->write_len = data->str->len;
2618         watch->write_data = (guint8 *) g_string_free (data->str, FALSE);
2619         watch->write_cseq = data->cseq;
2620
2621         g_slice_free (GstRTSPRec, data);
2622       }
2623
2624       res = write_bytes (watch->writefd.fd, watch->write_data,
2625           &watch->write_off, watch->write_len);
2626       if (res == GST_RTSP_EINTR)
2627         break;
2628       if (res != GST_RTSP_OK)
2629         goto error;
2630
2631       if (watch->funcs.message_sent && watch->write_cseq != -1)
2632         watch->funcs.message_sent (watch, watch->write_cseq, watch->user_data);
2633
2634     done:
2635       if (watch->messages == NULL && watch->write_added) {
2636         g_source_remove_poll ((GSource *) watch, &watch->writefd);
2637         watch->write_added = FALSE;
2638         watch->writefd.revents = 0;
2639       }
2640       g_free (watch->write_data);
2641       watch->write_data = NULL;
2642     } while (FALSE);
2643   }
2644
2645   return TRUE;
2646
2647   /* ERRORS */
2648 eof:
2649   {
2650     if (watch->funcs.closed)
2651       watch->funcs.closed (watch, watch->user_data);
2652     return FALSE;
2653   }
2654 error:
2655   {
2656     if (watch->funcs.error)
2657       watch->funcs.error (watch, res, watch->user_data);
2658     return FALSE;
2659   }
2660 }
2661
2662 static void
2663 gst_rtsp_source_finalize (GSource * source)
2664 {
2665   GstRTSPWatch *watch = (GstRTSPWatch *) source;
2666   GList *walk;
2667
2668   build_reset (&watch->builder);
2669
2670   for (walk = watch->messages; walk; walk = g_list_next (walk)) {
2671     GstRTSPRec *data = walk->data;
2672
2673     g_string_free (data->str, TRUE);
2674     g_slice_free (GstRTSPRec, data);
2675   }
2676   g_list_free (watch->messages);
2677   g_free (watch->write_data);
2678
2679   if (watch->notify)
2680     watch->notify (watch->user_data);
2681 }
2682
2683 static GSourceFuncs gst_rtsp_source_funcs = {
2684   gst_rtsp_source_prepare,
2685   gst_rtsp_source_check,
2686   gst_rtsp_source_dispatch,
2687   gst_rtsp_source_finalize
2688 };
2689
2690 /**
2691  * gst_rtsp_watch_new:
2692  * @conn: a #GstRTSPConnection
2693  * @funcs: watch functions
2694  * @user_data: user data to pass to @funcs
2695  * @notify: notify when @user_data is not referenced anymore
2696  *
2697  * Create a watch object for @conn. The functions provided in @funcs will be
2698  * called with @user_data when activity happened on the watch.
2699  *
2700  * The new watch is usually created so that it can be attached to a
2701  * maincontext with gst_rtsp_watch_attach(). 
2702  *
2703  * @conn must exist for the entire lifetime of the watch.
2704  *
2705  * Returns: a #GstRTSPWatch that can be used for asynchronous RTSP
2706  * communication. Free with gst_rtsp_watch_unref () after usage.
2707  *
2708  * Since: 0.10.23
2709  */
2710 GstRTSPWatch *
2711 gst_rtsp_watch_new (GstRTSPConnection * conn,
2712     GstRTSPWatchFuncs * funcs, gpointer user_data, GDestroyNotify notify)
2713 {
2714   GstRTSPWatch *result;
2715
2716   g_return_val_if_fail (conn != NULL, NULL);
2717   g_return_val_if_fail (funcs != NULL, NULL);
2718   g_return_val_if_fail (conn->readfd != NULL, NULL);
2719   g_return_val_if_fail (conn->writefd != NULL, NULL);
2720
2721   result = (GstRTSPWatch *) g_source_new (&gst_rtsp_source_funcs,
2722       sizeof (GstRTSPWatch));
2723
2724   result->conn = conn;
2725   result->builder.state = STATE_START;
2726
2727   result->readfd.fd = -1;
2728   result->writefd.fd = -1;
2729
2730   gst_rtsp_watch_reset (result);
2731
2732   result->funcs = *funcs;
2733   result->user_data = user_data;
2734   result->notify = notify;
2735
2736   /* only add the read fd, the write fd is only added when we have data
2737    * to send. */
2738   g_source_add_poll ((GSource *) result, &result->readfd);
2739
2740   return result;
2741 }
2742
2743 /**
2744  * gst_rtsp_watch_reset:
2745  * @watch: a #GstRTSPWatch
2746  *
2747  * Reset @watch, this is usually called after gst_rtsp_connection_do_tunnel()
2748  * when the file descriptors of the connection might have changed.
2749  *
2750  * Since: 0.10.23
2751  */
2752 void
2753 gst_rtsp_watch_reset (GstRTSPWatch * watch)
2754 {
2755   if (watch->readfd.fd != -1)
2756     g_source_remove_poll ((GSource *) watch, &watch->readfd);
2757   if (watch->writefd.fd != -1)
2758     g_source_remove_poll ((GSource *) watch, &watch->writefd);
2759
2760   watch->readfd.fd = watch->conn->readfd->fd;
2761   watch->readfd.events = READ_COND;
2762   watch->readfd.revents = 0;
2763
2764   watch->writefd.fd = watch->conn->writefd->fd;
2765   watch->writefd.events = WRITE_COND;
2766   watch->writefd.revents = 0;
2767   watch->write_added = FALSE;
2768
2769   g_source_add_poll ((GSource *) watch, &watch->readfd);
2770 }
2771
2772 /**
2773  * gst_rtsp_watch_attach:
2774  * @watch: a #GstRTSPWatch
2775  * @context: a GMainContext (if NULL, the default context will be used)
2776  *
2777  * Adds a #GstRTSPWatch to a context so that it will be executed within that context.
2778  *
2779  * Returns: the ID (greater than 0) for the watch within the GMainContext. 
2780  *
2781  * Since: 0.10.23
2782  */
2783 guint
2784 gst_rtsp_watch_attach (GstRTSPWatch * watch, GMainContext * context)
2785 {
2786   g_return_val_if_fail (watch != NULL, 0);
2787
2788   return g_source_attach ((GSource *) watch, context);
2789 }
2790
2791 /**
2792  * gst_rtsp_watch_unref:
2793  * @watch: a #GstRTSPWatch
2794  *
2795  * Decreases the reference count of @watch by one. If the resulting reference
2796  * count is zero the watch and associated memory will be destroyed.
2797  *
2798  * Since: 0.10.23
2799  */
2800 void
2801 gst_rtsp_watch_unref (GstRTSPWatch * watch)
2802 {
2803   g_return_if_fail (watch != NULL);
2804
2805   g_source_unref ((GSource *) watch);
2806 }
2807
2808 static GstRTSPRec *
2809 queue_response (GstRTSPWatch * watch, GString * str, guint cseq)
2810 {
2811   GstRTSPRec *data;
2812
2813   /* make a record with the message as a string ans cseq */
2814   data = g_slice_new (GstRTSPRec);
2815   data->str = str;
2816   data->cseq = cseq;
2817
2818   /* add the record to a queue */
2819   watch->messages = g_list_append (watch->messages, data);
2820
2821   /* make sure the main context will now also check for writability on the
2822    * socket */
2823   if (!watch->write_added) {
2824     g_source_add_poll ((GSource *) watch, &watch->writefd);
2825     watch->write_added = TRUE;
2826   }
2827
2828   return data;
2829 }
2830
2831 /**
2832  * gst_rtsp_watch_queue_message:
2833  * @watch: a #GstRTSPWatch
2834  * @message: a #GstRTSPMessage
2835  *
2836  * Queue a @message for transmission in @watch. The contents of this 
2837  * message will be serialized and transmitted when the connection of the
2838  * watch becomes writable.
2839  *
2840  * The return value of this function will be returned as the cseq argument in
2841  * the message_sent callback.
2842  *
2843  * Returns: the sequence number of the message or -1 if the cseq could not be
2844  * determined.
2845  *
2846  * Since: 0.10.23
2847  */
2848 guint
2849 gst_rtsp_watch_queue_message (GstRTSPWatch * watch, GstRTSPMessage * message)
2850 {
2851   GstRTSPRec *data;
2852   gchar *header;
2853   guint cseq;
2854
2855   g_return_val_if_fail (watch != NULL, GST_RTSP_EINVAL);
2856   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
2857
2858   /* get the cseq from the message, when we finish writing this message on the
2859    * socket we will have to pass the cseq to the callback. */
2860   if (gst_rtsp_message_get_header (message, GST_RTSP_HDR_CSEQ, &header,
2861           0) == GST_RTSP_OK) {
2862     cseq = atoi (header);
2863   } else {
2864     cseq = -1;
2865   }
2866
2867   /* make a record with the message as a string ans cseq */
2868   data = queue_response (watch, message_to_string (watch->conn, message), cseq);
2869
2870   return cseq;
2871 }