]> git.sesse.net Git - vlc/blob - src/extras/libc.c
OpenBSDish strlcpy()
[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 lldiv_t vlc_lldiv( long long numer, long long denom )
352 {
353     lldiv_t d;
354     d.quot = numer / denom;
355     d.rem  = numer % denom;
356     return d;
357 }
358 #endif
359
360
361 /**
362  * Copy a string to a sized buffer. The result is always nul-terminated
363  * (contrary to strncpy()).
364  *
365  * @param dest destination buffer
366  * @param src string to be copied
367  * @param len maximum number of characters to be copied plus one for the
368  * terminating nul.
369  *
370  * @return strlen(src)
371  */
372 #ifndef HAVE_STRLCPY
373 extern size_t vlc_strlcpy (char *tgt, const char *src, size_t bufsize)
374 {
375     size_t length;
376
377     for (length = 1; (length < bufsize) && *src; length++)
378         *tgt++ = *src++;
379
380     if (bufsize)
381         *tgt = '\0';
382
383     while (*src++)
384         length++;
385
386     return length - 1;
387 }
388 #endif
389
390 /*****************************************************************************
391  * vlc_*dir_wrapper: wrapper under Windows to return the list of drive letters
392  * when called with an empty argument or just '\'
393  *****************************************************************************/
394 #if defined(WIN32) && !defined(UNDER_CE)
395 typedef struct vlc_DIR
396 {
397     DIR *p_real_dir;
398     int i_drives;
399     struct dirent dd_dir;
400     vlc_bool_t b_insert_back;
401 } vlc_DIR;
402
403 void *vlc_opendir_wrapper( const char *psz_path )
404 {
405     vlc_DIR *p_dir;
406     DIR *p_real_dir;
407
408     if ( psz_path == NULL || psz_path[0] == '\0'
409           || (psz_path[0] == '\\' && psz_path[1] == '\0') )
410     {
411         /* Special mode to list drive letters */
412         p_dir = malloc( sizeof(vlc_DIR) );
413         p_dir->p_real_dir = NULL;
414         p_dir->i_drives = GetLogicalDrives();
415         return (void *)p_dir;
416     }
417
418     p_real_dir = opendir( psz_path );
419     if ( p_real_dir == NULL )
420         return NULL;
421
422     p_dir = malloc( sizeof(vlc_DIR) );
423     p_dir->p_real_dir = p_real_dir;
424     p_dir->b_insert_back = ( psz_path[1] == ':' && psz_path[2] == '\\'
425                               && psz_path[3] =='\0' );
426     return (void *)p_dir;
427 }
428
429 struct dirent *vlc_readdir_wrapper( void *_p_dir )
430 {
431     vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
432     unsigned int i;
433     DWORD i_drives;
434
435     if ( p_dir->p_real_dir != NULL )
436     {
437         if ( p_dir->b_insert_back )
438         {
439             p_dir->dd_dir.d_ino = 0;
440             p_dir->dd_dir.d_reclen = 0;
441             p_dir->dd_dir.d_namlen = 2;
442             strcpy( p_dir->dd_dir.d_name, ".." );
443             p_dir->b_insert_back = VLC_FALSE;
444             return &p_dir->dd_dir;
445         }
446
447         return readdir( p_dir->p_real_dir );
448     }
449
450     /* Drive letters mode */
451     i_drives = p_dir->i_drives;
452     if ( !i_drives )
453         return NULL; /* end */
454
455     for ( i = 0; i < sizeof(DWORD)*8; i++, i_drives >>= 1 )
456         if ( i_drives & 1 ) break;
457
458     if ( i >= 26 )
459         return NULL; /* this should not happen */
460
461     sprintf( p_dir->dd_dir.d_name, "%c:\\", 'A' + i );
462     p_dir->dd_dir.d_namlen = strlen(p_dir->dd_dir.d_name);
463     p_dir->i_drives &= ~(1UL << i);
464     return &p_dir->dd_dir;
465 }
466
467 int vlc_closedir_wrapper( void *_p_dir )
468 {
469     vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
470
471     if ( p_dir->p_real_dir != NULL )
472     {
473         int i_ret = closedir( p_dir->p_real_dir );
474         free( p_dir );
475         return i_ret;
476     }
477
478     free( p_dir );
479     return 0;
480 }
481 #else
482 void *vlc_opendir_wrapper( const char *psz_path )
483 {
484     return (void *)opendir( psz_path );
485 }
486 struct dirent *vlc_readdir_wrapper( void *_p_dir )
487 {
488     return readdir( (DIR *)_p_dir );
489 }
490 int vlc_closedir_wrapper( void *_p_dir )
491 {
492     return closedir( (DIR *)_p_dir );
493 }
494 #endif
495
496 /*****************************************************************************
497  * scandir: scan a directory alpha-sorted
498  *****************************************************************************/
499 #if !defined( HAVE_SCANDIR )
500 int vlc_alphasort( const struct dirent **a, const struct dirent **b )
501 {
502     return strcoll( (*a)->d_name, (*b)->d_name );
503 }
504
505 int vlc_scandir( const char *name, struct dirent ***namelist,
506                     int (*filter) ( const struct dirent * ),
507                     int (*compar) ( const struct dirent **,
508                                     const struct dirent ** ) )
509 {
510     DIR            * p_dir;
511     struct dirent  * p_content;
512     struct dirent ** pp_list;
513     int              ret, size;
514
515     if( !namelist || !( p_dir = vlc_opendir_wrapper( name ) ) ) return -1;
516
517     ret     = 0;
518     pp_list = NULL;
519     while( ( p_content = vlc_readdir_wrapper( p_dir ) ) )
520     {
521         if( filter && !filter( p_content ) )
522         {
523             continue;
524         }
525         pp_list = realloc( pp_list, ( ret + 1 ) * sizeof( struct dirent * ) );
526         size = sizeof( struct dirent ) + strlen( p_content->d_name ) + 1;
527         pp_list[ret] = malloc( size );
528         memcpy( pp_list[ret], p_content, size );
529         ret++;
530     }
531
532     vlc_closedir_wrapper( p_dir );
533
534     if( compar )
535     {
536         qsort( pp_list, ret, sizeof( struct dirent * ),
537                (int (*)(const void *, const void *)) compar );
538     }
539
540     *namelist = pp_list;
541     return ret;
542 }
543 #endif
544
545 #if defined (WIN32) || !defined (HAVE_SHARED_LIBVLC)
546 /*****************************************************************************
547  * dgettext: gettext for plugins.
548  *****************************************************************************/
549 char *vlc_dgettext( const char *package, const char *msgid )
550 {
551 #if defined( ENABLE_NLS ) \
552      && ( defined(HAVE_GETTEXT) || defined(HAVE_INCLUDED_GETTEXT) )
553     return dgettext( package, msgid );
554 #else
555     return (char *)msgid;
556 #endif
557 }
558 #endif
559
560 /*****************************************************************************
561  * count_utf8_string: returns the number of characters in the string.
562  *****************************************************************************/
563 static int count_utf8_string( const char *psz_string )
564 {
565     int i = 0, i_count = 0;
566     while( psz_string[ i ] != 0 )
567     {
568         if( ((unsigned char *)psz_string)[ i ] <  0x80UL ) i_count++;
569         i++;
570     }
571     return i_count;
572 }
573
574 /*****************************************************************************
575  * wraptext: inserts \n at convenient places to wrap the text.
576  *           Returns the modified string in a new buffer.
577  *****************************************************************************/
578 char *vlc_wraptext( const char *psz_text, int i_line )
579 {
580     int i_len;
581     char *psz_line, *psz_new_text;
582
583     psz_line = psz_new_text = strdup( psz_text );
584
585     i_len = count_utf8_string( psz_text );
586
587     while( i_len > i_line )
588     {
589         /* Look if there is a newline somewhere. */
590         char *psz_parser = psz_line;
591         int i_count = 0;
592         while( i_count <= i_line && *psz_parser != '\n' )
593         {
594             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
595             psz_parser++;
596             i_count++;
597         }
598         if( *psz_parser == '\n' )
599         {
600             i_len -= (i_count + 1);
601             psz_line = psz_parser + 1;
602             continue;
603         }
604
605         /* Find the furthest space. */
606         while( psz_parser > psz_line && *psz_parser != ' ' )
607         {
608             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser--;
609             psz_parser--;
610             i_count--;
611         }
612         if( *psz_parser == ' ' )
613         {
614             *psz_parser = '\n';
615             i_len -= (i_count + 1);
616             psz_line = psz_parser + 1;
617             continue;
618         }
619
620         /* Wrapping has failed. Find the first space or newline */
621         while( i_count < i_len && *psz_parser != ' ' && *psz_parser != '\n' )
622         {
623             while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
624             psz_parser++;
625             i_count++;
626         }
627         if( i_count < i_len ) *psz_parser = '\n';
628         i_len -= (i_count + 1);
629         psz_line = psz_parser + 1;
630     }
631
632     return psz_new_text;
633 }
634
635 /*****************************************************************************
636  * iconv wrapper
637  *****************************************************************************/
638 vlc_iconv_t vlc_iconv_open( const char *tocode, const char *fromcode )
639 {
640 #if defined(HAVE_ICONV)
641     return iconv_open( tocode, fromcode );
642 #else
643     return NULL;
644 #endif
645 }
646
647 size_t vlc_iconv( vlc_iconv_t cd, const char **inbuf, size_t *inbytesleft,
648                   char **outbuf, size_t *outbytesleft )
649 {
650 #if defined(HAVE_ICONV)
651     return iconv( cd, (ICONV_CONST char **)inbuf, inbytesleft,
652                   outbuf, outbytesleft );
653 #else
654     int i_bytes;
655
656     if (inbytesleft == NULL || outbytesleft == NULL)
657     {
658         return 0;
659     }
660
661     i_bytes = __MIN(*inbytesleft, *outbytesleft);
662     if( !inbuf || !outbuf || !i_bytes ) return (size_t)(-1);
663     memcpy( *outbuf, *inbuf, i_bytes );
664     inbuf += i_bytes;
665     outbuf += i_bytes;
666     inbytesleft -= i_bytes;
667     outbytesleft -= i_bytes;
668     return i_bytes;
669 #endif
670 }
671
672 int vlc_iconv_close( vlc_iconv_t cd )
673 {
674 #if defined(HAVE_ICONV)
675     return iconv_close( cd );
676 #else
677     return 0;
678 #endif
679 }
680
681 /*****************************************************************************
682  * reduce a fraction
683  *   (adapted from libavcodec, author Michael Niedermayer <michaelni@gmx.at>)
684  *****************************************************************************/
685 vlc_bool_t vlc_ureduce( unsigned *pi_dst_nom, unsigned *pi_dst_den,
686                         uint64_t i_nom, uint64_t i_den, uint64_t i_max )
687 {
688     vlc_bool_t b_exact = 1;
689     uint64_t i_gcd;
690
691     if( i_den == 0 )
692     {
693         *pi_dst_nom = 0;
694         *pi_dst_den = 1;
695         return 1;
696     }
697
698     i_gcd = GCD( i_nom, i_den );
699     i_nom /= i_gcd;
700     i_den /= i_gcd;
701
702     if( i_max == 0 ) i_max = I64C(0xFFFFFFFF);
703
704     if( i_nom > i_max || i_den > i_max )
705     {
706         uint64_t i_a0_num = 0, i_a0_den = 1, i_a1_num = 1, i_a1_den = 0;
707         b_exact = 0;
708
709         for( ; ; )
710         {
711             uint64_t i_x = i_nom / i_den;
712             uint64_t i_a2n = i_x * i_a1_num + i_a0_num;
713             uint64_t i_a2d = i_x * i_a1_den + i_a0_den;
714
715             if( i_a2n > i_max || i_a2d > i_max ) break;
716
717             i_nom %= i_den;
718
719             i_a0_num = i_a1_num; i_a0_den = i_a1_den;
720             i_a1_num = i_a2n; i_a1_den = i_a2d;
721             if( i_nom == 0 ) break;
722             i_x = i_nom; i_nom = i_den; i_den = i_x;
723         }
724         i_nom = i_a1_num;
725         i_den = i_a1_den;
726     }
727
728     *pi_dst_nom = i_nom;
729     *pi_dst_den = i_den;
730
731     return b_exact;
732 }
733
734 /*************************************************************************
735  * vlc_parse_cmdline: Command line parsing into elements.
736  *
737  * The command line is composed of space/tab separated arguments.
738  * Quotes can be used as argument delimiters and a backslash can be used to
739  * escape a quote.
740  *************************************************************************/
741 static void find_end_quote( char **s, char **ppsz_parser, int i_quote )
742 {
743     int i_bcount = 0;
744
745     while( **s )
746     {
747         if( **s == '\\' )
748         {
749             **ppsz_parser = **s;
750             (*ppsz_parser)++; (*s)++;
751             i_bcount++;
752         }
753         else if( **s == '"' || **s == '\'' )
754         {
755             /* Preceeded by a number of '\' which we erase. */
756             *ppsz_parser -= i_bcount / 2;
757             if( i_bcount & 1 )
758             {
759                 /* '\\' followed by a '"' or '\'' */
760                 *ppsz_parser -= 1;
761                 **ppsz_parser = **s;
762                 (*ppsz_parser)++; (*s)++;
763                 i_bcount = 0;
764                 continue;
765             }
766
767             if( **s == i_quote )
768             {
769                 /* End */
770                 return;
771             }
772             else
773             {
774                 /* Different quoting */
775                 int i_quote = **s;
776                 **ppsz_parser = **s;
777                 (*ppsz_parser)++; (*s)++;
778                 find_end_quote( s, ppsz_parser, i_quote );
779                 **ppsz_parser = **s;
780                 (*ppsz_parser)++; (*s)++;
781             }
782
783             i_bcount = 0;
784         }
785         else
786         {
787             /* A regular character */
788             **ppsz_parser = **s;
789             (*ppsz_parser)++; (*s)++;
790             i_bcount = 0;
791         }
792     }
793 }
794
795 char **vlc_parse_cmdline( const char *psz_cmdline, int *i_args )
796 {
797     int argc = 0;
798     char **argv = 0;
799     char *s, *psz_parser, *psz_arg, *psz_orig;
800     int i_bcount = 0;
801
802     if( !psz_cmdline ) return 0;
803     psz_orig = strdup( psz_cmdline );
804     psz_arg = psz_parser = s = psz_orig;
805
806     while( *s )
807     {
808         if( *s == '\t' || *s == ' ' )
809         {
810             /* We have a complete argument */
811             *psz_parser = 0;
812             TAB_APPEND( argc, argv, strdup(psz_arg) );
813
814             /* Skip trailing spaces/tabs */
815             do{ s++; } while( *s == '\t' || *s == ' ' );
816
817             /* New argument */
818             psz_arg = psz_parser = s;
819             i_bcount = 0;
820         }
821         else if( *s == '\\' )
822         {
823             *psz_parser++ = *s++;
824             i_bcount++;
825         }
826         else if( *s == '"' || *s == '\'' )
827         {
828             if( ( i_bcount & 1 ) == 0 )
829             {
830                 /* Preceeded by an even number of '\', this is half that
831                  * number of '\', plus a quote which we erase. */
832                 int i_quote = *s;
833                 psz_parser -= i_bcount / 2;
834                 s++;
835                 find_end_quote( &s, &psz_parser, i_quote );
836                 s++;
837             }
838             else
839             {
840                 /* Preceeded by an odd number of '\', this is half that
841                  * number of '\' followed by a '"' */
842                 psz_parser = psz_parser - i_bcount/2 - 1;
843                 *psz_parser++ = '"';
844                 s++;
845             }
846             i_bcount = 0;
847         }
848         else
849         {
850             /* A regular character */
851             *psz_parser++ = *s++;
852             i_bcount = 0;
853         }
854     }
855
856     /* Take care of the last arg */
857     if( *psz_arg )
858     {
859         *psz_parser = '\0';
860         TAB_APPEND( argc, argv, strdup(psz_arg) );
861     }
862
863     if( i_args ) *i_args = argc;
864     free( psz_orig );
865     return argv;
866 }
867
868 /*************************************************************************
869  * vlc_execve: Execute an external program with a given environment,
870  * wait until it finishes and return its standard output
871  *************************************************************************/
872 int __vlc_execve( vlc_object_t *p_object, int i_argc, char **ppsz_argv,
873                   char **ppsz_env, char *psz_cwd, char *p_in, int i_in,
874                   char **pp_data, int *pi_data )
875 {
876 #ifdef HAVE_FORK
877     int pi_stdin[2];
878     int pi_stdout[2];
879     pid_t i_child_pid;
880
881     pipe( pi_stdin );
882     pipe( pi_stdout );
883
884     if ( (i_child_pid = fork()) == -1 )
885     {
886         msg_Err( p_object, "unable to fork (%s)", strerror(errno) );
887         return -1;
888     }
889
890     if ( i_child_pid == 0 )
891     {
892         close(0);
893         dup(pi_stdin[1]);
894         close(pi_stdin[0]);
895
896         close(1);
897         dup(pi_stdout[1]);
898         close(pi_stdout[0]);
899
900         close(2);
901
902         if ( psz_cwd != NULL )
903             chdir( psz_cwd );
904         execve( ppsz_argv[0], ppsz_argv, ppsz_env );
905         exit(1);
906     }
907
908     close(pi_stdin[1]);
909     close(pi_stdout[1]);
910     if ( !i_in )
911         close( pi_stdin[0] );
912
913     *pi_data = 0;
914     *pp_data = malloc( 1025 );  /* +1 for \0 */
915
916     while ( !p_object->b_die )
917     {
918         int i_ret, i_status;
919         fd_set readfds, writefds;
920         struct timeval tv;
921
922         FD_ZERO( &readfds );
923         FD_ZERO( &writefds );
924         FD_SET( pi_stdout[0], &readfds );
925         if ( i_in )
926             FD_SET( pi_stdin[0], &writefds );
927
928         tv.tv_sec = 0;
929         tv.tv_usec = 10000;
930
931         i_ret = select( pi_stdin[0] > pi_stdout[0] ? pi_stdin[0] + 1 :
932                         pi_stdout[0] + 1, &readfds, &writefds, NULL, &tv );
933         if ( i_ret > 0 )
934         {
935             if ( FD_ISSET( pi_stdout[0], &readfds ) )
936             {
937                 ssize_t i_read = read( pi_stdout[0], &(*pp_data)[*pi_data],
938                                        1024 );
939                 if ( i_read > 0 )
940                 {
941                     *pi_data += i_read;
942                     *pp_data = realloc( *pp_data, *pi_data + 1025 );
943                 }
944             }
945             if ( FD_ISSET( pi_stdin[0], &writefds ) )
946             {
947                 ssize_t i_write = write( pi_stdin[0], p_in, __MIN(i_in, 1024) );
948
949                 if ( i_write > 0 )
950                 {
951                     p_in += i_write;
952                     i_in -= i_write;
953                 }
954                 if ( !i_in )
955                     close( pi_stdin[0] );
956             }
957         }
958
959         if ( waitpid( i_child_pid, &i_status, WNOHANG ) == i_child_pid )
960         {
961             if ( WIFEXITED( i_status ) )
962             {
963                 if ( WEXITSTATUS( i_status ) )
964                 {
965                     msg_Warn( p_object,
966                               "child %s returned with error code %d",
967                               ppsz_argv[0], WEXITSTATUS( i_status ) );
968                 }
969             }
970             else
971             {
972                 if ( WIFSIGNALED( i_status ) )
973                 {
974                     msg_Warn( p_object,
975                               "child %s quit on signal %d", ppsz_argv[0],
976                               WTERMSIG( i_status ) );
977                 }
978             }
979             if ( i_in )
980                 close( pi_stdin[0] );
981             close( pi_stdout[0] );
982             break;
983         }
984
985         if ( i_ret < 0 && errno != EINTR )
986         {
987             msg_Warn( p_object, "select failed (%s)", strerror(errno) );
988         }
989     }
990
991 #elif defined( WIN32 ) && !defined( UNDER_CE )
992     SECURITY_ATTRIBUTES saAttr;
993     PROCESS_INFORMATION piProcInfo;
994     STARTUPINFO siStartInfo;
995     BOOL bFuncRetn = FALSE;
996     HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr;
997     DWORD i_status;
998     char *psz_cmd, *p_env, *p;
999     char **ppsz_parser;
1000     int i_size;
1001
1002     /* Set the bInheritHandle flag so pipe handles are inherited. */
1003     saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
1004     saAttr.bInheritHandle = TRUE;
1005     saAttr.lpSecurityDescriptor = NULL;
1006
1007     /* Create a pipe for the child process's STDOUT. */
1008     if ( !CreatePipe( &hChildStdoutRd, &hChildStdoutWr, &saAttr, 0 ) )
1009     {
1010         msg_Err( p_object, "stdout pipe creation failed" );
1011         return -1;
1012     }
1013
1014     /* Ensure the read handle to the pipe for STDOUT is not inherited. */
1015     SetHandleInformation( hChildStdoutRd, HANDLE_FLAG_INHERIT, 0 );
1016
1017     /* Create a pipe for the child process's STDIN. */
1018     if ( !CreatePipe( &hChildStdinRd, &hChildStdinWr, &saAttr, 0 ) )
1019     {
1020         msg_Err( p_object, "stdin pipe creation failed" );
1021         return -1;
1022     }
1023
1024     /* Ensure the write handle to the pipe for STDIN is not inherited. */
1025     SetHandleInformation( hChildStdinWr, HANDLE_FLAG_INHERIT, 0 );
1026
1027     /* Set up members of the PROCESS_INFORMATION structure. */
1028     ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
1029
1030     /* Set up members of the STARTUPINFO structure. */
1031     ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
1032     siStartInfo.cb = sizeof(STARTUPINFO);
1033     siStartInfo.hStdError = hChildStdoutWr;
1034     siStartInfo.hStdOutput = hChildStdoutWr;
1035     siStartInfo.hStdInput = hChildStdinRd;
1036     siStartInfo.wShowWindow = SW_HIDE;
1037     siStartInfo.dwFlags |= STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
1038
1039     /* Set up the command line. */
1040     psz_cmd = malloc(32768);
1041     psz_cmd[0] = '\0';
1042     i_size = 32768;
1043     ppsz_parser = &ppsz_argv[0];
1044     while ( ppsz_parser[0] != NULL && i_size > 0 )
1045     {
1046         /* Protect the last argument with quotes ; the other arguments
1047          * are supposed to be already protected because they have been
1048          * passed as a command-line option. */
1049         if ( ppsz_parser[1] == NULL )
1050         {
1051             strncat( psz_cmd, "\"", i_size );
1052             i_size--;
1053         }
1054         strncat( psz_cmd, *ppsz_parser, i_size );
1055         i_size -= strlen( *ppsz_parser );
1056         if ( ppsz_parser[1] == NULL )
1057         {
1058             strncat( psz_cmd, "\"", i_size );
1059             i_size--;
1060         }
1061         strncat( psz_cmd, " ", i_size );
1062         i_size--;
1063         ppsz_parser++;
1064     }
1065
1066     /* Set up the environment. */
1067     p = p_env = malloc(32768);
1068     i_size = 32768;
1069     ppsz_parser = &ppsz_env[0];
1070     while ( *ppsz_parser != NULL && i_size > 0 )
1071     {
1072         memcpy( p, *ppsz_parser,
1073                 __MIN((int)(strlen(*ppsz_parser) + 1), i_size) );
1074         p += strlen(*ppsz_parser) + 1;
1075         i_size -= strlen(*ppsz_parser) + 1;
1076         ppsz_parser++;
1077     }
1078     *p = '\0';
1079
1080     /* Create the child process. */
1081     bFuncRetn = CreateProcess( NULL,
1082           psz_cmd,       // command line
1083           NULL,          // process security attributes
1084           NULL,          // primary thread security attributes
1085           TRUE,          // handles are inherited
1086           0,             // creation flags
1087           p_env,
1088           psz_cwd,
1089           &siStartInfo,  // STARTUPINFO pointer
1090           &piProcInfo ); // receives PROCESS_INFORMATION
1091
1092     free( psz_cmd );
1093     free( p_env );
1094
1095     if ( bFuncRetn == 0 )
1096     {
1097         msg_Err( p_object, "child creation failed" );
1098         return -1;
1099     }
1100
1101     /* Read from a file and write its contents to a pipe. */
1102     while ( i_in > 0 && !p_object->b_die )
1103     {
1104         DWORD i_written;
1105         if ( !WriteFile( hChildStdinWr, p_in, i_in, &i_written, NULL ) )
1106             break;
1107         i_in -= i_written;
1108         p_in += i_written;
1109     }
1110
1111     /* Close the pipe handle so the child process stops reading. */
1112     CloseHandle(hChildStdinWr);
1113
1114     /* Close the write end of the pipe before reading from the
1115      * read end of the pipe. */
1116     CloseHandle(hChildStdoutWr);
1117
1118     /* Read output from the child process. */
1119     *pi_data = 0;
1120     *pp_data = malloc( 1025 );  /* +1 for \0 */
1121
1122     while ( !p_object->b_die )
1123     {
1124         DWORD i_read;
1125         if ( !ReadFile( hChildStdoutRd, &(*pp_data)[*pi_data], 1024, &i_read,
1126                         NULL )
1127               || i_read == 0 )
1128             break;
1129         *pi_data += i_read;
1130         *pp_data = realloc( *pp_data, *pi_data + 1025 );
1131     }
1132
1133     while ( !p_object->b_die
1134              && !GetExitCodeProcess( piProcInfo.hProcess, &i_status )
1135              && i_status != STILL_ACTIVE )
1136         msleep( 10000 );
1137
1138     CloseHandle(piProcInfo.hProcess);
1139     CloseHandle(piProcInfo.hThread);
1140
1141     if ( i_status )
1142         msg_Warn( p_object,
1143                   "child %s returned with error code %ld",
1144                   ppsz_argv[0], i_status );
1145
1146 #else
1147     msg_Err( p_object, "vlc_execve called but no implementation is available" );
1148     return -1;
1149
1150 #endif
1151
1152     (*pp_data)[*pi_data] = '\0';
1153
1154     return 0;
1155 }