]> git.sesse.net Git - vlc/blob - src/extras/libc.c
Unicode support in directory access on windows. Patch by xxcv. Untested.
[vlc] / src / extras / libc.c
1 /*****************************************************************************
2  * libc.c: Extra libc function for some systems.
3  *****************************************************************************
4  * Copyright (C) 2002-2006 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Jon Lech Johansen <jon-vl@nanocrew.net>
8  *          Samuel Hocevar <sam@zoy.org>
9  *          Gildas Bazin <gbazin@videolan.org>
10  *          Derk-Jan Hartman <hartman at videolan dot org>
11  *          Christophe Massiot <massiot@via.ecp.fr>
12  *          Rémi Denis-Courmont <rem à videolan.org>
13  *
14  * This program is free software; you can redistribute it and/or modify
15  * it under the terms of the GNU General Public License as published by
16  * the Free Software Foundation; either version 2 of the License, or
17  * (at your option) any later version.
18  *
19  * This program is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public License
25  * along with this program; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
27  *****************************************************************************/
28 #include <string.h>                                              /* strdup() */
29 #include <stdlib.h>
30 #include <ctype.h>
31
32 #include <vlc/vlc.h>
33
34 #undef iconv_t
35 #undef iconv_open
36 #undef iconv
37 #undef iconv_close
38
39 #if defined(HAVE_ICONV)
40 #   include <iconv.h>
41 #endif
42
43 #ifdef HAVE_DIRENT_H
44 #   include <dirent.h>
45 #endif
46
47 #ifdef HAVE_FORK
48 #   include <sys/time.h>
49 #   include <unistd.h>
50 #   include <errno.h>
51 #   include <sys/wait.h>
52 #endif
53
54 #if defined(WIN32) || defined(UNDER_CE)
55 #   define WIN32_LEAN_AND_MEAN
56 #   include <windows.h>
57 #endif
58
59 #ifdef UNDER_CE
60 #   define strcoll strcmp
61 #endif
62
63 /*****************************************************************************
64  * getenv: just in case, but it should never be called
65  *****************************************************************************/
66 #if !defined( HAVE_GETENV )
67 char *vlc_getenv( const char *name )
68 {
69     return NULL;
70 }
71 #endif
72
73 /*****************************************************************************
74  * strdup: returns a malloc'd copy of a string
75  *****************************************************************************/
76 #if !defined( HAVE_STRDUP )
77 char *vlc_strdup( const char *string )
78 {
79     return strndup( string, strlen( string ) );
80 }
81 #endif
82
83 /*****************************************************************************
84  * strndup: returns a malloc'd copy of at most n bytes of string
85  * Does anyone know whether or not it will be present in Jaguar?
86  *****************************************************************************/
87 #if !defined( HAVE_STRNDUP )
88 char *vlc_strndup( const char *string, size_t n )
89 {
90     char *psz;
91     size_t len = strlen( string );
92
93     len = __MIN( len, n );
94     psz = (char*)malloc( len + 1 );
95
96     if( psz != NULL )
97     {
98         memcpy( (void*)psz, (const void*)string, len );
99         psz[ len ] = 0;
100     }
101
102     return psz;
103 }
104 #endif
105
106 /*****************************************************************************
107  * strcasecmp: compare two strings ignoring case
108  *****************************************************************************/
109 #if !defined( HAVE_STRCASECMP ) && !defined( HAVE_STRICMP )
110 int vlc_strcasecmp( const char *s1, const char *s2 )
111 {
112     int c1, c2;
113     if( !s1 || !s2 ) return  -1;
114
115     while( *s1 && *s2 )
116     {
117         c1 = tolower(*s1);
118         c2 = tolower(*s2);
119
120         if( c1 != c2 ) return (c1 < c2 ? -1 : 1);
121         s1++; s2++;
122     }
123
124     if( !*s1 && !*s2 ) return 0;
125     else return (*s1 ? 1 : -1);
126 }
127 #endif
128
129 /*****************************************************************************
130  * strncasecmp: compare n chars from two strings ignoring case
131  *****************************************************************************/
132 #if !defined( HAVE_STRNCASECMP ) && !defined( HAVE_STRNICMP )
133 int vlc_strncasecmp( const char *s1, const char *s2, size_t n )
134 {
135     int c1, c2;
136     if( !s1 || !s2 ) return  -1;
137
138     while( n > 0 && *s1 && *s2 )
139     {
140         c1 = tolower(*s1);
141         c2 = tolower(*s2);
142
143         if( c1 != c2 ) return (c1 < c2 ? -1 : 1);
144         s1++; s2++; n--;
145     }
146
147     if( !n || (!*s1 && !*s2) ) return 0;
148     else return (*s1 ? 1 : -1);
149 }
150 #endif
151
152 /******************************************************************************
153  * strcasestr: find a substring (little) in another substring (big)
154  * Case sensitive. Return NULL if not found, return big if little == null
155  *****************************************************************************/
156 #if !defined( HAVE_STRCASESTR ) && !defined( HAVE_STRISTR )
157 char * vlc_strcasestr( const char *psz_big, const char *psz_little )
158 {
159     char *p_pos = (char *)psz_big;
160
161     if( !psz_big || !psz_little || !*psz_little ) return p_pos;
162  
163     while( *p_pos ) 
164     {
165         if( toupper( *p_pos ) == toupper( *psz_little ) )
166         {
167             char * psz_cur1 = p_pos + 1;
168             char * psz_cur2 = (char *)psz_little + 1;
169             while( *psz_cur1 && *psz_cur2 &&
170                    toupper( *psz_cur1 ) == toupper( *psz_cur2 ) )
171             {
172                 psz_cur1++;
173                 psz_cur2++;
174             }
175             if( !*psz_cur2 ) return p_pos;
176         }
177         p_pos++;
178     }
179     return NULL;
180 }
181 #endif
182
183 /*****************************************************************************
184  * vasprintf:
185  *****************************************************************************/
186 #if !defined(HAVE_VASPRINTF) || defined(__APPLE__) || defined(SYS_BEOS)
187 int vlc_vasprintf(char **strp, const char *fmt, va_list ap)
188 {
189     /* Guess we need no more than 100 bytes. */
190     int     i_size = 100;
191     char    *p = malloc( i_size );
192     int     n;
193
194     if( p == NULL )
195     {
196         *strp = NULL;
197         return -1;
198     }
199
200     for( ;; )
201     {
202         /* Try to print in the allocated space. */
203         n = vsnprintf( p, i_size, fmt, ap );
204
205         /* If that worked, return the string. */
206         if (n > -1 && n < i_size)
207         {
208             *strp = p;
209             return strlen( p );
210         }
211         /* Else try again with more space. */
212         if (n > -1)    /* glibc 2.1 */
213         {
214            i_size = n+1; /* precisely what is needed */
215         }
216         else           /* glibc 2.0 */
217         {
218            i_size *= 2;  /* twice the old size */
219         }
220         if( (p = realloc( p, i_size ) ) == NULL)
221         {
222             *strp = NULL;
223             return -1;
224         }
225     }
226 }
227 #endif
228
229 /*****************************************************************************
230  * asprintf:
231  *****************************************************************************/
232 #if !defined(HAVE_ASPRINTF) || defined(__APPLE__) || defined(SYS_BEOS)
233 int vlc_asprintf( char **strp, const char *fmt, ... )
234 {
235     va_list args;
236     int i_ret;
237
238     va_start( args, fmt );
239     i_ret = vasprintf( strp, fmt, args );
240     va_end( args );
241
242     return i_ret;
243 }
244 #endif
245
246 /*****************************************************************************
247  * atof: convert a string to a double.
248  *****************************************************************************/
249 #if !defined( HAVE_ATOF )
250 double vlc_atof( const char *nptr )
251 {
252     double f_result;
253     wchar_t *psz_tmp;
254     int i_len = strlen( nptr ) + 1;
255
256     psz_tmp = malloc( i_len * sizeof(wchar_t) );
257     MultiByteToWideChar( CP_ACP, 0, nptr, -1, psz_tmp, i_len );
258     f_result = wcstod( psz_tmp, NULL );
259     free( psz_tmp );
260
261     return f_result;
262 }
263 #endif
264
265 /*****************************************************************************
266  * strtoll: convert a string to a 64 bits int.
267  *****************************************************************************/
268 #if !defined( HAVE_STRTOLL )
269 int64_t vlc_strtoll( const char *nptr, char **endptr, int base )
270 {
271     int64_t i_value = 0;
272     int sign = 1, newbase = base ? base : 10;
273
274     while( isspace(*nptr) ) nptr++;
275
276     if( *nptr == '-' )
277     {
278         sign = -1;
279         nptr++;
280     }
281
282     /* Try to detect base */
283     if( *nptr == '0' )
284     {
285         newbase = 8;
286         nptr++;
287
288         if( *nptr == 'x' )
289         {
290             newbase = 16;
291             nptr++;
292         }
293     }
294
295     if( base && newbase != base )
296     {
297         if( endptr ) *endptr = (char *)nptr;
298         return i_value;
299     }
300
301     switch( newbase )
302     {
303         case 10:
304             while( *nptr >= '0' && *nptr <= '9' )
305             {
306                 i_value *= 10;
307                 i_value += ( *nptr++ - '0' );
308             }
309             if( endptr ) *endptr = (char *)nptr;
310             break;
311
312         case 16:
313             while( (*nptr >= '0' && *nptr <= '9') ||
314                    (*nptr >= 'a' && *nptr <= 'f') ||
315                    (*nptr >= 'A' && *nptr <= 'F') )
316             {
317                 int i_valc = 0;
318                 if(*nptr >= '0' && *nptr <= '9') i_valc = *nptr - '0';
319                 else if(*nptr >= 'a' && *nptr <= 'f') i_valc = *nptr - 'a' +10;
320                 else if(*nptr >= 'A' && *nptr <= 'F') i_valc = *nptr - 'A' +10;
321                 i_value *= 16;
322                 i_value += i_valc;
323                 nptr++;
324             }
325             if( endptr ) *endptr = (char *)nptr;
326             break;
327
328         default:
329             i_value = strtol( nptr, endptr, newbase );
330             break;
331     }
332
333     return i_value * sign;
334 }
335 #endif
336
337 /*****************************************************************************
338  * atoll: convert a string to a 64 bits int.
339  *****************************************************************************/
340 #if !defined( HAVE_ATOLL )
341 int64_t vlc_atoll( const char *nptr )
342 {
343     return strtoll( nptr, (char **)NULL, 10 );
344 }
345 #endif
346
347 /*****************************************************************************
348  * lldiv: returns quotient and remainder
349  *****************************************************************************/
350 #if defined(SYS_BEOS) \
351  || (defined (__FreeBSD__) && (__FreeBSD__ < 5))
352 lldiv_t vlc_lldiv( long long numer, long long denom )
353 {
354     lldiv_t d;
355     d.quot = numer / denom;
356     d.rem  = numer % denom;
357     return d;
358 }
359 #endif
360
361
362 /**
363  * Copy a string to a sized buffer. The result is always nul-terminated
364  * (contrary to strncpy()).
365  *
366  * @param dest destination buffer
367  * @param src string to be copied
368  * @param len maximum number of characters to be copied plus one for the
369  * terminating nul.
370  *
371  * @return strlen(src)
372  */
373 #ifndef HAVE_STRLCPY
374 extern size_t vlc_strlcpy (char *tgt, const char *src, size_t bufsize)
375 {
376     size_t length;
377
378     for (length = 1; (length < bufsize) && *src; length++)
379         *tgt++ = *src++;
380
381     if (bufsize)
382         *tgt = '\0';
383
384     while (*src++)
385         length++;
386
387     return length - 1;
388 }
389 #endif
390
391 /*****************************************************************************
392  * vlc_*dir_wrapper: wrapper under Windows to return the list of drive letters
393  * when called with an empty argument or just '\'
394  *****************************************************************************/
395 #if defined(WIN32) && !defined(UNDER_CE)
396 typedef struct vlc_DIR
397 {
398     DIR *p_real_dir;
399     int i_drives;
400     struct dirent dd_dir;
401     vlc_bool_t b_insert_back;
402 } vlc_DIR;
403
404 void *vlc_opendir_wrapper( const char *psz_path )
405 {
406     vlc_DIR *p_dir;
407     DIR *p_real_dir;
408
409     if ( psz_path == NULL || psz_path[0] == '\0'
410           || (psz_path[0] == '\\' && psz_path[1] == '\0') )
411     {
412         /* Special mode to list drive letters */
413         p_dir = malloc( sizeof(vlc_DIR) );
414         p_dir->p_real_dir = NULL;
415         p_dir->i_drives = GetLogicalDrives();
416         return (void *)p_dir;
417     }
418
419     if (GetVersion() < 0x80000000)
420     {
421         /* for Windows NT and above */
422         wchar_t wpath[MAX_PATH + 1];
423
424         if (!MultiByteToWideChar (CP_UTF8, 0, psz_path, -1, wpath, MAX_PATH))
425             return NULL;
426         wpath[MAX_PATH] = L'\0';
427         p_real_dir = _wopendir( wpath );
428     }
429
430     if ( p_real_dir == NULL )
431         return NULL;
432
433     p_dir = malloc( sizeof(vlc_DIR) );
434     p_dir->p_real_dir = p_real_dir;
435     p_dir->b_insert_back = ( psz_path[1] == ':' && psz_path[2] == '\\'
436                               && psz_path[3] =='\0' );
437     return (void *)p_dir;
438 }
439
440 struct dirent *vlc_readdir_wrapper( void *_p_dir )
441 {
442     vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
443     unsigned int i;
444     DWORD i_drives;
445
446     if ( p_dir->p_real_dir != NULL )
447     {
448         if ( p_dir->b_insert_back )
449         {
450             p_dir->dd_dir.d_ino = 0;
451             p_dir->dd_dir.d_reclen = 0;
452             p_dir->dd_dir.d_namlen = 2;
453             strcpy( p_dir->dd_dir.d_name, ".." );
454             p_dir->b_insert_back = VLC_FALSE;
455             return &p_dir->dd_dir;
456         }
457
458         return _wreaddir( p_dir->p_real_dir );
459     }
460
461     /* Drive letters mode */
462     i_drives = p_dir->i_drives;
463     if ( !i_drives )
464         return NULL; /* end */
465
466     for ( i = 0; i < sizeof(DWORD)*8; i++, i_drives >>= 1 )
467         if ( i_drives & 1 ) break;
468
469     if ( i >= 26 )
470         return NULL; /* this should not happen */
471
472     sprintf( p_dir->dd_dir.d_name, "%c:\\", 'A' + i );
473     p_dir->dd_dir.d_namlen = strlen(p_dir->dd_dir.d_name);
474     p_dir->i_drives &= ~(1UL << i);
475     return &p_dir->dd_dir;
476 }
477
478 int vlc_closedir_wrapper( void *_p_dir )
479 {
480     vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
481
482     if ( p_dir->p_real_dir != NULL )
483     {
484         int i_ret = _wclosedir( p_dir->p_real_dir );
485         free( p_dir );
486         return i_ret;
487     }
488
489     free( p_dir );
490     return 0;
491 }
492 #else
493 void *vlc_opendir_wrapper( const char *psz_path )
494 {
495     return (void *)opendir( psz_path );
496 }
497 struct dirent *vlc_readdir_wrapper( void *_p_dir )
498 {
499     return readdir( (DIR *)_p_dir );
500 }
501 int vlc_closedir_wrapper( void *_p_dir )
502 {
503     return closedir( (DIR *)_p_dir );
504 }
505 #endif
506
507 /*****************************************************************************
508  * scandir: scan a directory alpha-sorted
509  *****************************************************************************/
510 #if !defined( HAVE_SCANDIR )
511 int vlc_alphasort( const struct dirent **a, const struct dirent **b )
512 {
513     return strcoll( (*a)->d_name, (*b)->d_name );
514 }
515
516 int vlc_scandir( const char *name, struct dirent ***namelist,
517                     int (*filter) ( const struct dirent * ),
518                     int (*compar) ( const struct dirent **,
519                                     const struct dirent ** ) )
520 {
521     DIR            * p_dir;
522     struct dirent  * p_content;
523     struct dirent ** pp_list;
524     int              ret, size;
525
526     if( !namelist || !( p_dir = vlc_opendir_wrapper( name ) ) ) return -1;
527
528     ret     = 0;
529     pp_list = NULL;
530     while( ( p_content = vlc_readdir_wrapper( p_dir ) ) )
531     {
532         if( filter && !filter( p_content ) )
533         {
534             continue;
535         }
536         pp_list = realloc( pp_list, ( ret + 1 ) * sizeof( struct dirent * ) );
537         size = sizeof( struct dirent ) + strlen( p_content->d_name ) + 1;
538         pp_list[ret] = malloc( size );
539         memcpy( pp_list[ret], p_content, size );
540         ret++;
541     }
542
543     vlc_closedir_wrapper( p_dir );
544
545     if( compar )
546     {
547         qsort( pp_list, ret, sizeof( struct dirent * ),
548                (int (*)(const void *, const void *)) compar );
549     }
550
551     *namelist = pp_list;
552     return ret;
553 }
554 #endif
555
556 #if defined (WIN32) || !defined (HAVE_SHARED_LIBVLC)
557 /*****************************************************************************
558  * dgettext: gettext for plugins.
559  *****************************************************************************/
560 char *vlc_dgettext( const char *package, const char *msgid )
561 {
562 #if defined( ENABLE_NLS ) \
563      && ( defined(HAVE_GETTEXT) || defined(HAVE_INCLUDED_GETTEXT) )
564     return dgettext( package, msgid );
565 #else
566     return (char *)msgid;
567 #endif
568 }
569 #endif
570
571 /*****************************************************************************
572  * count_utf8_string: returns the number of characters in the string.
573  *****************************************************************************/
574 static int count_utf8_string( const char *psz_string )
575 {
576     int i = 0, i_count = 0;
577     while( psz_string[ i ] != 0 )
578     {
579         if( ((unsigned char *)psz_string)[ i ] <  0x80UL ) i_count++;
580         i++;
581     }
582     return i_count;
583 }
584
585 /*****************************************************************************
586  * wraptext: inserts \n at convenient places to wrap the text.
587  *           Returns the modified string in a new buffer.
588  *****************************************************************************/
589 char *vlc_wraptext( const char *psz_text, int i_line )
590 {
591     int i_len;
592     char *psz_line, *psz_new_text;
593
594     psz_line = psz_new_text = strdup( psz_text );
595
596     i_len = count_utf8_string( psz_text );
597
598     while( i_len > i_line )
599     {
600         /* Look if there is a newline somewhere. */
601         char *psz_parser = psz_line;
602         int i_count = 0;
603         while( i_count <= i_line && *psz_parser != '\n' )
604         {
605             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
606             psz_parser++;
607             i_count++;
608         }
609         if( *psz_parser == '\n' )
610         {
611             i_len -= (i_count + 1);
612             psz_line = psz_parser + 1;
613             continue;
614         }
615
616         /* Find the furthest space. */
617         while( psz_parser > psz_line && *psz_parser != ' ' )
618         {
619             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser--;
620             psz_parser--;
621             i_count--;
622         }
623         if( *psz_parser == ' ' )
624         {
625             *psz_parser = '\n';
626             i_len -= (i_count + 1);
627             psz_line = psz_parser + 1;
628             continue;
629         }
630
631         /* Wrapping has failed. Find the first space or newline */
632         while( i_count < i_len && *psz_parser != ' ' && *psz_parser != '\n' )
633         {
634             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
635             psz_parser++;
636             i_count++;
637         }
638         if( i_count < i_len ) *psz_parser = '\n';
639         i_len -= (i_count + 1);
640         psz_line = psz_parser + 1;
641     }
642
643     return psz_new_text;
644 }
645
646 /*****************************************************************************
647  * iconv wrapper
648  *****************************************************************************/
649 vlc_iconv_t vlc_iconv_open( const char *tocode, const char *fromcode )
650 {
651 #if defined(HAVE_ICONV)
652     return iconv_open( tocode, fromcode );
653 #else
654     return NULL;
655 #endif
656 }
657
658 size_t vlc_iconv( vlc_iconv_t cd, const char **inbuf, size_t *inbytesleft,
659                   char **outbuf, size_t *outbytesleft )
660 {
661 #if defined(HAVE_ICONV)
662     return iconv( cd, (ICONV_CONST char **)inbuf, inbytesleft,
663                   outbuf, outbytesleft );
664 #else
665     int i_bytes;
666
667     if (inbytesleft == NULL || outbytesleft == NULL)
668     {
669         return 0;
670     }
671
672     i_bytes = __MIN(*inbytesleft, *outbytesleft);
673     if( !inbuf || !outbuf || !i_bytes ) return (size_t)(-1);
674     memcpy( *outbuf, *inbuf, i_bytes );
675     inbuf += i_bytes;
676     outbuf += i_bytes;
677     inbytesleft -= i_bytes;
678     outbytesleft -= i_bytes;
679     return i_bytes;
680 #endif
681 }
682
683 int vlc_iconv_close( vlc_iconv_t cd )
684 {
685 #if defined(HAVE_ICONV)
686     return iconv_close( cd );
687 #else
688     return 0;
689 #endif
690 }
691
692 /*****************************************************************************
693  * reduce a fraction
694  *   (adapted from libavcodec, author Michael Niedermayer <michaelni@gmx.at>)
695  *****************************************************************************/
696 vlc_bool_t vlc_ureduce( unsigned *pi_dst_nom, unsigned *pi_dst_den,
697                         uint64_t i_nom, uint64_t i_den, uint64_t i_max )
698 {
699     vlc_bool_t b_exact = 1;
700     uint64_t i_gcd;
701
702     if( i_den == 0 )
703     {
704         *pi_dst_nom = 0;
705         *pi_dst_den = 1;
706         return 1;
707     }
708
709     i_gcd = GCD( i_nom, i_den );
710     i_nom /= i_gcd;
711     i_den /= i_gcd;
712
713     if( i_max == 0 ) i_max = I64C(0xFFFFFFFF);
714
715     if( i_nom > i_max || i_den > i_max )
716     {
717         uint64_t i_a0_num = 0, i_a0_den = 1, i_a1_num = 1, i_a1_den = 0;
718         b_exact = 0;
719
720         for( ; ; )
721         {
722             uint64_t i_x = i_nom / i_den;
723             uint64_t i_a2n = i_x * i_a1_num + i_a0_num;
724             uint64_t i_a2d = i_x * i_a1_den + i_a0_den;
725
726             if( i_a2n > i_max || i_a2d > i_max ) break;
727
728             i_nom %= i_den;
729
730             i_a0_num = i_a1_num; i_a0_den = i_a1_den;
731             i_a1_num = i_a2n; i_a1_den = i_a2d;
732             if( i_nom == 0 ) break;
733             i_x = i_nom; i_nom = i_den; i_den = i_x;
734         }
735         i_nom = i_a1_num;
736         i_den = i_a1_den;
737     }
738
739     *pi_dst_nom = i_nom;
740     *pi_dst_den = i_den;
741
742     return b_exact;
743 }
744
745 /*************************************************************************
746  * vlc_parse_cmdline: Command line parsing into elements.
747  *
748  * The command line is composed of space/tab separated arguments.
749  * Quotes can be used as argument delimiters and a backslash can be used to
750  * escape a quote.
751  *************************************************************************/
752 static void find_end_quote( char **s, char **ppsz_parser, int i_quote )
753 {
754     int i_bcount = 0;
755
756     while( **s )
757     {
758         if( **s == '\\' )
759         {
760             **ppsz_parser = **s;
761             (*ppsz_parser)++; (*s)++;
762             i_bcount++;
763         }
764         else if( **s == '"' || **s == '\'' )
765         {
766             /* Preceeded by a number of '\' which we erase. */
767             *ppsz_parser -= i_bcount / 2;
768             if( i_bcount & 1 )
769             {
770                 /* '\\' followed by a '"' or '\'' */
771                 *ppsz_parser -= 1;
772                 **ppsz_parser = **s;
773                 (*ppsz_parser)++; (*s)++;
774                 i_bcount = 0;
775                 continue;
776             }
777
778             if( **s == i_quote )
779             {
780                 /* End */
781                 return;
782             }
783             else
784             {
785                 /* Different quoting */
786                 int i_quote = **s;
787                 **ppsz_parser = **s;
788                 (*ppsz_parser)++; (*s)++;
789                 find_end_quote( s, ppsz_parser, i_quote );
790                 **ppsz_parser = **s;
791                 (*ppsz_parser)++; (*s)++;
792             }
793
794             i_bcount = 0;
795         }
796         else
797         {
798             /* A regular character */
799             **ppsz_parser = **s;
800             (*ppsz_parser)++; (*s)++;
801             i_bcount = 0;
802         }
803     }
804 }
805
806 char **vlc_parse_cmdline( const char *psz_cmdline, int *i_args )
807 {
808     int argc = 0;
809     char **argv = 0;
810     char *s, *psz_parser, *psz_arg, *psz_orig;
811     int i_bcount = 0;
812
813     if( !psz_cmdline ) return 0;
814     psz_orig = strdup( psz_cmdline );
815     psz_arg = psz_parser = s = psz_orig;
816
817     while( *s )
818     {
819         if( *s == '\t' || *s == ' ' )
820         {
821             /* We have a complete argument */
822             *psz_parser = 0;
823             TAB_APPEND( argc, argv, strdup(psz_arg) );
824
825             /* Skip trailing spaces/tabs */
826             do{ s++; } while( *s == '\t' || *s == ' ' );
827
828             /* New argument */
829             psz_arg = psz_parser = s;
830             i_bcount = 0;
831         }
832         else if( *s == '\\' )
833         {
834             *psz_parser++ = *s++;
835             i_bcount++;
836         }
837         else if( *s == '"' || *s == '\'' )
838         {
839             if( ( i_bcount & 1 ) == 0 )
840             {
841                 /* Preceeded by an even number of '\', this is half that
842                  * number of '\', plus a quote which we erase. */
843                 int i_quote = *s;
844                 psz_parser -= i_bcount / 2;
845                 s++;
846                 find_end_quote( &s, &psz_parser, i_quote );
847                 s++;
848             }
849             else
850             {
851                 /* Preceeded by an odd number of '\', this is half that
852                  * number of '\' followed by a '"' */
853                 psz_parser = psz_parser - i_bcount/2 - 1;
854                 *psz_parser++ = '"';
855                 s++;
856             }
857             i_bcount = 0;
858         }
859         else
860         {
861             /* A regular character */
862             *psz_parser++ = *s++;
863             i_bcount = 0;
864         }
865     }
866
867     /* Take care of the last arg */
868     if( *psz_arg )
869     {
870         *psz_parser = '\0';
871         TAB_APPEND( argc, argv, strdup(psz_arg) );
872     }
873
874     if( i_args ) *i_args = argc;
875     free( psz_orig );
876     return argv;
877 }
878
879 /*************************************************************************
880  * vlc_execve: Execute an external program with a given environment,
881  * wait until it finishes and return its standard output
882  *************************************************************************/
883 int __vlc_execve( vlc_object_t *p_object, int i_argc, char **ppsz_argv,
884                   char **ppsz_env, char *psz_cwd, char *p_in, int i_in,
885                   char **pp_data, int *pi_data )
886 {
887 #ifdef HAVE_FORK
888     int pi_stdin[2];
889     int pi_stdout[2];
890     pid_t i_child_pid;
891
892     pipe( pi_stdin );
893     pipe( pi_stdout );
894
895     if ( (i_child_pid = fork()) == -1 )
896     {
897         msg_Err( p_object, "unable to fork (%s)", strerror(errno) );
898         return -1;
899     }
900
901     if ( i_child_pid == 0 )
902     {
903         close(0);
904         dup(pi_stdin[1]);
905         close(pi_stdin[0]);
906
907         close(1);
908         dup(pi_stdout[1]);
909         close(pi_stdout[0]);
910
911         close(2);
912
913         if ( psz_cwd != NULL )
914             chdir( psz_cwd );
915         execve( ppsz_argv[0], ppsz_argv, ppsz_env );
916         exit(1);
917     }
918
919     close(pi_stdin[1]);
920     close(pi_stdout[1]);
921     if ( !i_in )
922         close( pi_stdin[0] );
923
924     *pi_data = 0;
925     *pp_data = malloc( 1025 );  /* +1 for \0 */
926
927     while ( !p_object->b_die )
928     {
929         int i_ret, i_status;
930         fd_set readfds, writefds;
931         struct timeval tv;
932
933         FD_ZERO( &readfds );
934         FD_ZERO( &writefds );
935         FD_SET( pi_stdout[0], &readfds );
936         if ( i_in )
937             FD_SET( pi_stdin[0], &writefds );
938
939         tv.tv_sec = 0;
940         tv.tv_usec = 10000;
941
942         i_ret = select( pi_stdin[0] > pi_stdout[0] ? pi_stdin[0] + 1 :
943                         pi_stdout[0] + 1, &readfds, &writefds, NULL, &tv );
944         if ( i_ret > 0 )
945         {
946             if ( FD_ISSET( pi_stdout[0], &readfds ) )
947             {
948                 ssize_t i_read = read( pi_stdout[0], &(*pp_data)[*pi_data],
949                                        1024 );
950                 if ( i_read > 0 )
951                 {
952                     *pi_data += i_read;
953                     *pp_data = realloc( *pp_data, *pi_data + 1025 );
954                 }
955             }
956             if ( FD_ISSET( pi_stdin[0], &writefds ) )
957             {
958                 ssize_t i_write = write( pi_stdin[0], p_in, __MIN(i_in, 1024) );
959
960                 if ( i_write > 0 )
961                 {
962                     p_in += i_write;
963                     i_in -= i_write;
964                 }
965                 if ( !i_in )
966                     close( pi_stdin[0] );
967             }
968         }
969
970         if ( waitpid( i_child_pid, &i_status, WNOHANG ) == i_child_pid )
971         {
972             if ( WIFEXITED( i_status ) )
973             {
974                 if ( WEXITSTATUS( i_status ) )
975                 {
976                     msg_Warn( p_object,
977                               "child %s returned with error code %d",
978                               ppsz_argv[0], WEXITSTATUS( i_status ) );
979                 }
980             }
981             else
982             {
983                 if ( WIFSIGNALED( i_status ) )
984                 {
985                     msg_Warn( p_object,
986                               "child %s quit on signal %d", ppsz_argv[0],
987                               WTERMSIG( i_status ) );
988                 }
989             }
990             if ( i_in )
991                 close( pi_stdin[0] );
992             close( pi_stdout[0] );
993             break;
994         }
995
996         if ( i_ret < 0 && errno != EINTR )
997         {
998             msg_Warn( p_object, "select failed (%s)", strerror(errno) );
999         }
1000     }
1001
1002 #elif defined( WIN32 ) && !defined( UNDER_CE )
1003     SECURITY_ATTRIBUTES saAttr;
1004     PROCESS_INFORMATION piProcInfo;
1005     STARTUPINFO siStartInfo;
1006     BOOL bFuncRetn = FALSE;
1007     HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr;
1008     DWORD i_status;
1009     char *psz_cmd, *p_env, *p;
1010     char **ppsz_parser;
1011     int i_size;
1012
1013     /* Set the bInheritHandle flag so pipe handles are inherited. */
1014     saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
1015     saAttr.bInheritHandle = TRUE;
1016     saAttr.lpSecurityDescriptor = NULL;
1017
1018     /* Create a pipe for the child process's STDOUT. */
1019     if ( !CreatePipe( &hChildStdoutRd, &hChildStdoutWr, &saAttr, 0 ) )
1020     {
1021         msg_Err( p_object, "stdout pipe creation failed" );
1022         return -1;
1023     }
1024
1025     /* Ensure the read handle to the pipe for STDOUT is not inherited. */
1026     SetHandleInformation( hChildStdoutRd, HANDLE_FLAG_INHERIT, 0 );
1027
1028     /* Create a pipe for the child process's STDIN. */
1029     if ( !CreatePipe( &hChildStdinRd, &hChildStdinWr, &saAttr, 0 ) )
1030     {
1031         msg_Err( p_object, "stdin pipe creation failed" );
1032         return -1;
1033     }
1034
1035     /* Ensure the write handle to the pipe for STDIN is not inherited. */
1036     SetHandleInformation( hChildStdinWr, HANDLE_FLAG_INHERIT, 0 );
1037
1038     /* Set up members of the PROCESS_INFORMATION structure. */
1039     ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
1040
1041     /* Set up members of the STARTUPINFO structure. */
1042     ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
1043     siStartInfo.cb = sizeof(STARTUPINFO);
1044     siStartInfo.hStdError = hChildStdoutWr;
1045     siStartInfo.hStdOutput = hChildStdoutWr;
1046     siStartInfo.hStdInput = hChildStdinRd;
1047     siStartInfo.wShowWindow = SW_HIDE;
1048     siStartInfo.dwFlags |= STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
1049
1050     /* Set up the command line. */
1051     psz_cmd = malloc(32768);
1052     psz_cmd[0] = '\0';
1053     i_size = 32768;
1054     ppsz_parser = &ppsz_argv[0];
1055     while ( ppsz_parser[0] != NULL && i_size > 0 )
1056     {
1057         /* Protect the last argument with quotes ; the other arguments
1058          * are supposed to be already protected because they have been
1059          * passed as a command-line option. */
1060         if ( ppsz_parser[1] == NULL )
1061         {
1062             strncat( psz_cmd, "\"", i_size );
1063             i_size--;
1064         }
1065         strncat( psz_cmd, *ppsz_parser, i_size );
1066         i_size -= strlen( *ppsz_parser );
1067         if ( ppsz_parser[1] == NULL )
1068         {
1069             strncat( psz_cmd, "\"", i_size );
1070             i_size--;
1071         }
1072         strncat( psz_cmd, " ", i_size );
1073         i_size--;
1074         ppsz_parser++;
1075     }
1076
1077     /* Set up the environment. */
1078     p = p_env = malloc(32768);
1079     i_size = 32768;
1080     ppsz_parser = &ppsz_env[0];
1081     while ( *ppsz_parser != NULL && i_size > 0 )
1082     {
1083         memcpy( p, *ppsz_parser,
1084                 __MIN((int)(strlen(*ppsz_parser) + 1), i_size) );
1085         p += strlen(*ppsz_parser) + 1;
1086         i_size -= strlen(*ppsz_parser) + 1;
1087         ppsz_parser++;
1088     }
1089     *p = '\0';
1090
1091     /* Create the child process. */
1092     bFuncRetn = CreateProcess( NULL,
1093           psz_cmd,       // command line
1094           NULL,          // process security attributes
1095           NULL,          // primary thread security attributes
1096           TRUE,          // handles are inherited
1097           0,             // creation flags
1098           p_env,
1099           psz_cwd,
1100           &siStartInfo,  // STARTUPINFO pointer
1101           &piProcInfo ); // receives PROCESS_INFORMATION
1102
1103     free( psz_cmd );
1104     free( p_env );
1105
1106     if ( bFuncRetn == 0 )
1107     {
1108         msg_Err( p_object, "child creation failed" );
1109         return -1;
1110     }
1111
1112     /* Read from a file and write its contents to a pipe. */
1113     while ( i_in > 0 && !p_object->b_die )
1114     {
1115         DWORD i_written;
1116         if ( !WriteFile( hChildStdinWr, p_in, i_in, &i_written, NULL ) )
1117             break;
1118         i_in -= i_written;
1119         p_in += i_written;
1120     }
1121
1122     /* Close the pipe handle so the child process stops reading. */
1123     CloseHandle(hChildStdinWr);
1124
1125     /* Close the write end of the pipe before reading from the
1126      * read end of the pipe. */
1127     CloseHandle(hChildStdoutWr);
1128
1129     /* Read output from the child process. */
1130     *pi_data = 0;
1131     *pp_data = malloc( 1025 );  /* +1 for \0 */
1132
1133     while ( !p_object->b_die )
1134     {
1135         DWORD i_read;
1136         if ( !ReadFile( hChildStdoutRd, &(*pp_data)[*pi_data], 1024, &i_read,
1137                         NULL )
1138               || i_read == 0 )
1139             break;
1140         *pi_data += i_read;
1141         *pp_data = realloc( *pp_data, *pi_data + 1025 );
1142     }
1143
1144     while ( !p_object->b_die
1145              && !GetExitCodeProcess( piProcInfo.hProcess, &i_status )
1146              && i_status != STILL_ACTIVE )
1147         msleep( 10000 );
1148
1149     CloseHandle(piProcInfo.hProcess);
1150     CloseHandle(piProcInfo.hThread);
1151
1152     if ( i_status )
1153         msg_Warn( p_object,
1154                   "child %s returned with error code %ld",
1155                   ppsz_argv[0], i_status );
1156
1157 #else
1158     msg_Err( p_object, "vlc_execve called but no implementation is available" );
1159     return -1;
1160
1161 #endif
1162
1163     (*pp_data)[*pi_data] = '\0';
1164
1165     return 0;
1166 }