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