]> git.sesse.net Git - vlc/blob - src/stream_output/sap.c
10fddd48e141b1b835fd993112b08626107dca7c
[vlc] / src / stream_output / sap.c
1 /*****************************************************************************
2  * sap.c : SAP announce handler
3  *****************************************************************************
4  * Copyright (C) 2002-2005 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Clément Stenac <zorglub@videolan.org>
8  *          Rémi Denis-Courmont <rem # videolan.org>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
23  *****************************************************************************/
24
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 #include <stdlib.h>                                                /* free() */
29 #include <stdio.h>                                              /* sprintf() */
30 #include <string.h>                                            /* strerror() */
31 #include <ctype.h>                                  /* tolower(), isxdigit() */
32
33 #include <vlc/vlc.h>
34 #include <vlc/sout.h>
35
36 #include "network.h"
37 #if defined( WIN32 ) || defined( UNDER_CE )
38 #   if defined(UNDER_CE) && defined(sockaddr_storage)
39 #       undef sockaddr_storage
40 #   endif
41 #   include <winsock2.h>
42 #   include <ws2tcpip.h>
43 #else
44 #   include <netdb.h>
45 #endif
46 #include "charset.h"
47
48 /* SAP is always on that port */
49 #define SAP_PORT 9875
50
51 #define DEFAULT_PORT "1234"
52
53 #undef EXTRA_DEBUG
54
55 /* SAP Specific structures */
56
57 /* 100ms */
58 #define SAP_IDLE ((mtime_t)(0.100*CLOCK_FREQ))
59 #define SAP_MAX_BUFFER 65534
60 #define MIN_INTERVAL 2
61 #define MAX_INTERVAL 300
62
63 /* A SAP announce address. For each of these, we run the
64  * control flow algorithm */
65 struct sap_address_t
66 {
67     char *psz_address;
68     int i_port;
69     int i_rfd; /* Read socket */
70     int i_wfd; /* Write socket */
71
72     /* Used for flow control */
73     mtime_t t1;
74     vlc_bool_t b_enabled;
75     vlc_bool_t b_ready;
76     int i_interval;
77     int i_buff;
78     int i_limit;
79 };
80
81 /*****************************************************************************
82  * Local prototypes
83  *****************************************************************************/
84 static void RunThread( vlc_object_t *p_this);
85 static int CalculateRate( sap_handler_t *p_sap, sap_address_t *p_address );
86 static int SDPGenerate( sap_handler_t *p_sap, session_descriptor_t *p_session );
87
88 static int announce_SendSAPAnnounce( sap_handler_t *p_sap,
89                                      sap_session_t *p_session );
90
91
92 static int announce_SAPAnnounceAdd( sap_handler_t *p_sap,
93                              session_descriptor_t *p_session,
94                              announce_method_t *p_method );
95
96 static int announce_SAPAnnounceDel( sap_handler_t *p_sap,
97                              session_descriptor_t *p_session );
98 static char *convert_to_utf8( struct sap_handler_t *p_this, char *psz_local );
99
100 #define FREE( p ) if( p ) { free( p ); (p) = NULL; }
101
102
103 /**
104  * Create the SAP handler
105  *
106  * \param p_announce the parent announce_handler
107  * \return the newly created SAP handler or NULL on error
108  */
109 sap_handler_t *announce_SAPHandlerCreate( announce_handler_t *p_announce )
110 {
111     sap_handler_t *p_sap;
112     char *psz_charset;
113
114     p_sap = vlc_object_create( p_announce, sizeof( sap_handler_t ) );
115
116     if( !p_sap )
117     {
118         msg_Err( p_announce, "out of memory" );
119         return NULL;
120     }
121
122     vlc_mutex_init( p_sap, &p_sap->object_lock );
123
124     vlc_current_charset( &psz_charset );
125     p_sap->iconvHandle = vlc_iconv_open( "UTF-8", psz_charset );
126     free( psz_charset );
127     if( p_sap->iconvHandle == (vlc_iconv_t)(-1) )
128     {
129         msg_Warn( p_sap, "Unable to do requested conversion" );
130     }
131
132     p_sap->pf_add = announce_SAPAnnounceAdd;
133     p_sap->pf_del = announce_SAPAnnounceDel;
134
135     p_sap->i_sessions = 0;
136     p_sap->i_addresses = 0;
137     p_sap->i_current_session = 0;
138
139     p_sap->b_control = config_GetInt( p_sap, "sap-flow-control");
140
141     if( vlc_thread_create( p_sap, "sap handler", RunThread,
142                        VLC_THREAD_PRIORITY_LOW, VLC_FALSE ) )
143     {
144         msg_Dbg( p_announce, "Unable to spawn SAP handler thread");
145         free( p_sap );
146         return NULL;
147     };
148     msg_Dbg( p_announce, "thread created, %i sessions", p_sap->i_sessions);
149     return p_sap;
150 }
151
152 /**
153  *  Destroy the SAP handler
154  *  \param p_this the SAP Handler to destroy
155  *  \return nothing
156  */
157 void announce_SAPHandlerDestroy( sap_handler_t *p_sap )
158 {
159     int i;
160
161     vlc_mutex_destroy( &p_sap->object_lock );
162
163     /* Free the remaining sessions */
164     for( i = 0 ; i< p_sap->i_sessions ; i++)
165     {
166         sap_session_t *p_session = p_sap->pp_sessions[i];
167         FREE( p_session->psz_sdp );
168         FREE( p_session->psz_data );
169         REMOVE_ELEM( p_sap->pp_sessions, p_sap->i_sessions , i );
170         FREE( p_session );
171     }
172
173     /* Free the remaining addresses */
174     for( i = 0 ; i< p_sap->i_addresses ; i++)
175     {
176         sap_address_t *p_address = p_sap->pp_addresses[i];
177         FREE( p_address->psz_address );
178         if( p_address->i_rfd > -1 )
179         {
180             net_Close( p_address->i_rfd );
181         }
182         if( p_address->i_wfd > -1 && p_sap->b_control )
183         {
184             net_Close( p_address->i_wfd );
185         }
186         REMOVE_ELEM( p_sap->pp_addresses, p_sap->i_addresses, i );
187         FREE( p_address );
188     }
189
190     if( p_sap->iconvHandle != (vlc_iconv_t)(-1) )
191         vlc_iconv_close( p_sap->iconvHandle );
192
193     /* Free the structure */
194     vlc_object_destroy( p_sap );
195 }
196
197 /**
198  * main SAP handler thread
199  * \param p_this the SAP Handler object
200  * \return nothing
201  */
202 static void RunThread( vlc_object_t *p_this)
203 {
204     sap_handler_t *p_sap = (sap_handler_t*)p_this;
205     sap_session_t *p_session;
206
207     while( !p_sap->b_die )
208     {
209         int i;
210
211         /* If needed, get the rate info */
212         if( p_sap->b_control == VLC_TRUE )
213         {
214             for( i = 0 ; i< p_sap->i_addresses ; i++)
215             {
216                 if( p_sap->pp_addresses[i]->b_enabled == VLC_TRUE )
217                 {
218                     CalculateRate( p_sap, p_sap->pp_addresses[i] );
219                 }
220             }
221         }
222
223         /* Find the session to announce */
224         vlc_mutex_lock( &p_sap->object_lock );
225         if( p_sap->i_sessions > p_sap->i_current_session + 1)
226         {
227             p_sap->i_current_session++;
228         }
229         else if( p_sap->i_sessions > 0)
230         {
231             p_sap->i_current_session = 0;
232         }
233         else
234         {
235             vlc_mutex_unlock( &p_sap->object_lock );
236             msleep( SAP_IDLE );
237             continue;
238         }
239         p_session = p_sap->pp_sessions[p_sap->i_current_session];
240         vlc_mutex_unlock( &p_sap->object_lock );
241
242         /* And announce it */
243         if( p_session->p_address->b_enabled == VLC_TRUE &&
244             p_session->p_address->b_ready == VLC_TRUE )
245         {
246             announce_SendSAPAnnounce( p_sap, p_session );
247         }
248
249         msleep( SAP_IDLE );
250     }
251 }
252
253 /* Add a SAP announce */
254 static int announce_SAPAnnounceAdd( sap_handler_t *p_sap,
255                              session_descriptor_t *p_session,
256                              announce_method_t *p_method )
257 {
258     int i;
259     char *psz_type = "application/sdp";
260     int i_header_size;
261     char *psz_head;
262     vlc_bool_t b_found = VLC_FALSE, b_ipv6 = VLC_FALSE;
263     sap_session_t *p_sap_session;
264     mtime_t i_hash;
265
266     vlc_mutex_lock( &p_sap->object_lock );
267
268     /* If needed, build the SDP */
269     if( !p_session->psz_sdp )
270     {
271         if ( SDPGenerate( p_sap, p_session ) != VLC_SUCCESS )
272         {
273             vlc_mutex_unlock( &p_sap->object_lock );
274             return VLC_EGENERIC;
275         }
276     }
277
278     if( p_method->psz_address == NULL )
279     {
280         /* Determine SAP multicast address automatically */
281         char psz_buf[NI_MAXNUMERICHOST], *ptr;
282         const char *psz_addr;
283         struct addrinfo hints, *res;
284
285         if( p_session->psz_uri == NULL )
286         {
287             msg_Err( p_sap, "*FIXME* Unexpected NULL URI for SAP announce" );
288             msg_Err( p_sap, "This should not happen. VLC needs fixing." );
289             vlc_mutex_unlock( &p_sap->object_lock );
290             return VLC_EGENERIC;
291         }
292
293         /* Canonicalize IP address (e.g. 224.00.010.1 => 224.0.10.1) */
294         memset( &hints, 0, sizeof( hints ) );
295         hints.ai_socktype = SOCK_DGRAM;
296         hints.ai_flags = AI_NUMERICHOST;
297
298         i = vlc_getaddrinfo( (vlc_object_t *)p_sap, p_session->psz_uri, 0,
299                              &hints, &res );
300         if( i == 0 )
301             i = vlc_getnameinfo( res->ai_addr, res->ai_addrlen, psz_buf,
302                                  sizeof( psz_buf ), NULL, NI_NUMERICHOST );
303         if( i )
304         {
305             msg_Err( p_sap, "Invalid URI for SAP announce : %s : %s",
306                      p_session->psz_uri, vlc_gai_strerror( i ) );
307             vlc_mutex_unlock( &p_sap->object_lock );
308             return VLC_EGENERIC;
309         }
310
311         /* Remove interface specification if present */
312         ptr = strchr( psz_buf, '%' );
313         if( ptr != NULL )
314             *ptr = '\0';
315
316         if( strchr( psz_buf, ':' ) != NULL )
317         {
318             b_ipv6 = VLC_TRUE;
319
320             /* See RFC3513 for list of valid IPv6 scopes */
321             if( ( tolower( psz_buf[0] ) == 'f' )
322              && ( tolower( psz_buf[1] ) == 'f' )
323              && isxdigit( psz_buf[2] ) && isxdigit( psz_buf[3] ) )
324             {
325                 /* Multicast IPv6 */
326                 psz_buf[2] = '0'; /* force flags to zero */
327                 /* keep scope in psz_addr[3] */
328                 memcpy( &psz_buf[4], "::2:7ffe", sizeof( "::2:7ffe" ) );
329                 psz_addr = psz_buf;
330             }
331             else
332                 /* Unicast IPv6 - assume global scope */
333                 psz_addr = "ff0e::2:7ffe";
334         }
335         else
336         {
337             /* See RFC2365 for IPv4 scopes */
338             if( memcmp( psz_buf, "224.0.0.", 8 ) == 0 )
339                 psz_addr = "224.0.0.255";
340             else
341             if( memcmp( psz_buf, "239.255.", 8 ) == 0 )
342                 psz_addr = "239.255.255.255";
343             else
344             if( ( memcmp( psz_buf, "239.19", 6 ) == 0 )
345              && ( ( psz_buf[6] >= '2' ) && ( psz_buf[6] <= '5' ) ) )
346                 psz_addr = "239.195.255.255";
347             else
348                 /* assume global scope */
349                 psz_addr = "224.2.127.254";
350         }
351
352         p_method->psz_address = strdup( psz_addr );
353     }
354     else
355         b_ipv6 == (strchr( p_method->psz_address, ':' ) != NULL);
356
357     msg_Dbg( p_sap, "using SAP address: %s", p_method->psz_address);
358
359     /* XXX: Check for dupes */
360     p_sap_session = (sap_session_t*)malloc(sizeof(sap_session_t));
361
362     p_sap_session->psz_sdp = strdup( p_session->psz_sdp );
363     p_sap_session->i_last = 0;
364
365     /* Add the address to the buffer */
366     for( i = 0; i< p_sap->i_addresses; i++)
367     {
368         if( !strcmp( p_method->psz_address,
369              p_sap->pp_addresses[i]->psz_address ) )
370         {
371             p_sap_session->p_address = p_sap->pp_addresses[i];
372             b_found = VLC_TRUE;
373             break;
374         }
375     }
376     if( b_found == VLC_FALSE )
377     {
378         sap_address_t *p_address = (sap_address_t *)
379                                     malloc( sizeof(sap_address_t) );
380         if( !p_address )
381         {
382             msg_Err( p_sap, "out of memory" );
383             return VLC_ENOMEM;
384         }
385         p_address->psz_address = strdup( p_method->psz_address );
386         p_address->i_port  =  9875;
387         p_address->i_wfd = net_OpenUDP( p_sap, "", 0,
388                                         p_address->psz_address,
389                                         p_address->i_port );
390         if( p_address->i_wfd != -1 )
391             net_StopRecv( p_address->i_wfd );
392
393         if( p_sap->b_control == VLC_TRUE )
394         {
395             p_address->i_rfd = net_OpenUDP( p_sap, p_method->psz_address,
396                                             p_address->i_port,
397                                             "", 0 );
398             if( p_address->i_rfd != -1 )
399                 net_StopSend( p_address->i_rfd );
400             p_address->i_buff = 0;
401             p_address->b_enabled = VLC_TRUE;
402             p_address->b_ready = VLC_FALSE;
403             p_address->i_limit = 10000; /* 10000 bps */
404             p_address->t1 = 0;
405         }
406         else
407         {
408             p_address->b_enabled = VLC_TRUE;
409             p_address->b_ready = VLC_TRUE;
410             p_address->i_interval = config_GetInt( p_sap,"sap-interval");
411             p_address->i_rfd = -1;
412         }
413
414         if( p_address->i_wfd == -1 || (p_address->i_rfd == -1
415                                         && p_sap->b_control ) )
416         {
417             msg_Warn( p_sap, "disabling address" );
418             p_address->b_enabled = VLC_FALSE;
419         }
420
421         INSERT_ELEM( p_sap->pp_addresses,
422                      p_sap->i_addresses,
423                      p_sap->i_addresses,
424                      p_address );
425         p_sap_session->p_address = p_address;
426     }
427
428     /* Build the SAP Headers */
429     i_header_size = ( b_ipv6 ? 20 : 8 ) + strlen( psz_type ) + 1;
430     psz_head = (char *) malloc( i_header_size * sizeof( char ) );
431     if( ! psz_head )
432     {
433         msg_Err( p_sap, "out of memory" );
434         return VLC_ENOMEM;
435     }
436
437     psz_head[0] = 0x20; /* Means SAPv1, IPv4, not encrypted, not compressed */
438     psz_head[1] = 0x00; /* No authentification length */
439
440     i_hash = mdate();
441     psz_head[2] = (i_hash & 0xFF00) >> 8; /* Msg id hash */
442     psz_head[3] = (i_hash & 0xFF);        /* Msg id hash 2 */
443
444     if( b_ipv6 )
445     {
446         /* in_addr_t ip_server = inet_addr( ip ); */
447         psz_head[0] |= 0x10; /* Set IPv6 */
448
449         psz_head[4] = 0x01; /* Source IP  FIXME: we should get the real address */
450         psz_head[5] = 0x02; /* idem */
451         psz_head[6] = 0x03; /* idem */
452         psz_head[7] = 0x04; /* idem */
453
454         psz_head[8] = 0x01; /* Source IP  FIXME: we should get the real address */
455         psz_head[9] = 0x02; /* idem */
456         psz_head[10] = 0x03; /* idem */
457         psz_head[11] = 0x04; /* idem */
458
459         psz_head[12] = 0x01; /* Source IP  FIXME: we should get the real address */
460         psz_head[13] = 0x02; /* idem */
461         psz_head[14] = 0x03; /* idem */
462         psz_head[15] = 0x04; /* idem */
463
464         psz_head[16] = 0x01; /* Source IP  FIXME: we should get the real address */
465         psz_head[17] = 0x02; /* idem */
466         psz_head[18] = 0x03; /* idem */
467         psz_head[19] = 0x04; /* idem */
468
469         strncpy( psz_head + 20, psz_type, 15 );
470     }
471     else
472     {
473         /* in_addr_t ip_server = inet_addr( ip) */
474         /* Source IP  FIXME: we should get the real address */
475         psz_head[4] = 0x01; /* ip_server */
476         psz_head[5] = 0x02; /* ip_server>>8 */
477         psz_head[6] = 0x03; /* ip_server>>16 */
478         psz_head[7] = 0x04; /* ip_server>>24 */
479
480         strncpy( psz_head + 8, psz_type, 15 );
481     }
482
483     psz_head[ i_header_size-1 ] = '\0';
484     p_sap_session->i_length = i_header_size + strlen( p_sap_session->psz_sdp);
485
486     p_sap_session->psz_data = (char *)malloc( sizeof(char)*
487                                               p_sap_session->i_length );
488
489     /* Build the final message */
490     memcpy( p_sap_session->psz_data, psz_head, i_header_size );
491     memcpy( p_sap_session->psz_data+i_header_size, p_sap_session->psz_sdp,
492             strlen( p_sap_session->psz_sdp) );
493
494     free( psz_head );
495
496     /* Enqueue the announce */
497     INSERT_ELEM( p_sap->pp_sessions,
498                  p_sap->i_sessions,
499                  p_sap->i_sessions,
500                  p_sap_session );
501     msg_Dbg( p_sap,"Addresses: %i  Sessions: %i",
502                    p_sap->i_addresses,p_sap->i_sessions);
503
504     /* Remember the SAP session for later deletion */
505     p_session->p_sap = p_sap_session;
506
507     vlc_mutex_unlock( &p_sap->object_lock );
508
509     return VLC_SUCCESS;
510 }
511
512 /* Remove a SAP Announce */
513 static int announce_SAPAnnounceDel( sap_handler_t *p_sap,
514                              session_descriptor_t *p_session )
515 {
516     int i;
517     vlc_mutex_lock( &p_sap->object_lock );
518
519     msg_Dbg( p_sap,"removing SAP announce %p",p_session->p_sap);
520
521     /* Dequeue the announce */
522     for( i = 0; i< p_sap->i_sessions; i++)
523     {
524         if( p_session->p_sap == p_sap->pp_sessions[i] )
525         {
526             REMOVE_ELEM( p_sap->pp_sessions,
527                          p_sap->i_sessions,
528                          i );
529
530             FREE( p_session->p_sap->psz_sdp );
531             FREE( p_session->p_sap->psz_data );
532             free( p_session->p_sap );
533             break;
534         }
535     }
536
537     /* XXX: Dequeue the address too if it is not used anymore
538      * TODO: - address refcount
539              - send a SAP deletion packet */
540
541     msg_Dbg( p_sap,"%i announces remaining", p_sap->i_sessions );
542
543     vlc_mutex_unlock( &p_sap->object_lock );
544
545     return VLC_SUCCESS;
546 }
547
548 static int announce_SendSAPAnnounce( sap_handler_t *p_sap,
549                                      sap_session_t *p_session )
550 {
551     int i_ret;
552
553     /* This announce has never been sent yet */
554     if( p_session->i_last == 0 )
555     {
556         p_session->i_next = mdate()+ p_session->p_address->i_interval*1000000;
557         p_session->i_last = 1;
558         return VLC_SUCCESS;
559     }
560
561     if( p_session->i_next < mdate() )
562     {
563 #ifdef EXTRA_DEBUG
564         msg_Dbg( p_sap, "Sending announce");
565 #endif
566         i_ret = net_Write( p_sap, p_session->p_address->i_wfd, NULL,
567                            p_session->psz_data,
568                            p_session->i_length );
569         if( i_ret  != p_session->i_length )
570         {
571             msg_Warn( p_sap, "SAP send failed on address %s (%i %i)",
572                    p_session->p_address->psz_address,
573                    i_ret, p_session->i_length );
574         }
575         p_session->i_last = p_session->i_next;
576         p_session->i_next = p_session->i_last
577                             + p_session->p_address->i_interval*1000000;
578     }
579     else
580     {
581         return VLC_SUCCESS;
582     }
583     return VLC_SUCCESS;
584 }
585
586 static int SDPGenerate( sap_handler_t *p_sap, session_descriptor_t *p_session )
587 {
588     int64_t i_sdp_id = mdate();
589     int     i_sdp_version = 1 + p_sap->i_sessions + (rand()&0xfff);
590     char *psz_group, *psz_name, psz_uribuf[NI_MAXNUMERICHOST], *psz_uri;
591     char ipv;
592
593     psz_group = convert_to_utf8( p_sap, p_session->psz_group );
594     psz_name = convert_to_utf8( p_sap, p_session->psz_name );
595
596     /* FIXME: really check that psz_uri is a real IP address
597      * FIXME: make a common function to obtain a canonical IP address */
598     ipv = ( strchr( p_session->psz_uri, ':' )  != NULL) ? '6' : '4';
599     if( *p_session->psz_uri == '[' )
600     {
601         char *ptr;
602
603         strncpy( psz_uribuf, p_session->psz_uri + 1, sizeof( psz_uribuf ) );
604         psz_uribuf[sizeof( psz_uribuf ) - 1] = '\0';
605         ptr = strchr( psz_uribuf, '%' );
606         if( ptr != NULL)
607             *ptr = '\0';
608         ptr = strchr( psz_uribuf, ']' );
609         if( ptr != NULL)
610             *ptr = '\0';
611         psz_uri = psz_uribuf;
612     }
613     else
614         psz_uri = p_session->psz_uri;
615
616     /* see the lists in modules/stream_out/rtp.c for compliance stuff */
617     p_session->psz_sdp = (char *)malloc(
618                             sizeof("v=0\r\n"
619                                    "o=- 45383436098 45398  IN IP4 127.0.0.1\r\n" /* FIXME */
620                                    "s=\r\n"
621                                    "t=0 0\r\n"
622                                    "c=IN IP4 /\r\n"
623                                    "m=video  udp\r\n"
624                                    "a=tool:"PACKAGE_STRING"\r\n"
625                                    "a=type:broadcast\r\n")
626                            + strlen( psz_name )
627                            + strlen( psz_uri ) + 300
628                            + ( psz_group ? strlen( psz_group ) : 0 ) );
629
630     if( p_session->psz_sdp == NULL || psz_name == NULL )
631     {
632         msg_Err( p_sap, "out of memory" );
633         FREE( psz_name );
634         FREE( psz_group );
635         return VLC_ENOMEM;
636     }
637
638     sprintf( p_session->psz_sdp,
639                             "v=0\r\n"
640                             "o=- "I64Fd" %d IN IP4 127.0.0.1\r\n"
641                             "s=%s\r\n"
642                             "t=0 0\r\n"
643                             "c=IN IP%c %s/%d\r\n"
644                             "m=video %d udp %d\r\n"
645                             "a=tool:"PACKAGE_STRING"\r\n"
646                             "a=type:broadcast\r\n",
647                             i_sdp_id, i_sdp_version,
648                             psz_name, ipv,
649                             psz_uri, p_session->i_ttl,
650                             p_session->i_port, p_session->i_payload );
651     free( psz_name );
652
653     if( psz_group )
654     {
655         sprintf( p_session->psz_sdp, "%sa=x-plgroup:%s\r\n",
656                                      p_session->psz_sdp, psz_group );
657         free( psz_group );
658     }
659
660     msg_Dbg( p_sap, "Generated SDP (%i bytes):\n%s", strlen(p_session->psz_sdp),
661                     p_session->psz_sdp );
662     return VLC_SUCCESS;
663 }
664
665 static int CalculateRate( sap_handler_t *p_sap, sap_address_t *p_address )
666 {
667     int i_read;
668     char buffer[SAP_MAX_BUFFER];
669     int i_tot = 0;
670     mtime_t i_temp;
671     int i_rate;
672
673     if( p_address->t1 == 0 )
674     {
675         p_address->t1 = mdate();
676         return VLC_SUCCESS;
677     }
678     do
679     {
680         /* Might be too slow if we have huge data */
681         i_read = net_ReadNonBlock( p_sap, p_address->i_rfd, NULL, buffer,
682                                    SAP_MAX_BUFFER, 0 );
683         i_tot += i_read;
684     } while( i_read > 0 && i_tot < SAP_MAX_BUFFER );
685
686     i_temp = mdate();
687
688     /* We calculate the rate every 5 seconds */
689     if( i_temp - p_address->t1 < 5000000 )
690     {
691         p_address->i_buff += i_tot;
692         return VLC_SUCCESS;
693     }
694
695     /* Bits/second */
696     i_rate = (int)(8*1000000*((mtime_t)p_address->i_buff + (mtime_t)i_tot ) /
697                         (i_temp - p_address->t1 ));
698
699     p_address->i_limit = 10000;
700
701     p_address->i_interval = ((1000*i_rate / p_address->i_limit) *
702                             (MAX_INTERVAL - MIN_INTERVAL))/1000 + MIN_INTERVAL;
703
704     if( p_address->i_interval > MAX_INTERVAL || p_address->i_interval < 0 )
705     {
706         p_address->i_interval = MAX_INTERVAL;
707     }
708 #ifdef EXTRA_DEBUG
709     msg_Dbg( p_sap,"%s:%i : Rate=%i, Interval = %i s",
710                     p_address->psz_address,p_address->i_port,
711                     i_rate,
712                     p_address->i_interval );
713 #endif
714
715     p_address->b_ready = VLC_TRUE;
716
717     p_address->t1 = i_temp;
718     p_address->i_buff = 0;
719
720     return VLC_SUCCESS;
721 }
722
723
724 static char *convert_to_utf8( struct sap_handler_t *p_this, char *psz_local )
725 {
726     char *psz_unicode, *psz_in, *psz_out;
727     size_t ret, i_in, i_out;
728
729     if( psz_local == NULL )
730         return NULL;
731     if ( p_this->iconvHandle == (vlc_iconv_t)(-1) )
732         return strdup( psz_local );
733
734     psz_in = psz_local;
735     i_in = strlen( psz_local );
736
737     i_out = 6 * i_in;
738     psz_unicode = malloc( i_out + 1 );
739     if( psz_unicode == NULL )
740         return strdup( psz_local );
741     psz_out = psz_unicode;
742
743     ret = vlc_iconv( p_this->iconvHandle,
744                      &psz_in, &i_in, &psz_out, &i_out);
745     if( ret == (size_t)(-1) || i_in )
746     {
747         msg_Warn( p_this, "Failed to convert \"%s\" to UTF-8", psz_local );
748         return strdup( psz_local );
749     }
750     *psz_out = '\0';
751     return psz_unicode;
752 }