]> git.sesse.net Git - vlc/blobdiff - src/input/subtitles.c
* subtitles.c: case insensitive search for subtitle file extensions.
[vlc] / src / input / subtitles.c
index 4aee0ae05b2762024b7edeb88b3155cfbff3e8bd..dbd54788693eab608ae23733446f4f8dbbe8ea33 100644 (file)
@@ -1,7 +1,7 @@
 /*****************************************************************************
  * subtitles.c
  *****************************************************************************
- * Copyright (C) 2003-2004 the VideoLAN team
+ * Copyright (C) 2003-2006 the VideoLAN team
  * $Id$
  *
  * Authors: Derk-Jan Hartman <hartman at videolan.org>
 
 #include <stdlib.h>
 #include <vlc/vlc.h>
-#include <vlc/input.h>
-#include "charset.h"
+#include <vlc_input.h>
+#include <vlc_charset.h>
 
 #ifdef HAVE_DIRENT_H
 #   include <dirent.h>
 #endif
 
-#ifdef HAVE_LIMITS_H  
-#   include <limits.h>  
+#ifdef HAVE_LIMITS_H
+#   include <limits.h>
 #endif
 
 #ifdef HAVE_UNISTD_H
 #   include <unistd.h>
 #endif
+#include <sys/stat.h>
 
 #include <ctype.h>
-
-/**
- * What's between a directory and a filename?
- */
-#if defined( WIN32 )
-    #define DIRECTORY_SEPARATOR '\\'
-#else
-    #define DIRECTORY_SEPARATOR '/'
-#endif
+#include "input_internal.h"
 
 /**
  * We are not going to autodetect more subtitle files than this.
@@ -62,9 +55,9 @@
 
 
 /**
- * The possible extentions for subtitle files we support
+ * The possible extensions for subtitle files we support
  */
-static const char * sub_exts[] = {  "utf", "utf8", "utf-8", "sub", "srt", "smi", "txt", "ssa", "idx", NULL};
+static const char * sub_exts[] = {  "utf", "utf8", "utf-8", "sub", "srt", "smi", "txt", "ssa", "idx", NULL };
 /* extensions from unsupported types */
 /* rt, aqt, jss, js, ass */
 
@@ -104,10 +97,7 @@ static void strcpy_strip_ext( char *d, char *s )
         return;
     }
     else
-    {
-        strncpy(d, s, tmp - s);
-        d[tmp - s] = 0;
-    }
+        strlcpy(d, s, tmp - s + 1 );
     while( *d )
     {
         *d = tolower(*d);
@@ -161,37 +151,31 @@ static int compare_sub_priority( const void *a, const void *b )
 #endif
 }
 
-/* Utility function for scandir */  
-static int Filter( const struct dirent *p_dir_content )
+/*
+ * Check if a file ends with a subtitle extension
+ */
+int subtitles_Filter( const char *psz_dir_content )
 {
-    int i;
-    char *tmp = NULL;
-
-    if( p_dir_content == NULL || p_dir_content->d_name == NULL ) return VLC_FALSE;
-    /* does it end with a subtitle extension? */
-    tmp = strrchr( p_dir_content->d_name, '.');
-    if( !tmp )
-    {
-        return VLC_FALSE;
-    }
+    const char *tmp = strrchr( psz_dir_content, '.');
+    if( tmp == NULL )
+        return 0;
     else
     {
+        int i;
+        tmp++;
+
         for( i = 0; sub_exts[i]; i++ )
-        {
-            if( strcmp( sub_exts[i], tmp+1 ) == 0 )
-            {
-                return VLC_TRUE;
-            }
-        }
+            if( strcasecmp( sub_exts[i], tmp ) == 0 )
+                return 1;
     }
-    return VLC_FALSE;
+    return 0;
 }
 
 
 /**
  * Convert a list of paths separated by ',' to a char**
  */
-static char **paths_to_list( char *psz_dir, char *psz_path )
+static char **paths_to_list( const char *psz_dir, char *psz_path )
 {
     unsigned int i, k, i_nb_subdirs;
     char **subdirs; /* list of subdirectories to look in */
@@ -236,11 +220,11 @@ static char **paths_to_list( char *psz_dir, char *psz_path )
                                            + strlen(psz_subdir) + 2 );
                 if( psz_temp )
                 {
-                    sprintf( psz_temp, "%s%s%c", 
+                    sprintf( psz_temp, "%s%s%c",
                              psz_subdir[0] == '.' ? psz_dir : "",
                              psz_subdir,
                              psz_subdir[strlen(psz_subdir) - 1] ==
-                              DIRECTORY_SEPARATOR ? '\0' : DIRECTORY_SEPARATOR );
+                              DIR_SEP_CHAR ? '\0' : DIR_SEP_CHAR );
                     subdirs[i] = psz_temp;
                     i++;
                 }
@@ -272,18 +256,13 @@ static char **paths_to_list( char *psz_dir, char *psz_path )
  * The array contains max MAX_SUBTITLE_FILES items and you need to free it after use.
  */
 char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
-                         char *psz_name )
+                         const char *psz_name )
 {
     vlc_value_t fuzzy;
-    int j, i_result2, i_dir_content, i_sub_count = 0, i_fname_len = 0;
+    int j, i_result2, i_sub_count = 0, i_fname_len = 0;
     char *f_dir = NULL, *f_fname = NULL, *f_fname_noext = NULL, *f_fname_trim = NULL;
     char *tmp = NULL;
 
-    char tmp_fname_noext[PATH_MAX];
-    char tmp_fname_trim[PATH_MAX];
-    char tmp_fname_ext[PATH_MAX];
-
-    struct dirent **pp_dir_content;
     char **tmp_subdirs, **subdirs; /* list of subdirectories to look in */
 
     subfn *result = NULL; /* unsorted results */
@@ -300,21 +279,18 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
     }
 
     /* extract filename & dirname from psz_fname */
-    tmp = strrchr( psz_fname, DIRECTORY_SEPARATOR );
+    tmp = strrchr( psz_fname, DIR_SEP_CHAR );
     if( tmp )
     {
         int dirlen = 0;
 
         f_fname = malloc( strlen(tmp) );
         if( f_fname )
-            strcpy( f_fname, tmp+1 ); // we skip the seperator, so it will still fit in the allocated space
-        dirlen = strlen(psz_fname) - strlen(tmp) + 1; // add the seperator
+            strcpy( f_fname, tmp+1 ); // we skip the separator, so it will still fit in the allocated space
+        dirlen = strlen(psz_fname) - strlen(tmp) + 2; // add the separator
         f_dir = malloc( dirlen + 1 );
-        if( f_dir )
-        {
-            strncpy( f_dir, psz_fname, dirlen );
-            f_dir[dirlen] = 0;
-        }
+        if( f_dir != NULL )
+            strlcpy( f_dir, psz_fname, dirlen );
     }
     else
     {
@@ -330,7 +306,7 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
         }
         dirlen = strlen( f_dir );
         f_dir = (char *)realloc(f_dir, dirlen +2 );
-        f_dir[dirlen] = DIRECTORY_SEPARATOR;
+        f_dir[dirlen] = DIR_SEP_CHAR;
         f_dir[dirlen+1] = '\0';
         f_fname = strdup( psz_fname );
     }
@@ -351,39 +327,40 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
     tmp_subdirs = paths_to_list( f_dir, psz_path );
     subdirs = tmp_subdirs;
 
-    for( j = -1; (j == -1) || ( (j >= 0) && (subdirs != NULL) && (*subdirs != NULL) );
-         j++)
+    for( j = -1; (j == -1) || ( (j >= 0) && (subdirs != NULL) &&
+        (*subdirs != NULL) ); j++)
     {
-        char *psz_locale_dir;
+        const char *psz_dir = j < 0 ? f_dir : *subdirs;
+        char **ppsz_dir_content;
+        int i_dir_content;
 
-        pp_dir_content = NULL;
-        i_dir_content = 0;
-
-        if( j < 0 && f_dir == NULL )
+        if( psz_dir == NULL )
             continue;
 
         /* parse psz_src dir */
-        psz_locale_dir = ToLocale( j < 0 ? f_dir : *subdirs );
-        i_dir_content = scandir( psz_locale_dir, &pp_dir_content, Filter, NULL );
-        LocaleFree( psz_locale_dir );
+        i_dir_content = utf8_scandir( psz_dir, &ppsz_dir_content,
+                                      subtitles_Filter, NULL );
 
         if( i_dir_content != -1 )
         {
             int a;
 
-            msg_Dbg( p_this, "looking for a subtitle file in %s", j < 0 ? f_dir : *subdirs );
+            msg_Dbg( p_this, "looking for a subtitle file in %s", psz_dir );
             for( a = 0; a < i_dir_content; a++ )
             {
+                char *psz_name = ppsz_dir_content[a];
+                char tmp_fname_noext[strlen( psz_name ) + 1];
+                char tmp_fname_trim[strlen( psz_name ) + 1];
+                char tmp_fname_ext[strlen( psz_name ) + 1];
+
                 int i_prio = 0;
-                struct dirent *p_dir_content = pp_dir_content[a];
-                char *psz_inUTF8 = FromLocale( p_dir_content->d_name );
-                char *p_fixed_name = vlc_fix_readdir_charset( p_this, psz_inUTF8 );
 
-                LocaleFree( psz_inUTF8 );
+                if( psz_name == NULL )
+                    continue;
 
                 /* retrieve various parts of the filename */
-                strcpy_strip_ext( tmp_fname_noext, p_fixed_name );
-                strcpy_get_ext( tmp_fname_ext, p_fixed_name );
+                strcpy_strip_ext( tmp_fname_noext, psz_name );
+                strcpy_get_ext( tmp_fname_ext, psz_name );
                 strcpy_trim( tmp_fname_trim, tmp_fname_noext );
 
                 if( !i_prio && !strcmp( tmp_fname_trim, f_fname_trim ) )
@@ -415,34 +392,34 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
                 }
                 if( i_prio >= fuzzy.i_int )
                 {
-                    FILE *f;
-                    char *psz_path;
+                    char psz_path[strlen( psz_dir ) + strlen( psz_name ) + 1];
+                    struct stat st;
+
+                    sprintf( psz_path, "%s%s", psz_dir, psz_name );
+                    msg_Dbg( p_this,
+                                "autodetected subtitle: %s with priority %d",
+                                psz_path, i_prio );
 
-                    asprintf( &psz_path, "%s%s", j < 0 ? f_dir : *subdirs, p_fixed_name );
-                    msg_Dbg( p_this, "autodetected subtitle: %s with priority %d", p_fixed_name, i_prio );
-                   /* FIXME: a portable wrapper for stat() or access() would be more suited */
-                    if( ( f = vlc_fopen( psz_path, "rt" ) ) )
+                    if( !utf8_stat( psz_path, &st ) && S_ISREG( st.st_mode ) )
                     {
-                        fclose( f );
-                    msg_Dbg( p_this, "autodetected subtitle: %s with priority %d", p_fixed_name, i_prio );
+                        msg_Dbg( p_this,
+                                "autodetected subtitle: %s with priority %d",
+                                psz_path, i_prio );
                         result[i_sub_count].priority = i_prio;
-                        result[i_sub_count].psz_fname = psz_UTF8_path;
+                        result[i_sub_count].psz_fname = strdup( psz_path );
                         result[i_sub_count].psz_ext = strdup(tmp_fname_ext);
                         i_sub_count++;
                     }
                     else
                     {
-                        msg_Dbg( p_this, "fopen failed" );
-                        if( psz_UTF8_path ) free( psz_UTF8_path );
-                        LocaleFree( psz_locale_path );
+                        msg_Dbg( p_this, "stat failed" );
                     }
                 }
                 if( i_sub_count >= MAX_SUBTITLE_FILES ) break;
-                if( p_fixed_name ) free( p_fixed_name );
             }
             for( a = 0; a < i_dir_content; a++ )
-                if( pp_dir_content[a] ) free( pp_dir_content[a] );
-            if( pp_dir_content ) free( pp_dir_content );
+                free( ppsz_dir_content[a] );
+            if( ppsz_dir_content ) free( ppsz_dir_content );
         }
         if( j >= 0 ) if( *subdirs ) free( *subdirs++ );
     }
@@ -468,7 +445,8 @@ char **subtitles_Detect( input_thread_t *p_this, char *psz_path,
             for( i = 0; i < i_sub_count; i++ )
             {
                 if( result[i].psz_fname && result[j].psz_fname &&
-                    !strncasecmp( result[j].psz_fname, result[i].psz_fname, sizeof( result[j].psz_fname) - 4 ) && 
+                    !strncasecmp( result[j].psz_fname, result[i].psz_fname,
+                                sizeof( result[j].psz_fname) - 4 ) &&
                     !strcasecmp( result[i].psz_ext, "idx" ) )
                     break;
             }