1 /*****************************************************************************
2 * libc.c: Extra libc function for some systems.
3 *****************************************************************************
4 * Copyright (C) 2002-2006 the VideoLAN team
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>
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.
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.
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 *****************************************************************************/
42 #if defined(HAVE_ICONV)
55 # include <sys/time.h>
58 # include <sys/wait.h>
60 # include <sys/socket.h>
61 # include <sys/poll.h>
64 #if defined(WIN32) || defined(UNDER_CE)
69 # define WIN32_LEAN_AND_MEAN
74 # define strcoll strcmp
77 /******************************************************************************
78 * strcasestr: find a substring (little) in another substring (big)
79 * Case sensitive. Return NULL if not found, return big if little == null
80 *****************************************************************************/
81 #if !defined( HAVE_STRCASESTR ) && !defined( HAVE_STRISTR )
82 char * vlc_strcasestr( const char *psz_big, const char *psz_little )
84 char *p_pos = (char *)psz_big;
86 if( !psz_big || !psz_little || !*psz_little ) return p_pos;
90 if( toupper( *p_pos ) == toupper( *psz_little ) )
92 char * psz_cur1 = p_pos + 1;
93 char * psz_cur2 = (char *)psz_little + 1;
94 while( *psz_cur1 && *psz_cur2 &&
95 toupper( *psz_cur1 ) == toupper( *psz_cur2 ) )
100 if( !*psz_cur2 ) return p_pos;
108 /*****************************************************************************
110 *****************************************************************************/
111 #if !defined(HAVE_VASPRINTF) || defined(__APPLE__) || defined(SYS_BEOS)
112 int vlc_vasprintf(char **strp, const char *fmt, va_list ap)
114 /* Guess we need no more than 100 bytes. */
116 char *p = malloc( i_size );
127 /* Try to print in the allocated space. */
128 n = vsnprintf( p, i_size, fmt, ap );
130 /* If that worked, return the string. */
131 if (n > -1 && n < i_size)
136 /* Else try again with more space. */
137 if (n > -1) /* glibc 2.1 */
139 i_size = n+1; /* precisely what is needed */
143 i_size *= 2; /* twice the old size */
145 if( (p = realloc( p, i_size ) ) == NULL)
154 /*****************************************************************************
156 *****************************************************************************/
157 #if !defined(HAVE_ASPRINTF) || defined(__APPLE__) || defined(SYS_BEOS)
158 int vlc_asprintf( char **strp, const char *fmt, ... )
163 va_start( args, fmt );
164 i_ret = vasprintf( strp, fmt, args );
171 /*****************************************************************************
172 * strtoll: convert a string to a 64 bits int.
173 *****************************************************************************/
174 #if !defined( HAVE_STRTOLL )
175 int64_t vlc_strtoll( const char *nptr, char **endptr, int base )
178 int sign = 1, newbase = base ? base : 10;
180 while( isspace(*nptr) ) nptr++;
188 /* Try to detect base */
201 if( base && newbase != base )
203 if( endptr ) *endptr = (char *)nptr;
210 while( *nptr >= '0' && *nptr <= '9' )
213 i_value += ( *nptr++ - '0' );
215 if( endptr ) *endptr = (char *)nptr;
219 while( (*nptr >= '0' && *nptr <= '9') ||
220 (*nptr >= 'a' && *nptr <= 'f') ||
221 (*nptr >= 'A' && *nptr <= 'F') )
224 if(*nptr >= '0' && *nptr <= '9') i_valc = *nptr - '0';
225 else if(*nptr >= 'a' && *nptr <= 'f') i_valc = *nptr - 'a' +10;
226 else if(*nptr >= 'A' && *nptr <= 'F') i_valc = *nptr - 'A' +10;
231 if( endptr ) *endptr = (char *)nptr;
235 i_value = strtol( nptr, endptr, newbase );
239 return i_value * sign;
244 * Copy a string to a sized buffer. The result is always nul-terminated
245 * (contrary to strncpy()).
247 * @param dest destination buffer
248 * @param src string to be copied
249 * @param len maximum number of characters to be copied plus one for the
252 * @return strlen(src)
255 extern size_t vlc_strlcpy (char *tgt, const char *src, size_t bufsize)
259 for (length = 1; (length < bufsize) && *src; length++)
272 /*****************************************************************************
273 * vlc_*dir_wrapper: wrapper under Windows to return the list of drive letters
274 * when called with an empty argument or just '\'
275 *****************************************************************************/
276 #if defined(WIN32) && !defined(UNDER_CE)
279 typedef struct vlc_DIR
283 struct _wdirent dd_dir;
287 void *vlc_wopendir( const wchar_t *wpath )
289 vlc_DIR *p_dir = NULL;
290 _WDIR *p_real_dir = NULL;
292 if ( wpath == NULL || wpath[0] == '\0'
293 || (wcscmp (wpath, L"\\") == 0) )
295 /* Special mode to list drive letters */
296 p_dir = malloc( sizeof(vlc_DIR) );
299 p_dir->p_real_dir = NULL;
300 p_dir->i_drives = GetLogicalDrives();
301 return (void *)p_dir;
304 p_real_dir = _wopendir( wpath );
305 if ( p_real_dir == NULL )
308 p_dir = malloc( sizeof(vlc_DIR) );
311 _wclosedir( p_real_dir );
314 p_dir->p_real_dir = p_real_dir;
316 assert (wpath[0]); // wpath[1] is defined
317 p_dir->b_insert_back = !wcscmp (wpath + 1, L":\\");
318 return (void *)p_dir;
321 struct _wdirent *vlc_wreaddir( void *_p_dir )
323 vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
327 if ( p_dir->p_real_dir != NULL )
329 if ( p_dir->b_insert_back )
331 /* Adds "..", gruik! */
332 p_dir->dd_dir.d_ino = 0;
333 p_dir->dd_dir.d_reclen = 0;
334 p_dir->dd_dir.d_namlen = 2;
335 wcscpy( p_dir->dd_dir.d_name, L".." );
336 p_dir->b_insert_back = false;
337 return &p_dir->dd_dir;
340 return _wreaddir( p_dir->p_real_dir );
343 /* Drive letters mode */
344 i_drives = p_dir->i_drives;
346 return NULL; /* end */
348 for ( i = 0; i < sizeof(DWORD)*8; i++, i_drives >>= 1 )
349 if ( i_drives & 1 ) break;
352 return NULL; /* this should not happen */
354 swprintf( p_dir->dd_dir.d_name, L"%c:\\", 'A' + i );
355 p_dir->dd_dir.d_namlen = wcslen(p_dir->dd_dir.d_name);
356 p_dir->i_drives &= ~(1UL << i);
357 return &p_dir->dd_dir;
360 int vlc_wclosedir( void *_p_dir )
362 vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
365 if ( p_dir->p_real_dir != NULL )
366 i_ret = _wclosedir( p_dir->p_real_dir );
372 void vlc_rewinddir( void *_p_dir )
374 vlc_DIR *p_dir = (vlc_DIR *)_p_dir;
376 if ( p_dir->p_real_dir != NULL )
377 _wrewinddir( p_dir->p_real_dir );
381 /*****************************************************************************
382 * scandir: scan a directory alpha-sorted
383 *****************************************************************************/
384 #if !defined( HAVE_SCANDIR )
385 /* FIXME: I suspect this is dead code -> utf8_scandir */
391 int vlc_alphasort( const struct dirent **a, const struct dirent **b )
393 return strcoll( (*a)->d_name, (*b)->d_name );
396 int vlc_scandir( const char *name, struct dirent ***namelist,
397 int (*filter) ( const struct dirent * ),
398 int (*compar) ( const struct dirent **,
399 const struct dirent ** ) )
402 struct dirent * p_content;
403 struct dirent ** pp_list;
406 if( !namelist || !( p_dir = opendir( name ) ) ) return -1;
410 while( ( p_content = readdir( p_dir ) ) )
412 if( filter && !filter( p_content ) )
416 pp_list = realloc( pp_list, ( ret + 1 ) * sizeof( struct dirent * ) );
417 size = sizeof( struct dirent ) + strlen( p_content->d_name ) + 1;
418 pp_list[ret] = malloc( size );
421 memcpy( pp_list[ret], p_content, size );
426 /* Continuing is useless when no more memory can be allocted,
427 * so better return what we have found.
438 qsort( pp_list, ret, sizeof( struct dirent * ),
439 (int (*)(const void *, const void *)) compar );
449 * gettext callbacks for plugins.
450 * LibVLC links libintl statically on Windows.
452 char *vlc_dgettext( const char *package, const char *msgid )
454 return dgettext( package, msgid );
459 * In-tree plugins share their gettext domain with LibVLC.
461 char *vlc_gettext( const char *msgid )
463 return dgettext( PACKAGE_NAME, msgid );
466 /*****************************************************************************
467 * count_utf8_string: returns the number of characters in the string.
468 *****************************************************************************/
469 static int count_utf8_string( const char *psz_string )
471 int i = 0, i_count = 0;
472 while( psz_string[ i ] != 0 )
474 if( ((unsigned char *)psz_string)[ i ] < 0x80UL ) i_count++;
480 /*****************************************************************************
481 * wraptext: inserts \n at convenient places to wrap the text.
482 * Returns the modified string in a new buffer.
483 *****************************************************************************/
484 char *vlc_wraptext( const char *psz_text, int i_line )
487 char *psz_line, *psz_new_text;
489 psz_line = psz_new_text = strdup( psz_text );
491 i_len = count_utf8_string( psz_text );
493 while( i_len > i_line )
495 /* Look if there is a newline somewhere. */
496 char *psz_parser = psz_line;
498 while( i_count <= i_line && *psz_parser != '\n' )
500 while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
504 if( *psz_parser == '\n' )
506 i_len -= (i_count + 1);
507 psz_line = psz_parser + 1;
511 /* Find the furthest space. */
512 while( psz_parser > psz_line && *psz_parser != ' ' )
514 while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser--;
518 if( *psz_parser == ' ' )
521 i_len -= (i_count + 1);
522 psz_line = psz_parser + 1;
526 /* Wrapping has failed. Find the first space or newline */
527 while( i_count < i_len && *psz_parser != ' ' && *psz_parser != '\n' )
529 while( *((unsigned char *)psz_parser) >= 0x80UL ) psz_parser++;
533 if( i_count < i_len ) *psz_parser = '\n';
534 i_len -= (i_count + 1);
535 psz_line = psz_parser + 1;
541 /*****************************************************************************
543 *****************************************************************************/
544 vlc_iconv_t vlc_iconv_open( const char *tocode, const char *fromcode )
546 #if defined(HAVE_ICONV)
547 return iconv_open( tocode, fromcode );
553 size_t vlc_iconv( vlc_iconv_t cd, const char **inbuf, size_t *inbytesleft,
554 char **outbuf, size_t *outbytesleft )
556 #if defined(HAVE_ICONV)
557 return iconv( cd, (ICONV_CONST char **)inbuf, inbytesleft,
558 outbuf, outbytesleft );
562 if (inbytesleft == NULL || outbytesleft == NULL)
567 i_bytes = __MIN(*inbytesleft, *outbytesleft);
568 if( !inbuf || !outbuf || !i_bytes ) return (size_t)(-1);
569 memcpy( *outbuf, *inbuf, i_bytes );
572 inbytesleft -= i_bytes;
573 outbytesleft -= i_bytes;
578 int vlc_iconv_close( vlc_iconv_t cd )
580 #if defined(HAVE_ICONV)
581 return iconv_close( cd );
587 /*****************************************************************************
589 * (adapted from libavcodec, author Michael Niedermayer <michaelni@gmx.at>)
590 *****************************************************************************/
591 bool vlc_ureduce( unsigned *pi_dst_nom, unsigned *pi_dst_den,
592 uint64_t i_nom, uint64_t i_den, uint64_t i_max )
604 i_gcd = GCD( i_nom, i_den );
608 if( i_max == 0 ) i_max = INT64_C(0xFFFFFFFF);
610 if( i_nom > i_max || i_den > i_max )
612 uint64_t i_a0_num = 0, i_a0_den = 1, i_a1_num = 1, i_a1_den = 0;
617 uint64_t i_x = i_nom / i_den;
618 uint64_t i_a2n = i_x * i_a1_num + i_a0_num;
619 uint64_t i_a2d = i_x * i_a1_den + i_a0_den;
621 if( i_a2n > i_max || i_a2d > i_max ) break;
625 i_a0_num = i_a1_num; i_a0_den = i_a1_den;
626 i_a1_num = i_a2n; i_a1_den = i_a2d;
627 if( i_nom == 0 ) break;
628 i_x = i_nom; i_nom = i_den; i_den = i_x;
640 /*************************************************************************
641 * vlc_execve: Execute an external program with a given environment,
642 * wait until it finishes and return its standard output
643 *************************************************************************/
644 int __vlc_execve( vlc_object_t *p_object, int i_argc, char *const *ppsz_argv,
645 char *const *ppsz_env, const char *psz_cwd,
646 const char *p_in, size_t i_in,
647 char **pp_data, size_t *pi_data )
649 (void)i_argc; // <-- hmph
651 # define BUFSIZE 1024
652 int fds[2], i_status;
654 if (socketpair (AF_LOCAL, SOCK_STREAM, 0, fds))
658 if ((fds[0] > 2) && (fds[1] > 2))
664 msg_Err (p_object, "unable to fork (%m)");
673 pthread_sigmask (SIG_SETMASK, &set, NULL);
676 * Like it or not, close can fail (and not only with EBADF)
678 if ((close (0) == 0) && (close (1) == 0) && (close (2) == 0)
679 && (dup (fds[1]) == 0) && (dup (fds[1]) == 1)
680 && (open ("/dev/null", O_RDONLY) == 2)
681 && ((psz_cwd == NULL) || (chdir (psz_cwd) == 0)))
682 execve (ppsz_argv[0], ppsz_argv, ppsz_env);
696 shutdown (fds[0], SHUT_WR);
698 while (!p_object->b_die)
700 struct pollfd ufd[1];
701 memset (ufd, 0, sizeof (ufd));
703 ufd[0].events = POLLIN;
706 ufd[0].events |= POLLOUT;
708 if (poll (ufd, 1, 10) <= 0)
711 if (ufd[0].revents & ~POLLOUT)
713 char *ptr = realloc (*pp_data, *pi_data + BUFSIZE + 1);
715 break; /* safely abort */
719 ssize_t val = read (fds[0], ptr + *pi_data, BUFSIZE);
724 shutdown (fds[0], SHUT_RD);
732 if (ufd[0].revents & POLLOUT)
734 ssize_t val = write (fds[0], p_in, i_in);
740 shutdown (fds[0], SHUT_WR);
752 while (waitpid (pid, &i_status, 0) == -1);
754 if (WIFEXITED (i_status))
756 i_status = WEXITSTATUS (i_status);
758 msg_Warn (p_object, "child %s (PID %d) exited with error code %d",
759 ppsz_argv[0], (int)pid, i_status);
762 if (WIFSIGNALED (i_status)) // <-- this should be redumdant a check
764 i_status = WTERMSIG (i_status);
765 msg_Warn (p_object, "child %s (PID %d) exited on signal %d (%s)",
766 ppsz_argv[0], (int)pid, i_status, strsignal (i_status));
769 #elif defined( WIN32 ) && !defined( UNDER_CE )
770 SECURITY_ATTRIBUTES saAttr;
771 PROCESS_INFORMATION piProcInfo;
772 STARTUPINFO siStartInfo;
773 BOOL bFuncRetn = FALSE;
774 HANDLE hChildStdinRd, hChildStdinWr, hChildStdoutRd, hChildStdoutWr;
776 char *psz_cmd = NULL, *p_env = NULL, *p = NULL;
780 /* Set the bInheritHandle flag so pipe handles are inherited. */
781 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
782 saAttr.bInheritHandle = TRUE;
783 saAttr.lpSecurityDescriptor = NULL;
785 /* Create a pipe for the child process's STDOUT. */
786 if ( !CreatePipe( &hChildStdoutRd, &hChildStdoutWr, &saAttr, 0 ) )
788 msg_Err( p_object, "stdout pipe creation failed" );
792 /* Ensure the read handle to the pipe for STDOUT is not inherited. */
793 SetHandleInformation( hChildStdoutRd, HANDLE_FLAG_INHERIT, 0 );
795 /* Create a pipe for the child process's STDIN. */
796 if ( !CreatePipe( &hChildStdinRd, &hChildStdinWr, &saAttr, 0 ) )
798 msg_Err( p_object, "stdin pipe creation failed" );
802 /* Ensure the write handle to the pipe for STDIN is not inherited. */
803 SetHandleInformation( hChildStdinWr, HANDLE_FLAG_INHERIT, 0 );
805 /* Set up members of the PROCESS_INFORMATION structure. */
806 ZeroMemory( &piProcInfo, sizeof(PROCESS_INFORMATION) );
808 /* Set up members of the STARTUPINFO structure. */
809 ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
810 siStartInfo.cb = sizeof(STARTUPINFO);
811 siStartInfo.hStdError = hChildStdoutWr;
812 siStartInfo.hStdOutput = hChildStdoutWr;
813 siStartInfo.hStdInput = hChildStdinRd;
814 siStartInfo.wShowWindow = SW_HIDE;
815 siStartInfo.dwFlags |= STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
817 /* Set up the command line. */
818 psz_cmd = malloc(32768);
823 ppsz_parser = &ppsz_argv[0];
824 while ( ppsz_parser[0] != NULL && i_size > 0 )
826 /* Protect the last argument with quotes ; the other arguments
827 * are supposed to be already protected because they have been
828 * passed as a command-line option. */
829 if ( ppsz_parser[1] == NULL )
831 strncat( psz_cmd, "\"", i_size );
834 strncat( psz_cmd, *ppsz_parser, i_size );
835 i_size -= strlen( *ppsz_parser );
836 if ( ppsz_parser[1] == NULL )
838 strncat( psz_cmd, "\"", i_size );
841 strncat( psz_cmd, " ", i_size );
846 /* Set up the environment. */
847 p = p_env = malloc(32768);
855 ppsz_parser = &ppsz_env[0];
856 while ( *ppsz_parser != NULL && i_size > 0 )
858 memcpy( p, *ppsz_parser,
859 __MIN((int)(strlen(*ppsz_parser) + 1), i_size) );
860 p += strlen(*ppsz_parser) + 1;
861 i_size -= strlen(*ppsz_parser) + 1;
866 /* Create the child process. */
867 bFuncRetn = CreateProcess( NULL,
868 psz_cmd, // command line
869 NULL, // process security attributes
870 NULL, // primary thread security attributes
871 TRUE, // handles are inherited
875 &siStartInfo, // STARTUPINFO pointer
876 &piProcInfo ); // receives PROCESS_INFORMATION
881 if ( bFuncRetn == 0 )
883 msg_Err( p_object, "child creation failed" );
887 /* Read from a file and write its contents to a pipe. */
888 while ( i_in > 0 && !p_object->b_die )
891 if ( !WriteFile( hChildStdinWr, p_in, i_in, &i_written, NULL ) )
897 /* Close the pipe handle so the child process stops reading. */
898 CloseHandle(hChildStdinWr);
900 /* Close the write end of the pipe before reading from the
901 * read end of the pipe. */
902 CloseHandle(hChildStdoutWr);
904 /* Read output from the child process. */
909 *pp_data = malloc( 1025 ); /* +1 for \0 */
911 while ( !p_object->b_die )
914 if ( !ReadFile( hChildStdoutRd, &(*pp_data)[*pi_data], 1024, &i_read,
919 *pp_data = realloc( *pp_data, *pi_data + 1025 );
922 while ( !p_object->b_die
923 && !GetExitCodeProcess( piProcInfo.hProcess, &i_status )
924 && i_status != STILL_ACTIVE )
927 CloseHandle(piProcInfo.hProcess);
928 CloseHandle(piProcInfo.hThread);
932 "child %s returned with error code %ld",
933 ppsz_argv[0], i_status );
936 msg_Err( p_object, "vlc_execve called but no implementation is available" );
941 if (*pp_data == NULL)
944 (*pp_data)[*pi_data] = '\0';