]> git.sesse.net Git - vlc/blobdiff - modules/access/mms/mmsh.c
* modules/access/mms/mmsh.c: fixed changeset 23489. Please compile before you submit.
[vlc] / modules / access / mms / mmsh.c
index 18d1c73bf0637ed82021420fb7dca67afeb3602d..d32758a005af71d54c58e128b14225c51bc89c4a 100644 (file)
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
-#include <stdlib.h>
 
 #include <vlc/vlc.h>
 #include <vlc_access.h>
 #include "vlc_playlist.h"
+#include "vlc_strings.h"
 
 #include <vlc_network.h>
 #include "vlc_url.h"
@@ -39,7 +39,6 @@
 #include "mmsh.h"
 
 /* TODO:
- *  - http_proxy
  *  - authentication
  */
 
@@ -74,6 +73,7 @@ int E_(MMSHOpen)( access_t *p_access )
 {
     access_sys_t    *p_sys;
     char            *psz_location = NULL;
+    char            *psz_proxy;
 
     /* init p_sys */
 
@@ -88,17 +88,80 @@ int E_(MMSHOpen)( access_t *p_access )
     p_access->info.b_eof = VLC_FALSE;
     p_access->info.i_title = 0;
     p_access->info.i_seekpoint = 0;
+
     p_access->p_sys = p_sys = malloc( sizeof( access_sys_t ) );
+    if( !p_sys )
+        return VLC_ENOMEM;
+
     memset( p_sys, 0, sizeof( access_sys_t ) );
     p_sys->i_proto= MMS_PROTO_HTTP;
     p_sys->fd     = -1;
     p_sys->i_start= 0;
 
+    /* Handle proxy */
+    p_sys->b_proxy = VLC_FALSE;
+    memset( &p_sys->proxy, 0, sizeof(p_sys->proxy) );
+
+    /* Check proxy */
+    /* TODO reuse instead http-proxy from http access ? */
+    psz_proxy = var_CreateGetString( p_access, "mmsh-proxy" );
+    if( !*psz_proxy )
+    {
+        char *psz_http_proxy = config_GetPsz( p_access, "http-proxy" );
+        if( psz_http_proxy && *psz_http_proxy )
+        {
+            free( psz_proxy );
+            psz_proxy = psz_http_proxy;
+            var_SetString( p_access, "mmsh-proxy", psz_proxy );
+        }
+        else
+        {
+            free( psz_http_proxy );
+        }
+    }
+
+    if( *psz_proxy )
+    {
+        p_sys->b_proxy = VLC_TRUE;
+        vlc_UrlParse( &p_sys->proxy, psz_proxy, 0 );
+    }
+#ifdef HAVE_GETENV
+    else
+    {
+        char *psz_proxy = getenv( "http_proxy" );
+        if( psz_proxy && *psz_proxy )
+        {
+            p_sys->b_proxy = VLC_TRUE;
+            vlc_UrlParse( &p_sys->proxy, psz_proxy, 0 );
+        }
+    }
+#endif
+    free( psz_proxy );
+
+    if( p_sys->b_proxy )
+    {
+        if( ( p_sys->proxy.psz_host == NULL ) ||
+            ( *p_sys->proxy.psz_host == '\0' ) )
+        {
+            msg_Warn( p_access, "invalid proxy host" );
+            vlc_UrlClean( &p_sys->proxy );
+            free( p_sys );
+            return VLC_EGENERIC;
+        }
+
+        if( p_sys->proxy.i_port <= 0 )
+            p_sys->proxy.i_port = 80;
+        msg_Dbg( p_access, "Using http proxy %s:%d",
+                 p_sys->proxy.psz_host, p_sys->proxy.i_port );
+    }
+
     /* open a tcp connection */
     vlc_UrlParse( &p_sys->url, p_access->psz_path, 0 );
-    if( p_sys->url.psz_host == NULL || *p_sys->url.psz_host == '\0' )
+    if( ( p_sys->url.psz_host == NULL ) ||
+        ( *p_sys->url.psz_host == '\0' ) )
     {
         msg_Err( p_access, "invalid host" );
+        vlc_UrlClean( &p_sys->proxy );
         vlc_UrlClean( &p_sys->url );
         free( p_sys );
         return VLC_EGENERIC;
@@ -108,6 +171,7 @@ int E_(MMSHOpen)( access_t *p_access )
 
     if( Describe( p_access, &psz_location ) )
     {
+        vlc_UrlClean( &p_sys->proxy );
         vlc_UrlClean( &p_sys->url );
         free( p_sys );
         return VLC_EGENERIC;
@@ -135,6 +199,7 @@ int E_(MMSHOpen)( access_t *p_access )
     {
         msg_Err( p_access, "cannot start stream" );
         free( p_sys->p_header );
+        vlc_UrlClean( &p_sys->proxy );
         vlc_UrlClean( &p_sys->url );
         free( p_sys );
         return VLC_EGENERIC;
@@ -156,6 +221,12 @@ void E_( MMSHClose )( access_t *p_access )
     access_sys_t *p_sys = p_access->p_sys;
 
     Stop( p_access );
+
+    if( p_sys->p_header )
+        free( p_sys->p_header  );
+
+    vlc_UrlClean( &p_sys->proxy );
+    vlc_UrlClean( &p_sys->url );
     free( p_sys );
 }
 
@@ -186,11 +257,6 @@ static int Control( access_t *p_access, int i_query, va_list args )
 
         case ACCESS_CAN_CONTROL_PACE:
             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
-
-#if 0       /* Disable for now until we have a clock synchro algo
-             * which works with something else than MPEG over UDP */
-            *pb_bool = VLC_FALSE;
-#endif
             *pb_bool = VLC_TRUE;
             break;
 
@@ -209,7 +275,7 @@ static int Control( access_t *p_access, int i_query, va_list args )
             i_int = (int)va_arg( args, int );
             pb_bool = (vlc_bool_t *)va_arg( args, vlc_bool_t * );
 
-            if( i_int < 0 || i_int > 127 )
+            if( (i_int < 0) || (i_int > 127) )
                 return VLC_EGENERIC;
             *pb_bool =  p_sys->asfh.stream[i_int].i_selected ? VLC_TRUE : VLC_FALSE;
             break;
@@ -220,6 +286,7 @@ static int Control( access_t *p_access, int i_query, va_list args )
         case ACCESS_SET_TITLE:
         case ACCESS_SET_SEEKPOINT:
         case ACCESS_SET_PRIVATE_ID_STATE:
+        case ACCESS_GET_CONTENT_TYPE:
             return VLC_EGENERIC;
 
         default:
@@ -250,6 +317,7 @@ static int Seek( access_t *p_access, int64_t i_pos )
 
     while( !p_access->b_die )
     {
+        msg_Warn( p_access, "GetPacket 1" );
         if( GetPacket( p_access, &ck ) )
             break;
 
@@ -272,7 +340,7 @@ static int Seek( access_t *p_access, int64_t i_pos )
  *****************************************************************************/
 static int ReadRedirect( access_t *p_access, uint8_t *p, int i_len )
 {
-    return 0;
+    return VLC_SUCCESS;
 }
 
 /*****************************************************************************
@@ -285,14 +353,15 @@ static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
     size_t       i_data = 0;
 
     if( p_access->info.b_eof )
-        return 0;
+        return VLC_SUCCESS;
 
     while( i_data < (size_t) i_len )
     {
-        if( p_access->info.i_pos < p_sys->i_start + p_sys->i_header )
+        if( p_access->info.i_pos < (p_sys->i_start + p_sys->i_header) )
         {
             int i_offset = p_access->info.i_pos - p_sys->i_start;
-            i_copy = __MIN( p_sys->i_header - i_offset, (int)((size_t)i_len - i_data) );
+            i_copy = __MIN( p_sys->i_header - i_offset,
+                            (int)((size_t)i_len - i_data) );
             memcpy( &p_buffer[i_data], &p_sys->p_header[i_offset], i_copy );
 
             i_data += i_copy;
@@ -302,7 +371,6 @@ static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
         {
             i_copy = __MIN( p_sys->i_packet_length - p_sys->i_packet_used,
                             i_len - i_data );
-
             memcpy( &p_buffer[i_data],
                     &p_sys->p_packet[p_sys->i_packet_used],
                     i_copy );
@@ -316,7 +384,6 @@ static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
         {
             i_copy = __MIN( p_sys->asfh.i_min_data_packet_size - p_sys->i_packet_used,
                             i_len - i_data );
-
             memset( &p_buffer[i_data], 0, i_copy );
 
             i_data += i_copy;
@@ -326,12 +393,13 @@ static int Read( access_t *p_access, uint8_t *p_buffer, int i_len )
         else
         {
             chunk_t ck;
+        msg_Warn( p_access, "GetPacket 2" );
             if( GetPacket( p_access, &ck ) )
             {
                 int i_ret = -1;
                 if( p_sys->b_broadcast )
                 {
-                    if( ck.i_type == 0x4524 && ck.i_sequence != 0 )
+                    if( (ck.i_type == 0x4524) && (ck.i_sequence != 0) )
                         i_ret = Restart( p_access );
                     else if( ck.i_type == 0x4324 )
                         i_ret = Reset( p_access );
@@ -371,15 +439,15 @@ static int Restart( access_t *p_access )
     if( Describe( p_access, &psz_location ) )
     {
         msg_Err( p_access, "describe failed" );
-        return -1;
+        return VLC_EGENERIC;
     }
     /* */
     if( Start( p_access, 0 ) )
     {
         msg_Err( p_access, "Start failed" );
-        return -1;
+        return VLC_EGENERIC;
     }
-    return 0;
+    return VLC_SUCCESS;
 }
 static int Reset( access_t *p_access )
 {
@@ -399,7 +467,7 @@ static int Reset( access_t *p_access )
     /* Get the next header FIXME memory loss ? */
     GetHeader( p_access );
     if( p_sys->i_header <= 0 )
-        return -1;
+        return VLC_EGENERIC;
 
     E_( asf_HeaderParse )( &p_sys->asfh,
                            p_sys->p_header, p_sys->i_header );
@@ -431,8 +499,61 @@ static int Reset( access_t *p_access )
     /* */
     p_sys->i_packet_used = 0;
     p_sys->i_packet_length = 0;
-    return 0;
+    return VLC_SUCCESS;
+}
+
+static int OpenConnection( access_t *p_access )
+{
+    access_sys_t *p_sys = p_access->p_sys;
+    vlc_url_t    srv = p_sys->b_proxy ? p_sys->proxy : p_sys->url;
+
+    if( ( p_sys->fd = net_ConnectTCP( p_access,
+                                      srv.psz_host, srv.i_port ) ) < 0 )
+    {
+        msg_Err( p_access, "cannot connect to %s:%d",
+                 srv.psz_host, srv.i_port );
+        return VLC_EGENERIC;
+    }
+
+    if( p_sys->b_proxy )
+    {
+        net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
+                    "GET http://%s:%d%s HTTP/1.0\r\n",
+                    p_sys->url.psz_host, p_sys->url.i_port,
+                    ( (p_sys->url.psz_path == NULL) ||
+                      (*p_sys->url.psz_path == '\0') ) ?
+                         "/" : p_sys->url.psz_path );
+
+        /* Proxy Authentication */
+        if( p_sys->proxy.psz_username && *p_sys->proxy.psz_username )
+        {
+            char *buf;
+            char *b64;
+
+            asprintf( &buf, "%s:%s", p_sys->proxy.psz_username,
+                       p_sys->proxy.psz_password ? p_sys->proxy.psz_password : "" );
+
+            b64 = vlc_b64_encode( buf );
+            free( buf );
+
+            net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
+                        "Proxy-Authorization: Basic %s\r\n", b64 );
+            free( b64 );
+        }
+    }
+    else
+    {
+        net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
+                    "GET %s HTTP/1.0\r\n"
+                    "Host: %s:%d\r\n",
+                    ( (p_sys->url.psz_path == NULL) ||
+                      (*p_sys->url.psz_path == '\0') ) ?
+                            "/" : p_sys->url.psz_path,
+                    p_sys->url.psz_host, p_sys->url.i_port );
+    }
+    return VLC_SUCCESS;
 }
+
 /*****************************************************************************
  * Describe:
  *****************************************************************************/
@@ -452,25 +573,15 @@ static int Describe( access_t  *p_access, char **ppsz_location )
     p_sys->p_packet = NULL;
     E_( GenerateGuid )( &p_sys->guid );
 
-    if( ( p_sys->fd = net_ConnectTCP( p_access, p_sys->url.psz_host,
-                                            p_sys->url.i_port ) ) < 0 )
-    {
-        msg_Err( p_access, "cannot connect to %s:%d", p_sys->url.psz_host,
-                 p_sys->url.i_port );
-        goto error;
-    }
+    if( OpenConnection( p_access ) )
+        return VLC_EGENERIC;
 
-    /* send first request */
     net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
-                "GET %s HTTP/1.0\r\n"
                 "Accept: */*\r\n"
                 "User-Agent: "MMSH_USER_AGENT"\r\n"
-                "Host: %s:%d\r\n"
                 "Pragma: no-cache,rate=1.000000,stream-time=0,stream-offset=0:0,request-context=%d,max-duration=0\r\n"
                 "Pragma: xClientGUID={"GUID_FMT"}\r\n"
                 "Connection: Close\r\n",
-                ( p_sys->url.psz_path == NULL || *p_sys->url.psz_path == '\0' ) ? "/" : p_sys->url.psz_path,
-                p_sys->url.psz_host, p_sys->url.i_port,
                 p_sys->i_request_context++,
                 GUID_PRINT( p_sys->guid ) );
 
@@ -486,6 +597,7 @@ static int Describe( access_t  *p_access, char **ppsz_location )
         msg_Err( p_access, "failed to read answer" );
         goto error;
     }
+
     if( strncmp( psz, "HTTP/1.", 7 ) )
     {
         msg_Err( p_access, "invalid HTTP reply '%s'", psz );
@@ -561,8 +673,8 @@ static int Describe( access_t  *p_access, char **ppsz_location )
     }
 
     /* Handle the redirection */
-    if( ( i_code == 301 || i_code == 302 ||
-          i_code == 303 || i_code == 307 ) &&
+    if( ( (i_code == 301) || (i_code == 302) ||
+          (i_code == 303) || (i_code == 307) ) &&
         psz_location && *psz_location )
     {
         msg_Dbg( p_access, "redirection to %s", psz_location );
@@ -580,7 +692,8 @@ static int Describe( access_t  *p_access, char **ppsz_location )
         goto error;
     }
     /* close this connection */
-    net_Close( p_sys->fd ); p_sys->fd = -1;
+    net_Close( p_sys->fd );
+    p_sys->fd = -1;
 
     /* *** parse header and get stream and their id *** */
     /* get all streams properties,
@@ -608,12 +721,15 @@ error:
     }
     return VLC_EGENERIC;
 }
+
 static void GetHeader( access_t *p_access )
 {
     access_sys_t *p_sys = p_access->p_sys;
 
     /* Read the asf header */
     p_sys->i_header = 0;
+    if( p_sys->p_header )
+        free( p_sys->p_header  );
     p_sys->p_header = NULL;
     for( ;; )
     {
@@ -634,26 +750,18 @@ static void GetHeader( access_t *p_access )
 
 
 /*****************************************************************************
- *
- *****************************************************************************/
+ * Start stream
+ ****************************************************************************/
 static int Start( access_t *p_access, off_t i_pos )
 {
     access_sys_t *p_sys = p_access->p_sys;
     int  i_streams = 0;
     int  i_streams_selected = 0;
     int  i;
-    char *psz;
+    char *psz = NULL;
 
     msg_Dbg( p_access, "starting stream" );
 
-    if( ( p_sys->fd = net_ConnectTCP( p_access, p_sys->url.psz_host,
-                                      p_sys->url.i_port ) ) < 0 )
-    {
-        /* should not occur */
-        msg_Err( p_access, "cannot connect to the server" );
-        return VLC_EGENERIC;
-    }
-
     for( i = 1; i < 128; i++ )
     {
         if( p_sys->asfh.stream[i].i_cat == ASF_STREAM_UNKNOWN )
@@ -662,19 +770,18 @@ static int Start( access_t *p_access, off_t i_pos )
         if( p_sys->asfh.stream[i].i_selected )
             i_streams_selected++;
     }
-
     if( i_streams_selected <= 0 )
     {
         msg_Err( p_access, "no stream selected" );
         return VLC_EGENERIC;
     }
+
+    if( OpenConnection( p_access ) )
+        return VLC_EGENERIC;
+
     net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
-                "GET %s HTTP/1.0\r\n"
                 "Accept: */*\r\n"
-                "User-Agent: "MMSH_USER_AGENT"\r\n"
-                "Host: %s:%d\r\n",
-                ( p_sys->url.psz_path == NULL || *p_sys->url.psz_path == '\0' ) ? "/" : p_sys->url.psz_path,
-                p_sys->url.psz_host, p_sys->url.i_port );
+                "User-Agent: "MMSH_USER_AGENT"\r\n" );
     if( p_sys->b_broadcast )
     {
         net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
@@ -706,12 +813,12 @@ static int Start( access_t *p_access, off_t i_pos )
             {
                 i_select = 0;
             }
-
             net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
                         "ffff:%d:%d ", i, i_select );
         }
     }
     net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL, "\r\n" );
+#if 0
     net_Printf( VLC_OBJECT(p_access), p_sys->fd, NULL,
                 "Connection: Close\r\n" );
 
@@ -720,12 +827,15 @@ static int Start( access_t *p_access, off_t i_pos )
         msg_Err( p_access, "failed to send request" );
         return VLC_EGENERIC;
     }
+#endif
 
-    if( ( psz = net_Gets( VLC_OBJECT(p_access), p_sys->fd, NULL ) ) == NULL )
+    psz = net_Gets( VLC_OBJECT(p_access), p_sys->fd, NULL );
+    if( psz == NULL )
     {
-        msg_Err( p_access, "cannot read data" );
+        msg_Err( p_access, "cannot read data 0" );
         return VLC_EGENERIC;
     }
+
     if( atoi( &psz[9] ) >= 400 )
     {
         msg_Err( p_access, "error: %s", psz );
@@ -741,7 +851,7 @@ static int Start( access_t *p_access, off_t i_pos )
         char *psz = net_Gets( p_access, p_sys->fd, NULL );
         if( psz == NULL )
         {
-            msg_Err( p_access, "cannot read data" );
+            msg_Err( p_access, "cannot read data 1" );
             return VLC_EGENERIC;
         }
         if( *psz == '\0' )
@@ -760,7 +870,7 @@ static int Start( access_t *p_access, off_t i_pos )
 }
 
 /*****************************************************************************
- *
+ * closing stream
  *****************************************************************************/
 static void Stop( access_t *p_access )
 {
@@ -775,7 +885,7 @@ static void Stop( access_t *p_access )
 }
 
 /*****************************************************************************
- *
+ * get packet
  *****************************************************************************/
 static int GetPacket( access_t * p_access, chunk_t *p_ck )
 {
@@ -792,7 +902,10 @@ static int GetPacket( access_t * p_access, chunk_t *p_ck )
      * entire header.
      */
     if( net_Read( p_access, p_sys->fd, NULL, p_sys->buffer, 4, VLC_TRUE ) < 4 )
+    {
+       msg_Err( p_access, "cannot read data 2" );
        return VLC_EGENERIC;
+    }
 
     p_ck->i_type = GetWLE( p_sys->buffer);
     p_ck->i_size = GetWLE( p_sys->buffer + 2);
@@ -803,7 +916,7 @@ static int GetPacket( access_t * p_access, chunk_t *p_ck )
 
     if( net_Read( p_access, p_sys->fd, NULL, p_sys->buffer + 4, restsize, VLC_TRUE ) < restsize )
     {
-        msg_Err( p_access, "cannot read data" );
+        msg_Err( p_access, "cannot read data 3" );
         return VLC_EGENERIC;
     }
     p_ck->i_sequence  = GetDWLE( p_sys->buffer + 4);
@@ -850,7 +963,7 @@ static int GetPacket( access_t * p_access, chunk_t *p_ck )
         (net_Read( p_access, p_sys->fd, NULL, &p_sys->buffer[12],
                    p_ck->i_data, VLC_TRUE ) < p_ck->i_data) )
     {
-        msg_Err( p_access, "cannot read data" );
+        msg_Err( p_access, "cannot read data 4" );
         return VLC_EGENERIC;
     }