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