Switch MMS to poll to avoid select() issues. Needs review/testing.
[vlc.git] / modules / access / mms / mmstu.c
1 /*****************************************************************************
2  * mms.c: MMS access plug-in
3  *****************************************************************************
4  * Copyright (C) 2001, 2002 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 #include <vlc/vlc.h>
29 #include <vlc_access.h>
30
31 #include <errno.h>
32
33 #ifdef HAVE_UNISTD_H
34 #   include <unistd.h>
35 #endif
36 #ifdef HAVE_FCNTL_H
37 #   include <fcntl.h>
38 #endif
39 #ifdef HAVE_SYS_TIME_H
40 #   include <sys/time.h>
41 #endif
42 #ifdef HAVE_SYS_TYPES_H
43 #   include <sys/types.h>
44 #endif
45 #ifdef HAVE_SYS_STAT_H
46 #   include <sys/stat.h>
47 #endif
48
49 #include <vlc_network.h>
50 #include <poll.h>
51 #include "vlc_url.h"
52 #include "asf.h"
53 #include "buffer.h"
54
55 #include "mms.h"
56 #include "mmstu.h"
57
58 #undef MMS_DEBUG
59
60 /****************************************************************************
61  * NOTES:
62  *  MMSProtocole documentation found at http://get.to/sdp
63  ****************************************************************************/
64
65 /*****************************************************************************
66  * Local prototypes
67  *****************************************************************************/
68 int  E_( MMSTUOpen )  ( access_t * );
69 void E_( MMSTUClose ) ( access_t * );
70
71
72 static ssize_t Read( access_t *, uint8_t *, size_t );
73 static int Seek( access_t *, int64_t );
74 static int Control( access_t *, int, va_list );
75
76 static int  MMSOpen ( access_t *, vlc_url_t *, int );
77 static int  MMSStart( access_t *, uint32_t );
78 static int  MMSStop ( access_t * );
79 static void MMSClose( access_t * );
80
81
82 static int  mms_CommandRead( access_t *p_access, int i_command1, int i_command2 );
83 static int  mms_CommandSend( access_t *, int, uint32_t, uint32_t, uint8_t *, int );
84
85 static int  mms_HeaderMediaRead( access_t *, int );
86
87 static int  mms_ReceivePacket( access_t * );
88
89
90 int  E_(MMSTUOpen)( access_t *p_access )
91 {
92     access_sys_t   *p_sys;
93     int             i_proto;
94     int             i_status;
95
96     /* Set up p_access */
97     p_access->pf_read = Read;
98     p_access->pf_block = NULL;
99     p_access->pf_control = Control;
100     p_access->pf_seek = Seek;
101     p_access->info.i_update = 0;
102     p_access->info.i_size = 0;
103     p_access->info.i_pos = 0;
104     p_access->info.b_eof = VLC_FALSE;
105     p_access->info.i_title = 0;
106     p_access->info.i_seekpoint = 0;
107     p_access->p_sys = p_sys = malloc( sizeof( access_sys_t ) );
108     memset( p_sys, 0, sizeof( access_sys_t ) );
109
110     /* *** Parse URL and get server addr/port and path *** */
111     vlc_UrlParse( &p_sys->url, p_access->psz_path, 0 );
112     if( p_sys->url.psz_host == NULL || *p_sys->url.psz_host == '\0' )
113     {
114         msg_Err( p_access, "invalid server name" );
115         vlc_UrlClean( &p_sys->url );
116         free( p_sys );
117         return VLC_EGENERIC;
118     }
119     if( p_sys->url.i_port <= 0 )
120     {
121         p_sys->url.i_port = 1755;
122     }
123
124     /* *** connect to this server *** */
125     /* look at  requested protocol (udp/tcp) */
126     i_proto = MMS_PROTO_AUTO;
127     if( *p_access->psz_access )
128     {
129         if( !strncmp( p_access->psz_access, "mmsu", 4 ) )
130         {
131             i_proto = MMS_PROTO_UDP;
132         }
133         else if( !strncmp( p_access->psz_access, "mmst", 4 ) )
134         {
135             i_proto = MMS_PROTO_TCP;
136         }
137     }
138
139     /* connect */
140     if( i_proto == MMS_PROTO_AUTO )
141     {   /* first try with TCP and then UDP*/
142         if( ( i_status = MMSOpen( p_access, &p_sys->url, MMS_PROTO_TCP ) ) )
143         {
144             i_status = MMSOpen( p_access, &p_sys->url, MMS_PROTO_UDP );
145         }
146     }
147     else
148     {
149         i_status = MMSOpen( p_access, &p_sys->url, i_proto );
150     }
151
152     if( i_status )
153     {
154         msg_Err( p_access, "cannot connect to server" );
155         vlc_UrlClean( &p_sys->url );
156         free( p_sys );
157         return VLC_EGENERIC;
158     }
159
160     msg_Dbg( p_access, "connected to %s:%d", p_sys->url.psz_host, p_sys->url.i_port );
161     /*
162      * i_flags_broadcast
163      *  yy xx ?? ??
164      *  broadcast    yy=0x02, xx= 0x00
165      *  pre-recorded yy=0x01, xx= 0x80 if video, 0x00 no video
166      */
167     if( p_sys->i_packet_count <= 0 && p_sys->asfh.i_data_packets_count > 0 )
168     {
169         p_sys->i_packet_count = p_sys->asfh.i_data_packets_count;
170     }
171     if( p_sys->i_packet_count <= 0 || ( p_sys->i_flags_broadcast >> 24 ) == 0x02 )
172     {
173         p_sys->b_seekable = VLC_FALSE;
174     }
175     else
176     {
177         p_sys->b_seekable = VLC_TRUE;
178         p_access->info.i_size =
179             (uint64_t)p_sys->i_header +
180             (uint64_t)p_sys->i_packet_count * (uint64_t)p_sys->i_packet_length;
181     }
182
183     /* *** Start stream *** */
184     if( MMSStart( p_access, 0xffffffff ) < 0 )
185     {
186         msg_Err( p_access, "cannot start stream" );
187         E_(MMSTUClose) ( p_access );
188         return VLC_EGENERIC;
189     }
190     return VLC_SUCCESS;
191 }
192
193 /*****************************************************************************
194  * Close: free unused data structures
195  *****************************************************************************/
196 void E_(MMSTUClose)( access_t *p_access )
197 {
198     access_sys_t *p_sys = p_access->p_sys;
199
200     /* close connection with server */
201     MMSClose( p_access );
202
203     /* free memory */
204     vlc_UrlClean( &p_sys->url );
205
206     free( p_sys );
207 }
208
209 /*****************************************************************************
210  * Control:
211  *****************************************************************************/
212 static int Control( access_t *p_access, int i_query, va_list args )
213 {
214     access_sys_t *p_sys = p_access->p_sys;
215     vlc_bool_t   *pb_bool;
216     int          *pi_int;
217     int64_t      *pi_64;
218     int           i_int;
219     vlc_value_t  val;
220
221     switch( i_query )
222     {
223         /* */
224         case ACCESS_CAN_SEEK:
225             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
226             *pb_bool = p_sys->b_seekable;
227             break;
228
229         case ACCESS_CAN_FASTSEEK:
230         case ACCESS_CAN_PAUSE:
231             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
232             *pb_bool = VLC_FALSE;
233             break;
234
235         case ACCESS_CAN_CONTROL_PACE:
236             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
237
238 #if 0       /* Disable for now until we have a clock synchro algo
239              * which works with something else than MPEG over UDP */
240             *pb_bool = VLC_FALSE;
241 #endif
242             *pb_bool = VLC_TRUE;
243             break;
244
245         /* */
246         case ACCESS_GET_MTU:
247             pi_int = (int*)va_arg( args, int * );
248             *pi_int = 3 * p_sys->i_packet_length;
249             break;
250
251         case ACCESS_GET_PTS_DELAY:
252             pi_64 = (int64_t*)va_arg( args, int64_t * );
253             var_Get( p_access, "mms-caching", &val );
254             *pi_64 = (int64_t)var_GetInteger( p_access, "mms-caching" ) * I64C(1000);
255             break;
256
257         case ACCESS_GET_PRIVATE_ID_STATE:
258             i_int = (int)va_arg( args, int );
259             pb_bool = (vlc_bool_t *)va_arg( args, vlc_bool_t * );
260
261             if( i_int < 0 || i_int > 127 )
262                 return VLC_EGENERIC;
263             *pb_bool =  p_sys->asfh.stream[i_int].i_selected ? VLC_TRUE : VLC_FALSE;
264             break;
265
266         /* */
267         case ACCESS_SET_PAUSE_STATE:
268         case ACCESS_GET_TITLE_INFO:
269         case ACCESS_SET_TITLE:
270         case ACCESS_SET_SEEKPOINT:
271         case ACCESS_SET_PRIVATE_ID_STATE:
272         case ACCESS_GET_CONTENT_TYPE:
273             return VLC_EGENERIC;
274
275
276         default:
277             msg_Warn( p_access, "unimplemented query in control" );
278             return VLC_EGENERIC;
279
280     }
281     return VLC_SUCCESS;
282 }
283
284 /*****************************************************************************
285  * Seek: try to go at the right place
286  *****************************************************************************/
287 static int Seek( access_t * p_access, int64_t i_pos )
288 {
289     access_sys_t *p_sys = p_access->p_sys;
290     uint32_t    i_packet;
291     uint32_t    i_offset;
292     var_buffer_t buffer;
293
294     if( i_pos < 0 )
295         return VLC_EGENERIC;
296
297     if( i_pos < p_sys->i_header)
298     {
299
300         if( p_access->info.i_pos < p_sys->i_header )
301         {
302             /* no need to restart stream, it was already one
303              * or no stream was yet read */
304             p_access->info.i_pos = i_pos;
305             return VLC_SUCCESS;
306         }
307         else
308         {
309             i_packet = 0xffffffff;
310             i_offset = 0;
311         }
312     }
313     else
314     {
315         i_packet = ( i_pos - p_sys->i_header ) / p_sys->i_packet_length;
316         i_offset = ( i_pos - p_sys->i_header ) % p_sys->i_packet_length;
317     }
318     msg_Dbg( p_access, "seeking to "I64Fd " (packet:%d)", i_pos, i_packet );
319
320     MMSStop( p_access );
321     msg_Dbg( p_access, "stream stopped (seek)" );
322
323     /* *** restart stream *** */
324     var_buffer_initwrite( &buffer, 0 );
325     var_buffer_add64( &buffer, 0 ); /* seek point in second */
326     var_buffer_add32( &buffer, 0xffffffff );
327     var_buffer_add32( &buffer, i_packet ); // begin from start
328     var_buffer_add8( &buffer, 0xff ); // stream time limit
329     var_buffer_add8( &buffer, 0xff ); //  on 3bytes ...
330     var_buffer_add8( &buffer, 0xff ); //
331     var_buffer_add8( &buffer, 0x00 ); // don't use limit
332     var_buffer_add32( &buffer, p_sys->i_media_packet_id_type );
333
334     mms_CommandSend( p_access, 0x07, p_sys->i_command_level, 0x0001ffff,
335                      buffer.p_data, buffer.i_data );
336
337     var_buffer_free( &buffer );
338
339
340     while( !p_access->b_die )
341     {
342         mms_HeaderMediaRead( p_access, MMS_PACKET_CMD );
343         if( p_sys->i_command == 0x1e )
344         {
345             msg_Dbg( p_access, "received 0x1e (seek)" );
346             break;
347         }
348     }
349
350     while( !p_access->b_die )
351     {
352         mms_HeaderMediaRead( p_access, MMS_PACKET_CMD );
353         if( p_sys->i_command == 0x05 )
354         {
355             msg_Dbg( p_access, "received 0x05 (seek)" );
356             break;
357         }
358     }
359
360     /* get a packet */
361     mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA );
362     msg_Dbg( p_access, "Streaming restarted" );
363
364     p_sys->i_media_used += i_offset;
365     p_access->info.i_pos = i_pos;
366     p_access->info.b_eof = VLC_FALSE;
367
368     return VLC_SUCCESS;
369 }
370
371 /*****************************************************************************
372  * Read:
373  *****************************************************************************/
374 static ssize_t Read( access_t *p_access, uint8_t *p_buffer, size_t i_len )
375 {
376     access_sys_t *p_sys = p_access->p_sys;
377     size_t      i_data;
378     size_t      i_copy;
379
380     i_data = 0;
381
382     /* *** now send data if needed *** */
383     while( i_data < (size_t)i_len )
384     {
385         if( p_access->info.i_pos < p_sys->i_header )
386         {
387             i_copy = __MIN( i_len, p_sys->i_header - p_access->info.i_pos );
388             memcpy( &p_buffer[i_data], &p_sys->p_header[p_access->info.i_pos], i_copy );
389             i_data += i_copy;
390             p_access->info.i_pos += i_copy;
391         }
392         else if( p_sys->i_media_used < p_sys->i_media )
393         {
394             i_copy = __MIN( i_len - i_data ,
395                             p_sys->i_media - p_sys->i_media_used );
396             memcpy( &p_buffer[i_data], &p_sys->p_media[p_sys->i_media_used], i_copy );
397             i_data += i_copy;
398             p_sys->i_media_used += i_copy;
399             p_access->info.i_pos += i_copy;
400         }
401         else if( p_sys->p_media != NULL &&
402                  p_sys->i_media_used < p_sys->i_packet_length )
403         {
404             i_copy = __MIN( i_len - i_data,
405                             p_sys->i_packet_length - p_sys->i_media_used);
406             memset( &p_buffer[i_data], 0, i_copy );
407
408             i_data += i_copy;
409             p_sys->i_media_used += i_copy;
410             p_access->info.i_pos += i_copy;
411         }
412         else if( p_access->info.b_eof ||
413                  mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA ) < 0 )
414         {
415             break;
416         }
417     }
418
419     return i_data;
420 }
421
422 /****************************************************************************
423  * MMSOpen : Open a connection with the server over mmst or mmsu
424  ****************************************************************************/
425 static int MMSOpen( access_t  *p_access, vlc_url_t *p_url, int  i_proto )
426 {
427     access_sys_t *p_sys = p_access->p_sys;
428     int           b_udp = ( i_proto == MMS_PROTO_UDP ) ? 1 : 0;
429
430     var_buffer_t buffer;
431     char         tmp[4096];
432     uint16_t     *p;
433     int          i_server_version;
434     int          i_tool_version;
435     int          i_update_player_url;
436     int          i_encryption_type;
437     int          i;
438     int          i_streams;
439     int          i_first;
440
441
442     /* *** Open a TCP connection with server *** */
443     msg_Dbg( p_access, "waiting for connection..." );
444     p_sys->i_handle_tcp = net_ConnectTCP( p_access, p_url->psz_host, p_url->i_port );
445     if( p_sys->i_handle_tcp < 0 )
446     {
447         msg_Err( p_access, "failed to open a connection (tcp)" );
448         return VLC_EGENERIC;
449     }
450     msg_Dbg( p_access,
451              "connection(tcp) with \"%s:%d\" successful",
452              p_url->psz_host,
453              p_url->i_port );
454
455     /* *** Bind port if UDP protocol is selected *** */
456     if( b_udp )
457     {
458         if( net_GetSockAddress( p_sys->i_handle_tcp, p_sys->sz_bind_addr,
459                                 NULL ) )
460         {
461             net_Close( p_sys->i_handle_tcp );
462             return VLC_EGENERIC;
463         }
464
465         p_sys->i_handle_udp = net_ListenUDP1( (vlc_object_t *)p_access, p_sys->sz_bind_addr,
466                                               7000 );
467         if( p_sys->i_handle_udp < 0 )
468         {
469             msg_Err( p_access, "failed to open a connection (udp)" );
470             net_Close( p_sys->i_handle_tcp );
471             return VLC_EGENERIC;
472         }
473         msg_Dbg( p_access,
474                  "connection(udp) at \"%s:%d\" successful",
475                  p_sys->sz_bind_addr, 7000 );
476     }
477
478     /* *** Init context for mms prototcol *** */
479     E_( GenerateGuid )( &p_sys->guid );    /* used to identify client by server */
480     msg_Dbg( p_access,
481              "generated guid: "GUID_FMT,
482              GUID_PRINT( p_sys->guid ) );
483     p_sys->i_command_level = 1;          /* updated after 0x1A command */
484     p_sys->i_seq_num = 0;
485     p_sys->i_media_packet_id_type  = 0x04;
486     p_sys->i_header_packet_id_type = 0x02;
487     p_sys->i_proto = i_proto;
488     p_sys->i_packet_seq_num = 0;
489     p_sys->p_header = NULL;
490     p_sys->i_header = 0;
491     p_sys->p_media = NULL;
492     p_sys->i_media = 0;
493     p_sys->i_media_used = 0;
494
495     p_access->info.i_pos = 0;
496     p_sys->i_buffer_tcp = 0;
497     p_sys->i_buffer_udp = 0;
498     p_sys->p_cmd = NULL;
499     p_sys->i_cmd = 0;
500     p_access->info.b_eof = 0;
501
502     /* *** send command 1 : connection request *** */
503     var_buffer_initwrite( &buffer, 0 );
504     var_buffer_add16( &buffer, 0x001c );
505     var_buffer_add16( &buffer, 0x0003 );
506     sprintf( tmp,
507              "NSPlayer/7.0.0.1956; {"GUID_FMT"}; Host: %s",
508              GUID_PRINT( p_sys->guid ),
509              p_url->psz_host );
510     var_buffer_addUTF16( &buffer, tmp );
511
512     mms_CommandSend( p_access,
513                      0x01,          /* connexion request */
514                      0x00000000,    /* flags, FIXME */
515                      0x0004000b,    /* ???? */
516                      buffer.p_data,
517                      buffer.i_data );
518
519     if( mms_CommandRead( p_access, 0x01, 0 ) < 0 )
520     {
521         var_buffer_free( &buffer );
522         MMSClose( p_access );
523         return VLC_EGENERIC;
524     }
525
526     i_server_version = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 32 );
527     i_tool_version = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 36 );
528     i_update_player_url = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 40 );
529     i_encryption_type = GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 44 );
530     p = (uint16_t*)( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 48 );
531 #define GETUTF16( psz, size ) \
532     { \
533         int i; \
534         psz = malloc( size + 1); \
535         for( i = 0; i < size; i++ ) \
536         { \
537             psz[i] = p[i]; \
538         } \
539         psz[size] = '\0'; \
540         p += ( size ); \
541     }
542     GETUTF16( p_sys->psz_server_version, i_server_version );
543     GETUTF16( p_sys->psz_tool_version, i_tool_version );
544     GETUTF16( p_sys->psz_update_player_url, i_update_player_url );
545     GETUTF16( p_sys->psz_encryption_type, i_encryption_type );
546 #undef GETUTF16
547     msg_Dbg( p_access,
548              "0x01 --> server_version:\"%s\" tool_version:\"%s\" update_player_url:\"%s\" encryption_type:\"%s\"",
549              p_sys->psz_server_version,
550              p_sys->psz_tool_version,
551              p_sys->psz_update_player_url,
552              p_sys->psz_encryption_type );
553
554     /* *** should make an 18 command to make data timing *** */
555
556     /* *** send command 2 : transport protocol selection *** */
557     var_buffer_reinitwrite( &buffer, 0 );
558     var_buffer_add32( &buffer, 0x00000000 );
559     var_buffer_add32( &buffer, 0x000a0000 );
560     var_buffer_add32( &buffer, 0x00000002 );
561     if( b_udp )
562     {
563         sprintf( tmp,
564                  "\\\\%s\\UDP\\%d",
565                  p_sys->sz_bind_addr,
566                  7000 ); // FIXME
567     }
568     else
569     {
570         sprintf( tmp, "\\\\192.168.0.1\\TCP\\1242"  );
571     }
572     var_buffer_addUTF16( &buffer, tmp );
573     var_buffer_add16( &buffer, '0' );
574
575     mms_CommandSend( p_access,
576                      0x02,          /* connexion request */
577                      0x00000000,    /* flags, FIXME */
578                      0xffffffff,    /* ???? */
579                      buffer.p_data,
580                      buffer.i_data );
581
582     /* *** response from server, should be 0x02 or 0x03 *** */
583     mms_CommandRead( p_access, 0x02, 0x03 );
584     if( p_sys->i_command == 0x03 )
585     {
586         msg_Err( p_access,
587                  "%s protocol selection failed", b_udp ? "UDP" : "TCP" );
588         var_buffer_free( &buffer );
589         MMSClose( p_access );
590         return VLC_EGENERIC;
591     }
592     else if( p_sys->i_command != 0x02 )
593     {
594         msg_Warn( p_access, "received command isn't 0x02 in reponse to 0x02" );
595     }
596
597     /* *** send command 5 : media file name/path requested *** */
598     var_buffer_reinitwrite( &buffer, 0 );
599     var_buffer_add64( &buffer, 0 );
600     var_buffer_addUTF16( &buffer, p_url->psz_path );
601
602     mms_CommandSend( p_access,
603                      0x05,
604                      p_sys->i_command_level,
605                      0xffffffff,
606                      buffer.p_data,
607                      buffer.i_data );
608
609     /* *** wait for reponse *** */
610     mms_CommandRead( p_access, 0x1a, 0x06 );
611
612     /* test if server send 0x1A answer */
613     if( p_sys->i_command == 0x1A )
614     {
615         msg_Err( p_access, "id/password requested (not yet supported)" );
616         /*  FIXME */
617         var_buffer_free( &buffer );
618         MMSClose( p_access );
619         return VLC_EGENERIC;
620     }
621     if( p_sys->i_command != 0x06 )
622     {
623         msg_Err( p_access,
624                  "unknown answer (0x%x instead of 0x06)",
625                  p_sys->i_command );
626         var_buffer_free( &buffer );
627         MMSClose( p_access );
628         return( -1 );
629     }
630
631     /*  1 for file ok, 2 for authen ok */
632     switch( GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE ) )
633     {
634         case 0x0001:
635             msg_Dbg( p_access, "media file name/path accepted" );
636             break;
637         case 0x0002:
638             msg_Dbg( p_access, "authentication accepted" );
639             break;
640         case -1:
641         default:
642         msg_Err( p_access, "error while asking for file %d",
643                  GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE ) );
644         var_buffer_free( &buffer );
645         MMSClose( p_access );
646         return VLC_EGENERIC;
647     }
648
649     p_sys->i_flags_broadcast =
650         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 12 );
651     p_sys->i_media_length =
652         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 24 );
653     p_sys->i_packet_length =
654         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 44 );
655     p_sys->i_packet_count =
656         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 48 );
657     p_sys->i_max_bit_rate =
658         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 56 );
659     p_sys->i_header_size =
660         GetDWLE( p_sys->p_cmd + MMS_CMD_HEADERSIZE + 60 );
661
662     msg_Dbg( p_access,
663              "answer 0x06 flags:0x%8.8x media_length:%us "
664              "packet_length:%ul packet_count:%d max_bit_rate:%d "
665              "header_size:%d",
666              p_sys->i_flags_broadcast,
667              p_sys->i_media_length,
668              (unsigned)p_sys->i_packet_length,
669              p_sys->i_packet_count,
670              p_sys->i_max_bit_rate,
671              p_sys->i_header_size );
672
673     /* *** send command 15 *** */
674
675     var_buffer_reinitwrite( &buffer, 0 );
676     var_buffer_add32( &buffer, 0 );
677     var_buffer_add32( &buffer, 0x8000 );
678     var_buffer_add32( &buffer, 0xffffffff );
679     var_buffer_add32( &buffer, 0x00 );
680     var_buffer_add32( &buffer, 0x00 );
681     var_buffer_add32( &buffer, 0x00 );
682     var_buffer_add64( &buffer, (((uint64_t)0x40ac2000)<<32) );
683     var_buffer_add32( &buffer, p_sys->i_header_packet_id_type );
684     var_buffer_add32( &buffer, 0x00 );
685     mms_CommandSend( p_access, 0x15, p_sys->i_command_level, 0x00,
686                      buffer.p_data, buffer.i_data );
687
688     /* *** wait for reponse *** */
689     /* Commented out because it fails on some stream (no 0x11 answer) */
690 #if 0
691     mms_CommandRead( p_access, 0x11, 0 );
692
693     if( p_sys->i_command != 0x11 )
694     {
695         msg_Err( p_access,
696                  "unknown answer (0x%x instead of 0x11)",
697                  p_sys->i_command );
698         var_buffer_free( &buffer );
699         MMSClose( p_access );
700         return( -1 );
701     }
702 #endif
703
704     /* *** now read header packet *** */
705     /* XXX could be split over multiples packets */
706     msg_Dbg( p_access, "reading header" );
707     for( ;; )
708     {
709         if( mms_HeaderMediaRead( p_access, MMS_PACKET_HEADER ) < 0 )
710         {
711             msg_Err( p_access, "cannot receive header" );
712             var_buffer_free( &buffer );
713             MMSClose( p_access );
714             return VLC_EGENERIC;
715         }
716         if( p_sys->i_header >= p_sys->i_header_size )
717         {
718             msg_Dbg( p_access,
719                      "header complete(%d)",
720                      p_sys->i_header );
721             break;
722         }
723         msg_Dbg( p_access,
724                  "header incomplete (%d/%d), reading more",
725                  p_sys->i_header,
726                  p_sys->i_header_size );
727     }
728
729     /* *** parse header and get stream and their id *** */
730     /* get all streams properties,
731      *
732      * TODO : stream bitrates properties(optional)
733      *        and bitrate mutual exclusion(optional) */
734     E_( asf_HeaderParse )( &p_sys->asfh,
735                            p_sys->p_header, p_sys->i_header );
736     E_( asf_StreamSelect)( &p_sys->asfh,
737                            var_CreateGetInteger( p_access, "mms-maxbitrate" ),
738                            var_CreateGetInteger( p_access, "mms-all" ),
739                            var_CreateGetInteger( p_access, "audio" ),
740                            var_CreateGetInteger( p_access, "video" ) );
741
742     /* *** now select stream we want to receive *** */
743     /* TODO take care of stream bitrate TODO */
744     i_streams = 0;
745     i_first = -1;
746     var_buffer_reinitwrite( &buffer, 0 );
747     /* for now, select first audio and video stream */
748     for( i = 1; i < 128; i++ )
749     {
750
751         if( p_sys->asfh.stream[i].i_cat != ASF_STREAM_UNKNOWN )
752         {
753             i_streams++;
754             if( i_first != -1 )
755             {
756                 var_buffer_add16( &buffer, 0xffff );
757                 var_buffer_add16( &buffer, i );
758             }
759             else
760             {
761                 i_first = i;
762             }
763             if( p_sys->asfh.stream[i].i_selected )
764             {
765                 var_buffer_add16( &buffer, 0x0000 );
766                 msg_Info( p_access,
767                           "selecting stream[0x%x] %s (%d kb/s)",
768                           i,
769                           ( p_sys->asfh.stream[i].i_cat == ASF_STREAM_AUDIO  ) ?
770                                                   "audio" : "video" ,
771                           p_sys->asfh.stream[i].i_bitrate / 1024);
772             }
773             else
774             {
775                 var_buffer_add16( &buffer, 0x0002 );
776                 msg_Info( p_access,
777                           "ignoring stream[0x%x] %s (%d kb/s)",
778                           i,
779                           ( p_sys->asfh.stream[i].i_cat == ASF_STREAM_AUDIO  ) ?
780                                     "audio" : "video" ,
781                           p_sys->asfh.stream[i].i_bitrate / 1024);
782
783             }
784         }
785     }
786
787     if( i_streams == 0 )
788     {
789         msg_Err( p_access, "cannot find any stream" );
790         var_buffer_free( &buffer );
791         MMSClose( p_access );
792         return VLC_EGENERIC;
793     }
794     mms_CommandSend( p_access, 0x33,
795                      i_streams,
796                      0xffff | ( i_first << 16 ),
797                      buffer.p_data, buffer.i_data );
798
799     mms_CommandRead( p_access, 0x21, 0 );
800     if( p_sys->i_command != 0x21 )
801     {
802         msg_Err( p_access,
803                  "unknown answer (0x%x instead of 0x21)",
804                  p_sys->i_command );
805         var_buffer_free( &buffer );
806         MMSClose( p_access );
807         return VLC_EGENERIC;
808     }
809
810
811     var_buffer_free( &buffer );
812
813     msg_Info( p_access, "connection successful" );
814
815     return VLC_SUCCESS;
816 }
817
818 /****************************************************************************
819  * MMSStart : Start streaming
820  ****************************************************************************/
821 static int MMSStart( access_t  *p_access, uint32_t i_packet )
822 {
823     access_sys_t        *p_sys = p_access->p_sys;
824     var_buffer_t    buffer;
825
826     /* *** start stream from packet 0 *** */
827     var_buffer_initwrite( &buffer, 0 );
828     var_buffer_add64( &buffer, 0 ); /* seek point in second */
829     var_buffer_add32( &buffer, 0xffffffff );
830     var_buffer_add32( &buffer, i_packet ); // begin from start
831     var_buffer_add8( &buffer, 0xff ); // stream time limit
832     var_buffer_add8( &buffer, 0xff ); //  on 3bytes ...
833     var_buffer_add8( &buffer, 0xff ); //
834     var_buffer_add8( &buffer, 0x00 ); // don't use limit
835     var_buffer_add32( &buffer, p_sys->i_media_packet_id_type );
836
837     mms_CommandSend( p_access, 0x07, p_sys->i_command_level, 0x0001ffff,
838                      buffer.p_data, buffer.i_data );
839
840     var_buffer_free( &buffer );
841
842     mms_CommandRead( p_access, 0x05, 0 );
843
844     if( p_sys->i_command != 0x05 )
845     {
846         msg_Err( p_access,
847                  "unknown answer (0x%x instead of 0x05)",
848                  p_sys->i_command );
849         return( -1 );
850     }
851     else
852     {
853         /* get a packet */
854         mms_HeaderMediaRead( p_access, MMS_PACKET_MEDIA );
855         msg_Dbg( p_access, "streaming started" );
856         return( 0 );
857     }
858 }
859
860 /****************************************************************************
861  * MMSStop : Stop streaming
862  ****************************************************************************/
863 static int MMSStop( access_t  *p_access )
864 {
865     access_sys_t *p_sys = p_access->p_sys;
866
867     /* *** stop stream but keep connection alive *** */
868     mms_CommandSend( p_access,
869                      0x09,
870                      p_sys->i_command_level,
871                      0x001fffff,
872                      NULL, 0 );
873     return( 0 );
874 }
875
876 /****************************************************************************
877  * MMSClose : Close streaming and connection
878  ****************************************************************************/
879 static void MMSClose( access_t  *p_access )
880 {
881     access_sys_t        *p_sys = p_access->p_sys;
882
883     msg_Dbg( p_access, "Connection closed" );
884
885     /* *** tell server that we will disconnect *** */
886     mms_CommandSend( p_access,
887                      0x0d,
888                      p_sys->i_command_level,
889                      0x00000001,
890                      NULL, 0 );
891
892     /* *** close sockets *** */
893     net_Close( p_sys->i_handle_tcp );
894     if( p_sys->i_proto == MMS_PROTO_UDP )
895     {
896         net_Close( p_sys->i_handle_udp );
897     }
898
899     FREENULL( p_sys->p_cmd );
900     FREENULL( p_sys->p_media );
901     FREENULL( p_sys->p_header );
902
903     FREENULL( p_sys->psz_server_version );
904     FREENULL( p_sys->psz_tool_version );
905     FREENULL( p_sys->psz_update_player_url );
906     FREENULL( p_sys->psz_encryption_type );
907 }
908
909 /****************************************************************************
910  *
911  * MMS specific functions
912  *
913  ****************************************************************************/
914 static int mms_CommandSend( access_t *p_access, int i_command,
915                             uint32_t i_prefix1, uint32_t i_prefix2,
916                             uint8_t *p_data, int i_data_old )
917 {
918     var_buffer_t buffer;
919     access_sys_t *p_sys = p_access->p_sys;
920     int i_data_by8, i_ret;
921     int i_data = i_data_old;
922
923     while( i_data & 0x7 ) i_data++;
924     i_data_by8 = i_data >> 3;
925
926     /* first init buffer */
927     var_buffer_initwrite( &buffer, 0 );
928
929     var_buffer_add32( &buffer, 0x00000001 );    /* start sequence */
930     var_buffer_add32( &buffer, 0xB00BFACE );
931     /* size after protocol type */
932     var_buffer_add32( &buffer, i_data + MMS_CMD_HEADERSIZE - 16 );
933     var_buffer_add32( &buffer, 0x20534d4d );    /* protocol "MMS " */
934     var_buffer_add32( &buffer, i_data_by8 + 4 );
935     var_buffer_add32( &buffer, p_sys->i_seq_num ); p_sys->i_seq_num++;
936     var_buffer_add64( &buffer, 0 );
937     var_buffer_add32( &buffer, i_data_by8 + 2 );
938     var_buffer_add32( &buffer, 0x00030000 | i_command ); /* dir | command */
939     var_buffer_add32( &buffer, i_prefix1 );    /* command specific */
940     var_buffer_add32( &buffer, i_prefix2 );    /* command specific */
941
942     /* specific command data */
943     if( p_data && i_data > 0 )
944     {
945         var_buffer_addmemory( &buffer, p_data, i_data_old );
946     }
947
948     /* Append padding to the command data */
949     var_buffer_add64( &buffer, 0 );
950
951     /* send it */
952     i_ret = net_Write( p_access, p_sys->i_handle_tcp, NULL, buffer.p_data,
953                        buffer.i_data - ( 8 - ( i_data - i_data_old ) ) );
954     if( i_ret != buffer.i_data - ( 8 - ( i_data - i_data_old ) ) )
955     {
956         msg_Err( p_access, "failed to send command" );
957         return VLC_EGENERIC;
958     }
959
960     var_buffer_free( &buffer );
961     return VLC_SUCCESS;
962 }
963
964 static int NetFillBuffer( access_t *p_access )
965 {
966 #ifdef UNDER_CE
967     return -1;
968
969 #else
970     access_sys_t    *p_sys = p_access->p_sys;
971     int             i_ret;
972     struct pollfd   ufd[2];
973     unsigned        timeout, nfd;
974
975     /* FIXME when using udp */
976     ssize_t i_tcp, i_udp;
977     ssize_t i_tcp_read, i_udp_read;
978     int i_try = 0;
979
980     i_tcp = MMS_BUFFER_SIZE/2 - p_sys->i_buffer_tcp;
981
982     if( p_sys->i_proto == MMS_PROTO_UDP )
983     {
984         i_udp = MMS_BUFFER_SIZE/2 - p_sys->i_buffer_udp;
985     }
986     else
987     {
988         i_udp = 0;  /* there isn't udp socket */
989     }
990
991     if( ( i_udp <= 0 ) && ( i_tcp <= 0 ) )
992     {
993         msg_Warn( p_access, "nothing to read %d:%d", (int)i_tcp, (int)i_udp );
994         return 0;
995     }
996     else
997     {
998         /* msg_Warn( p_access, "ask for tcp:%d udp:%d", i_tcp, i_udp ); */
999     }
1000
1001     /* Find if some data is available */
1002     do
1003     {
1004         i_try++;
1005
1006         /* Initialize file descriptor set */
1007         memset (ufd, 0, sizeof (ufd));
1008         nfd = 0;
1009
1010         if( i_tcp > 0 )
1011         {
1012             ufd[nfd].fd = p_sys->i_handle_tcp;
1013             ufd[nfd].events = POLLIN;
1014             nfd++;
1015         }
1016         if( i_udp > 0 )
1017         {
1018             ufd[nfd].fd = p_sys->i_handle_tcp;
1019             ufd[nfd].events = POLLIN;
1020             nfd++;
1021         }
1022
1023         /* We'll wait 0.5 second if nothing happens */
1024         timeout = 500;
1025
1026         if( i_try > 3 && (p_sys->i_buffer_tcp > 0 || p_sys->i_buffer_udp > 0) )
1027         {
1028             return -1;
1029         }
1030
1031         if( p_access->b_die || p_access->b_error ) return -1;
1032
1033         //msg_Dbg( p_access, "NetFillBuffer: trying again (select)" );
1034
1035     } while( !(i_ret = poll( ufd, nfd, timeout)) ||
1036              (i_ret < 0 && errno == EINTR) );
1037
1038     if( i_ret < 0 )
1039     {
1040         msg_Err( p_access, "network poll error (%m)" );
1041         return -1;
1042     }
1043
1044     i_tcp_read = i_udp_read = 0;
1045
1046     if( ( i_tcp > 0 ) && ufd[0].revents )
1047     {
1048         i_tcp_read =
1049             recv( p_sys->i_handle_tcp,
1050                   p_sys->buffer_tcp + p_sys->i_buffer_tcp,
1051                   i_tcp + MMS_BUFFER_SIZE/2, 0 );
1052     }
1053
1054     if( i_udp > 0 && ufd[i_tcp > 0].revents )
1055     {
1056         i_udp_read = recv( p_sys->i_handle_udp,
1057                            p_sys->buffer_udp + p_sys->i_buffer_udp,
1058                            i_udp + MMS_BUFFER_SIZE/2, 0 );
1059     }
1060
1061 #ifdef MMS_DEBUG
1062     if( p_sys->i_proto == MMS_PROTO_UDP )
1063     {
1064         msg_Dbg( p_access, "filling buffer TCP:%d+%d UDP:%d+%d",
1065                  p_sys->i_buffer_tcp, i_tcp_read,
1066                  p_sys->i_buffer_udp, i_udp_read );
1067     }
1068     else
1069     {
1070         msg_Dbg( p_access, "filling buffer TCP:%d+%d",
1071                  p_sys->i_buffer_tcp, i_tcp_read );
1072     }
1073 #endif
1074
1075     if( i_tcp_read > 0 ) p_sys->i_buffer_tcp += i_tcp_read;
1076     if( i_udp_read > 0 ) p_sys->i_buffer_udp += i_udp_read;
1077
1078     return i_tcp_read + i_udp_read;
1079 #endif
1080 }
1081
1082 static int  mms_ParseCommand( access_t *p_access,
1083                               uint8_t *p_data,
1084                               int i_data,
1085                               int *pi_used )
1086 {
1087  #define GET32( i_pos ) \
1088     ( p_sys->p_cmd[i_pos] + ( p_sys->p_cmd[i_pos +1] << 8 ) + \
1089       ( p_sys->p_cmd[i_pos + 2] << 16 ) + \
1090       ( p_sys->p_cmd[i_pos + 3] << 24 ) )
1091
1092     access_sys_t        *p_sys = p_access->p_sys;
1093     int         i_length;
1094     uint32_t    i_id;
1095
1096     if( p_sys->p_cmd )
1097     {
1098         free( p_sys->p_cmd );
1099     }
1100     p_sys->i_cmd = i_data;
1101     p_sys->p_cmd = malloc( i_data );
1102     memcpy( p_sys->p_cmd, p_data, i_data );
1103
1104     *pi_used = i_data; /* by default */
1105
1106     if( i_data < MMS_CMD_HEADERSIZE )
1107     {
1108         msg_Warn( p_access, "truncated command (header incomplete)" );
1109         p_sys->i_command = 0;
1110         return -1;
1111     }
1112     i_id =  GetDWLE( p_data + 4 );
1113     i_length = GetDWLE( p_data + 8 ) + 16;
1114
1115     if( i_id != 0xb00bface )
1116     {
1117         msg_Err( p_access,
1118                  "incorrect command header (0x%x)", i_id );
1119         p_sys->i_command = 0;
1120         return -1;
1121     }
1122
1123     if( i_length > p_sys->i_cmd )
1124     {
1125         msg_Warn( p_access,
1126                   "truncated command (missing %d bytes)",
1127                    i_length - i_data  );
1128         p_sys->i_command = 0;
1129         return -1;
1130     }
1131     else if( i_length < p_sys->i_cmd )
1132     {
1133         p_sys->i_cmd = i_length;
1134         *pi_used = i_length;
1135     }
1136
1137     msg_Dbg( p_access,
1138              "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",
1139              GET32( 0 ),
1140              GET32( 4 ),
1141              GET32( 8 ),
1142              /* 12: protocol type "MMS " */
1143              GET32( 16 ),
1144              GET32( 20 ),
1145              /* 24: unknown (0) */
1146              /* 28: unknown (0) */
1147              GET32( 32 ),
1148              GET32( 36 )
1149              /* 40: switches */
1150              /* 44: extra */ );
1151
1152     p_sys->i_command = GET32( 36 ) & 0xffff;
1153 #undef GET32
1154
1155     return MMS_PACKET_CMD;
1156 }
1157
1158 static int  mms_ParsePacket( access_t *p_access,
1159                              uint8_t *p_data, size_t i_data,
1160                              int *pi_used )
1161 {
1162     access_sys_t        *p_sys = p_access->p_sys;
1163     int i_packet_seq_num;
1164     size_t i_packet_length;
1165     uint32_t i_packet_id;
1166
1167     uint8_t  *p_packet;
1168
1169
1170     *pi_used = i_data; /* default */
1171     if( i_data <= 8 )
1172     {
1173         msg_Warn( p_access, "truncated packet (header incomplete)" );
1174         return -1;
1175     }
1176
1177     i_packet_id = p_data[4];
1178     i_packet_seq_num = GetDWLE( p_data );
1179     i_packet_length = GetWLE( p_data + 6 );
1180
1181     //msg_Warn( p_access, "------->i_packet_length=%d, i_data=%d", i_packet_length, i_data );
1182
1183     if( i_packet_length > i_data || i_packet_length <= 8)
1184     {
1185      /*   msg_Dbg( p_access,
1186                  "truncated packet (Declared %d bytes, Actual %d bytes)",
1187                  i_packet_length, i_data  ); */
1188         *pi_used = 0;
1189         return -1;
1190     }
1191     else if( i_packet_length < i_data )
1192     {
1193         *pi_used = i_packet_length;
1194     }
1195
1196     if( i_packet_id == 0xff )
1197     {
1198         msg_Warn( p_access,
1199                   "receive MMS UDP pair timing" );
1200         return( MMS_PACKET_UDP_TIMING );
1201     }
1202
1203     if( i_packet_id != p_sys->i_header_packet_id_type &&
1204         i_packet_id != p_sys->i_media_packet_id_type )
1205     {
1206         msg_Warn( p_access, "incorrect Packet Id Type (0x%x)", i_packet_id );
1207         return -1;
1208     }
1209
1210     /* we now have a media or a header packet */
1211     p_packet = malloc( i_packet_length - 8 ); // don't bother with preheader
1212     memcpy( p_packet, p_data + 8, i_packet_length - 8 );
1213
1214     if( i_packet_seq_num != p_sys->i_packet_seq_num )
1215     {
1216 #if 0
1217         /* FIXME for udp could be just wrong order ? */
1218         msg_Warn( p_access,
1219                   "detected packet lost (%d != %d)",
1220                   i_packet_seq_num,
1221                   p_sys->i_packet_seq_num );
1222 #endif
1223     }
1224     p_sys->i_packet_seq_num = i_packet_seq_num + 1;
1225
1226     if( i_packet_id == p_sys->i_header_packet_id_type )
1227     {
1228         if( p_sys->p_header )
1229         {
1230             p_sys->p_header = realloc( p_sys->p_header,
1231                                           p_sys->i_header + i_packet_length - 8 );
1232             memcpy( &p_sys->p_header[p_sys->i_header],
1233                     p_packet,
1234                     i_packet_length - 8 );
1235             p_sys->i_header += i_packet_length - 8;
1236
1237             free( p_packet );
1238         }
1239         else
1240         {
1241             p_sys->p_header = p_packet;
1242             p_sys->i_header = i_packet_length - 8;
1243         }
1244 /*        msg_Dbg( p_access,
1245                  "receive header packet (%d bytes)",
1246                  i_packet_length - 8 ); */
1247
1248         return MMS_PACKET_HEADER;
1249     }
1250     else
1251     {
1252         FREENULL( p_sys->p_media );
1253         p_sys->p_media = p_packet;
1254         p_sys->i_media = i_packet_length - 8;
1255         p_sys->i_media_used = 0;
1256 /*        msg_Dbg( p_access,
1257                  "receive media packet (%d bytes)",
1258                  i_packet_length - 8 ); */
1259
1260         return MMS_PACKET_MEDIA;
1261     }
1262 }
1263
1264 static int mms_ReceivePacket( access_t *p_access )
1265 {
1266     access_sys_t *p_sys = p_access->p_sys;
1267     int i_packet_tcp_type;
1268     int i_packet_udp_type;
1269
1270     for( ;; )
1271     {
1272         vlc_bool_t b_refill = VLC_TRUE;
1273
1274         /* first if we need to refill buffer */
1275         if( p_sys->i_buffer_tcp >= MMS_CMD_HEADERSIZE )
1276         {
1277             if( GetDWLE( p_sys->buffer_tcp + 4 ) == 0xb00bface  )
1278             {
1279                 if( GetDWLE( p_sys->buffer_tcp + 8 ) + 16 <=
1280                     (uint32_t)p_sys->i_buffer_tcp )
1281                 {
1282                     b_refill = VLC_FALSE;
1283                 }
1284             }
1285             else if( GetWLE( p_sys->buffer_tcp + 6 ) <= p_sys->i_buffer_tcp )
1286             {
1287                 b_refill = VLC_FALSE;
1288             }
1289         }
1290         if( p_sys->i_proto == MMS_PROTO_UDP && p_sys->i_buffer_udp >= 8 &&
1291             GetWLE( p_sys->buffer_udp + 6 ) <= p_sys->i_buffer_udp )
1292         {
1293             b_refill = VLC_FALSE;
1294         }
1295
1296         if( b_refill && NetFillBuffer( p_access ) < 0 )
1297         {
1298             msg_Warn( p_access, "cannot fill buffer" );
1299             return -1;
1300         }
1301
1302         i_packet_tcp_type = -1;
1303         i_packet_udp_type = -1;
1304
1305         if( p_sys->i_buffer_tcp > 0 )
1306         {
1307             int i_used;
1308
1309             if( GetDWLE( p_sys->buffer_tcp + 4 ) == 0xb00bface )
1310             {
1311                 i_packet_tcp_type =
1312                     mms_ParseCommand( p_access, p_sys->buffer_tcp,
1313                                       p_sys->i_buffer_tcp, &i_used );
1314
1315             }
1316             else
1317             {
1318                 i_packet_tcp_type =
1319                     mms_ParsePacket( p_access, p_sys->buffer_tcp,
1320                                      p_sys->i_buffer_tcp, &i_used );
1321             }
1322             if( i_used > 0 && i_used < MMS_BUFFER_SIZE )
1323             {
1324                 memmove( p_sys->buffer_tcp, p_sys->buffer_tcp + i_used,
1325                          MMS_BUFFER_SIZE - i_used );
1326             }
1327             p_sys->i_buffer_tcp -= i_used;
1328         }
1329         else if( p_sys->i_buffer_udp > 0 )
1330         {
1331             int i_used;
1332
1333             i_packet_udp_type =
1334                 mms_ParsePacket( p_access, p_sys->buffer_udp,
1335                                  p_sys->i_buffer_udp, &i_used );
1336
1337             if( i_used > 0 && i_used < MMS_BUFFER_SIZE )
1338             {
1339                 memmove( p_sys->buffer_udp, p_sys->buffer_udp + i_used,
1340                          MMS_BUFFER_SIZE - i_used );
1341             }
1342             p_sys->i_buffer_udp -= i_used;
1343         }
1344
1345         if( i_packet_tcp_type == MMS_PACKET_CMD && p_sys->i_command == 0x1b )
1346         {
1347             mms_CommandSend( p_access, 0x1b, 0, 0, NULL, 0 );
1348             i_packet_tcp_type = -1;
1349         }
1350
1351         if( i_packet_tcp_type != -1 )
1352         {
1353             return i_packet_tcp_type;
1354         }
1355         else if( i_packet_udp_type != -1 )
1356         {
1357             return i_packet_udp_type;
1358         }
1359     }
1360 }
1361
1362 static int mms_ReceiveCommand( access_t *p_access )
1363 {
1364     access_sys_t *p_sys = p_access->p_sys;
1365
1366     for( ;; )
1367     {
1368         int i_used;
1369         int i_status;
1370
1371         if( NetFillBuffer( p_access ) < 0 )
1372         {
1373             msg_Warn( p_access, "cannot fill buffer" );
1374             return VLC_EGENERIC;
1375         }
1376         if( p_sys->i_buffer_tcp > 0 )
1377         {
1378             i_status = mms_ParseCommand( p_access, p_sys->buffer_tcp,
1379                                          p_sys->i_buffer_tcp, &i_used );
1380             if( i_used < MMS_BUFFER_SIZE )
1381             {
1382                 memmove( p_sys->buffer_tcp, p_sys->buffer_tcp + i_used,
1383                          MMS_BUFFER_SIZE - i_used );
1384             }
1385             p_sys->i_buffer_tcp -= i_used;
1386
1387             if( i_status < 0 )
1388             {
1389                 return VLC_EGENERIC;
1390             }
1391
1392             if( p_sys->i_command == 0x1b )
1393             {
1394                 mms_CommandSend( p_access, 0x1b, 0, 0, NULL, 0 );
1395             }
1396             else
1397             {
1398                 break;
1399             }
1400         }
1401         else
1402         {
1403             return VLC_EGENERIC;
1404         }
1405     }
1406
1407     return VLC_SUCCESS;
1408 }
1409
1410 #define MMS_RETRY_MAX       10
1411 #define MMS_RETRY_SLEEP     50000
1412
1413 static int mms_CommandRead( access_t *p_access, int i_command1,
1414                             int i_command2 )
1415 {
1416     access_sys_t *p_sys = p_access->p_sys;
1417     int i_count;
1418     int i_status;
1419
1420     for( i_count = 0; i_count < MMS_RETRY_MAX; )
1421     {
1422         i_status = mms_ReceiveCommand( p_access );
1423         if( i_status < 0 || p_sys->i_command == 0 )
1424         {
1425             i_count++;
1426             msleep( MMS_RETRY_SLEEP );
1427         }
1428         else if( i_command1 == 0 && i_command2 == 0)
1429         {
1430             return VLC_SUCCESS;
1431         }
1432         else if( p_sys->i_command == i_command1 ||
1433                  p_sys->i_command == i_command2 )
1434         {
1435             return VLC_SUCCESS;
1436         }
1437         else
1438         {
1439             switch( p_sys->i_command )
1440             {
1441                 case 0x03:
1442                     msg_Warn( p_access, "socket closed by server" );
1443                     p_access->info.b_eof = 1;
1444                     return VLC_EGENERIC;
1445                 case 0x1e:
1446                     msg_Warn( p_access, "end of media stream" );
1447                     p_access->info.b_eof = 1;
1448                     return VLC_EGENERIC;
1449                 default:
1450                     break;
1451             }
1452         }
1453     }
1454     msg_Warn( p_access, "failed to receive command (aborting)" );
1455
1456     return VLC_EGENERIC;
1457 }
1458
1459
1460 static int mms_HeaderMediaRead( access_t *p_access, int i_type )
1461 {
1462     access_sys_t *p_sys = p_access->p_sys;
1463     int          i_count;
1464
1465     for( i_count = 0; i_count < MMS_RETRY_MAX; )
1466     {
1467         int i_status;
1468
1469         if( p_access->b_die )
1470             return -1;
1471
1472         i_status = mms_ReceivePacket( p_access );
1473         if( i_status < 0 )
1474         {
1475             i_count++;
1476             msg_Warn( p_access, "cannot receive header (%d/%d)",
1477                       i_count, MMS_RETRY_MAX );
1478             msleep( MMS_RETRY_SLEEP );
1479         }
1480         else if( i_status == i_type || i_type == MMS_PACKET_ANY )
1481         {
1482             return i_type;
1483         }
1484         else if( i_status == MMS_PACKET_CMD )
1485         {
1486             switch( p_sys->i_command )
1487             {
1488                 case 0x03:
1489                     msg_Warn( p_access, "socket closed by server" );
1490                     p_access->info.b_eof = 1;
1491                     return -1;
1492                 case 0x1e:
1493                     msg_Warn( p_access, "end of media stream" );
1494                     p_access->info.b_eof = 1;
1495                     return -1;
1496                 case 0x20:
1497                     /* XXX not too dificult to be done EXCEPT that we
1498                      * need to restart demuxer... and I don't see how we
1499                      * could do that :p */
1500                     msg_Err( p_access,
1501                              "reinitialization needed --> unsupported" );
1502                     p_access->info.b_eof = VLC_TRUE;
1503                     return -1;
1504                 default:
1505                     break;
1506             }
1507         }
1508     }
1509
1510     msg_Err( p_access, "cannot receive %s (aborting)",
1511              ( i_type == MMS_PACKET_HEADER ) ? "header" : "media data" );
1512     return -1;
1513 }
1514