/***************************************************************************** * mmstu.c: MMS access plug-in ***************************************************************************** * Copyright (C) 2001, 2002 VLC authors and VideoLAN * $Id$ * * Authors: Laurent Aimar * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as published by * the Free Software Foundation; either version 2.1 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program; if not, write to the Free Software Foundation, * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************/ /***************************************************************************** * Preamble *****************************************************************************/ #ifdef HAVE_CONFIG_H # include "config.h" #endif #include #include #include #include #include #include #include #ifdef HAVE_POLL # include #endif #include #include #include #include #include #include "asf.h" #include "buffer.h" #include "mms.h" #include "mmstu.h" #undef MMS_DEBUG /**************************************************************************** * NOTES: * MMSProtocole documentation found at http://get.to/sdp ****************************************************************************/ /***************************************************************************** * Local prototypes *****************************************************************************/ int MMSTUOpen ( stream_t * ); void MMSTUClose ( stream_t * ); static block_t *Block( stream_t *, bool * ); static int Seek( stream_t *, uint64_t ); static int Control( stream_t *, int, va_list ); static int MMSOpen ( stream_t *, vlc_url_t *, int ); static int MMSStart( stream_t *, uint32_t ); static int MMSStop ( stream_t * ); static void MMSClose( stream_t * ); static int mms_CommandRead( stream_t *p_access, int i_command1, int i_command2 ); static int mms_CommandSend( stream_t *, int, uint32_t, uint32_t, uint8_t *, int ); static int mms_HeaderMediaRead( stream_t *, int ); static int mms_ReceivePacket( stream_t * ); static void KeepAliveStart( stream_t * ); static void KeepAliveStop( stream_t * ); int MMSTUOpen( stream_t *p_access ) { access_sys_t *p_sys; int i_proto; int i_status; /* Set up p_access */ p_access->pf_read = NULL; p_access->pf_block = Block; p_access->pf_control = Control; p_access->pf_seek = Seek; p_access->p_sys = p_sys = calloc( 1, sizeof( access_sys_t ) ); if( !p_sys ) return VLC_ENOMEM; p_sys->i_timeout = var_CreateGetInteger( p_access, "mms-timeout" ); vlc_mutex_init( &p_sys->lock_netwrite ); /* *** Parse URL and get server addr/port and path *** */ vlc_UrlParse( &p_sys->url, p_access->psz_url ); if( p_sys->url.psz_host == NULL || *p_sys->url.psz_host == '\0' ) { msg_Err( p_access, "invalid server name" ); vlc_UrlClean( &p_sys->url ); vlc_mutex_destroy( &p_sys->lock_netwrite ); free( p_sys ); return VLC_EGENERIC; } if( p_sys->url.i_port <= 0 ) { p_sys->url.i_port = 1755; } /* *** connect to this server *** */ /* look at requested protocol (udp/tcp) */ i_proto = MMS_PROTO_AUTO; if( !strncmp( p_access->psz_name, "mmsu", 4 ) ) { i_proto = MMS_PROTO_UDP; } else if( !strncmp( p_access->psz_name, "mmst", 4 ) ) { i_proto = MMS_PROTO_TCP; } /* connect */ if( i_proto == MMS_PROTO_AUTO ) { /* first try with TCP and then UDP*/ i_status = MMSOpen( p_access, &p_sys->url, MMS_PROTO_TCP ); if( i_status ) i_status = MMSOpen( p_access, &p_sys->url, MMS_PROTO_UDP ); } else { i_status = MMSOpen( p_access, &p_sys->url, i_proto ); } if( i_status ) { msg_Err( p_access, "cannot connect to server" ); vlc_UrlClean( &p_sys->url ); vlc_mutex_destroy( &p_sys->lock_netwrite ); free( p_sys ); return VLC_EGENERIC; } msg_Dbg( p_access, "connected to %s:%d", p_sys->url.psz_host, p_sys->url.i_port ); /* * i_flags_broadcast * yy xx ?? ?? * broadcast yy=0x02, xx= 0x00 * pre-recorded yy=0x01, xx= 0x80 if video, 0x00 no video */ if( p_sys->i_packet_count <= 0 && p_sys->asfh.i_data_packets_count > 0 ) { p_sys->i_packet_count = p_sys->asfh.i_data_packets_count; } if( p_sys->i_packet_count <= 0 || ( p_sys->i_flags_broadcast >> 24 ) == 0x02 ) { p_sys->b_seekable = false; } else { p_sys->b_seekable = true; p_sys->i_size = (uint64_t)p_sys->i_header + (uint64_t)p_sys->i_packet_count * (uint64_t)p_sys->i_packet_length; } p_sys->b_keep_alive = false; /* *** Start stream *** */ if( MMSStart( p_access, 0xffffffff ) < 0 ) { msg_Err( p_access, "cannot start stream" ); MMSTUClose ( p_access ); return VLC_EGENERIC; } return VLC_SUCCESS; } /***************************************************************************** * Close: free unused data structures *****************************************************************************/ void MMSTUClose( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; KeepAliveStop( p_access ); /* close connection with server */ MMSClose( p_access ); /* free memory */ vlc_UrlClean( &p_sys->url ); free( p_sys ); } /***************************************************************************** * Control: *****************************************************************************/ static int Control( stream_t *p_access, int i_query, va_list args ) { access_sys_t *p_sys = p_access->p_sys; bool *pb_bool; bool b_bool; int64_t *pi_64; int i_int; switch( i_query ) { case STREAM_CAN_SEEK: pb_bool = va_arg( args, bool * ); *pb_bool = p_sys->b_seekable; break; case STREAM_CAN_FASTSEEK: pb_bool = va_arg( args, bool * ); *pb_bool = false; break; case STREAM_CAN_PAUSE: pb_bool = va_arg( args, bool * ); *pb_bool = true; break; case STREAM_CAN_CONTROL_PACE: pb_bool = va_arg( args, bool * ); #if 0 /* Disable for now until we have a clock synchro algo * which works with something else than MPEG over UDP */ *pb_bool = false; #endif *pb_bool = true; break; case STREAM_GET_SIZE: if( !p_sys->b_seekable ) return VLC_EGENERIC; *va_arg( args, uint64_t * ) = p_sys->i_size; break; case STREAM_GET_PTS_DELAY: pi_64 = va_arg( args, int64_t * ); *pi_64 = INT64_C(1000) * var_InheritInteger( p_access, "network-caching" ); break; case STREAM_GET_PRIVATE_ID_STATE: i_int = va_arg( args, int ); pb_bool = (bool *)va_arg( args, bool * ); if( i_int < 0 || i_int > 127 ) return VLC_EGENERIC; *pb_bool = p_sys->asfh.stream[i_int].i_selected ? true : false; break; case STREAM_SET_PRIVATE_ID_STATE: { i_int = va_arg( args, int ); b_bool = (bool)va_arg( args, int ); int i_cat; if( i_int > 127 ) return VLC_EGENERIC; else if ( i_int < 0 ) { /* Deselecting all ES in this category */ assert( !b_bool ); i_cat = -1 * i_int; if ( i_cat > ES_CATEGORY_COUNT ) return VLC_EGENERIC; } else { /* Chose another ES */ assert( b_bool ); i_cat = p_sys->asfh.stream[i_int].i_cat; } for ( int i=0; i< 128; i++ ) { /* First unselect all streams from the same cat */ if ( i_cat == p_sys->asfh.stream[i].i_cat ) p_sys->asfh.stream[i].i_selected = false; } if ( i_int > 0 ) p_sys->asfh.stream[i_int].i_selected = true; MMSStop( p_access ); Seek( p_access, p_sys->i_position ); return VLC_SUCCESS; } case STREAM_SET_PAUSE_STATE: b_bool = (bool)va_arg( args, int ); if( b_bool ) { MMSStop( p_access ); KeepAliveStart( p_access ); } else { KeepAliveStop( p_access ); Seek( p_access, p_sys->i_position ); } break; default: return VLC_EGENERIC; } return VLC_SUCCESS; } /***************************************************************************** * Seek: try to go at the right place *****************************************************************************/ static int Seek( stream_t * p_access, uint64_t i_pos ) { access_sys_t *p_sys = p_access->p_sys; uint32_t i_packet; uint32_t i_offset; var_buffer_t buffer; if( i_pos < p_sys->i_header) { if( p_sys->i_position < p_sys->i_header ) { /* no need to restart stream, it was already one * or no stream was yet read */ p_sys->i_position = i_pos; return VLC_SUCCESS; } else { i_packet = 0xffffffff; i_offset = 0; } } else { i_packet = ( i_pos - p_sys->i_header ) / p_sys->i_packet_length; i_offset = ( i_pos - p_sys->i_header ) % p_sys->i_packet_length; } if( p_sys->b_seekable && i_packet >= p_sys->i_packet_count ) return VLC_EGENERIC; msg_Dbg( p_access, "seeking to %"PRIu64 " (packet:%u)", i_pos, i_packet ); MMSStop( p_access ); msg_Dbg( p_access, "stream stopped (seek)" ); /* *** restart stream *** */ var_buffer_initwrite( &buffer, 0 ); var_buffer_add64( &buffer, 0 ); /* seek point in second */ var_buffer_add32( &buffer, 0xffffffff ); var_buffer_add32( &buffer, i_packet ); // begin from start var_buffer_add8( &buffer, 0xff ); // stream time limit var_buffer_add8( &buffer, 0xff ); // on 3bytes ... var_buffer_add8( &buffer, 0xff ); // var_buffer_add8( &buffer, 0x00 ); // don't use limit var_buffer_add32( &buffer, p_sys->i_media_packet_id_type ); mms_CommandSend( p_access, 0x07, p_sys->i_command_level, 0x0001ffff, buffer.p_data, buffer.i_data ); var_buffer_free( &buffer ); for( ;; ) { if( mms_HeaderMediaRead( p_access, MMS_PACKET_CMD ) < 0 ) { p_sys->b_eof = true; return VLC_EGENERIC; } if( p_sys->i_command == 0x1e ) { msg_Dbg( p_access, "received 0x1e (seek)" ); break; } } for( ;; ) { if( mms_HeaderMediaRead( p_access, MMS_PACKET_CMD ) < 0 ) { p_sys->b_eof = true; return VLC_EGENERIC; } if( p_sys->i_command == 0x05 ) { msg_Dbg( p_access, "received 0x05 (seek)" ); break; } } /* get a packet */ if( mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA ) < 0 ) { p_sys->b_eof = true; return VLC_EGENERIC; } msg_Dbg( p_access, "Streaming restarted" ); p_sys->i_media_used += i_offset; p_sys->i_position = i_pos; return VLC_SUCCESS; } /***************************************************************************** * Block: *****************************************************************************/ static block_t *Block( stream_t *p_access, bool *restrict eof ) { access_sys_t *p_sys = p_access->p_sys; if( p_sys->b_eof ) { *eof = true; return NULL; } if( p_sys->i_position < p_sys->i_header ) { const size_t i_copy = p_sys->i_header - p_sys->i_position; block_t *p_block = block_Alloc( i_copy ); if( !p_block ) return NULL; memcpy( p_block->p_buffer, &p_sys->p_header[p_sys->i_position], i_copy ); p_sys->i_position += i_copy; return p_block; } else if( p_sys->p_media && p_sys->i_media_used < __MAX( p_sys->i_media, p_sys->i_packet_length ) ) { size_t i_copy = 0; size_t i_padding = 0; if( p_sys->i_media_used < p_sys->i_media ) i_copy = p_sys->i_media - p_sys->i_media_used; if( __MAX( p_sys->i_media, p_sys->i_media_used ) < p_sys->i_packet_length ) i_padding = p_sys->i_packet_length - __MAX( p_sys->i_media, p_sys->i_media_used ); block_t *p_block = block_Alloc( i_copy + i_padding ); if( !p_block ) return NULL; if( i_copy > 0 ) memcpy( &p_block->p_buffer[0], &p_sys->p_media[p_sys->i_media_used], i_copy ); if( i_padding > 0 ) memset( &p_block->p_buffer[i_copy], 0, i_padding ); p_sys->i_media_used += i_copy + i_padding; p_sys->i_position += i_copy + i_padding; return p_block; } mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA ); return NULL; } /**************************************************************************** * MMSOpen : Open a connection with the server over mmst or mmsu ****************************************************************************/ static int MMSOpen( stream_t *p_access, vlc_url_t *p_url, int i_proto ) { access_sys_t *p_sys = p_access->p_sys; int b_udp = ( i_proto == MMS_PROTO_UDP ) ? 1 : 0; var_buffer_t buffer; char *tmp; const uint8_t *p; const uint8_t *p_cmdend; uint32_t i_server_version; uint32_t i_tool_version; uint32_t i_update_player_url; uint32_t i_encryption_type; int i_streams; int i_first; char *mediapath; /* *** Open a TCP connection with server *** */ msg_Dbg( p_access, "waiting for connection..." ); p_sys->i_handle_tcp = net_ConnectTCP( p_access, p_url->psz_host, p_url->i_port ); if( p_sys->i_handle_tcp < 0 ) { msg_Err( p_access, "failed to open a connection (tcp)" ); return VLC_EGENERIC; } msg_Dbg( p_access, "connection(tcp) with \"%s:%d\" successful", p_url->psz_host, p_url->i_port ); /* *** Bind port if UDP protocol is selected *** */ if( b_udp ) { if( net_GetSockAddress( p_sys->i_handle_tcp, p_sys->sz_bind_addr, NULL ) ) { net_Close( p_sys->i_handle_tcp ); return VLC_EGENERIC; } p_sys->i_handle_udp = net_ListenUDP1( (vlc_object_t *)p_access, p_sys->sz_bind_addr, 7000 ); if( p_sys->i_handle_udp < 0 ) { msg_Err( p_access, "failed to open a connection (udp)" ); net_Close( p_sys->i_handle_tcp ); return VLC_EGENERIC; } msg_Dbg( p_access, "connection(udp) at \"%s:%d\" successful", p_sys->sz_bind_addr, 7000 ); } /* *** Init context for mms prototcol *** */ GenerateGuid ( &p_sys->guid ); /* used to identify client by server */ msg_Dbg( p_access, "generated guid: "GUID_FMT, GUID_PRINT( p_sys->guid ) ); p_sys->i_command_level = 1; /* updated after 0x1A command */ p_sys->i_seq_num = 0; p_sys->i_media_packet_id_type = 0x04; p_sys->i_header_packet_id_type = 0x02; p_sys->i_proto = i_proto; p_sys->i_packet_seq_num = 0; p_sys->p_header = NULL; p_sys->i_header = 0; p_sys->p_media = NULL; p_sys->i_media = 0; p_sys->i_media_used = 0; p_sys->i_position = 0; p_sys->i_buffer_tcp = 0; p_sys->i_buffer_udp = 0; p_sys->p_cmd = NULL; p_sys->i_cmd = 0; p_sys->b_eof = false; /* *** send command 1 : connection request *** */ var_buffer_initwrite( &buffer, 0 ); var_buffer_add16( &buffer, 0x001c ); var_buffer_add16( &buffer, 0x0003 ); if( asprintf( &tmp, "NSPlayer/7.0.0.1956; {"GUID_FMT"}; Host: %s", GUID_PRINT( p_sys->guid ), p_url->psz_host ) < 0 ) { var_buffer_free( &buffer ); net_Close( p_sys->i_handle_tcp ); return VLC_ENOMEM; } var_buffer_addUTF16( p_access, &buffer, tmp ); free( tmp ); mms_CommandSend( p_access, 0x01, /* connexion request */ 0x00000000, /* flags, FIXME */ 0x0004000b, /* ???? */ buffer.p_data, buffer.i_data ); if( mms_CommandRead( p_access, 0x01, 0 ) < 0 || p_sys->i_cmd < MMS_CMD_HEADERSIZE + 48 ) { var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } i_server_version = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 32 ); i_tool_version = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 36 ); i_update_player_url = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 40 ); i_encryption_type = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 44 ); p = p_sys->p_cmd + MMS_CMD_HEADERSIZE + 48; p_cmdend = &p_sys->p_cmd[p_sys->i_cmd]; #define GETUTF16( fmt, size ) \ do \ { \ if( (p_cmdend - p) / 2u < (size) ) \ {\ var_buffer_free( &buffer );\ MMSClose( p_access );\ return VLC_EBADVAR;\ }\ char *str = FromCharset( "UTF-16LE", p, (size) * 2 ); \ p += (size) * 2; \ if( str != NULL ) \ { \ msg_Dbg( p_access, fmt " %s", str ); \ free( str ); \ } \ } while (0) GETUTF16( "server version: ", i_server_version ); GETUTF16( "tool version: ", i_tool_version ); GETUTF16( "update player URL:", i_update_player_url ); GETUTF16( "encryption type: ", i_encryption_type ); #undef GETUTF16 /* *** should make an 18 command to make data timing *** */ /* *** send command 2 : transport protocol selection *** */ var_buffer_reinitwrite( &buffer, 0 ); var_buffer_add32( &buffer, 0x00000000 ); var_buffer_add32( &buffer, 0x000a0000 ); var_buffer_add32( &buffer, 0x00000002 ); if( b_udp ) { if( asprintf( &tmp, "\\\\%s\\UDP\\%d", p_sys->sz_bind_addr, 7000 ) < 0) // FIXME { var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } } else { if( asprintf( &tmp, "\\\\192.168.0.1\\TCP\\1242" ) < 0 ) { var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } } var_buffer_addUTF16( p_access, &buffer, tmp ); var_buffer_add16( &buffer, '0' ); free( tmp ); mms_CommandSend( p_access, 0x02, /* connexion request */ 0x00000000, /* flags, FIXME */ 0xffffffff, /* ???? */ buffer.p_data, buffer.i_data ); /* *** response from server, should be 0x02 or 0x03 *** */ mms_CommandRead( p_access, 0x02, 0x03 ); if( p_sys->i_command == 0x03 ) { msg_Err( p_access, "%s protocol selection failed", b_udp ? "UDP" : "TCP" ); var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } else if( p_sys->i_command != 0x02 ) { msg_Warn( p_access, "received command isn't 0x02 in response to 0x02" ); } /* *** send command 5 : media file name/path requested *** */ var_buffer_reinitwrite( &buffer, 0 ); var_buffer_add64( &buffer, 0 ); /* media file path shouldn't start with / character */ mediapath = p_url->psz_path; if ( mediapath && *mediapath == '/' ) { mediapath++; } var_buffer_addUTF16( p_access, &buffer, mediapath ); mms_CommandSend( p_access, 0x05, p_sys->i_command_level, 0xffffffff, buffer.p_data, buffer.i_data ); /* *** wait for response *** */ mms_CommandRead( p_access, 0x1a, 0x06 ); /* test if server send 0x1A answer */ if( p_sys->i_command == 0x1A ) { msg_Err( p_access, "id/password requested (not yet supported)" ); /* FIXME */ var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } if( p_sys->i_command != 0x06 ) { msg_Err( p_access, "unknown answer (0x%x instead of 0x06)", p_sys->i_command ); var_buffer_free( &buffer ); MMSClose( p_access ); return( -1 ); } if( p_sys->i_cmd < MMS_CMD_HEADERSIZE + 64 ) { var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EBADVAR; } /* 1 for file ok, 2 for authen ok */ switch( GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE ) ) { case 0x0001: msg_Dbg( p_access, "media file name/path accepted" ); break; case 0x0002: msg_Dbg( p_access, "authentication accepted" ); break; case -1: default: msg_Err( p_access, "error while asking for file %d", GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE ) ); var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } p_sys->i_flags_broadcast = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 12 ); p_sys->i_media_length = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 24 ); p_sys->i_packet_length = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 44 ); p_sys->i_packet_count = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 48 ); p_sys->i_max_bit_rate = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 56 ); p_sys->i_header_size = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 60 ); msg_Dbg( p_access, "answer 0x06 flags:0x%8.8"PRIx32" media_length:%"PRIu32"s " "packet_length:%zu packet_count:%"PRIu32" max_bit_rate:%"PRIu32 "header_size:%zu", p_sys->i_flags_broadcast, p_sys->i_media_length, p_sys->i_packet_length, p_sys->i_packet_count, p_sys->i_max_bit_rate, p_sys->i_header_size ); /* *** send command 15 *** */ var_buffer_reinitwrite( &buffer, 0 ); var_buffer_add32( &buffer, 0 ); var_buffer_add32( &buffer, 0x8000 ); var_buffer_add32( &buffer, 0xffffffff ); var_buffer_add32( &buffer, 0x00 ); var_buffer_add32( &buffer, 0x00 ); var_buffer_add32( &buffer, 0x00 ); var_buffer_add64( &buffer, (((uint64_t)0x40ac2000)<<32) ); var_buffer_add32( &buffer, p_sys->i_header_packet_id_type ); var_buffer_add32( &buffer, 0x00 ); mms_CommandSend( p_access, 0x15, p_sys->i_command_level, 0x00, buffer.p_data, buffer.i_data ); /* *** wait for response *** */ /* Commented out because it fails on some stream (no 0x11 answer) */ #if 0 mms_CommandRead( p_access, 0x11, 0 ); if( p_sys->i_command != 0x11 ) { msg_Err( p_access, "unknown answer (0x%x instead of 0x11)", p_sys->i_command ); var_buffer_free( &buffer ); MMSClose( p_access ); return( -1 ); } #endif /* *** now read header packet *** */ /* XXX could be split over multiples packets */ msg_Dbg( p_access, "reading header" ); for( ;; ) { if( mms_HeaderMediaRead( p_access, MMS_PACKET_HEADER ) < 0 ) { msg_Err( p_access, "cannot receive header" ); var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } if( p_sys->i_header >= p_sys->i_header_size ) { msg_Dbg( p_access, "header complete(%zu)", p_sys->i_header ); break; } msg_Dbg( p_access, "header incomplete (%zu/%zu), reading more", p_sys->i_header, p_sys->i_header_size ); } /* *** parse header and get stream and their id *** */ /* get all streams properties, * * TODO : stream bitrates properties(optional) * and bitrate mutual exclusion(optional) */ asf_HeaderParse ( &p_sys->asfh, p_sys->p_header, p_sys->i_header ); asf_StreamSelect( &p_sys->asfh, var_InheritInteger( p_access, "mms-maxbitrate" ), var_InheritBool( p_access, "mms-all" ), var_InheritBool( p_access, "audio" ), var_InheritBool( p_access, "video" ) ); /* *** now select stream we want to receive *** */ /* TODO take care of stream bitrate TODO */ i_streams = 0; i_first = -1; var_buffer_reinitwrite( &buffer, 0 ); /* for now, select first audio and video stream */ for( int i = 1; i < 128; i++ ) { if( p_sys->asfh.stream[i].i_cat != ASF_CODEC_TYPE_UNKNOWN ) { i_streams++; if( i_first != -1 ) { var_buffer_add16( &buffer, 0xffff ); var_buffer_add16( &buffer, i ); } else { i_first = i; } if( p_sys->asfh.stream[i].i_selected ) { var_buffer_add16( &buffer, 0x0000 ); msg_Info( p_access, "selecting stream[0x%x] %s (%d Kib/s)", i, ( p_sys->asfh.stream[i].i_cat == ASF_CODEC_TYPE_AUDIO ) ? "audio" : "video" , p_sys->asfh.stream[i].i_bitrate / 1024); } else { var_buffer_add16( &buffer, 0x0002 ); msg_Info( p_access, "ignoring stream[0x%x] %s (%d Kib/s)", i, ( p_sys->asfh.stream[i].i_cat == ASF_CODEC_TYPE_AUDIO ) ? "audio" : "video" , p_sys->asfh.stream[i].i_bitrate / 1024); } } } if( i_streams == 0 ) { msg_Err( p_access, "cannot find any stream" ); var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } mms_CommandSend( p_access, 0x33, i_streams, 0xffff | ( i_first << 16 ), buffer.p_data, buffer.i_data ); mms_CommandRead( p_access, 0x21, 0 ); if( p_sys->i_command != 0x21 ) { msg_Err( p_access, "unknown answer (0x%x instead of 0x21)", p_sys->i_command ); var_buffer_free( &buffer ); MMSClose( p_access ); return VLC_EGENERIC; } var_buffer_free( &buffer ); msg_Info( p_access, "connection successful" ); return VLC_SUCCESS; } /**************************************************************************** * MMSStart : Start streaming ****************************************************************************/ static int MMSStart( stream_t *p_access, uint32_t i_packet ) { access_sys_t *p_sys = p_access->p_sys; var_buffer_t buffer; /* *** start stream from packet 0 *** */ var_buffer_initwrite( &buffer, 0 ); var_buffer_add64( &buffer, 0 ); /* seek point in second */ var_buffer_add32( &buffer, 0xffffffff ); var_buffer_add32( &buffer, i_packet ); // begin from start var_buffer_add8( &buffer, 0xff ); // stream time limit var_buffer_add8( &buffer, 0xff ); // on 3bytes ... var_buffer_add8( &buffer, 0xff ); // var_buffer_add8( &buffer, 0x00 ); // don't use limit var_buffer_add32( &buffer, p_sys->i_media_packet_id_type ); mms_CommandSend( p_access, 0x07, p_sys->i_command_level, 0x0001ffff, buffer.p_data, buffer.i_data ); var_buffer_free( &buffer ); mms_CommandRead( p_access, 0x05, 0 ); if( p_sys->i_command != 0x05 ) { msg_Err( p_access, "unknown answer (0x%x instead of 0x05)", p_sys->i_command ); return -1; } else { /* get a packet */ if( mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA ) < 0 ) return -1; msg_Dbg( p_access, "streaming started" ); return 0; } } /**************************************************************************** * MMSStop : Stop streaming ****************************************************************************/ static int MMSStop( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; /* *** stop stream but keep connection alive *** */ mms_CommandSend( p_access, 0x09, p_sys->i_command_level, 0x001fffff, NULL, 0 ); return( 0 ); } /**************************************************************************** * MMSClose : Close streaming and connection ****************************************************************************/ static void MMSClose( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; msg_Dbg( p_access, "Connection closed" ); /* *** tell server that we will disconnect *** */ mms_CommandSend( p_access, 0x0d, p_sys->i_command_level, 0x00000001, NULL, 0 ); /* *** close sockets *** */ net_Close( p_sys->i_handle_tcp ); if( p_sys->i_proto == MMS_PROTO_UDP ) { net_Close( p_sys->i_handle_udp ); } FREENULL( p_sys->p_cmd ); FREENULL( p_sys->p_media ); FREENULL( p_sys->p_header ); p_sys->i_header = 0; } /**************************************************************************** * * MMS specific functions * ****************************************************************************/ static int mms_CommandSend( stream_t *p_access, int i_command, uint32_t i_prefix1, uint32_t i_prefix2, uint8_t *p_data, int i_data_old ) { var_buffer_t buffer; access_sys_t *p_sys = p_access->p_sys; int i_data_by8, i_ret; int i_data = i_data_old; while( i_data & 0x7 ) i_data++; i_data_by8 = i_data >> 3; /* first init buffer */ var_buffer_initwrite( &buffer, 0 ); var_buffer_add32( &buffer, 0x00000001 ); /* start sequence */ var_buffer_add32( &buffer, 0xB00BFACE ); /* size after protocol type */ var_buffer_add32( &buffer, i_data + MMS_CMD_HEADERSIZE - 16 ); var_buffer_add32( &buffer, 0x20534d4d ); /* protocol "MMS " */ var_buffer_add32( &buffer, i_data_by8 + 4 ); var_buffer_add32( &buffer, p_sys->i_seq_num ); p_sys->i_seq_num++; var_buffer_add64( &buffer, 0 ); var_buffer_add32( &buffer, i_data_by8 + 2 ); var_buffer_add32( &buffer, 0x00030000 | i_command ); /* dir | command */ var_buffer_add32( &buffer, i_prefix1 ); /* command specific */ var_buffer_add32( &buffer, i_prefix2 ); /* command specific */ /* specific command data */ if( p_data && i_data > 0 ) { var_buffer_addmemory( &buffer, p_data, i_data_old ); } /* Append padding to the command data */ var_buffer_add64( &buffer, 0 ); /* send it */ vlc_mutex_lock( &p_sys->lock_netwrite ); i_ret = net_Write( p_access, p_sys->i_handle_tcp, buffer.p_data, buffer.i_data - ( 8 - ( i_data - i_data_old ) ) ); vlc_mutex_unlock( &p_sys->lock_netwrite ); if( i_ret != buffer.i_data - ( 8 - ( i_data - i_data_old ) ) ) { var_buffer_free( &buffer ); msg_Err( p_access, "failed to send command" ); return VLC_EGENERIC; } var_buffer_free( &buffer ); return VLC_SUCCESS; } static int NetFillBuffer( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; int i_ret; struct pollfd ufd[2]; unsigned timeout = p_sys->i_timeout, nfd = 0; /* FIXME when using udp */ ssize_t i_tcp, i_udp; ssize_t i_tcp_read, i_udp_read; i_tcp = MMS_BUFFER_SIZE/2 - p_sys->i_buffer_tcp; if( p_sys->i_proto == MMS_PROTO_UDP ) { i_udp = MMS_BUFFER_SIZE/2 - p_sys->i_buffer_udp; } else { i_udp = 0; /* there isn't udp socket */ } if( ( i_udp <= 0 ) && ( i_tcp <= 0 ) ) { msg_Warn( p_access, "nothing to read %d:%d", (int)i_tcp, (int)i_udp ); return 0; } else { /* msg_Warn( p_access, "ask for tcp:%d udp:%d", i_tcp, i_udp ); */ } /* Initialize file descriptor set */ if( i_tcp > 0 ) { ufd[nfd].fd = p_sys->i_handle_tcp; ufd[nfd].events = POLLIN; nfd++; } if( i_udp > 0 ) { ufd[nfd].fd = p_sys->i_handle_udp; ufd[nfd].events = POLLIN; nfd++; } /* Find if some data is available */ if( p_sys->i_buffer_tcp > 0 || p_sys->i_buffer_udp > 0 ) timeout = 2000; do { if( vlc_killed() ) return -1; i_ret = vlc_poll_i11e(ufd, nfd, timeout); } while( i_ret < 0 && errno == EINTR ); /* FIXME: recompute timeout */ if( i_ret == 0 ) { msg_Err(p_access, "no data received"); return -1; } if( i_ret < 0 ) { msg_Err( p_access, "network poll error: %s", vlc_strerror_c(errno) ); return -1; } i_tcp_read = i_udp_read = 0; if( ( i_tcp > 0 ) && ufd[0].revents ) { i_tcp_read = recv( p_sys->i_handle_tcp, p_sys->buffer_tcp + p_sys->i_buffer_tcp, MMS_BUFFER_SIZE - p_sys->i_buffer_tcp, 0 ); } if( i_udp > 0 && ufd[i_tcp > 0].revents ) { i_udp_read = recv( p_sys->i_handle_udp, p_sys->buffer_udp + p_sys->i_buffer_udp, MMS_BUFFER_SIZE - p_sys->i_buffer_udp, 0 ); } #ifdef MMS_DEBUG if( p_sys->i_proto == MMS_PROTO_UDP ) { msg_Dbg( p_access, "filling buffer TCP:%d+%d UDP:%d+%d", p_sys->i_buffer_tcp, i_tcp_read, p_sys->i_buffer_udp, i_udp_read ); } else { msg_Dbg( p_access, "filling buffer TCP:%d+%d", p_sys->i_buffer_tcp, i_tcp_read ); } #endif if( i_tcp_read > 0 ) p_sys->i_buffer_tcp += (size_t) i_tcp_read; if( i_udp_read > 0 ) p_sys->i_buffer_udp += (size_t) i_udp_read; return i_tcp_read + i_udp_read; } static int mms_ParseCommand( stream_t *p_access, uint8_t *p_data, size_t i_data, size_t *pi_used ) { #define GET32( i_pos ) \ ( p_sys->p_cmd[i_pos] + ( p_sys->p_cmd[i_pos +1] << 8 ) + \ ( p_sys->p_cmd[i_pos + 2] << 16 ) + \ ( p_sys->p_cmd[i_pos + 3] << 24 ) ) access_sys_t *p_sys = p_access->p_sys; uint32_t i_length; uint32_t i_id; free( p_sys->p_cmd ); p_sys->p_cmd = NULL; p_sys->i_cmd = 0; p_sys->i_command = 0; *pi_used = 0; if( i_data < MMS_CMD_HEADERSIZE ) { msg_Warn( p_access, "truncated command (header incomplete)" ); return -1; } i_id = GetDWLE( p_data + 4 ); i_length = GetDWLE( p_data + 8 ) + 16; if( i_id != 0xb00bface || i_length < 16 ) { msg_Err( p_access, "incorrect command header (0x%"PRIx32")", i_id ); return -1; } if( i_length > i_data ) { msg_Warn( p_access, "truncated command (missing %zu bytes)", (size_t)i_length - i_data ); return -1; } if( (p_sys->p_cmd = malloc( i_length )) ) { memcpy( p_sys->p_cmd, p_data, i_length ); } else { return -1; } p_sys->i_cmd = i_length; *pi_used = i_length; msg_Dbg( p_access, "recv command start_sequence:0x%8.8x command_id:0x%8.8x length:%d len8:%d sequence 0x%8.8x len8_II:%d dir_comm:0x%8.8x", GET32( 0 ), GET32( 4 ), GET32( 8 ), /* 12: protocol type "MMS " */ GET32( 16 ), GET32( 20 ), /* 24: unknown (0) */ /* 28: unknown (0) */ GET32( 32 ), GET32( 36 ) /* 40: switches */ /* 44: extra */ ); p_sys->i_command = GET32( 36 ) & 0xffff; #undef GET32 return MMS_PACKET_CMD; } static int mms_ParsePacket( stream_t *p_access, const uint8_t *p_data, size_t i_data, size_t *pi_used ) { access_sys_t *p_sys = p_access->p_sys; int i_packet_seq_num; size_t i_packet_length; uint32_t i_packet_id; *pi_used = i_data; /* default */ if( i_data <= 8 ) { msg_Warn( p_access, "truncated packet (header incomplete)" ); return -1; } i_packet_id = p_data[4]; i_packet_seq_num = GetDWLE( p_data ); i_packet_length = GetWLE( p_data + 6 ); //msg_Warn( p_access, "------->i_packet_length=%d, i_data=%d", i_packet_length, i_data ); if( i_packet_length > i_data || i_packet_length <= 8) { /* msg_Dbg( p_access, "truncated packet (Declared %d bytes, Actual %d bytes)", i_packet_length, i_data ); */ *pi_used = 0; return -1; } else if( i_packet_length < i_data ) { *pi_used = i_packet_length; } if( i_packet_id == 0xff ) { msg_Warn( p_access, "receive MMS UDP pair timing" ); return( MMS_PACKET_UDP_TIMING ); } if( i_packet_id != p_sys->i_header_packet_id_type && i_packet_id != p_sys->i_media_packet_id_type ) { msg_Warn( p_access, "incorrect Packet Id Type (0x%x)", i_packet_id ); return -1; } /* we now have a media or a header packet */ if( i_packet_seq_num != p_sys->i_packet_seq_num ) { #if 0 /* FIXME for udp could be just wrong order ? */ msg_Warn( p_access, "detected packet lost (%d != %d)", i_packet_seq_num, p_sys->i_packet_seq_num ); #endif } p_sys->i_packet_seq_num = i_packet_seq_num + 1; i_packet_length -= 8; // don't bother with preheader if( i_packet_id == p_sys->i_header_packet_id_type ) { size_t new_header_size; if( add_overflow( p_sys->i_header, i_packet_length, &new_header_size ) ) return -1; uint8_t *p_reaced = realloc( p_sys->p_header, new_header_size ); if( !p_reaced ) return -1; memcpy( &p_reaced[p_sys->i_header], p_data + 8, i_packet_length ); p_sys->p_header = p_reaced; p_sys->i_header = new_header_size; /* msg_Dbg( p_access, "receive header packet (%d bytes)", i_packet_length ); */ return MMS_PACKET_HEADER; } else { free( p_sys->p_media ); p_sys->i_media = 0; p_sys->i_media_used = 0; p_sys->p_media = malloc( i_packet_length ); if( !p_sys->p_media ) return -1; p_sys->i_media = i_packet_length; memcpy( p_sys->p_media, p_data + 8, p_sys->i_media ); /* msg_Dbg( p_access, "receive media packet (%d bytes)", i_packet_length ); */ return MMS_PACKET_MEDIA; } } static int mms_ReceivePacket( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; int i_packet_tcp_type; int i_packet_udp_type; for( ;; ) { bool b_refill = true; /* first if we need to refill buffer */ if( p_sys->i_buffer_tcp >= MMS_CMD_HEADERSIZE ) { if( GetDWLE( p_sys->buffer_tcp + 4 ) == 0xb00bface ) { if( GetDWLE( p_sys->buffer_tcp + 8 ) + 16 <= (size_t)p_sys->i_buffer_tcp ) { b_refill = false; } } else if( GetWLE( p_sys->buffer_tcp + 6 ) <= p_sys->i_buffer_tcp ) { b_refill = false; } } if( p_sys->i_proto == MMS_PROTO_UDP && p_sys->i_buffer_udp >= 8 && GetWLE( p_sys->buffer_udp + 6 ) <= p_sys->i_buffer_udp ) { b_refill = false; } if( b_refill && NetFillBuffer( p_access ) < 0 ) { msg_Warn( p_access, "cannot fill buffer" ); return -1; } i_packet_tcp_type = -1; i_packet_udp_type = -1; if( p_sys->i_buffer_tcp > 0 ) { size_t i_used; if( GetDWLE( p_sys->buffer_tcp + 4 ) == 0xb00bface ) { i_packet_tcp_type = mms_ParseCommand( p_access, p_sys->buffer_tcp, p_sys->i_buffer_tcp, &i_used ); } else { i_packet_tcp_type = mms_ParsePacket( p_access, p_sys->buffer_tcp, p_sys->i_buffer_tcp, &i_used ); } if( i_used > 0 && i_used < MMS_BUFFER_SIZE ) { memmove( p_sys->buffer_tcp, p_sys->buffer_tcp + i_used, MMS_BUFFER_SIZE - i_used ); } p_sys->i_buffer_tcp -= i_used; } else if( p_sys->i_buffer_udp > 0 ) { size_t i_used; i_packet_udp_type = mms_ParsePacket( p_access, p_sys->buffer_udp, p_sys->i_buffer_udp, &i_used ); if( i_used > 0 && i_used < MMS_BUFFER_SIZE ) { memmove( p_sys->buffer_udp, p_sys->buffer_udp + i_used, MMS_BUFFER_SIZE - i_used ); } p_sys->i_buffer_udp -= i_used; } if( i_packet_tcp_type == MMS_PACKET_CMD && p_sys->i_command == 0x1b ) { mms_CommandSend( p_access, 0x1b, 0, 0, NULL, 0 ); i_packet_tcp_type = -1; } if( i_packet_tcp_type != -1 ) { return i_packet_tcp_type; } else if( i_packet_udp_type != -1 ) { return i_packet_udp_type; } } } static int mms_ReceiveCommand( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; for( ;; ) { size_t i_used; int i_status; if( NetFillBuffer( p_access ) < 0 ) { msg_Warn( p_access, "cannot fill buffer" ); return VLC_EGENERIC; } if( p_sys->i_buffer_tcp > 0 ) { i_status = mms_ParseCommand( p_access, p_sys->buffer_tcp, p_sys->i_buffer_tcp, &i_used ); if( i_used < MMS_BUFFER_SIZE ) { memmove( p_sys->buffer_tcp, p_sys->buffer_tcp + i_used, MMS_BUFFER_SIZE - i_used ); } p_sys->i_buffer_tcp -= i_used; if( i_status < 0 ) { return VLC_EGENERIC; } if( p_sys->i_command == 0x1b ) { mms_CommandSend( p_access, 0x1b, 0, 0, NULL, 0 ); } else { break; } } else { return VLC_EGENERIC; } } return VLC_SUCCESS; } #define MMS_RETRY_MAX 10 static int mms_CommandRead( stream_t *p_access, int i_command1, int i_command2 ) { access_sys_t *p_sys = p_access->p_sys; int i_count; int i_status; for( i_count = 0; i_count < MMS_RETRY_MAX; ) { i_status = mms_ReceiveCommand( p_access ); if( i_status < 0 || p_sys->i_command == 0 ) { i_count++; } else if( i_command1 == 0 && i_command2 == 0) { return VLC_SUCCESS; } else if( p_sys->i_command == i_command1 || p_sys->i_command == i_command2 ) { return VLC_SUCCESS; } else { switch( p_sys->i_command ) { case 0x03: msg_Warn( p_access, "socket closed by server" ); p_sys->b_eof = true; return VLC_EGENERIC; case 0x1e: msg_Warn( p_access, "end of media stream" ); p_sys->b_eof = true; return VLC_EGENERIC; default: break; } } } p_sys->b_eof = true; msg_Warn( p_access, "failed to receive command (aborting)" ); return VLC_EGENERIC; } static int mms_HeaderMediaRead( stream_t *p_access, int i_type ) { access_sys_t *p_sys = p_access->p_sys; int i_count; for( i_count = 0; i_count < MMS_RETRY_MAX; ) { int i_status; i_status = mms_ReceivePacket( p_access ); if( i_status < 0 ) { i_count++; msg_Warn( p_access, "cannot receive header (%d/%d)", i_count, MMS_RETRY_MAX ); } else if( i_status == i_type || i_type == MMS_PACKET_ANY ) { return i_type; } else if( i_status == MMS_PACKET_CMD ) { switch( p_sys->i_command ) { case 0x03: msg_Warn( p_access, "socket closed by server" ); p_sys->b_eof = true; return -1; case 0x1e: msg_Warn( p_access, "end of media stream" ); p_sys->b_eof = true; return -1; case 0x20: /* XXX not too dificult to be done EXCEPT that we * need to restart demuxer... and I don't see how we * could do that :p */ msg_Err( p_access, "reinitialization needed --> unsupported" ); p_sys->b_eof = true; return -1; default: break; } } } msg_Err( p_access, "cannot receive %s (aborting)", ( i_type == MMS_PACKET_HEADER ) ? "header" : "media data" ); p_sys->b_eof = true; return -1; } noreturn static void *KeepAliveThread( void *p_data ) { stream_t *p_access = p_data; for( ;; ) { /* Send keep-alive every ten seconds */ int canc = vlc_savecancel(); mms_CommandSend( p_access, 0x1b, 0, 0, NULL, 0 ); vlc_restorecancel( canc ); msleep( 10 * CLOCK_FREQ ); } vlc_assert_unreachable(); } static void KeepAliveStart( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; if( p_sys->b_keep_alive ) return; p_sys->b_keep_alive = !vlc_clone( &p_sys->keep_alive, KeepAliveThread, p_access, VLC_THREAD_PRIORITY_LOW ); } static void KeepAliveStop( stream_t *p_access ) { access_sys_t *p_sys = p_access->p_sys; if( !p_sys->b_keep_alive ) return; vlc_cancel( p_sys->keep_alive ); vlc_join( p_sys->keep_alive, NULL ); p_sys->b_keep_alive = false; }